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