Fix RBS translation of non-block type in block position

Signed-off-by: Alexandre Terrasa <alexandre.terrasa@shopify.com>
Shopify · Jan 14, 2025 · 64312e5 · 64312e5
1 parent e9c6406
commit 64312e5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/lib/rbi/rbs_printer.rb b/lib/rbi/rbs_printer.rb
@@ -388,9 +388,7 @@ def print_method_sig(node, sig)
           type_string = "(?) -> untyped"
           block_is_nilable = true
         when Type::Simple
-          if block_type.name == "Proc"
-            type_string = "(?) -> untyped"
-          end
+          type_string = "(?) -> untyped"
           skip = true if block_type.name == "NilClass"
         end
 

diff --git a/test/rbi/rbs_printer_test.rb b/test/rbi/rbs_printer_test.rb
@@ -530,6 +530,21 @@ def foo: -> instance
       RBI
     end
 
+    def test_print_block_type_alias
+      rbi = parse_rbi(<<~RBI)
+        BLOCK = T.type_alias { T.proc.void }
+
+        sig { params(block: BLOCK) }
+        def foo(&block); end
+      RBI
+
+      assert_equal(<<~RBI, rbi.rbs_string)
+        BLOCK: untyped
+
+        def foo: { (?) -> untyped } -> void
+      RBI
+    end
+
     def test_print_t_structs
       rbi = parse_rbi(<<~RBI)
         class Foo < T::Struct; end