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