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.



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

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 



32
33
34
35
36
37
 
# File 'lib/dm-types/enum.rb', line 32

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

#load(value) ⇒ Object 



28
29
30
 
# File 'lib/dm-types/enum.rb', line 28

def load(value)
  flag_map[value]
end

#typecast_to_primitive(value) ⇒ Object 



39
40
41
42
43
44
45
46
47
 
# File 'lib/dm-types/enum.rb', line 39

def typecast_to_primitive(value)
  # 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 ::Fixnum then value.to_i
  else               value
  end
end