Module: Sequel::Plugins::Enum::ClassMethods

Defined in:
lib/sequel_enum.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#enums ⇒ Object (readonly)

Returns the value of attribute enums.



11
12
13
 
# File 'lib/sequel_enum.rb', line 11

def enums
  @enums
end

Instance Method Details

#enum(column, values) ⇒ Object 



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/sequel_enum.rb', line 13

def enum(column, values)
  if values.is_a? Hash
    values.each do |key,val|
      raise ArgumentError, "index should be numeric, #{key} provided wich it's a #{key.class}" unless key.is_a? Fixnum
      raise ArgumentError "value should be a symbol, #{val} provided wich it's a #{val.class}" unless val.is_a? Symbol
    end
  elsif values.is_a? Array
    values = Hash[values.map.with_index { |v, i| [i,v] }]
  else
    raise ArgumentError, "#enum expects the second argument to be an array of symbols or a hash like { index => :value }"
  end

  define_method "#{column}=" do |value|
    index = self.class.enums[column].rassoc(value)
    self[column] = index && index.first
  end

  define_method "#{column}" do
    self.class.enums[column].fetch(self[column], nil)
  end

  values.each do |key, value|
    define_method "#{value}?" do
      self.send(column) == value
    end
  end

  enums[column] = values
end