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 com.google.protobuf.Descriptors; 36 import org.jruby.RubyModule; 37 import org.jruby.RubyNumeric; 38 import org.jruby.anno.JRubyMethod; 39 import org.jruby.runtime.ThreadContext; 40 import org.jruby.runtime.builtin.IRubyObject; 41 42 public class RubyEnum { 43 /* 44 * call-seq: 45 * Enum.lookup(number) => name 46 * 47 * This module method, provided on each generated enum module, looks up an enum 48 * value by number and returns its name as a Ruby symbol, or nil if not found. 49 */ 50 @JRubyMethod(meta = true) lookup(ThreadContext context, IRubyObject recv, IRubyObject number)51 public static IRubyObject lookup(ThreadContext context, IRubyObject recv, IRubyObject number) { 52 RubyEnumDescriptor rubyEnumDescriptorescriptor = (RubyEnumDescriptor) getDescriptor(context, recv); 53 Descriptors.EnumDescriptor descriptor = rubyEnumDescriptorescriptor.getDescriptor(); 54 Descriptors.EnumValueDescriptor value = descriptor.findValueByNumber(RubyNumeric.num2int(number)); 55 if (value == null) return context.runtime.getNil(); 56 return context.runtime.newSymbol(value.getName()); 57 } 58 59 /* 60 * call-seq: 61 * Enum.resolve(name) => number 62 * 63 * This module method, provided on each generated enum module, looks up an enum 64 * value by name (as a Ruby symbol) and returns its name, or nil if not found. 65 */ 66 @JRubyMethod(meta = true) resolve(ThreadContext context, IRubyObject recv, IRubyObject name)67 public static IRubyObject resolve(ThreadContext context, IRubyObject recv, IRubyObject name) { 68 RubyEnumDescriptor rubyEnumDescriptorescriptor = (RubyEnumDescriptor) getDescriptor(context, recv); 69 Descriptors.EnumDescriptor descriptor = rubyEnumDescriptorescriptor.getDescriptor(); 70 Descriptors.EnumValueDescriptor value = descriptor.findValueByName(name.asJavaString()); 71 if (value == null) return context.runtime.getNil(); 72 return context.runtime.newFixnum(value.getNumber()); 73 } 74 75 /* 76 * call-seq: 77 * Enum.descriptor 78 * 79 * This module method, provided on each generated enum module, returns the 80 * EnumDescriptor corresponding to this enum type. 81 */ 82 @JRubyMethod(meta = true, name = "descriptor") getDescriptor(ThreadContext context, IRubyObject recv)83 public static IRubyObject getDescriptor(ThreadContext context, IRubyObject recv) { 84 return ((RubyModule) recv).getInstanceVariable(Utils.DESCRIPTOR_INSTANCE_VAR); 85 } 86 } 87