opal-cj 0.7.0.beta1

data/lib/opal/nodes/singleton_class.rb

@@ -0,0 +1,25 @@
+require 'opal/nodes/scope'
+
+module Opal
+  module Nodes
+    class SingletonClassNode < ScopeNode
+      handle :sclass
+
+      children :object, :body
+
+      def compile
+        push "(function(self) {"
+
+        in_scope do
+          add_temp '$scope = self.$$scope'
+          add_temp 'def = self.$$proto'
+
+          line scope.to_vars
+          line stmt(compiler.returns(body))
+        end
+
+        line "})(", recv(object), ".$singleton_class())"
+      end
+    end
+  end
+end

data/lib/opal/nodes/super.rb

@@ -0,0 +1,97 @@
+require 'opal/nodes/base'
+
+module Opal
+  module Nodes
+    # This base class is used just to child the find_super_dispatcher method
+    # body. This is then used by actual super calls, or a defined?(super) style
+    # call.
+    class BaseSuperNode < Base
+      children :arglist, :iter
+
+      def compile_dispatcher
+        if arglist or iter
+          iter = expr(iter_sexp)
+        else
+          scope.uses_block!
+          iter = '$iter'
+        end
+        if scope.def?
+          scope.uses_block!
+          scope_name = scope.identify!
+          class_name = scope.parent.name ? "$#{scope.parent.name}" : 'self.$$class.$$proto'
+
+          if scope.defs
+            push "$opal.find_super_dispatcher(self, '#{scope.mid.to_s}', #{scope_name}, "
+            push iter
+            push ", #{class_name})"
+          else
+            push "$opal.find_super_dispatcher(self, '#{scope.mid.to_s}', #{scope_name}, "
+            push iter
+            push ")"
+          end
+        elsif scope.iter?
+          chain, cur_defn, mid = scope.get_super_chain
+          trys = chain.map { |c| "#{c}.$$def" }.join(' || ')
+
+          push "$opal.find_iter_super_dispatcher(self, #{mid}, (#{trys} || #{cur_defn}), null)"
+        else
+          raise "Cannot call super() from outside a method block"
+        end
+      end
+
+      def args
+        arglist || s(:arglist)
+      end
+
+      def iter_sexp
+        iter || s(:js_tmp, 'null')
+      end
+    end
+
+    class DefinedSuperNode < BaseSuperNode
+      handle :defined_super
+
+      def compile
+        # insert method body to find super method
+        self.compile_dispatcher
+
+        wrap '((', ') != null ? "super" : nil)'
+      end
+    end
+
+    class SuperNode < BaseSuperNode
+      handle :super
+
+      children :arglist, :iter
+
+      def compile
+        if arglist or iter
+          splat = has_splat?
+          args = expr(self.args)
+
+          unless splat
+            args = [fragment('['), args, fragment(']')]
+          end
+        else
+          if scope.def?
+            scope.uses_zuper = true
+            args = fragment('$zuper')
+          else
+            args = fragment('$slice.call(arguments)')
+          end
+        end
+
+        # compile our call to runtime to get super method
+        self.compile_dispatcher
+
+        push ".apply(self, "
+        push(*args)
+        push ")"
+      end
+
+      def has_splat?
+        args.children.any? { |child| child.type == :splat }
+      end
+    end
+  end
+end

data/lib/opal/nodes/top.rb

@@ -0,0 +1,101 @@
+require 'opal/version'
+require 'opal/nodes/scope'
+
+module Opal
+  module Nodes
+    # Generates code for an entire file, i.e. the base sexp
+    class TopNode < ScopeNode
+      handle :top
+
+      children :body
+
+      def compile
+        push version_comment
+
+        opening
+        in_scope do
+          line "$opal.dynamic_require_severity = #{compiler.dynamic_require_severity.to_s.inspect};"
+          body_code = stmt(stmts)
+          body_code = [body_code] unless body_code.is_a?(Array)
+
+          add_temp 'self = $opal.top'
+          add_temp '$scope = $opal'
+          add_temp 'nil = $opal.nil'
+
+          add_used_helpers
+          add_used_operators
+          line scope.to_vars
+
+          compile_method_stubs
+          compile_irb_vars
+          compile_end_construct
+
+          line body_code
+        end
+
+        closing
+      end
+
+      def opening
+        if compiler.requirable?
+          line "Opal.modules[#{compiler.file.inspect}] = function($opal) {"
+        else
+          line "(function($opal) {"
+        end
+      end
+
+      def closing
+        if compiler.requirable?
+          line "};\n"
+        else
+          line "})(Opal);\n"
+        end
+      end
+
+      def stmts
+        compiler.returns(body)
+      end
+
+      def compile_irb_vars
+        if compiler.irb?
+          line "if (!$opal.irb_vars) { $opal.irb_vars = {}; }"
+        end
+      end
+
+      def add_used_helpers
+        helpers = compiler.helpers.to_a
+        helpers.to_a.each { |h| add_temp "$#{h} = $opal.#{h}" }
+      end
+
+      def add_used_operators
+        operators = compiler.operator_helpers.to_a
+        operators.each do |op|
+          name = Nodes::CallNode::OPERATORS[op]
+          line "function $rb_#{name}(lhs, rhs) {"
+          line "  return (typeof(lhs) === 'number' && typeof(rhs) === 'number') ? lhs #{op} rhs : lhs['$#{op}'](rhs);"
+          line "}"
+        end
+      end
+
+      def compile_method_stubs
+        if compiler.method_missing?
+          calls = compiler.method_calls
+          stubs = calls.to_a.map { |k| "'$#{k}'" }.join(', ')
+          line "$opal.add_stubs([#{stubs}]);"
+        end
+      end
+
+      # Any special __END__ content in code
+      def compile_end_construct
+        if content = compiler.eof_content
+          line "var $__END__ = Opal.Object.$new();"
+          line "$__END__.$read = function() { return #{content.inspect}; };"
+        end
+      end
+
+      def version_comment
+        "/* Generated by Opal #{Opal::VERSION} */"
+      end
+    end
+  end
+end

data/lib/opal/nodes/variables.rb

@@ -0,0 +1,158 @@
+require 'opal/nodes/base'
+
+module Opal
+  module Nodes
+    class LocalVariableNode < Base
+      handle :lvar
+
+      children :var_name
+
+      def using_irb?
+        compiler.irb? and scope.top?
+      end
+
+      def compile
+        return push(variable(var_name.to_s)) unless using_irb?
+
+        with_temp do |tmp|
+          push property(var_name.to_s)
+          wrap "((#{tmp} = $opal.irb_vars", ") == null ? nil : #{tmp})"
+        end
+      end
+    end
+
+    class LocalAssignNode < Base
+      handle :lasgn
+
+      children :var_name, :value
+
+      def using_irb?
+        compiler.irb? and scope.top?
+      end
+
+      def compile
+        if using_irb?
+          push "$opal.irb_vars#{property var_name.to_s} = "
+          push expr(value)
+        else
+          add_local variable(var_name.to_s)
+
+          push "#{variable(var_name.to_s)} = "
+          push expr(value)
+        end
+
+        wrap '(', ')' if recv?
+      end
+    end
+
+    class InstanceVariableNode < Base
+      handle :ivar
+
+      children :name
+
+      def var_name
+        name.to_s[1..-1]
+      end
+
+      def compile
+        name = property var_name
+        add_ivar name
+        push "self#{name}"
+      end
+    end
+
+    class InstanceAssignNode < Base
+      handle :iasgn
+
+      children :name, :value
+
+      def var_name
+        name.to_s[1..-1]
+      end
+
+      def compile
+        name = property var_name
+        push "self#{name} = "
+        push expr(value)
+      end
+    end
+
+    class GlobalVariableNode < Base
+      handle :gvar
+
+      children :name
+
+      def var_name
+        name.to_s[1..-1]
+      end
+
+      def compile
+        helper :gvars
+        name = property var_name
+        add_gvar name
+        push "$gvars#{name}"
+      end
+    end
+
+    class GlobalAssignNode < Base
+      handle :gasgn
+
+      children :name, :value
+
+      def var_name
+        name.to_s[1..-1]
+      end
+
+      def compile
+        helper :gvars
+        name = property var_name
+        push "$gvars#{name} = "
+        push expr(value)
+      end
+    end
+
+    class BackrefNode < Base
+      handle :nth_ref
+
+      def compile
+        push "nil"
+      end
+    end
+
+    class ClassVariableNode < Base
+      handle :cvar
+
+      children :name
+
+      def compile
+        with_temp do |tmp|
+          push "((#{tmp} = $opal.cvars['#{name}']) == null ? nil : #{tmp})"
+        end
+      end
+    end
+
+    class ClassVarAssignNode < Base
+      handle :casgn
+
+      children :name, :value
+
+      def compile
+        push "($opal.cvars['#{name}'] = "
+        push expr(value)
+        push ")"
+      end
+    end
+
+    class ClassVarDeclNode < Base
+      handle :cvdecl
+
+      children :name, :value
+
+      def compile
+        push "($opal.cvars['#{name}'] = "
+        push expr(value)
+        push ")"
+      end
+    end
+  end
+end

data/lib/opal/nodes/while.rb

@@ -0,0 +1,65 @@
+require 'opal/nodes/base'
+
+module Opal
+  module Nodes
+    class WhileNode < Base
+      handle :while
+
+      children :test, :body
+
+      def compile
+        with_temp do |redo_var|
+          test_code = js_truthy(test)
+
+          compiler.in_while do
+            while_loop[:closure] = true if wrap_in_closure?
+            while_loop[:redo_var] = redo_var
+
+            body_code = stmt(body)
+
+            if uses_redo?
+              push "#{redo_var} = false; #{while_open}#{redo_var} || "
+              push test_code
+              push while_close
+            else
+              push while_open, test_code, while_close
+            end
+
+            push "#{redo_var} = false;" if uses_redo?
+            line body_code, "}"
+          end
+        end
+
+        wrap '(function() {', '; return nil; })()' if wrap_in_closure?
+      end
+
+      def while_open
+        "while ("
+      end
+
+      def while_close
+        ") {"
+      end
+
+      def uses_redo?
+        while_loop[:use_redo]
+      end
+
+      def wrap_in_closure?
+        expr? or recv?
+      end
+    end
+
+    class UntilNode < WhileNode
+      handle :until
+
+      def while_open
+        "while (!("
+      end
+
+      def while_close
+        ")) {"
+      end
+    end
+  end
+end