first commit
This commit is contained in:
commit
91a5eb75b8
GPG Key ID:
A23CF9620CBB55F9
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
.idea
|
||||||
|
autogen_*
|
||||||
|
*.pb.go
|
||||||
132
comment_parse.go
Normal file
132
comment_parse.go
Normal file
@ -0,0 +1,132 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func IsCommentRouterGroup(comments protogen.CommentSet) bool {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
if len(split) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
re := regexp.MustCompile("@route_group:\\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 false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentGenerateTo(comments protogen.CommentSet) string {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@gen_to:\\s*([\\w|/|\\.]*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return match[0][1]
|
||||||
|
}
|
||||||
|
return "/"
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentBaseURL(comments protogen.CommentSet) string {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@base_url:\\s*([\\w|/]*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return match[0][1]
|
||||||
|
}
|
||||||
|
return "/"
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentHttpMethod(comments protogen.CommentSet) string {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@method:\\s*([\\w]*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return match[0][1]
|
||||||
|
}
|
||||||
|
return "GET"
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentAuthor(comments protogen.CommentSet) string {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@author:\\s*(.*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return match[0][1]
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentDescribe(comments protogen.CommentSet) string {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@desc:\\s*(.*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return match[0][1]
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentApiURL(method *protogen.Method) string {
|
||||||
|
raw := method.Comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@api:\\s*([\\w|/]*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return match[0][1]
|
||||||
|
}
|
||||||
|
return "/" + CamelCaseToJavascriptCase(method.GoName)
|
||||||
|
}
|
||||||
34
go.mod
Normal file
34
go.mod
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
module gitter.top/coco/protoc-gen-coco
|
||||||
|
|
||||||
|
go 1.19
|
||||||
|
|
||||||
|
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
|
||||||
|
)
|
||||||
83
go.sum
Normal file
83
go.sum
Normal file
@ -0,0 +1,83 @@
|
|||||||
|
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/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=
|
||||||
|
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=
|
||||||
|
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=
|
||||||
58
main.go
Normal file
58
main.go
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Coco struct{}
|
||||||
|
|
||||||
|
// Generate generate coco router map
|
||||||
|
func (c *Coco) Generate(plugin *protogen.Plugin) error {
|
||||||
|
if len(plugin.Files) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
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
|
||||||
|
if !IsCommentRouterGroup(pbFile.Services[0].Comments) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(service.Methods) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
values, err := GenerateRouterMap(service)
|
||||||
|
if err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate router map failed: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
g.P(values)
|
||||||
|
g.P()
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
var coco = new(Coco)
|
||||||
|
protogen.Options{}.Run(coco.Generate)
|
||||||
|
}
|
||||||
75
router_generate.go
Normal file
75
router_generate.go
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
|
"text/template"
|
||||||
|
)
|
||||||
|
|
||||||
|
type generateRouterMethod struct {
|
||||||
|
MethodName string
|
||||||
|
InputName string
|
||||||
|
OutputName string
|
||||||
|
API string
|
||||||
|
Author string
|
||||||
|
Describe string
|
||||||
|
Method string
|
||||||
|
}
|
||||||
|
|
||||||
|
type generateRouters struct {
|
||||||
|
StructName string
|
||||||
|
BaseURL string
|
||||||
|
GenTo string
|
||||||
|
Methods []generateRouterMethod
|
||||||
|
}
|
||||||
|
|
||||||
|
const routerGenerateTpl = `type AutoGen{{.StructName}}Impl interface { ` + `{{range .Methods}}
|
||||||
|
{{.MethodName}}(ctx *core.Context, req *{{.InputName}}) (resp *{{.OutputName}}, err error){{end}}
|
||||||
|
}
|
||||||
|
var (
|
||||||
|
AutoGen{{.StructName}} = &core.Routers{
|
||||||
|
BaseURL: "{{.BaseURL}}",
|
||||||
|
Apis: core.RouterMap{ {{range .Methods}}
|
||||||
|
"{{.MethodName}}": {
|
||||||
|
API: "{{.API}}",
|
||||||
|
Method: "{{.Method}}",
|
||||||
|
Author: "{{.Author}}",
|
||||||
|
Describe: "{{.Describe}}",
|
||||||
|
},{{end}}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
`
|
||||||
|
|
||||||
|
func GenerateRouterMap(srv *protogen.Service) (string, error) {
|
||||||
|
var generate = new(generateRouters)
|
||||||
|
generate.StructName = srv.GoName
|
||||||
|
generate.BaseURL = GetCommentBaseURL(srv.Comments)
|
||||||
|
generate.GenTo = GetCommentGenerateTo(srv.Comments)
|
||||||
|
|
||||||
|
for _, method := range srv.Methods {
|
||||||
|
generate.Methods = append(generate.Methods, generateRouterMethod{
|
||||||
|
MethodName: method.GoName,
|
||||||
|
InputName: method.Input.GoIdent.GoName,
|
||||||
|
OutputName: method.Output.GoIdent.GoName,
|
||||||
|
API: GetCommentApiURL(method),
|
||||||
|
Author: GetCommentAuthor(method.Comments),
|
||||||
|
Describe: GetCommentDescribe(method.Comments),
|
||||||
|
Method: GetCommentHttpMethod(method.Comments),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
tpl, err := template.New("group_router").Parse(routerGenerateTpl)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("parse %s group router template failed: %v", srv.GoName, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tpl.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = tpl.Execute(&buf, generate); err != nil {
|
||||||
|
return "", fmt.Errorf("execute %s group router template failed: %v", srv.GoName, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return buf.String(), nil
|
||||||
|
}
|
||||||
41
test.proto
Normal file
41
test.proto
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
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 {}
|
||||||
40
util.go
Normal file
40
util.go
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "unicode"
|
||||||
|
|
||||||
|
// CamelCaseToUnderscore converts CamelCase to camel_case
|
||||||
|
func CamelCaseToUnderscore(str string) string {
|
||||||
|
var result string
|
||||||
|
for idx, ch := range str {
|
||||||
|
// first letter will just be lowered
|
||||||
|
if idx == 0 {
|
||||||
|
result = string(unicode.ToLower(ch))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// anywhere else
|
||||||
|
if unicode.IsUpper(ch) {
|
||||||
|
result = result + "_" + string(unicode.ToLower(ch))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// nothing to see here, just accept it
|
||||||
|
result += string(ch)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// CamelCaseToJavascriptCase convert CamelCase to camelCase
|
||||||
|
func CamelCaseToJavascriptCase(str string) string {
|
||||||
|
var result string
|
||||||
|
for idx, ch := range str {
|
||||||
|
if idx == 0 {
|
||||||
|
result = string(unicode.ToLower(ch))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
result += string(ch)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
14
util_test.go
Normal file
14
util_test.go
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCamelCaseToUnderscore(t *testing.T) {
|
||||||
|
fmt.Println(CamelCaseToUnderscore("HelloWorld"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCamelCaseToJavascriptCase(t *testing.T) {
|
||||||
|
fmt.Println(CamelCaseToJavascriptCase("HelloWorld"))
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user