-
Notifications
You must be signed in to change notification settings - Fork 21.6k
/
has_one.rb
61 lines (50 loc) · 1.86 KB
/
has_one.rb
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
# frozen_string_literal: true
module ActiveRecord::Associations::Builder # :nodoc:
class HasOne < SingularAssociation # :nodoc:
def self.macro
:has_one
end
def self.valid_options(options)
valid = super + [:as, :through]
valid += [:foreign_type] if options[:as]
valid += [:ensuring_owner_was] if options[:dependent] == :destroy_async
valid += [:source, :source_type, :disable_joins] if options[:through]
valid
end
def self.valid_dependent_options
[:destroy, :destroy_async, :delete, :nullify, :restrict_with_error, :restrict_with_exception]
end
def self.define_callbacks(model, reflection)
super
add_touch_callbacks(model, reflection) if reflection.options[:touch]
end
def self.add_destroy_callbacks(model, reflection)
super unless reflection.options[:through]
end
def self.define_validations(model, reflection)
super
if reflection.options[:required]
model.validates_presence_of reflection.name, message: :required
end
end
def self.touch_record(record, name, touch)
instance = record.send(name)
if instance&.persisted?
touch != true ?
instance.touch(touch) : instance.touch
end
end
def self.add_touch_callbacks(model, reflection)
name = reflection.name
touch = reflection.options[:touch]
callback = -> (record) { HasOne.touch_record(record, name, touch) }
model.after_create callback, if: :saved_changes?
model.after_create_commit { association(name).reset_negative_cache }
model.after_update callback, if: :saved_changes?
model.after_destroy callback
model.after_touch callback
end
private_class_method :macro, :valid_options, :valid_dependent_options, :add_destroy_callbacks,
:define_callbacks, :define_validations, :add_touch_callbacks
end
end