Module: StoreField::ClassMethods

Defined in:
lib/store_field.rb

Instance Method Summary collapse

Instance Method Details

#store_field(key, options = {}) ⇒ Object

Raises:

  • (ArgumentError) 


9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/store_field.rb', line 9

def store_field(key, options = {})
  raise ArgumentError.new(':in is invalid')   if options[:in] and serialized_attributes[options[:in].to_s].nil?
  raise ArgumentError.new(':type is invalid') if options[:type] and ![ Hash, Set ].include?(options[:type])

  klass = options[:type]
  store_attribute = options[:in] || serialized_attributes.keys.first
  raise ArgumentError.new('store method must be defined before store_field') if store_attribute.nil?

  # Accessor
  define_method(key) do
    value = send("#{store_attribute}")[key]
    if value.nil?
      value = klass ? klass.new : {}
      send("#{store_attribute}")[key] = value
    end
    value
  end

  # Utility methods for Set
  if options[:type] == Set
    define_method("set_#{key}")     {|value| send(key).add(value); self }
    define_method("unset_#{key}")   {|value| send(key).delete(value); self }
    define_method("set_#{key}?")    {|value| send(key).include?(value) }
    define_method("unset_#{key}?")  {|value| !send(key).include?(value) }
  end
end