Generate less code inside def/defs to gather block from method

opal · Aug 2, 2013 · c018aed · c018aed
1 parent 775d2c6
commit c018aed
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/lib/opal/parser.rb b/lib/opal/parser.rb
@@ -1203,13 +1203,12 @@ def js_def(recvr, mid, args, stmts, line, end_line, sexp)
 
           if @scope.uses_block?
             @scope.add_temp "$iter = #{scope_name}._p"
-            @scope.add_temp yielder
-            blk = fragment(("\n%s%s = %s._p || nil, %s._p = null;\n%s" %
-                            [@indent, yielder, scope_name, scope_name, @indent]), sexp)
+            @scope.add_temp "#{yielder} = $iter || nil"
+
+            code.unshift f("#{scope_name}._p = null;", sexp)
           end
 
           code.push(*stmt_code)
-          code.unshift blk if blk
 
           uses_super = @scope.uses_super