#include "util/date.hpp"

#if __has_include(<catch2/catch_test_macros.hpp>)
#include <catch2/catch_test_macros.hpp>
#include <catch2/matchers/catch_matchers_all.hpp>
#else
#include <catch2/catch.hpp>
#endif
#include <chrono>
#include <stdexcept>

using namespace std::literals::chrono_literals;

/*
 * Check that the date/time formatter with locale and timezone support is working as expected.
 */

const date::zoned_time<std::chrono::seconds> TEST_TIME = date::zoned_time{
    "UTC", date::local_days{date::Monday[1] / date::January / 2022} + 13h + 4min + 5s};

TEST_CASE("Format UTC time", "[clock][util]") {
  const auto loc = std::locale("C");
  const auto tm = TEST_TIME;

  REQUIRE(fmt::format(loc, "{}", tm).empty());  // no format specified
  REQUIRE(fmt::format(loc, "{:%c %Z}", tm) == "Mon Jan  3 13:04:05 2022 UTC");
  REQUIRE(fmt::format(loc, "{arg:%Y%m%d%H%M%S}", fmt::arg("arg", tm)) == "20220103130405");

  /* Test a few locales that are most likely to be present */
  SECTION("US locale") {
    try {
      const auto loc = std::locale("en_US");

      REQUIRE(fmt::format(loc, "{}", tm).empty());  // no format specified
      REQUIRE_THAT(fmt::format(loc, "{:%c}", tm),   // HowardHinnant/date#704
                   Catch::Matchers::StartsWith("Mon 03 Jan 2022 01:04:05 PM"));
      REQUIRE(fmt::format(loc, "{:%x %X}", tm) == "01/03/2022 01:04:05 PM");
      REQUIRE(fmt::format(loc, "{arg:%Y%m%d%H%M%S}", fmt::arg("arg", tm)) == "20220103130405");
    } catch (const std::runtime_error&) {
      // locale not found; ignore
    }
  }
  SECTION("GB locale") {
    try {
      const auto loc = std::locale("en_GB");

      REQUIRE(fmt::format(loc, "{}", tm).empty());  // no format specified
      REQUIRE_THAT(fmt::format(loc, "{:%c}", tm),   // HowardHinnant/date#704
                   Catch::Matchers::StartsWith("Mon 03 Jan 2022 13:04:05"));
      REQUIRE(fmt::format(loc, "{:%x %X}", tm) == "03/01/22 13:04:05");
      REQUIRE(fmt::format(loc, "{arg:%Y%m%d%H%M%S}", fmt::arg("arg", tm)) == "20220103130405");
    } catch (const std::runtime_error&) {
      // locale not found; ignore
    }
  }
}

TEST_CASE("Format zoned time", "[clock][util]") {
  const auto loc = std::locale("C");
  const auto tm = date::zoned_time{"America/New_York", TEST_TIME};

  REQUIRE(fmt::format(loc, "{}", tm).empty());  // no format specified
  REQUIRE(fmt::format(loc, "{:%c %Z}", tm) == "Mon Jan  3 08:04:05 2022 EST");
  REQUIRE(fmt::format(loc, "{arg:%Y%m%d%H%M%S}", fmt::arg("arg", tm)) == "20220103080405");

  /* Test a few locales that are most likely to be present */
  SECTION("US locale") {
    try {
      const auto loc = std::locale("en_US");

      REQUIRE(fmt::format(loc, "{}", tm).empty());  // no format specified
      REQUIRE_THAT(fmt::format(loc, "{:%c}", tm),   // HowardHinnant/date#704
                   Catch::Matchers::StartsWith("Mon 03 Jan 2022 08:04:05 AM"));
      REQUIRE(fmt::format(loc, "{:%x %X}", tm) == "01/03/2022 08:04:05 AM");
      REQUIRE(fmt::format(loc, "{arg:%Y%m%d%H%M%S}", fmt::arg("arg", tm)) == "20220103080405");
    } catch (const std::runtime_error&) {
      // locale not found; ignore
    }
  }

  SECTION("GB locale") {
    try {
      const auto loc = std::locale("en_GB");

      REQUIRE(fmt::format(loc, "{}", tm).empty());  // no format specified
      REQUIRE_THAT(fmt::format(loc, "{:%c}", tm),   // HowardHinnant/date#704
                   Catch::Matchers::StartsWith("Mon 03 Jan 2022 08:04:05"));
      REQUIRE(fmt::format(loc, "{:%x %X}", tm) == "03/01/22 08:04:05");
      REQUIRE(fmt::format(loc, "{arg:%Y%m%d%H%M%S}", fmt::arg("arg", tm)) == "20220103080405");
    } catch (const std::runtime_error&) {
      // locale not found; ignore
    }
  }
}