setblockparam

Author: kddnewton

lib/syntax_tree/compiler.rb

@@ -415,8 +415,13 @@ def visit_assign(node)
         when GVar
           iseq.setglobal(node.target.value.value.to_sym)
         when Ident
-          local_variable = visit(node.target)
-          iseq.setlocal(local_variable.index, local_variable.level)
+          lookup = visit(node.target)
+
+          if lookup.local.is_a?(YARV::LocalTable::BlockLocal)
+            iseq.setblockparam(lookup.index, lookup.level)
+          else
+            iseq.setlocal(lookup.index, lookup.level)
+          end
         when IVar
           iseq.setinstancevariable(node.target.value.value.to_sym)
         end

lib/syntax_tree/yarv.rb

@@ -252,13 +252,13 @@ def to_a
               insn
             when Array
               case insn[0]
-              when :setlocal_WC_0, :setlocal_WC_1, :setlocal
+              when :setlocal_WC_0, :setlocal_WC_1, :setlocal, :setblockparam
                 iseq = self
 
                 case insn[0]
                 when :setlocal_WC_1
                   iseq = iseq.parent_iseq
-                when :setlocal
+                when :setlocal, :setblockparam
                   insn[2].times { iseq = iseq.parent_iseq }
                 end
 
@@ -704,6 +704,11 @@ def send(method_id, argc, flag = VM_CALL_ARGS_SIMPLE, block_iseq = nil)
         end
       end
 
+      def setblockparam(index, level)
+        stack.change_by(-1)
+        push([:setblockparam, index, level])
+      end
+
       def setclassvariable(name)
         stack.change_by(-1)
 

test/compiler_test.rb

@@ -361,6 +361,7 @@ class CompilerTest < Minitest::Test
       "def foo(bar, *baz, &qux); end",
       "def foo(&qux); qux; end",
       "def foo(&qux); qux.call; end",
+      "def foo(&qux); qux = bar; end",
       "def foo(bar:); end",
       "def foo(bar:, baz:); end",
       "def foo(bar: 1); end",