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
|
# File 'lib/drafting/base_class_methods.rb', line 5
def has_drafts(options={})
raise ArgumentError unless options.is_a?(Hash)
raise ArgumentError unless options.keys.all? { |k| ALLOWED_DRAFT_OPTION_KEYS.include?(k) }
class_attribute :draft_parent
if options[:parent]
parent_class = self.reflect_on_all_associations(:belongs_to).find { |a| a.name == options[:parent] }.try(:klass)
raise ArgumentError unless parent_class
unless parent_class.method_defined? :drafts
parent_class.class_eval do
def drafts(user)
Draft.where(
user: user,
user_type: user.try(:class).try(:name),
parent: self
)
end
def self.child_drafts(user)
Draft.where(
user: user,
user_type: user.try(:class).try(:name),
parent_type: self.base_class.name
)
end
end
end
self.draft_parent = options[:parent]
end
include Drafting::InstanceMethods
extend Drafting::ClassMethods
attr_accessor :draft_id
after_create :clear_draft
end
|