release: new version

README.md

@@ -0,0 +1,20 @@
+# protoc-gen-coco
+
+## install
+
+```shell
+go install gitter.top/coco/protoc-gen-coco@latest
+```
+
+## usage
+
+```shell
+ protoc example.proto --coco_out=. --go_out=.
+```
+
+## debug
+
+```shell
+go install .
+protoc --coco_out=. .\example_errcode.proto --go_out=.
+```

comment_parse.go

@@ -1,9 +1,10 @@
 package main
 
 import (
-	"google.golang.org/protobuf/compiler/protogen"
 	"regexp"
 	"strings"
+
+	"google.golang.org/protobuf/compiler/protogen"
 )
 
 func IsCommentRouterGroup(comments protogen.CommentSet) bool {
@@ -41,9 +42,26 @@ func GetCommentGenerateTo(comments protogen.CommentSet) string {
 		if len(match[0]) != 2 {
 			continue
 		}
-		return match[0][1]
+		return trim(match[0][1])
 	}
-	return "/"
+	return ""
+}
+
+func GetCommentRpcGenerateTo(comments protogen.CommentSet) string {
+	raw := comments.Leading.String()
+	split := strings.Split(raw, "\n")
+	re := regexp.MustCompile("@rpc_to:\\s*([\\w|/|\\.]*)")
+	for _, s := range split {
+		match := re.FindAllStringSubmatch(s, -1)
+		if len(match) == 0 {
+			continue
+		}
+		if len(match[0]) != 2 {
+			continue
+		}
+		return trim(match[0][1])
+	}
+	return ""
 }
 
 func GetCommentBaseURL(comments protogen.CommentSet) string {
@@ -58,7 +76,7 @@ func GetCommentBaseURL(comments protogen.CommentSet) string {
 		if len(match[0]) != 2 {
 			continue
 		}
-		return match[0][1]
+		return trim(match[0][1])
 	}
 	return "/"
 }
@@ -75,7 +93,7 @@ func GetCommentHttpMethod(comments protogen.CommentSet) string {
 		if len(match[0]) != 2 {
 			continue
 		}
-		return match[0][1]
+		return trim(match[0][1])
 	}
 	return "GET"
 }
@@ -92,7 +110,7 @@ func GetCommentAuthor(comments protogen.CommentSet) string {
 		if len(match[0]) != 2 {
 			continue
 		}
-		return match[0][1]
+		return trim(match[0][1])
 	}
 	return ""
 }
@@ -109,7 +127,7 @@ func GetCommentDescribe(comments protogen.CommentSet) string {
 		if len(match[0]) != 2 {
 			continue
 		}
-		return match[0][1]
+		return trim(match[0][1])
 	}
 	return ""
 }
@@ -130,3 +148,86 @@ func GetCommentApiURL(method *protogen.Method) string {
 	}
 	return "/" + CamelCaseToJavascriptCase(method.GoName)
 }
+
+func GetCommentTableName(message *protogen.Message) string {
+	raw := message.Comments.Leading.String()
+	split := strings.Split(raw, "\n")
+	re := regexp.MustCompile("@table_name:\\s*(.*)")
+	for _, s := range split {
+		match := re.FindAllStringSubmatch(s, -1)
+		if len(match) == 0 {
+			continue
+		}
+		if len(match[0]) != 2 {
+			continue
+		}
+		return trim(match[0][1])
+	}
+	return CamelCaseToUnderscore(message.GoIdent.GoName)
+}
+
+func GetCommentBsonName(field *protogen.Field) string {
+	raw := field.Comments.Leading.String()
+	split := strings.Split(raw, "\n")
+	re := regexp.MustCompile("@bson:\\s*(.*)")
+	var value string
+	for _, s := range split {
+		match := re.FindAllStringSubmatch(s, -1)
+		if len(match) == 0 {
+			continue
+		}
+		if len(match[0]) != 2 {
+			continue
+		}
+		value = trim(match[0][1])
+	}
+	if value == "" {
+		value = CamelCaseToUnderscore(field.GoName)
+	}
+	if strings.EqualFold(value, "id") {
+		value = "_id"
+	}
+	return value
+}
+
+func GetRPCInfoList(services *protogen.Service) []*rpcInfo {
+	if len(services.Methods) == 0 {
+		return nil
+	}
+	var list []*rpcInfo
+	for _, method := range services.Methods {
+		node := &rpcInfo{
+			FuncName:   method.GoName,
+			RouterPath: GetCommentApiURL(method),
+			Method:     GetCommentHttpMethod(method.Comments),
+			Author:     GetCommentAuthor(method.Comments),
+			Describe:   GetCommentDescribe(method.Comments),
+			ReqName:    method.Input.GoIdent.GoName,
+			RespName:   method.Output.GoIdent.GoName,
+		}
+		list = append(list, node)
+	}
+	return list
+}
+
+func IsCommentModel(message *protogen.Message) bool {
+	raw := message.Comments.Leading.String()
+	split := strings.Split(raw, "\n")
+	re := regexp.MustCompile("@model:\\s*(true|false)")
+	for _, s := range split {
+		match := re.FindAllStringSubmatch(s, -1)
+		if len(match) == 0 {
+			continue
+		}
+		if len(match[0]) != 2 {
+			continue
+		}
+		if match[0][1] == "true" {
+			return true
+		}
+	}
+	if strings.HasPrefix(message.GoIdent.GoName, "Model") {
+		return true
+	}
+	return false
+}

error_code_generate.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+	"path/filepath"
+	"text/template"
+	"time"
+
+	"google.golang.org/protobuf/compiler/protogen"
+)
+
+func (c *Coco) generateErrorCode(plugin *protogen.Plugin) {
+	for _, pbFile := range plugin.Files {
+		for _, e := range pbFile.Enums {
+			if e.Desc.Name() != "ErrCode" {
+				continue
+			}
+			values, err := c.errcodeDefine(string(pbFile.GoPackageName), e)
+			if err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "generate error code failed: %v", err)
+				continue
+			}
+			filename := fmt.Sprintf("%s/autogen_errcode_%s.go",
+				filepath.Dir(pbFile.GeneratedFilenamePrefix), pbFile.GoPackageName)
+			g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
+			g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
+			g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
+			g.P("// generate at: ", time.Now().Format(time.DateTime))
+			g.P()
+			g.P("package ", pbFile.GoPackageName)
+			g.P()
+			g.P(`import (
+	"gitter.top/coco/coco/core"
+)`)
+			g.P()
+			g.P(values)
+			g.P()
+		}
+	}
+}
+
+func (c *Coco) errcodeDefine(pkgName string, enum *protogen.Enum) (string, error) {
+	tpl, err := template.New("errcode_map").Parse(generateErrorCode)
+	if err != nil {
+		return "", fmt.Errorf("parse error code template failed: %v", err)
+	}
+
+	type T struct {
+		ErrorCode   int32
+		ErrorMsg    string
+		ErrorDefine string
+	}
+	var ts []T
+	for _, enumItem := range enum.Values {
+		ts = append(ts, T{
+			ErrorCode:   int32(enumItem.Desc.Number()),
+			ErrorMsg:    getCommentFromEnumItem(enumItem.Comments),
+			ErrorDefine: string(enumItem.Desc.Name()),
+		})
+	}
+	type TT struct {
+		PkgName    string
+		ErrorCodes []T
+	}
+
+	var tt = TT{PkgName: JavascriptCaseToCamelCase(pkgName), ErrorCodes: ts}
+	tpl.DefinedTemplates()
+	var buf bytes.Buffer
+	if err = tpl.Execute(&buf, tt); err != nil {
+		return "", fmt.Errorf("execute error code template failed: %v", err)
+	}
+	return buf.String(), nil
+}
+
+const generateErrorCode = `var (
+	{{.PkgName}}ErrcodeMap = map[int32]string{ {{range $errcode := .ErrorCodes}}
+		int32(ErrCode_{{$errcode.ErrorDefine}}):	"{{$errcode.ErrorMsg}}",{{end}}
+	}
+)
+
+var (
+	{{range $errcode := .ErrorCodes}}{{$errcode.ErrorDefine}} = &core.ErrMsg{ErrCode: {{$errcode.ErrorCode}}, ErrMsg: "{{$errcode.ErrorMsg}}"}
+{{end}}
+)
+`

go.mod

@@ -1,34 +1,12 @@
 module gitter.top/coco/protoc-gen-coco
 
-go 1.19
+go 1.21
+
+toolchain go1.21.5
 
 require (
-	gitter.top/coco/coco v0.0.0-20230312134419-0dd59eb0e955
-	google.golang.org/protobuf v1.28.1
-)
-
-require (
-	github.com/bytedance/sonic v1.8.0 // indirect
-	github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
-	github.com/gin-contrib/sse v0.1.0 // indirect
-	github.com/gin-gonic/gin v1.9.0 // indirect
-	github.com/go-playground/locales v0.14.1 // indirect
-	github.com/go-playground/universal-translator v0.18.1 // indirect
-	github.com/go-playground/validator/v10 v10.11.2 // indirect
-	github.com/goccy/go-json v0.10.0 // indirect
-	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/klauspost/cpuid/v2 v2.0.9 // indirect
-	github.com/leodido/go-urn v1.2.1 // indirect
-	github.com/mattn/go-isatty v0.0.17 // indirect
-	github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
-	github.com/modern-go/reflect2 v1.0.2 // indirect
-	github.com/pelletier/go-toml/v2 v2.0.6 // indirect
-	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
-	github.com/ugorji/go/codec v1.2.9 // indirect
-	golang.org/x/arch v0.0.0-20210923205945-b76863e36670 // indirect
-	golang.org/x/crypto v0.5.0 // indirect
-	golang.org/x/net v0.7.0 // indirect
-	golang.org/x/sys v0.5.0 // indirect
-	golang.org/x/text v0.7.0 // indirect
-	gopkg.in/yaml.v3 v3.0.1 // indirect
+	github.com/emicklei/proto v1.13.2
+	gitter.top/common/goast v0.0.1
+	gitter.top/common/protofmt v0.0.1
+	google.golang.org/protobuf v1.33.0
 )

go.sum

@@ -1,83 +1,20 @@
-github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
-github.com/bytedance/sonic v1.8.0 h1:ea0Xadu+sHlu7x5O3gKhRpQ1IKiMrSiHttPF0ybECuA=
-github.com/bytedance/sonic v1.8.0/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
-github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
-github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
-github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
-github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
-github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
-github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
-github.com/gin-gonic/gin v1.9.0 h1:OjyFBKICoexlu99ctXNR2gg+c5pKrKMuyjgARg9qeY8=
-github.com/gin-gonic/gin v1.9.0/go.mod h1:W1Me9+hsUSyj3CePGrd1/QrKJMSJ1Tu/0hFEH89961k=
-github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
-github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
-github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
-github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
-github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
-github.com/go-playground/validator/v10 v10.11.2 h1:q3SHpufmypg+erIExEKUmsgmhDTyhcJ38oeKGACXohU=
-github.com/go-playground/validator/v10 v10.11.2/go.mod h1:NieE624vt4SCTJtD87arVLvdmjPAeV8BQlHtMnw9D7s=
-github.com/goccy/go-json v0.10.0 h1:mXKd9Qw4NuzShiRlOXKews24ufknHO7gx30lsDyokKA=
-github.com/goccy/go-json v0.10.0/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
-github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
+github.com/emicklei/proto v1.13.2 h1:z/etSFO3uyXeuEsVPzfl56WNgzcvIr42aQazXaQmFZY=
+github.com/emicklei/proto v1.13.2/go.mod h1:rn1FgRS/FANiZdD2djyH7TMA9jdRDcYQ9IEN9yvjX0A=
 github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
 github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
-github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
-github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
-github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
-github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4=
-github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
-github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
-github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
-github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w=
-github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY=
-github.com/mattn/go-isatty v0.0.17 h1:BTarxUcIeDqL27Mc+vyvdWYSL28zpIhv3RoTdsLMPng=
-github.com/mattn/go-isatty v0.0.17/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
-github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
-github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
-github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
-github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
-github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU=
-github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
-github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
-github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
-github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
-github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
-github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
-github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
-github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
-github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
-github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
-github.com/ugorji/go/codec v1.2.9 h1:rmenucSohSTiyL09Y+l2OCk+FrMxGMzho2+tjr5ticU=
-github.com/ugorji/go/codec v1.2.9/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
-gitter.top/coco/coco v0.0.0-20230312134419-0dd59eb0e955 h1:dLnS34ByD6KgTiUHBPQSqhvh1yk9pR10P3So/Gz9Zo0=
-gitter.top/coco/coco v0.0.0-20230312134419-0dd59eb0e955/go.mod h1:IptVJiCXf87Xywfw26zrvXQjTjF8ZsIHUTJ8yPWSpWE=
-golang.org/x/arch v0.0.0-20210923205945-b76863e36670 h1:18EFjUmQOcUvxNYSkA6jO9VAiXCnxFY6NyDX0bHDmkU=
-golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
-golang.org/x/crypto v0.5.0 h1:U/0M97KRkSFvyD/3FSmdP5W5swImpNgle/EHFhOsQPE=
-golang.org/x/crypto v0.5.0/go.mod h1:NK/OQwhpMQP3MwtdjgLlYHnH9ebylxKWv3e0fK+mkQU=
-golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g=
-golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
-golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.5.0 h1:MUK/U/4lj1t1oPg0HfuXDN/Z1wv31ZJ/YcPiGccS4DU=
-golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo=
-golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+gitter.top/common/goast v0.0.1 h1:YYpEOVXdlkORHAExgZSSUuT1MaebGE46EGcJLBxu/Sw=
+gitter.top/common/goast v0.0.1/go.mod h1:+7nJ3ablVgA5+56qmTnIqrQhD9AjIN28YVFQYylDtm0=
+gitter.top/common/protofmt v0.0.1 h1:aIRLb4vFKu1u3R7e/foijaNzls6N31F2yhnw2yUSXl0=
+gitter.top/common/protofmt v0.0.1/go.mod h1:LrTenAqDqQQ74bBmpVwEOqZhmhcobtE79j0hQbeairs=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
-google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
-google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w=
-google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
-gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
-gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
-gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
+google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
-rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=

main.go

@@ -1,44 +1,122 @@
 package main
 
 import (
+	"bytes"
+	"flag"
 	"fmt"
-	"google.golang.org/protobuf/compiler/protogen"
 	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/emicklei/proto"
+	"gitter.top/common/protofmt"
+	"google.golang.org/protobuf/compiler/protogen"
 )
 
-type Coco struct{}
+type Coco struct {
+	DisableGenerateRouter     bool // 禁止路由信息生成
+	DisableGenerateMongoModel bool // 禁止mongo信息生成
+	DisableGenerateErrorCode  bool // 禁止错误码信息生成
+	DisableGenerateRouterWire bool // 禁止wire路由信息生成
+
+	Prefix      string // Prefix 当遇到proto是相对路径生成的时候 指定prefix
+	ProjectName string // 项目名称
+}
 
 // Generate generate coco router map
 func (c *Coco) Generate(plugin *protogen.Plugin) error {
 	if len(plugin.Files) == 0 {
 		return nil
 	}
+
+	if c.Prefix != "" {
+		if !strings.HasSuffix(c.Prefix, "/") {
+			c.Prefix += "/"
+		}
+	}
+
+	c.format(plugin)
+
+	if !c.DisableGenerateRouter {
+		c.generateRouterMap(plugin)
+		c.generateRouterImpl(plugin)
+		c.generateGrpcImpl(plugin)
+		c.generateRouterWire(plugin)
+	}
+
+	if !c.DisableGenerateMongoModel {
+		c.generateMongoModel(plugin)
+	}
+
+	if !c.DisableGenerateErrorCode {
+		c.generateErrorCode(plugin)
+	}
+
+	return nil
+}
+
+func (c *Coco) format(plugin *protogen.Plugin) {
+	for _, pbFile := range plugin.Files {
+		filename := pbFile.Desc.Path()
+		if c.Prefix != "" {
+			filename = c.Prefix + filename
+		}
+		file, err := os.Open(filename)
+		if err != nil {
+			_, _ = fmt.Fprintf(os.Stderr, "format pbFile %s failed: %v\n", filename, err)
+			continue
+		}
+		buf := new(bytes.Buffer)
+		parser := proto.NewParser(file)
+		parser.Filename(filename)
+		def, err := parser.Parse()
+		if err != nil {
+			_, _ = fmt.Fprintf(os.Stderr, "parse pbFile %s failed: %v\n", filename, err)
+			continue
+		}
+		protofmt.NewFormatter(buf, "    ").Format(def)
+		if err := os.WriteFile(filename, buf.Bytes(), os.ModePerm); err != nil {
+			_, _ = fmt.Fprintf(os.Stderr, "rewrite pbFile %s failed: %v\n", filename, err)
+			continue
+		}
+	}
+}
+
+func (c *Coco) generateRouterMap(plugin *protogen.Plugin) {
 	for _, pbFile := range plugin.Files {
 		// service empty
 		if len(pbFile.Services) == 0 {
 			continue
 		}
 
-		// write file header
-		filename := fmt.Sprintf("autogen_router_%s.go", pbFile.GeneratedFilenamePrefix)
-		g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
-		g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
-		g.P()
-		g.P("package ", pbFile.GoPackageName)
-		g.P()
-		g.P(`import (
-	"gitter.top/coco/coco/core"
-)`)
-		g.P()
-
 		for _, service := range pbFile.Services {
-			// service router group
+			// is service router group
 			if !IsCommentRouterGroup(pbFile.Services[0].Comments) {
 				continue
 			}
 			if len(service.Methods) == 0 {
 				continue
 			}
+
+			if GetCommentGenerateTo(service.Comments) == "" {
+				continue
+			}
+
+			// write file header
+			filename := fmt.Sprintf("%s/autogen_router_%s.go",
+				separator(filepath.Dir(pbFile.GeneratedFilenamePrefix)), CamelCaseToUnderscore(service.GoName))
+			g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
+			g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
+			g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
+			g.P("// generate at: ", time.Now().Format("2006-01-02 15:04:05"))
+			g.P()
+			g.P("package ", pbFile.GoPackageName)
+			g.P()
+			g.P(`import (
+	"gitter.top/coco/coco"
+)`)
+			g.P()
 			values, err := GenerateRouterMap(service)
 			if err != nil {
 				_, _ = fmt.Fprintf(os.Stderr, "generate router map failed: %v", err)
@@ -47,12 +125,168 @@ func (c *Coco) Generate(plugin *protogen.Plugin) error {
 			g.P(values)
 			g.P()
 		}
-
 	}
-	return nil
+}
+
+func (c *Coco) generateRouterImpl(plugin *protogen.Plugin) {
+	for _, pbFile := range plugin.Files {
+		// service empty
+		if len(pbFile.Services) == 0 {
+			continue
+		}
+
+		for _, service := range pbFile.Services {
+			// is service router group
+			if !IsCommentRouterGroup(service.Comments) {
+				continue
+			}
+
+			genTo := GetCommentGenerateTo(service.Comments)
+			if genTo == "" {
+				continue
+			}
+			// mkdir when Dir(genTo) not found
+			if err := mkdir(filepath.Dir(genTo)); err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "mkdir genTo path failed: %s\n", err)
+				return
+			}
+			// new version, package name only support `controller`+`api_version`
+			// so old value `string(pbFile.GoPackageName)` Deprecated.
+			version := filepath.Base(filepath.Dir(pbFile.GeneratedFilenamePrefix))
+			generator := newRouterImpl(genTo, "controller"+version, string(pbFile.GoPackageName), service.GoName, service)
+			generator.ProjectName = c.ProjectName
+			if !generator.IsFileExist() {
+				if err := generator.generateNewFile(); err != nil {
+					_, _ = fmt.Fprintf(os.Stderr, "generate all router failed: %v\n", err)
+				}
+				return
+			}
+			if err := generator.generateNewRouters(); err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "generate router impl failed: %v\n", err)
+			}
+		}
+	}
+}
+
+func (c *Coco) generateGrpcImpl(plugin *protogen.Plugin) {
+	for _, pbFile := range plugin.Files {
+		// service empty
+		if len(pbFile.Services) == 0 {
+			continue
+		}
+
+		for _, service := range pbFile.Services {
+			genTo := GetCommentRpcGenerateTo(service.Comments)
+			if genTo == "" {
+				continue
+			}
+			// mkdir when Dir(genTo) not found
+			if err := mkdir(filepath.Dir(genTo)); err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "mkdir genTo path failed: %s\n", err)
+				return
+			}
+			version := filepath.Base(filepath.Dir(pbFile.GeneratedFilenamePrefix))
+			generator := newRouterImpl(genTo, "rpc"+version, string(pbFile.GoPackageName), service.GoName, service)
+			generator.ProjectName = c.ProjectName
+			if !generator.IsFileExist() {
+				if err := generator.generateNewGrpcFile(); err != nil {
+					_, _ = fmt.Fprintf(os.Stderr, "generate all grpc failed: %v\n", err)
+				}
+				return
+			}
+			if err := generator.generateNewGrpcServers(); err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "generate rpc impl failed: %v\n", err)
+			}
+		}
+	}
+}
+
+func (c *Coco) generateMongoModel(plugin *protogen.Plugin) {
+	for _, pbFile := range plugin.Files {
+		if len(pbFile.Messages) == 0 {
+			return
+		}
+
+		var needMessages []*protogen.Message
+		for _, message := range pbFile.Messages {
+			// 非model
+			if !IsCommentModel(message) {
+				continue
+			}
+			needMessages = append(needMessages, message)
+		}
+
+		if len(needMessages) == 0 {
+			continue
+		}
+		filename := fmt.Sprintf("%s/autogen_model_%s.go",
+			separator(filepath.Dir(pbFile.GeneratedFilenamePrefix)), filepath.Base(pbFile.GeneratedFilenamePrefix))
+		g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
+		g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
+		g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
+		g.P("// generate at: ", time.Now().Format("2006-01-02 15:04:05"))
+		g.P()
+		g.P("package ", pbFile.GoPackageName)
+		g.P()
+		for _, message := range needMessages {
+			value, err := GenerateModel(message, string(pbFile.GoPackageName))
+			if err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "generate model failed: %v", err)
+				continue
+			}
+			g.P(value)
+			g.P()
+		}
+	}
+}
+
+func (c *Coco) generateRouterWire(plugin *protogen.Plugin) {
+	if c.DisableGenerateRouterWire {
+		return
+	}
+	for _, pbFile := range plugin.Files {
+		if len(pbFile.Services) == 0 {
+			continue
+		}
+		for _, service := range pbFile.Services {
+			// is service router group
+			if !IsCommentRouterGroup(service.Comments) {
+				continue
+			}
+
+			if err := mkdir("gen/wire"); err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "mkdir gen/wire path failed: %s\n", err)
+				return
+			}
+			wireDir := "gen/wire/wire.go"
+			generator := newRouterImplWire(wireDir, "wire", string(pbFile.GoPackageName), service.GoName)
+			generator.Version = filepath.Base(filepath.Dir(pbFile.GeneratedFilenamePrefix))
+			generator.ProjectName = c.ProjectName
+			if !generator.IsFileExist() {
+				if err := generator.generateNewWireFile(); err != nil {
+					_, _ = fmt.Fprintf(os.Stderr, "generate router wire failed: %v\n", err)
+					return
+				}
+			}
+			if err := generator.generateServicesWire(); err != nil {
+				_, _ = fmt.Fprintf(os.Stderr, "generate router wire failed: %v\n", err)
+			}
+		}
+	}
 }
 
 func main() {
 	var coco = new(Coco)
-	protogen.Options{}.Run(coco.Generate)
+	var flags flag.FlagSet
+
+	flags.BoolVar(&coco.DisableGenerateErrorCode, "disable_error_code", false, "disable generate error code")
+	flags.BoolVar(&coco.DisableGenerateMongoModel, "disable_mongodb_model", false, "disable generate mongodb model")
+	flags.BoolVar(&coco.DisableGenerateRouter, "disable_router", false, "disable generate router code")
+	flags.BoolVar(&coco.DisableGenerateRouterWire, "disable_router_wire", false, "disable generate router wire")
+	flags.StringVar(&coco.Prefix, "prefix", "", "proto prefix")
+	flags.StringVar(&coco.ProjectName, "project_name", "", "project name")
+
+	protogen.Options{
+		ParamFunc: flags.Set,
+	}.Run(coco.Generate)
 }

model_generate.go

@@ -0,0 +1,63 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"text/template"
+
+	"google.golang.org/protobuf/compiler/protogen"
+)
+
+const ModelTpl = `const TableName{{.ModelName}} = "{{.TableName}}"
+func (t *{{.ModelName}}) TableName() string {
+	return "{{.TableName}}"
+}
+
+{{range $structs := .DbFields}}
+func (m *{{$.ModelName}}) Get{{$structs.FieldName}}Field() string {
+	return "{{$structs.BsonField}}"
+}
+{{end}}
+`
+
+type DbField struct {
+	FieldName string
+	BsonField string
+}
+
+type model struct {
+	ModelName   string
+	TableName   string
+	PackageName string
+	DbFields    []*DbField
+}
+
+func GenerateModel(message *protogen.Message, packageName string) (string, error) {
+	var name = message.GoIdent.GoName
+	var entry = &model{
+		ModelName:   name,
+		TableName:   GetCommentTableName(message),
+		PackageName: packageName,
+	}
+
+	for _, field := range message.Fields {
+		var dbField = &DbField{
+			FieldName: field.GoName,
+			BsonField: GetCommentBsonName(field),
+		}
+		entry.DbFields = append(entry.DbFields, dbField)
+	}
+
+	tpl, err := template.New("model_tpl").Parse(ModelTpl)
+	if err != nil {
+		return "", fmt.Errorf("parse %s template failed: %v", name, err)
+	}
+
+	tpl.DefinedTemplates()
+	var buf bytes.Buffer
+	if err = tpl.Execute(&buf, entry); err != nil {
+		return "", fmt.Errorf("execute %s template failed: %v", name, err)
+	}
+
+	return buf.String(), nil
+}

router_generate.go

@@ -2,9 +2,14 @@ package main
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
-	"google.golang.org/protobuf/compiler/protogen"
+	"os"
+	"strings"
 	"text/template"
+
+	"gitter.top/common/goast"
+	"google.golang.org/protobuf/compiler/protogen"
 )
 
 type generateRouterMethod struct {
@@ -25,12 +30,15 @@ type generateRouters struct {
 }
 
 const routerGenerateTpl = `type AutoGen{{.StructName}}Impl interface { ` + `{{range .Methods}}
-	{{.MethodName}}(ctx *core.Context, req *{{.InputName}}) (resp *{{.OutputName}}, err error){{end}}
+	{{.MethodName}}(ctx *coco.Context, req *{{.InputName}}) (resp *{{.OutputName}}, err error){{end}}
 }
-var (
-	AutoGen{{.StructName}} = &core.Routers{
-			BaseURL: "{{.BaseURL}}",
-			Apis: core.RouterMap{ {{range .Methods}}
+
+type {{.StructName}}_RouterMap struct {}
+
+func (receiver *{{.StructName}}_RouterMap) GetRouterMap() *coco.Routers {
+        return &coco.Routers {
+            BaseURL: "{{.BaseURL}}",
+			Apis: coco.RouterMap{ {{range .Methods}}
 				"{{.MethodName}}": {
 					API:      "{{.API}}",
 					Method:   "{{.Method}}",
@@ -38,8 +46,8 @@ var (
 					Describe: "{{.Describe}}",
 				},{{end}}
 			},
+        }
 	}
-)
 `
 
 func GenerateRouterMap(srv *protogen.Service) (string, error) {
@@ -73,3 +81,425 @@ func GenerateRouterMap(srv *protogen.Service) (string, error) {
 
 	return buf.String(), nil
 }
+
+type rpcInfo struct {
+	FuncName   string // 函数名
+	RouterPath string // 路由路径
+	Method     string // 请求类型 POST/GET...
+	Author     string // 接口作者
+	Describe   string // 描述
+	ReqName    string // request name
+	RespName   string // response name
+}
+
+type routerImpl struct {
+	Filename    string
+	PkgName     string
+	SrvName     string
+	GoPkgName   string
+	ProjectName string
+	Version     string
+	Routers     []*rpcInfo
+}
+
+// newRouterImpl 生成路由信息
+// filename 文件生成地址
+// pkgName 文件包名
+// srvName 被控制的service名称
+// service service信息
+func newRouterImpl(filename, pkgName, goPkgName, srvName string, service *protogen.Service) *routerImpl {
+	if filename == "" {
+		filename = "./"
+	}
+	return &routerImpl{
+		Filename:  filename,
+		SrvName:   srvName,
+		PkgName:   pkgName,
+		GoPkgName: goPkgName,
+		Routers:   GetRPCInfoList(service),
+	}
+}
+
+func newRouterImplWire(filename, pkgName, goPkgName, srvName string) *routerImpl {
+	return &routerImpl{
+		Filename:  filename,
+		SrvName:   srvName,
+		PkgName:   pkgName,
+		GoPkgName: goPkgName,
+	}
+}
+
+// fixPkgName 修正不规范的包名
+func fixPkgName(name string) string {
+	name = strings.ReplaceAll(name, "-", "_")
+	name = strings.ReplaceAll(name, ".", "_")
+	return name
+}
+
+func (ri *routerImpl) IsFileExist() bool {
+	_, err := os.Stat(ri.Filename)
+	if err != nil {
+		if errors.Is(err, os.ErrNotExist) {
+			return false
+		} else {
+			_, _ = fmt.Fprintf(os.Stderr, "get file stat failed: %v", err)
+		}
+		return false
+	}
+	return true
+}
+
+func (ri *routerImpl) GetFileInfo(filename string) (os.FileInfo, error) {
+	return os.Stat(filename)
+}
+
+const CompleteRouteGenerateAndPackageTpl = `package {{.PkgName}}
+
+import (
+	"gitter.top/coco/coco"
+
+	{{$.GoPkgName}} "{{.ProjectName}}/gen/v1"
+)
+
+type {{.SrvName}} struct {}
+
+// IDE: {{.SrvName}} implemented {{.GoPkgName}}.AutoGen{{.SrvName}}Impl interface
+var _ {{.GoPkgName}}.AutoGen{{.SrvName}}Impl = (*{{.SrvName}})(nil)
+
+{{range $route := .Routers}}
+// {{$route.FuncName}} {{$route.Describe}}
+func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx *coco.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
+	resp = new({{$.GoPkgName}}.{{$route.RespName}})
+	
+	// TODO impl...
+
+	return resp, nil
+}
+{{end}}
+`
+
+func (ri *routerImpl) generateNewFile() error {
+	t, err := template.New("generate_new_file").Parse(CompleteRouteGenerateAndPackageTpl)
+	if err != nil {
+		return err
+	}
+	t.DefinedTemplates()
+	var buf bytes.Buffer
+	if err = t.Execute(&buf, ri); err != nil {
+		return err
+	}
+	f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	if _, err = f.Write(buf.Bytes()); err != nil {
+		return err
+	}
+	return nil
+}
+
+const CompleteGrpcGenerateAndPackageTpl = `package {{.PkgName}}
+
+import (
+	"context"
+
+	{{.GoPkgName}} "{{.ProjectName}}/gen/v1"
+)
+
+type {{.SrvName}} struct {}
+
+// IDE: {{.SrvName}} implemented {{.GoPkgName}}.{{.SrvName}}Server interface
+var _ {{.GoPkgName}}.{{.SrvName}}Server = (*{{.SrvName}})(nil)
+
+{{range $route := .Routers}}
+// {{$route.FuncName}} {{$route.Describe}}
+func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx context.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
+	resp = new({{$.GoPkgName}}.{{$route.RespName}})
+	
+	// TODO impl...
+
+	return resp, nil
+}
+{{end}}
+`
+
+func (ri *routerImpl) generateNewGrpcFile() error {
+	t, err := template.New("generate_new_grpc_file").Parse(CompleteGrpcGenerateAndPackageTpl)
+	if err != nil {
+		return err
+	}
+	t.DefinedTemplates()
+	var buf bytes.Buffer
+	if err = t.Execute(&buf, ri); err != nil {
+		return err
+	}
+	f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	if _, err = f.Write(buf.Bytes()); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (ri *routerImpl) generateNewRouters() error {
+	info, err := ri.GetFileInfo(ri.Filename)
+	if err != nil {
+		return err
+	}
+	if info.IsDir() {
+		return nil
+	}
+	parser, err := goast.NewParser(ri.Filename)
+	if err != nil {
+		return err
+	}
+
+	definer, err := parser.Parse()
+	if err != nil {
+		return err
+	}
+
+	if !definer.ExistType(ri.SrvName) {
+		tpl := `
+type {{.SrvName}} struct {}
+
+// IDE: {{.SrvName}} implemented {{.GoPkgName}}.AutoGen{{.SrvName}}Impl interface
+var _ {{.GoPkgName}}.AutoGen{{.SrvName}}Impl = (*{{.SrvName}})(nil)
+`
+		t, err := template.New("generate_type").Parse(tpl)
+		if err != nil {
+			return err
+		}
+		t.DefinedTemplates()
+		var buf bytes.Buffer
+		if err = t.Execute(&buf, ri); err != nil {
+			return err
+		}
+		f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+		if _, err = f.Write(buf.Bytes()); err != nil {
+			return err
+		}
+	}
+
+	var needImpls []*rpcInfo
+	for _, router := range ri.Routers {
+		if !definer.ExistMethod(fmt.Sprintf("%s.%s", ri.SrvName, router.FuncName)) {
+			needImpls = append(needImpls, router)
+		}
+	}
+	if len(needImpls) != 0 {
+		tpl := `{{range $route := .Routers}}
+// {{$route.FuncName}} {{$route.Describe}}
+func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx *coco.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
+	resp = new({{$.GoPkgName}}.{{$route.RespName}})
+	
+	// TODO impl...
+
+	return resp, nil
+}
+{{end}}`
+		ri.Routers = needImpls
+		t, err := template.New("generate_impls").Parse(tpl)
+		if err != nil {
+			return err
+		}
+		t.DefinedTemplates()
+		var buf bytes.Buffer
+		if err = t.Execute(&buf, ri); err != nil {
+			return err
+		}
+		f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+		if _, err = f.Write(buf.Bytes()); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (ri *routerImpl) generateNewGrpcServers() error {
+	info, err := ri.GetFileInfo(ri.Filename)
+	if err != nil {
+		return err
+	}
+	if info.IsDir() {
+		return nil
+	}
+	parser, err := goast.NewParser(ri.Filename)
+	if err != nil {
+		return err
+	}
+
+	definer, err := parser.Parse()
+	if err != nil {
+		return err
+	}
+
+	if !definer.ExistType(ri.SrvName) {
+		tpl := `
+type {{.SrvName}} struct {}
+
+// IDE: {{.SrvName}} implemented {{.GoPkgName}}.{{.SrvName}}Server interface
+var _ {{.GoPkgName}}.{{.SrvName}}Server = (*{{.SrvName}})(nil)
+`
+		t, err := template.New("generate_type").Parse(tpl)
+		if err != nil {
+			return err
+		}
+		t.DefinedTemplates()
+		var buf bytes.Buffer
+		if err = t.Execute(&buf, ri); err != nil {
+			return err
+		}
+		f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+		if _, err = f.Write(buf.Bytes()); err != nil {
+			return err
+		}
+	}
+
+	var needImpls []*rpcInfo
+	for _, router := range ri.Routers {
+		if !definer.ExistMethod(fmt.Sprintf("%s.%s", ri.SrvName, router.FuncName)) {
+			needImpls = append(needImpls, router)
+		}
+	}
+	if len(needImpls) != 0 {
+		tpl := `{{range $route := .Routers}}
+// {{$route.FuncName}} {{$route.Describe}}
+func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx context.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
+	resp = new({{$.GoPkgName}}.{{$route.RespName}})
+	
+	// TODO impl...
+
+	return resp, nil
+}
+{{end}}`
+		ri.Routers = needImpls
+		t, err := template.New("generate_impls").Parse(tpl)
+		if err != nil {
+			return err
+		}
+		t.DefinedTemplates()
+		var buf bytes.Buffer
+		if err = t.Execute(&buf, ri); err != nil {
+			return err
+		}
+		f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+		if _, err = f.Write(buf.Bytes()); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+const routeWireCompleteTpl = `//go:build wireinject
+// +build wireinject
+
+package wire
+
+import (
+	"github.com/google/wire"
+
+	"{{.ProjectName}}/gen/{{.Version}}"
+	"{{.ProjectName}}/services/controller/{{.Version}}"
+)
+
+`
+
+func (ri *routerImpl) generateNewWireFile() error {
+	t, err := template.New("generate_new_wire").Parse(routeWireCompleteTpl)
+	if err != nil {
+		return err
+	}
+	t.DefinedTemplates()
+	var buf bytes.Buffer
+	if err = t.Execute(&buf, ri); err != nil {
+		return err
+	}
+	f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	if _, err = f.Write(buf.Bytes()); err != nil {
+		return err
+	}
+	return nil
+}
+
+const routeWireTpl = `type {{.SrvName}} struct {
+	*controller{{.Version}}.{{.SrvName}}
+	*gen{{.Version}}.{{.SrvName}}_RouterMap
+}
+
+func New{{.SrvName}}(c *controller{{.Version}}.{{.SrvName}}, r *gen{{.Version}}.{{.SrvName}}_RouterMap) *{{.SrvName}} {
+	return &{{.SrvName}}{c, r}
+}
+
+func Init{{.SrvName}}() *{{.SrvName}} {
+	wire.Build(wire.Struct(new(controller{{.Version}}.{{.SrvName}})), wire.Struct(new(gen{{.Version}}.{{.SrvName}}_RouterMap)), New{{.SrvName}})
+	return &{{.SrvName}}{}
+}
+`
+
+func (ri *routerImpl) generateServicesWire() error {
+	info, err := ri.GetFileInfo(ri.Filename)
+	if err != nil {
+		return err
+	}
+	if info.IsDir() {
+		return nil
+	}
+	parser, err := goast.NewParser(ri.Filename)
+	if err != nil {
+		return err
+	}
+
+	definer, err := parser.Parse()
+	if err != nil {
+		return err
+	}
+	if definer.ExistType(ri.SrvName) {
+		return nil
+	}
+
+	t, err := template.New("generate_router_wire").Parse(routeWireTpl)
+	if err != nil {
+		return err
+	}
+	t.DefinedTemplates()
+	var buf bytes.Buffer
+	if err = t.Execute(&buf, ri); err != nil {
+		return err
+	}
+	f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	if _, err = f.Write(buf.Bytes()); err != nil {
+		return err
+	}
+	return nil
+}

test.proto

@@ -1,41 +0,0 @@
-syntax = "proto3";
-
-package v1;
-
-option go_package = "./;main";
-
-// @route_group: true
-// @base_url: /api/v1
-service ExampleService { // tail
-  rpc ExampleCall1(ExampleMessage1) returns(ReturnType) {}
-  rpc ExampleCall2(ExampleMessage2) returns(ReturnType) {}
-}
-
-// @route_group: true
-// @base_url: /api/v2
-service Example1Service { // tail
-  rpc ExampleCall1(ExampleMessage1) returns(ReturnType) {}
-  rpc ExampleCall2(ExampleMessage2) returns(ReturnType) {}
-}
-
-// ExampleMessage1 - Example Leading Comment for ExampleMessage1
-message ExampleMessage1 {
-  string MyString = 1;
-}
-
-/*
-ExampleMessage2 - Example Leading Comment for ExampleMessage2
-*/
-message ExampleMessage2 {
-  int32 MyInt = 1;
-  // MyInt - Example trailing Comment
-  message ExampleNested {
-    bytes data = 1;
-  }
-  ExampleNested nested = 2;
-}
-
-/*
-ReturnType - Empty Structure Placeholder
-*/
-message ReturnType {}

util.go

@@ -1,6 +1,12 @@
 package main
 
-import "unicode"
+import (
+	"os"
+	"strings"
+	"unicode"
+
+	"google.golang.org/protobuf/compiler/protogen"
+)
 
 // CamelCaseToUnderscore converts CamelCase to camel_case
 func CamelCaseToUnderscore(str string) string {
@@ -38,3 +44,46 @@ func CamelCaseToJavascriptCase(str string) string {
 	}
 	return result
 }
+
+// JavascriptCaseToCamelCase convert camelCase to CamelCase
+func JavascriptCaseToCamelCase(str string) string {
+	var result string
+	for idx, ch := range str {
+		if idx == 0 {
+			result = string(unicode.ToUpper(ch))
+			continue
+		}
+
+		result += string(ch)
+	}
+	return result
+}
+
+func getCommentFromEnumItem(comment protogen.CommentSet) string {
+	c := strings.Trim(string(comment.Trailing), " \n\t\r")
+	if c == "" {
+		c = "unknown error"
+	}
+	return c
+}
+
+func separator(name string) string {
+	name = strings.ReplaceAll(name, "\\\\", "/")
+	return strings.ReplaceAll(name, "\\", "/")
+}
+
+func trim(s string) string {
+	return strings.Trim(s, "\r\t\n")
+}
+
+func mkdir(p string) error {
+	_, err := os.Stat(p)
+	if os.IsNotExist(err) {
+		err := os.MkdirAll(p, os.ModePerm)
+		if err != nil {
+			return err
+		}
+		return nil
+	}
+	return err
+}