xdrgen 0.0.1

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

@@ -0,0 +1,12 @@
+module Xdrgen::AST
+  module Definitions
+    class StructMember < Base
+      delegate :name, to: :declaration
+      delegate :type, to: :declaration
+
+      def optional?
+        declaration.is_a?(Declarations::Optional)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,19 @@
+module Xdrgen::AST
+  module Definitions
+    class Typedef < Base
+      include Concerns::Contained
+      delegate :name, to: :declaration
+      delegate :type, to: :declaration
+      delegate :sub_type, to: :type
+
+      def resolved_type
+        cur = self
+
+        cur = root.find_definition(cur.type.text_value) while cur.is_a?(Typedef)
+
+        cur
+      end
+
+    end
+  end
+end

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

@@ -0,0 +1,55 @@
+module Xdrgen::AST
+  module Definitions
+    class Union < Base
+      extend Memoist
+      include Concerns::Named
+      include Concerns::Namespace
+      include Concerns::Contained
+
+      delegate :discriminant, to: :union_body
+      delegate :name, to: :discriminant, prefix:true
+      delegate :arms, to: :union_body
+      delegate :normal_arms, to: :union_body
+      delegate :default_arm, to: :union_body
+
+      memoize def discriminant_type
+        return nil unless discriminant.type.is_a?(Identifier)
+
+        root.find_definition discriminant.type.name
+      end
+
+      def resolved_case(kase)
+        if discriminant_type.nil? then
+          # discriminant_type has not been found we need to search for the value in namespace's enum constants.
+          # It's a case where union discriminant is a standard type (like `int`):
+          #
+          # enum StellarValueType
+          # {
+          #     STELLAR_VALUE_BASIC = 0,
+          #     STELLAR_VALUE_SIGNED = 1
+          # };
+          #
+          # union switch (int v)
+          # {
+          # case STELLAR_VALUE_BASIC:
+          #     void;
+          #     ...
+          found = namespace.find_enum_value(kase.value_s)
+          raise "Case error:  #{kase} (#{kase.value_s}) constant not found" if found.nil?
+        else
+          found = discriminant_type.members.find{|m| m.name == kase.value_s}
+          raise "Case error:  #{kase} is not a member of #{discriminant_type.name}" if found.nil?
+        end
+        found
+      end
+
+      def nested_definitions
+        arms.
+          map(&:declaration).
+          reject{|d| d.is_a?(Declarations::Void)}.
+          map(&:type).
+          select{|d| d.is_a?(Concerns::NestedDefinition)}
+      end
+    end
+  end
+end

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

@@ -0,0 +1,33 @@
+module Xdrgen::AST
+  module Definitions
+    class UnionArm < Base
+      extend Memoist
+      include Concerns::Contained
+
+      delegate :name, to: :declaration
+      delegate :type, to: :declaration
+
+      memoize def union
+        find_ancestors(Union).last
+      end
+
+      def cases
+        cases_n.elements
+      end
+
+      def resolved_case(kase)
+        union.resolved_case kase
+      end
+
+      def resolved_cases
+        enum = union.discriminant_type
+
+        cases.map{|kase| resolved_case kase}
+      end
+
+      def void?
+        declaration.is_a?(Declarations::Void)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,11 @@
+module Xdrgen::AST
+  module Definitions
+    class UnionArmCase < Base
+
+      def value_s
+        value.text_value
+      end
+      
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+module Xdrgen::AST
+  module Definitions
+    class UnionBody < Base
+      extend Memoist
+
+      memoize def arms
+        [
+          normal_arms,
+          default_arm,
+        ].flatten.compact
+      end
+
+      def normal_arms
+        arms_n.elements.select{|c| c.is_a?(UnionArm)}
+      end
+
+      def default_arm
+        default_case_n unless default_case_n.terminal?
+      end
+    end
+  end
+end

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

@@ -0,0 +1,19 @@
+module Xdrgen::AST
+  module Definitions
+    class UnionDefaultArm < Base
+      extend Memoist
+      include Concerns::Contained
+      
+      delegate :name, to: :declaration
+      delegate :type, to: :declaration
+
+      memoize def union
+        find_ancestors(Union).last
+      end
+
+      def void?
+        declaration.is_a?(Declarations::Void)
+      end
+    end
+  end
+end

data/lib/xdrgen/ast/fixed_size.rb

@@ -0,0 +1,23 @@
+module Xdrgen::AST
+  class FixedSize < Treetop::Runtime::SyntaxNode
+    def size
+      size_t.text_value
+    end
+
+    def resolved_size
+      return size unless named?
+
+      resolved = root.find_definition(size)
+
+      if resolved.blank?
+        raise "Could not resolve constant: #{size}"
+      end
+
+      resolved.value
+    end
+
+    def named?
+      size_t.is_a?(Xdrgen::AST::Identifier)
+    end
+  end
+end

data/lib/xdrgen/ast/hexadecimal_constant.rb

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

data/lib/xdrgen/ast/identifier.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST
+  class Identifier < Treetop::Runtime::SyntaxNode
+    alias_method :name, :text_value
+  end
+end

data/lib/xdrgen/ast/octal_constant.rb

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

data/lib/xdrgen/ast/top.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST
+  class Top < Treetop::Runtime::SyntaxNode
+    include Concerns::HasDefinitions
+  end
+end

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

@@ -0,0 +1,27 @@
+module Xdrgen::AST::Typespecs
+  module Base
+    include Xdrgen::AST::Concerns::Contained
+
+    def decl
+      find_ancestors(Xdrgen::AST::Declarations::Base).last
+    end
+
+    def sub_type
+      case decl
+      when Xdrgen::AST::Declarations::Optional
+        :optional
+      when Xdrgen::AST::Declarations::Array
+        decl.fixed? ? :array : :var_array
+      else
+        :simple
+      end
+    end
+
+    def array_size
+      raise "Called array_size on a non-array!" unless decl.is_a?(Xdrgen::AST::Declarations::Array)
+
+      [decl.size_spec.named?, decl.size]
+    end
+
+  end
+end

data/lib/xdrgen/ast/typespecs/bool.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Typespecs
+  class Bool < Treetop::Runtime::SyntaxNode
+    include Base
+  end
+end

data/lib/xdrgen/ast/typespecs/double.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Typespecs
+  class Double < Treetop::Runtime::SyntaxNode
+    include Base
+  end
+end

data/lib/xdrgen/ast/typespecs/float.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Typespecs
+  class Float < Treetop::Runtime::SyntaxNode
+    include Base
+  end
+end

data/lib/xdrgen/ast/typespecs/hyper.rb

@@ -0,0 +1,6 @@
+module Xdrgen::AST::Typespecs
+  class Hyper < Treetop::Runtime::SyntaxNode
+    include Base
+
+  end
+end

data/lib/xdrgen/ast/typespecs/int.rb

@@ -0,0 +1,6 @@
+module Xdrgen::AST::Typespecs
+  class Int < Treetop::Runtime::SyntaxNode
+    include Base
+
+  end
+end

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

@@ -0,0 +1,10 @@
+module Xdrgen::AST::Typespecs
+  class Opaque < Treetop::Runtime::SyntaxNode
+    include Base
+
+    delegate :size, to: :decl
+    delegate :name, to: :decl
+    delegate :fixed?, to: :decl
+
+  end
+end

data/lib/xdrgen/ast/typespecs/quadruple.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Typespecs
+  class Quadruple < Treetop::Runtime::SyntaxNode
+    include Base
+  end
+end

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

@@ -0,0 +1,14 @@
+module Xdrgen::AST::Typespecs
+  module Simple
+    include Base
+
+    def resolved_type
+      name = self.text_value.split("::").last
+      result = root.find_definition name
+
+      raise Xdrgen::TypeResolutionError, "Cannot resolve type `#{name}`" if result.blank?
+      result
+    end
+
+  end
+end

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

@@ -0,0 +1,9 @@
+module Xdrgen::AST::Typespecs
+  class String < Treetop::Runtime::SyntaxNode
+    include Base
+    
+    delegate :size, to: :decl
+    delegate :name, to: :decl
+
+  end
+end

data/lib/xdrgen/ast/typespecs/unsigned_hyper.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Typespecs
+  class UnsignedHyper < Treetop::Runtime::SyntaxNode
+    include Base
+  end
+end

data/lib/xdrgen/ast/typespecs/unsigned_int.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST::Typespecs
+  class UnsignedInt < Treetop::Runtime::SyntaxNode
+    include Base
+  end
+end

data/lib/xdrgen/ast/var_size.rb

@@ -0,0 +1,26 @@
+module Xdrgen::AST
+  class VarSize < Treetop::Runtime::SyntaxNode
+    include Concerns::Contained
+
+    def size
+      return nil if size_t.text_value.blank?
+      size_t.text_value
+    end
+
+    def resolved_size
+      return size unless named?
+
+      resolved = root.find_definition(size)
+
+      if resolved.blank?
+        raise "Could not resolve constant: #{size}"
+      end
+
+      resolved.value
+    end
+
+    def named?
+      size_t.is_a?(Xdrgen::AST::Identifier)
+    end
+  end
+end

data/lib/xdrgen/ast/whitespace.rb

@@ -0,0 +1,5 @@
+module Xdrgen::AST
+  module Whitespace
+
+  end
+end

data/lib/xdrgen/cli.rb

@@ -0,0 +1,31 @@
+require 'slop'
+
+module Xdrgen
+  module CLI
+    def self.run(args)
+      args = args.dup
+      opts = Slop.parse! args do
+        banner 'Usage: xdrgen -o OUTPUT_DIR INPUT --gen=ruby'
+        on 'o', 'output=', 'The output directory'
+        on 'l', 'language=', 'The output language', default: 'ruby'
+        on 'n', 'namespace=', '"namespace" to generate code within (language-specific)'
+      end
+
+      fail(opts) if args.blank?
+      fail(opts) if opts[:output].blank?
+
+      compilation = Compilation.new(
+        args,
+        output_dir: opts[:output], 
+        language:   opts[:language].to_sym,
+        namespace:  opts[:namespace]
+      )
+      compilation.compile
+    end
+
+    def self.fail(slop, code=1)
+      STDERR.puts slop
+      exit(code)
+    end
+  end
+end

data/lib/xdrgen/compilation.rb

@@ -0,0 +1,31 @@
+module Xdrgen
+  class Compilation
+    extend Memoist
+
+    def initialize(source_paths, output_dir:".", language: :ruby, namespace: nil)
+      @source_paths = source_paths
+      @output_dir  = output_dir
+      @namespace   = namespace
+      @language    = language
+    end
+
+    memoize def source
+     @source_paths.map{|p| IO.read(p)}.join("\n")
+    end
+
+    memoize def ast
+      parser = Parser.new
+      parser.parse(source)
+    end
+
+    def compile
+      output = Output.new(@source_paths, @output_dir)
+
+      
+      generator = Generators.for_language(@language).new(ast, output, @namespace)
+      generator.generate
+    ensure
+      output.close
+    end
+  end
+end