blob: 0473d6aafc5ea0662548115dfbc863fe23fde222 (
plain) (
blame)
1
2
3
4
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
module ActiveRecord
# = Active Record Dynamic Finder Match
#
# Refer to ActiveRecord::Base documentation for Dynamic attribute-based finders for detailed info
#
class DynamicFinderMatch
def self.match(method)
method = method.to_s
klass = klasses.find do |_klass|
_klass.matches?(method)
end
klass.new(method) if klass
end
def self.matches?(method)
method =~ self::METHOD_PATTERN
end
def self.klasses
[FindBy, FindByBang, FindOrInitializeCreateBy, FindOrCreateByBang]
end
def initialize(method)
@finder = :first
@instantiator = nil
match_data = method.match(self.class::METHOD_PATTERN)
@attribute_names = match_data[-1].split("_and_")
initialize_from_match_data(match_data)
end
attr_reader :finder, :attribute_names, :instantiator
def finder?
@finder && !@instantiator
end
def creator?
@finder == :first && @instantiator == :create
end
def instantiator?
@instantiator
end
def bang?
false
end
def valid_arguments?(arguments)
arguments.size >= @attribute_names.size
end
def save_record?
@instantiator == :create
end
def save_method
bang? ? :save! : :save
end
private
def initialize_from_match_data(match_data)
end
end
class FindBy < DynamicFinderMatch
METHOD_PATTERN = /^find_(all_|last_)?by_([_a-zA-Z]\w*)$/
def initialize_from_match_data(match_data)
@finder = :last if match_data[1] == 'last_'
@finder = :all if match_data[1] == 'all_'
end
end
class FindByBang < DynamicFinderMatch
METHOD_PATTERN = /^find_by_([_a-zA-Z]\w*)\!$/
def bang?
true
end
end
class FindOrInitializeCreateBy < DynamicFinderMatch
METHOD_PATTERN = /^find_or_(initialize|create)_by_([_a-zA-Z]\w*)$/
def initialize_from_match_data(match_data)
@instantiator = match_data[1] == 'initialize' ? :new : :create
end
def valid_arguments?(arguments)
arguments.size == 1 && arguments.first.is_a?(Hash) || super
end
end
class FindOrCreateByBang < DynamicFinderMatch
METHOD_PATTERN = /^find_or_create_by_([_a-zA-Z]\w*)\!$/
def initialize_from_match_data(match_data)
@instantiator = :create
end
def bang?
true
end
end
end
|