bigdir/lib/redmine/activity/fetcher.rb

# Redmine - project management software
# Copyright (C) 2006-2013  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.

module Redmine
  module Activity
    # Class used to retrieve activity events
    class Fetcher
      attr_reader :user, :project, :scope,:course

      # Needs to be unloaded in development mode
      @@constantized_providers = Hash.new {|h,k| h[k] = Redmine::Activity.providers[k].collect {|t| t.constantize } }

      def initialize(user, options={})
        if options.has_key?(:project)
          options.assert_valid_keys(:project, :with_subprojects, :author)
          @user = user
          @project = options[:project]
        else
          options.assert_valid_keys(:course, :with_subprojects, :author)
          @user = user
          @course = options[:course]
        end
        @options = options

        @scope = event_types
      end

      # Returns an array of available event types
      def event_types
        return @event_types unless @event_types.nil?

        @event_types = Redmine::Activity.available_event_types
        if @project
          @event_types = @event_types.select { |o| @project.self_and_descendants.detect { |p| @user.allowed_to?("view_#{o}".to_sym, p) } }
        elsif @course
          @event_types = @event_types.select { |o|  @user.allowed_to?("view_#{o}".to_sym,@course) }
        else
          @event_types = @event_types.select { |o| @user.allowed_to?("view_#{o}".to_sym, @course) } if @course
        end
        @event_types
      end

      # Yields to filter the activity scope
      def scope_select(&block)
        @scope = @scope.select {|t| yield t }
      end

      # Sets the scope
      # Argument can be :all, :default or an array of event types
      def scope=(s)
        case s
        when :all
          @scope = event_types
        when :default
          default_scope!
        else
        @scope = s & event_types
        end
      end

      # Resets the scope to the default scope
      def default_scope!
        @scope = Redmine::Activity.default_event_types
      end

      # Returns an array of events for the given date range
      # sorted in reverse chronological order
      def events(from = nil, to = nil, options={})
        e = []
        @options[:limit] = options[:limit]

        @scope.each do |event_type|
          constantized_providers(event_type).each do |provider|
            e += provider.find_events(event_type, @user, from, to, @options)
          end
        end

        e.sort! {|a,b| b.event_datetime <=> a.event_datetime}

        if options[:limit]
          e = e.slice(0, options[:limit])
        end
        e
      end

      def events_welcome(from = nil, to = nil, options={})
        e = []
        @options[:limit] = options[:limit]
        @scope.each do |event_type|
          constantized_providers(event_type).each do |provider|
            if event_type.to_s != 'course_journals_for_messages'
              cur_objs = provider.find_events1(event_type, @user, from, to, @options)
            end
            if cur_objs
              cur_objs.each do |cur_obj|
                if cur_obj.class == Issue
                  if cur_obj.project != nil && cur_obj.project.project_status != nil
                    e += [cur_obj]
                  end
                else
                  e += [cur_obj]
                end
              end
            end
          end
          if e.count > options[:limit]
            break
          end
        end

        e.sort! {|a,b| b.event_datetime <=> a.event_datetime}

        if options[:limit]
          e = e.slice(0, options[:limit])
        end
        e
      end

      private

      def constantized_providers(event_type)
        @@constantized_providers[event_type]
      end
    end
  end
end