Merge pull request #179 from rhiroe/patch-hiroe-2

Set the generics of the struct based on the type of its attributes

Author: soutaro

lib/rbs/inline/writer.rb

@@ -428,7 +428,12 @@ def translate_struct_assign_decl(decl, rbs) #: void
           members: [*attributes, new],
           super_class: RBS::AST::Declarations::Class::Super.new(
             name: RBS::TypeName.new(name: :Struct, namespace: RBS::Namespace.empty),
-            args: [RBS::Types::Bases::Any.new(location: nil)],
+            args: [
+              RBS::Types::Union.new(
+                types: decl.each_attribute.map { |_, attr| attr&.type || RBS::Types::Bases::Any.new(location: nil) }.uniq,
+                location: nil
+              )
+            ],
             location: nil
           ),
           annotations: decl.class_annotations,

test/rbs/inline/writer_test.rb

@@ -1011,7 +1011,7 @@ class Account
 
     assert_equal <<~RBS, output
       # Account record
-      class Account < Struct[untyped]
+      class Account < Struct[Integer | String]
         attr_accessor id(): Integer
 
         attr_accessor email(): String
@@ -1029,7 +1029,7 @@ def self.new: (?name: untyped) -> instance
         end
       end
 
-      class Item < Struct[untyped]
+      class Item < Struct[String | Integer]
         attr_accessor sku(): String
 
         attr_accessor price(): Integer
@@ -1041,7 +1041,7 @@ def self.new: (?String sku, ?Integer price) -> instance
       # @rbs %a{rbs-inline:readonly-attributes=true}
       %a{rbs-inline:new-args=required}
       %a{rbs-inline:readonly-attributes=true}
-      class User < Struct[untyped]
+      class User < Struct[String]
         attr_reader name(): String
 
         def self.new: (String name) -> instance