forked from mongomapper/mongomapper
-
Notifications
You must be signed in to change notification settings - Fork 0
/
associations.rb
90 lines (75 loc) · 2.43 KB
/
associations.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
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
# encoding: UTF-8
module MongoMapper
module Plugins
module Associations
extend ActiveSupport::Concern
module ClassMethods
def inherited(subclass)
subclass.associations = associations.dup
super
end
def belongs_to(association_id, options={}, &extension)
create_association(BelongsToAssociation.new(association_id, options, &extension))
end
def many(association_id, options={}, &extension)
create_association(ManyAssociation.new(association_id, options, &extension))
end
def one(association_id, options={}, &extension)
create_association(OneAssociation.new(association_id, options, &extension))
end
def associations
@associations ||= {}
end
def associations=(hash)
@associations = hash
end
def associations_module_defined?
if method(:const_defined?).arity == 1 # Ruby 1.9 compat check
const_defined?('MongoMapperAssociations')
else
const_defined?('MongoMapperAssociations', false)
end
end
def associations_module
if associations_module_defined?
const_get 'MongoMapperAssociations'
else
Module.new.tap do |m|
const_set 'MongoMapperAssociations', m
include m
end
end
end
private
def create_association(association)
associations[association.name] = association
association.setup(self)
end
end
def associations
self.class.associations
end
def embedded_associations
associations.values.select { |assoc| assoc.embeddable? }
end
def build_proxy(association)
proxy = association.proxy_class.new(self, association)
self.instance_variable_set(association.ivar, proxy)
proxy
end
def get_proxy(association)
unless proxy = self.instance_variable_get(association.ivar)
proxy = build_proxy(association)
end
proxy
end
def save_to_collection(options={})
super if defined?(super)
associations.each do |association_name, association|
proxy = get_proxy(association)
proxy.save_to_collection(options) if proxy.proxy_respond_to?(:save_to_collection) && association.autosave?
end
end
end
end
end