goctl added

tools/goctl/api/gogen/genlogic.go

@@ -0,0 +1,130 @@
+package gogen
+
+import (
+	"bytes"
+	"fmt"
+	"path"
+	"strings"
+	"text/template"
+
+	"zero/tools/goctl/api/spec"
+	"zero/tools/goctl/api/util"
+)
+
+const logicTemplate = `package logic
+
+import (
+	{{.imports}}
+)
+
+type {{.logic}} struct {
+	ctx context.Context
+	logx.Logger
+}
+
+func New{{.logic}}(ctx context.Context, svcCtx *svc.ServiceContext) {{.logic}} {
+	return {{.logic}}{
+		ctx:    ctx,
+		Logger: logx.WithContext(ctx),
+	}
+	// TODO need set model here from svc
+}
+
+func (l *{{.logic}}) {{.function}}({{.request}}) {{.responseType}} {
+	{{.returnString}}
+}
+
+`
+
+func genLogic(dir string, api *spec.ApiSpec) error {
+	for _, g := range api.Service.Groups {
+		for _, r := range g.Routes {
+			err := genLogicByRoute(dir, g, r)
+			if err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}
+
+func genLogicByRoute(dir string, group spec.Group, route spec.Route) error {
+	handler, ok := util.GetAnnotationValue(route.Annotations, "server", "handler")
+	if !ok {
+		return fmt.Errorf("missing handler annotation for %q", route.Path)
+	}
+	handler = strings.TrimSuffix(handler, "handler")
+	handler = strings.TrimSuffix(handler, "Handler")
+	filename := strings.ToLower(handler)
+	goFile := filename + "logic.go"
+	fp, created, err := util.MaybeCreateFile(dir, getLogicFolderPath(group, route), goFile)
+	if err != nil {
+		return err
+	}
+	if !created {
+		return nil
+	}
+	defer fp.Close()
+
+	parentPkg, err := getParentPackage(dir)
+	if err != nil {
+		return err
+	}
+	imports := genLogicImports(route, parentPkg)
+
+	responseString := ""
+	returnString := ""
+	requestString := ""
+	if len(route.ResponseType.Name) > 0 {
+		responseString = "(*types." + strings.Title(route.ResponseType.Name) + ", error)"
+		returnString = "return nil, nil"
+	} else {
+		responseString = "error"
+		returnString = "return nil"
+	}
+	if len(route.RequestType.Name) > 0 {
+		requestString = "req " + "types." + strings.Title(route.RequestType.Name)
+	}
+
+	t := template.Must(template.New("logicTemplate").Parse(logicTemplate))
+	buffer := new(bytes.Buffer)
+	err = t.Execute(fp, map[string]string{
+		"imports":      imports,
+		"logic":        strings.Title(handler) + "Logic",
+		"function":     strings.Title(strings.TrimSuffix(handler, "Handler")),
+		"responseType": responseString,
+		"returnString": returnString,
+		"request":      requestString,
+	})
+	if err != nil {
+		return nil
+	}
+	formatCode := formatCode(buffer.String())
+	_, err = fp.WriteString(formatCode)
+	return err
+}
+
+func getLogicFolderPath(group spec.Group, route spec.Route) string {
+	folder, ok := util.GetAnnotationValue(route.Annotations, "server", folderProperty)
+	if !ok {
+		folder, ok = util.GetAnnotationValue(group.Annotations, "server", folderProperty)
+		if !ok {
+			return logicDir
+		}
+	}
+	folder = strings.TrimPrefix(folder, "/")
+	folder = strings.TrimSuffix(folder, "/")
+	return path.Join(logicDir, folder)
+}
+
+func genLogicImports(route spec.Route, parentPkg string) string {
+	var imports []string
+	imports = append(imports, `"context"`)
+	imports = append(imports, "\n")
+	imports = append(imports, `"zero/core/logx"`)
+	if len(route.ResponseType.Name) > 0 || len(route.RequestType.Name) > 0 {
+		imports = append(imports, fmt.Sprintf("\"%s\"", path.Join(parentPkg, typesDir)))
+	}
+	imports = append(imports, fmt.Sprintf("\"%s\"", path.Join(parentPkg, contextDir)))
+	return strings.Join(imports, "\n\t")
+}