require("aocli/advent_of_code/client")

module Aocli
  module AdventOfCode
    MAX_RETRY_COUNT = 5
    module_function

    def get_problem_description(year:, day:)
      wait_for_start(year: year, day: day)
      response = fetch_description(year: year, day: day)
      extract_text(response)
    end

    def get_problem_inputs(year:, day:)
      response = client.get_problem_inputs(year: year, day: day)
      unless response.ok?
        raise(StandardError, "unable to fetch input - #{response.inspect}")
      end

      response.body
    end

    def client
      @client ||= Aocli::AdventOfCode::Client.new
    end

    def wait_for_start(day:, year:)
      while true
        current_time = DateTime.now.new_offset("EST")
        live_time = DateTime.new(year, 12, day, 0, 0, 0, "EST")
        time_to_wait = ((live_time - current_time) * 24 * 60 * 60).to_i
        break if time_to_wait < 1

        print("\rWaiting #{time_to_wait} seconds before fetching...".ljust(40, " "))
        sleep(1)
      end
      puts
    end

    def fetch_description(year:, day:)
      response = nil
      attempt_count = 0
      while attempt_count < MAX_RETRY_COUNT
        attempt_count += 1
        print("\rFetching problem and input attempt ##{attempt_count}".ljust(40, " "))
        response = client.get_problem_description(year: year, day: day)
        break unless response.should_retry?

        sleep(1)
      end
      puts
      raise(StandardError, "Unable to fetch todays problem: #{response.inspect}") unless response.ok?

      response
    end

    def extract_text(response)
      html = Nokogiri::HTML(response.body)
      article = html.css("article").first
      article.search("p,h2").each { |element| element.after("\n") }
      article.text
    end
  end
end