Support forwarded arguments in super and nested blocks

Co-authored-by: OpenAI Codex <codex@openai.com>

Author: 2 people

lib/typeprof/core/ast/call.rb

@@ -132,6 +132,7 @@ def install0(genv)
         if @block_body
           block_body = @block_body # kinda type annotationty
           block_tbl = @block_tbl || raise
+          block_body.lenv.forward_args = @lenv.forward_args
           @lenv.locals.each {|var, vtx| block_body.lenv.locals[var] = vtx }
           block_tbl.each {|var| block_body.lenv.locals[var] = Source.new(genv.nil_type) }
           block_body.lenv.locals[:"*self"] = block_body.lenv.cref.get_self(genv)
@@ -314,9 +315,9 @@ def initialize(raw_node,  lenv)
 
     class ForwardingSuperNode < CallBaseNode
       def initialize(raw_node,  lenv)
-        raw_args = nil # TODO: forward args properly
+        raw_args = nil
         raw_block = raw_node.block
-        super(raw_node, nil, :"*super", nil, raw_args, nil, raw_block, lenv)
+        super(raw_node, nil, :"*super", nil, raw_args, nil, raw_block, lenv, forwarding_arguments: true)
       end
     end
 

scenario/args/forwarding_arguments.rb

@@ -36,3 +36,20 @@ class Object
   def foo: (*Integer, **Integer) -> [Array[Integer], { x: Integer, y: Integer }]
   def bar: (*Integer, **Integer) -> [Array[Integer], { x: Integer, y: Integer }]
 end
+
+## update
+def foo(...)
+  1.times { bar(...) }
+end
+
+def bar(*a, **b)
+  [a, b]
+end
+
+foo(1, x: 4, y: 5)
+
+## assert
+class Object
+  def foo: (*Integer, **Integer) -> Integer
+  def bar: (*Integer, **Integer) -> [Array[Integer], { x: Integer, y: Integer }]
+end

scenario/misc/super.rb

@@ -36,3 +36,26 @@ def [](key)
 class StringifyKeyHash < Hash
   def []: (untyped) -> untyped
 end
+
+## update
+class SuperBase
+  def foo(*a, **b)
+    [a, b]
+  end
+end
+
+class SuperChild < SuperBase
+  def foo(...)
+    super(...)
+  end
+end
+
+SuperChild.new.foo(1, x: 4, y: 5)
+
+## assert
+class SuperBase
+  def foo: (*Integer, **Integer) -> [Array[Integer], { x: Integer, y: Integer }]
+end
+class SuperChild < SuperBase
+  def foo: (*Integer, **Integer) -> [Array[Integer], { x: Integer, y: Integer }]
+end