Class: DataMapper::Property::Flag

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

Instance Method Summary collapse

Methods included from Flags

#custom?, included

Constructor Details

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

Returns a new instance of Flag.



12
13
14
15
16
17
18
19
20
21
# File 'lib/dm-types/flag.rb', line 12

def initialize(model, name, options = {})
  super

  @flag_map = {}

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

Instance Method Details

#dump(value) ⇒ Object



39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# File 'lib/dm-types/flag.rb', line 39

def dump(value)
  unless value.nil?
    flags = Array(value).map { |flag| flag.to_sym }
    flags.uniq!

    flag = 0

    flag_map.invert.values_at(*flags).each do |i|
      next if i.nil?
      flag += (1 << i)
    end

    flag
  end
end

#load(value) ⇒ Object



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# File 'lib/dm-types/flag.rb', line 23

def load(value)
  return [] if value.nil? || value <= 0

  begin
    matches = []

    0.upto(flag_map.size - 1) do |i|
      matches << flag_map[i] if value[i] == 1
    end

    matches.compact
  rescue TypeError, Errno::EDOM
    []
  end
end

#typecast(value) ⇒ Object



55
56
57
58
59
60
61
# File 'lib/dm-types/flag.rb', line 55

def typecast(value)
  case value
    when nil     then nil
    when ::Array then value.map { |v| v.to_sym }
    else [value.to_sym]
  end
end