# frozen_string_literal: true

require 'thor'
require 'kovid'

module Kovid
  class CLI < Thor
    FULL_FLAG = %w[-f --full].freeze

    desc 'check COUNTRY or check "COUNTRY NAME"', 'Returns reported data on provided country. eg: "kovid check "hong kong".'
    method_option :full, aliases: '-f'
    def check(name)
      fetch_country_stats(name)
      data_source
    end

    desc 'country COUNTRY or country "COUNTRY NAME"', 'Returns reported data on provided country. eg: "kovid country "hong kong".'
    method_option :full, aliases: '-f'
    def country(name)
      fetch_country_stats(name)
      data_source
    end

    desc 'state STATE', 'Return reported data on provided state.'
    def state(state)
      puts Kovid.state(state)
      data_source
    end

    desc 'compare COUNTRY COUNTRY', 'Returns full comparison table for given countries. Accepts multiple countries.'
    def compare(*name)
      if FULL_FLAG.include?(name.fetch(-1))
        puts Kovid.country_comparison_full(name[0..-2])
      else
        puts Kovid.country_comparison(name)
      end
      data_source
    end

    desc 'world', 'Returns total number of cases, deaths and recoveries.'
    def world
      puts Kovid.cases
      data_source
    end

    desc 'history COUNTRY or history COUNTRY N', 'Return history of incidents of COUNTRY (in the last N days)'
    def history(*params)
      if params.size == 2
        puts Kovid.history(params.first, params.last)
      else
        puts Kovid.history(params.first, nil)
      end
      data_source
    end

    desc 'eu', 'Returns aggregated data on the EU.'
    def eu
      puts Kovid.eu_aggregate
      data_source
    end

    desc 'europe', 'Returns aggregated data on Europe.'
    def europe
      puts Kovid.europe_aggregate
      data_source
    end

    desc 'africa', 'Returns aggregated data on Africa.'
    def africa
      puts Kovid.africa_aggregate
      data_source
    end

    desc 'sa', 'Returns aggregated data on South America.'
    def sa
      puts Kovid.south_america_aggregate
      data_source
    end

    desc 'asia', 'Returns aggregated data on Asia.'
    def asia
      puts Kovid.asia_aggregate
      data_source
    end

    desc 'version', 'Returns version of kovid'
    def version
      puts Kovid::VERSION
    end

    desc 'histogram', 'Returns a histogram of incidents.'
    def histogram(country, date = nil)
      if date.nil?
        Kovid.info_table("Please add a month and year in the form 'M.YY'")
      else
        puts Kovid.histogram(country, date)
        data_source
      end
    end

    private

    def fetch_country_stats(country)
      if options[:full]
        puts Kovid.country_full(country)
      else
        puts Kovid.country(country)
      end
    end

    def data_source
      source = <<~TEXT
        Sources:
         * JHU CSSE GISand Data
         * https://www.worldometers.info/coronavirus/
      TEXT
      puts source
      puts
    end
  end
end