Module: EnumMachine::DriverSimpleClass

Defined in:
lib/enum_machine/driver_simple_class.rb

Class Method Summary collapse

Class Method Details

.call(args) ⇒ Object

include EnumMachine[

state: { enum: %w[choice in_delivery], i18n_scope: 'line_item.state' },
color: { enum: %w[red green yellow] },
type:  { enum: %w[CartType BookmarkType] },

]



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
42
43
44
45
46
47
48
49
50
 
# File 'lib/enum_machine/driver_simple_class.rb', line 10

def self.call(args)
  Module.new do
    define_singleton_method(:included) do |klass|
      args.each do |attr, params|
        enum_values     = params.fetch(:enum)
        i18n_scope      = params.fetch(:i18n_scope, nil)
        value_decorator = params.fetch(:value_decorator, nil)

        if defined?(ActiveRecord) && klass <= ActiveRecord::Base
          klass.enum_machine(attr, enum_values, i18n_scope: i18n_scope)
        else
          enum_const_name = attr.to_s.upcase
          value_class = BuildValueClass.call(enum_values: enum_values, i18n_scope: i18n_scope, value_decorator: value_decorator)
          enum_class = BuildEnumClass.call(enum_values: enum_values, i18n_scope: i18n_scope, value_class: value_class)

          define_methods =
            Module.new do
              define_method(attr) do
                enum_value = super()
                return unless enum_value

                enum_class.value_attribute_mapping.fetch(enum_value)
              end
            end

          enum_decorator =
            Module.new do
              define_singleton_method(:included) do |decorating_class|
                decorating_class.prepend define_methods
                decorating_class.const_set enum_const_name, enum_class
              end
            end
          enum_class.define_singleton_method(:enum_decorator) { enum_decorator }

          klass.include(enum_decorator)
          enum_decorator
        end
      end
    end
  end
end