-
Notifications
You must be signed in to change notification settings - Fork 528
/
paranoia.rb
59 lines (48 loc) · 1.29 KB
/
paranoia.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
module Paranoia
def self.included(klazz)
klazz.extend Query
end
module Query
def paranoid? ; true ; end
def only_deleted
unscoped {
where("deleted_at is not null")
}
end
end
def destroy
_run_destroy_callbacks { delete }
end
def delete
update_attribute_or_column(:deleted_at, Time.now) if !deleted? && persisted?
freeze
end
def restore!
update_attribute_or_column :deleted_at, nil
end
def destroyed?
!self.deleted_at.nil?
end
alias :deleted? :destroyed?
private
# Rails 3.1 adds update_column. Rails > 3.2.6 deprecates update_attribute, gone in Rails 4.
def update_attribute_or_column(*args)
respond_to?(:update_column) ? update_column(*args) : update_attribute(*args)
end
end
class ActiveRecord::Base
def self.acts_as_paranoid
alias_method :destroy!, :destroy
alias_method :delete!, :delete
include Paranoia
default_scope :conditions => { :deleted_at => nil }
end
def self.paranoid? ; false ; end
def paranoid? ; self.class.paranoid? ; end
# Override the persisted method to allow for the paranoia gem.
# If a paranoid record is selected, then we only want to check
# if it's a new record, not if it is "destroyed".
def persisted?
paranoid? ? !new_record? : super
end
end