feat: new command pbf

coco/main.go

@@ -30,6 +30,7 @@ func init() {
 	rootCmd.AddCommand(bootstrap.AddServiceCommand())
 	rootCmd.AddCommand(bootstrap.GenerateProtoFile())
 	rootCmd.AddCommand(bootstrap.InjectProtoFile())
+	rootCmd.AddCommand(bootstrap.AddProtoFormat())
 }
 
 func main() {

coco_proto_format.go

@@ -0,0 +1,54 @@
+package bootstrap
+
+import (
+	"bytes"
+	"github.com/emicklei/proto"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"gitter.top/sync/proto-contrib/pkg/protofmt"
+	"os"
+)
+
+func AddProtoFormat() *cobra.Command {
+	var protoFile string
+	cmd := &cobra.Command{
+		Use:   "pbf",
+		Short: "protobuf format",
+		PreRun: func(_ *cobra.Command, args []string) {
+			if protoFile == "" && len(args) != 0 {
+				protoFile = args[0]
+			}
+			stat, err := os.Stat(protoFile)
+			if err != nil {
+				logrus.Fatalf("read proto file %s failed: %v", protoFile, err)
+				return
+			}
+			if stat.IsDir() {
+				logrus.Fatalf("proto file can not be directory")
+				return
+			}
+		},
+		Run: func(cmd *cobra.Command, args []string) {
+			file, err := os.Open(protoFile)
+			if err != nil {
+				logrus.Errorf("format proto file %s failed: %v", protoFile, err)
+				return
+			}
+			buf := new(bytes.Buffer)
+			parser := proto.NewParser(file)
+			parser.Filename(protoFile)
+			def, err := parser.Parse()
+			if err != nil {
+				logrus.Errorf("parse proto file %s failed: %v", protoFile, err)
+				return
+			}
+			protofmt.NewFormatter(buf, "    ").Format(def)
+			if err := os.WriteFile(protoFile, buf.Bytes(), os.ModePerm); err != nil {
+				logrus.Errorf("write proto file %s failed: %v", protoFile, err)
+				return
+			}
+		},
+	}
+	cmd.Flags().StringVar(&protoFile, "path", "", "proto file path")
+	return cmd
+}

create_api.go

@@ -55,11 +55,18 @@ func AddServiceCommand() *cobra.Command {
 				return
 			}
 
+			err := gobuf.CreateFile(pbPath)
+			if err != nil {
+				logrus.Errorf("read proto file failed (when create new file): %v", err)
+				return
+			}
+
 			buf, err := gobuf.NewParser(pbPath)
 			if err != nil {
 				logrus.Errorf("read proto file failed: %v", err)
 				return
 			}
+
 			if buf.ExistService(svcName) {
 				logrus.Errorf("router group name exist")
 				return

go.mod

@@ -6,7 +6,7 @@ require (
 	code.gitea.io/sdk/gitea v0.17.1
 	github.com/sirupsen/logrus v1.9.3
 	github.com/spf13/cobra v1.8.0
-	gitter.top/common/gobuf v0.0.2
+	gitter.top/common/gobuf v0.0.4
 	gopkg.in/yaml.v3 v3.0.1
 )
 

template/template_apiservices.go

@@ -10,7 +10,7 @@ option go_package = "{{.Name}}/gen;genv1";
 // @route_group: true
 // @base_url: /v1/file
 // @gen_to: ./services/controller/v1/file_controller.go
-// @rpc_to: ./services/rpc/v1/file_impl.go
+// @rpc_to: ./services/microservice/v1/file_service.go
 service FileService {
     // @desc: 列表
     // @author: Young Xu