watcher.rb
81 lines
| 2.5 KiB
| text/x-ruby
|
RubyLexer
|
r6390 | # Redmine - project management software | ||
|
r13490 | # Copyright (C) 2006-2015 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 | ||
|
r13100 | attr_protected :id | ||
|
r3053 | |||
|
r11729 | # Returns true if at least one object among objects is watched by user | ||
def self.any_watched?(objects, user) | ||||
objects = objects.reject(&:new_record?) | ||||
if objects.any? | ||||
objects.group_by {|object| object.class.base_class}.each do |base_class, objects| | ||||
if Watcher.where(:watchable_type => base_class.name, :watchable_id => objects.map(&:id), :user_id => user.id).exists? | ||||
return true | ||||
end | ||||
end | ||||
end | ||||
false | ||||
end | ||||
|
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 | ||||
|
r12297 | User.where("id IN (SELECT DISTINCT user_id FROM #{table_name})").each do |user| | ||
|
r3053 | 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 | ||||
|
r12294 | where(:user_id => user.id).each do |watcher| | ||
|
r3053 | next if watcher.watchable.nil? | ||
if options.has_key?(:project) | ||||
|
r12293 | unless watcher.watchable.respond_to?(:project) && | ||
watcher.watchable.project == options[:project] | ||||
next | ||||
end | ||||
|
r3053 | end | ||
if watcher.watchable.respond_to?(:visible?) | ||||
unless watcher.watchable.visible?(user) | ||||
watcher.destroy | ||||
pruned += 1 | ||||
end | ||||
end | ||||
end | ||||
pruned | ||||
end | ||||
|
r450 | end | ||