foss/mongoid-enum
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactored accessor definition into methods

Browse Source
This commit is contained in:
Nicholas Bruning 2014-03-17 23:44:40 +11:00
parent d4280d3eca
commit fde35639d8
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
lib/mongoid/enum.rb
View File

@ -51,15 +51,21 @@ module Mongoid
values.each do |value|
scope value, where(field_name => value)
if options[:multiple]
class_eval "def #{value}?() self.#{field_name}.include?(:#{value}) end"
class_eval "def #{value}!() update_attributes! :#{field_name} => (self.#{field_name} || []) + [:#{value}] end"
else
class_eval "def #{value}?() self.#{field_name} == :#{value} end"
class_eval "def #{value}!() update_attributes! :#{field_name} => :#{value} end"
end
options[:multiple] &&
define_array_accessor(field_name, value) ||
define_string_accessor(field_name, value)
end
end
def define_array_accessor(field_name, value)
class_eval "def #{value}?() self.#{field_name}.include?(:#{value}) end"
class_eval "def #{value}!() update_attributes! :#{field_name} => (self.#{field_name} || []) + [:#{value}] end"
end
def define_string_accessor(field_name, value)
class_eval "def #{value}?() self.#{field_name} == :#{value} end"
class_eval "def #{value}!() update_attributes! :#{field_name} => :#{value} end"
end
end
end
end