lib/opal/nodes/node_with_args.rb in opal-1.3.2 vs lib/opal/nodes/node_with_args.rb in opal-1.4.0.alpha1

@@ -27,16 +27,10 @@
         push process(arity_check_node)
       end
 
       def compile_block_arg
         if scope.uses_block?
-          scope_name  = scope.identity
-          yielder     = scope.block_name || '$yield'
-
-          add_temp "$iter = #{scope_name}.$$p"
-          add_temp "#{yielder} = $iter || nil"
-
-          line "if ($iter) #{scope_name}.$$p = null;"
+          scope.prepare_block
         end
       end
 
       def parameters_code
         Args::Parameters.new(original_args).to_code