inject go module variable for go template

tools/goctl/api/gogen/genconfig.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/vars"
 )
@@ -48,6 +49,11 @@ func genConfig(dir string, cfg *config.Config, api *spec.ApiSpec) error {
 	}
 	authImportStr := fmt.Sprintf("\"%s/rest\"", vars.ProjectOpenSourceURL)
 
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
+
 	return genFile(fileGenConfig{
 		dir:             dir,
 		subdir:          configDir,
@@ -60,6 +66,7 @@ func genConfig(dir string, cfg *config.Config, api *spec.ApiSpec) error {
 			"authImport": authImportStr,
 			"auth":       strings.Join(auths, "\n"),
 			"jwtTrans":   strings.Join(jwtTransList, "\n"),
+			"goModule":   projectCtx.Path,
 		},
 	})
 }

tools/goctl/api/gogen/genhandlers.go

@@ -9,6 +9,7 @@ import (
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
 	"github.com/zeromicro/go-zero/tools/goctl/util"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
 )
@@ -29,6 +30,7 @@ type handlerInfo struct {
 	Call               string
 	HasResp            bool
 	HasRequest         bool
+	GoModule           string
 }
 
 func genHandler(dir, rootPkg string, cfg *config.Config, group spec.Group, route spec.Route) error {
@@ -41,6 +43,10 @@ func genHandler(dir, rootPkg string, cfg *config.Config, group spec.Group, route
 		logicName = pkgName
 	}
 
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
 	return doGenToFile(dir, handler, cfg, group, route, handlerInfo{
 		PkgName:        pkgName,
 		ImportPackages: genHandlerImports(group, route, rootPkg),
@@ -51,6 +57,7 @@ func genHandler(dir, rootPkg string, cfg *config.Config, group spec.Group, route
 		Call:           strings.Title(strings.TrimSuffix(handler, "Handler")),
 		HasResp:        len(route.ResponseTypeName()) > 0,
 		HasRequest:     len(route.RequestTypeName()) > 0,
+		GoModule:       projectCtx.Path,
 	})
 }
 

tools/goctl/api/gogen/genlogic.go

@@ -10,6 +10,7 @@ import (
 	"github.com/zeromicro/go-zero/tools/goctl/api/parser/g4/gen/api"
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
 	"github.com/zeromicro/go-zero/tools/goctl/vars"
@@ -54,6 +55,11 @@ func genLogicByRoute(dir, rootPkg string, cfg *config.Config, group spec.Group,
 	}
 
 	subDir := getLogicFolderPath(group, route)
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
+
 	return genFile(fileGenConfig{
 		dir:             dir,
 		subdir:          subDir,
@@ -70,6 +76,7 @@ func genLogicByRoute(dir, rootPkg string, cfg *config.Config, group spec.Group,
 			"responseType": responseString,
 			"returnString": returnString,
 			"request":      requestString,
+			"goModule":     projectCtx.Path,
 		},
 	})
 }

tools/goctl/api/gogen/genmain.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
 	"github.com/zeromicro/go-zero/tools/goctl/vars"
@@ -27,6 +28,11 @@ func genMain(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpec) error {
 		filename = strings.ReplaceAll(filename, "-api", "")
 	}
 
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
+
 	return genFile(fileGenConfig{
 		dir:             dir,
 		subdir:          "",
@@ -38,6 +44,7 @@ func genMain(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpec) error {
 		data: map[string]string{
 			"importPackages": genMainImports(rootPkg),
 			"serviceName":    configName,
+			"goModule":       projectCtx.Path,
 		},
 	})
 }

tools/goctl/api/gogen/genmiddleware.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 )
 
@@ -14,6 +15,11 @@ var middlewareImplementCode string
 
 func genMiddleware(dir string, cfg *config.Config, api *spec.ApiSpec) error {
 	middlewares := getMiddleware(api)
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
+
 	for _, item := range middlewares {
 		middlewareFilename := strings.TrimSuffix(strings.ToLower(item), "middleware") + "_middleware"
 		filename, err := format.FileNamingFormat(cfg.NamingFormat, middlewareFilename)
@@ -31,7 +37,8 @@ func genMiddleware(dir string, cfg *config.Config, api *spec.ApiSpec) error {
 			templateFile:    middlewareImplementCodeFile,
 			builtinTemplate: middlewareImplementCode,
 			data: map[string]string{
-				"name": strings.Title(name),
+				"name":     strings.Title(name),
+				"goModule": projectCtx.Path,
 			},
 		})
 		if err != nil {

tools/goctl/api/gogen/genroutes.go

@@ -13,6 +13,7 @@ import (
 	"github.com/zeromicro/go-zero/core/collection"
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
 	"github.com/zeromicro/go-zero/tools/goctl/vars"
@@ -158,13 +159,19 @@ rest.WithPrefix("%s"),`, g.prefix)
 			routes = strings.TrimSpace(gbuilder.String())
 		}
 
+		projectCtx, err := ctx.Prepare(dir)
+		if err != nil {
+			return err
+		}
+
 		if err := gt.Execute(&builder, map[string]string{
 			"routes":    routes,
 			"jwt":       jwt,
 			"signature": signature,
 			"prefix":    prefix,
 			"timeout":   timeout,
 			"maxBytes":  maxBytes,
+			"goModule":  projectCtx.Path,
 		}); err != nil {
 			return err
 		}

tools/goctl/api/gogen/gensvc.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
 	"github.com/zeromicro/go-zero/tools/goctl/vars"
@@ -40,6 +41,10 @@ func genServiceContext(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpe
 		configImport += fmt.Sprintf("\n\t\"%s/rest\"", vars.ProjectOpenSourceURL)
 	}
 
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
 	return genFile(fileGenConfig{
 		dir:             dir,
 		subdir:          contextDir,
@@ -53,6 +58,7 @@ func genServiceContext(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpe
 			"config":               "config.Config",
 			"middleware":           middlewareStr,
 			"middlewareAssignment": middlewareAssignment,
+			"goModule":             projectCtx.Path,
 		},
 	})
 }

tools/goctl/api/gogen/gentypes.go

@@ -12,6 +12,7 @@ import (
 	apiutil "github.com/zeromicro/go-zero/tools/goctl/api/util"
 	"github.com/zeromicro/go-zero/tools/goctl/config"
 	"github.com/zeromicro/go-zero/tools/goctl/util"
+	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 )
 
@@ -53,6 +54,10 @@ func genTypes(dir string, cfg *config.Config, api *spec.ApiSpec) error {
 	filename := path.Join(dir, typesDir, typeFilename)
 	os.Remove(filename)
 
+	projectCtx, err := ctx.Prepare(dir)
+	if err != nil {
+		return err
+	}
 	return genFile(fileGenConfig{
 		dir:             dir,
 		subdir:          typesDir,
@@ -64,6 +69,7 @@ func genTypes(dir string, cfg *config.Config, api *spec.ApiSpec) error {
 		data: map[string]any{
 			"types":        val,
 			"containsTime": false,
+			"goModule":     projectCtx.Path,
 		},
 	})
 }

tools/goctl/rpc/generator/gencall.go

@@ -114,6 +114,7 @@ func (g *Generator) genCallGroup(ctx DirContext, proto parser.Proto, cfg *conf.C
 			"serviceName":    serviceName,
 			"functions":      strings.Join(functions, pathx.NL),
 			"interface":      strings.Join(iFunctions, pathx.NL),
+			"goModule":       ctx.GetModule(),
 		}, filename, true); err != nil {
 			return err
 		}

tools/goctl/rpc/generator/genlogic.go

@@ -65,6 +65,7 @@ func (g *Generator) genLogicInCompatibility(ctx DirContext, proto parser.Proto,
 			"functions":   functions,
 			"packageName": "logic",
 			"imports":     strings.Join(imports.KeysStr(), pathx.NL),
+			"goModule":    ctx.GetModule(),
 		}, filename, false)
 		if err != nil {
 			return err
@@ -119,6 +120,7 @@ func (g *Generator) genLogicGroup(ctx DirContext, proto parser.Proto, cfg *conf.
 				"functions":   functions,
 				"packageName": packageName,
 				"imports":     strings.Join(imports.KeysStr(), pathx.NL),
+				"goModule":    ctx.GetModule(),
 			}, filename, false); err != nil {
 				return err
 			}

tools/goctl/rpc/generator/genmain.go

@@ -78,5 +78,6 @@ func (g *Generator) GenMain(ctx DirContext, proto parser.Proto, cfg *conf.Config
 		"imports":      strings.Join(imports, pathx.NL),
 		"pkg":          proto.PbPackage,
 		"serviceNames": serviceNames,
+		"goModule":     ctx.GetModule(),
 	}, fileName, false)
 }

tools/goctl/rpc/generator/genserver.go

@@ -97,6 +97,7 @@ func (g *Generator) genServerGroup(ctx DirContext, proto parser.Proto, cfg *conf
 			"imports":   strings.Join(imports.KeysStr(), pathx.NL),
 			"funcs":     strings.Join(funcList, pathx.NL),
 			"notStream": notStream,
+			"goModule":  ctx.GetModule(),
 		}, serverFile, true); err != nil {
 			return err
 		}
@@ -148,6 +149,7 @@ func (g *Generator) genServerInCompatibility(ctx DirContext, proto parser.Proto,
 		"imports":   strings.Join(imports.KeysStr(), pathx.NL),
 		"funcs":     strings.Join(funcList, pathx.NL),
 		"notStream": notStream,
+		"goModule":  ctx.GetModule(),
 	}, serverFile, true)
 }
 

tools/goctl/rpc/generator/gensvc.go

@@ -31,6 +31,7 @@ func (g *Generator) GenSvc(ctx DirContext, _ parser.Proto, cfg *conf.Config) err
 	}
 
 	return util.With("svc").GoFmt(true).Parse(text).SaveTo(map[string]any{
-		"imports": fmt.Sprintf(`"%v"`, ctx.GetConfig().Package),
+		"imports":  fmt.Sprintf(`"%v"`, ctx.GetConfig().Package),
+		"goModule": ctx.GetModule(),
 	}, fileName, false)
 }

tools/goctl/rpc/generator/mkdir.go

@@ -1,13 +1,13 @@
 package generator
 
 import (
-	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"path/filepath"
 	"strings"
 
 	conf "github.com/zeromicro/go-zero/tools/goctl/config"
 	"github.com/zeromicro/go-zero/tools/goctl/rpc/parser"
 	"github.com/zeromicro/go-zero/tools/goctl/util/ctx"
+	"github.com/zeromicro/go-zero/tools/goctl/util/format"
 	"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
 	"github.com/zeromicro/go-zero/tools/goctl/util/stringx"
 )
@@ -40,6 +40,7 @@ type (
 		GetMain() Dir
 		GetServiceName() stringx.String
 		SetPbDir(pbDir, grpcDir string)
+		GetModule() string
 	}
 
 	// Dir defines a directory
@@ -266,6 +267,10 @@ func (d *defaultDirContext) GetServiceName() stringx.String {
 	return d.serviceName
 }
 
+func (d *defaultDirContext) GetModule() string {
+	return d.ctx.Path
+}
+
 // Valid returns true if the directory is valid
 func (d *Dir) Valid() bool {
 	return len(d.Filename) > 0 && len(d.Package) > 0