From 91a5eb75b857a9737b9df4e90c80c85070229f67 Mon Sep 17 00:00:00 2001 From: Young Xu Date: Sun, 12 Mar 2023 23:37:02 +0800 Subject: [PATCH] first commit --- .gitignore | 3 ++ comment_parse.go | 132 +++++++++++++++++++++++++++++++++++++++++++++ go.mod | 34 ++++++++++++ go.sum | 83 ++++++++++++++++++++++++++++ main.go | 58 ++++++++++++++++++++ router_generate.go | 75 ++++++++++++++++++++++++++ test.proto | 41 ++++++++++++++ util.go | 40 ++++++++++++++ util_test.go | 14 +++++ 9 files changed, 480 insertions(+) create mode 100644 .gitignore create mode 100644 comment_parse.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go create mode 100644 router_generate.go create mode 100644 test.proto create mode 100644 util.go create mode 100644 util_test.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..fe6c39a --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +.idea +autogen_* +*.pb.go \ No newline at end of file diff --git a/comment_parse.go b/comment_parse.go new file mode 100644 index 0000000..d620207 --- /dev/null +++ b/comment_parse.go @@ -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) +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..c3e411c --- /dev/null +++ b/go.mod @@ -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 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..81078d2 --- /dev/null +++ b/go.sum @@ -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= diff --git a/main.go b/main.go new file mode 100644 index 0000000..2c1cffb --- /dev/null +++ b/main.go @@ -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) +} diff --git a/router_generate.go b/router_generate.go new file mode 100644 index 0000000..5f4c113 --- /dev/null +++ b/router_generate.go @@ -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 +} diff --git a/test.proto b/test.proto new file mode 100644 index 0000000..e1a0a44 --- /dev/null +++ b/test.proto @@ -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 {} \ No newline at end of file diff --git a/util.go b/util.go new file mode 100644 index 0000000..4bb2178 --- /dev/null +++ b/util.go @@ -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 +} diff --git a/util_test.go b/util_test.go new file mode 100644 index 0000000..f46ba82 --- /dev/null +++ b/util_test.go @@ -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")) +}