require "csv"
require "sun_calc"

tries = 0
days_range = 0..365
latitude_range = (-89..89).step(5)
longitude_range = (-179..179).step(10)
estimated_tries = days_range.size * latitude_range.size * longitude_range.size

CSV.open("sun_calc.csv", "wb") do |csv|
  csv << %w[
    date
    latitude
    longitude
    sun_rising_time
    sun_transit_time
    sun_setting_time
    moon_rising_time
    moon_transit_time
    moon_setting_time
    illuminated_fraction
  ]

  days_range.each do |i|
    latitude_range.each do |latitude|
      longitude_range.each do |longitude|
        puts "Try #{tries}/#{estimated_tries}"

        date = Date.new(2024, 1, 1).next_day(i)
        noon = Time.utc(date.year, date.month, date.day, noon)
        sun_results = SunCalc.sun_times(noon, latitude, longitude)
        moon_results = SunCalc.moon_times(noon, latitude, longitude)
        moon_illumination_results = SunCalc.moon_illumination(noon)

        csv << [
          date,
          latitude,
          longitude,
          sun_results[:sunrise],
          sun_results[:solar_noon],
          sun_results[:sunset],
          moon_results[:moonrise],
          moon_results[:lunar_noon],
          moon_results[:moonset],
          moon_illumination_results[:fraction]
        ]

        tries += 1
      end
    end
  end
end

# frozen_string_literal: true

require "csv"
require "faraday"

class DataExtractorMiddleware < Faraday::Middleware
  def on_complete(env)
    json_response = JSON.parse(env.body, symbolize_names: true)
    env.body = json_response.dig(:response, :data)
  end
end

Faraday::Response.register_middleware(
  data_extractor_middleware: DataExtractorMiddleware
)

class OpaleClient
  BASE_URL = "https://opale.imcce.fr/api/v1"

  CELESTIAL_BODIES = [
    SUN = 10,
    MOON = 301
  ].freeze

  def initialize(latitude, longitude)
    @latitude = latitude
    @longitude = longitude
  end

  def self.sun_rise_transit_set(latitude, longitude)
    new(latitude, longitude).sun_rise_transit_set
  end

  def self.moon_rise_transit_set(latitude, longitude)
    new(latitude, longitude).moon_rise_transit_set
  end

  def self.moon_illuminated_fraction(latitude, longitude)
    new(latitude, longitude).moon_illuminated_fraction
  end

  def sun_rise_transit_set
    request(
      "#{BASE_URL}/phenomena/rts/#{SUN}/2024-01-01/#{latitude},#{longitude}?twilight=true&nbd=366"
    )[0][SUN.to_s.to_sym]
  end

  def moon_rise_transit_set
    request(
      "#{BASE_URL}/phenomena/rts/#{MOON}/2024-01-01/#{latitude},#{longitude}?nbd=366"
    )[0][MOON.to_s.to_sym]
  end

  def moon_illuminated_fraction
    request(
      "#{BASE_URL}/positions/#{MOON}?date=2024-01-01T12:00:00&step=1D&nbd=366&quantities=il"
    )
  end

  private

  attr_reader :latitude, :longitude

  def request(url)
    conn = Faraday.new(url: url) do |faraday|
      faraday.use DataExtractorMiddleware
    end

    conn.get(url).body
  end
end

tries = 0
latitude_range = (-89..89).step(5)
longitude_range = (-179..179).step(10)
estimated_tries = latitude_range.size * longitude_range.size

CSV.open("imcce.csv", "wb") do |csv|
  csv << %w[
    date
    latitude
    longitude
    sun_rising_time
    sun_transit_time
    sun_setting_time
    sun_civil_dawn_time
    sun_civil_dusk_time
    sun_nautical_dawn_time
    sun_nautical_dusk_time
    sun_astronomical_dawn_time
    sun_astronomical_dusk_time
    moon_rising_time
    moon_transit_time
    moon_setting_time
    illuminated_fraction
  ]

  latitude_range.each do |latitude|
    longitude_range.each do |longitude|
      puts "Try #{tries}/#{estimated_tries}"

      sun_results = OpaleClient.sun_rise_transit_set(latitude, longitude)
      moon_results = OpaleClient.moon_rise_transit_set(latitude, longitude)
      moon_illumination_results = OpaleClient.moon_illuminated_fraction(latitude, longitude)

      sun_results.each do |sun_result|
        date = sun_result[:calendarDate]
        moon_result = moon_results.find { _1[:calendarDate] == date }
        moon_illumination_result = moon_illumination_results.find { _1[:date] == "#{date}T12:00:00.000" }

        csv << [
          date,
          latitude,
          longitude,
          sun_result[:rise].dig(0, :date),
          sun_result[:upperTransit].dig(0, :date),
          sun_result[:set].dig(0, :date),
          sun_result[:civilDawn].dig(0, :date),
          sun_result[:civilDusk].dig(0, :date),
          sun_result[:nauticalDawn].dig(0, :date),
          sun_result[:nauticalDusk].dig(0, :date),
          sun_result[:astronomicalDawn].dig(0, :date),
          sun_result[:astronomicalDusk].dig(0, :date),
          moon_result[:rise].dig(0, :date),
          moon_result[:upperTransit].dig(0, :date),
          moon_result[:set].dig(0, :date),
          moon_illumination_result[:illuminatedFraction]
        ]
      end

      sleep(1)
      tries += 1
    end
  end
end