adding ecma visitr

git-svn-id: svn+ssh://rubyforge.org/var/svn/johnson/trunk@98 54575175-8111-4fdf-a583-07ff49f40e23

lib/johnson/nodes/node.rb

@@ -41,6 +41,10 @@ def initialize(line, column, value)
       def to_sexp
         SexpVisitor.new.accept(self)
       end
+
+      def to_ecma
+        EcmaVisitor.new.accept(self)
+      end
     end
     SINGLE_NODES.each { |se| const_set(se.to_sym, Class.new(Node)) }
   end

lib/johnson/visitors.rb

@@ -1 +1,2 @@
 require 'johnson/visitors/sexp_visitor'
+require 'johnson/visitors/ecma_visitor'

lib/johnson/visitors/ecma_visitor.rb

@@ -0,0 +1,71 @@
+module Johnson
+  module Visitors
+    class EcmaVisitor
+      def initialize
+        @depth = 0
+      end
+
+      def visit_SourceElements(o)
+        (@depth == 0 ? '' : '{') +
+          o.value.map { |x| "#{indent}#{x.accept(self)}" }.join("\n") +
+          (@depth == 0 ? '' : '}')
+      end
+
+      %w{ Name Number Regexp String }.each do |type|
+        define_method(:"visit_#{type}") do |o|
+          o.value
+        end
+      end
+
+      {
+        'OpEqual'             => '=',
+        'StrictNotEqual'      => '!==',
+        'StrictEqual'         => '===',
+        'Or'                  => '||',
+        'OpURShift'           => '>>>',
+        'OpURShiftEqual'      => '>>>=',
+        'OpSubtract'          => '-',
+        'OpSubtractEqual'     => '-=',
+        'OpRShift'            => '>>',
+        'OpRShiftEqual'       => '>>=',
+        'OpMultiply'          => '*',
+        'OpMultiplyEqual'     => '*=',
+        'OpMod'               => '%',
+        'OpModEqual'          => '%=',
+        'OpLShift'            => '<<',
+        'OpLShiftEqual'       => '<<=',
+        'OpDivide'            => '/',
+        'OpDivideEqual'       => '/=',
+        'OpBitXor'            => '^',
+        'OpBitXorEqual'       => '^=',
+        'OpBitOr'             => '|',
+        'OpBitOrEqual'        => '|=',
+        'OpBitAnd'            => '&',
+        'OpBitAndEqual'       => '&=',
+        'OpAdd'               => '+',
+        'OpAddEqual'          => '+=',
+        'NotEqual'            => '!=',
+        'LessThan'            => '<',
+        'LessThanOrEqual'     => '<=',
+        'GreaterThan'         => '>',
+        'GreaterThanOrEqual'  => '>=',
+        'And'                 => '&&',
+        'InstanceOf'          => 'instanceof',
+        'Equal'               => '==',
+      }.each do |type,op|
+        define_method(:"visit_#{type}") do |o|
+          "#{o.left && o.left.accept(self)}" \
+            " #{op} " \
+            "#{o.right && o.right.accept(self)}"
+        end
+      end
+
+      def accept(target)
+        target.accept(self)
+      end
+
+      private
+      def indent; ' ' * @depth * 2; end
+    end
+  end
+end

test/helper.rb

@@ -37,5 +37,9 @@ def setup
     def assert_sexp(expected, actual)
       assert_equal(expected, actual.to_sexp)
     end
+
+    def assert_ecma(expected, actual)
+      assert_equal(expected, actual.to_ecma)
+    end
   end
 end

test/johnson/nodes/binary_node_test.rb

@@ -51,6 +51,10 @@ class BinaryNodeTest < Johnson::NodeTestCase
                    )
       end
     end
+
+    define_method(:"test_#{op}_to_ecma") do
+      assert_ecma("i #{sym} 10", @parser.parse("i #{sym} 10"))
+    end
   end
   {
     :and                => '&&',