#include "osm_time_range.hpp"

#include <iostream>
#include <fstream>
#include <map>
#include <locale>

#define BOOST_TEST_MODULE OpeningHours

#if defined(__clang__)
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wshorten-64-to-32"
#include <boost/test/included/unit_test.hpp>
#pragma clang diagnostic pop
#else
#include <boost/test/included/unit_test.hpp>
#endif

#include <boost/spirit/include/qi.hpp>

template <typename Char, typename Parser>
bool test(Char const* in, Parser const& p, bool full_match = true)
{
  // we don't care about the result of the "what" function.
  // we only care that all parsers have it:
  boost::spirit::qi::what(p);

  Char const* last = in;
  while (*last)
    last++;
  return boost::spirit::qi::parse(in, last, p) && (!full_match || (in == last));
}


BOOST_AUTO_TEST_CASE(OpeningHours_Locale)
{
  namespace charset = boost::spirit::standard_wide;
  namespace qi = boost::spirit::qi;

  class alltime_ : public qi::symbols<wchar_t>
  {
  public:
    alltime_()
    {
      add
      (L"пн")(L"uu")(L"œæ")
      ;
    }
  } alltime;

  std::locale loc("en_US");
  std::locale prev = std::locale::global(loc);

  BOOST_CHECK(test(L"TeSt",charset::no_case[qi::lit("test")]));
  BOOST_CHECK(test(L"Пн",charset::no_case[alltime]));
  BOOST_CHECK(test(L"UU",charset::no_case[alltime]));
  BOOST_CHECK(test(L"ή",charset::no_case[alltime]));
  BOOST_CHECK(test(L"КАР",charset::no_case[charset::string(L"кар")]));
  BOOST_CHECK(test(L"КрУглосуточно",charset::no_case[qi::lit(L"круглосуточно")]));

  std::locale::global(prev);
}

BOOST_AUTO_TEST_CASE(OpeningHours_TimeHit)
{
  {
    OSMTimeRange oh("06:13-15:00; 16:30+");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("12-12-2013 7:00").IsOpen());
    BOOST_CHECK(oh("12-12-2013 16:00").IsClosed());
    BOOST_CHECK(oh("12-12-2013 20:00").IsOpen());
  }
  {
    OSMTimeRange oh("We-Sa; Mo[1,3] closed; Su[-1,-2] closed; Fr[2] open; Fr[-2], Fr open; Su[-2] -2 days");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("20-03-2015 18:00").IsOpen());
    BOOST_CHECK(oh("17-03-2015 18:00").IsClosed());
  }

  {
    OSMTimeRange oh("We-Fr; Mo[1,3] closed; Su[-1,-2] closed");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("20-03-2015 18:00").IsOpen());
    BOOST_CHECK(oh("17-03-2015 18:00").IsClosed());
  }
  {
    OSMTimeRange oh("We-Fr; Mo[1,3] +1 day closed; Su[-1,-2] -3 days closed");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("20-03-2015 18:00").IsOpen());
    BOOST_CHECK(oh("17-03-2015 18:00").IsClosed());
  }
  {
    OSMTimeRange oh("Mo-Su 14:30-17:00; Mo[1] closed; Su[-1] closed");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("09-03-2015 16:00").IsOpen());
    BOOST_CHECK(oh("02-03-2015 16:00").IsClosed());
    BOOST_CHECK(oh("22-03-2015 16:00").IsOpen());
    BOOST_CHECK(oh("29-03-2015 16:00").IsClosed());
  }
  {
    OSMTimeRange oh("PH,Tu-Su 10:00-18:00; Sa[1] 10:00-18:00 open \"Eintritt ins gesamte Haus frei\"; Jan 1,Dec 24,Dec 25,easter -2 days: closed");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("03-03-2015 16:00").IsOpen());
    BOOST_CHECK(oh.Comment().empty());
    BOOST_CHECK(oh("07-03-2015 16:00").IsOpen());
    BOOST_CHECK(oh.Comment().empty() == false);
  }
  {
    OSMTimeRange oh("Mo-Su 11:00+; Mo [1,3] off");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("04-03-2015 16:00").IsOpen());
    BOOST_CHECK(oh("09-03-2015 16:00").IsOpen());
    BOOST_CHECK(oh("02-03-2015 16:00").IsClosed());
    BOOST_CHECK(oh("16-03-2015 16:00").IsClosed());
  }
  {
    OSMTimeRange oh("08:00-16:00 open, 16:00-03:00 open \"public room\"");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("01-03-2015 20:00").IsOpen());
    BOOST_CHECK(oh("01-03-2015 20:00").Comment() == "public room");
  }
  {
    OSMTimeRange oh("9:00－02:00");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("01-01-2000 07:00").IsClosed());
    BOOST_CHECK(oh("01-01-2000 09:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 12:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 20:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 24:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 00:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 01:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 01:59").IsOpen());
    BOOST_CHECK(oh("01-01-2000 02:00").IsClosed());
  }
  {
    OSMTimeRange oh("09:00am-19:00pm"); // hours > 12, ignore pm
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("01-01-2000 20:00").IsClosed());
    BOOST_CHECK(oh("01-01-2000 8:00").IsClosed());
    BOOST_CHECK(oh("01-01-2000 14:00").IsOpen());
  }
  {
    OSMTimeRange oh("09:00h-7:00 pm"); // symbols case
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK(oh("01-01-2000 20:00").IsClosed());
    BOOST_CHECK(oh("01-01-2000 8:00").IsClosed());
    BOOST_CHECK(oh("01-01-2000 14:00").IsOpen());
  }
  {
    OSMTimeRange oh("Mo-Fr: 11-19 Uhr;Sa: 10-18 Uhr");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK_EQUAL(oh("08-03-2015 20:00").IsClosed(), true);
    BOOST_CHECK_EQUAL(oh("18-03-2015 12:00").IsClosed(), false);
    BOOST_CHECK_EQUAL(oh("16-03-2015 10:00").IsOpen(), false);
    BOOST_CHECK(oh("14-03-2015 10:00").IsOpen());
    BOOST_CHECK(oh("16-03-2015 11:00").IsOpen());
    BOOST_CHECK(oh("01-01-2000 14:00").IsOpen());
  }
  {
    OSMTimeRange oh("Apr 9-19");
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK_EQUAL(oh("01-01-2000 20:00").IsClosed(), false);
    BOOST_CHECK_EQUAL(oh("01-01-2000 8:00").IsClosed(), false);
    BOOST_CHECK(oh("01-01-2000 14:00").IsOpen());
  }
  {
    OSMTimeRange oh("9-19"); // it's no time, it's days of month
    BOOST_CHECK(oh.IsValid());
    BOOST_CHECK_EQUAL(oh("01-01-2000 20:00").IsClosed(), false);
    BOOST_CHECK_EQUAL(oh("01-01-2000 8:00").IsClosed(), false);
    BOOST_CHECK(oh("01-01-2000 14:00").IsOpen());
  }
}

BOOST_AUTO_TEST_CASE(OpeningHours_StaticSet)
{
  {
    OSMTimeRange oh("06:00-02:00/21:03");
    BOOST_CHECK(oh.IsValid());
  }

  {
    OSMTimeRange oh("06:00-09:00/03");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:00-07:00/03");
    BOOST_CHECK(oh.IsValid() == false);
  }
  {
    OSMTimeRange oh("sunrise-sunset");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Su-Th sunset-24:00, 04:00-sunrise; Fr-Sa sunset-sunrise");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr-Sep Su [1,3] 14:30-17:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:00+");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:00-07:00+");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("24/7");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:13-15:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-Su 08:00-23:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("(sunrise+02:00)-(sunset-04:12)");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-Sa; PH off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Jan-Mar 07:00-19:00;Apr-Sep 07:00-22:00;Oct-Dec 07:00-19:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo closed");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:00-23:00 open \"Dining in\"");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:00-23:00 open \"Dining in\" || 00:00-24:00 open \"Drive-through\"");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Tu-Th 20:00-03:00 open \"Club and bar\"; Fr-Sa 20:00-04:00 open \"Club and bar\" || Su-Mo 18:00-02:00 open \"bar\" || Tu-Th 18:00-03:00 open \"bar\" || Fr-Sa 18:00-04:00 open \"bar\"");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00-21:00 \"call us\"");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("10:00-13:30,17:00-20:30");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr-Sep: Mo-Fr 09:00-13:00,14:00-18:00; Apr-Sep: Sa 10:00-13:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo,We,Th,Fr 12:00-18:00; Sa-Su 12:00-17:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Su-Th 11:00-03:00, Fr-Sa 11:00-05:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-We 17:00-01:00, Th,Fr 15:00-01:00; PH off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Tu-Su 10:00-18:00, Mo 12:00-17:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("sunset-sunrise");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("9:00－22:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("jun 16-mar 14 sunrise-sunset");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Sa-Su; PH");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Su; PH");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-Sa; PH off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-Sa; PH off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Пн. — пт.: 09:00 — 21:00; сб.: 09:00 — 19:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("May 15-Sep 23 10:00-18:00; Sep 24 - May 14 \"by appointment\"");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("May-Aug: Mo-Sa 14:30-Sunset; Su 10:30-Sunset; Sep-Apr off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("May-Oct; Nov-Apr off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Fr-Sa");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr 01-Oct 03: Mo-Th 07:00-20:00; Apr 01-Oct 03: Fr-Su 07:00-21:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr 01-Oct 14 07:00-13:00, 15:00-22:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("06:00-08:30; 15:30-16:30");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr-Sep: sunrise-sunset; Dec 1-20: dusk-dawn off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr-Sep: sunrise-sunset; Jan 1 off; Dec 25-26 off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr-Sep: sunrise-sunset; Jan 1: off; Dec 25-26: off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-Fr: 09:00-18:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Apr-Sep sunrise-sunset; Dec 1-20 dusk-dawn off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo off; Tu-Sa 09:30-19:00; Su 10:00-14:30; Jan 1 off; Dec 25-26 off");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo-Th 08:00-19:00, Fr 08:00-17:00, Su[-2] 08:00-15:00");
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Lu-Di 10:00-18:00"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("sunset-sunrise; Sa 09:00-18:00;TU,su,pH OFF"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00h-19:00 h"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00h to 19:00 h"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09h to 19:00"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("9h-19h"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("9am-9pm"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00H-19:00 h"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00am-19:00"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00-19:00;Sa 09:00-18:00;Tu,Su,PH OFF"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00-19:00;Sa 09:00-18:00;Tu,Su,ph Off"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("05:00 – 22:00"); // long dash instead minus
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("05:00 - 22:00"); // minus
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00-20:00 open \"Bei schönem Wetter. Falls unklar kann angerufen werden\""); // charset
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00-22:00; Tu off; dec 31 off; Jan 1 off"); // symbols case
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("9:00-22:00"); // leading zeros
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("09:00-9:30"); // leading zeros
    BOOST_CHECK(oh.IsValid());
  }
  {
    OSMTimeRange oh("Mo 08:00-11:00,14:00-17:00; Tu 08:00-11:00, 14:00-17:00; We 08:00-11:00; Th 08:00-11:00, 14:00-16:00; Fr 08:00-11:00");
    BOOST_CHECK(oh.IsValid());
  }
}


BOOST_AUTO_TEST_CASE( OpeningHours_CountFailed )
{
  std::ifstream datalist("opening-count.lst");
  BOOST_REQUIRE_MESSAGE(datalist.is_open(), "Can't open ./opening-count.lst: " << std::strerror(errno));

  std::string line;
  size_t line_num = 0;
  size_t num_failed = 0;
  size_t num_total = 0;
  std::map<size_t, size_t> desc;

  while (std::getline(datalist, line))
  {
    size_t count = 1;
    std::string datastr;

    auto d = line.find('|');
    if (d == std::string::npos)
    {
      BOOST_WARN_MESSAGE((d != std::string::npos), "Incorrect line " << line_num << " format: " << line);
      datastr = line;
    }
    else
    {
      count = std::stol(line.substr(0,d));
      datastr = line.substr(d+1);
    }

    line_num++;

    OSMTimeRange oh(datastr);
    if (!oh.IsValid()) {
      num_failed += count;
      desc[count]++;
      BOOST_TEST_MESSAGE("-- " << count << " :[" << datastr << "]");
    }
    num_total += count;
  }
  BOOST_CHECK_MESSAGE((num_failed == 0), "Failed " << num_failed << " of " << num_total << " (" << double(num_failed)/(double(num_total)/100) << "%)");
  std::stringstream desc_message;
  for (auto const & e : desc) {
    desc_message << "Weight: " << e.first << " Count: " << e.second << std::endl;
  }
  BOOST_TEST_MESSAGE(desc_message.str());
}