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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
# File 'lib/lookup_by/association.rb', line 17
def lookup_for field, options = {}
begin
return unless table_exists?
rescue => error
Rails.logger.error "lookup_by caught #{error.class.name} when connecting - skipping initialization (#{error.inspect})"
return
end
options.symbolize_keys!
options.assert_valid_keys(:class_name, :foreign_key, :symbolize, :strict, :scope, :inverse_scope, :belongs_to)
field = field.to_sym
%W(#{field} raw_#{field} #{field}= #{field}_before_type_cast #{field}?).map(&:to_sym).each do |method|
raise Error, "method `#{method}` already exists on #{self.inspect}" if instance_methods.include? method
end
singleton_class.class_eval do
attr_reader :lookups
end
@lookups ||= []
@lookups << field
scope_name =
if options[:scope] == false
nil
elsif !options.key?(:scope) || options[:scope] == true
"with_#{field}"
else
options[:scope].to_s
end
inverse_scope_name =
if options[:inverse_scope] == false
nil
elsif !options.key?(:inverse_scope) || options[:inverse_scope] == true
"without_#{field}"
else
options[:inverse_scope].to_s
end
if scope_name && respond_to?(scope_name)
raise Error, "#{scope_name} already exists on #{self}."
end
if inverse_scope_name && respond_to?(inverse_scope_name)
raise Error, "#{inverse_scope_name} already exists on #{self}."
end
class_name = options[:class_name] || field
class_name = class_name.to_s.camelize
begin
klass = class_name.constantize
rescue NameError
raise Error, "uninitialized constant #{class_name}, call lookup_for with `class_name` option if it doesn't match the foreign key"
end
raise Error, "class #{class_name} does not use lookup_by" unless klass.respond_to?(:lookup)
foreign_key = options[:foreign_key] || "#{field}_id"
foreign_key = foreign_key.to_sym
Rails.logger.error "foreign key `#{foreign_key}` is required on #{self}" unless attribute_names.include?(foreign_key.to_s)
belongs = options[:belongs_to] || false
class_eval " belongs_to :\#{field}, class_name: \"\#{class_name}\", foreign_key: :\#{foreign_key}, autosave: false, optional: true\n BELONGS_TO\n\n class_eval <<-SCOPES, __FILE__, __LINE__.next if scope_name\n scope :\#{scope_name}, ->(*names) { where(\#{foreign_key}: \#{class_name}[*names]) }\n SCOPES\n\n class_eval <<-SCOPES, __FILE__, __LINE__.next if inverse_scope_name\n scope :\#{inverse_scope_name}, ->(*names) {\n if names.length != 1\n where('\#{foreign_key} NOT IN (?)', \#{class_name}[*names])\n else\n where('\#{foreign_key} <> ?', \#{class_name}[*names])\n end\n }\n SCOPES\n\n cast = options[:symbolize] ? \".to_sym\" : \"\"\n\n lookup_field = klass.lookup.field\n lookup_object = \"\#{class_name}[\#{foreign_key}]\"\n\n strict = options[:strict]\n strict = true if strict.nil?\n\n class_eval <<-METHODS, __FILE__, __LINE__.next\n def raw_\#{field}\n \#{lookup_object}\n end\n\n def \#{field}\n value = \#{lookup_object}\n value ? value.\#{lookup_field}\#{cast} : nil\n end\n\n def \#{field}?(name)\n raise ArgumentError, \"Invalid \#{field} \\\#{name.inspect}\" unless object = \#{class_name}[name]\n \#{foreign_key} == object.id\n end\n\n def \#{field}_before_type_cast\n \#{lookup_object}.\#{lookup_field}_before_type_cast\n end\n\n def \#{field}=(arg)\n result = case arg\n when nil\n nil\n when String, Integer, IPAddr\n \#{class_name}[arg]\n when Symbol\n \#{%Q(raise ArgumentError, \"\#{foreign_key}=(Symbol): use `lookup_for :column, symbolize: true` to allow symbols\") unless options[:symbolize]}\n \#{class_name}[arg]\n when \#{class_name}\n raise ArgumentError, \"self.\#{foreign_key}=(\#{class_name}): must be saved\" unless arg.persisted?\n arg\n else\n raise TypeError, \"\#{foreign_key}=(arg): arg must be a String, Symbol, Integer, IPAddr, nil, or \#{class_name}\"\n end\n\n \#{ %Q(raise LookupBy::Error, \"\\\#{arg.inspect} is not in the <\#{class_name}> lookup cache\" if arg.present? && result.nil?) if strict }\n\n if result.blank?\n self.\#{foreign_key} = nil\n elsif result.persisted?\n self.\#{foreign_key} = result.id\n elsif lookup_errors = result.errors[:\#{lookup_field}]\n lookup_errors.each do |msg|\n errors.add :\#{field}, msg\n end\n end\n end\n METHODS\nend\n", __FILE__, __LINE__.next if belongs
|