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