active_model: Support ActiveModel::* mix-ins

pocke · Feb 4, 2024 · 2c09a4f · 2c09a4f
1 parent 6b4d6c5
commit 2c09a4f
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/lib/rbs_rails/active_model.rb b/lib/rbs_rails/active_model.rb
@@ -42,7 +42,7 @@ def generate
       private def klass_decl
         <<~RBS
           #{header}
-          def self.attribute: (Symbol name, ?Symbol? cast_type, ?untyped default, **untyped) -> void
+          #{mixins}
 
           #{attributes}
           #{footer}
@@ -69,6 +69,22 @@ def self.attribute: (Symbol name, ?Symbol? cast_type, ?untyped default, **untype
         end.join("\n")
       end
 
+      private def mixins
+        mixins = []
+        if klass < ::ActiveModel::Model
+          mixins << "include ::ActiveModel::Model"
+        end
+        if klass < ::ActiveModel::Attributes
+          mixins << "include ::ActiveModel::Attributes"
+          mixins << "extend ::ActiveModel::Attributes::ClassMethods"
+        end
+        if klass < ::ActiveModel::Validations
+          mixins << "include ::ActiveModel::Validations"
+          mixins << "extend ::ActiveModel::Validations::ClassMethods"
+        end
+        mixins.join("\n")
+      end
+
       private def attributes
         # @type var model: untyped
         model = klass

diff --git a/sig/rbs_rails/active_model.rbs b/sig/rbs_rails/active_model.rbs
@@ -15,6 +15,7 @@ module RbsRails
 
       def klass_decl: () -> String
       def header: () -> String
+      def mixins: () -> String
       def attributes: () -> String
       def footer: () -> String
 

diff --git a/test/expectations/article.rbs b/test/expectations/article.rbs
@@ -1,5 +1,9 @@
 class Article < ::Object
-  def self.attribute: (Symbol name, ?Symbol? cast_type, ?untyped default, **untyped) -> void
+  include ::ActiveModel::Model
+  include ::ActiveModel::Attributes
+  extend ::ActiveModel::Attributes::ClassMethods
+  include ::ActiveModel::Validations
+  extend ::ActiveModel::Validations::ClassMethods
 
   def blog_id: () -> Integer?
   def blog_id=: (Integer? value) -> Integer?