Add support for lambda and block locals

Co-authored-by: Kevin Newton <[email protected]>

Author: vinistock

lib/syntax_tree/with_scope.rb

@@ -189,6 +189,15 @@ def visit_blockarg(node)
       super
     end
 
+    def visit_block_var(node)
+      node.locals.each do |local|
+        current_scope.add_local_definition(local, :variable)
+      end
+
+      super
+    end
+    alias visit_lambda_var visit_block_var
+
     # Visit for keeping track of local variable definitions
     def visit_var_field(node)
       value = node.value

test/with_scope_test.rb

@@ -356,6 +356,27 @@ def test_double_nested_arguments
       assert_argument(collector, "four", definitions: [1], usages: [5])
     end
 
+    def test_block_locals
+      collector = Collector.collect(<<~RUBY)
+        [].each do |; a|
+        end
+      RUBY
+
+      assert_equal(1, collector.variables.length)
+
+      assert_variable(collector, "a", definitions: [1])
+    end
+
+    def test_lambda_locals
+      collector = Collector.collect(<<~RUBY)
+        ->(;a) { }
+      RUBY
+
+      assert_equal(1, collector.variables.length)
+
+      assert_variable(collector, "a", definitions: [1])
+    end
+
     def test_regex_named_capture_groups
       collector = Collector.collect(<<~RUBY)
         if /(?<one>\\w+)-(?<two>\\w+)/ =~ "something-else"