class OverheadTimesheetHook < Redmine::Hook::ViewListener
  include OverheadHelper

  # Adds a table header "Billable" to the report results
  def plugin_timesheet_views_timesheet_group_header(context={})
    return content_tag(:th, l(:overhead_field_billable))
  end

  # Adds a blank table cell to the issue grouping option
  def plugin_timesheet_views_timesheet_time_entry_sum(context={})
    return content_tag(:td, '')
  end
  
  # Adds the gree checkbox image if the TimeEntry is billable
  def plugin_timesheet_views_timesheet_time_entry(context={})
    time_entry = context[:time_entry]
    if time_entry && time_entry.billable? 
      content_tag(:td, image_tag('true.png'), :style => "text-align:center;")
    else 
      content_tag(:td,'') 
    end 
  end

  def plugin_timesheet_model_timesheet_csv_header(context={})
    context[:csv_data] << l(:overhead_field_billable)
  end

  def plugin_timesheet_model_timesheet_time_entry_to_csv(context={})
    if context[:time_entry]
      if context[:time_entry].billable?
        context[:csv_data] << l(:overhead_field_billable)
      else
        context[:csv_data] << l(:overhead_field_overhead)
      end
    end
  end

  # Added a new field for filtering based on "billable?"
  def plugin_timesheet_views_timesheet_form(context={})
    if context[:params] &&
        context[:params][:timesheet] &&
        context[:params][:timesheet][:billable]
      selected_values = context[:params][:timesheet][:billable]
    else
      selected_values = []
    end

    context[:controller].send(:render_to_string,
                              :partial => 'timesheet/overhead_form',
                              :layout => false,
                              :locals => {
                                :list_size => context[:list_size] || 5,
                                :selected_values => selected_values
                              })
  end

  def plugin_timesheet_controller_report_pre_fetch_time_entries(context = {})
    if context[:params] && context[:params][:timesheet] && context[:params][:timesheet][:billable]
      billable_options = context[:params][:timesheet][:billable]
      activities = []

      if billable_options.include?("billable") && context[:timesheet].projects.present?
        context[:timesheet].projects.each do |project|
          activities << project.billable_activities
        end
      end

      if billable_options.include?("overhead") && context[:timesheet].projects.present?
        context[:timesheet].projects.each do |project|
          activities << project.non_billable_activities
        end
      end
      # Overrides the activities directly
      context[:timesheet].activities = activities.flatten.uniq.compact.collect(&:id).sort unless activities.empty?
    end
  end
end