##// END OF EJS Templates
remove unneeded Relation#all from Watcher#prune_single_user...
Toshi MARUYAMA -
r12294:6dbb9c3677c3
parent child
Show More
@@ -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).all.each do |watcher|
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