Chris@1115: # Redmine - project management software Chris@1494: # Copyright (C) 2006-2014 Jean-Philippe Lang Chris@1115: # Chris@1115: # This program is free software; you can redistribute it and/or Chris@1115: # modify it under the terms of the GNU General Public License Chris@1115: # as published by the Free Software Foundation; either version 2 Chris@1115: # of the License, or (at your option) any later version. Chris@1115: # Chris@1115: # This program is distributed in the hope that it will be useful, Chris@1115: # but WITHOUT ANY WARRANTY; without even the implied warranty of Chris@1115: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Chris@1115: # GNU General Public License for more details. Chris@1115: # Chris@1115: # You should have received a copy of the GNU General Public License Chris@1115: # along with this program; if not, write to the Free Software Chris@1115: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. Chris@1115: Chris@1115: module Redmine Chris@1115: module Helpers Chris@1115: class TimeReport Chris@1464: attr_reader :criteria, :columns, :hours, :total_hours, :periods Chris@1115: Chris@1464: def initialize(project, issue, criteria, columns, time_entry_scope) Chris@1115: @project = project Chris@1115: @issue = issue Chris@1115: Chris@1115: @criteria = criteria || [] Chris@1115: @criteria = @criteria.select{|criteria| available_criteria.has_key? criteria} Chris@1115: @criteria.uniq! Chris@1115: @criteria = @criteria[0,3] Chris@1115: Chris@1115: @columns = (columns && %w(year month week day).include?(columns)) ? columns : 'month' Chris@1464: @scope = time_entry_scope Chris@1115: Chris@1115: run Chris@1115: end Chris@1115: Chris@1115: def available_criteria Chris@1115: @available_criteria || load_available_criteria Chris@1115: end Chris@1115: Chris@1115: private Chris@1115: Chris@1115: def run Chris@1115: unless @criteria.empty? Chris@1115: time_columns = %w(tyear tmonth tweek spent_on) Chris@1115: @hours = [] Chris@1517: @scope.includes(:issue, :activity). Chris@1517: group(@criteria.collect{|criteria| @available_criteria[criteria][:sql]} + time_columns). Chris@1517: joins(@criteria.collect{|criteria| @available_criteria[criteria][:joins]}.compact). Chris@1517: sum(:hours).each do |hash, hours| Chris@1115: h = {'hours' => hours} Chris@1115: (@criteria + time_columns).each_with_index do |name, i| Chris@1115: h[name] = hash[i] Chris@1115: end Chris@1115: @hours << h Chris@1115: end Chris@1115: Chris@1115: @hours.each do |row| Chris@1115: case @columns Chris@1115: when 'year' Chris@1115: row['year'] = row['tyear'] Chris@1115: when 'month' Chris@1115: row['month'] = "#{row['tyear']}-#{row['tmonth']}" Chris@1115: when 'week' Chris@1464: row['week'] = "#{row['spent_on'].cwyear}-#{row['tweek']}" Chris@1115: when 'day' Chris@1115: row['day'] = "#{row['spent_on']}" Chris@1115: end Chris@1115: end Chris@1115: Chris@1464: min = @hours.collect {|row| row['spent_on']}.min Chris@1464: @from = min ? min.to_date : Date.today Chris@1115: Chris@1464: max = @hours.collect {|row| row['spent_on']}.max Chris@1464: @to = max ? max.to_date : Date.today Chris@1115: Chris@1115: @total_hours = @hours.inject(0) {|s,k| s = s + k['hours'].to_f} Chris@1115: Chris@1115: @periods = [] Chris@1115: # Date#at_beginning_of_ not supported in Rails 1.2.x Chris@1115: date_from = @from.to_time Chris@1115: # 100 columns max Chris@1115: while date_from <= @to.to_time && @periods.length < 100 Chris@1115: case @columns Chris@1115: when 'year' Chris@1115: @periods << "#{date_from.year}" Chris@1115: date_from = (date_from + 1.year).at_beginning_of_year Chris@1115: when 'month' Chris@1115: @periods << "#{date_from.year}-#{date_from.month}" Chris@1115: date_from = (date_from + 1.month).at_beginning_of_month Chris@1115: when 'week' Chris@1464: @periods << "#{date_from.to_date.cwyear}-#{date_from.to_date.cweek}" Chris@1115: date_from = (date_from + 7.day).at_beginning_of_week Chris@1115: when 'day' Chris@1115: @periods << "#{date_from.to_date}" Chris@1115: date_from = date_from + 1.day Chris@1115: end Chris@1115: end Chris@1115: end Chris@1115: end Chris@1115: Chris@1115: def load_available_criteria Chris@1115: @available_criteria = { 'project' => {:sql => "#{TimeEntry.table_name}.project_id", Chris@1115: :klass => Project, Chris@1115: :label => :label_project}, Chris@1115: 'status' => {:sql => "#{Issue.table_name}.status_id", Chris@1115: :klass => IssueStatus, Chris@1115: :label => :field_status}, Chris@1115: 'version' => {:sql => "#{Issue.table_name}.fixed_version_id", Chris@1115: :klass => Version, Chris@1115: :label => :label_version}, Chris@1115: 'category' => {:sql => "#{Issue.table_name}.category_id", Chris@1115: :klass => IssueCategory, Chris@1115: :label => :field_category}, Chris@1464: 'user' => {:sql => "#{TimeEntry.table_name}.user_id", Chris@1115: :klass => User, Chris@1464: :label => :label_user}, Chris@1115: 'tracker' => {:sql => "#{Issue.table_name}.tracker_id", Chris@1115: :klass => Tracker, Chris@1115: :label => :label_tracker}, Chris@1115: 'activity' => {:sql => "#{TimeEntry.table_name}.activity_id", Chris@1115: :klass => TimeEntryActivity, Chris@1115: :label => :label_activity}, Chris@1115: 'issue' => {:sql => "#{TimeEntry.table_name}.issue_id", Chris@1115: :klass => Issue, Chris@1115: :label => :label_issue} Chris@1115: } Chris@1115: Chris@1464: # Add time entry custom fields Chris@1464: custom_fields = TimeEntryCustomField.all Chris@1464: # Add project custom fields Chris@1464: custom_fields += ProjectCustomField.all Chris@1464: # Add issue custom fields Chris@1464: custom_fields += (@project.nil? ? IssueCustomField.for_all : @project.all_issue_custom_fields) Chris@1464: # Add time entry activity custom fields Chris@1464: custom_fields += TimeEntryActivityCustomField.all Chris@1464: Chris@1115: # Add list and boolean custom fields as available criteria Chris@1517: custom_fields.select {|cf| %w(list bool).include?(cf.field_format) && !cf.multiple?}.each do |cf| Chris@1517: @available_criteria["cf_#{cf.id}"] = {:sql => cf.group_statement, Chris@1464: :joins => cf.join_for_order_statement, Chris@1115: :format => cf.field_format, Chris@1517: :custom_field => cf, Chris@1115: :label => cf.name} Chris@1115: end Chris@1115: Chris@1115: @available_criteria Chris@1115: end Chris@1115: end Chris@1115: end Chris@1115: end