You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
2.7 KiB
105 lines
2.7 KiB
2 years ago
|
package main
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"google.golang.org/protobuf/compiler/protogen"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
func main() {
|
||
|
u := &Kit{
|
||
|
imports: map[string]string{},
|
||
|
}
|
||
|
protogen.Options{}.Run(u.Generate)
|
||
|
}
|
||
|
|
||
|
type Kit struct {
|
||
|
imports map[string]string
|
||
|
}
|
||
|
|
||
|
func (u *Kit) addImports(imp string) {
|
||
|
u.imports[imp] = imp
|
||
|
}
|
||
|
func (u *Kit) Generate(plugin *protogen.Plugin) error {
|
||
|
if len(plugin.Files) < 1 {
|
||
|
return nil
|
||
|
}
|
||
|
for _, f := range plugin.Files {
|
||
|
if len(f.Services) == 0 {
|
||
|
continue
|
||
|
}
|
||
|
fname := f.GeneratedFilenamePrefix + ".ts"
|
||
|
t := plugin.NewGeneratedFile(fname, f.GoImportPath)
|
||
|
for _, s := range f.Messages {
|
||
|
t.P(`export interface `, s.Desc.Name(), ` {`)
|
||
|
for _, vv := range s.Fields {
|
||
|
common := strings.TrimSpace(vv.Comments.Leading.String())
|
||
|
if common != "" {
|
||
|
t.P(` `, common)
|
||
|
}
|
||
|
typ := []string{}
|
||
|
isRequired := ":"
|
||
|
if vv.Desc.IsMap() {
|
||
|
fields := vv.Desc.Message().Fields()
|
||
|
typ = append(typ, fmt.Sprintf(`{ [key: %s]: %s }`, getType(fields.Get(0).Kind().String()), getType(fields.Get(1).Kind().String())))
|
||
|
} else if vv.Desc.Kind().String() == "message" {
|
||
|
//t.P(vv.Desc.Message().Fields())
|
||
|
//t.P(vv.Desc.Message().Name())
|
||
|
//if vv.Desc.Message().Fields().Get(0).Name() == "value" {
|
||
|
// isRequired = "?:"
|
||
|
// //typ = append(typ, getType(vv.Desc.Message().Fields().Get(0).Kind().String()))
|
||
|
//} else {
|
||
|
fullName := fmt.Sprintf("%v", vv.Desc.Message().FullName())
|
||
|
if strings.HasPrefix(fullName, "google.protobuf") && strings.HasSuffix(fullName, "Value") {
|
||
|
isRequired = "?:"
|
||
|
}
|
||
|
typ = append(typ, getType(fmt.Sprintf("%v", vv.Desc.Message().Name())))
|
||
|
//}
|
||
|
} else {
|
||
|
typ = append(typ, getType(vv.Desc.Kind().String()))
|
||
|
}
|
||
|
|
||
|
//vtype := fmt.Sprintf("%v", vv.Desc.Kind())
|
||
|
//if vtype == "bytes" {
|
||
|
// typ = append(typ, "Uint8Array")
|
||
|
//} else if vtype == "message" {
|
||
|
// if vv.Desc.IsMap() {
|
||
|
// typ = append(typ, fmt.Sprintf(`{ [key: %s]: %s }`, fields.Get(0).Kind().String(), fields.Get(1).Kind().String()))
|
||
|
// } else {
|
||
|
//
|
||
|
// }
|
||
|
//
|
||
|
// //for _,v:=range vv.Desc.Message().Fields(){
|
||
|
// //
|
||
|
// //}
|
||
|
// //t.P()
|
||
|
// //typ = append(typ, "Uint8Array")
|
||
|
//} else {
|
||
|
// typ = append(typ, vtype)
|
||
|
//}
|
||
|
if vv.Desc.IsList() {
|
||
|
typ = append(typ, "[]")
|
||
|
}
|
||
|
//t.P(vv.Desc)
|
||
|
t.P(` `, vv.Desc.JSONName(), isRequired, ` `, strings.Join(typ, ""))
|
||
|
}
|
||
|
t.P(`}`)
|
||
|
t.P()
|
||
|
}
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
func getType(key string) string {
|
||
|
mps := map[string]string{
|
||
|
"int32": "number",
|
||
|
"bytes": "Uint8Array",
|
||
|
"Int32Value": "number",
|
||
|
"StringValue": "string",
|
||
|
"bool": "boolean",
|
||
|
}
|
||
|
if v, ok := mps[key]; ok {
|
||
|
return v
|
||
|
}
|
||
|
return key
|
||
|
}
|