1 /* 2 * Protocol Buffers - Google's data interchange format 3 * Copyright 2014 Google Inc. All rights reserved. 4 * https://developers.google.com/protocol-buffers/ 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions are 8 * met: 9 * 10 * * Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * * Redistributions in binary form must reproduce the above 13 * copyright notice, this list of conditions and the following disclaimer 14 * in the documentation and/or other materials provided with the 15 * distribution. 16 * * Neither the name of Google Inc. nor the names of its 17 * contributors may be used to endorse or promote products derived from 18 * this software without specific prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31 */ 32 33 package com.google.protobuf.jruby; 34 35 import org.jruby.Ruby; 36 import org.jruby.RubyClass; 37 import org.jruby.RubyModule; 38 import org.jruby.RubyObject; 39 import org.jruby.anno.JRubyClass; 40 import org.jruby.anno.JRubyMethod; 41 import org.jruby.runtime.ObjectAllocator; 42 import org.jruby.runtime.ThreadContext; 43 import org.jruby.runtime.builtin.IRubyObject; 44 45 @JRubyClass(name = "OneofBuilderContext") 46 public class RubyOneofBuilderContext extends RubyObject { createRubyOneofBuilderContext(Ruby runtime)47 public static void createRubyOneofBuilderContext(Ruby runtime) { 48 RubyModule protobuf = runtime.getClassFromPath("Google::Protobuf"); 49 RubyModule internal = protobuf.defineModuleUnder("Internal"); 50 RubyClass cRubyOneofBuidlerContext = internal.defineClassUnder("OneofBuilderContext", runtime.getObject(), new ObjectAllocator() { 51 @Override 52 public IRubyObject allocate(Ruby ruby, RubyClass rubyClass) { 53 return new RubyOneofBuilderContext(ruby, rubyClass); 54 } 55 }); 56 cRubyOneofBuidlerContext.defineAnnotatedMethods(RubyOneofBuilderContext.class); 57 } 58 RubyOneofBuilderContext(Ruby ruby, RubyClass rubyClass)59 public RubyOneofBuilderContext(Ruby ruby, RubyClass rubyClass) { 60 super(ruby, rubyClass); 61 } 62 63 @JRubyMethod initialize(ThreadContext context, IRubyObject oneofdef)64 public IRubyObject initialize(ThreadContext context, IRubyObject oneofdef) { 65 this.descriptor = (RubyOneofDescriptor) oneofdef; 66 this.cFieldDescriptor = (RubyClass) context.runtime.getClassFromPath("Google::Protobuf::FieldDescriptor"); 67 return this; 68 } 69 70 @JRubyMethod(required = 3, optional = 1) optional(ThreadContext context, IRubyObject[] args)71 public IRubyObject optional(ThreadContext context, IRubyObject[] args) { 72 IRubyObject name = args[0]; 73 IRubyObject type = args[1]; 74 IRubyObject number = args[2]; 75 IRubyObject typeClass = args.length > 3 ? args[3] : context.runtime.getNil(); 76 RubyFieldDescriptor fieldDescriptor = Utils.msgdefCreateField(context, "optional", 77 name, type, number, typeClass, cFieldDescriptor); 78 descriptor.addField(context, fieldDescriptor); 79 return this; 80 } 81 82 private RubyOneofDescriptor descriptor; 83 private RubyClass cFieldDescriptor; 84 } 85