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.
36 lines
659 B
36 lines
659 B
4 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"google.golang.org/protobuf/compiler/protogen"
|
||
|
|
"google.golang.org/protobuf/types/pluginpb"
|
||
|
|
)
|
||
|
|
|
||
|
|
const version = "0.0.1"
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
showVersion := flag.Bool("version", false, "print the version and exit")
|
||
|
|
flag.Parse()
|
||
|
|
if *showVersion {
|
||
|
|
fmt.Printf("protoc-gen-go-http %v\n", version)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
var flags flag.FlagSet
|
||
|
|
|
||
|
|
protogen.Options{
|
||
|
|
ParamFunc: flags.Set,
|
||
|
|
}.Run(func(gen *protogen.Plugin) error {
|
||
|
|
gen.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
|
||
|
|
for _, f := range gen.Files {
|
||
|
|
if !f.Generate {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
generateFile(gen, f)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
}
|