Add support for 'it' local variable in AST and parameter handling

Author: White-Green

lib/typeprof/core/ast.rb

@@ -92,6 +92,8 @@ def self.create_node(raw_node, lenv, use_result = true)
       # variables
       when :local_variable_read_node
         LocalVariableReadNode.new(raw_node, lenv)
+      when :it_local_variable_read_node
+        ItLocalVariableReadNode.new(raw_node, lenv)
       when :local_variable_write_node
         LocalVariableWriteNode.new(raw_node, AST.create_node(raw_node.value, lenv), lenv)
       when :local_variable_operator_write_node

lib/typeprof/core/ast/call.rb

@@ -55,6 +55,8 @@ def initialize(raw_node, recv, mid, mid_code_range, raw_args, last_arg, raw_bloc
                               raw_block.parameters.parameters.requireds.map {|n| n.is_a?(Prism::MultiTargetNode) ? nil : n.name }
                             when Prism::NumberedParametersNode
                               1.upto(raw_block.parameters.maximum).map { |n| :"_#{n}" }
+                            when Prism::ItParametersNode
+                              [:it]
                             when nil
                               []
                             else

lib/typeprof/core/ast/variable.rb

@@ -19,6 +19,22 @@ def retrieve_at(pos)
       end
     end
 
+    class ItLocalVariableReadNode < Node
+      def initialize(raw_node, lenv)
+        super(raw_node, lenv)
+      end
+
+      def attrs = {}
+
+      def install0(genv)
+        @lenv.get_var(:it)
+      end
+
+      def retrieve_at(pos)
+        yield self if code_range.include?(pos)
+      end
+    end
+
     class LocalVariableWriteNode < Node
       def initialize(raw_node, rhs, lenv)
         super(raw_node, lenv)