Support global variable target in for loop

natalie-lang · Jun 23, 2024 · ae5ea19 · ae5ea19
1 parent f323e7f
commit ae5ea19
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/natalie/compiler/args.rb b/lib/natalie/compiler/args.rb
@@ -50,6 +50,8 @@ def transform_arg(arg)
           transform_instance_variable_arg(arg)
         when ::Prism::ClassVariableTargetNode
           transform_class_variable_arg(arg)
+        when ::Prism::GlobalVariableTargetNode
+          transform_global_variable_arg(arg)
         when ::Prism::ConstantTargetNode
           transform_constant_arg(arg)
         when ::Prism::RequiredParameterNode
@@ -174,6 +176,11 @@ def transform_class_variable_arg(arg)
         @instructions << ClassVariableSetInstruction.new(arg.name)
       end
 
+      def transform_global_variable_arg(arg)
+        @instructions << ArrayShiftInstruction.new
+        @instructions << GlobalVariableSetInstruction.new(arg.name)
+      end
+
       def transform_constant_arg(arg)
         @instructions << ArrayShiftInstruction.new
         @instructions << PushSelfInstruction.new

diff --git a/lib/natalie/compiler/pass1.rb b/lib/natalie/compiler/pass1.rb
@@ -2671,6 +2671,7 @@ def maximum_arg_count(args)
           %i[
             class_variable_target_node
             constant_target_node
+            global_variable_target_node
             instance_variable_target_node
             local_variable_target_node
             multi_target_node