702 lines
21 KiB
Go
702 lines
21 KiB
Go
|
// Copyright 2020-2021 Buf Technologies, Inc.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
package protosource
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"google.golang.org/protobuf/types/descriptorpb"
|
||
|
|
)
|
||
|
|
|
||
|
|
type file struct {
|
||
|
|
FileInfo
|
||
|
|
descriptor
|
||
|
|
|
||
|
|
fileDescriptorProto *descriptorpb.FileDescriptorProto
|
||
|
|
syntax Syntax
|
||
|
|
fileImports []FileImport
|
||
|
|
messages []Message
|
||
|
|
enums []Enum
|
||
|
|
services []Service
|
||
|
|
optimizeMode FileOptionsOptimizeMode
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) Syntax() Syntax {
|
||
|
|
return f.syntax
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) Package() string {
|
||
|
|
return f.fileDescriptorProto.GetPackage()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) FileImports() []FileImport {
|
||
|
|
return f.fileImports
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) Messages() []Message {
|
||
|
|
return f.messages
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) Enums() []Enum {
|
||
|
|
return f.enums
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) Services() []Service {
|
||
|
|
return f.services
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) CsharpNamespace() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetCsharpNamespace()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) GoPackage() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetGoPackage()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaMultipleFiles() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetJavaMultipleFiles()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaOuterClassname() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetJavaOuterClassname()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaPackage() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetJavaPackage()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaStringCheckUtf8() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetJavaStringCheckUtf8()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) ObjcClassPrefix() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetObjcClassPrefix()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpClassPrefix() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetPhpClassPrefix()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpNamespace() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetPhpNamespace()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpMetadataNamespace() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetPhpMetadataNamespace()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) RubyPackage() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetRubyPackage()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) SwiftPrefix() string {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetSwiftPrefix()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) OptimizeFor() FileOptionsOptimizeMode {
|
||
|
|
return f.optimizeMode
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) CcGenericServices() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetCcGenericServices()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaGenericServices() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetJavaGenericServices()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PyGenericServices() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetPyGenericServices()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpGenericServices() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetPhpGenericServices()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) CcEnableArenas() bool {
|
||
|
|
return f.fileDescriptorProto.GetOptions().GetCcEnableArenas()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PackageLocation() Location {
|
||
|
|
return f.getLocationByPathKey(packagePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) CsharpNamespaceLocation() Location {
|
||
|
|
return f.getLocationByPathKey(csharpNamespacePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) GoPackageLocation() Location {
|
||
|
|
return f.getLocationByPathKey(goPackagePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaMultipleFilesLocation() Location {
|
||
|
|
return f.getLocationByPathKey(javaMultipleFilesPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaOuterClassnameLocation() Location {
|
||
|
|
return f.getLocationByPathKey(javaOuterClassnamePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaPackageLocation() Location {
|
||
|
|
return f.getLocationByPathKey(javaPackagePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaStringCheckUtf8Location() Location {
|
||
|
|
return f.getLocationByPathKey(javaStringCheckUtf8PathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) ObjcClassPrefixLocation() Location {
|
||
|
|
return f.getLocationByPathKey(objcClassPrefixPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpClassPrefixLocation() Location {
|
||
|
|
return f.getLocationByPathKey(phpClassPrefixPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpNamespaceLocation() Location {
|
||
|
|
return f.getLocationByPathKey(phpNamespacePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpMetadataNamespaceLocation() Location {
|
||
|
|
return f.getLocationByPathKey(phpMetadataNamespacePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) RubyPackageLocation() Location {
|
||
|
|
return f.getLocationByPathKey(rubyPackagePathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) SwiftPrefixLocation() Location {
|
||
|
|
return f.getLocationByPathKey(swiftPrefixPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) OptimizeForLocation() Location {
|
||
|
|
return f.getLocationByPathKey(optimizeForPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) CcGenericServicesLocation() Location {
|
||
|
|
return f.getLocationByPathKey(ccGenericServicesPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) JavaGenericServicesLocation() Location {
|
||
|
|
return f.getLocationByPathKey(javaGenericServicesPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PyGenericServicesLocation() Location {
|
||
|
|
return f.getLocationByPathKey(pyGenericServicesPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) PhpGenericServicesLocation() Location {
|
||
|
|
return f.getLocationByPathKey(phpGenericServicesPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) CcEnableArenasLocation() Location {
|
||
|
|
return f.getLocationByPathKey(ccEnableArenasPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) SyntaxLocation() Location {
|
||
|
|
return f.getLocationByPathKey(syntaxPathKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
// does not validation of the fileDescriptorProto - this is assumed to be done elsewhere
|
||
|
|
// does no duplicate checking by name - could just have maps ie importToFileImport, enumNameToEnum, etc
|
||
|
|
func newFile(inputFile InputFile) (*file, error) {
|
||
|
|
f := &file{
|
||
|
|
FileInfo: inputFile,
|
||
|
|
fileDescriptorProto: inputFile.Proto(),
|
||
|
|
}
|
||
|
|
descriptor := newDescriptor(
|
||
|
|
f,
|
||
|
|
newLocationStore(f.fileDescriptorProto.GetSourceCodeInfo().GetLocation()),
|
||
|
|
)
|
||
|
|
f.descriptor = descriptor
|
||
|
|
|
||
|
|
syntaxString := f.fileDescriptorProto.GetSyntax()
|
||
|
|
if syntaxString == "" || syntaxString == "proto2" {
|
||
|
|
f.syntax = SyntaxProto2
|
||
|
|
} else if syntaxString == "proto3" {
|
||
|
|
f.syntax = SyntaxProto3
|
||
|
|
} else {
|
||
|
|
return nil, fmt.Errorf("unknown syntax: %q", syntaxString)
|
||
|
|
}
|
||
|
|
|
||
|
|
for dependencyIndex, dependency := range f.fileDescriptorProto.GetDependency() {
|
||
|
|
fileImport, err := newFileImport(
|
||
|
|
f.descriptor,
|
||
|
|
dependency,
|
||
|
|
getDependencyPath(dependencyIndex),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
f.fileImports = append(f.fileImports, fileImport)
|
||
|
|
}
|
||
|
|
for _, dependencyIndex := range f.fileDescriptorProto.GetPublicDependency() {
|
||
|
|
if len(f.fileImports) <= int(dependencyIndex) {
|
||
|
|
return nil, fmt.Errorf("got dependency index of %d but length of imports is %d", dependencyIndex, len(f.fileImports))
|
||
|
|
}
|
||
|
|
fileImport, ok := f.fileImports[dependencyIndex].(*fileImport)
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("could not cast %T to a *fileImport", f.fileImports[dependencyIndex])
|
||
|
|
}
|
||
|
|
fileImport.setIsPublic()
|
||
|
|
}
|
||
|
|
for _, dependencyIndex := range f.fileDescriptorProto.GetWeakDependency() {
|
||
|
|
if len(f.fileImports) <= int(dependencyIndex) {
|
||
|
|
return nil, fmt.Errorf("got dependency index of %d but length of imports is %d", dependencyIndex, len(f.fileImports))
|
||
|
|
}
|
||
|
|
fileImport, ok := f.fileImports[dependencyIndex].(*fileImport)
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("could not cast %T to a *fileImport", f.fileImports[dependencyIndex])
|
||
|
|
}
|
||
|
|
fileImport.setIsWeak()
|
||
|
|
}
|
||
|
|
for enumIndex, enumDescriptorProto := range f.fileDescriptorProto.GetEnumType() {
|
||
|
|
enum, err := f.populateEnum(
|
||
|
|
enumDescriptorProto,
|
||
|
|
enumIndex,
|
||
|
|
nil,
|
||
|
|
nil,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
f.enums = append(f.enums, enum)
|
||
|
|
}
|
||
|
|
for messageIndex, descriptorProto := range f.fileDescriptorProto.GetMessageType() {
|
||
|
|
message, err := f.populateMessage(
|
||
|
|
descriptorProto,
|
||
|
|
messageIndex,
|
||
|
|
nil,
|
||
|
|
nil,
|
||
|
|
nil,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
f.messages = append(f.messages, message)
|
||
|
|
}
|
||
|
|
for serviceIndex, serviceDescriptorProto := range f.fileDescriptorProto.GetService() {
|
||
|
|
service, err := f.populateService(
|
||
|
|
serviceDescriptorProto,
|
||
|
|
serviceIndex,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
f.services = append(f.services, service)
|
||
|
|
}
|
||
|
|
optimizeMode, err := getFileOptionsOptimizeMode(f.fileDescriptorProto.GetOptions().GetOptimizeFor())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
f.optimizeMode = optimizeMode
|
||
|
|
|
||
|
|
return f, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) populateEnum(
|
||
|
|
enumDescriptorProto *descriptorpb.EnumDescriptorProto,
|
||
|
|
enumIndex int,
|
||
|
|
// all message indexes leading to this enum
|
||
|
|
nestedMessageIndexes []int,
|
||
|
|
// all message names leading to this enum
|
||
|
|
nestedMessageNames []string,
|
||
|
|
) (Enum, error) {
|
||
|
|
enumNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getEnumPath(enumIndex, nestedMessageIndexes...),
|
||
|
|
),
|
||
|
|
enumDescriptorProto.GetName(),
|
||
|
|
getEnumNamePath(enumIndex, nestedMessageIndexes...),
|
||
|
|
nestedMessageNames,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
enum := newEnum(
|
||
|
|
enumNamedDescriptor,
|
||
|
|
enumDescriptorProto.GetOptions().GetAllowAlias(),
|
||
|
|
getEnumAllowAliasPath(enumIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
|
||
|
|
for enumValueIndex, enumValueDescriptorProto := range enumDescriptorProto.GetValue() {
|
||
|
|
enumValueNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getEnumValuePath(enumIndex, enumValueIndex, nestedMessageIndexes...),
|
||
|
|
),
|
||
|
|
enumValueDescriptorProto.GetName(),
|
||
|
|
getEnumValueNamePath(enumIndex, enumValueIndex, nestedMessageIndexes...),
|
||
|
|
append(nestedMessageNames, enum.Name()),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
enumValue := newEnumValue(
|
||
|
|
enumValueNamedDescriptor,
|
||
|
|
enum,
|
||
|
|
int(enumValueDescriptorProto.GetNumber()),
|
||
|
|
getEnumValueNumberPath(enumIndex, enumValueIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
enum.addValue(enumValue)
|
||
|
|
}
|
||
|
|
|
||
|
|
for reservedRangeIndex, reservedRangeDescriptorProto := range enumDescriptorProto.GetReservedRange() {
|
||
|
|
reservedRangeLocationDescriptor := newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getEnumReservedRangePath(enumIndex, reservedRangeIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
reservedEnumRange := newEnumRange(
|
||
|
|
reservedRangeLocationDescriptor,
|
||
|
|
enum,
|
||
|
|
int(reservedRangeDescriptorProto.GetStart()),
|
||
|
|
int(reservedRangeDescriptorProto.GetEnd()),
|
||
|
|
)
|
||
|
|
enum.addReservedEnumRange(reservedEnumRange)
|
||
|
|
}
|
||
|
|
for reservedNameIndex, reservedNameValue := range enumDescriptorProto.GetReservedName() {
|
||
|
|
reservedNameLocationDescriptor := newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getEnumReservedNamePath(enumIndex, reservedNameIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
reservedName, err := newReservedName(
|
||
|
|
reservedNameLocationDescriptor,
|
||
|
|
reservedNameValue,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
enum.addReservedName(reservedName)
|
||
|
|
}
|
||
|
|
return enum, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) populateMessage(
|
||
|
|
descriptorProto *descriptorpb.DescriptorProto,
|
||
|
|
// always stays the same on every recursive call
|
||
|
|
topLevelMessageIndex int,
|
||
|
|
// includes descriptorProto index
|
||
|
|
nestedMessageIndexes []int,
|
||
|
|
// does NOT include descriptorProto.GetName()
|
||
|
|
nestedMessageNames []string,
|
||
|
|
parent Message,
|
||
|
|
) (Message, error) {
|
||
|
|
messageNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessagePath(topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
),
|
||
|
|
descriptorProto.GetName(),
|
||
|
|
getMessageNamePath(topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
nestedMessageNames,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
message := newMessage(
|
||
|
|
messageNamedDescriptor,
|
||
|
|
parent,
|
||
|
|
descriptorProto.GetOptions().GetMapEntry(),
|
||
|
|
descriptorProto.GetOptions().GetMessageSetWireFormat(),
|
||
|
|
descriptorProto.GetOptions().GetNoStandardDescriptorAccessor(),
|
||
|
|
getMessageMessageSetWireFormatPath(topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageNoStandardDescriptorAccessorPath(topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
oneofIndexToOneof := make(map[int]*oneof)
|
||
|
|
for oneofIndex, oneofDescriptorProto := range descriptorProto.GetOneofDecl() {
|
||
|
|
oneofNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessageOneofPath(oneofIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
),
|
||
|
|
oneofDescriptorProto.GetName(),
|
||
|
|
getMessageOneofNamePath(oneofIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
append(nestedMessageNames, message.Name()),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
oneof := newOneof(
|
||
|
|
oneofNamedDescriptor,
|
||
|
|
message,
|
||
|
|
)
|
||
|
|
message.addOneof(oneof)
|
||
|
|
oneofIndexToOneof[oneofIndex] = oneof
|
||
|
|
}
|
||
|
|
for fieldIndex, fieldDescriptorProto := range descriptorProto.GetField() {
|
||
|
|
// TODO: not working for map entries
|
||
|
|
fieldNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessageFieldPath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
),
|
||
|
|
fieldDescriptorProto.GetName(),
|
||
|
|
getMessageFieldNamePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
append(nestedMessageNames, message.Name()),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var packed *bool
|
||
|
|
if fieldDescriptorProto.Options != nil {
|
||
|
|
packed = fieldDescriptorProto.GetOptions().Packed
|
||
|
|
}
|
||
|
|
label, err := getFieldDescriptorProtoLabel(fieldDescriptorProto.GetLabel())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
typ, err := getFieldDescriptorProtoType(fieldDescriptorProto.GetType())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
jsType, err := getFieldOptionsJSType(fieldDescriptorProto.GetOptions().GetJstype())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
cType, err := getFieldOptionsCType(fieldDescriptorProto.GetOptions().GetCtype())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var oneof *oneof
|
||
|
|
var ok bool
|
||
|
|
if fieldDescriptorProto.OneofIndex != nil {
|
||
|
|
oneofIndex := int(*fieldDescriptorProto.OneofIndex)
|
||
|
|
oneof, ok = oneofIndexToOneof[oneofIndex]
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("no oneof for index %d", oneofIndex)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
field := newField(
|
||
|
|
fieldNamedDescriptor,
|
||
|
|
message,
|
||
|
|
int(fieldDescriptorProto.GetNumber()),
|
||
|
|
label,
|
||
|
|
typ,
|
||
|
|
fieldDescriptorProto.GetTypeName(),
|
||
|
|
oneof,
|
||
|
|
fieldDescriptorProto.GetProto3Optional(),
|
||
|
|
fieldDescriptorProto.GetJsonName(),
|
||
|
|
jsType,
|
||
|
|
cType,
|
||
|
|
packed,
|
||
|
|
getMessageFieldNumberPath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageFieldTypePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageFieldTypeNamePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageFieldJSONNamePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageFieldJSTypePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageFieldCTypePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageFieldPackedPath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
message.addField(field)
|
||
|
|
if oneof != nil {
|
||
|
|
oneof.addField(field)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// TODO: is this right?
|
||
|
|
for fieldIndex, fieldDescriptorProto := range descriptorProto.GetExtension() {
|
||
|
|
fieldNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessageExtensionPath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
),
|
||
|
|
fieldDescriptorProto.GetName(),
|
||
|
|
getMessageExtensionNamePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
append(nestedMessageNames, message.Name()),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var packed *bool
|
||
|
|
if fieldDescriptorProto.Options != nil {
|
||
|
|
packed = fieldDescriptorProto.GetOptions().Packed
|
||
|
|
}
|
||
|
|
label, err := getFieldDescriptorProtoLabel(fieldDescriptorProto.GetLabel())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
typ, err := getFieldDescriptorProtoType(fieldDescriptorProto.GetType())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
jsType, err := getFieldOptionsJSType(fieldDescriptorProto.GetOptions().GetJstype())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
cType, err := getFieldOptionsCType(fieldDescriptorProto.GetOptions().GetCtype())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var oneof *oneof
|
||
|
|
var ok bool
|
||
|
|
if fieldDescriptorProto.OneofIndex != nil {
|
||
|
|
oneofIndex := int(*fieldDescriptorProto.OneofIndex)
|
||
|
|
oneof, ok = oneofIndexToOneof[oneofIndex]
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("no oneof for index %d", oneofIndex)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
field := newField(
|
||
|
|
fieldNamedDescriptor,
|
||
|
|
message,
|
||
|
|
int(fieldDescriptorProto.GetNumber()),
|
||
|
|
label,
|
||
|
|
typ,
|
||
|
|
fieldDescriptorProto.GetTypeName(),
|
||
|
|
oneof,
|
||
|
|
fieldDescriptorProto.GetProto3Optional(),
|
||
|
|
fieldDescriptorProto.GetJsonName(),
|
||
|
|
jsType,
|
||
|
|
cType,
|
||
|
|
packed,
|
||
|
|
getMessageExtensionNumberPath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageExtensionTypePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageExtensionTypeNamePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageExtensionJSONNamePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageExtensionJSTypePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageExtensionCTypePath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
getMessageExtensionPackedPath(fieldIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
message.addExtension(field)
|
||
|
|
if oneof != nil {
|
||
|
|
oneof.addField(field)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for reservedRangeIndex, reservedRangeDescriptorProto := range descriptorProto.GetReservedRange() {
|
||
|
|
reservedRangeLocationDescriptor := newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessageReservedRangePath(reservedRangeIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
reservedMessageRange := newMessageRange(
|
||
|
|
reservedRangeLocationDescriptor,
|
||
|
|
message,
|
||
|
|
int(reservedRangeDescriptorProto.GetStart()),
|
||
|
|
int(reservedRangeDescriptorProto.GetEnd()),
|
||
|
|
)
|
||
|
|
message.addReservedMessageRange(reservedMessageRange)
|
||
|
|
}
|
||
|
|
for reservedNameIndex, reservedNameValue := range descriptorProto.GetReservedName() {
|
||
|
|
reservedNameLocationDescriptor := newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessageReservedNamePath(reservedNameIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
reservedName, err := newReservedName(
|
||
|
|
reservedNameLocationDescriptor,
|
||
|
|
reservedNameValue,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
message.addReservedName(reservedName)
|
||
|
|
}
|
||
|
|
for extensionRangeIndex, extensionRangeDescriptorProto := range descriptorProto.GetExtensionRange() {
|
||
|
|
extensionRangeLocationDescriptor := newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMessageExtensionRangePath(extensionRangeIndex, topLevelMessageIndex, nestedMessageIndexes...),
|
||
|
|
)
|
||
|
|
extensionMessageRange := newMessageRange(
|
||
|
|
extensionRangeLocationDescriptor,
|
||
|
|
message,
|
||
|
|
int(extensionRangeDescriptorProto.GetStart()),
|
||
|
|
int(extensionRangeDescriptorProto.GetEnd()),
|
||
|
|
)
|
||
|
|
message.addExtensionMessageRange(extensionMessageRange)
|
||
|
|
}
|
||
|
|
for enumIndex, enumDescriptorProto := range descriptorProto.GetEnumType() {
|
||
|
|
nestedEnum, err := f.populateEnum(
|
||
|
|
enumDescriptorProto,
|
||
|
|
enumIndex,
|
||
|
|
// this is all of the message indexes including this one
|
||
|
|
// TODO we should refactor get.*Path messages to be more consistent
|
||
|
|
append([]int{topLevelMessageIndex}, nestedMessageIndexes...),
|
||
|
|
append(nestedMessageNames, message.Name()),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
message.addNestedEnum(nestedEnum)
|
||
|
|
}
|
||
|
|
for nestedMessageIndex, nestedMessageDescriptorProto := range descriptorProto.GetNestedType() {
|
||
|
|
nestedMessage, err := f.populateMessage(
|
||
|
|
nestedMessageDescriptorProto,
|
||
|
|
topLevelMessageIndex,
|
||
|
|
append(nestedMessageIndexes, nestedMessageIndex),
|
||
|
|
append(nestedMessageNames, message.Name()),
|
||
|
|
message,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
message.addNestedMessage(nestedMessage)
|
||
|
|
}
|
||
|
|
return message, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *file) populateService(
|
||
|
|
serviceDescriptorProto *descriptorpb.ServiceDescriptorProto,
|
||
|
|
serviceIndex int,
|
||
|
|
) (Service, error) {
|
||
|
|
serviceNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getServicePath(serviceIndex),
|
||
|
|
),
|
||
|
|
serviceDescriptorProto.GetName(),
|
||
|
|
getServiceNamePath(serviceIndex),
|
||
|
|
nil,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
service := newService(
|
||
|
|
serviceNamedDescriptor,
|
||
|
|
)
|
||
|
|
for methodIndex, methodDescriptorProto := range serviceDescriptorProto.GetMethod() {
|
||
|
|
methodNamedDescriptor, err := newNamedDescriptor(
|
||
|
|
newLocationDescriptor(
|
||
|
|
f.descriptor,
|
||
|
|
getMethodPath(serviceIndex, methodIndex),
|
||
|
|
),
|
||
|
|
methodDescriptorProto.GetName(),
|
||
|
|
getMethodNamePath(serviceIndex, methodIndex),
|
||
|
|
[]string{service.Name()},
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
idempotencyLevel, err := getMethodOptionsIdempotencyLevel(methodDescriptorProto.GetOptions().GetIdempotencyLevel())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
method, err := newMethod(
|
||
|
|
methodNamedDescriptor,
|
||
|
|
service,
|
||
|
|
methodDescriptorProto.GetInputType(),
|
||
|
|
methodDescriptorProto.GetOutputType(),
|
||
|
|
methodDescriptorProto.GetClientStreaming(),
|
||
|
|
methodDescriptorProto.GetServerStreaming(),
|
||
|
|
getMethodInputTypePath(serviceIndex, methodIndex),
|
||
|
|
getMethodOutputTypePath(serviceIndex, methodIndex),
|
||
|
|
idempotencyLevel,
|
||
|
|
getMethodIdempotencyLevelPath(serviceIndex, methodIndex),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
service.addMethod(method)
|
||
|
|
}
|
||
|
|
return service, nil
|
||
|
|
}
|