| @@ -1,65 +1,66 | |||
|
|
1 | 1 | # Redmine - project management software |
|
|
2 | 2 | # Copyright (C) 2006-2011 Jean-Philippe Lang |
|
|
3 | 3 | # |
|
|
4 | 4 | # This program is free software; you can redistribute it and/or |
|
|
5 | 5 | # modify it under the terms of the GNU General Public License |
|
|
6 | 6 | # as published by the Free Software Foundation; either version 2 |
|
|
7 | 7 | # of the License, or (at your option) any later version. |
|
|
8 | 8 | # |
|
|
9 | 9 | # This program is distributed in the hope that it will be useful, |
|
|
10 | 10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
11 | 11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
12 | 12 | # GNU General Public License for more details. |
|
|
13 | 13 | # |
|
|
14 | 14 | # You should have received a copy of the GNU General Public License |
|
|
15 | 15 | # along with this program; if not, write to the Free Software |
|
|
16 | 16 | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
|
|
17 | 17 | |
|
|
18 | 18 | class Watcher < ActiveRecord::Base |
|
|
19 | 19 | belongs_to :watchable, :polymorphic => true |
|
|
20 | 20 | belongs_to :user |
|
|
21 | 21 | |
|
|
22 | 22 | validates_presence_of :user |
|
|
23 | 23 | validates_uniqueness_of :user_id, :scope => [:watchable_type, :watchable_id] |
|
|
24 | validate :validate_user | |
|
|
24 | 25 | |
|
|
25 | 26 | # Unwatch things that users are no longer allowed to view |
|
|
26 | 27 | def self.prune(options={}) |
|
|
27 | 28 | if options.has_key?(:user) |
|
|
28 | 29 | prune_single_user(options[:user], options) |
|
|
29 | 30 | else |
|
|
30 | 31 | pruned = 0 |
|
|
31 | 32 | User.find(:all, :conditions => "id IN (SELECT DISTINCT user_id FROM #{table_name})").each do |user| |
|
|
32 | 33 | pruned += prune_single_user(user, options) |
|
|
33 | 34 | end |
|
|
34 | 35 | pruned |
|
|
35 | 36 | end |
|
|
36 | 37 | end |
|
|
37 | 38 | |
|
|
38 | 39 | protected |
|
|
39 | 40 | |
|
|
40 | def validate | |
|
|
41 | def validate_user | |
|
|
41 | 42 | errors.add :user_id, :invalid unless user.nil? || user.active? |
|
|
42 | 43 | end |
|
|
43 | 44 | |
|
|
44 | 45 | private |
|
|
45 | 46 | |
|
|
46 | 47 | def self.prune_single_user(user, options={}) |
|
|
47 | 48 | return unless user.is_a?(User) |
|
|
48 | 49 | pruned = 0 |
|
|
49 | 50 | find(:all, :conditions => {:user_id => user.id}).each do |watcher| |
|
|
50 | 51 | next if watcher.watchable.nil? |
|
|
51 | 52 | |
|
|
52 | 53 | if options.has_key?(:project) |
|
|
53 | 54 | next unless watcher.watchable.respond_to?(:project) && watcher.watchable.project == options[:project] |
|
|
54 | 55 | end |
|
|
55 | 56 | |
|
|
56 | 57 | if watcher.watchable.respond_to?(:visible?) |
|
|
57 | 58 | unless watcher.watchable.visible?(user) |
|
|
58 | 59 | watcher.destroy |
|
|
59 | 60 | pruned += 1 |
|
|
60 | 61 | end |
|
|
61 | 62 | end |
|
|
62 | 63 | end |
|
|
63 | 64 | pruned |
|
|
64 | 65 | end |
|
|
65 | 66 | end |
General Comments 0
You need to be logged in to leave comments.
Login now