You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
go-zero/tools/goctl/api/dartgen/util.go

207 lines
4.3 KiB
Go

4 years ago
package dartgen
import (
"errors"
"fmt"
4 years ago
"os"
"path"
4 years ago
"strings"
"github.com/zeromicro/go-zero/tools/goctl/api/spec"
"github.com/zeromicro/go-zero/tools/goctl/api/util"
4 years ago
)
const (
formTagKey = "form"
pathTagKey = "path"
headerTagKey = "header"
)
func normalizeHandlerName(handlerName string) string {
handler := strings.Replace(handlerName, "Handler", "", 1)
handler = lowCamelCase(handler)
return handler
}
4 years ago
func lowCamelCase(s string) string {
if len(s) < 1 {
return ""
}
4 years ago
s = util.ToCamelCase(util.ToSnakeCase(s))
return util.ToLower(s[:1]) + s[1:]
}
func getBaseName(str string) string {
return path.Base(str)
}
func getPropertyFromMember(member spec.Member) string {
name, err := member.GetPropertyName()
if err != nil {
panic(fmt.Sprintf("cannot get property name of %q", member.Name))
}
return name
4 years ago
}
func isDirectType(s string) bool {
return isAtomicType(s) || isListType(s) && isAtomicType(getCoreType(s))
}
func isAtomicType(s string) bool {
switch s {
case "String", "int", "double", "bool":
return true
default:
return false
}
}
func isListType(s string) bool {
return strings.HasPrefix(s, "List<")
}
func isClassListType(s string) bool {
return strings.HasPrefix(s, "List<") && !isAtomicType(getCoreType(s))
}
func getCoreType(s string) string {
if isAtomicType(s) {
return s
}
if isListType(s) {
s = strings.Replace(s, "List<", "", -1)
return strings.Replace(s, ">", "", -1)
}
return s
}
func fileExists(path string) bool {
_, err := os.Stat(path)
return !os.IsNotExist(err)
}
func buildSpecType(tp spec.Type, name string) spec.Type {
switch v := tp.(type) {
case spec.PrimitiveType:
return spec.PrimitiveType{RawName: name}
case spec.MapType:
return spec.MapType{RawName: name, Key: v.Key, Value: v.Value}
case spec.ArrayType:
return spec.ArrayType{RawName: name, Value: v.Value}
case spec.InterfaceType:
return spec.InterfaceType{RawName: name}
case spec.PointerType:
return spec.PointerType{RawName: name, Type: v.Type}
}
return tp
}
func specTypeToDart(tp spec.Type) (string, error) {
switch v := tp.(type) {
case spec.DefineStruct:
return tp.Name(), nil
case spec.PrimitiveType:
r, ok := primitiveType(tp.Name())
if !ok {
return "", errors.New("unsupported primitive type " + tp.Name())
}
return r, nil
case spec.MapType:
valueType, err := specTypeToDart(v.Value)
if err != nil {
return "", err
}
return fmt.Sprintf("Map<String, %s>", valueType), nil
case spec.ArrayType:
if tp.Name() == "[]byte" {
return "List<int>", nil
}
valueType, err := specTypeToDart(v.Value)
if err != nil {
return "", err
}
s := getBaseType(valueType)
if len(s) != 0 {
return s, nil
}
return fmt.Sprintf("List<%s>", valueType), nil
case spec.InterfaceType:
return "Object", nil
case spec.PointerType:
return specTypeToDart(v.Type)
}
return "", errors.New("unsupported primitive type " + tp.Name())
}
func getBaseType(valueType string) string {
switch valueType {
case "int":
return "List<int>"
case "double":
return "List<double>"
case "boolean":
return "List<bool>"
case "String":
return "List<String>"
default:
return ""
}
}
func primitiveType(tp string) (string, bool) {
switch tp {
case "string":
return "String", true
case "int", "int8", "int16", "int32", "int64", "uint", "uint8", "uint16", "uint32", "uint64", "rune":
return "int", true
case "float32", "float64":
return "double", true
case "bool":
return "bool", true
}
return "", false
}
func hasUrlPathParams(route spec.Route) bool {
ds, ok := route.RequestType.(spec.DefineStruct)
if !ok {
return false
}
return len(route.RequestTypeName()) > 0 && len(ds.GetTagMembers(pathTagKey)) > 0
}
func extractPositionalParamsFromPath(route spec.Route) string {
ds, ok := route.RequestType.(spec.DefineStruct)
if !ok {
return ""
}
var params []string
for _, member := range ds.GetTagMembers(pathTagKey) {
params = append(params, fmt.Sprintf("%s %s", member.Type.Name(), getPropertyFromMember(member)))
}
return strings.Join(params, ", ")
}
func makeDartRequestUrlPath(route spec.Route) string {
path := route.Path
ds, ok := route.RequestType.(spec.DefineStruct)
if !ok {
return path
}
for _, member := range ds.GetTagMembers(pathTagKey) {
path = strings.ReplaceAll(path, ":"+pathTagKey, "${"+getPropertyFromMember(member)+"}")
}
return `"` + path + `"`
}