rpc service generation (#26)

* add execute files * add protoc-osx * add rpc generation * add rpc generation * add: rpc template generation * update usage * fixed env prepare for project in go path * optimize gomod cache * add README.md * format error * reactor templatex.go * remove waste code
2020-08-28 19:24:58 +08:00
parent 71bbf91a63
commit db16115037
36 changed files with 2021 additions and 79 deletions
--- a/tools/goctl/rpc/ctx/ctx.go
+++ b/tools/goctl/rpc/ctx/ctx.go
@@ -0,0 +1,111 @@
+package ctx
+
+import (
+	"fmt"
+	"path/filepath"
+	"runtime"
+	"strings"
+
+	"github.com/urfave/cli"
+
+	"github.com/tal-tech/go-zero/core/logx"
+	"github.com/tal-tech/go-zero/tools/goctl/util"
+	"github.com/tal-tech/go-zero/tools/goctl/util/console"
+	"github.com/tal-tech/go-zero/tools/goctl/util/stringx"
+)
+
+const (
+	flagSrc     = "src"
+	flagDir     = "dir"
+	flagShared  = "shared"
+	flagService = "service"
+	flagIdea    = "idea"
+)
+
+type (
+	RpcContext struct {
+		ProjectPath  string
+		ProjectName  stringx.String
+		ServiceName  stringx.String
+		CurrentPath  string
+		Module       string
+		ProtoFileSrc string
+		ProtoSource  string
+		TargetDir    string
+		SharedDir    string
+		GoPath       string
+		console.Console
+	}
+)
+
+func MustCreateRpcContext(protoSrc, targetDir, sharedDir, serviceName string, idea bool) *RpcContext {
+	log := console.NewConsole(idea)
+	info, err := prepare(log)
+	log.Must(err)
+
+	if stringx.From(protoSrc).IsEmptyOrSpace() {
+		log.Fatalln("expected proto source, but nothing found")
+	}
+	srcFp, err := filepath.Abs(protoSrc)
+	log.Must(err)
+
+	if !util.FileExists(srcFp) {
+		log.Fatalln("%s is not exists", srcFp)
+	}
+	current := filepath.Dir(srcFp)
+	if stringx.From(targetDir).IsEmptyOrSpace() {
+		targetDir = current
+	}
+	if stringx.From(sharedDir).IsEmptyOrSpace() {
+		sharedDir = filepath.Join(current, "shared")
+	}
+	targetDirFp, err := filepath.Abs(targetDir)
+	log.Must(err)
+
+	sharedFp, err := filepath.Abs(sharedDir)
+	log.Must(err)
+
+	if stringx.From(serviceName).IsEmptyOrSpace() {
+		serviceName = getServiceFromRpcStructure(targetDirFp)
+	}
+	serviceNameString := stringx.From(serviceName)
+	if serviceNameString.IsEmptyOrSpace() {
+		log.Fatalln("service name is not found")
+	}
+
+	return &RpcContext{
+		ProjectPath:  info.Path,
+		ProjectName:  stringx.From(info.Name),
+		ServiceName:  serviceNameString,
+		CurrentPath:  current,
+		Module:       info.GoMod.Module,
+		ProtoFileSrc: srcFp,
+		ProtoSource:  filepath.Base(srcFp),
+		TargetDir:    targetDirFp,
+		SharedDir:    sharedFp,
+		GoPath:       info.GoPath,
+		Console:      log,
+	}
+}
+func MustCreateRpcContextFromCli(ctx *cli.Context) *RpcContext {
+	os := runtime.GOOS
+	switch os {
+	case "darwin":
+	case "windows":
+		logx.Must(fmt.Errorf("windows will support soon"))
+	default:
+		logx.Must(fmt.Errorf("unexpected os: %s", os))
+	}
+	protoSrc := ctx.String(flagSrc)
+	targetDir := ctx.String(flagDir)
+	sharedDir := ctx.String(flagShared)
+	serviceName := ctx.String(flagService)
+	idea := ctx.Bool(flagIdea)
+	return MustCreateRpcContext(protoSrc, targetDir, sharedDir, serviceName, idea)
+}
+
+func getServiceFromRpcStructure(targetDir string) string {
+	targetDir = filepath.Clean(targetDir)
+	suffix := filepath.Join("cmd", "rpc")
+	return filepath.Base(strings.TrimSuffix(targetDir, suffix))
+}