protobuf 1.0.0

data/lib/protobuf/descriptor/descriptor.rb

@@ -0,0 +1,55 @@
+module Protobuf
+  module Descriptor
+    class Descriptor
+      def initialize(message_class)
+        @message_class = message_class
+      end
+
+      def proto_type
+        'Google::Protobuf::DescriptorProto'
+      end
+
+      def build(proto, opt={})
+        mod = opt[:module]
+        cls = mod.const_set(proto.name, Class.new(@message_class))
+        proto.nested_type.each do |message_proto|
+          Protobuf::Message.descriptor.build(message_proto, :module => cls)
+        end
+        proto.enum_type.each do |enum_proto|
+          Protobuf::Enum.descriptor.build(enum_proto, :module => cls)
+        end
+        proto.field.each do |field_proto|
+          Protobuf::Field::BaseField.descriptor.build(field_proto, :class => cls)
+        end
+      end
+
+      def unbuild(parent_proto)
+        message_proto = Google::Protobuf::DescriptorProto.new
+        message_proto.name = @message_class.to_s.split('::').last
+        @message_class.fields.each do |tag, field|
+          field.descriptor.unbuild(message_proto)
+        end
+        ObjectSpace.each_object(Class) do |cls|
+          if innerclass?(@message_class, cls)
+            cls.descriptor.unbuild(message_proto)
+          end
+        end
+
+        case parent_proto
+        when Google::Protobuf::FileDescriptorProto
+          parent_proto.message_type << message_proto
+        when Google::Protobuf::DescriptorProto
+          parent_proto.nested_type << message_proto
+        else
+          raise TypeError, parent_proto.class.name
+        end
+      end
+
+      private
+
+      def innerclass?(parent, child)
+        child.name =~ /::/ && child.name.split('::')[0..-2].join('::') == parent.name
+      end
+    end
+  end
+end

data/lib/protobuf/descriptor/descriptor_builder.rb

@@ -0,0 +1,143 @@
+require 'protobuf/descriptor/file_descriptor'
+
+module Protobuf
+  module Descriptor
+    def self.id2type(type_id)
+      require 'protobuf/descriptor/descriptor_proto'
+      case type_id
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_DOUBLE
+        :double
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_FLOAT
+        :float
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_INT64
+        :int64
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_UINT64
+        :unit64
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_INT32
+        :int64
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_FIXED64
+        :fixed64
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_FIXED32
+        :fixed32
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_BOOL
+        :bool
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_STRING
+        :string
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_GROUP
+        :group
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_MESSAGE
+        :message
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_BYTES
+        :bytes
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_UINT32
+        :uint32
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_ENUM
+        :enum
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_SFIXED32
+        :sfixed32
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_SFIXED64
+        :sfixed64
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_SINT32
+        :sint32
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_SINT64
+        :sint64
+      else
+        raise ArgumentError, "Invalid type: #{proto.type}"
+      end
+    end
+
+    def self.type2id(type)
+      require 'protobuf/descriptor/descriptor_proto'
+      case type
+      when :double
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_DOUBLE
+      when :float
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_FLOAT
+      when :int64
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_INT64
+      when :unit64
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_UINT64
+      when :int64
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_INT32
+      when :fixed64
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_FIXED64
+      when :fixed32
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_FIXED32
+      when Google::Protobuf::FieldDescriptorProto::Type::TYPE_BOOL
+        :bool
+      when :string
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_STRING
+      when :group
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_GROUP
+      when :message
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_MESSAGE
+      when :bytes
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_BYTES
+      when :uint32
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_UINT32
+      when :enum
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_ENUM
+      when :sfixed32
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_SFIXED32
+      when :sfixed64
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_SFIXED64
+      when :sint32
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_SINT32
+      when :sint64
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_SINT64
+      else
+        Google::Protobuf::FieldDescriptorProto::Type::TYPE_MESSAGE
+      end
+    end
+
+    def self.id2label(label_id)
+      require 'protobuf/descriptor/descriptor_proto'
+      case label_id
+      when Google::Protobuf::FieldDescriptorProto::Label::LABEL_REQUIRED
+        :required
+      when Google::Protobuf::FieldDescriptorProto::Label::LABEL_OPTIONAL
+        :optional
+      when Google::Protobuf::FieldDescriptorProto::Label::LABEL_REPEATED
+        :repeated
+      else
+        raise ArgumentError, "Invalid label: #{proto.label}"
+      end
+    end
+
+    def self.label2id(label)
+      require 'protobuf/descriptor/descriptor_proto'
+      case label
+      when :required
+        Google::Protobuf::FieldDescriptorProto::Label::LABEL_REQUIRED
+      when :optional
+        Google::Protobuf::FieldDescriptorProto::Label::LABEL_OPTIONAL
+      when :repeated
+        Google::Protobuf::FieldDescriptorProto::Label::LABEL_REPEATED
+      else
+        raise ArgumentError, "Invalid label: #{label}"
+      end
+    end
+
+    class DescriptorBuilder
+      class <<self
+
+        def proto_type
+          nil
+        end
+
+        def build(proto, opt={})
+          acceptable_descriptor(proto).build(proto)
+        end
+
+        def acceptable_descriptor(proto)
+          Protobuf::Descriptor.constants.each do |class_name|
+            descriptor_class = Protobuf::Descriptor.const_get(class_name)
+            if descriptor_class.respond_to?(:proto_type) && descriptor_class.proto_type == proto.class.name
+              return descriptor_class
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/protobuf/descriptor/descriptor_proto.rb

@@ -0,0 +1,138 @@
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/extend'
+
+module Google
+  module Protobuf
+    ::Protobuf::OPTIONS[:"java_package"] = "com.google.protobuf"
+    ::Protobuf::OPTIONS[:"java_outer_classname"] = "DescriptorProtos"
+    ::Protobuf::OPTIONS[:"optimize_for"] = :SPEED
+    class FileDescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :name, 1
+      optional :string, :package, 2
+      repeated :string, :dependency, 3
+      repeated :DescriptorProto, :message_type, 4
+      repeated :EnumDescriptorProto, :enum_type, 5
+      repeated :ServiceDescriptorProto, :service, 6
+      repeated :FieldDescriptorProto, :extension, 7
+      optional :FileOptions, :options, 8
+    end
+    class DescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :name, 1
+      repeated :FieldDescriptorProto, :field, 2
+      repeated :FieldDescriptorProto, :extension, 6
+      repeated :DescriptorProto, :nested_type, 3
+      repeated :EnumDescriptorProto, :enum_type, 4
+      class ExtensionRange < ::Protobuf::Message
+        defined_in __FILE__
+        optional :int32, :start, 1
+        optional :int32, :end, 2
+      end
+      repeated :ExtensionRange, :extension_range, 5
+      optional :MessageOptions, :options, 7
+    end
+    class FieldDescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      class Type < ::Protobuf::Enum
+        defined_in __FILE__
+        define :TYPE_DOUBLE, 1
+        define :TYPE_FLOAT, 2
+        define :TYPE_INT64, 3
+        define :TYPE_UINT64, 4
+        define :TYPE_INT32, 5
+        define :TYPE_FIXED64, 6
+        define :TYPE_FIXED32, 7
+        define :TYPE_BOOL, 8
+        define :TYPE_STRING, 9
+        define :TYPE_GROUP, 10
+        define :TYPE_MESSAGE, 11
+        define :TYPE_BYTES, 12
+        define :TYPE_UINT32, 13
+        define :TYPE_ENUM, 14
+        define :TYPE_SFIXED32, 15
+        define :TYPE_SFIXED64, 16
+        define :TYPE_SINT32, 17
+        define :TYPE_SINT64, 18
+      end
+      class Label < ::Protobuf::Enum
+        defined_in __FILE__
+        define :LABEL_OPTIONAL, 1
+        define :LABEL_REQUIRED, 2
+        define :LABEL_REPEATED, 3
+      end
+      optional :string, :name, 1
+      optional :int32, :number, 3
+      optional :Label, :label, 4
+      optional :Type, :type, 5
+      optional :string, :type_name, 6
+      optional :string, :extendee, 2
+      optional :string, :default_value, 7
+      optional :FieldOptions, :options, 8
+    end
+    class EnumDescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :name, 1
+      repeated :EnumValueDescriptorProto, :value, 2
+      optional :EnumOptions, :options, 3
+    end
+    class EnumValueDescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :name, 1
+      optional :int32, :number, 2
+      optional :EnumValueOptions, :options, 3
+    end
+    class ServiceDescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :name, 1
+      repeated :MethodDescriptorProto, :method, 2
+      optional :ServiceOptions, :options, 3
+    end
+    class MethodDescriptorProto < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :name, 1
+      optional :string, :input_type, 2
+      optional :string, :output_type, 3
+      optional :MethodOptions, :options, 4
+    end
+    class FileOptions < ::Protobuf::Message
+      defined_in __FILE__
+      optional :string, :java_package, 1
+      optional :string, :java_outer_classname, 8
+      optional :bool, :java_multiple_files, 10, :default => false
+      class OptimizeMode < ::Protobuf::Enum
+        defined_in __FILE__
+        define :SPEED, 1
+        define :CODE_SIZE, 2
+      end
+      optional :OptimizeMode, :optimize_for, 9, :default => :CODE_SIZE
+    end
+    class MessageOptions < ::Protobuf::Message
+      defined_in __FILE__
+      optional :bool, :message_set_wire_format, 1, :default => false
+    end
+    class FieldOptions < ::Protobuf::Message
+      defined_in __FILE__
+      optional :CType, :ctype, 1
+      class CType < ::Protobuf::Enum
+        defined_in __FILE__
+        define :CORD, 1
+        define :STRING_PIECE, 2
+      end
+      optional :string, :experimental_map_key, 9
+    end
+    class EnumOptions < ::Protobuf::Message
+      defined_in __FILE__
+    end
+    class EnumValueOptions < ::Protobuf::Message
+      defined_in __FILE__
+    end
+    class ServiceOptions < ::Protobuf::Message
+      defined_in __FILE__
+    end
+    class MethodOptions < ::Protobuf::Message
+      defined_in __FILE__
+    end
+  end
+end

data/lib/protobuf/descriptor/enum_descriptor.rb

@@ -0,0 +1,33 @@
+module Protobuf
+  module Descriptor
+    class EnumDescriptor
+      def initialize(enum_class)
+        @enum_class = enum_class
+      end
+
+      def proto_type
+        Google::Protobuf::EnumDescriptorProto
+      end
+
+      def build(proto, opt)
+        mod = opt[:module]
+        cls = mod.const_set(proto.name, Class.new(Protobuf::Enum))
+        proto.value.each do |value_proto|
+          cls.class_eval { define value_proto.name, value_proto.number }
+        end
+      end
+
+      def unbuild(parent_proto)
+        enum_proto = Google::Protobuf::EnumDescriptorProto.new
+        enum_proto.name = @enum_class.name.split('::').last
+        @enum_class.values.each do |name, value|
+          enum_value_proto = Google::Protobuf::EnumValueDescriptorProto.new
+          enum_value_proto.name = name.to_s
+          enum_value_proto.number = value.value
+          enum_proto.value << enum_value_proto
+        end
+        parent_proto.enum_type << enum_proto
+      end
+    end
+  end
+end

data/lib/protobuf/descriptor/field_descriptor.rb

@@ -0,0 +1,49 @@
+module Protobuf
+  module Descriptor
+    class FieldDescriptor
+      def initialize(field_instance=nil)
+        @field_instance = field_instance
+      end
+
+      def proto_type
+        'Google::Protobuf::FieldDescriptorProto'
+      end
+
+      def build(proto, opt={})
+        cls = opt[:class]
+        rule = Protobuf::Descriptor.id2label(proto.label)
+        type = Protobuf::Descriptor.id2type(proto.type)
+        type = proto.type_name.to_sym if [:message, :enum].include?(type)
+        opts = {}
+        opts[:default] = proto.default_value if proto.default_value
+        cls.define_field(rule, type, proto.name, proto.number, opts)
+      end
+
+      def unbuild(parent_proto, extension=false)
+        field_proto = Google::Protobuf::FieldDescriptorProto.new
+        field_proto.name = @field_instance.name.to_s
+        field_proto.number = @field_instance.tag
+        field_proto.label = Protobuf::Descriptor.label2id(@field_instance.rule)
+        field_proto.type = Protobuf::Descriptor.type2id(@field_instance.type)
+        if [Google::Protobuf::FieldDescriptorProto::Type::TYPE_MESSAGE,
+          Google::Protobuf::FieldDescriptorProto::Type::TYPE_ENUM].include? field_proto.type
+          field_proto.type_name = @field_instance.type.to_s.split('::').last
+        end
+        field_proto.default_value = @field_instance.default.to_s if @field_instance.default
+
+        case parent_proto
+        when Google::Protobuf::FileDescriptorProto
+          parent_proto.extension << field_proto
+        when Google::Protobuf::DescriptorProto
+          if extension
+            parent_proto.extension << field_proto
+          else
+            parent_proto.field << field_proto
+          end
+        else
+          raise TypeError, parent_proto.class.name
+        end
+      end
+    end
+  end
+end

data/lib/protobuf/descriptor/file_descriptor.rb

@@ -0,0 +1,37 @@
+module Protobuf
+  module Descriptor
+    class FileDescriptor
+      class <<self
+        def proto_type
+          'Google::Protobuf::FileDescriptorProto'
+        end
+
+        def build(proto, opt={})
+          mod = Object
+          if package = proto.package and not package.empty?
+            module_names = package.split '::'
+            module_names.each do |module_name|
+              mod = mod.const_set module_name, Module.new
+            end
+          end
+          proto.message_type.each do |message_proto|
+            Protobuf::Message.descriptor.build message_proto, :module => mod
+          end
+          proto.enum_type.each do |enum_proto|
+            Protobuf::Enum.descriptor.build enum_proto, :module => mod
+          end
+        end
+
+        def unbuild(messages)
+          messages = [messages] unless messages.is_a? Array
+          proto = Google::Protobuf::FileDescriptorProto.new
+          proto.package = messages.first.to_s.split('::')[0..-2].join('::') if messages.first.to_s =~ /::/
+          messages.each do |message|
+            message.descriptor.unbuild proto
+          end
+          proto
+        end
+      end
+    end
+  end
+end