Chris@1494: # Redmine - project management software
Chris@1494: # Copyright (C) 2006-2014  Jean-Philippe Lang
Chris@1494: #
Chris@1494: # This program is free software; you can redistribute it and/or
Chris@1494: # modify it under the terms of the GNU General Public License
Chris@1494: # as published by the Free Software Foundation; either version 2
Chris@1494: # of the License, or (at your option) any later version.
Chris@1494: #
Chris@1494: # This program is distributed in the hope that it will be useful,
Chris@1494: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1494: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1494: # GNU General Public License for more details.
Chris@1494: #
Chris@1494: # You should have received a copy of the GNU General Public License
Chris@1494: # along with this program; if not, write to the Free Software
Chris@1494: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1494: 
Chris@1494: class TimelogController < ApplicationController
Chris@1494:   menu_item :issues
Chris@1494: 
Chris@1494:   before_filter :find_project_for_new_time_entry, :only => [:create]
Chris@1494:   before_filter :find_time_entry, :only => [:show, :edit, :update]
Chris@1494:   before_filter :find_time_entries, :only => [:bulk_edit, :bulk_update, :destroy]
Chris@1494:   before_filter :authorize, :except => [:new, :index, :report]
Chris@1494: 
Chris@1494:   before_filter :find_optional_project, :only => [:index, :report]
Chris@1494:   before_filter :find_optional_project_for_new_time_entry, :only => [:new]
Chris@1494:   before_filter :authorize_global, :only => [:new, :index, :report]
Chris@1494: 
Chris@1494:   accept_rss_auth :index
Chris@1494:   accept_api_auth :index, :show, :create, :update, :destroy
Chris@1494: 
Chris@1494:   rescue_from Query::StatementInvalid, :with => :query_statement_invalid
Chris@1494: 
Chris@1494:   helper :sort
Chris@1494:   include SortHelper
Chris@1494:   helper :issues
Chris@1494:   include TimelogHelper
Chris@1494:   helper :custom_fields
Chris@1494:   include CustomFieldsHelper
Chris@1494:   helper :queries
Chris@1494:   include QueriesHelper
Chris@1494: 
Chris@1494:   def index
Chris@1494:     @query = TimeEntryQuery.build_from_params(params, :project => @project, :name => '_')
Chris@1494: 
Chris@1494:     sort_init(@query.sort_criteria.empty? ? [['spent_on', 'desc']] : @query.sort_criteria)
Chris@1494:     sort_update(@query.sortable_columns)
Chris@1494:     scope = time_entry_scope(:order => sort_clause).
Chris@1494:       includes(:project, :user, :issue).
Chris@1494:       preload(:issue => [:project, :tracker, :status, :assigned_to, :priority])
Chris@1494: 
Chris@1494:     respond_to do |format|
Chris@1494:       format.html {
Chris@1494:         @entry_count = scope.count
Chris@1494:         @entry_pages = Paginator.new @entry_count, per_page_option, params['page']
Chris@1494:         @entries = scope.offset(@entry_pages.offset).limit(@entry_pages.per_page).all
Chris@1494:         @total_hours = scope.sum(:hours).to_f
Chris@1494: 
Chris@1494:         render :layout => !request.xhr?
Chris@1494:       }
Chris@1494:       format.api  {
Chris@1494:         @entry_count = scope.count
Chris@1494:         @offset, @limit = api_offset_and_limit
Chris@1494:         @entries = scope.offset(@offset).limit(@limit).preload(:custom_values => :custom_field).all
Chris@1494:       }
Chris@1494:       format.atom {
Chris@1494:         entries = scope.limit(Setting.feeds_limit.to_i).reorder("#{TimeEntry.table_name}.created_on DESC").all
Chris@1494:         render_feed(entries, :title => l(:label_spent_time))
Chris@1494:       }
Chris@1494:       format.csv {
Chris@1494:         # Export all entries
Chris@1494:         @entries = scope.all
Chris@1494:         send_data(query_to_csv(@entries, @query, params), :type => 'text/csv; header=present', :filename => 'timelog.csv')
Chris@1494:       }
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def report
Chris@1494:     @query = TimeEntryQuery.build_from_params(params, :project => @project, :name => '_')
Chris@1494:     scope = time_entry_scope
Chris@1494: 
Chris@1494:     @report = Redmine::Helpers::TimeReport.new(@project, @issue, params[:criteria], params[:columns], scope)
Chris@1494: 
Chris@1494:     respond_to do |format|
Chris@1494:       format.html { render :layout => !request.xhr? }
Chris@1494:       format.csv  { send_data(report_to_csv(@report), :type => 'text/csv; header=present', :filename => 'timelog.csv') }
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def show
Chris@1494:     respond_to do |format|
Chris@1494:       # TODO: Implement html response
Chris@1494:       format.html { render :nothing => true, :status => 406 }
Chris@1494:       format.api
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def new
Chris@1494:     @time_entry ||= TimeEntry.new(:project => @project, :issue => @issue, :user => User.current, :spent_on => User.current.today)
Chris@1494:     @time_entry.safe_attributes = params[:time_entry]
Chris@1494:   end
Chris@1494: 
Chris@1494:   def create
Chris@1494:     @time_entry ||= TimeEntry.new(:project => @project, :issue => @issue, :user => User.current, :spent_on => User.current.today)
Chris@1494:     @time_entry.safe_attributes = params[:time_entry]
Chris@1494: 
Chris@1494:     call_hook(:controller_timelog_edit_before_save, { :params => params, :time_entry => @time_entry })
Chris@1494: 
Chris@1494:     if @time_entry.save
Chris@1494:       respond_to do |format|
Chris@1494:         format.html {
Chris@1494:           flash[:notice] = l(:notice_successful_create)
Chris@1494:           if params[:continue]
Chris@1494:             if params[:project_id]
Chris@1494:               options = {
Chris@1494:                 :time_entry => {:issue_id => @time_entry.issue_id, :activity_id => @time_entry.activity_id},
Chris@1494:                 :back_url => params[:back_url]
Chris@1494:               }
Chris@1494:               if @time_entry.issue
Chris@1494:                 redirect_to new_project_issue_time_entry_path(@time_entry.project, @time_entry.issue, options)
Chris@1494:               else
Chris@1494:                 redirect_to new_project_time_entry_path(@time_entry.project, options)
Chris@1494:               end
Chris@1494:             else
Chris@1494:               options = {
Chris@1494:                 :time_entry => {:project_id => @time_entry.project_id, :issue_id => @time_entry.issue_id, :activity_id => @time_entry.activity_id},
Chris@1494:                 :back_url => params[:back_url]
Chris@1494:               }
Chris@1494:               redirect_to new_time_entry_path(options)
Chris@1494:             end
Chris@1494:           else
Chris@1494:             redirect_back_or_default project_time_entries_path(@time_entry.project)
Chris@1494:           end
Chris@1494:         }
Chris@1494:         format.api  { render :action => 'show', :status => :created, :location => time_entry_url(@time_entry) }
Chris@1494:       end
Chris@1494:     else
Chris@1494:       respond_to do |format|
Chris@1494:         format.html { render :action => 'new' }
Chris@1494:         format.api  { render_validation_errors(@time_entry) }
Chris@1494:       end
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def edit
Chris@1494:     @time_entry.safe_attributes = params[:time_entry]
Chris@1494:   end
Chris@1494: 
Chris@1494:   def update
Chris@1494:     @time_entry.safe_attributes = params[:time_entry]
Chris@1494: 
Chris@1494:     call_hook(:controller_timelog_edit_before_save, { :params => params, :time_entry => @time_entry })
Chris@1494: 
Chris@1494:     if @time_entry.save
Chris@1494:       respond_to do |format|
Chris@1494:         format.html {
Chris@1494:           flash[:notice] = l(:notice_successful_update)
Chris@1494:           redirect_back_or_default project_time_entries_path(@time_entry.project)
Chris@1494:         }
Chris@1494:         format.api  { render_api_ok }
Chris@1494:       end
Chris@1494:     else
Chris@1494:       respond_to do |format|
Chris@1494:         format.html { render :action => 'edit' }
Chris@1494:         format.api  { render_validation_errors(@time_entry) }
Chris@1494:       end
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def bulk_edit
Chris@1494:     @available_activities = TimeEntryActivity.shared.active
Chris@1494:     @custom_fields = TimeEntry.first.available_custom_fields
Chris@1494:   end
Chris@1494: 
Chris@1494:   def bulk_update
Chris@1494:     attributes = parse_params_for_bulk_time_entry_attributes(params)
Chris@1494: 
Chris@1494:     unsaved_time_entry_ids = []
Chris@1494:     @time_entries.each do |time_entry|
Chris@1494:       time_entry.reload
Chris@1494:       time_entry.safe_attributes = attributes
Chris@1494:       call_hook(:controller_time_entries_bulk_edit_before_save, { :params => params, :time_entry => time_entry })
Chris@1494:       unless time_entry.save
Chris@1494:         logger.info "time entry could not be updated: #{time_entry.errors.full_messages}" if logger && logger.info
Chris@1494:         # Keep unsaved time_entry ids to display them in flash error
Chris@1494:         unsaved_time_entry_ids << time_entry.id
Chris@1494:       end
Chris@1494:     end
Chris@1494:     set_flash_from_bulk_time_entry_save(@time_entries, unsaved_time_entry_ids)
Chris@1494:     redirect_back_or_default project_time_entries_path(@projects.first)
Chris@1494:   end
Chris@1494: 
Chris@1494:   def destroy
Chris@1494:     destroyed = TimeEntry.transaction do
Chris@1494:       @time_entries.each do |t|
Chris@1494:         unless t.destroy && t.destroyed?
Chris@1494:           raise ActiveRecord::Rollback
Chris@1494:         end
Chris@1494:       end
Chris@1494:     end
Chris@1494: 
Chris@1494:     respond_to do |format|
Chris@1494:       format.html {
Chris@1494:         if destroyed
Chris@1494:           flash[:notice] = l(:notice_successful_delete)
Chris@1494:         else
Chris@1494:           flash[:error] = l(:notice_unable_delete_time_entry)
Chris@1494:         end
Chris@1494:         redirect_back_or_default project_time_entries_path(@projects.first)
Chris@1494:       }
Chris@1494:       format.api  {
Chris@1494:         if destroyed
Chris@1494:           render_api_ok
Chris@1494:         else
Chris@1494:           render_validation_errors(@time_entries)
Chris@1494:         end
Chris@1494:       }
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494: private
Chris@1494:   def find_time_entry
Chris@1494:     @time_entry = TimeEntry.find(params[:id])
Chris@1494:     unless @time_entry.editable_by?(User.current)
Chris@1494:       render_403
Chris@1494:       return false
Chris@1494:     end
Chris@1494:     @project = @time_entry.project
Chris@1494:   rescue ActiveRecord::RecordNotFound
Chris@1494:     render_404
Chris@1494:   end
Chris@1494: 
Chris@1494:   def find_time_entries
Chris@1494:     @time_entries = TimeEntry.find_all_by_id(params[:id] || params[:ids])
Chris@1494:     raise ActiveRecord::RecordNotFound if @time_entries.empty?
Chris@1494:     @projects = @time_entries.collect(&:project).compact.uniq
Chris@1494:     @project = @projects.first if @projects.size == 1
Chris@1494:   rescue ActiveRecord::RecordNotFound
Chris@1494:     render_404
Chris@1494:   end
Chris@1494: 
Chris@1494:   def set_flash_from_bulk_time_entry_save(time_entries, unsaved_time_entry_ids)
Chris@1494:     if unsaved_time_entry_ids.empty?
Chris@1494:       flash[:notice] = l(:notice_successful_update) unless time_entries.empty?
Chris@1494:     else
Chris@1494:       flash[:error] = l(:notice_failed_to_save_time_entries,
Chris@1494:                         :count => unsaved_time_entry_ids.size,
Chris@1494:                         :total => time_entries.size,
Chris@1494:                         :ids => '#' + unsaved_time_entry_ids.join(', #'))
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def find_optional_project_for_new_time_entry
Chris@1494:     if (project_id = (params[:project_id] || params[:time_entry] && params[:time_entry][:project_id])).present?
Chris@1494:       @project = Project.find(project_id)
Chris@1494:     end
Chris@1494:     if (issue_id = (params[:issue_id] || params[:time_entry] && params[:time_entry][:issue_id])).present?
Chris@1494:       @issue = Issue.find(issue_id)
Chris@1494:       @project ||= @issue.project
Chris@1494:     end
Chris@1494:   rescue ActiveRecord::RecordNotFound
Chris@1494:     render_404
Chris@1494:   end
Chris@1494: 
Chris@1494:   def find_project_for_new_time_entry
Chris@1494:     find_optional_project_for_new_time_entry
Chris@1494:     if @project.nil?
Chris@1494:       render_404
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def find_optional_project
Chris@1494:     if !params[:issue_id].blank?
Chris@1494:       @issue = Issue.find(params[:issue_id])
Chris@1494:       @project = @issue.project
Chris@1494:     elsif !params[:project_id].blank?
Chris@1494:       @project = Project.find(params[:project_id])
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   # Returns the TimeEntry scope for index and report actions
Chris@1494:   def time_entry_scope(options={})
Chris@1494:     scope = @query.results_scope(options)
Chris@1494:     if @issue
Chris@1494:       scope = scope.on_issue(@issue)
Chris@1494:     end
Chris@1494:     scope
Chris@1494:   end
Chris@1494: 
Chris@1494:   def parse_params_for_bulk_time_entry_attributes(params)
Chris@1494:     attributes = (params[:time_entry] || {}).reject {|k,v| v.blank?}
Chris@1494:     attributes.keys.each {|k| attributes[k] = '' if attributes[k] == 'none'}
Chris@1494:     attributes[:custom_field_values].reject! {|k,v| v.blank?} if attributes[:custom_field_values]
Chris@1494:     attributes
Chris@1494:   end
Chris@1494: end