google-protoc 3.0.0.alpha.6.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: bb2c13b622a5efdad8178da103cc797481290c8a
4
+ data.tar.gz: 9aa3c5af920e66a5cd7c2e6d49f53fb94e405627
5
+ SHA512:
6
+ metadata.gz: 44d70d837f75845648d0eb9b1e7b0d2249d2049b49be41c62e84533f1d9d7399fa1306eb444985313a58ba85e17a8d914e5bc9a00baa3fc03762daa9f451ac9f
7
+ data.tar.gz: 3d5075aa870a60fed2cf53a5143b931af06e0a7f3ea70fd531bb35c64a26fe3e540259ebf223176dd295adac9429481485e441b52031c531a8e1c368402c4e56
@@ -0,0 +1,1763 @@
1
+ // Protocol Buffers - Google's data interchange format
2
+ // Copyright 2014 Google Inc. All rights reserved.
3
+ // https://developers.google.com/protocol-buffers/
4
+ //
5
+ // Redistribution and use in source and binary forms, with or without
6
+ // modification, are permitted provided that the following conditions are
7
+ // met:
8
+ //
9
+ // * Redistributions of source code must retain the above copyright
10
+ // notice, this list of conditions and the following disclaimer.
11
+ // * Redistributions in binary form must reproduce the above
12
+ // copyright notice, this list of conditions and the following disclaimer
13
+ // in the documentation and/or other materials provided with the
14
+ // distribution.
15
+ // * Neither the name of Google Inc. nor the names of its
16
+ // contributors may be used to endorse or promote products derived from
17
+ // this software without specific prior written permission.
18
+ //
19
+ // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20
+ // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21
+ // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22
+ // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23
+ // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24
+ // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25
+ // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26
+ // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27
+ // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28
+ // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29
+ // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
+
31
+ #include "protobuf.h"
32
+
33
+ // -----------------------------------------------------------------------------
34
+ // Common utilities.
35
+ // -----------------------------------------------------------------------------
36
+
37
+ static const char* get_str(VALUE str) {
38
+ Check_Type(str, T_STRING);
39
+ return RSTRING_PTR(str);
40
+ }
41
+
42
+ static VALUE rb_str_maybe_null(const char* s) {
43
+ if (s == NULL) {
44
+ s = "";
45
+ }
46
+ return rb_str_new2(s);
47
+ }
48
+
49
+ static upb_def* check_notfrozen(const upb_def* def) {
50
+ if (upb_def_isfrozen(def)) {
51
+ rb_raise(rb_eRuntimeError,
52
+ "Attempt to modify a frozen descriptor. Once descriptors are "
53
+ "added to the descriptor pool, they may not be modified.");
54
+ }
55
+ return (upb_def*)def;
56
+ }
57
+
58
+ static upb_msgdef* check_msg_notfrozen(const upb_msgdef* def) {
59
+ return upb_downcast_msgdef_mutable(check_notfrozen((const upb_def*)def));
60
+ }
61
+
62
+ static upb_fielddef* check_field_notfrozen(const upb_fielddef* def) {
63
+ return upb_downcast_fielddef_mutable(check_notfrozen((const upb_def*)def));
64
+ }
65
+
66
+ static upb_oneofdef* check_oneof_notfrozen(const upb_oneofdef* def) {
67
+ return (upb_oneofdef*)check_notfrozen((const upb_def*)def);
68
+ }
69
+
70
+ static upb_enumdef* check_enum_notfrozen(const upb_enumdef* def) {
71
+ return (upb_enumdef*)check_notfrozen((const upb_def*)def);
72
+ }
73
+
74
+ // -----------------------------------------------------------------------------
75
+ // DescriptorPool.
76
+ // -----------------------------------------------------------------------------
77
+
78
+ #define DEFINE_CLASS(name, string_name) \
79
+ VALUE c ## name; \
80
+ const rb_data_type_t _ ## name ## _type = { \
81
+ string_name, \
82
+ { name ## _mark, name ## _free, NULL }, \
83
+ }; \
84
+ name* ruby_to_ ## name(VALUE val) { \
85
+ name* ret; \
86
+ TypedData_Get_Struct(val, name, &_ ## name ## _type, ret); \
87
+ return ret; \
88
+ } \
89
+
90
+ #define DEFINE_SELF(type, var, rb_var) \
91
+ type* var = ruby_to_ ## type(rb_var)
92
+
93
+ // Global singleton DescriptorPool. The user is free to create others, but this
94
+ // is used by generated code.
95
+ VALUE generated_pool;
96
+
97
+ DEFINE_CLASS(DescriptorPool, "Google::Protobuf::DescriptorPool");
98
+
99
+ void DescriptorPool_mark(void* _self) {
100
+ }
101
+
102
+ void DescriptorPool_free(void* _self) {
103
+ DescriptorPool* self = _self;
104
+ upb_symtab_unref(self->symtab, &self->symtab);
105
+ xfree(self);
106
+ }
107
+
108
+ /*
109
+ * call-seq:
110
+ * DescriptorPool.new => pool
111
+ *
112
+ * Creates a new, empty, descriptor pool.
113
+ */
114
+ VALUE DescriptorPool_alloc(VALUE klass) {
115
+ DescriptorPool* self = ALLOC(DescriptorPool);
116
+ self->symtab = upb_symtab_new(&self->symtab);
117
+ return TypedData_Wrap_Struct(klass, &_DescriptorPool_type, self);
118
+ }
119
+
120
+ void DescriptorPool_register(VALUE module) {
121
+ VALUE klass = rb_define_class_under(
122
+ module, "DescriptorPool", rb_cObject);
123
+ rb_define_alloc_func(klass, DescriptorPool_alloc);
124
+ rb_define_method(klass, "add", DescriptorPool_add, 1);
125
+ rb_define_method(klass, "build", DescriptorPool_build, 0);
126
+ rb_define_method(klass, "lookup", DescriptorPool_lookup, 1);
127
+ rb_define_singleton_method(klass, "generated_pool",
128
+ DescriptorPool_generated_pool, 0);
129
+ cDescriptorPool = klass;
130
+ rb_gc_register_address(&cDescriptorPool);
131
+
132
+ generated_pool = rb_class_new_instance(0, NULL, klass);
133
+ rb_gc_register_address(&generated_pool);
134
+ }
135
+
136
+ static void add_descriptor_to_pool(DescriptorPool* self,
137
+ Descriptor* descriptor) {
138
+ CHECK_UPB(
139
+ upb_symtab_add(self->symtab, (upb_def**)&descriptor->msgdef, 1,
140
+ NULL, &status),
141
+ "Adding Descriptor to DescriptorPool failed");
142
+ }
143
+
144
+ static void add_enumdesc_to_pool(DescriptorPool* self,
145
+ EnumDescriptor* enumdesc) {
146
+ CHECK_UPB(
147
+ upb_symtab_add(self->symtab, (upb_def**)&enumdesc->enumdef, 1,
148
+ NULL, &status),
149
+ "Adding EnumDescriptor to DescriptorPool failed");
150
+ }
151
+
152
+ /*
153
+ * call-seq:
154
+ * DescriptorPool.add(descriptor)
155
+ *
156
+ * Adds the given Descriptor or EnumDescriptor to this pool. All references to
157
+ * other types in a Descriptor's fields must be resolvable within this pool or
158
+ * an exception will be raised.
159
+ */
160
+ VALUE DescriptorPool_add(VALUE _self, VALUE def) {
161
+ DEFINE_SELF(DescriptorPool, self, _self);
162
+ VALUE def_klass = rb_obj_class(def);
163
+ if (def_klass == cDescriptor) {
164
+ add_descriptor_to_pool(self, ruby_to_Descriptor(def));
165
+ } else if (def_klass == cEnumDescriptor) {
166
+ add_enumdesc_to_pool(self, ruby_to_EnumDescriptor(def));
167
+ } else {
168
+ rb_raise(rb_eArgError,
169
+ "Second argument must be a Descriptor or EnumDescriptor.");
170
+ }
171
+ return Qnil;
172
+ }
173
+
174
+ /*
175
+ * call-seq:
176
+ * DescriptorPool.build(&block)
177
+ *
178
+ * Invokes the block with a Builder instance as self. All message and enum types
179
+ * added within the block are committed to the pool atomically, and may refer
180
+ * (co)recursively to each other. The user should call Builder#add_message and
181
+ * Builder#add_enum within the block as appropriate. This is the recommended,
182
+ * idiomatic way to define new message and enum types.
183
+ */
184
+ VALUE DescriptorPool_build(VALUE _self) {
185
+ VALUE ctx = rb_class_new_instance(0, NULL, cBuilder);
186
+ VALUE block = rb_block_proc();
187
+ rb_funcall_with_block(ctx, rb_intern("instance_eval"), 0, NULL, block);
188
+ rb_funcall(ctx, rb_intern("finalize_to_pool"), 1, _self);
189
+ return Qnil;
190
+ }
191
+
192
+ /*
193
+ * call-seq:
194
+ * DescriptorPool.lookup(name) => descriptor
195
+ *
196
+ * Finds a Descriptor or EnumDescriptor by name and returns it, or nil if none
197
+ * exists with the given name.
198
+ */
199
+ VALUE DescriptorPool_lookup(VALUE _self, VALUE name) {
200
+ DEFINE_SELF(DescriptorPool, self, _self);
201
+ const char* name_str = get_str(name);
202
+ const upb_def* def = upb_symtab_lookup(self->symtab, name_str);
203
+ if (!def) {
204
+ return Qnil;
205
+ }
206
+ return get_def_obj(def);
207
+ }
208
+
209
+ /*
210
+ * call-seq:
211
+ * DescriptorPool.generated_pool => descriptor_pool
212
+ *
213
+ * Class method that returns the global DescriptorPool. This is a singleton into
214
+ * which generated-code message and enum types are registered. The user may also
215
+ * register types in this pool for convenience so that they do not have to hold
216
+ * a reference to a private pool instance.
217
+ */
218
+ VALUE DescriptorPool_generated_pool(VALUE _self) {
219
+ return generated_pool;
220
+ }
221
+
222
+ // -----------------------------------------------------------------------------
223
+ // Descriptor.
224
+ // -----------------------------------------------------------------------------
225
+
226
+ DEFINE_CLASS(Descriptor, "Google::Protobuf::Descriptor");
227
+
228
+ void Descriptor_mark(void* _self) {
229
+ Descriptor* self = _self;
230
+ rb_gc_mark(self->klass);
231
+ rb_gc_mark(self->typeclass_references);
232
+ }
233
+
234
+ void Descriptor_free(void* _self) {
235
+ Descriptor* self = _self;
236
+ upb_msgdef_unref(self->msgdef, &self->msgdef);
237
+ if (self->layout) {
238
+ free_layout(self->layout);
239
+ }
240
+ if (self->fill_handlers) {
241
+ upb_handlers_unref(self->fill_handlers, &self->fill_handlers);
242
+ }
243
+ if (self->fill_method) {
244
+ upb_pbdecodermethod_unref(self->fill_method, &self->fill_method);
245
+ }
246
+ if (self->json_fill_method) {
247
+ upb_json_parsermethod_unref(self->json_fill_method,
248
+ &self->json_fill_method);
249
+ }
250
+ if (self->pb_serialize_handlers) {
251
+ upb_handlers_unref(self->pb_serialize_handlers,
252
+ &self->pb_serialize_handlers);
253
+ }
254
+ if (self->json_serialize_handlers) {
255
+ upb_handlers_unref(self->json_serialize_handlers,
256
+ &self->json_serialize_handlers);
257
+ }
258
+ if (self->json_serialize_handlers_preserve) {
259
+ upb_handlers_unref(self->json_serialize_handlers_preserve,
260
+ &self->json_serialize_handlers_preserve);
261
+ }
262
+ xfree(self);
263
+ }
264
+
265
+ /*
266
+ * call-seq:
267
+ * Descriptor.new => descriptor
268
+ *
269
+ * Creates a new, empty, message type descriptor. At a minimum, its name must be
270
+ * set before it is added to a pool. It cannot be used to create messages until
271
+ * it is added to a pool, after which it becomes immutable (as part of a
272
+ * finalization process).
273
+ */
274
+ VALUE Descriptor_alloc(VALUE klass) {
275
+ Descriptor* self = ALLOC(Descriptor);
276
+ VALUE ret = TypedData_Wrap_Struct(klass, &_Descriptor_type, self);
277
+ self->msgdef = upb_msgdef_new(&self->msgdef);
278
+ self->klass = Qnil;
279
+ self->layout = NULL;
280
+ self->fill_handlers = NULL;
281
+ self->fill_method = NULL;
282
+ self->json_fill_method = NULL;
283
+ self->pb_serialize_handlers = NULL;
284
+ self->json_serialize_handlers = NULL;
285
+ self->json_serialize_handlers_preserve = NULL;
286
+ self->typeclass_references = rb_ary_new();
287
+ return ret;
288
+ }
289
+
290
+ void Descriptor_register(VALUE module) {
291
+ VALUE klass = rb_define_class_under(
292
+ module, "Descriptor", rb_cObject);
293
+ rb_define_alloc_func(klass, Descriptor_alloc);
294
+ rb_define_method(klass, "each", Descriptor_each, 0);
295
+ rb_define_method(klass, "lookup", Descriptor_lookup, 1);
296
+ rb_define_method(klass, "add_field", Descriptor_add_field, 1);
297
+ rb_define_method(klass, "add_oneof", Descriptor_add_oneof, 1);
298
+ rb_define_method(klass, "each_oneof", Descriptor_each_oneof, 0);
299
+ rb_define_method(klass, "lookup_oneof", Descriptor_lookup_oneof, 1);
300
+ rb_define_method(klass, "msgclass", Descriptor_msgclass, 0);
301
+ rb_define_method(klass, "name", Descriptor_name, 0);
302
+ rb_define_method(klass, "name=", Descriptor_name_set, 1);
303
+ rb_include_module(klass, rb_mEnumerable);
304
+ cDescriptor = klass;
305
+ rb_gc_register_address(&cDescriptor);
306
+ }
307
+
308
+ /*
309
+ * call-seq:
310
+ * Descriptor.name => name
311
+ *
312
+ * Returns the name of this message type as a fully-qualfied string (e.g.,
313
+ * My.Package.MessageType).
314
+ */
315
+ VALUE Descriptor_name(VALUE _self) {
316
+ DEFINE_SELF(Descriptor, self, _self);
317
+ return rb_str_maybe_null(upb_msgdef_fullname(self->msgdef));
318
+ }
319
+
320
+ /*
321
+ * call-seq:
322
+ * Descriptor.name = name
323
+ *
324
+ * Assigns a name to this message type. The descriptor must not have been added
325
+ * to a pool yet.
326
+ */
327
+ VALUE Descriptor_name_set(VALUE _self, VALUE str) {
328
+ DEFINE_SELF(Descriptor, self, _self);
329
+ upb_msgdef* mut_def = check_msg_notfrozen(self->msgdef);
330
+ const char* name = get_str(str);
331
+ CHECK_UPB(
332
+ upb_msgdef_setfullname(mut_def, name, &status),
333
+ "Error setting Descriptor name");
334
+ return Qnil;
335
+ }
336
+
337
+ /*
338
+ * call-seq:
339
+ * Descriptor.each(&block)
340
+ *
341
+ * Iterates over fields in this message type, yielding to the block on each one.
342
+ */
343
+ VALUE Descriptor_each(VALUE _self) {
344
+ DEFINE_SELF(Descriptor, self, _self);
345
+
346
+ upb_msg_field_iter it;
347
+ for (upb_msg_field_begin(&it, self->msgdef);
348
+ !upb_msg_field_done(&it);
349
+ upb_msg_field_next(&it)) {
350
+ const upb_fielddef* field = upb_msg_iter_field(&it);
351
+ VALUE obj = get_def_obj(field);
352
+ rb_yield(obj);
353
+ }
354
+ return Qnil;
355
+ }
356
+
357
+ /*
358
+ * call-seq:
359
+ * Descriptor.lookup(name) => FieldDescriptor
360
+ *
361
+ * Returns the field descriptor for the field with the given name, if present,
362
+ * or nil if none.
363
+ */
364
+ VALUE Descriptor_lookup(VALUE _self, VALUE name) {
365
+ DEFINE_SELF(Descriptor, self, _self);
366
+ const char* s = get_str(name);
367
+ const upb_fielddef* field = upb_msgdef_ntofz(self->msgdef, s);
368
+ if (field == NULL) {
369
+ return Qnil;
370
+ }
371
+ return get_def_obj(field);
372
+ }
373
+
374
+ /*
375
+ * call-seq:
376
+ * Descriptor.add_field(field) => nil
377
+ *
378
+ * Adds the given FieldDescriptor to this message type. This descriptor must not
379
+ * have been added to a pool yet. Raises an exception if a field with the same
380
+ * name or number already exists. Sub-type references (e.g. for fields of type
381
+ * message) are not resolved at this point.
382
+ */
383
+ VALUE Descriptor_add_field(VALUE _self, VALUE obj) {
384
+ DEFINE_SELF(Descriptor, self, _self);
385
+ upb_msgdef* mut_def = check_msg_notfrozen(self->msgdef);
386
+ FieldDescriptor* def = ruby_to_FieldDescriptor(obj);
387
+ upb_fielddef* mut_field_def = check_field_notfrozen(def->fielddef);
388
+ CHECK_UPB(
389
+ upb_msgdef_addfield(mut_def, mut_field_def, NULL, &status),
390
+ "Adding field to Descriptor failed");
391
+ add_def_obj(def->fielddef, obj);
392
+ return Qnil;
393
+ }
394
+
395
+ /*
396
+ * call-seq:
397
+ * Descriptor.add_oneof(oneof) => nil
398
+ *
399
+ * Adds the given OneofDescriptor to this message type. This descriptor must not
400
+ * have been added to a pool yet. Raises an exception if a oneof with the same
401
+ * name already exists, or if any of the oneof's fields' names or numbers
402
+ * conflict with an existing field in this message type. All fields in the oneof
403
+ * are added to the message descriptor. Sub-type references (e.g. for fields of
404
+ * type message) are not resolved at this point.
405
+ */
406
+ VALUE Descriptor_add_oneof(VALUE _self, VALUE obj) {
407
+ DEFINE_SELF(Descriptor, self, _self);
408
+ upb_msgdef* mut_def = check_msg_notfrozen(self->msgdef);
409
+ OneofDescriptor* def = ruby_to_OneofDescriptor(obj);
410
+ upb_oneofdef* mut_oneof_def = check_oneof_notfrozen(def->oneofdef);
411
+ CHECK_UPB(
412
+ upb_msgdef_addoneof(mut_def, mut_oneof_def, NULL, &status),
413
+ "Adding oneof to Descriptor failed");
414
+ add_def_obj(def->oneofdef, obj);
415
+ return Qnil;
416
+ }
417
+
418
+ /*
419
+ * call-seq:
420
+ * Descriptor.each_oneof(&block) => nil
421
+ *
422
+ * Invokes the given block for each oneof in this message type, passing the
423
+ * corresponding OneofDescriptor.
424
+ */
425
+ VALUE Descriptor_each_oneof(VALUE _self) {
426
+ DEFINE_SELF(Descriptor, self, _self);
427
+
428
+ upb_msg_oneof_iter it;
429
+ for (upb_msg_oneof_begin(&it, self->msgdef);
430
+ !upb_msg_oneof_done(&it);
431
+ upb_msg_oneof_next(&it)) {
432
+ const upb_oneofdef* oneof = upb_msg_iter_oneof(&it);
433
+ VALUE obj = get_def_obj(oneof);
434
+ rb_yield(obj);
435
+ }
436
+ return Qnil;
437
+ }
438
+
439
+ /*
440
+ * call-seq:
441
+ * Descriptor.lookup_oneof(name) => OneofDescriptor
442
+ *
443
+ * Returns the oneof descriptor for the oneof with the given name, if present,
444
+ * or nil if none.
445
+ */
446
+ VALUE Descriptor_lookup_oneof(VALUE _self, VALUE name) {
447
+ DEFINE_SELF(Descriptor, self, _self);
448
+ const char* s = get_str(name);
449
+ const upb_oneofdef* oneof = upb_msgdef_ntooz(self->msgdef, s);
450
+ if (oneof == NULL) {
451
+ return Qnil;
452
+ }
453
+ return get_def_obj(oneof);
454
+ }
455
+
456
+ /*
457
+ * call-seq:
458
+ * Descriptor.msgclass => message_klass
459
+ *
460
+ * Returns the Ruby class created for this message type. Valid only once the
461
+ * message type has been added to a pool.
462
+ */
463
+ VALUE Descriptor_msgclass(VALUE _self) {
464
+ DEFINE_SELF(Descriptor, self, _self);
465
+ if (!upb_def_isfrozen((const upb_def*)self->msgdef)) {
466
+ rb_raise(rb_eRuntimeError,
467
+ "Cannot fetch message class from a Descriptor not yet in a pool.");
468
+ }
469
+ if (self->klass == Qnil) {
470
+ self->klass = build_class_from_descriptor(self);
471
+ }
472
+ return self->klass;
473
+ }
474
+
475
+ // -----------------------------------------------------------------------------
476
+ // FieldDescriptor.
477
+ // -----------------------------------------------------------------------------
478
+
479
+ DEFINE_CLASS(FieldDescriptor, "Google::Protobuf::FieldDescriptor");
480
+
481
+ void FieldDescriptor_mark(void* _self) {
482
+ }
483
+
484
+ void FieldDescriptor_free(void* _self) {
485
+ FieldDescriptor* self = _self;
486
+ upb_fielddef_unref(self->fielddef, &self->fielddef);
487
+ xfree(self);
488
+ }
489
+
490
+ /*
491
+ * call-seq:
492
+ * FieldDescriptor.new => field
493
+ *
494
+ * Returns a new field descriptor. Its name, type, etc. must be set before it is
495
+ * added to a message type.
496
+ */
497
+ VALUE FieldDescriptor_alloc(VALUE klass) {
498
+ FieldDescriptor* self = ALLOC(FieldDescriptor);
499
+ VALUE ret = TypedData_Wrap_Struct(klass, &_FieldDescriptor_type, self);
500
+ upb_fielddef* fielddef = upb_fielddef_new(&self->fielddef);
501
+ upb_fielddef_setpacked(fielddef, false);
502
+ self->fielddef = fielddef;
503
+ return ret;
504
+ }
505
+
506
+ void FieldDescriptor_register(VALUE module) {
507
+ VALUE klass = rb_define_class_under(
508
+ module, "FieldDescriptor", rb_cObject);
509
+ rb_define_alloc_func(klass, FieldDescriptor_alloc);
510
+ rb_define_method(klass, "name", FieldDescriptor_name, 0);
511
+ rb_define_method(klass, "name=", FieldDescriptor_name_set, 1);
512
+ rb_define_method(klass, "type", FieldDescriptor_type, 0);
513
+ rb_define_method(klass, "type=", FieldDescriptor_type_set, 1);
514
+ rb_define_method(klass, "label", FieldDescriptor_label, 0);
515
+ rb_define_method(klass, "label=", FieldDescriptor_label_set, 1);
516
+ rb_define_method(klass, "number", FieldDescriptor_number, 0);
517
+ rb_define_method(klass, "number=", FieldDescriptor_number_set, 1);
518
+ rb_define_method(klass, "submsg_name", FieldDescriptor_submsg_name, 0);
519
+ rb_define_method(klass, "submsg_name=", FieldDescriptor_submsg_name_set, 1);
520
+ rb_define_method(klass, "subtype", FieldDescriptor_subtype, 0);
521
+ rb_define_method(klass, "get", FieldDescriptor_get, 1);
522
+ rb_define_method(klass, "set", FieldDescriptor_set, 2);
523
+ cFieldDescriptor = klass;
524
+ rb_gc_register_address(&cFieldDescriptor);
525
+ }
526
+
527
+ /*
528
+ * call-seq:
529
+ * FieldDescriptor.name => name
530
+ *
531
+ * Returns the name of this field.
532
+ */
533
+ VALUE FieldDescriptor_name(VALUE _self) {
534
+ DEFINE_SELF(FieldDescriptor, self, _self);
535
+ return rb_str_maybe_null(upb_fielddef_name(self->fielddef));
536
+ }
537
+
538
+ /*
539
+ * call-seq:
540
+ * FieldDescriptor.name = name
541
+ *
542
+ * Sets the name of this field. Cannot be called once the containing message
543
+ * type, if any, is added to a pool.
544
+ */
545
+ VALUE FieldDescriptor_name_set(VALUE _self, VALUE str) {
546
+ DEFINE_SELF(FieldDescriptor, self, _self);
547
+ upb_fielddef* mut_def = check_field_notfrozen(self->fielddef);
548
+ const char* name = get_str(str);
549
+ CHECK_UPB(upb_fielddef_setname(mut_def, name, &status),
550
+ "Error setting FieldDescriptor name");
551
+ return Qnil;
552
+ }
553
+
554
+ upb_fieldtype_t ruby_to_fieldtype(VALUE type) {
555
+ if (TYPE(type) != T_SYMBOL) {
556
+ rb_raise(rb_eArgError, "Expected symbol for field type.");
557
+ }
558
+
559
+ #define CONVERT(upb, ruby) \
560
+ if (SYM2ID(type) == rb_intern( # ruby )) { \
561
+ return UPB_TYPE_ ## upb; \
562
+ }
563
+
564
+ CONVERT(FLOAT, float);
565
+ CONVERT(DOUBLE, double);
566
+ CONVERT(BOOL, bool);
567
+ CONVERT(STRING, string);
568
+ CONVERT(BYTES, bytes);
569
+ CONVERT(MESSAGE, message);
570
+ CONVERT(ENUM, enum);
571
+ CONVERT(INT32, int32);
572
+ CONVERT(INT64, int64);
573
+ CONVERT(UINT32, uint32);
574
+ CONVERT(UINT64, uint64);
575
+
576
+ #undef CONVERT
577
+
578
+ rb_raise(rb_eArgError, "Unknown field type.");
579
+ return 0;
580
+ }
581
+
582
+ VALUE fieldtype_to_ruby(upb_fieldtype_t type) {
583
+ switch (type) {
584
+ #define CONVERT(upb, ruby) \
585
+ case UPB_TYPE_ ## upb : return ID2SYM(rb_intern( # ruby ));
586
+ CONVERT(FLOAT, float);
587
+ CONVERT(DOUBLE, double);
588
+ CONVERT(BOOL, bool);
589
+ CONVERT(STRING, string);
590
+ CONVERT(BYTES, bytes);
591
+ CONVERT(MESSAGE, message);
592
+ CONVERT(ENUM, enum);
593
+ CONVERT(INT32, int32);
594
+ CONVERT(INT64, int64);
595
+ CONVERT(UINT32, uint32);
596
+ CONVERT(UINT64, uint64);
597
+ #undef CONVERT
598
+ }
599
+ return Qnil;
600
+ }
601
+
602
+ upb_descriptortype_t ruby_to_descriptortype(VALUE type) {
603
+ if (TYPE(type) != T_SYMBOL) {
604
+ rb_raise(rb_eArgError, "Expected symbol for field type.");
605
+ }
606
+
607
+ #define CONVERT(upb, ruby) \
608
+ if (SYM2ID(type) == rb_intern( # ruby )) { \
609
+ return UPB_DESCRIPTOR_TYPE_ ## upb; \
610
+ }
611
+
612
+ CONVERT(FLOAT, float);
613
+ CONVERT(DOUBLE, double);
614
+ CONVERT(BOOL, bool);
615
+ CONVERT(STRING, string);
616
+ CONVERT(BYTES, bytes);
617
+ CONVERT(MESSAGE, message);
618
+ CONVERT(GROUP, group);
619
+ CONVERT(ENUM, enum);
620
+ CONVERT(INT32, int32);
621
+ CONVERT(INT64, int64);
622
+ CONVERT(UINT32, uint32);
623
+ CONVERT(UINT64, uint64);
624
+ CONVERT(SINT32, sint32);
625
+ CONVERT(SINT64, sint64);
626
+ CONVERT(FIXED32, fixed32);
627
+ CONVERT(FIXED64, fixed64);
628
+ CONVERT(SFIXED32, sfixed32);
629
+ CONVERT(SFIXED64, sfixed64);
630
+
631
+ #undef CONVERT
632
+
633
+ rb_raise(rb_eArgError, "Unknown field type.");
634
+ return 0;
635
+ }
636
+
637
+ VALUE descriptortype_to_ruby(upb_descriptortype_t type) {
638
+ switch (type) {
639
+ #define CONVERT(upb, ruby) \
640
+ case UPB_DESCRIPTOR_TYPE_ ## upb : return ID2SYM(rb_intern( # ruby ));
641
+ CONVERT(FLOAT, float);
642
+ CONVERT(DOUBLE, double);
643
+ CONVERT(BOOL, bool);
644
+ CONVERT(STRING, string);
645
+ CONVERT(BYTES, bytes);
646
+ CONVERT(MESSAGE, message);
647
+ CONVERT(GROUP, group);
648
+ CONVERT(ENUM, enum);
649
+ CONVERT(INT32, int32);
650
+ CONVERT(INT64, int64);
651
+ CONVERT(UINT32, uint32);
652
+ CONVERT(UINT64, uint64);
653
+ CONVERT(SINT32, sint32);
654
+ CONVERT(SINT64, sint64);
655
+ CONVERT(FIXED32, fixed32);
656
+ CONVERT(FIXED64, fixed64);
657
+ CONVERT(SFIXED32, sfixed32);
658
+ CONVERT(SFIXED64, sfixed64);
659
+ #undef CONVERT
660
+ }
661
+ return Qnil;
662
+ }
663
+
664
+ /*
665
+ * call-seq:
666
+ * FieldDescriptor.type => type
667
+ *
668
+ * Returns this field's type, as a Ruby symbol, or nil if not yet set.
669
+ *
670
+ * Valid field types are:
671
+ * :int32, :int64, :uint32, :uint64, :float, :double, :bool, :string,
672
+ * :bytes, :message.
673
+ */
674
+ VALUE FieldDescriptor_type(VALUE _self) {
675
+ DEFINE_SELF(FieldDescriptor, self, _self);
676
+ if (!upb_fielddef_typeisset(self->fielddef)) {
677
+ return Qnil;
678
+ }
679
+ return descriptortype_to_ruby(upb_fielddef_descriptortype(self->fielddef));
680
+ }
681
+
682
+ /*
683
+ * call-seq:
684
+ * FieldDescriptor.type = type
685
+ *
686
+ * Sets this field's type. Cannot be called if field is part of a message type
687
+ * already in a pool.
688
+ */
689
+ VALUE FieldDescriptor_type_set(VALUE _self, VALUE type) {
690
+ DEFINE_SELF(FieldDescriptor, self, _self);
691
+ upb_fielddef* mut_def = check_field_notfrozen(self->fielddef);
692
+ upb_fielddef_setdescriptortype(mut_def, ruby_to_descriptortype(type));
693
+ return Qnil;
694
+ }
695
+
696
+ /*
697
+ * call-seq:
698
+ * FieldDescriptor.label => label
699
+ *
700
+ * Returns this field's label (i.e., plurality), as a Ruby symbol.
701
+ *
702
+ * Valid field labels are:
703
+ * :optional, :repeated
704
+ */
705
+ VALUE FieldDescriptor_label(VALUE _self) {
706
+ DEFINE_SELF(FieldDescriptor, self, _self);
707
+ switch (upb_fielddef_label(self->fielddef)) {
708
+ #define CONVERT(upb, ruby) \
709
+ case UPB_LABEL_ ## upb : return ID2SYM(rb_intern( # ruby ));
710
+
711
+ CONVERT(OPTIONAL, optional);
712
+ CONVERT(REQUIRED, required);
713
+ CONVERT(REPEATED, repeated);
714
+
715
+ #undef CONVERT
716
+ }
717
+
718
+ return Qnil;
719
+ }
720
+
721
+ /*
722
+ * call-seq:
723
+ * FieldDescriptor.label = label
724
+ *
725
+ * Sets the label on this field. Cannot be called if field is part of a message
726
+ * type already in a pool.
727
+ */
728
+ VALUE FieldDescriptor_label_set(VALUE _self, VALUE label) {
729
+ DEFINE_SELF(FieldDescriptor, self, _self);
730
+ upb_fielddef* mut_def = check_field_notfrozen(self->fielddef);
731
+ upb_label_t upb_label = -1;
732
+ bool converted = false;
733
+
734
+ if (TYPE(label) != T_SYMBOL) {
735
+ rb_raise(rb_eArgError, "Expected symbol for field label.");
736
+ }
737
+
738
+ #define CONVERT(upb, ruby) \
739
+ if (SYM2ID(label) == rb_intern( # ruby )) { \
740
+ upb_label = UPB_LABEL_ ## upb; \
741
+ converted = true; \
742
+ }
743
+
744
+ CONVERT(OPTIONAL, optional);
745
+ CONVERT(REQUIRED, required);
746
+ CONVERT(REPEATED, repeated);
747
+
748
+ #undef CONVERT
749
+
750
+ if (!converted) {
751
+ rb_raise(rb_eArgError, "Unknown field label.");
752
+ }
753
+
754
+ upb_fielddef_setlabel(mut_def, upb_label);
755
+
756
+ return Qnil;
757
+ }
758
+
759
+ /*
760
+ * call-seq:
761
+ * FieldDescriptor.number => number
762
+ *
763
+ * Returns the tag number for this field.
764
+ */
765
+ VALUE FieldDescriptor_number(VALUE _self) {
766
+ DEFINE_SELF(FieldDescriptor, self, _self);
767
+ return INT2NUM(upb_fielddef_number(self->fielddef));
768
+ }
769
+
770
+ /*
771
+ * call-seq:
772
+ * FieldDescriptor.number = number
773
+ *
774
+ * Sets the tag number for this field. Cannot be called if field is part of a
775
+ * message type already in a pool.
776
+ */
777
+ VALUE FieldDescriptor_number_set(VALUE _self, VALUE number) {
778
+ DEFINE_SELF(FieldDescriptor, self, _self);
779
+ upb_fielddef* mut_def = check_field_notfrozen(self->fielddef);
780
+ CHECK_UPB(upb_fielddef_setnumber(mut_def, NUM2INT(number), &status),
781
+ "Error setting field number");
782
+ return Qnil;
783
+ }
784
+
785
+ /*
786
+ * call-seq:
787
+ * FieldDescriptor.submsg_name => submsg_name
788
+ *
789
+ * Returns the name of the message or enum type corresponding to this field, if
790
+ * it is a message or enum field (respectively), or nil otherwise. This type
791
+ * name will be resolved within the context of the pool to which the containing
792
+ * message type is added.
793
+ */
794
+ VALUE FieldDescriptor_submsg_name(VALUE _self) {
795
+ DEFINE_SELF(FieldDescriptor, self, _self);
796
+ if (!upb_fielddef_hassubdef(self->fielddef)) {
797
+ return Qnil;
798
+ }
799
+ return rb_str_maybe_null(upb_fielddef_subdefname(self->fielddef));
800
+ }
801
+
802
+ /*
803
+ * call-seq:
804
+ * FieldDescriptor.submsg_name = submsg_name
805
+ *
806
+ * Sets the name of the message or enum type corresponding to this field, if it
807
+ * is a message or enum field (respectively). This type name will be resolved
808
+ * within the context of the pool to which the containing message type is added.
809
+ * Cannot be called on field that are not of message or enum type, or on fields
810
+ * that are part of a message type already added to a pool.
811
+ */
812
+ VALUE FieldDescriptor_submsg_name_set(VALUE _self, VALUE value) {
813
+ DEFINE_SELF(FieldDescriptor, self, _self);
814
+ upb_fielddef* mut_def = check_field_notfrozen(self->fielddef);
815
+ const char* str = get_str(value);
816
+ if (!upb_fielddef_hassubdef(self->fielddef)) {
817
+ rb_raise(rb_eTypeError, "FieldDescriptor does not have subdef.");
818
+ }
819
+ CHECK_UPB(upb_fielddef_setsubdefname(mut_def, str, &status),
820
+ "Error setting submessage name");
821
+ return Qnil;
822
+ }
823
+
824
+ /*
825
+ * call-seq:
826
+ * FieldDescriptor.subtype => message_or_enum_descriptor
827
+ *
828
+ * Returns the message or enum descriptor corresponding to this field's type if
829
+ * it is a message or enum field, respectively, or nil otherwise. Cannot be
830
+ * called *until* the containing message type is added to a pool (and thus
831
+ * resolved).
832
+ */
833
+ VALUE FieldDescriptor_subtype(VALUE _self) {
834
+ DEFINE_SELF(FieldDescriptor, self, _self);
835
+ const upb_def* def;
836
+
837
+ if (!upb_fielddef_hassubdef(self->fielddef)) {
838
+ return Qnil;
839
+ }
840
+ def = upb_fielddef_subdef(self->fielddef);
841
+ if (def == NULL) {
842
+ return Qnil;
843
+ }
844
+ return get_def_obj(def);
845
+ }
846
+
847
+ /*
848
+ * call-seq:
849
+ * FieldDescriptor.get(message) => value
850
+ *
851
+ * Returns the value set for this field on the given message. Raises an
852
+ * exception if message is of the wrong type.
853
+ */
854
+ VALUE FieldDescriptor_get(VALUE _self, VALUE msg_rb) {
855
+ DEFINE_SELF(FieldDescriptor, self, _self);
856
+ MessageHeader* msg;
857
+ TypedData_Get_Struct(msg_rb, MessageHeader, &Message_type, msg);
858
+ if (msg->descriptor->msgdef != upb_fielddef_containingtype(self->fielddef)) {
859
+ rb_raise(rb_eTypeError, "get method called on wrong message type");
860
+ }
861
+ return layout_get(msg->descriptor->layout, Message_data(msg), self->fielddef);
862
+ }
863
+
864
+ /*
865
+ * call-seq:
866
+ * FieldDescriptor.set(message, value)
867
+ *
868
+ * Sets the value corresponding to this field to the given value on the given
869
+ * message. Raises an exception if message is of the wrong type. Performs the
870
+ * ordinary type-checks for field setting.
871
+ */
872
+ VALUE FieldDescriptor_set(VALUE _self, VALUE msg_rb, VALUE value) {
873
+ DEFINE_SELF(FieldDescriptor, self, _self);
874
+ MessageHeader* msg;
875
+ TypedData_Get_Struct(msg_rb, MessageHeader, &Message_type, msg);
876
+ if (msg->descriptor->msgdef != upb_fielddef_containingtype(self->fielddef)) {
877
+ rb_raise(rb_eTypeError, "set method called on wrong message type");
878
+ }
879
+ layout_set(msg->descriptor->layout, Message_data(msg), self->fielddef, value);
880
+ return Qnil;
881
+ }
882
+
883
+ // -----------------------------------------------------------------------------
884
+ // OneofDescriptor.
885
+ // -----------------------------------------------------------------------------
886
+
887
+ DEFINE_CLASS(OneofDescriptor, "Google::Protobuf::OneofDescriptor");
888
+
889
+ void OneofDescriptor_mark(void* _self) {
890
+ }
891
+
892
+ void OneofDescriptor_free(void* _self) {
893
+ OneofDescriptor* self = _self;
894
+ upb_oneofdef_unref(self->oneofdef, &self->oneofdef);
895
+ xfree(self);
896
+ }
897
+
898
+ /*
899
+ * call-seq:
900
+ * OneofDescriptor.new => oneof_descriptor
901
+ *
902
+ * Creates a new, empty, oneof descriptor. The oneof may only be modified prior
903
+ * to being added to a message descriptor which is subsequently added to a pool.
904
+ */
905
+ VALUE OneofDescriptor_alloc(VALUE klass) {
906
+ OneofDescriptor* self = ALLOC(OneofDescriptor);
907
+ VALUE ret = TypedData_Wrap_Struct(klass, &_OneofDescriptor_type, self);
908
+ self->oneofdef = upb_oneofdef_new(&self->oneofdef);
909
+ return ret;
910
+ }
911
+
912
+ void OneofDescriptor_register(VALUE module) {
913
+ VALUE klass = rb_define_class_under(
914
+ module, "OneofDescriptor", rb_cObject);
915
+ rb_define_alloc_func(klass, OneofDescriptor_alloc);
916
+ rb_define_method(klass, "name", OneofDescriptor_name, 0);
917
+ rb_define_method(klass, "name=", OneofDescriptor_name_set, 1);
918
+ rb_define_method(klass, "add_field", OneofDescriptor_add_field, 1);
919
+ rb_define_method(klass, "each", OneofDescriptor_each, 0);
920
+ rb_include_module(klass, rb_mEnumerable);
921
+ cOneofDescriptor = klass;
922
+ rb_gc_register_address(&cOneofDescriptor);
923
+ }
924
+
925
+ /*
926
+ * call-seq:
927
+ * OneofDescriptor.name => name
928
+ *
929
+ * Returns the name of this oneof.
930
+ */
931
+ VALUE OneofDescriptor_name(VALUE _self) {
932
+ DEFINE_SELF(OneofDescriptor, self, _self);
933
+ return rb_str_maybe_null(upb_oneofdef_name(self->oneofdef));
934
+ }
935
+
936
+ /*
937
+ * call-seq:
938
+ * OneofDescriptor.name = name
939
+ *
940
+ * Sets a new name for this oneof. The oneof must not have been added to a
941
+ * message descriptor yet.
942
+ */
943
+ VALUE OneofDescriptor_name_set(VALUE _self, VALUE value) {
944
+ DEFINE_SELF(OneofDescriptor, self, _self);
945
+ upb_oneofdef* mut_def = check_oneof_notfrozen(self->oneofdef);
946
+ const char* str = get_str(value);
947
+ CHECK_UPB(upb_oneofdef_setname(mut_def, str, &status),
948
+ "Error setting oneof name");
949
+ return Qnil;
950
+ }
951
+
952
+ /*
953
+ * call-seq:
954
+ * OneofDescriptor.add_field(field) => nil
955
+ *
956
+ * Adds a field to this oneof. The field may have been added to this oneof in
957
+ * the past, or the message to which this oneof belongs (if any), but may not
958
+ * have already been added to any other oneof or message. Otherwise, an
959
+ * exception is raised.
960
+ *
961
+ * All fields added to the oneof via this method will be automatically added to
962
+ * the message to which this oneof belongs, if it belongs to one currently, or
963
+ * else will be added to any message to which the oneof is later added at the
964
+ * time that it is added.
965
+ */
966
+ VALUE OneofDescriptor_add_field(VALUE _self, VALUE obj) {
967
+ DEFINE_SELF(OneofDescriptor, self, _self);
968
+ upb_oneofdef* mut_def = check_oneof_notfrozen(self->oneofdef);
969
+ FieldDescriptor* def = ruby_to_FieldDescriptor(obj);
970
+ upb_fielddef* mut_field_def = check_field_notfrozen(def->fielddef);
971
+ CHECK_UPB(
972
+ upb_oneofdef_addfield(mut_def, mut_field_def, NULL, &status),
973
+ "Adding field to OneofDescriptor failed");
974
+ add_def_obj(def->fielddef, obj);
975
+ return Qnil;
976
+ }
977
+
978
+ /*
979
+ * call-seq:
980
+ * OneofDescriptor.each(&block) => nil
981
+ *
982
+ * Iterates through fields in this oneof, yielding to the block on each one.
983
+ */
984
+ VALUE OneofDescriptor_each(VALUE _self, VALUE field) {
985
+ DEFINE_SELF(OneofDescriptor, self, _self);
986
+ upb_oneof_iter it;
987
+ for (upb_oneof_begin(&it, self->oneofdef);
988
+ !upb_oneof_done(&it);
989
+ upb_oneof_next(&it)) {
990
+ const upb_fielddef* f = upb_oneof_iter_field(&it);
991
+ VALUE obj = get_def_obj(f);
992
+ rb_yield(obj);
993
+ }
994
+ return Qnil;
995
+ }
996
+
997
+ // -----------------------------------------------------------------------------
998
+ // EnumDescriptor.
999
+ // -----------------------------------------------------------------------------
1000
+
1001
+ DEFINE_CLASS(EnumDescriptor, "Google::Protobuf::EnumDescriptor");
1002
+
1003
+ void EnumDescriptor_mark(void* _self) {
1004
+ EnumDescriptor* self = _self;
1005
+ rb_gc_mark(self->module);
1006
+ }
1007
+
1008
+ void EnumDescriptor_free(void* _self) {
1009
+ EnumDescriptor* self = _self;
1010
+ upb_enumdef_unref(self->enumdef, &self->enumdef);
1011
+ xfree(self);
1012
+ }
1013
+
1014
+ /*
1015
+ * call-seq:
1016
+ * EnumDescriptor.new => enum_descriptor
1017
+ *
1018
+ * Creates a new, empty, enum descriptor. Must be added to a pool before the
1019
+ * enum type can be used. The enum type may only be modified prior to adding to
1020
+ * a pool.
1021
+ */
1022
+ VALUE EnumDescriptor_alloc(VALUE klass) {
1023
+ EnumDescriptor* self = ALLOC(EnumDescriptor);
1024
+ VALUE ret = TypedData_Wrap_Struct(klass, &_EnumDescriptor_type, self);
1025
+ self->enumdef = upb_enumdef_new(&self->enumdef);
1026
+ self->module = Qnil;
1027
+ return ret;
1028
+ }
1029
+
1030
+ void EnumDescriptor_register(VALUE module) {
1031
+ VALUE klass = rb_define_class_under(
1032
+ module, "EnumDescriptor", rb_cObject);
1033
+ rb_define_alloc_func(klass, EnumDescriptor_alloc);
1034
+ rb_define_method(klass, "name", EnumDescriptor_name, 0);
1035
+ rb_define_method(klass, "name=", EnumDescriptor_name_set, 1);
1036
+ rb_define_method(klass, "add_value", EnumDescriptor_add_value, 2);
1037
+ rb_define_method(klass, "lookup_name", EnumDescriptor_lookup_name, 1);
1038
+ rb_define_method(klass, "lookup_value", EnumDescriptor_lookup_value, 1);
1039
+ rb_define_method(klass, "each", EnumDescriptor_each, 0);
1040
+ rb_define_method(klass, "enummodule", EnumDescriptor_enummodule, 0);
1041
+ rb_include_module(klass, rb_mEnumerable);
1042
+ cEnumDescriptor = klass;
1043
+ rb_gc_register_address(&cEnumDescriptor);
1044
+ }
1045
+
1046
+ /*
1047
+ * call-seq:
1048
+ * EnumDescriptor.name => name
1049
+ *
1050
+ * Returns the name of this enum type.
1051
+ */
1052
+ VALUE EnumDescriptor_name(VALUE _self) {
1053
+ DEFINE_SELF(EnumDescriptor, self, _self);
1054
+ return rb_str_maybe_null(upb_enumdef_fullname(self->enumdef));
1055
+ }
1056
+
1057
+ /*
1058
+ * call-seq:
1059
+ * EnumDescriptor.name = name
1060
+ *
1061
+ * Sets the name of this enum type. Cannot be called if the enum type has
1062
+ * already been added to a pool.
1063
+ */
1064
+ VALUE EnumDescriptor_name_set(VALUE _self, VALUE str) {
1065
+ DEFINE_SELF(EnumDescriptor, self, _self);
1066
+ upb_enumdef* mut_def = check_enum_notfrozen(self->enumdef);
1067
+ const char* name = get_str(str);
1068
+ CHECK_UPB(upb_enumdef_setfullname(mut_def, name, &status),
1069
+ "Error setting EnumDescriptor name");
1070
+ return Qnil;
1071
+ }
1072
+
1073
+ /*
1074
+ * call-seq:
1075
+ * EnumDescriptor.add_value(key, value)
1076
+ *
1077
+ * Adds a new key => value mapping to this enum type. Key must be given as a
1078
+ * Ruby symbol. Cannot be called if the enum type has already been added to a
1079
+ * pool. Will raise an exception if the key or value is already in use.
1080
+ */
1081
+ VALUE EnumDescriptor_add_value(VALUE _self, VALUE name, VALUE number) {
1082
+ DEFINE_SELF(EnumDescriptor, self, _self);
1083
+ upb_enumdef* mut_def = check_enum_notfrozen(self->enumdef);
1084
+ const char* name_str = rb_id2name(SYM2ID(name));
1085
+ int32_t val = NUM2INT(number);
1086
+ CHECK_UPB(upb_enumdef_addval(mut_def, name_str, val, &status),
1087
+ "Error adding value to enum");
1088
+ return Qnil;
1089
+ }
1090
+
1091
+ /*
1092
+ * call-seq:
1093
+ * EnumDescriptor.lookup_name(name) => value
1094
+ *
1095
+ * Returns the numeric value corresponding to the given key name (as a Ruby
1096
+ * symbol), or nil if none.
1097
+ */
1098
+ VALUE EnumDescriptor_lookup_name(VALUE _self, VALUE name) {
1099
+ DEFINE_SELF(EnumDescriptor, self, _self);
1100
+ const char* name_str= rb_id2name(SYM2ID(name));
1101
+ int32_t val = 0;
1102
+ if (upb_enumdef_ntoiz(self->enumdef, name_str, &val)) {
1103
+ return INT2NUM(val);
1104
+ } else {
1105
+ return Qnil;
1106
+ }
1107
+ }
1108
+
1109
+ /*
1110
+ * call-seq:
1111
+ * EnumDescriptor.lookup_value(name) => value
1112
+ *
1113
+ * Returns the key name (as a Ruby symbol) corresponding to the integer value,
1114
+ * or nil if none.
1115
+ */
1116
+ VALUE EnumDescriptor_lookup_value(VALUE _self, VALUE number) {
1117
+ DEFINE_SELF(EnumDescriptor, self, _self);
1118
+ int32_t val = NUM2INT(number);
1119
+ const char* name = upb_enumdef_iton(self->enumdef, val);
1120
+ if (name != NULL) {
1121
+ return ID2SYM(rb_intern(name));
1122
+ } else {
1123
+ return Qnil;
1124
+ }
1125
+ }
1126
+
1127
+ /*
1128
+ * call-seq:
1129
+ * EnumDescriptor.each(&block)
1130
+ *
1131
+ * Iterates over key => value mappings in this enum's definition, yielding to
1132
+ * the block with (key, value) arguments for each one.
1133
+ */
1134
+ VALUE EnumDescriptor_each(VALUE _self) {
1135
+ DEFINE_SELF(EnumDescriptor, self, _self);
1136
+
1137
+ upb_enum_iter it;
1138
+ for (upb_enum_begin(&it, self->enumdef);
1139
+ !upb_enum_done(&it);
1140
+ upb_enum_next(&it)) {
1141
+ VALUE key = ID2SYM(rb_intern(upb_enum_iter_name(&it)));
1142
+ VALUE number = INT2NUM(upb_enum_iter_number(&it));
1143
+ rb_yield_values(2, key, number);
1144
+ }
1145
+
1146
+ return Qnil;
1147
+ }
1148
+
1149
+ /*
1150
+ * call-seq:
1151
+ * EnumDescriptor.enummodule => module
1152
+ *
1153
+ * Returns the Ruby module corresponding to this enum type. Cannot be called
1154
+ * until the enum descriptor has been added to a pool.
1155
+ */
1156
+ VALUE EnumDescriptor_enummodule(VALUE _self) {
1157
+ DEFINE_SELF(EnumDescriptor, self, _self);
1158
+ if (!upb_def_isfrozen((const upb_def*)self->enumdef)) {
1159
+ rb_raise(rb_eRuntimeError,
1160
+ "Cannot fetch enum module from an EnumDescriptor not yet "
1161
+ "in a pool.");
1162
+ }
1163
+ if (self->module == Qnil) {
1164
+ self->module = build_module_from_enumdesc(self);
1165
+ }
1166
+ return self->module;
1167
+ }
1168
+
1169
+ // -----------------------------------------------------------------------------
1170
+ // MessageBuilderContext.
1171
+ // -----------------------------------------------------------------------------
1172
+
1173
+ DEFINE_CLASS(MessageBuilderContext,
1174
+ "Google::Protobuf::Internal::MessageBuilderContext");
1175
+
1176
+ void MessageBuilderContext_mark(void* _self) {
1177
+ MessageBuilderContext* self = _self;
1178
+ rb_gc_mark(self->descriptor);
1179
+ rb_gc_mark(self->builder);
1180
+ }
1181
+
1182
+ void MessageBuilderContext_free(void* _self) {
1183
+ MessageBuilderContext* self = _self;
1184
+ xfree(self);
1185
+ }
1186
+
1187
+ VALUE MessageBuilderContext_alloc(VALUE klass) {
1188
+ MessageBuilderContext* self = ALLOC(MessageBuilderContext);
1189
+ VALUE ret = TypedData_Wrap_Struct(
1190
+ klass, &_MessageBuilderContext_type, self);
1191
+ self->descriptor = Qnil;
1192
+ self->builder = Qnil;
1193
+ return ret;
1194
+ }
1195
+
1196
+ void MessageBuilderContext_register(VALUE module) {
1197
+ VALUE klass = rb_define_class_under(
1198
+ module, "MessageBuilderContext", rb_cObject);
1199
+ rb_define_alloc_func(klass, MessageBuilderContext_alloc);
1200
+ rb_define_method(klass, "initialize",
1201
+ MessageBuilderContext_initialize, 2);
1202
+ rb_define_method(klass, "optional", MessageBuilderContext_optional, -1);
1203
+ rb_define_method(klass, "required", MessageBuilderContext_required, -1);
1204
+ rb_define_method(klass, "repeated", MessageBuilderContext_repeated, -1);
1205
+ rb_define_method(klass, "map", MessageBuilderContext_map, -1);
1206
+ rb_define_method(klass, "oneof", MessageBuilderContext_oneof, 1);
1207
+ cMessageBuilderContext = klass;
1208
+ rb_gc_register_address(&cMessageBuilderContext);
1209
+ }
1210
+
1211
+ /*
1212
+ * call-seq:
1213
+ * MessageBuilderContext.new(desc, builder) => context
1214
+ *
1215
+ * Create a new message builder context around the given message descriptor and
1216
+ * builder context. This class is intended to serve as a DSL context to be used
1217
+ * with #instance_eval.
1218
+ */
1219
+ VALUE MessageBuilderContext_initialize(VALUE _self,
1220
+ VALUE msgdef,
1221
+ VALUE builder) {
1222
+ DEFINE_SELF(MessageBuilderContext, self, _self);
1223
+ self->descriptor = msgdef;
1224
+ self->builder = builder;
1225
+ return Qnil;
1226
+ }
1227
+
1228
+ static VALUE msgdef_add_field(VALUE msgdef,
1229
+ const char* label, VALUE name,
1230
+ VALUE type, VALUE number,
1231
+ VALUE type_class) {
1232
+ VALUE fielddef = rb_class_new_instance(0, NULL, cFieldDescriptor);
1233
+ VALUE name_str = rb_str_new2(rb_id2name(SYM2ID(name)));
1234
+
1235
+ rb_funcall(fielddef, rb_intern("label="), 1, ID2SYM(rb_intern(label)));
1236
+ rb_funcall(fielddef, rb_intern("name="), 1, name_str);
1237
+ rb_funcall(fielddef, rb_intern("type="), 1, type);
1238
+ rb_funcall(fielddef, rb_intern("number="), 1, number);
1239
+
1240
+ if (type_class != Qnil) {
1241
+ if (TYPE(type_class) != T_STRING) {
1242
+ rb_raise(rb_eArgError, "Expected string for type class");
1243
+ }
1244
+ // Make it an absolute type name by prepending a dot.
1245
+ type_class = rb_str_append(rb_str_new2("."), type_class);
1246
+ rb_funcall(fielddef, rb_intern("submsg_name="), 1, type_class);
1247
+ }
1248
+
1249
+ rb_funcall(msgdef, rb_intern("add_field"), 1, fielddef);
1250
+ return fielddef;
1251
+ }
1252
+
1253
+ /*
1254
+ * call-seq:
1255
+ * MessageBuilderContext.optional(name, type, number, type_class = nil)
1256
+ *
1257
+ * Defines a new optional field on this message type with the given type, tag
1258
+ * number, and type class (for message and enum fields). The type must be a Ruby
1259
+ * symbol (as accepted by FieldDescriptor#type=) and the type_class must be a
1260
+ * string, if present (as accepted by FieldDescriptor#submsg_name=).
1261
+ */
1262
+ VALUE MessageBuilderContext_optional(int argc, VALUE* argv, VALUE _self) {
1263
+ DEFINE_SELF(MessageBuilderContext, self, _self);
1264
+ VALUE name, type, number, type_class;
1265
+
1266
+ if (argc < 3) {
1267
+ rb_raise(rb_eArgError, "Expected at least 3 arguments.");
1268
+ }
1269
+ name = argv[0];
1270
+ type = argv[1];
1271
+ number = argv[2];
1272
+ type_class = (argc > 3) ? argv[3] : Qnil;
1273
+
1274
+ return msgdef_add_field(self->descriptor, "optional",
1275
+ name, type, number, type_class);
1276
+ }
1277
+
1278
+ /*
1279
+ * call-seq:
1280
+ * MessageBuilderContext.required(name, type, number, type_class = nil)
1281
+ *
1282
+ * Defines a new required field on this message type with the given type, tag
1283
+ * number, and type class (for message and enum fields). The type must be a Ruby
1284
+ * symbol (as accepted by FieldDescriptor#type=) and the type_class must be a
1285
+ * string, if present (as accepted by FieldDescriptor#submsg_name=).
1286
+ *
1287
+ * Proto3 does not have required fields, but this method exists for
1288
+ * completeness. Any attempt to add a message type with required fields to a
1289
+ * pool will currently result in an error.
1290
+ */
1291
+ VALUE MessageBuilderContext_required(int argc, VALUE* argv, VALUE _self) {
1292
+ DEFINE_SELF(MessageBuilderContext, self, _self);
1293
+ VALUE name, type, number, type_class;
1294
+
1295
+ if (argc < 3) {
1296
+ rb_raise(rb_eArgError, "Expected at least 3 arguments.");
1297
+ }
1298
+ name = argv[0];
1299
+ type = argv[1];
1300
+ number = argv[2];
1301
+ type_class = (argc > 3) ? argv[3] : Qnil;
1302
+
1303
+ return msgdef_add_field(self->descriptor, "required",
1304
+ name, type, number, type_class);
1305
+ }
1306
+
1307
+ /*
1308
+ * call-seq:
1309
+ * MessageBuilderContext.repeated(name, type, number, type_class = nil)
1310
+ *
1311
+ * Defines a new repeated field on this message type with the given type, tag
1312
+ * number, and type class (for message and enum fields). The type must be a Ruby
1313
+ * symbol (as accepted by FieldDescriptor#type=) and the type_class must be a
1314
+ * string, if present (as accepted by FieldDescriptor#submsg_name=).
1315
+ */
1316
+ VALUE MessageBuilderContext_repeated(int argc, VALUE* argv, VALUE _self) {
1317
+ DEFINE_SELF(MessageBuilderContext, self, _self);
1318
+ VALUE name, type, number, type_class;
1319
+
1320
+ if (argc < 3) {
1321
+ rb_raise(rb_eArgError, "Expected at least 3 arguments.");
1322
+ }
1323
+ name = argv[0];
1324
+ type = argv[1];
1325
+ number = argv[2];
1326
+ type_class = (argc > 3) ? argv[3] : Qnil;
1327
+
1328
+ return msgdef_add_field(self->descriptor, "repeated",
1329
+ name, type, number, type_class);
1330
+ }
1331
+
1332
+ /*
1333
+ * call-seq:
1334
+ * MessageBuilderContext.map(name, key_type, value_type, number,
1335
+ * value_type_class = nil)
1336
+ *
1337
+ * Defines a new map field on this message type with the given key and value
1338
+ * types, tag number, and type class (for message and enum value types). The key
1339
+ * type must be :int32/:uint32/:int64/:uint64, :bool, or :string. The value type
1340
+ * type must be a Ruby symbol (as accepted by FieldDescriptor#type=) and the
1341
+ * type_class must be a string, if present (as accepted by
1342
+ * FieldDescriptor#submsg_name=).
1343
+ */
1344
+ VALUE MessageBuilderContext_map(int argc, VALUE* argv, VALUE _self) {
1345
+ DEFINE_SELF(MessageBuilderContext, self, _self);
1346
+ VALUE name, key_type, value_type, number, type_class;
1347
+ VALUE mapentry_desc, mapentry_desc_name;
1348
+
1349
+ if (argc < 4) {
1350
+ rb_raise(rb_eArgError, "Expected at least 4 arguments.");
1351
+ }
1352
+ name = argv[0];
1353
+ key_type = argv[1];
1354
+ value_type = argv[2];
1355
+ number = argv[3];
1356
+ type_class = (argc > 4) ? argv[4] : Qnil;
1357
+
1358
+ // Validate the key type. We can't accept enums, messages, or floats/doubles
1359
+ // as map keys. (We exclude these explicitly, and the field-descriptor setter
1360
+ // below then ensures that the type is one of the remaining valid options.)
1361
+ if (SYM2ID(key_type) == rb_intern("float") ||
1362
+ SYM2ID(key_type) == rb_intern("double") ||
1363
+ SYM2ID(key_type) == rb_intern("enum") ||
1364
+ SYM2ID(key_type) == rb_intern("message")) {
1365
+ rb_raise(rb_eArgError,
1366
+ "Cannot add a map field with a float, double, enum, or message "
1367
+ "type.");
1368
+ }
1369
+
1370
+ // Create a new message descriptor for the map entry message, and create a
1371
+ // repeated submessage field here with that type.
1372
+ mapentry_desc = rb_class_new_instance(0, NULL, cDescriptor);
1373
+ mapentry_desc_name = rb_funcall(self->descriptor, rb_intern("name"), 0);
1374
+ mapentry_desc_name = rb_str_cat2(mapentry_desc_name, "_MapEntry_");
1375
+ mapentry_desc_name = rb_str_cat2(mapentry_desc_name,
1376
+ rb_id2name(SYM2ID(name)));
1377
+ Descriptor_name_set(mapentry_desc, mapentry_desc_name);
1378
+
1379
+ {
1380
+ // The 'mapentry' attribute has no Ruby setter because we do not want the
1381
+ // user attempting to DIY the setup below; we want to ensure that the fields
1382
+ // are correct. So we reach into the msgdef here to set the bit manually.
1383
+ Descriptor* mapentry_desc_self = ruby_to_Descriptor(mapentry_desc);
1384
+ upb_msgdef_setmapentry((upb_msgdef*)mapentry_desc_self->msgdef, true);
1385
+ }
1386
+
1387
+ {
1388
+ // optional <type> key = 1;
1389
+ VALUE key_field = rb_class_new_instance(0, NULL, cFieldDescriptor);
1390
+ FieldDescriptor_name_set(key_field, rb_str_new2("key"));
1391
+ FieldDescriptor_label_set(key_field, ID2SYM(rb_intern("optional")));
1392
+ FieldDescriptor_number_set(key_field, INT2NUM(1));
1393
+ FieldDescriptor_type_set(key_field, key_type);
1394
+ Descriptor_add_field(mapentry_desc, key_field);
1395
+ }
1396
+
1397
+ {
1398
+ // optional <type> value = 2;
1399
+ VALUE value_field = rb_class_new_instance(0, NULL, cFieldDescriptor);
1400
+ FieldDescriptor_name_set(value_field, rb_str_new2("value"));
1401
+ FieldDescriptor_label_set(value_field, ID2SYM(rb_intern("optional")));
1402
+ FieldDescriptor_number_set(value_field, INT2NUM(2));
1403
+ FieldDescriptor_type_set(value_field, value_type);
1404
+ if (type_class != Qnil) {
1405
+ VALUE submsg_name = rb_str_new2("."); // prepend '.' to make absolute.
1406
+ submsg_name = rb_str_append(submsg_name, type_class);
1407
+ FieldDescriptor_submsg_name_set(value_field, submsg_name);
1408
+ }
1409
+ Descriptor_add_field(mapentry_desc, value_field);
1410
+ }
1411
+
1412
+ {
1413
+ // Add the map-entry message type to the current builder, and use the type
1414
+ // to create the map field itself.
1415
+ Builder* builder_self = ruby_to_Builder(self->builder);
1416
+ rb_ary_push(builder_self->pending_list, mapentry_desc);
1417
+ }
1418
+
1419
+ {
1420
+ VALUE map_field = rb_class_new_instance(0, NULL, cFieldDescriptor);
1421
+ VALUE name_str = rb_str_new2(rb_id2name(SYM2ID(name)));
1422
+ VALUE submsg_name;
1423
+
1424
+ FieldDescriptor_name_set(map_field, name_str);
1425
+ FieldDescriptor_number_set(map_field, number);
1426
+ FieldDescriptor_label_set(map_field, ID2SYM(rb_intern("repeated")));
1427
+ FieldDescriptor_type_set(map_field, ID2SYM(rb_intern("message")));
1428
+ submsg_name = rb_str_new2("."); // prepend '.' to make name absolute.
1429
+ submsg_name = rb_str_append(submsg_name, mapentry_desc_name);
1430
+ FieldDescriptor_submsg_name_set(map_field, submsg_name);
1431
+ Descriptor_add_field(self->descriptor, map_field);
1432
+ }
1433
+
1434
+ return Qnil;
1435
+ }
1436
+
1437
+ /*
1438
+ * call-seq:
1439
+ * MessageBuilderContext.oneof(name, &block) => nil
1440
+ *
1441
+ * Creates a new OneofDescriptor with the given name, creates a
1442
+ * OneofBuilderContext attached to that OneofDescriptor, evaluates the given
1443
+ * block in the context of that OneofBuilderContext with #instance_eval, and
1444
+ * then adds the oneof to the message.
1445
+ *
1446
+ * This is the recommended, idiomatic way to build oneof definitions.
1447
+ */
1448
+ VALUE MessageBuilderContext_oneof(VALUE _self, VALUE name) {
1449
+ DEFINE_SELF(MessageBuilderContext, self, _self);
1450
+ VALUE oneofdef = rb_class_new_instance(0, NULL, cOneofDescriptor);
1451
+ VALUE args[2] = { oneofdef, self->builder };
1452
+ VALUE ctx = rb_class_new_instance(2, args, cOneofBuilderContext);
1453
+ VALUE block = rb_block_proc();
1454
+ VALUE name_str = rb_str_new2(rb_id2name(SYM2ID(name)));
1455
+ rb_funcall(oneofdef, rb_intern("name="), 1, name_str);
1456
+ rb_funcall_with_block(ctx, rb_intern("instance_eval"), 0, NULL, block);
1457
+ Descriptor_add_oneof(self->descriptor, oneofdef);
1458
+
1459
+ return Qnil;
1460
+ }
1461
+
1462
+ // -----------------------------------------------------------------------------
1463
+ // OneofBuilderContext.
1464
+ // -----------------------------------------------------------------------------
1465
+
1466
+ DEFINE_CLASS(OneofBuilderContext,
1467
+ "Google::Protobuf::Internal::OneofBuilderContext");
1468
+
1469
+ void OneofBuilderContext_mark(void* _self) {
1470
+ OneofBuilderContext* self = _self;
1471
+ rb_gc_mark(self->descriptor);
1472
+ rb_gc_mark(self->builder);
1473
+ }
1474
+
1475
+ void OneofBuilderContext_free(void* _self) {
1476
+ OneofBuilderContext* self = _self;
1477
+ xfree(self);
1478
+ }
1479
+
1480
+ VALUE OneofBuilderContext_alloc(VALUE klass) {
1481
+ OneofBuilderContext* self = ALLOC(OneofBuilderContext);
1482
+ VALUE ret = TypedData_Wrap_Struct(
1483
+ klass, &_OneofBuilderContext_type, self);
1484
+ self->descriptor = Qnil;
1485
+ self->builder = Qnil;
1486
+ return ret;
1487
+ }
1488
+
1489
+ void OneofBuilderContext_register(VALUE module) {
1490
+ VALUE klass = rb_define_class_under(
1491
+ module, "OneofBuilderContext", rb_cObject);
1492
+ rb_define_alloc_func(klass, OneofBuilderContext_alloc);
1493
+ rb_define_method(klass, "initialize",
1494
+ OneofBuilderContext_initialize, 2);
1495
+ rb_define_method(klass, "optional", OneofBuilderContext_optional, -1);
1496
+ cOneofBuilderContext = klass;
1497
+ rb_gc_register_address(&cOneofBuilderContext);
1498
+ }
1499
+
1500
+ /*
1501
+ * call-seq:
1502
+ * OneofBuilderContext.new(desc, builder) => context
1503
+ *
1504
+ * Create a new oneof builder context around the given oneof descriptor and
1505
+ * builder context. This class is intended to serve as a DSL context to be used
1506
+ * with #instance_eval.
1507
+ */
1508
+ VALUE OneofBuilderContext_initialize(VALUE _self,
1509
+ VALUE oneofdef,
1510
+ VALUE builder) {
1511
+ DEFINE_SELF(OneofBuilderContext, self, _self);
1512
+ self->descriptor = oneofdef;
1513
+ self->builder = builder;
1514
+ return Qnil;
1515
+ }
1516
+
1517
+ /*
1518
+ * call-seq:
1519
+ * OneofBuilderContext.optional(name, type, number, type_class = nil)
1520
+ *
1521
+ * Defines a new optional field in this oneof with the given type, tag number,
1522
+ * and type class (for message and enum fields). The type must be a Ruby symbol
1523
+ * (as accepted by FieldDescriptor#type=) and the type_class must be a string,
1524
+ * if present (as accepted by FieldDescriptor#submsg_name=).
1525
+ */
1526
+ VALUE OneofBuilderContext_optional(int argc, VALUE* argv, VALUE _self) {
1527
+ DEFINE_SELF(OneofBuilderContext, self, _self);
1528
+ VALUE name, type, number, type_class;
1529
+
1530
+ if (argc < 3) {
1531
+ rb_raise(rb_eArgError, "Expected at least 3 arguments.");
1532
+ }
1533
+ name = argv[0];
1534
+ type = argv[1];
1535
+ number = argv[2];
1536
+ type_class = (argc > 3) ? argv[3] : Qnil;
1537
+
1538
+ return msgdef_add_field(self->descriptor, "optional",
1539
+ name, type, number, type_class);
1540
+ }
1541
+
1542
+ // -----------------------------------------------------------------------------
1543
+ // EnumBuilderContext.
1544
+ // -----------------------------------------------------------------------------
1545
+
1546
+ DEFINE_CLASS(EnumBuilderContext,
1547
+ "Google::Protobuf::Internal::EnumBuilderContext");
1548
+
1549
+ void EnumBuilderContext_mark(void* _self) {
1550
+ EnumBuilderContext* self = _self;
1551
+ rb_gc_mark(self->enumdesc);
1552
+ }
1553
+
1554
+ void EnumBuilderContext_free(void* _self) {
1555
+ EnumBuilderContext* self = _self;
1556
+ xfree(self);
1557
+ }
1558
+
1559
+ VALUE EnumBuilderContext_alloc(VALUE klass) {
1560
+ EnumBuilderContext* self = ALLOC(EnumBuilderContext);
1561
+ VALUE ret = TypedData_Wrap_Struct(
1562
+ klass, &_EnumBuilderContext_type, self);
1563
+ self->enumdesc = Qnil;
1564
+ return ret;
1565
+ }
1566
+
1567
+ void EnumBuilderContext_register(VALUE module) {
1568
+ VALUE klass = rb_define_class_under(
1569
+ module, "EnumBuilderContext", rb_cObject);
1570
+ rb_define_alloc_func(klass, EnumBuilderContext_alloc);
1571
+ rb_define_method(klass, "initialize",
1572
+ EnumBuilderContext_initialize, 1);
1573
+ rb_define_method(klass, "value", EnumBuilderContext_value, 2);
1574
+ cEnumBuilderContext = klass;
1575
+ rb_gc_register_address(&cEnumBuilderContext);
1576
+ }
1577
+
1578
+ /*
1579
+ * call-seq:
1580
+ * EnumBuilderContext.new(enumdesc) => context
1581
+ *
1582
+ * Create a new builder context around the given enum descriptor. This class is
1583
+ * intended to serve as a DSL context to be used with #instance_eval.
1584
+ */
1585
+ VALUE EnumBuilderContext_initialize(VALUE _self, VALUE enumdef) {
1586
+ DEFINE_SELF(EnumBuilderContext, self, _self);
1587
+ self->enumdesc = enumdef;
1588
+ return Qnil;
1589
+ }
1590
+
1591
+ static VALUE enumdef_add_value(VALUE enumdef,
1592
+ VALUE name, VALUE number) {
1593
+ rb_funcall(enumdef, rb_intern("add_value"), 2, name, number);
1594
+ return Qnil;
1595
+ }
1596
+
1597
+ /*
1598
+ * call-seq:
1599
+ * EnumBuilder.add_value(name, number)
1600
+ *
1601
+ * Adds the given name => number mapping to the enum type. Name must be a Ruby
1602
+ * symbol.
1603
+ */
1604
+ VALUE EnumBuilderContext_value(VALUE _self, VALUE name, VALUE number) {
1605
+ DEFINE_SELF(EnumBuilderContext, self, _self);
1606
+ return enumdef_add_value(self->enumdesc, name, number);
1607
+ }
1608
+
1609
+ // -----------------------------------------------------------------------------
1610
+ // Builder.
1611
+ // -----------------------------------------------------------------------------
1612
+
1613
+ DEFINE_CLASS(Builder, "Google::Protobuf::Internal::Builder");
1614
+
1615
+ void Builder_mark(void* _self) {
1616
+ Builder* self = _self;
1617
+ rb_gc_mark(self->pending_list);
1618
+ }
1619
+
1620
+ void Builder_free(void* _self) {
1621
+ Builder* self = _self;
1622
+ xfree(self->defs);
1623
+ xfree(self);
1624
+ }
1625
+
1626
+ /*
1627
+ * call-seq:
1628
+ * Builder.new => builder
1629
+ *
1630
+ * Creates a new Builder. A Builder can accumulate a set of new message and enum
1631
+ * descriptors and atomically register them into a pool in a way that allows for
1632
+ * (co)recursive type references.
1633
+ */
1634
+ VALUE Builder_alloc(VALUE klass) {
1635
+ Builder* self = ALLOC(Builder);
1636
+ VALUE ret = TypedData_Wrap_Struct(
1637
+ klass, &_Builder_type, self);
1638
+ self->pending_list = rb_ary_new();
1639
+ self->defs = NULL;
1640
+ return ret;
1641
+ }
1642
+
1643
+ void Builder_register(VALUE module) {
1644
+ VALUE klass = rb_define_class_under(module, "Builder", rb_cObject);
1645
+ rb_define_alloc_func(klass, Builder_alloc);
1646
+ rb_define_method(klass, "add_message", Builder_add_message, 1);
1647
+ rb_define_method(klass, "add_enum", Builder_add_enum, 1);
1648
+ rb_define_method(klass, "finalize_to_pool", Builder_finalize_to_pool, 1);
1649
+ cBuilder = klass;
1650
+ rb_gc_register_address(&cBuilder);
1651
+ }
1652
+
1653
+ /*
1654
+ * call-seq:
1655
+ * Builder.add_message(name, &block)
1656
+ *
1657
+ * Creates a new, empty descriptor with the given name, and invokes the block in
1658
+ * the context of a MessageBuilderContext on that descriptor. The block can then
1659
+ * call, e.g., MessageBuilderContext#optional and MessageBuilderContext#repeated
1660
+ * methods to define the message fields.
1661
+ *
1662
+ * This is the recommended, idiomatic way to build message definitions.
1663
+ */
1664
+ VALUE Builder_add_message(VALUE _self, VALUE name) {
1665
+ DEFINE_SELF(Builder, self, _self);
1666
+ VALUE msgdef = rb_class_new_instance(0, NULL, cDescriptor);
1667
+ VALUE args[2] = { msgdef, _self };
1668
+ VALUE ctx = rb_class_new_instance(2, args, cMessageBuilderContext);
1669
+ VALUE block = rb_block_proc();
1670
+ rb_funcall(msgdef, rb_intern("name="), 1, name);
1671
+ rb_funcall_with_block(ctx, rb_intern("instance_eval"), 0, NULL, block);
1672
+ rb_ary_push(self->pending_list, msgdef);
1673
+ return Qnil;
1674
+ }
1675
+
1676
+ /*
1677
+ * call-seq:
1678
+ * Builder.add_enum(name, &block)
1679
+ *
1680
+ * Creates a new, empty enum descriptor with the given name, and invokes the
1681
+ * block in the context of an EnumBuilderContext on that descriptor. The block
1682
+ * can then call EnumBuilderContext#add_value to define the enum values.
1683
+ *
1684
+ * This is the recommended, idiomatic way to build enum definitions.
1685
+ */
1686
+ VALUE Builder_add_enum(VALUE _self, VALUE name) {
1687
+ DEFINE_SELF(Builder, self, _self);
1688
+ VALUE enumdef = rb_class_new_instance(0, NULL, cEnumDescriptor);
1689
+ VALUE ctx = rb_class_new_instance(1, &enumdef, cEnumBuilderContext);
1690
+ VALUE block = rb_block_proc();
1691
+ rb_funcall(enumdef, rb_intern("name="), 1, name);
1692
+ rb_funcall_with_block(ctx, rb_intern("instance_eval"), 0, NULL, block);
1693
+ rb_ary_push(self->pending_list, enumdef);
1694
+ return Qnil;
1695
+ }
1696
+
1697
+ static void validate_msgdef(const upb_msgdef* msgdef) {
1698
+ // Verify that no required fields exist. proto3 does not support these.
1699
+ upb_msg_field_iter it;
1700
+ for (upb_msg_field_begin(&it, msgdef);
1701
+ !upb_msg_field_done(&it);
1702
+ upb_msg_field_next(&it)) {
1703
+ const upb_fielddef* field = upb_msg_iter_field(&it);
1704
+ if (upb_fielddef_label(field) == UPB_LABEL_REQUIRED) {
1705
+ rb_raise(rb_eTypeError, "Required fields are unsupported in proto3.");
1706
+ }
1707
+ }
1708
+ }
1709
+
1710
+ static void validate_enumdef(const upb_enumdef* enumdef) {
1711
+ // Verify that an entry exists with integer value 0. (This is the default
1712
+ // value.)
1713
+ const char* lookup = upb_enumdef_iton(enumdef, 0);
1714
+ if (lookup == NULL) {
1715
+ rb_raise(rb_eTypeError,
1716
+ "Enum definition does not contain a value for '0'.");
1717
+ }
1718
+ }
1719
+
1720
+ /*
1721
+ * call-seq:
1722
+ * Builder.finalize_to_pool(pool)
1723
+ *
1724
+ * Adds all accumulated message and enum descriptors created in this builder
1725
+ * context to the given pool. The operation occurs atomically, and all
1726
+ * descriptors can refer to each other (including in cycles). This is the only
1727
+ * way to build (co)recursive message definitions.
1728
+ *
1729
+ * This method is usually called automatically by DescriptorPool#build after it
1730
+ * invokes the given user block in the context of the builder. The user should
1731
+ * not normally need to call this manually because a Builder is not normally
1732
+ * created manually.
1733
+ */
1734
+ VALUE Builder_finalize_to_pool(VALUE _self, VALUE pool_rb) {
1735
+ DEFINE_SELF(Builder, self, _self);
1736
+
1737
+ DescriptorPool* pool = ruby_to_DescriptorPool(pool_rb);
1738
+
1739
+ REALLOC_N(self->defs, upb_def*, RARRAY_LEN(self->pending_list));
1740
+
1741
+ for (int i = 0; i < RARRAY_LEN(self->pending_list); i++) {
1742
+ VALUE def_rb = rb_ary_entry(self->pending_list, i);
1743
+ if (CLASS_OF(def_rb) == cDescriptor) {
1744
+ self->defs[i] = (upb_def*)ruby_to_Descriptor(def_rb)->msgdef;
1745
+ validate_msgdef((const upb_msgdef*)self->defs[i]);
1746
+ } else if (CLASS_OF(def_rb) == cEnumDescriptor) {
1747
+ self->defs[i] = (upb_def*)ruby_to_EnumDescriptor(def_rb)->enumdef;
1748
+ validate_enumdef((const upb_enumdef*)self->defs[i]);
1749
+ }
1750
+ }
1751
+
1752
+ CHECK_UPB(upb_symtab_add(pool->symtab, (upb_def**)self->defs,
1753
+ RARRAY_LEN(self->pending_list), NULL, &status),
1754
+ "Unable to add defs to DescriptorPool");
1755
+
1756
+ for (int i = 0; i < RARRAY_LEN(self->pending_list); i++) {
1757
+ VALUE def_rb = rb_ary_entry(self->pending_list, i);
1758
+ add_def_obj(self->defs[i], def_rb);
1759
+ }
1760
+
1761
+ self->pending_list = rb_ary_new();
1762
+ return Qnil;
1763
+ }