Module: EnumMachine::DriverActiveRecord

Defined in:
lib/enum_machine/driver_active_record.rb

Instance Method Summary collapse

Instance Method Details

#enum_machine(attr, enum_values, i18n_scope: nil, value_decorator: nil, &block) ⇒ Object



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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# File 'lib/enum_machine/driver_active_record.rb', line 5

def enum_machine(attr, enum_values, i18n_scope: nil, value_decorator: nil, &block)
  klass = self

  i18n_scope ||= "#{klass.base_class.to_s.underscore}.#{attr}"

  enum_const_name = attr.to_s.upcase
  machine = Machine.new(enum_values, klass, enum_const_name, attr)
  machine.instance_eval(&block) if block

  value_class = BuildValueClass.call(enum_values: enum_values, i18n_scope: i18n_scope, machine: machine, value_decorator: value_decorator)
  enum_class = BuildEnumClass.call(enum_values: enum_values, i18n_scope: i18n_scope, machine: machine, value_class: value_class)

  value_class.extend(AttributePersistenceMethods[attr, enum_values])

  # default_proc for working with custom values not defined in enum list but may exists in db
  enum_class.value_attribute_mapping.default_proc =
    proc do |hash, enum_value|
      hash[enum_value] = value_class.new(enum_value).freeze
    end

  if machine.transitions?
    klass.class_eval "      before_save :__enum_machine_\#{attr}_before_save\n      after_save :__enum_machine_\#{attr}_after_save\n\n      def __enum_machine_\#{attr}_before_save\n        if (attr_changes = changes['\#{attr}']) && !@__enum_machine_\#{attr}_skip_transitions\n          value_was, value_new = *attr_changes\n          self.class::\#{enum_const_name}.machine.fetch_before_transitions(attr_changes).each do |block|\n            @__enum_machine_\#{attr}_forced_value = value_was\n            instance_exec(self, value_was, value_new, &block)\n          ensure\n            @__enum_machine_\#{attr}_forced_value = nil\n          end\n        end\n      end\n\n      def __enum_machine_\#{attr}_after_save\n        if (attr_changes = previous_changes['\#{attr}']) && !@__enum_machine_\#{attr}_skip_transitions\n          self.class::\#{enum_const_name}.machine.fetch_after_transitions(attr_changes).each { |block| instance_exec(self, *attr_changes, &block) }\n        end\n      end\n    RUBY\n  end\n\n  define_methods = Module.new\n  define_methods.class_eval <<-RUBY, __FILE__, __LINE__ + 1\n    # def state\n    #   enum_value = @__enum_machine_state_forced_value || super()\n    #   return unless enum_value\n    #\n    #   unless @__enum_value_state == enum_value\n    #     @__enum_value_state = self.class::STATE.value_attribute_mapping[enum_value].dup\n    #     @__enum_value_state.parent = self\n    #     @__enum_value_state.freeze\n    #   end\n    #\n    #   @__enum_value_state\n    # end\n    #\n    # def skip_state_transitions\n    #   @__enum_machine_state_skip_transitions = true\n    #   yield\n    # ensure\n    #   @__enum_machine_state_skip_transitions = false\n    # end\n    #\n    # def initialize_dup(other)\n    #   @__enum_value_state = nil\n    #   super\n    # end\n\n    def \#{attr}\n      enum_value = @__enum_machine_\#{attr}_forced_value || super()\n      return unless enum_value\n\n      unless @__enum_value_\#{attr} == enum_value\n        @__enum_value_\#{attr} = self.class::\#{enum_const_name}.value_attribute_mapping[enum_value].dup\n        @__enum_value_\#{attr}.parent = self\n        @__enum_value_\#{attr}.freeze\n      end\n\n      @__enum_value_\#{attr}\n    end\n\n    def skip_\#{attr}_transitions\n      @__enum_machine_\#{attr}_skip_transitions = true\n      yield\n    ensure\n      @__enum_machine_\#{attr}_skip_transitions = false\n    end\n\n    def initialize_dup(other)\n      @__enum_value_\#{attr} = nil\n      super\n    end\n  RUBY\n\n  enum_decorator =\n    Module.new do\n      define_singleton_method(:included) do |decorating_class|\n        decorating_class.prepend define_methods\n        decorating_class.const_set enum_const_name, enum_class\n      end\n    end\n  enum_class.define_singleton_method(:enum_decorator) { enum_decorator }\n\n  klass.include(enum_decorator)\n\n  enum_decorator\nend\n", __FILE__, __LINE__ + 1 # rubocop:disable Style/DocumentDynamicEvalDefinition