Class: SorbetRails::ModelPlugins::EnumerableCollections

Inherits:
Base
  • Object
show all
Defined in:
lib/sorbet-rails/model_plugins/enumerable_collections.rb

Constant Summary

Constants inherited from Base

Base::Parameter

Instance Attribute Summary

Attributes inherited from Base

#available_classes, #model_class

Instance Method Summary collapse

Methods inherited from Base

#attribute_has_unconditional_presence_validation?, #initialize

Methods included from SorbetRails::ModelUtils

#add_relation_query_method, #exists_class_method?, #exists_instance_method?, #habtm_class?, #model_assoc_proxy_class_name, #model_assoc_relation_class_name, #model_class, #model_class_name, #model_module_name, #model_query_methods_returning_assoc_relation_module_name, #model_query_methods_returning_relation_module_name, #model_relation_class_name, #model_relation_type_alias, #model_relation_type_class_name

Constructor Details

This class inherits a constructor from SorbetRails::ModelPlugins::Base

Instance Method Details

#generate(root) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/sorbet-rails/model_plugins/enumerable_collections.rb', line 6

def generate(root)
  model_assoc_proxy_class_rbi = root.create_class(self.model_assoc_proxy_class_name)

  # following methods only exists in an association proxy
  ["<<", "append", "push", "concat"].each do |method_name|
    elem = self.model_class_name
    model_assoc_proxy_class_rbi.create_method(
      method_name,
      parameters: [
        Parameter.new("*records", type: "T.any(#{elem}, T::Array[#{elem}])"),
      ],
      return_type: "T.self_type",
    )
  end
end