twostroke 0.0.4 → 0.1.0

data/lib/twostroke/ast/array.rb

@@ -10,5 +10,13 @@ module Twostroke::AST
     def collapse
       self.class.new items: items.map(&:collapse)
     end
+    
+    def walk(&bk)
+      if yield self
+        items.each do |item|
+          item.walk &bk
+        end
+      end
+    end
   end
 end

data/lib/twostroke/ast/assignment.rb

@@ -5,5 +5,12 @@ module Twostroke::AST
     def collapse
       self.class.new left: left.collapse, right: right.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        left.walk &bk
+        right.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/binary_operators.rb

@@ -16,6 +16,13 @@ module Twostroke::AST
       def collapse
         self.class.new left: left.collapse, right: right.collapse, assign_result_left: assign_result_left
       end
+      
+      def walk(&bk)
+        if yield self
+          left.walk &bk
+          right.walk &bk
+        end
+      end
     end
     const_set op, klass
   end

data/lib/twostroke/ast/body.rb

@@ -10,5 +10,11 @@ module Twostroke::AST
     def collapse
       self.class.new statements: statements.reject(&:nil?).map(&:collapse)
     end
+    
+    def walk(&bk)
+      if yield self
+        statements.each { |s| s.walk &bk }
+      end
+    end
   end
 end

data/lib/twostroke/ast/break.rb

@@ -3,5 +3,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/call.rb

@@ -10,5 +10,12 @@ module Twostroke::AST
     def collapse
       self.class.new callee: callee.collapse, arguments: arguments.map(&:collapse)
     end
+    
+    def walk(&bk)
+      if yield self
+        callee.walk &bk
+        arguments.each { |arg| arg.walk &bk }
+      end
+    end
   end
 end

data/lib/twostroke/ast/case.rb

@@ -3,10 +3,18 @@ module Twostroke::AST
     attr_accessor :expression, :statements
     def initialize(*args)
       @statements = []
+      @is_default = false
       super *args
     end
     def collapse
-      self.class.new expression: expression.collapse, statements: statements.collect(&:collapse)
+      self.class.new expression: expression && expression.collapse, statements: statements.collect(&:collapse)
+    end
+    
+    def walk(&bk)
+      if yield self
+        expression.walk &bk
+        statements.each { |s| s.walk &bk }
+      end
     end
   end
 end

data/lib/twostroke/ast/continue.rb

@@ -0,0 +1,11 @@
+module Twostroke::AST
+  class Continue < Base
+    def collapse
+      self
+    end
+    
+    def walk
+      yield self
+    end
+  end
+end

data/lib/twostroke/ast/declaration.rb

@@ -5,5 +5,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/delete.rb

@@ -5,5 +5,11 @@ module Twostroke::AST
     def collapse
       self.class.new expression: expression.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        expression.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/do_while.rb

@@ -5,5 +5,12 @@ module Twostroke::AST
     def collapse
       self.class.new body: body.collapse, condition: condition.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        condition.walk &bk
+        body.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/false.rb

@@ -0,0 +1,11 @@
+module Twostroke::AST
+  class False < Base    
+    def collapse
+      self
+    end
+    
+    def walk(&bk)
+      yield self
+    end
+  end
+end

data/lib/twostroke/ast/for_in.rb

@@ -5,5 +5,13 @@ module Twostroke::AST
     def collapse
       self.class.new lval: lval.collapse, object: object.collapse, body: body.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        lval.walk &bk
+        object.walk &bk
+        body.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/for_loop.rb

@@ -6,7 +6,16 @@ module Twostroke::AST
       self.class.new initializer: initializer && initializer.collapse,
         condition: condition && condition.collapse,
         increment: increment && increment.collapse,
-        body: body.collapse
+        body: body && body.collapse
+    end
+    
+    def walk(&bk)
+      if yield self
+        initializer.walk(&bk) if initializer
+        condition.walk(&bk) if condition
+        increment.walk(&bk) if increment
+        body.walk &bk if body
+      end
     end
   end
 end

data/lib/twostroke/ast/function.rb

@@ -11,5 +11,11 @@ module Twostroke::AST
     def collapse
       self.class.new name: name, arguments: arguments, statements: statements.reject(&:nil?).map(&:collapse)
     end
+    
+    def walk(&bk)
+      if yield self
+        statements.each { |s| s.walk &bk }
+      end
+    end
   end
 end

data/lib/twostroke/ast/if.rb

@@ -5,5 +5,13 @@ module Twostroke::AST
     def collapse
       self.class.new condition: condition.collapse, then: @then.collapse, else: @else && @else.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        condition.walk &bk
+        @then.walk &bk
+        @else.walk &bk if @else
+      end
+    end
   end
 end

data/lib/twostroke/ast/index.rb

@@ -5,5 +5,12 @@ module Twostroke::AST
     def collapse
       self.class.new object: object.collapse, index: index.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        object.walk &bk
+        index.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/member_access.rb

@@ -5,5 +5,11 @@ module Twostroke::AST
     def collapse
       self.class.new object: object.collapse, member: member
     end
+    
+    def walk(&bk)
+      if yield self
+        object.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/multi_expression.rb

@@ -5,5 +5,12 @@ module Twostroke::AST
     def collapse
       self.class.new left: left.collapse, right: right.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        left.walk &bk
+        right.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/new.rb

@@ -10,5 +10,12 @@ module Twostroke::AST
     def collapse
       self.class.new callee: callee.collapse, arguments: arguments.map(&:collapse)
     end
+    
+    def walk(&bk)
+      if yield self
+        callee.walk &bk
+        arguments.each { |a| a.walk &bk }
+      end
+    end
   end
 end

data/lib/twostroke/ast/null.rb

@@ -3,5 +3,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/number.rb

@@ -5,5 +5,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/object_literal.rb

@@ -11,5 +11,11 @@ module Twostroke::AST
       collapsed = items.map { |k,v| [k, v.collapse] }
       self.class.new items: collapsed
     end
+    
+    def walk(&bk)
+      if yield self
+        items.each { |i| i[1].walk &bk }
+      end
+    end
   end
 end

data/lib/twostroke/ast/regexp.rb

@@ -5,5 +5,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk(&bk)
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/return.rb

@@ -5,5 +5,11 @@ module Twostroke::AST
     def collapse
       self.class.new expression: expression && expression.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        expression.walk &bk if expression
+      end
+    end
   end
 end

data/lib/twostroke/ast/string.rb

@@ -5,5 +5,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/switch.rb

@@ -8,5 +8,12 @@ module Twostroke::AST
     def collapse
       self.class.new expression: expression.collapse, cases: cases.collect(&:collapse)
     end
+    
+    def walk(&bk)
+      if yield self
+        expression.walk &bk
+        cases.each { |c| c.walk &bk }
+      end
+    end
   end
 end

data/lib/twostroke/ast/ternary.rb

@@ -5,5 +5,13 @@ module Twostroke::AST
     def collapse
       self.class.new condition: condition.collapse, if_true: if_true.collapse, if_false: if_false.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        condition.walk &bk
+        if_true.walk &bk
+        if_false.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/this.rb

@@ -1,5 +1,9 @@
 module Twostroke::AST
   class This < Base
     def collapse; self; end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/throw.rb

@@ -5,5 +5,11 @@ module Twostroke::AST
     def collapse
       self.class.new expression: expression.collapse
     end
+    
+    def walk(&bk)
+      if yield self
+        expression.walk &bk
+      end
+    end
   end
 end

data/lib/twostroke/ast/true.rb

@@ -0,0 +1,11 @@
+module Twostroke::AST
+  class True < Base    
+    def collapse
+      self
+    end
+    
+    def walk(&bk)
+      yield self
+    end
+  end
+end

data/lib/twostroke/ast/try.rb

@@ -7,5 +7,13 @@ module Twostroke::AST
         catch_statements: (catch_statements && catch_statements.map(&:collapse)),
         finally_statements: (finally_statements && finally_statements.map(&:collapse))
     end
+    
+    def walk(&bk)
+      if yield self
+        try_statements.each { |s| s.walk &bk }
+        catch_statements.each { |s| s.walk &bk } if catch_statements
+        finally_statements.each { |s| s.walk &bk } if finally_statements
+      end
+    end
   end
 end

data/lib/twostroke/ast/unary_operators.rb

@@ -1,12 +1,18 @@
 module Twostroke::AST
   [ :PostIncrement, :PreIncrement, :PostDecrement, :PreDecrement,
-    :BinaryNot, :UnaryPlus, :Negation, :TypeOf, :Not, :Void ].each do |op|
+    :BinaryNot, :UnaryPlus, :Negation, :TypeOf, :Not, :Void, :BracketedExpression ].each do |op|
       klass = Class.new Base do
         attr_accessor :value
       
         def collapse
           self.class.new value: value.collapse
         end
+        
+        def walk(&bk)
+          if yield self
+            value.walk &bk
+          end
+        end
       end
       const_set op, klass
     end

data/lib/twostroke/ast/variable.rb

@@ -5,5 +5,9 @@ module Twostroke::AST
     def collapse
       self
     end
+    
+    def walk
+      yield self
+    end
   end
 end

data/lib/twostroke/ast/while.rb

@@ -3,7 +3,14 @@ module Twostroke::AST
     attr_accessor :condition, :body
     
     def collapse
-      self.class.new condition: condition.collapse, body: body.collapse
+      self.class.new condition: condition.collapse, body: body && body.collapse
+    end
+    
+    def walk(&bk)
+      if yield self
+        condition.walk &bk
+        body.walk &bk if body
+      end
     end
   end
 end

data/lib/twostroke/ast/with.rb

@@ -0,0 +1,16 @@
+module Twostroke::AST
+  class With < Base
+    attr_accessor :object, :statement
+    
+    def collapse
+      self.class.new object: object.collapse, statement: statement && statement.collapse
+    end
+    
+    def walk(&bk)
+      if yield self
+        object.walk &bk
+        statement.walk &bk
+      end
+    end
+  end
+end