From d1f399b03dec39b9616c5fc78abe7392a2c6b1e0 Mon Sep 17 00:00:00 2001 From: Young Xu Date: Sun, 12 Mar 2023 23:56:26 +0800 Subject: [PATCH] release: new version --- README.md | 20 ++ comment_parse.go | 115 ++++++++++- error_code_generate.go | 87 ++++++++ go.mod | 36 +--- go.sum | 83 +------- main.go | 270 +++++++++++++++++++++++-- model_generate.go | 63 ++++++ router_generate.go | 444 ++++++++++++++++++++++++++++++++++++++++- test.proto | 41 ---- util.go | 51 ++++- 10 files changed, 1034 insertions(+), 176 deletions(-) create mode 100644 README.md create mode 100644 error_code_generate.go create mode 100644 model_generate.go delete mode 100644 test.proto diff --git a/README.md b/README.md new file mode 100644 index 0000000..c3de090 --- /dev/null +++ b/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=. +``` \ No newline at end of file diff --git a/comment_parse.go b/comment_parse.go index d620207..afd26d9 100644 --- a/comment_parse.go +++ b/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 +} diff --git a/error_code_generate.go b/error_code_generate.go new file mode 100644 index 0000000..117920b --- /dev/null +++ b/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}} +) +` diff --git a/go.mod b/go.mod index c3e411c..9b285de 100644 --- a/go.mod +++ b/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 ) diff --git a/go.sum b/go.sum index 81078d2..f74eaea 100644 --- a/go.sum +++ b/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= diff --git a/main.go b/main.go index 2c1cffb..ec702a3 100644 --- a/main.go +++ b/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) } diff --git a/model_generate.go b/model_generate.go new file mode 100644 index 0000000..4ff986c --- /dev/null +++ b/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 +} diff --git a/router_generate.go b/router_generate.go index 5f4c113..93548b4 100644 --- a/router_generate.go +++ b/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 +} diff --git a/test.proto b/test.proto deleted file mode 100644 index e1a0a44..0000000 --- a/test.proto +++ /dev/null @@ -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 {} \ No newline at end of file diff --git a/util.go b/util.go index 4bb2178..4af3cd0 100644 --- a/util.go +++ b/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 +}