Support anonymous block forwarding

Author: mame

lib/typeprof/core/ast/call.rb

@@ -18,6 +18,7 @@ def initialize(raw_node, recv, mid, mid_code_range, raw_args, last_arg, raw_bloc
         @block_f_args = nil
         @block_body = nil
         @safe_navigation = raw_node.respond_to?(:safe_navigation?) && raw_node.safe_navigation?
+        @anonymous_block_forwarding = false
 
         if raw_args
           args = []
@@ -45,8 +46,11 @@ def initialize(raw_node, recv, mid, mid_code_range, raw_args, last_arg, raw_bloc
 
         if raw_block
           if raw_block.type == :block_argument_node
-            # TODO: Support anonymous block forwarding
-            @block_pass = AST.create_node(raw_block.expression, lenv) if raw_block.expression
+            if raw_block.expression
+              @block_pass = AST.create_node(raw_block.expression, lenv)
+            else
+              @anonymous_block_forwarding = true
+            end
           else
             @block_pass = nil
             @block_tbl = raw_block.locals
@@ -74,11 +78,11 @@ def initialize(raw_node, recv, mid, mid_code_range, raw_args, last_arg, raw_bloc
 
       attr_reader :recv, :mid, :mid_code_range, :yield
       attr_reader :positional_args, :splat_flags, :keyword_args
-      attr_reader :block_tbl, :block_f_args, :block_body, :block_pass
+      attr_reader :block_tbl, :block_f_args, :block_body, :block_pass, :anonymous_block_forwarding
       attr_reader :safe_navigation
 
       def subnodes = { recv:, positional_args:, keyword_args:, block_body:, block_pass: }
-      def attrs = { mid:, splat_flags:, block_tbl:, block_f_args:, yield:, safe_navigation: }
+      def attrs = { mid:, splat_flags:, block_tbl:, block_f_args:, yield:, safe_navigation:, anonymous_block_forwarding: }
 
       def install0(genv)
         recv = @recv ? @recv.install(genv) : @yield ? @lenv.get_var(:"*given_block") : @lenv.get_var(:"*self")
@@ -133,6 +137,8 @@ def install0(genv)
           blk_ty = Source.new(Type::Proc.new(genv, block))
         elsif @block_pass
           blk_ty = @block_pass.install(genv)
+        elsif @anonymous_block_forwarding
+          blk_ty = @lenv.get_var(:"*anonymous_block")
         end
 
         a_args = ActualArguments.new(positional_args, @splat_flags, keyword_args, blk_ty)

lib/typeprof/core/ast/method.rb

@@ -95,7 +95,7 @@ def self.parse_params(tbl, raw_args, lenv)
         raise "unexpected keyword rest: #{ raw_args.keyword_rest.class }"
       end
 
-      block = raw_args.block.name if raw_args.block
+      block = raw_args.block.name || :"*anonymous_block" if raw_args.block
 
       {
         req_positionals:,

scenario/block/anonymous_block_forwarding.rb

@@ -13,6 +13,6 @@ def bar(&b)
 
 ## assert
 class Object
-  def foo: -> untyped
-  def bar: { (Float) -> untyped } -> untyped
+  def foo: { (Float) -> String } -> String
+  def bar: { (Float) -> String } -> String
 end