5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
# File 'lib/mongoid_enumerable.rb', line 5
def self.included(base)
base.class.redefine_method :enumerable do |field_name, values, options = {}|
field_name = String(field_name)
values.collect!(&:to_s)
prefix = options.fetch(:prefix, "")
default = options.fetch(:default, values.first)
field(field_name, type: String, default: default)
values.each do |value|
method_name = "#{prefix}#{value}"
define_method("#{method_name}!") do
update!(field_name => value)
end
define_method("#{method_name}?") do
send(field_name) == value
end
base.class.redefine_method(method_name) do
base.where(field_name => value)
end
end
end
end
|