Module: EnumeratedAttribute::Integrations::ActiveRecord

Defined in:
lib/enumerated_attribute/integrations/active_record.rb

Defined Under Namespace

Modules: ClassMethods

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(klass) ⇒ Object 



5
6
7
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 5

def self.included(klass)
  klass.extend(ClassMethods)
end

Instance Method Details

#[](attr_name) ⇒ Object 



62
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 62

def [](attr_name); read_enumerated_attribute(attr_name); end

#[]=(attr_name, value) ⇒ Object 



63
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 63

def []=(attr_name, value); write_enumerated_attribute(attr_name, value); end

#attributes ⇒ Object 



52
53
54
55
56
57
58
59
60
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 52

def attributes
  atts = super
  atts.each do |k,v|
    if self.class.has_enumerated_attribute?(k)
      atts[k] = v.to_sym if v
    end
  end
  atts
end

#attributes=(attrs, guard_protected_attributes = true) ⇒ Object 



37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 37

def attributes=(attrs, guard_protected_attributes=true)
  return if attrs.nil?
  #check the attributes then turn them over 
  attrs.each do |k, v|
    if self.class.has_enumerated_attribute?(k)
      unless self.class.enumerated_attribute_allows_value?(k, v)
        raise InvalidEnumeration, ":#{v.to_s} is not a defined enumeration value for the '#{k.to_s}' attribute", caller
      end
      attrs[k] = v.to_s
    end
  end
  
  super
end

#read_enumerated_attribute(name) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 24

def read_enumerated_attribute(name)
  name = name.to_s
  #if not enumerated - let active record handle it
  return read_attribute(name) unless self.class.has_enumerated_attribute?(name)
  #if enumerated, is it an active record attribute, if not, the value is stored in an instance variable
  name_sym = name.to_sym
  return instance_variable_get('@'+name) unless self.has_attribute?(name)
  #this is an enumerated active record attribute
  val = read_attribute(name)
  val = val.to_sym if !!val
  val
end

#write_enumerated_attribute(name, val) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/enumerated_attribute/integrations/active_record.rb', line 9

def write_enumerated_attribute(name, val)
  name = name.to_s
  return write_attribute(name, val) unless self.class.has_enumerated_attribute?(name)
  val = nil if val == ''
  val_str = val.to_s if val
  val_sym = val.to_sym if val
  unless self.class.enumerated_attribute_allows_value?(name, val_sym)
    raise(InvalidEnumeration, "nil is not allowed on '#{name}' attribute, set :nil=>true option", caller) unless val
    raise(InvalidEnumeration, ":#{val_str} is not a defined enumeration value for the '#{name}' attribute", caller)
  end
  return instance_variable_set('@'+name, val_sym) unless self.has_attribute?(name)
  write_attribute(name, val_str)
  val_sym
end