package main import ( "fmt" "os" "runtime" "github.com/tal-tech/go-zero/core/logx" "github.com/tal-tech/go-zero/tools/goctl/api/apigen" "github.com/tal-tech/go-zero/tools/goctl/api/dartgen" "github.com/tal-tech/go-zero/tools/goctl/api/docgen" "github.com/tal-tech/go-zero/tools/goctl/api/format" "github.com/tal-tech/go-zero/tools/goctl/api/gogen" "github.com/tal-tech/go-zero/tools/goctl/api/javagen" "github.com/tal-tech/go-zero/tools/goctl/api/ktgen" "github.com/tal-tech/go-zero/tools/goctl/api/new" "github.com/tal-tech/go-zero/tools/goctl/api/tsgen" "github.com/tal-tech/go-zero/tools/goctl/api/validate" "github.com/tal-tech/go-zero/tools/goctl/configgen" "github.com/tal-tech/go-zero/tools/goctl/docker" "github.com/tal-tech/go-zero/tools/goctl/kube" model "github.com/tal-tech/go-zero/tools/goctl/model/sql/command" "github.com/tal-tech/go-zero/tools/goctl/plugin" rpc "github.com/tal-tech/go-zero/tools/goctl/rpc/cli" "github.com/tal-tech/go-zero/tools/goctl/tpl" "github.com/tal-tech/go-zero/tools/goctl/upgrade" "github.com/urfave/cli" ) var ( buildVersion = "1.1.5" commands = []cli.Command{ { Name: "upgrade", Usage: "upgrade goctl to latest version", Action: upgrade.Upgrade, }, { Name: "api", Usage: "generate api related files", Flags: []cli.Flag{ cli.StringFlag{ Name: "o", Usage: "the output api file", }, }, Action: apigen.ApiCommand, Subcommands: []cli.Command{ { Name: "new", Usage: "fast create api service", Action: new.CreateServiceCommand, }, { Name: "format", Usage: "format api files", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the format target dir", }, cli.BoolFlag{ Name: "iu", Usage: "ignore update", }, cli.BoolFlag{ Name: "stdin", Usage: "use stdin to input api doc content, press \"ctrl + d\" to send EOF", }, }, Action: format.GoFormatApi, }, { Name: "validate", Usage: "validate api file", Flags: []cli.Flag{ cli.StringFlag{ Name: "api", Usage: "validate target api file", }, }, Action: validate.GoValidateApi, }, { Name: "doc", Usage: "generate doc files", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the target dir", }, cli.StringFlag{ Name: "o", Required: false, Usage: "the output markdown directory", }, }, Action: docgen.DocCommand, }, { Name: "go", Usage: "generate go files for provided api in yaml file", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the target dir", }, cli.StringFlag{ Name: "api", Usage: "the api file", }, cli.StringFlag{ Name: "style", Usage: "the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]", }, }, Action: gogen.GoCommand, }, { Name: "java", Usage: "generate java files for provided api in api file", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the target dir", }, cli.StringFlag{ Name: "api", Usage: "the api file", }, }, Action: javagen.JavaCommand, }, { Name: "ts", Usage: "generate ts files for provided api in api file", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the target dir", }, cli.StringFlag{ Name: "api", Usage: "the api file", }, cli.StringFlag{ Name: "webapi", Usage: "the web api file path", }, cli.StringFlag{ Name: "caller", Usage: "the web api caller", }, cli.BoolFlag{ Name: "unwrap", Usage: "unwrap the webapi caller for import", }, }, Action: tsgen.TsCommand, }, { Name: "dart", Usage: "generate dart files for provided api in api file", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the target dir", }, cli.StringFlag{ Name: "api", Usage: "the api file", }, }, Action: dartgen.DartCommand, }, { Name: "kt", Usage: "generate kotlin code for provided api file", Flags: []cli.Flag{ cli.StringFlag{ Name: "dir", Usage: "the target directory", }, cli.StringFlag{ Name: "api", Usage: "the api file", }, cli.StringFlag{ Name: "pkg", Usage: "define package name for kotlin file", }, }, Action: ktgen.KtCommand, }, { Name: "plugin", Usage: "custom file generator", Flags: []cli.Flag{ cli.StringFlag{ Name: "plugin, p", Usage: "the plugin file", }, cli.StringFlag{ Name: "dir", Usage: "the target directory", }, cli.StringFlag{ Name: "api", Usage: "the api file", }, cli.StringFlag{ Name: "style", Usage: "the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]", }, }, Action: plugin.PluginCommand, }, }, }, { Name: "docker", Usage: "generate Dockerfile", Flags: []cli.Flag{ cli.StringFlag{ Name: "go", Usage: "the file that contains main function", }, cli.IntFlag{ Name: "port", Usage: "the port to expose, default none", Value: 0, }, }, Action: docker.DockerCommand, }, { Name: "kube", Usage: "generate kubernetes files", Subcommands: []cli.Command{ { Name: "deploy", Usage: "generate deployment yaml file", Flags: []cli.Flag{ cli.StringFlag{ Name: "name", Usage: "the name of deployment", Required: true, }, cli.StringFlag{ Name: "namespace", Usage: "the namespace of deployment", Required: true, }, cli.StringFlag{ Name: "image", Usage: "the docker image of deployment", Required: true, }, cli.StringFlag{ Name: "secret", Usage: "the secret to image pull from registry", }, cli.IntFlag{ Name: "requestCpu", Usage: "the request cpu to deploy", Value: 500, }, cli.IntFlag{ Name: "requestMem", Usage: "the request memory to deploy", Value: 512, }, cli.IntFlag{ Name: "limitCpu", Usage: "the limit cpu to deploy", Value: 1000, }, cli.IntFlag{ Name: "limitMem", Usage: "the limit memory to deploy", Value: 1024, }, cli.StringFlag{ Name: "o", Usage: "the output yaml file", Required: true, }, cli.IntFlag{ Name: "replicas", Usage: "the number of replicas to deploy", Value: 3, }, cli.IntFlag{ Name: "revisions", Usage: "the number of revision history to limit", Value: 5, }, cli.IntFlag{ Name: "port", Usage: "the port of the deployment to listen on pod", Required: true, }, cli.IntFlag{ Name: "nodePort", Usage: "the nodePort of the deployment to expose", Value: 0, }, cli.IntFlag{ Name: "minReplicas", Usage: "the min replicas to deploy", Value: 3, }, cli.IntFlag{ Name: "maxReplicas", Usage: "the max replicas of deploy", Value: 10, }, }, Action: kube.DeploymentCommand, }, }, }, { Name: "rpc", Usage: "generate rpc code", Subcommands: []cli.Command{ { Name: "new", Usage: `generate rpc demo service`, Flags: []cli.Flag{ cli.StringFlag{ Name: "style", Usage: "the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]", }, cli.BoolFlag{ Name: "idea", Usage: "whether the command execution environment is from idea plugin. [optional]", }, }, Action: rpc.RPCNew, }, { Name: "template", Usage: `generate proto template`, Flags: []cli.Flag{ cli.StringFlag{ Name: "out, o", Usage: "the target path of proto", }, }, Action: rpc.RPCTemplate, }, { Name: "proto", Usage: `generate rpc from proto`, Flags: []cli.Flag{ cli.StringFlag{ Name: "src, s", Usage: "the file path of the proto source file", }, cli.StringSliceFlag{ Name: "proto_path, I", Usage: `native command of protoc, specify the directory in which to search for imports. [optional]`, }, cli.StringFlag{ Name: "dir, d", Usage: `the target path of the code`, }, cli.StringFlag{ Name: "style", Usage: "the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]", }, cli.BoolFlag{ Name: "idea", Usage: "whether the command execution environment is from idea plugin. [optional]", }, }, Action: rpc.RPC, }, }, }, { Name: "model", Usage: "generate model code", Subcommands: []cli.Command{ { Name: "mysql", Usage: `generate mysql model`, Subcommands: []cli.Command{ { Name: "ddl", Usage: `generate mysql model from ddl`, Flags: []cli.Flag{ cli.StringFlag{ Name: "src, s", Usage: "the path or path globbing patterns of the ddl", }, cli.StringFlag{ Name: "dir, d", Usage: "the target dir", }, cli.StringFlag{ Name: "style", Usage: "the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]", }, cli.BoolFlag{ Name: "cache, c", Usage: "generate code with cache [optional]", }, cli.BoolFlag{ Name: "idea", Usage: "for idea plugin [optional]", }, }, Action: model.MysqlDDL, }, { Name: "datasource", Usage: `generate model from datasource`, Flags: []cli.Flag{ cli.StringFlag{ Name: "url", Usage: `the data source of database,like "root:password@tcp(127.0.0.1:3306)/database`, }, cli.StringFlag{ Name: "table, t", Usage: `the table or table globbing patterns in the database`, }, cli.BoolFlag{ Name: "cache, c", Usage: "generate code with cache [optional]", }, cli.StringFlag{ Name: "dir, d", Usage: "the target dir", }, cli.StringFlag{ Name: "style", Usage: "the file naming format, see [https://github.com/tal-tech/go-zero/tree/master/tools/goctl/config/readme.md]", }, cli.BoolFlag{ Name: "idea", Usage: "for idea plugin [optional]", }, }, Action: model.MyDataSource, }, }, }, }, }, { Name: "config", Usage: "generate config json", Flags: []cli.Flag{ cli.StringFlag{ Name: "path, p", Usage: "the target config go file", }, }, Action: configgen.GenConfigCommand, }, { Name: "template", Usage: "template operation", Subcommands: []cli.Command{ { Name: "init", Usage: "initialize the all templates(force update)", Action: tpl.GenTemplates, }, { Name: "clean", Usage: "clean the all cache templates", Action: tpl.CleanTemplates, }, { Name: "update", Usage: "update template of the target category to the latest", Flags: []cli.Flag{ cli.StringFlag{ Name: "category,c", Usage: "the category of template, enum [api,rpc,model,docker,kube]", }, }, Action: tpl.UpdateTemplates, }, { Name: "revert", Usage: "revert the target template to the latest", Flags: []cli.Flag{ cli.StringFlag{ Name: "category,c", Usage: "the category of template, enum [api,rpc,model,docker,kube]", }, cli.StringFlag{ Name: "name,n", Usage: "the target file name of template", }, }, Action: tpl.RevertTemplates, }, }, }, } ) func main() { logx.Disable() app := cli.NewApp() app.Usage = "a cli tool to generate code" app.Version = fmt.Sprintf("%s %s/%s", buildVersion, runtime.GOOS, runtime.GOARCH) app.Commands = commands // cli already print error messages if err := app.Run(os.Args); err != nil { fmt.Println("error:", err) } }