|
|
|
package gogen
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"fmt"
|
|
|
|
goformat "go/format"
|
|
|
|
"io"
|
|
|
|
"path/filepath"
|
|
|
|
"strings"
|
|
|
|
"text/template"
|
|
|
|
|
|
|
|
"github.com/tal-tech/go-zero/core/collection"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/api/spec"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/api/util"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/util/ctx"
|
|
|
|
ctlutil "github.com/tal-tech/go-zero/tools/goctl/util/pathx"
|
|
|
|
)
|
|
|
|
|
|
|
|
type fileGenConfig struct {
|
|
|
|
dir string
|
|
|
|
subdir string
|
|
|
|
filename string
|
|
|
|
templateName string
|
|
|
|
category string
|
|
|
|
templateFile string
|
|
|
|
builtinTemplate string
|
|
|
|
data interface{}
|
|
|
|
}
|
|
|
|
|
|
|
|
func genFile(c fileGenConfig) error {
|
|
|
|
fp, created, err := util.MaybeCreateFile(c.dir, c.subdir, c.filename)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if !created {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
defer fp.Close()
|
|
|
|
|
|
|
|
var text string
|
|
|
|
if len(c.category) == 0 || len(c.templateFile) == 0 {
|
|
|
|
text = c.builtinTemplate
|
|
|
|
} else {
|
|
|
|
text, err = ctlutil.LoadTemplate(c.category, c.templateFile, c.builtinTemplate)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
t := template.Must(template.New(c.templateName).Parse(text))
|
|
|
|
buffer := new(bytes.Buffer)
|
|
|
|
err = t.Execute(buffer, c.data)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
code := formatCode(buffer.String())
|
|
|
|
_, err = fp.WriteString(code)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func getParentPackage(dir string) (string, error) {
|
|
|
|
abs, err := filepath.Abs(dir)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
projectCtx, err := ctx.Prepare(abs)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
// fix https://github.com/zeromicro/go-zero/issues/1058
|
|
|
|
wd := projectCtx.WorkDir
|
|
|
|
d := projectCtx.Dir
|
|
|
|
same, err := ctlutil.SameFile(wd, d)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
trim := strings.TrimPrefix(projectCtx.WorkDir, projectCtx.Dir)
|
|
|
|
if same {
|
|
|
|
trim = strings.TrimPrefix(strings.ToLower(projectCtx.WorkDir), strings.ToLower(projectCtx.Dir))
|
|
|
|
}
|
|
|
|
|
|
|
|
return filepath.ToSlash(filepath.Join(projectCtx.Path, trim)), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func writeProperty(writer io.Writer, name, tag, comment string, tp spec.Type, indent int) error {
|
|
|
|
util.WriteIndent(writer, indent)
|
|
|
|
var err error
|
|
|
|
if len(comment) > 0 {
|
|
|
|
comment = strings.TrimPrefix(comment, "//")
|
|
|
|
comment = "//" + comment
|
|
|
|
_, err = fmt.Fprintf(writer, "%s %s %s %s\n", strings.Title(name), tp.Name(), tag, comment)
|
|
|
|
} else {
|
|
|
|
_, err = fmt.Fprintf(writer, "%s %s %s\n", strings.Title(name), tp.Name(), tag)
|
|
|
|
}
|
|
|
|
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func getAuths(api *spec.ApiSpec) []string {
|
|
|
|
authNames := collection.NewSet()
|
|
|
|
for _, g := range api.Service.Groups {
|
|
|
|
jwt := g.GetAnnotation("jwt")
|
|
|
|
if len(jwt) > 0 {
|
|
|
|
authNames.Add(jwt)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return authNames.KeysStr()
|
|
|
|
}
|
|
|
|
|
|
|
|
func getMiddleware(api *spec.ApiSpec) []string {
|
|
|
|
result := collection.NewSet()
|
|
|
|
for _, g := range api.Service.Groups {
|
|
|
|
middleware := g.GetAnnotation("middleware")
|
|
|
|
if len(middleware) > 0 {
|
|
|
|
for _, item := range strings.Split(middleware, ",") {
|
|
|
|
result.Add(strings.TrimSpace(item))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return result.KeysStr()
|
|
|
|
}
|
|
|
|
|
|
|
|
func formatCode(code string) string {
|
|
|
|
ret, err := goformat.Source([]byte(code))
|
|
|
|
if err != nil {
|
|
|
|
return code
|
|
|
|
}
|
|
|
|
|
|
|
|
return string(ret)
|
|
|
|
}
|
|
|
|
|
|
|
|
func responseGoTypeName(r spec.Route, pkg ...string) string {
|
|
|
|
if r.ResponseType == nil {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
|
|
|
resp := golangExpr(r.ResponseType, pkg...)
|
|
|
|
switch r.ResponseType.(type) {
|
|
|
|
case spec.DefineStruct:
|
|
|
|
if !strings.HasPrefix(resp, "*") {
|
|
|
|
return "*" + resp
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return resp
|
|
|
|
}
|
|
|
|
|
|
|
|
func requestGoTypeName(r spec.Route, pkg ...string) string {
|
|
|
|
if r.RequestType == nil {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
|
|
|
return golangExpr(r.RequestType, pkg...)
|
|
|
|
}
|
|
|
|
|
|
|
|
func golangExpr(ty spec.Type, pkg ...string) string {
|
|
|
|
switch v := ty.(type) {
|
|
|
|
case spec.PrimitiveType:
|
|
|
|
return v.RawName
|
|
|
|
case spec.DefineStruct:
|
|
|
|
if len(pkg) > 1 {
|
|
|
|
panic("package cannot be more than 1")
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(pkg) == 0 {
|
|
|
|
return v.RawName
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Sprintf("%s.%s", pkg[0], strings.Title(v.RawName))
|
|
|
|
case spec.ArrayType:
|
|
|
|
if len(pkg) > 1 {
|
|
|
|
panic("package cannot be more than 1")
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(pkg) == 0 {
|
|
|
|
return v.RawName
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Sprintf("[]%s", golangExpr(v.Value, pkg...))
|
|
|
|
case spec.MapType:
|
|
|
|
if len(pkg) > 1 {
|
|
|
|
panic("package cannot be more than 1")
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(pkg) == 0 {
|
|
|
|
return v.RawName
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Sprintf("map[%s]%s", v.Key, golangExpr(v.Value, pkg...))
|
|
|
|
case spec.PointerType:
|
|
|
|
if len(pkg) > 1 {
|
|
|
|
panic("package cannot be more than 1")
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(pkg) == 0 {
|
|
|
|
return v.RawName
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Sprintf("*%s", golangExpr(v.Type, pkg...))
|
|
|
|
case spec.InterfaceType:
|
|
|
|
return v.RawName
|
|
|
|
}
|
|
|
|
|
|
|
|
return ""
|
|
|
|
}
|