rkelly-turbo 0.1.0

data/lib/rkelly/visitors/function_visitor.rb

@@ -0,0 +1,46 @@
+module RKelly
+  module Visitors
+    class FunctionVisitor < Visitor
+      attr_reader :scope_chain
+      def initialize(scope)
+        super()
+        @scope_chain = scope
+      end
+
+      def visit_SourceElementsNode(o)
+        o.value.each { |x| x.accept(self) }
+      end
+
+      def visit_FunctionDeclNode(o)
+        if o.value
+          scope_chain[o.value].value = RKelly::JS::Function.new(o.function_body, o.arguments)
+        end
+      end
+
+      %w{
+        AddNode ArgumentsNode ArrayNode AssignExprNode BitAndNode BitOrNode
+        BitXOrNode BitwiseNotNode BlockNode BracketAccessorNode BreakNode
+        CaseBlockNode CaseClauseNode CommaNode ConditionalNode
+        ConstStatementNode ContinueNode DeleteNode DivideNode
+        DoWhileNode DotAccessorNode ElementNode EmptyStatementNode EqualNode
+        ExpressionStatementNode FalseNode ForInNode ForNode FunctionBodyNode
+        FunctionExprNode GetterPropertyNode GreaterNode GreaterOrEqualNode
+        IfNode InNode InstanceOfNode LabelNode LeftShiftNode LessNode
+        LessOrEqualNode LogicalAndNode LogicalNotNode LogicalOrNode ModulusNode
+        MultiplyNode NewExprNode NotEqualNode NotStrictEqualNode NullNode
+        NumberNode ObjectLiteralNode OpAndEqualNode OpDivideEqualNode
+        OpEqualNode OpLShiftEqualNode OpMinusEqualNode OpModEqualNode
+        OpMultiplyEqualNode OpOrEqualNode OpPlusEqualNode OpRShiftEqualNode
+        OpURShiftEqualNode OpXOrEqualNode ParameterNode PostfixNode PrefixNode
+        PropertyNode RegexpNode ResolveNode ReturnNode RightShiftNode
+        SetterPropertyNode StrictEqualNode StringNode
+        SubtractNode SwitchNode ThisNode ThrowNode TrueNode TryNode TypeOfNode
+        UnaryMinusNode UnaryPlusNode UnsignedRightShiftNode VarDeclNode
+        VarStatementNode VoidNode WhileNode WithNode
+      }.each do |type|
+        define_method(:"visit_#{type}") do |o|
+        end
+      end
+    end
+  end
+end

data/lib/rkelly/visitors/pointcut_visitor.rb

@@ -0,0 +1,31 @@
+module RKelly
+  module Visitors
+    class PointcutVisitor < Visitor
+      attr_reader :matches
+      def initialize(pattern, matches = [])
+        @pattern  = pattern
+        @matches  = matches
+      end
+
+      def >(pattern)
+        pattern =
+          case pattern
+          when Class
+            pattern.new(Object)
+          else
+            pattern
+          end
+        self.class.new(nil, @matches.map do |m|
+          m.pointcut(pattern).matches
+        end.flatten)
+      end
+
+      ALL_NODES.each do |type|
+        define_method(:"visit_#{type}Node") do |o|
+          @matches << o if @pattern === o
+          super(o)
+        end
+      end
+    end
+  end
+end

data/lib/rkelly/visitors/real_sexp_visitor.rb

@@ -0,0 +1,16 @@
+module RKelly
+  module Visitors
+    class RealSexpVisitor < Visitor
+      ALL_NODES.each do |type|
+        eval <<-RUBY
+          def visit_#{type}Node(o)
+            sexp = s(:#{type.scan(/[A-Z][a-z]+/).join('_').downcase}, *super(o))
+            sexp.line = o.line if o.line
+            sexp.file = o.filename
+            sexp
+          end
+        RUBY
+      end
+    end
+  end
+end

data/lib/rkelly/visitors/sexp_visitor.rb

@@ -0,0 +1,373 @@
+module RKelly
+  module Visitors
+    class SexpVisitor < Visitor
+      def visit_NumberNode(o)
+        [:lit, o.value]
+      end
+
+      def visit_RegexpNode(o)
+        [:lit, o.value]
+      end
+
+      def visit_AssignExprNode(o)
+        [:assign, super]
+      end
+
+      def visit_VarDeclNode(o)
+        [ o.constant? ? :const_decl : :var_decl ] + super(o)
+      end
+
+      def visit_VarStatementNode(o)
+        [:var, super]
+      end
+
+      def visit_PostfixNode(o)
+        [:postfix, super, o.value]
+      end
+
+      def visit_PrefixNode(o)
+        [:prefix, super, o.value]
+      end
+
+      def visit_DeleteNode(o)
+        [:delete, super]
+      end
+
+      def visit_VoidNode(o)
+        [:void, super]
+      end
+
+      def visit_TypeOfNode(o)
+        [:typeof, super]
+      end
+
+      def visit_UnaryPlusNode(o)
+        [:u_plus, super]
+      end
+
+      def visit_UnaryMinusNode(o)
+        [:u_minus, super]
+      end
+
+      def visit_BitwiseNotNode(o)
+        [:bitwise_not, super]
+      end
+
+      def visit_LogicalNotNode(o)
+        [:not, super]
+      end
+
+      def visit_ConstStatementNode(o)
+        [:const, super]
+      end
+
+      def visit_MultiplyNode(o)
+        [:multiply, *super]
+      end
+
+      def visit_DivideNode(o)
+        [:divide, *super]
+      end
+
+      def visit_ModulusNode(o)
+        [:modulus, *super]
+      end
+
+      def visit_AddNode(o)
+        [:add, *super]
+      end
+
+      def visit_LeftShiftNode(o)
+        [:lshift, *super]
+      end
+
+      def visit_RightShiftNode(o)
+        [:rshift, *super]
+      end
+
+      def visit_UnsignedRightShiftNode(o)
+        [:urshift, *super]
+      end
+
+      def visit_SubtractNode(o)
+        [:subtract, *super]
+      end
+
+      def visit_LessNode(o)
+        [:less, *super]
+      end
+
+      def visit_GreaterNode(o)
+        [:greater, *super]
+      end
+
+      def visit_LessOrEqualNode(o)
+        [:less_or_equal, *super]
+      end
+
+      def visit_GreaterOrEqualNode(o)
+        [:greater_or_equal, *super]
+      end
+
+      def visit_InstanceOfNode(o)
+        [:instance_of, *super]
+      end
+
+      def visit_EqualNode(o)
+        [:equal, *super]
+      end
+
+      def visit_NotEqualNode(o)
+        [:not_equal, *super]
+      end
+
+      def visit_StrictEqualNode(o)
+        [:strict_equal, *super]
+      end
+
+      def visit_NotStrictEqualNode(o)
+        [:not_strict_equal, *super]
+      end
+
+      def visit_BitAndNode(o)
+        [:bit_and, *super]
+      end
+
+      def visit_BitOrNode(o)
+        [:bit_or, *super]
+      end
+
+      def visit_BitXOrNode(o)
+        [:bit_xor, *super]
+      end
+
+      def visit_LogicalAndNode(o)
+        [:and, *super]
+      end
+
+      def visit_LogicalOrNode(o)
+        [:or, *super]
+      end
+
+      def visit_InNode(o)
+        [:in, *super]
+      end
+
+      def visit_DoWhileNode(o)
+        [:do_while, *super]
+      end
+
+      def visit_WhileNode(o)
+        [:while, *super]
+      end
+
+      def visit_WithNode(o)
+        [:with, *super]
+      end
+
+      def visit_CaseClauseNode(o)
+        [:case, *super]
+      end
+
+      def visit_CaseBlockNode(o)
+        [:case_block, super]
+      end
+
+      def visit_SwitchNode(o)
+        [:switch, *super]
+      end
+
+      def visit_ForNode(o)
+        [ :for, *super]
+      end
+
+      def visit_BlockNode(o)
+        [:block, super]
+      end
+
+      def visit_IfNode(o)
+        [:if, *super].compact
+      end
+
+      def visit_ConditionalNode(o)
+        [:conditional, *super]
+      end
+
+      def visit_ForInNode(o)
+        [ :for_in, *super]
+      end
+
+      def visit_TryNode(o)
+        [ :try, *super]
+      end
+
+      def visit_EmptyStatementNode(o)
+        [:empty]
+      end
+
+      def visit_FunctionBodyNode(o)
+        [:func_body, super]
+      end
+
+      def visit_ResolveNode(o)
+        [:resolve, o.value]
+      end
+
+      def visit_BracketAccessorNode(o)
+        [:bracket_access, *super]
+      end
+
+      def visit_NewExprNode(o)
+        [:new_expr, *super]
+      end
+
+      def visit_ParameterNode(o)
+        [:param, o.value]
+      end
+
+      def visit_BreakNode(o)
+        [:break, o.value].compact
+      end
+
+      def visit_ContinueNode(o)
+        [:continue, o.value].compact
+      end
+
+      def visit_LabelNode(o)
+        [:label ] + super
+      end
+
+      def visit_ThrowNode(o)
+        [:throw, super]
+      end
+
+      def visit_ObjectLiteralNode(o)
+        [:object, super]
+      end
+
+      def visit_PropertyNode(o)
+        [ :property ] + super
+      end
+
+      def visit_GetterPropertyNode(o)
+        [ :getter ] + super
+      end
+
+      def visit_SetterPropertyNode(o)
+        [ :setter ] + super
+      end
+
+      def visit_ElementNode(o)
+        [:element, super ]
+      end
+
+      def visit_ExpressionStatementNode(o)
+        [:expression, super ]
+      end
+
+      def visit_OpEqualNode(o)
+        [:op_equal, *super ]
+      end
+
+      def visit_OpPlusEqualNode(o)
+        [:op_plus_equal, *super ]
+      end
+
+      def visit_OpMinusEqualNode(o)
+        [:op_minus_equal, *super ]
+      end
+
+      def visit_OpMultiplyEqualNode(o)
+        [:op_multiply_equal, *super ]
+      end
+
+      def visit_OpDivideEqualNode(o)
+        [:op_divide_equal, *super]
+      end
+
+      def visit_OpLShiftEqualNode(o)
+        [:op_lshift_equal, *super ]
+      end
+
+      def visit_OpRShiftEqualNode(o)
+        [:op_rshift_equal, *super ]
+      end
+
+      def visit_OpURShiftEqualNode(o)
+        [:op_urshift_equal, *super ]
+      end
+
+      def visit_OpAndEqualNode(o)
+        [:op_and_equal, *super ]
+      end
+
+      def visit_OpXOrEqualNode(o)
+        [:op_xor_equal, *super ]
+      end
+
+      def visit_OpOrEqualNode(o)
+        [:op_or_equal, *super ]
+      end
+
+      def visit_OpModEqualNode(o)
+        [:op_mod_equal, *super]
+      end
+
+      def visit_CommaNode(o)
+        [:comma, *super]
+      end
+
+      def visit_FunctionCallNode(o)
+        [:function_call, *super]
+      end
+
+      def visit_ArrayNode(o)
+        [:array, super]
+      end
+
+      def visit_ThisNode(o)
+        [:this]
+      end
+
+      def visit_ReturnNode(o)
+        o.value ? [:return, super] : [:return]
+      end
+
+      def visit_FunctionExprNode(o)
+        [ :func_expr, *super]
+      end
+
+      def visit_FunctionDeclNode(o)
+        [ :func_decl, *super]
+      end
+
+      def visit_ArgumentsNode(o)
+        [:args, super]
+      end
+
+      def visit_DotAccessorNode(o)
+        [:dot_access,
+          super,
+          o.accessor
+        ]
+      end
+
+      def visit_NullNode(o)
+        [:nil]
+      end
+      
+      def visit_StringNode(o)
+        [:str, o.value]
+      end
+
+      def visit_FalseNode(o)
+        [:false]
+      end
+
+      def visit_TrueNode(o)
+        [:true]
+      end
+
+    end
+  end
+end