gen

func ApplyServiceFile ¶

func ApplyServiceFile(file *ast.File, action *dsl.Action, servicePkgName string) bool

ApplyServiceFile will apply the dsl.Action to the ast.File. It will modify the struct type and struct methods if Payload or Result is changed, and returns true. Otherwise returns false. The servicePkgName parameter specifies the expected package name for the service file. This should match the package name used in service registration to maintain consistency.

func ApplyServiceFileWithModelSync ¶ added in v0.10.0

func ApplyServiceFileWithModelSync(file *ast.File, action *dsl.Action, servicePkgName, correctModelImportPath, correctModelPkgName string) bool

ApplyServiceFileWithModelSync extends ApplyServiceFile to handle import path and package name updates. It will update import statements and package references when model packages are renamed.

Design Philosophy: This function uses AST manipulation instead of regenerating files to preserve user's code formatting and custom modifications. Different developers have different code formatting preferences, and we should not force our formatting on their existing code. We only update the necessary parts (imports and type references) while keeping everything else intact.

Example transformation when "model/oldpkg" is renamed to "model/newpkg": - Import statement: "myproject/model/oldpkg" -> "myproject/model/newpkg" - Type references: oldpkg.User -> newpkg.User, oldpkg.UserReq -> newpkg.UserReq, oldpkg.UserRsp -> newpkg.UserRsp

Parameters: - file: The AST file to process - action: The DSL action configuration - servicePkgName: The expected service package name - correctModelImportPath: The correct model import path (e.g., "myproject/model/auth") - correctModelPkgName: The correct model package name (e.g., "auth")

Returns true if any changes were made to the file.

func BuildMainFile ¶

func BuildMainFile(projectName string) (string, error)

BuildMainFile generates a main.go file, the content like below:

package main

import (

• "helloworld/configx" _ "helloworld/cronjob" _ "helloworld/middleware" _ "helloworld/model" _ "helloworld/module" "helloworld/router" _ "helloworld/service"

  "github.com/forbearing/gst/bootstrap" . "github.com/forbearing/gst/util"

)

func main() {
	RunOrDie(bootstrap.Bootstrap)
	RunOrDie(configx.Init)
	RunOrDie(cronjob.Init)
	RunOrDie(service.Init)
	RunOrDie(router.Init)
	RunOrDie(bootstrap.Run)
}

func BuildModelFile ¶

func BuildModelFile(pkgName string, modelImports []string, stmts ...ast.Stmt) (string, error)

BuildModelFile generates a model.go file, the content like below:

package model

import "github.com/forbearing/gst/model"

func init() {
	model.Register[*Group]()
	model.Register[*User]()
}

func BuildRouterFile ¶

func BuildRouterFile(pkgName string, modelImports []string, stmts ...ast.Stmt) (string, error)

BuildRouterFile generates a router.go file, the content like below:

package router

import (

"helloworld/model"

"github.com/forbearing/gst/router"

)

func Init() error {
	router.Register[*model.Group, *model.Group, *model.Group](router.Auth(), "group")
	router.Register[*model.User, *model.User, *model.User](router.Pub(), "user")
	return nil
}

FIXME: process imports automatically problem.

func BuildServiceFile ¶

func BuildServiceFile(pkgName string, modelImports []string, stmts ...ast.Stmt) (string, error)

BuildServiceFile generates a service.go file, the content like below:

package service

import "github.com/forbearing/gst/service"

func Init() error {
	service.Register[*group]()
	service.Register[*user]()
	return nil
}

FIXME: process imports automatically problem.

func EmptyLine ¶

func EmptyLine() *ast.EmptyStmt

func FormatNode ¶

func FormatNode(node ast.Node, processImport ...bool) (string, error)

FormatNode use go standard lib "go/format" to format ast.Node into code.

func FormatNodeExtra ¶

func FormatNodeExtra(node ast.Node, processImport ...bool) (string, error)

FormatNodeExtra use "https://github.com/mvdan/gofumpt" to format ast.Node into code.

func FormatNodeExtraWithFileSet ¶

func FormatNodeExtraWithFileSet(node ast.Node, fset *token.FileSet, processImport ...bool) (string, error)

FormatNodeExtraWithFileSet 使用指定的 FileSet 格式化节点，保持注释位置

func GenerateService ¶

func GenerateService(info *ModelInfo, action *dsl.Action, phase consts.Phase) *ast.File

func GetModulePath ¶

func GetModulePath() (string, error)

GetModulePath parses go.mod to get module path

func MethodAddComments ¶

func MethodAddComments(code string, modelName string) string

func ResolveImportConflicts ¶

func ResolveImportConflicts(imports []string) map[string]string

ResolveImportConflicts detects import conflicts and generates unique aliases Returns a map where key is the import path and value is the alias (empty string means no alias needed)

func Returns ¶

func Returns(exprs ...ast.Expr) *ast.ReturnStmt

func StmtLogInfo ¶

func StmtLogInfo(str string) *ast.ExprStmt

StmtLogInfo create *ast.ExprStmt represents `log.Info(str)`

func StmtLogWithServiceContext ¶

func StmtLogWithServiceContext(modelVarName string) *ast.AssignStmt

StmtLogWithServiceContext create *ast.AssignStmt represents `log := u.WithServiceContext(ctx, ctx.GetPhase())` modelVarName is model variable name.

func StmtModelRegister ¶

func StmtModelRegister(modelName string) *ast.ExprStmt

StmtModelRegister creates a *ast.ExprStmt represents golang code like below:

model.Register[*User]()

func StmtRouterRegister ¶

func StmtRouterRegister(modelPkgName, modelName, reqName, rspName string, router string, endpoint string, paramName string, verb string) *ast.ExprStmt

StmtRouterRegister creates a *ast.ExprStmt represents golang code like below:

router.Register[*model.Group, *model.Group, *model.Group](router.Auth(), "group", &types.ControllerConfig[*model.Group]{}, consts.Create)
router.Register[*model.Group, *model.Group, *model.Group](router.Pub(), "login", &types.ControllerConfig[*auth.LoginReq]{}, consts.Create)

func StmtServiceRegister ¶

func StmtServiceRegister(serviceImport string, phase consts.Phase) *ast.ExprStmt

StmtServiceRegister creates a *ast.ExprStmt represents golang code like below:

service.Register[*user.Creator](consts.PHASE_CREATE)