watcher.rb
66 lines
| 2.1 KiB
| text/x-ruby
|
RubyLexer
|
r6390 | # Redmine - project management software | ||
# Copyright (C) 2006-2011 Jean-Philippe Lang | ||||
|
r450 | # | ||
# This program is free software; you can redistribute it and/or | ||||
# modify it under the terms of the GNU General Public License | ||||
# as published by the Free Software Foundation; either version 2 | ||||
# of the License, or (at your option) any later version. | ||||
|
r6390 | # | ||
|
r450 | # This program is distributed in the hope that it will be useful, | ||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
|
r6390 | # | ||
|
r450 | # You should have received a copy of the GNU General Public License | ||
# along with this program; if not, write to the Free Software | ||||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. | ||||
class Watcher < ActiveRecord::Base | ||||
belongs_to :watchable, :polymorphic => true | ||||
belongs_to :user | ||||
|
r6390 | |||
|
r1711 | validates_presence_of :user | ||
|
r450 | validates_uniqueness_of :user_id, :scope => [:watchable_type, :watchable_id] | ||
|
r7952 | validate :validate_user | ||
|
r3053 | |||
# Unwatch things that users are no longer allowed to view | ||||
def self.prune(options={}) | ||||
if options.has_key?(:user) | ||||
prune_single_user(options[:user], options) | ||||
else | ||||
pruned = 0 | ||||
User.find(:all, :conditions => "id IN (SELECT DISTINCT user_id FROM #{table_name})").each do |user| | ||||
pruned += prune_single_user(user, options) | ||||
end | ||||
pruned | ||||
end | ||||
end | ||||
|
r6390 | |||
|
r1711 | protected | ||
|
r6390 | |||
|
r7952 | def validate_user | ||
|
r2430 | errors.add :user_id, :invalid unless user.nil? || user.active? | ||
|
r1711 | end | ||
|
r6390 | |||
|
r3053 | private | ||
|
r6390 | |||
|
r3053 | def self.prune_single_user(user, options={}) | ||
return unless user.is_a?(User) | ||||
pruned = 0 | ||||
find(:all, :conditions => {:user_id => user.id}).each do |watcher| | ||||
next if watcher.watchable.nil? | ||||
|
r6390 | |||
|
r3053 | if options.has_key?(:project) | ||
next unless watcher.watchable.respond_to?(:project) && watcher.watchable.project == options[:project] | ||||
end | ||||
|
r6390 | |||
|
r3053 | if watcher.watchable.respond_to?(:visible?) | ||
unless watcher.watchable.visible?(user) | ||||
watcher.destroy | ||||
pruned += 1 | ||||
end | ||||
end | ||||
end | ||||
pruned | ||||
end | ||||
|
r450 | end | ||