J3RN/time-tracker

require "csv"

class TimeEntriesController < ApplicationController
  before_action :set_time_entry, only: [:show, :edit, :update, :destroy,
                                        :stop_time, :start_time]
  before_action :set_tasks, only: [:new, :edit, :create, :update]
  before_action :set_date, only: [:index, :updates_all_time_entries]
  before_action :set_time_entries, only: [:index, :updates_all_time_entries]
  before_action :set_overrun_entries,
                only: [:index, :updates_all_time_entries],
                if: -> { @date == Date.today }
  respond_to :html, :js

  def index; end

  def show
    respond_with(@time_entry)
  end

  def new
    @time_entry = TimeEntry.new(running: true, start_time: Time.now)
    respond_with(@time_entry)
  end

  def edit; end

  def create
    @time_entry = TimeEntry.new(time_entry_params)
    @time_entry.user = current_user
    @time_entry.start_time ||= Time.now

    if @time_entry.save
      redirect_to time_entries_path(date: @time_entry.start_time.to_date)
    else
      render "new"
    end
  end

  def start_time
    @time_entry.start_time = Time.now
    @time_entry.running = true
    if @time_entry.save
      redirect_to time_entries_path
    else
      redirect_to time_entries_path, alert: "Failed to start timer"
    end
  end

  def stop_time
    @time_entry.duration = @time_entry.real_duration
    @time_entry.running = false
    if @time_entry.save
      redirect_to time_entries_path
    else
      redirect_to time_entries_path, alert: "Failed to stop timer"
    end
  end

  def update
    if @time_entry.update(time_entry_params)
      redirect_to time_entries_path(date: @time_entry.start_time.to_date)
    else
      render "edit"
    end
  end

  def destroy
    @time_entry.destroy
    respond_with(@time_entry)
  end

  def export
    @time_entries = TimeEntry.includes(task: [:tags])
    @time_entries = @time_entries.where(user: current_user) unless current_user.admin?
    send_data @time_entries.to_csv
  end

  def updates_all_time_entries
    render action: "updates_all_time_entries", layout: nil
  end

  private

  def set_tasks
    @tasks = Task.includes(:time_entries, :tags)
    @tasks = @tasks.where(user: current_user) unless current_user.admin?
    @tasks = @tasks.active.order_todo
    if @time_entry.try(:task) && @tasks.exclude?(@time_entry.task)
      @tasks << @time_entry.task
    end
    @tasks = @tasks.map { |task| [task.explicit_name, task.id] }
  end

  def set_time_entry
    @time_entry = TimeEntry.find(params[:id])
  end

  def set_date
    # If there's a date, make sure all time entries are from that date
    # Default to today
    @date = params[:date] ? Date.parse(params[:date]) : Date.today
  end

  def set_time_entries
    @time_entries = TimeEntry.filter_by_date(@date)
                             .includes(task: :tags)
                             .order(start_time: :desc)
    @time_entries = @time_entries.where(user: current_user) unless @admin
    @total = @time_entries.total_real_duration
  end

  def set_overrun_entries
    entries = TimeEntry.overrun
                       .includes(task: :tags)
                       .order(start_time: :desc)
    entries = entries.where(user: current_user) unless @admin
    @overrun_entries = entries.group_by { |e| e.start_time.to_date }
                              .sort_by { |date, _entries| date }
                              .to_h
  end

  def time_entry_params
    new_params = params.require(:time_entry).permit(:user_id, :task_id,
                                                    :duration, :start_time,
                                                    :note, :running, :goal,
                                                    :result)

    if new_params[:running] == "1"
      new_params = remove_stopped_elements(new_params)
    else
      start_time = Time.american_date(new_params[:start_time])
      new_params[:start_time] = Time.zone.local_to_utc(start_time)
    end

    new_params
  end

  def remove_stopped_elements(new_params)
    new_params.reject do |key, _|
      [:duration, :start_time, :result].include? key.to_sym
    end
  end
end