1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980 |
- # Redmine - project management software
- # Copyright (C) 2006-2017 Jean-Philippe Lang
- #
- # 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.
- #
- # 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.
- #
- # 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
-
- validates_presence_of :user
- validates_uniqueness_of :user_id, :scope => [:watchable_type, :watchable_id]
- validate :validate_user
-
- # 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
-
- # 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.where("id IN (SELECT DISTINCT user_id FROM #{table_name})").each do |user|
- pruned += prune_single_user(user, options)
- end
- pruned
- end
- end
-
- protected
-
- def validate_user
- errors.add :user_id, :invalid unless user.nil? || user.active?
- end
-
- private
-
- def self.prune_single_user(user, options={})
- return unless user.is_a?(User)
- pruned = 0
- where(:user_id => user.id).each do |watcher|
- next if watcher.watchable.nil?
- if options.has_key?(:project)
- unless watcher.watchable.respond_to?(:project) &&
- watcher.watchable.project == options[:project]
- next
- end
- end
- if watcher.watchable.respond_to?(:visible?)
- unless watcher.watchable.visible?(user)
- watcher.destroy
- pruned += 1
- end
- end
- end
- pruned
- end
- end
|