require 'csv'

module Twstats
  class CSVReader
    attr_reader :logs
    attr_reader :tags
    attr_reader :people
    attr_reader :tasks
    attr_reader :projects
    # Create an object structure from the csv file which allows to easily read the content
    def initialize(file)
      @logs = Set.new([])
      @tags = Set.new([])
      @people = Set.new([])
      @tasks = Set.new([])
      @projects = Set.new([])
      # Initialize the reader with a csv file and fill the times as an array of columns
      CSV.foreach(file, DEFAULT_OPTIONS) do |row|
        update_info TwLog.new(row)
      end
    end

    def update_info(log)
      @projects << log.project
      @people << log.who
      log.tags.each{|x| @tags << x}
      @tasks << log.task
      @logs << log
    end

    def mean_time
      logs.inject(0) {|sum, log| sum + log.decimal_time }.to_f/logs.size
    end

    def get_total_time(object, element, filter_billable)
      billable = 0
      non_billable = 0
      case object
        when :projects
          billable = logs.select{|log| log.project == element && log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
          non_billable = logs.select{|log| log.project == element && !log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
        when :tags
          billable = logs.select{|log| log.tags.include?(element) && log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
          non_billable = logs.select{|log| log.tags.include?(element) && !log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
        when :people
          billable = logs.select{|log| log.who == element && log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
          non_billable = logs.select{|log| log.who == element && !log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
        when :tasks
          billable = logs.select{|log| log.task == element && log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
          non_billable = logs.select{|log| log.task == element && !log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
        when :all
          billable = logs.select{|log| log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
          non_billable = logs.select{|log| !log.billable}.inject(0){ |sum, l| sum + l.decimal_time}
        else
          return -99999
      end
      puts "bill: #{billable}, nonbill: #{non_billable}"
      if filter_billable
        return [billable, non_billable]
      else
        return billable + non_billable
      end
    end

    def is_weekly?
      sorted_logs = logs.map{|l| l.date}.sort
      (sorted_logs[-1] - sorted_logs[0]).to_i <= 7
    end

    def not_tagged_tasks
      total_time = 0
      list = []
      logs.each do |log|
        if log.tags.empty?
          list << log
          total_time += log.decimal_time
        end
      end
      {
          total_time: total_time,
          list: list
      }
    end
  end
end