xdrgen 0.0.1

data/lib/xdrgen/ast/concerns/contained.rb

@@ -0,0 +1,32 @@
+module Xdrgen::AST
+  module Concerns
+    module Contained      
+      extend Memoist
+
+      protected
+      memoize def ancestors
+        current = self.parent
+        result = []
+
+        while current.present?
+          result.unshift(current)
+          current = current.parent
+        end
+        
+        result
+      end
+
+      def root
+        ancestors.first
+      end
+
+      def namespace
+        find_ancestors(HasDefinitions).last
+      end
+
+      def find_ancestors(type)
+        ancestors.select{|a| a.is_a?(type)}
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/concerns/has_children.rb

@@ -0,0 +1,9 @@
+module Xdrgen::AST
+  module Concerns
+    module HasChildren      
+      def children
+        children_n.elements
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/concerns/has_definitions.rb

@@ -0,0 +1,90 @@
+module Xdrgen::AST
+  module Concerns
+    module HasDefinitions
+      include HasChildren
+      
+      def typedefs
+        find_children(Definitions::Typedef)
+      end  
+
+      def consts
+        find_children(Definitions::Const)
+      end
+
+      def structs
+        find_children(Definitions::Struct)
+      end
+
+      def enums
+        find_children(Definitions::Enum)
+      end
+
+      def unions
+        find_children(Definitions::Union)
+      end
+
+      def namespaces
+        find_children(Definitions::Namespace)
+      end
+
+      def definitions
+        find_children(Definitions::Base)
+      end
+
+      def find_definition(name)
+        found = definitions.find{|d| d.name == name}
+        return found if found
+
+        namespaces.each do |ns|
+          found = ns.find_definition(name)
+          return found if found
+        end
+
+        nil
+      end
+
+      def find_enum_value(name)
+        enums.each do |e|
+          found = e.members.find{|d| d.name == name}
+          return found if found
+        end
+        raise "Could not find enum value #{name}"
+      end
+
+      ##
+      # Collapse the flat list of definitions in this 
+      # container into a nested array, grouping the
+      # definitions by contiguous types:
+      # 
+      # Example:
+      # 
+      # [Typedef, Typedef, Typedef, Const, Struct, Struct, Typedef]
+      # 
+      # becomes:
+      # 
+      # [[Typedef, Typedef, Typedef], [Const], [Struct, Struct], [Typedef]]
+      # 
+      # 
+      def definition_blocks
+        children.each_with_object([]) do |child, result|
+          next unless child.is_a?(Definitions::Base)
+
+          current_group = result.last
+          
+          if current_group.blank?
+            result.push [child]
+          elsif current_group.last.is_a?(child.class)
+            current_group.push child
+          else
+            result.push [child]
+          end
+        end
+      end
+
+      private
+      def find_children(type)
+        children.select{|c| c.is_a? type}
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/concerns/named.rb

@@ -0,0 +1,16 @@
+module Xdrgen::AST
+  module Concerns
+    module Named
+      delegate :name, to: :identifier
+
+      def namespaces
+        return [] unless self.is_a?(Contained)
+        self.find_ancestors(Concerns::Namespace)
+      end
+
+      def fully_qualified_name
+        namespaces.map(&:name) + [name]
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/concerns/namespace.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST
+  module Concerns
+    module Namespace
+      
+    end
+  end
+end

data/lib/xdrgen/ast/concerns/nested_definition.rb

@@ -0,0 +1,16 @@
+module Xdrgen::AST
+  module Concerns
+    module NestedDefinition
+
+      def name
+        find_ancestors(Declarations::Base).last.name
+      end
+
+      def parent_defn
+        find_ancestors(Definitions::Struct).last ||
+        find_ancestors(Definitions::Union).last
+      end
+      
+    end
+  end
+end

data/lib/xdrgen/ast/constant.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST
+  class Constant < Treetop::Runtime::SyntaxNode
+    def value
+      raise NotImplementedError, "implement in subclass"
+    end
+  end
+end

data/lib/xdrgen/ast/decimal_constant.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST
+  class DecimalConstant < Constant
+    def value
+      text_value.to_i
+    end
+  end
+end

data/lib/xdrgen/ast/declarations/array.rb

@@ -0,0 +1,15 @@
+module Xdrgen::AST::Declarations
+  class Array < Base
+    delegate :name, to: :identifier
+    delegate :size, to: :size_spec
+    delegate :resolved_size, to: :size_spec
+
+    def fixed?
+      size_spec.is_a?(Xdrgen::AST::FixedSize)
+    end
+
+    def child_type
+      type.text_value
+    end
+  end
+end

data/lib/xdrgen/ast/declarations/base.rb

@@ -0,0 +1,28 @@
+module Xdrgen::AST
+  module Declarations
+    class Base < Treetop::Runtime::SyntaxNode
+      TYPE_NODES = [
+        Typespecs::Base,
+        Concerns::NestedDefinition,
+        Concerns::Contained,
+      ]
+
+      def type
+        search(type_s) do |node|
+          TYPE_NODES.any?{|t| node.is_a?(t)}
+        end
+      end
+
+      private
+      def search(cur_el, &predicate)
+        return cur_el if predicate.call(cur_el)
+        return if cur_el.elements.blank?
+
+        cur_el.elements.each do |next_el|
+          child_result = search(next_el, &predicate)
+          return child_result if child_result.present?
+        end
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/declarations/opaque.rb

@@ -0,0 +1,11 @@
+module Xdrgen::AST::Declarations
+  class Opaque < Base
+    delegate :name, to: :identifier
+    delegate :size, to: :size_spec
+    delegate :resolved_size, to: :size_spec
+
+    def fixed?
+      size_spec.is_a?(Xdrgen::AST::FixedSize)
+    end
+  end
+end

data/lib/xdrgen/ast/declarations/optional.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Declarations
+  class Optional < Base
+    delegate :name, to: :identifier
+  end
+end

data/lib/xdrgen/ast/declarations/simple.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST::Declarations
+  class Simple < Base
+    delegate :name, to: :identifier
+    
+
+  end
+end

data/lib/xdrgen/ast/declarations/string.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST::Declarations
+  class String < Base
+    delegate :name, to: :identifier
+    delegate :size, to: :size_spec
+    delegate :resolved_size, to: :size_spec
+  end
+end

data/lib/xdrgen/ast/declarations/void.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST::Declarations
+  class Void < Base
+    def name
+      "void"
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/base.rb

@@ -0,0 +1,9 @@
+module Xdrgen::AST::Definitions
+  class Base < Treetop::Runtime::SyntaxNode
+
+    def sub_type
+      :simple
+    end
+    
+  end
+end

data/lib/xdrgen/ast/definitions/const.rb

@@ -0,0 +1,12 @@
+module Xdrgen::AST
+  module Definitions
+    class Const < Base
+      include Concerns::Named
+      include Concerns::Contained
+
+      def value
+        value_n.text_value
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/enum.rb

@@ -0,0 +1,17 @@
+module Xdrgen::AST
+  module Definitions
+    class Enum < Base
+      extend Memoist
+      include Concerns::Named
+      include Concerns::Contained
+
+      memoize def members
+        results = [enum_body.first_member_n]
+        enum_body.additional_members_n.elements.each do |n|
+          results.push n.enum_member
+        end
+        results
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/enum_member.rb

@@ -0,0 +1,44 @@
+module Xdrgen::AST
+  module Definitions
+    class EnumMember < Base
+      extend Memoist
+
+      include Concerns::Named
+      include Concerns::Contained
+
+      def value
+        unsigned_value = defined_value || auto_value
+
+        # enums are signed in xdr, so...
+        # convert to twos complement value
+        [unsigned_value].pack("l>").unpack("l>").first
+      end
+
+      memoize def enum
+        find_ancestors(Enum).last
+      end
+
+
+      def auto_value
+        index = enum.members.index(self)
+        if index == 0
+          0
+        else
+          # use the previous members value + 1
+          enum.members[index - 1].value + 1
+        end
+      end
+
+      def defined_value
+        return if value_n.terminal?
+
+        case value_n.val
+        when Constant
+          value_n.val.value
+        when Identifier
+          namespace.find_enum_value(value_n.val.name).defined_value
+        end
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/namespace.rb

@@ -0,0 +1,9 @@
+module Xdrgen::AST
+  module Definitions
+    class Namespace < Base
+      include Concerns::Named
+      include Concerns::Namespace
+      include Concerns::HasDefinitions
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/nested_enum.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST
+  module Definitions
+    class NestedEnum < Enum
+      include Concerns::NestedDefinition
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/nested_struct.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST
+  module Definitions
+    class NestedStruct < Struct
+      include Concerns::NestedDefinition
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/nested_union.rb

@@ -0,0 +1,7 @@
+module Xdrgen::AST
+  module Definitions
+    class NestedUnion < Union
+      include Concerns::NestedDefinition
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/struct.rb

@@ -0,0 +1,18 @@
+module Xdrgen::AST
+  module Definitions
+    class Struct < Base
+      include Concerns::Named
+      include Concerns::Namespace
+      include Concerns::Contained
+
+      delegate :members, to: :struct_body
+
+      def nested_definitions
+        members.
+          map(&:declaration).
+          map(&:type).
+          select{|d| d.is_a?(Concerns::NestedDefinition)}
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/definitions/struct_body.rb

@@ -0,0 +1,11 @@
+module Xdrgen::AST
+  module Definitions
+    class StructBody < Base
+      include Concerns::HasChildren
+
+      def members
+        children.select{|c| c.is_a?(StructMember)}
+      end
+    end
+  end
+end