Class: FlagShihTzuPlugin

Inherits:
SorbetRails::ModelPlugins::Base show all
Defined in:
lib/sorbet-rails/gem_plugins/flag_shih_tzu_plugin.rb

Overview

typed: false

Constant Summary

Constants inherited from SorbetRails::ModelPlugins::Base

SorbetRails::ModelPlugins::Base::Parameter

Instance Attribute Summary

Attributes inherited from SorbetRails::ModelPlugins::Base

#available_classes, #model_class

Instance Method Summary collapse

Methods inherited from SorbetRails::ModelPlugins::Base

#initialize, #serialization_coder_for_column

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_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

Methods included from SorbetRails::ModelColumnUtils

#active_record_type_to_sorbet_type, #attribute_has_unconditional_presence_validation?, #model_class, #nilable_column?, #time_zone_aware_column?, #type_for_column_def

Constructor Details

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

Instance Method Details

#generate(root) ⇒ Object



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# File 'lib/sorbet-rails/gem_plugins/flag_shih_tzu_plugin.rb', line 4

def generate(root)
  return unless @model_class.include?(::FlagShihTzu)

  obj_custom_module_name = self.model_module_name("GeneratedFlagShihTzuMethods")
  obj_custom_module_rbi  = root.create_module(obj_custom_module_name)

  klass_custom_module_name = self.model_module_name("GeneratedFlagShihTzuClassMethods")
  klass_custom_module_rbi  = root.create_module(klass_custom_module_name)


  # here we re-create the model class!
  model_class_rbi = root.create_class(self.model_class_name)
  model_class_rbi.create_include(obj_custom_module_name)
  model_class_rbi.create_extend(klass_custom_module_name)

  # then create custom methods, constants, etc. for this module.
  add_class_methods(klass_custom_module_rbi)

  if @model_class.flag_columns.present?
    @model_class
      .flag_columns
      .each do |column|
        add_methods_for_column(column, obj_custom_module_rbi)
      end
  end

  if @model_class.flag_mapping.present?
    @model_class
      .flag_mapping
      .each do |column, flags|
        flags.keys
          .select { |flag_key| really_has_the_flag?(flag_key) }
          .each do |flag_key|
            add_methods_for_flag(column, flag_key, obj_custom_module_rbi)
          end
      end
  end
end