Class: DataMapper::Property::Enum

Inherits:
Integer
  • Object
show all
Includes:
Flags
Defined in:
lib/dm-types/enum.rb

Instance Method Summary collapse

Methods included from Flags

#custom?, included

Constructor Details

#initialize(model, name, options = {}) ⇒ Enum

Returns a new instance of Enum.



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/dm-types/enum.rb', line 14

def initialize(model, name, options = {})
  @flag_map = {}

  flags = options.fetch(:flags, self.class.flags)
  flags.each_with_index do |flag, i|
    @flag_map[i.succ] = flag
  end

  if self.class.accepted_options.include?(:set) && !options.include?(:set)
    options[:set] = @flag_map.values_at(*@flag_map.keys.sort)
  end

  options[:max] = @flag_map.size

  super
end

Instance Method Details

#dump(value) ⇒ Object 



35
36
37
38
39
40
 
# File 'lib/dm-types/enum.rb', line 35

def dump(value)
  case value
  when ::Array then value.collect { |v| dump(v) }
  else              flag_map.invert[value]
  end
end

#load(value) ⇒ Object 



31
32
33
 
# File 'lib/dm-types/enum.rb', line 31

def load(value)
  flag_map[value]
end

#typecast(value) ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/dm-types/enum.rb', line 42

def typecast(value)
  return if value.nil?

  # Attempt to typecast using the class of the first item in the map.
  case flag_map[1]
  when ::Symbol then value.to_sym
  when ::String then value.to_s
  when ::Integer then value.to_i
  else value
  end
end