Sha256: 706354e170396a648de00ef387a0ccd43224ab5a848a49cc902d8a9d3871f2c7
Contents?: true
Size: 1.94 KB
Versions: 15
Compression:
Stored size: 1.94 KB
Contents
require 'opal/nodes/base'
module Opal
module Nodes
class BaseYieldNode < Base
def compile_call(children, level)
yielding_scope = find_yielding_scope
yielding_scope.uses_block!
block_name = yielding_scope.block_name || '$yield'
if yields_single_arg?(children)
push expr(children.first)
wrap "Opal.yield1(#{block_name}, ", ')'
else
push expr(s(:arglist, *children))
if uses_splat?(children)
wrap "Opal.yieldX(#{block_name}, ", ')'
else
wrap "Opal.yieldX(#{block_name}, [", '])'
end
end
end
def find_yielding_scope
working = scope
while working
if working.block_name or working.def?
break
end
working = working.parent
end
working
end
def yields_single_arg?(children)
!uses_splat?(children) and children.size == 1
end
def uses_splat?(children)
children.any? { |child| child.type == :splat }
end
end
class YieldNode < BaseYieldNode
handle :yield
def compile
compile_call(children, @level)
end
end
# special opal yield assign, for `a = yield(arg1, arg2)` to assign
# to a temp value to make yield expr into stmt.
#
# level will always be stmt as its the reason for this to exist
#
# s(:yasgn, :a, s(:yield, arg1, arg2))
class YasgnNode < BaseYieldNode
handle :yasgn
children :var_name, :yield_args
def compile
compile_call(s(*yield_args[1..-1]), :stmt)
wrap "(#{var_name} = ", ")"
end
end
# Created by `#returns()` for when a yield statement should return
# it's value (its last in a block etc).
class ReturnableYieldNode < BaseYieldNode
handle :returnable_yield
def compile
compile_call children, @level
wrap "return ", ";"
end
end
end
end
Version data entries
15 entries across 15 versions & 1 rubygems