Module: ActiveRecord::Sti::Enum::ClassMethods

Defined in:
lib/activerecord/sti/enum.rb

Instance Method Summary collapse

Instance Method Details

#add_enum(definitions) ⇒ Object



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# File 'lib/activerecord/sti/enum.rb', line 25

def add_enum definitions
  klass = self
  definitions.each do |name, values|
    enum_values = defined_enums[name.to_s] ||= ActiveSupport::HashWithIndifferentAccess.new
    name        = name.to_sym

    klass.singleton_class.send(:define_method, name.to_s.pluralize) { enum_values }

    _enum_methods_module.module_eval do
      pairs = values.respond_to?(:each_pair) ? values.each_pair : values.each_with_index
      pairs.each do |value, i|
        enum_values[value] = i

        define_method("#{value}?") { self[name] == i }
        define_method("#{value}!") { update! name => value }

        klass.scope value, -> { klass.where name => i }
      end
    end
    defined_enums[name.to_s] = enum_values
  end
end