blob: dcdc2202fada8b87c063dbc721c83f44a728dea4 [file] [log] [blame]
mpagenkoaf801632020-07-03 10:00:42 +00001// Copyright 2010 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
Holger Hildebrandtfa074992020-03-27 15:42:06 +00004
5package proto
6
Holger Hildebrandtfa074992020-03-27 15:42:06 +00007import (
8 "fmt"
Holger Hildebrandtfa074992020-03-27 15:42:06 +00009 "reflect"
Holger Hildebrandtfa074992020-03-27 15:42:06 +000010 "strconv"
11 "strings"
12 "sync"
mpagenkoaf801632020-07-03 10:00:42 +000013
14 "google.golang.org/protobuf/reflect/protoreflect"
15 "google.golang.org/protobuf/runtime/protoimpl"
Holger Hildebrandtfa074992020-03-27 15:42:06 +000016)
17
mpagenkoaf801632020-07-03 10:00:42 +000018// StructProperties represents protocol buffer type information for a
19// generated protobuf message in the open-struct API.
20//
21// Deprecated: Do not use.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000022type StructProperties struct {
mpagenkoaf801632020-07-03 10:00:42 +000023 // Prop are the properties for each field.
24 //
25 // Fields belonging to a oneof are stored in OneofTypes instead, with a
26 // single Properties representing the parent oneof held here.
27 //
28 // The order of Prop matches the order of fields in the Go struct.
29 // Struct fields that are not related to protobufs have a "XXX_" prefix
30 // in the Properties.Name and must be ignored by the user.
31 Prop []*Properties
Holger Hildebrandtfa074992020-03-27 15:42:06 +000032
33 // OneofTypes contains information about the oneof fields in this message.
mpagenkoaf801632020-07-03 10:00:42 +000034 // It is keyed by the protobuf field name.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000035 OneofTypes map[string]*OneofProperties
36}
37
mpagenkoaf801632020-07-03 10:00:42 +000038// Properties represents the type information for a protobuf message field.
39//
40// Deprecated: Do not use.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000041type Properties struct {
mpagenkoaf801632020-07-03 10:00:42 +000042 // Name is a placeholder name with little meaningful semantic value.
43 // If the name has an "XXX_" prefix, the entire Properties must be ignored.
44 Name string
45 // OrigName is the protobuf field name or oneof name.
46 OrigName string
47 // JSONName is the JSON name for the protobuf field.
48 JSONName string
49 // Enum is a placeholder name for enums.
50 // For historical reasons, this is neither the Go name for the enum,
51 // nor the protobuf name for the enum.
52 Enum string // Deprecated: Do not use.
53 // Weak contains the full name of the weakly referenced message.
54 Weak string
55 // Wire is a string representation of the wire type.
56 Wire string
57 // WireType is the protobuf wire type for the field.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000058 WireType int
mpagenkoaf801632020-07-03 10:00:42 +000059 // Tag is the protobuf field number.
60 Tag int
61 // Required reports whether this is a required field.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000062 Required bool
mpagenkoaf801632020-07-03 10:00:42 +000063 // Optional reports whether this is a optional field.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000064 Optional bool
mpagenkoaf801632020-07-03 10:00:42 +000065 // Repeated reports whether this is a repeated field.
Holger Hildebrandtfa074992020-03-27 15:42:06 +000066 Repeated bool
mpagenkoaf801632020-07-03 10:00:42 +000067 // Packed reports whether this is a packed repeated field of scalars.
68 Packed bool
69 // Proto3 reports whether this field operates under the proto3 syntax.
70 Proto3 bool
71 // Oneof reports whether this field belongs within a oneof.
72 Oneof bool
Holger Hildebrandtfa074992020-03-27 15:42:06 +000073
mpagenkoaf801632020-07-03 10:00:42 +000074 // Default is the default value in string form.
75 Default string
76 // HasDefault reports whether the field has a default value.
77 HasDefault bool
Holger Hildebrandtfa074992020-03-27 15:42:06 +000078
mpagenkoaf801632020-07-03 10:00:42 +000079 // MapKeyProp is the properties for the key field for a map field.
80 MapKeyProp *Properties
81 // MapValProp is the properties for the value field for a map field.
82 MapValProp *Properties
83}
Holger Hildebrandtfa074992020-03-27 15:42:06 +000084
mpagenkoaf801632020-07-03 10:00:42 +000085// OneofProperties represents the type information for a protobuf oneof.
86//
87// Deprecated: Do not use.
88type OneofProperties struct {
89 // Type is a pointer to the generated wrapper type for the field value.
90 // This is nil for messages that are not in the open-struct API.
91 Type reflect.Type
92 // Field is the index into StructProperties.Prop for the containing oneof.
93 Field int
94 // Prop is the properties for the field.
95 Prop *Properties
Holger Hildebrandtfa074992020-03-27 15:42:06 +000096}
97
98// String formats the properties in the protobuf struct field tag style.
99func (p *Properties) String() string {
100 s := p.Wire
mpagenkoaf801632020-07-03 10:00:42 +0000101 s += "," + strconv.Itoa(p.Tag)
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000102 if p.Required {
103 s += ",req"
104 }
105 if p.Optional {
106 s += ",opt"
107 }
108 if p.Repeated {
109 s += ",rep"
110 }
111 if p.Packed {
112 s += ",packed"
113 }
114 s += ",name=" + p.OrigName
mpagenkoaf801632020-07-03 10:00:42 +0000115 if p.JSONName != "" {
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000116 s += ",json=" + p.JSONName
117 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000118 if len(p.Enum) > 0 {
119 s += ",enum=" + p.Enum
120 }
mpagenkoaf801632020-07-03 10:00:42 +0000121 if len(p.Weak) > 0 {
122 s += ",weak=" + p.Weak
123 }
124 if p.Proto3 {
125 s += ",proto3"
126 }
127 if p.Oneof {
128 s += ",oneof"
129 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000130 if p.HasDefault {
131 s += ",def=" + p.Default
132 }
133 return s
134}
135
136// Parse populates p by parsing a string in the protobuf struct field tag style.
mpagenkoaf801632020-07-03 10:00:42 +0000137func (p *Properties) Parse(tag string) {
138 // For example: "bytes,49,opt,name=foo,def=hello!"
139 for len(tag) > 0 {
140 i := strings.IndexByte(tag, ',')
141 if i < 0 {
142 i = len(tag)
143 }
144 switch s := tag[:i]; {
145 case strings.HasPrefix(s, "name="):
146 p.OrigName = s[len("name="):]
147 case strings.HasPrefix(s, "json="):
148 p.JSONName = s[len("json="):]
149 case strings.HasPrefix(s, "enum="):
150 p.Enum = s[len("enum="):]
151 case strings.HasPrefix(s, "weak="):
152 p.Weak = s[len("weak="):]
153 case strings.Trim(s, "0123456789") == "":
154 n, _ := strconv.ParseUint(s, 10, 32)
155 p.Tag = int(n)
156 case s == "opt":
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000157 p.Optional = true
mpagenkoaf801632020-07-03 10:00:42 +0000158 case s == "req":
159 p.Required = true
160 case s == "rep":
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000161 p.Repeated = true
mpagenkoaf801632020-07-03 10:00:42 +0000162 case s == "varint" || s == "zigzag32" || s == "zigzag64":
163 p.Wire = s
164 p.WireType = WireVarint
165 case s == "fixed32":
166 p.Wire = s
167 p.WireType = WireFixed32
168 case s == "fixed64":
169 p.Wire = s
170 p.WireType = WireFixed64
171 case s == "bytes":
172 p.Wire = s
173 p.WireType = WireBytes
174 case s == "group":
175 p.Wire = s
176 p.WireType = WireStartGroup
177 case s == "packed":
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000178 p.Packed = true
mpagenkoaf801632020-07-03 10:00:42 +0000179 case s == "proto3":
180 p.Proto3 = true
181 case s == "oneof":
182 p.Oneof = true
183 case strings.HasPrefix(s, "def="):
184 // The default tag is special in that everything afterwards is the
185 // default regardless of the presence of commas.
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000186 p.HasDefault = true
mpagenkoaf801632020-07-03 10:00:42 +0000187 p.Default, i = tag[len("def="):], len(tag)
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000188 }
mpagenkoaf801632020-07-03 10:00:42 +0000189 tag = strings.TrimPrefix(tag[i:], ",")
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000190 }
191}
192
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000193// Init populates the properties from a protocol buffer struct tag.
mpagenkoaf801632020-07-03 10:00:42 +0000194//
195// Deprecated: Do not use.
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000196func (p *Properties) Init(typ reflect.Type, name, tag string, f *reflect.StructField) {
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000197 p.Name = name
198 p.OrigName = name
199 if tag == "" {
200 return
201 }
202 p.Parse(tag)
mpagenkoaf801632020-07-03 10:00:42 +0000203
204 if typ != nil && typ.Kind() == reflect.Map {
205 p.MapKeyProp = new(Properties)
206 p.MapKeyProp.Init(nil, "Key", f.Tag.Get("protobuf_key"), nil)
207 p.MapValProp = new(Properties)
208 p.MapValProp.Init(nil, "Value", f.Tag.Get("protobuf_val"), nil)
209 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000210}
211
mpagenkoaf801632020-07-03 10:00:42 +0000212var propertiesCache sync.Map // map[reflect.Type]*StructProperties
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000213
mpagenkoaf801632020-07-03 10:00:42 +0000214// GetProperties returns the list of properties for the type represented by t,
215// which must be a generated protocol buffer message in the open-struct API,
216// where protobuf message fields are represented by exported Go struct fields.
217//
218// Deprecated: Use protobuf reflection instead.
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000219func GetProperties(t reflect.Type) *StructProperties {
mpagenkoaf801632020-07-03 10:00:42 +0000220 if p, ok := propertiesCache.Load(t); ok {
221 return p.(*StructProperties)
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000222 }
mpagenkoaf801632020-07-03 10:00:42 +0000223 p, _ := propertiesCache.LoadOrStore(t, newProperties(t))
224 return p.(*StructProperties)
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000225}
226
mpagenkoaf801632020-07-03 10:00:42 +0000227func newProperties(t reflect.Type) *StructProperties {
228 if t.Kind() != reflect.Struct {
229 panic(fmt.Sprintf("%v is not a generated message in the open-struct API", t))
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000230 }
231
mpagenkoaf801632020-07-03 10:00:42 +0000232 var hasOneof bool
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000233 prop := new(StructProperties)
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000234
mpagenkoaf801632020-07-03 10:00:42 +0000235 // Construct a list of properties for each field in the struct.
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000236 for i := 0; i < t.NumField(); i++ {
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000237 p := new(Properties)
mpagenkoaf801632020-07-03 10:00:42 +0000238 f := t.Field(i)
239 tagField := f.Tag.Get("protobuf")
240 p.Init(f.Type, f.Name, tagField, &f)
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000241
mpagenkoaf801632020-07-03 10:00:42 +0000242 tagOneof := f.Tag.Get("protobuf_oneof")
243 if tagOneof != "" {
244 hasOneof = true
245 p.OrigName = tagOneof
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000246 }
mpagenkoaf801632020-07-03 10:00:42 +0000247
248 // Rename unrelated struct fields with the "XXX_" prefix since so much
249 // user code simply checks for this to exclude special fields.
250 if tagField == "" && tagOneof == "" && !strings.HasPrefix(p.Name, "XXX_") {
251 p.Name = "XXX_" + p.Name
252 p.OrigName = "XXX_" + p.OrigName
253 } else if p.Weak != "" {
254 p.Name = p.OrigName // avoid possible "XXX_" prefix on weak field
255 }
256
257 prop.Prop = append(prop.Prop, p)
258 }
259
260 // Construct a mapping of oneof field names to properties.
261 if hasOneof {
262 var oneofWrappers []interface{}
263 if fn, ok := reflect.PtrTo(t).MethodByName("XXX_OneofFuncs"); ok {
264 oneofWrappers = fn.Func.Call([]reflect.Value{reflect.Zero(fn.Type.In(0))})[3].Interface().([]interface{})
265 }
266 if fn, ok := reflect.PtrTo(t).MethodByName("XXX_OneofWrappers"); ok {
267 oneofWrappers = fn.Func.Call([]reflect.Value{reflect.Zero(fn.Type.In(0))})[0].Interface().([]interface{})
268 }
269 if m, ok := reflect.Zero(reflect.PtrTo(t)).Interface().(protoreflect.ProtoMessage); ok {
270 if m, ok := m.ProtoReflect().(interface{ ProtoMessageInfo() *protoimpl.MessageInfo }); ok {
271 oneofWrappers = m.ProtoMessageInfo().OneofWrappers
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000272 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000273 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000274
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000275 prop.OneofTypes = make(map[string]*OneofProperties)
mpagenkoaf801632020-07-03 10:00:42 +0000276 for _, wrapper := range oneofWrappers {
277 p := &OneofProperties{
278 Type: reflect.ValueOf(wrapper).Type(), // *T
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000279 Prop: new(Properties),
280 }
mpagenkoaf801632020-07-03 10:00:42 +0000281 f := p.Type.Elem().Field(0)
282 p.Prop.Name = f.Name
283 p.Prop.Parse(f.Tag.Get("protobuf"))
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000284
mpagenkoaf801632020-07-03 10:00:42 +0000285 // Determine the struct field that contains this oneof.
286 // Each wrapper is assignable to exactly one parent field.
287 var foundOneof bool
288 for i := 0; i < t.NumField() && !foundOneof; i++ {
289 if p.Type.AssignableTo(t.Field(i).Type) {
290 p.Field = i
291 foundOneof = true
292 }
293 }
294 if !foundOneof {
295 panic(fmt.Sprintf("%v is not a generated message in the open-struct API", t))
296 }
297 prop.OneofTypes[p.Prop.OrigName] = p
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000298 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000299 }
Holger Hildebrandtfa074992020-03-27 15:42:06 +0000300
301 return prop
302}
303
mpagenkoaf801632020-07-03 10:00:42 +0000304func (sp *StructProperties) Len() int { return len(sp.Prop) }
305func (sp *StructProperties) Less(i, j int) bool { return false }
306func (sp *StructProperties) Swap(i, j int) { return }