#include "core/anime_season.h"
#include "core/session.h"

namespace Anime {

static bool WinterStartsInDecember()
{
	switch (session.config.service) {
	case Service::MyAnimeList:
		return false;
	default:
		return true;
	}
}

static Season::Name GetSeasonForMonth(Date::Month month)
{
	if (WinterStartsInDecember()) {
		switch (month) {
		case Date::Month::Dec:
		case Date::Month::Jan:
		case Date::Month::Feb:
			return Season::Name::Winter;
		case Date::Month::Mar:
		case Date::Month::Apr:
		case Date::Month::May:
			return Season::Name::Spring;
		case Date::Month::Jun:
		case Date::Month::Jul:
		case Date::Month::Aug:
			return Season::Name::Summer;
		case Date::Month::Sep:
		case Date::Month::Oct:
		case Date::Month::Nov:
			return Season::Name::Autumn;
		default:
			return Season::Name::Unknown;
		}
	} else {
		switch (month) {
		case Date::Month::Jan:
		case Date::Month::Feb:
		case Date::Month::Mar:
			return Season::Name::Winter;
		case Date::Month::Apr:
		case Date::Month::May:
		case Date::Month::Jun:
			return Season::Name::Spring;
		case Date::Month::Jul:
		case Date::Month::Aug:
		case Date::Month::Sep:
			return Season::Name::Summer;
		case Date::Month::Oct:
		case Date::Month::Nov:
		case Date::Month::Dec:
			return Season::Name::Autumn;
		default:
			return Season::Name::Unknown;
		}
	}
}

Season::Season(Season::Name s, Date::Year y)
{
	season = s;
	year = y;
}

Season::Season(const Date &date)
{
	std::optional<Date::Month> month = date.GetMonth();
	if (month)
		season = GetSeasonForMonth(month.value());

	std::optional<Date::Year> y = date.GetYear();
	if (y)
		year = y.value();
}

bool Season::operator==(const Season &o) const
{
	return (season == o.season && year == o.year);
}

bool Season::operator!=(const Season &o) const
{
	return !(*this == o);
}

bool Season::operator<(const Season &o) const
{
	return std::tie(season, year) < std::tie(o.season, o.year);
}

bool Season::operator>(const Season &o) const
{
	return (o < *this);
}

bool Season::operator<=(const Season &o) const
{
	return !(o > *this);
}

bool Season::operator>=(const Season &o) const
{
	return !(*this < o);
}

Season &Season::operator++()
{
	switch (season) {
	case Season::Name::Winter:
		season = Season::Name::Spring;
		break;
	case Season::Name::Spring:
		season = Season::Name::Summer;
		break;
	case Season::Name::Summer:
		season = Season::Name::Autumn;
		break;
	case Season::Name::Autumn:
		season = Season::Name::Winter;
		year++;
		break;
	default:
		season = Season::Name::Unknown;
		break;
	}

	return *this;
}

Season &Season::operator--()
{
	switch (season) {
	case Season::Name::Winter:
		season = Season::Name::Autumn;
		year--;
		break;
	case Season::Name::Spring:
		season = Season::Name::Winter;
		break;
	case Season::Name::Summer:
		season = Season::Name::Spring;
		break;
	case Season::Name::Autumn:
		season = Season::Name::Summer;
		break;
	default:
		season = Season::Name::Unknown;
		break;
	}

	return *this;
}

} // namespace Anime