feat(main): add --mode

Author: lvan100

gen/gen_test.go

@@ -34,6 +34,7 @@ func testProject(t *testing.T, dir string) {
 
 	var m map[string]struct {
 		Server bool
+		Client bool
 	}
 	if err = json.Unmarshal(b, &m); err != nil {
 		t.Fatal(err)
@@ -42,18 +43,16 @@ func testProject(t *testing.T, dir string) {
 	idlDir := filepath.Join(dir, "idl")
 	for lang, c := range m {
 		projectDir := filepath.Join(dir, lang)
-
-		if c.Server {
-			dstDir := filepath.Join(projectDir, "server")
-			copyFiles(t, idlDir, dstDir)
-			config := &generator.Config{
-				ProjectDir: dstDir,
-				Version:    "v0.0.0",
-				Server:     true,
-			}
-			if err = Gen(lang, config); err != nil {
-				t.Fatal(err)
-			}
+		dstDir := filepath.Join(projectDir, "server")
+		copyFiles(t, idlDir, dstDir)
+		config := &generator.Config{
+			ProjectDir: dstDir,
+			Version:    "v0.0.0",
+			Server:     c.Server,
+			Client:     c.Client,
+		}
+		if err = Gen(lang, config); err != nil {
+			t.Fatal(err)
 		}
 	}
 }

gen/generator/generator.go

@@ -11,6 +11,7 @@ type Config struct {
 	ProjectDir string
 	Version    string
 	Server     bool
+	Client     bool
 }
 
 var generators = map[string]Generator{}

gen/generator/golang/main.go

@@ -44,10 +44,13 @@ func (g *Generator) Gen(config *generator.Config, files map[string]parser.Docume
 		if err := g.genServer(ctx, rpcs); err != nil {
 			return err
 		}
-	} else {
+	}
+
+	if config.Client {
 		if err := g.genClient(ctx, rpcs); err != nil {
 			return err
 		}
 	}
+
 	return nil
 }

gen/generator/golang/type.go

@@ -14,8 +14,8 @@ const Version = "v0.0.1"
 func main() {
 	var (
 		version  bool
-		server   bool
 		language string
+		genMode  string
 	)
 
 	root := &cobra.Command{
@@ -25,8 +25,8 @@ func main() {
 	}
 
 	root.Flags().BoolVar(&version, "version", false, "show version")
-	root.Flags().BoolVar(&server, "server", false, "gen server code or not")
 	root.Flags().StringVar(&language, "lang", "go", "language, go/php/java")
+	root.Flags().StringVar(&genMode, "mode", "server", "server, client, type")
 
 	root.RunE = func(cmd *cobra.Command, args []string) error {
 		if version {
@@ -36,9 +36,15 @@ func main() {
 		}
 		config := &generator.Config{
 			ProjectDir: ".",
-			Server:     server,
 			Version:    Version,
 		}
+		switch genMode {
+		case "server":
+			config.Server = true
+		case "client":
+			config.Client = true
+		default:
+		}
 		return gen.Gen(language, config)
 	}