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.
28 lines
706 B
28 lines
706 B
6 years ago
|
package naming
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path"
|
||
|
|
|
||
|
|
"github.com/golang/protobuf/protoc-gen-go/descriptor"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GoFileName returns the output name for the generated Go file.
|
||
|
|
func GoFileName(f *descriptor.FileDescriptorProto, suffix string) string {
|
||
|
|
name := *f.Name
|
||
|
|
if ext := path.Ext(name); ext == ".pb" || ext == ".proto" || ext == ".protodevel" {
|
||
|
|
name = name[:len(name)-len(ext)]
|
||
|
|
}
|
||
|
|
name += suffix
|
||
|
|
|
||
|
|
// Does the file have a "go_package" option? If it does, it may override the
|
||
|
|
// filename.
|
||
|
|
if impPath, _, ok := goPackageOption(f); ok && impPath != "" {
|
||
|
|
// Replace the existing dirname with the declared import path.
|
||
|
|
_, name = path.Split(name)
|
||
|
|
name = path.Join(impPath, name)
|
||
|
|
return name
|
||
|
|
}
|
||
|
|
|
||
|
|
return name
|
||
|
|
}
|