233 lines
9.6 KiB
Go
233 lines
9.6 KiB
Go
|
/*
|
||
|
|
* Copyright 2022 CloudWeGo Authors
|
||
|
|
*
|
||
|
|
* 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.
|
||
|
|
*
|
||
|
|
* Copyright (c) 2018 The Go Authors. All rights reserved.
|
||
|
|
*
|
||
|
|
* Redistribution and use in source and binary forms, with or without
|
||
|
|
* modification, are permitted provided that the following conditions are
|
||
|
|
* met:
|
||
|
|
*
|
||
|
|
* * Redistributions of source code must retain the above copyright
|
||
|
|
* notice, this list of conditions and the following disclaimer.
|
||
|
|
* * Redistributions in binary form must reproduce the above
|
||
|
|
* copyright notice, this list of conditions and the following disclaimer
|
||
|
|
* in the documentation and/or other materials provided with the
|
||
|
|
* distribution.
|
||
|
|
* * Neither the name of Google Inc. nor the names of its
|
||
|
|
* contributors may be used to endorse or promote products derived from
|
||
|
|
* this software without specific prior written permission.
|
||
|
|
*
|
||
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
*
|
||
|
|
* This file may have been modified by CloudWeGo authors. All CloudWeGo
|
||
|
|
* Modifications are Copyright 2022 CloudWeGo Authors.
|
||
|
|
*/
|
||
|
|
|
||
|
|
package protobuf
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
"unicode"
|
||
|
|
"unicode/utf8"
|
||
|
|
_ "unsafe"
|
||
|
|
|
||
|
|
"google.golang.org/protobuf/compiler/protogen"
|
||
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Field numbers for google.protobuf.FileDescriptorProto.
|
||
|
|
const (
|
||
|
|
FileDescriptorProto_Name_field_number protoreflect.FieldNumber = 1
|
||
|
|
FileDescriptorProto_Package_field_number protoreflect.FieldNumber = 2
|
||
|
|
FileDescriptorProto_Dependency_field_number protoreflect.FieldNumber = 3
|
||
|
|
FileDescriptorProto_PublicDependency_field_number protoreflect.FieldNumber = 10
|
||
|
|
FileDescriptorProto_WeakDependency_field_number protoreflect.FieldNumber = 11
|
||
|
|
FileDescriptorProto_MessageType_field_number protoreflect.FieldNumber = 4
|
||
|
|
FileDescriptorProto_EnumType_field_number protoreflect.FieldNumber = 5
|
||
|
|
FileDescriptorProto_Service_field_number protoreflect.FieldNumber = 6
|
||
|
|
FileDescriptorProto_Extension_field_number protoreflect.FieldNumber = 7
|
||
|
|
FileDescriptorProto_Options_field_number protoreflect.FieldNumber = 8
|
||
|
|
FileDescriptorProto_SourceCodeInfo_field_number protoreflect.FieldNumber = 9
|
||
|
|
FileDescriptorProto_Syntax_field_number protoreflect.FieldNumber = 12
|
||
|
|
)
|
||
|
|
|
||
|
|
const WeakFieldPrefix_goname = "XXX_weak_"
|
||
|
|
|
||
|
|
type fileInfo struct {
|
||
|
|
*protogen.File
|
||
|
|
|
||
|
|
allEnums []*enumInfo
|
||
|
|
allMessages []*messageInfo
|
||
|
|
allExtensions []*extensionInfo
|
||
|
|
|
||
|
|
allEnumsByPtr map[*enumInfo]int // value is index into allEnums
|
||
|
|
allMessagesByPtr map[*messageInfo]int // value is index into allMessages
|
||
|
|
allMessageFieldsByPtr map[*messageInfo]*structFields
|
||
|
|
|
||
|
|
// needRawDesc specifies whether the generator should emit logic to provide
|
||
|
|
// the legacy raw descriptor in GZIP'd form.
|
||
|
|
// This is updated by enum and message generation logic as necessary,
|
||
|
|
// and checked at the end of file generation.
|
||
|
|
needRawDesc bool
|
||
|
|
}
|
||
|
|
|
||
|
|
type enumInfo struct {
|
||
|
|
*protogen.Enum
|
||
|
|
|
||
|
|
genJSONMethod bool
|
||
|
|
genRawDescMethod bool
|
||
|
|
}
|
||
|
|
|
||
|
|
type messageInfo struct {
|
||
|
|
*protogen.Message
|
||
|
|
|
||
|
|
genRawDescMethod bool
|
||
|
|
genExtRangeMethod bool
|
||
|
|
|
||
|
|
isTracked bool
|
||
|
|
hasWeak bool
|
||
|
|
}
|
||
|
|
|
||
|
|
type extensionInfo struct {
|
||
|
|
*protogen.Extension
|
||
|
|
}
|
||
|
|
|
||
|
|
type structFields struct {
|
||
|
|
count int
|
||
|
|
unexported map[int]string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (sf *structFields) append(name string) {
|
||
|
|
if r, _ := utf8.DecodeRuneInString(name); !unicode.IsUpper(r) {
|
||
|
|
if sf.unexported == nil {
|
||
|
|
sf.unexported = make(map[int]string)
|
||
|
|
}
|
||
|
|
sf.unexported[sf.count] = name
|
||
|
|
}
|
||
|
|
sf.count++
|
||
|
|
}
|
||
|
|
|
||
|
|
type structTags [][2]string
|
||
|
|
|
||
|
|
func (tags structTags) String() string {
|
||
|
|
if len(tags) == 0 {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
var ss []string
|
||
|
|
for _, tag := range tags {
|
||
|
|
// NOTE: When quoting the value, we need to make sure the backtick
|
||
|
|
// character does not appear. Convert all cases to the escaped hex form.
|
||
|
|
key := tag[0]
|
||
|
|
val := strings.Replace(strconv.Quote(tag[1]), "`", `\x60`, -1)
|
||
|
|
ss = append(ss, fmt.Sprintf("%s:%s", key, val))
|
||
|
|
}
|
||
|
|
return "`" + strings.Join(ss, " ") + "`"
|
||
|
|
}
|
||
|
|
|
||
|
|
type goImportPath interface {
|
||
|
|
String() string
|
||
|
|
Ident(string) protogen.GoIdent
|
||
|
|
}
|
||
|
|
|
||
|
|
type trailingComment protogen.Comments
|
||
|
|
|
||
|
|
func (c trailingComment) String() string {
|
||
|
|
s := strings.TrimSuffix(protogen.Comments(c).String(), "\n")
|
||
|
|
if strings.Contains(s, "\n") {
|
||
|
|
// We don't support multi-lined trailing comments as it is unclear
|
||
|
|
// how to best render them in the generated code.
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
//go:linkname gotrackTags google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.gotrackTags
|
||
|
|
var gotrackTags structTags
|
||
|
|
|
||
|
|
var (
|
||
|
|
protoPackage goImportPath = protogen.GoImportPath("google.golang.org/protobuf/proto")
|
||
|
|
protoifacePackage goImportPath = protogen.GoImportPath("google.golang.org/protobuf/runtime/protoiface")
|
||
|
|
protoimplPackage goImportPath = protogen.GoImportPath("google.golang.org/protobuf/runtime/protoimpl")
|
||
|
|
protojsonPackage goImportPath = protogen.GoImportPath("google.golang.org/protobuf/encoding/protojson")
|
||
|
|
protoreflectPackage goImportPath = protogen.GoImportPath("google.golang.org/protobuf/reflect/protoreflect")
|
||
|
|
protoregistryPackage goImportPath = protogen.GoImportPath("google.golang.org/protobuf/reflect/protoregistry")
|
||
|
|
)
|
||
|
|
|
||
|
|
//go:linkname newFileInfo google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.newFileInfo
|
||
|
|
func newFileInfo(file *protogen.File) *fileInfo
|
||
|
|
|
||
|
|
//go:linkname genPackageKnownComment google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genPackageKnownComment
|
||
|
|
func genPackageKnownComment(f *fileInfo) protogen.Comments
|
||
|
|
|
||
|
|
//go:linkname genStandaloneComments google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genStandaloneComments
|
||
|
|
func genStandaloneComments(g *protogen.GeneratedFile, f *fileInfo, n int32)
|
||
|
|
|
||
|
|
//go:linkname genGeneratedHeader google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genGeneratedHeader
|
||
|
|
func genGeneratedHeader(gen *protogen.Plugin, g *protogen.GeneratedFile, f *fileInfo)
|
||
|
|
|
||
|
|
//go:linkname genImport google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genImport
|
||
|
|
func genImport(gen *protogen.Plugin, g *protogen.GeneratedFile, f *fileInfo, imp protoreflect.FileImport)
|
||
|
|
|
||
|
|
//go:linkname genEnum google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genEnum
|
||
|
|
func genEnum(g *protogen.GeneratedFile, f *fileInfo, e *enumInfo)
|
||
|
|
|
||
|
|
//go:linkname genMessageInternalFields google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genMessageInternalFields
|
||
|
|
func genMessageInternalFields(g *protogen.GeneratedFile, f *fileInfo, m *messageInfo, sf *structFields)
|
||
|
|
|
||
|
|
//go:linkname genExtensions google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genExtensions
|
||
|
|
func genExtensions(g *protogen.GeneratedFile, f *fileInfo)
|
||
|
|
|
||
|
|
//go:linkname genReflectFileDescriptor google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genReflectFileDescriptor
|
||
|
|
func genReflectFileDescriptor(gen *protogen.Plugin, g *protogen.GeneratedFile, f *fileInfo)
|
||
|
|
|
||
|
|
//go:linkname appendDeprecationSuffix google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.appendDeprecationSuffix
|
||
|
|
func appendDeprecationSuffix(prefix protogen.Comments, deprecated bool) protogen.Comments
|
||
|
|
|
||
|
|
//go:linkname genMessageDefaultDecls google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genMessageDefaultDecls
|
||
|
|
func genMessageDefaultDecls(g *protogen.GeneratedFile, f *fileInfo, m *messageInfo)
|
||
|
|
|
||
|
|
//go:linkname genMessageKnownFunctions google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genMessageKnownFunctions
|
||
|
|
func genMessageKnownFunctions(g *protogen.GeneratedFile, f *fileInfo, m *messageInfo)
|
||
|
|
|
||
|
|
//go:linkname genMessageMethods google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genMessageMethods
|
||
|
|
func genMessageMethods(g *protogen.GeneratedFile, f *fileInfo, m *messageInfo)
|
||
|
|
|
||
|
|
//go:linkname genMessageOneofWrapperTypes google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.genMessageOneofWrapperTypes
|
||
|
|
func genMessageOneofWrapperTypes(g *protogen.GeneratedFile, f *fileInfo, m *messageInfo)
|
||
|
|
|
||
|
|
//go:linkname oneofInterfaceName google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.oneofInterfaceName
|
||
|
|
func oneofInterfaceName(oneof *protogen.Oneof) string
|
||
|
|
|
||
|
|
//go:linkname fieldGoType google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.fieldGoType
|
||
|
|
func fieldGoType(g *protogen.GeneratedFile, f *fileInfo, field *protogen.Field) (goType string, pointer bool)
|
||
|
|
|
||
|
|
//go:linkname fieldProtobufTagValue google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.fieldProtobufTagValue
|
||
|
|
func fieldProtobufTagValue(field *protogen.Field) string
|
||
|
|
|
||
|
|
//go:linkname fieldJSONTagValue google.golang.org/protobuf/cmd/protoc-gen-go/internal_gengo.fieldJSONTagValue
|
||
|
|
func fieldJSONTagValue(field *protogen.Field) string
|