You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
2.4 KiB
Go
111 lines
2.4 KiB
Go
package tpl
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/logrusorgru/aurora"
|
|
"github.com/tal-tech/go-zero/core/errorx"
|
|
"github.com/tal-tech/go-zero/tools/goctl/api/gogen"
|
|
"github.com/tal-tech/go-zero/tools/goctl/docker"
|
|
"github.com/tal-tech/go-zero/tools/goctl/kube"
|
|
modelgen "github.com/tal-tech/go-zero/tools/goctl/model/sql/gen"
|
|
rpcgen "github.com/tal-tech/go-zero/tools/goctl/rpc/generator"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
const templateParentPath = "/"
|
|
|
|
func GenTemplates(ctx *cli.Context) error {
|
|
if err := errorx.Chain(
|
|
func() error {
|
|
return gogen.GenTemplates(ctx)
|
|
},
|
|
func() error {
|
|
return modelgen.GenTemplates(ctx)
|
|
},
|
|
func() error {
|
|
return rpcgen.GenTemplates(ctx)
|
|
},
|
|
func() error {
|
|
return docker.GenTemplates(ctx)
|
|
},
|
|
func() error {
|
|
return kube.GenTemplates(ctx)
|
|
},
|
|
); err != nil {
|
|
return err
|
|
}
|
|
|
|
dir, err := util.GetTemplateDir(templateParentPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("Templates are generated in %s, %s\n", aurora.Green(dir),
|
|
aurora.Red("edit on your risk!"))
|
|
|
|
return nil
|
|
}
|
|
|
|
func CleanTemplates(_ *cli.Context) error {
|
|
err := errorx.Chain(
|
|
func() error {
|
|
return gogen.Clean()
|
|
},
|
|
func() error {
|
|
return modelgen.Clean()
|
|
},
|
|
func() error {
|
|
return rpcgen.Clean()
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("%s\n", aurora.Green("template are clean!"))
|
|
return nil
|
|
}
|
|
|
|
func UpdateTemplates(ctx *cli.Context) (err error) {
|
|
category := ctx.String("category")
|
|
defer func() {
|
|
if err == nil {
|
|
fmt.Println(aurora.Green(fmt.Sprintf("%s template are update!", category)).String())
|
|
}
|
|
}()
|
|
switch category {
|
|
case gogen.GetCategory():
|
|
return gogen.Update(category)
|
|
case rpcgen.GetCategory():
|
|
return rpcgen.Update(category)
|
|
case modelgen.GetCategory():
|
|
return modelgen.Update(category)
|
|
default:
|
|
err = fmt.Errorf("unexpected category: %s", category)
|
|
return
|
|
}
|
|
}
|
|
|
|
func RevertTemplates(ctx *cli.Context) (err error) {
|
|
category := ctx.String("category")
|
|
filename := ctx.String("name")
|
|
defer func() {
|
|
if err == nil {
|
|
fmt.Println(aurora.Green(fmt.Sprintf("%s template are reverted!", filename)).String())
|
|
}
|
|
}()
|
|
switch category {
|
|
case gogen.GetCategory():
|
|
return gogen.RevertTemplate(filename)
|
|
case rpcgen.GetCategory():
|
|
return rpcgen.RevertTemplate(filename)
|
|
case modelgen.GetCategory():
|
|
return modelgen.RevertTemplate(filename)
|
|
default:
|
|
err = fmt.Errorf("unexpected category: %s", category)
|
|
return
|
|
}
|
|
}
|