#include "DiplomaticHistory.hpp"

#include "openvic-simulation/GameManager.hpp"
#include "openvic-simulation/dataloader/NodeTools.hpp"

using namespace OpenVic;
using namespace OpenVic::NodeTools;

WarHistory::added_wargoal_t::added_wargoal_t(
	Date new_added,
	Country const* new_actor,
	Country const* new_receiver,
	WargoalType const* new_wargoal,
	std::optional<Country const*> new_third_party,
	std::optional<Province const*> new_target
) : added { new_added }, actor { new_actor }, receiver { new_receiver }, wargoal { new_wargoal }, third_party { new_third_party }, target { new_target } {}

WarHistory::war_participant_t::war_participant_t(
	Country const* new_country,
	Date new_joined,
	std::optional<Date> new_exited
) : country { new_country }, joined { new_joined }, exited { new_exited } {}

WarHistory::WarHistory(
	std::string_view new_war_name,
	std::vector<war_participant_t>&& new_attackers,
	std::vector<war_participant_t>&& new_defenders,
	std::vector<added_wargoal_t>&& new_wargoals
) : war_name { new_war_name }, attackers { std::move(new_attackers) }, defenders { std::move(new_defenders) }, wargoals { std::move(new_wargoals) } {}

AllianceHistory::AllianceHistory(
	Country const* new_first, Country const* new_second,
	Date new_start, Date new_end
) : first { new_first }, second { new_second }, start { new_start }, end { new_end } {}

SubjectHistory::SubjectHistory(
	Country const* new_overlord, Country const* new_subject,
	const type_t new_type, const Date new_start, const Date new_end
) : overlord { new_overlord }, subject { new_subject }, type { new_type }, start { new_start }, end { new_end } {}

void DiplomaticHistoryManager::lock_diplomatic_history() {
	Logger::info("Locked diplomacy history registry after registering ", alliances.size() + subjects.size() + wars.size(), " items");
	locked = true;
}

bool DiplomaticHistoryManager::is_locked() const {
	return locked;
}

std::vector<AllianceHistory const*> DiplomaticHistoryManager::get_alliances(Date date) const {
	std::vector<AllianceHistory const*> ret;
	for (const auto& alliance : alliances) {
		if (alliance.start <= date && alliance.end >= date) {
			ret.push_back(&alliance);
		}
	}
	return ret;
}

std::vector<SubjectHistory const*> DiplomaticHistoryManager::get_subjects(Date date) const {
	std::vector<SubjectHistory const*> ret;
	for (const auto& subject : subjects) {
		if (subject.start <= date && subject.end >= date) {
			ret.push_back(&subject);
		}
	}
	return ret;
}

std::vector<WarHistory const*> DiplomaticHistoryManager::get_wars(Date date) const {
	std::vector<WarHistory const*> ret;
	for (const auto& war : wars) {
		Date start;
		for (const auto& wargoal : war.wargoals) {
			if (wargoal.added < start) start = wargoal.added;
		}
		if (start >= date) ret.push_back(&war);
	}
	return ret;
}

bool DiplomaticHistoryManager::load_diplomacy_history_file(CountryManager const& country_manager, ast::NodeCPtr root) {
	return expect_dictionary_keys(
		"alliance", ZERO_OR_MORE, [this, &country_manager](ast::NodeCPtr node) -> bool {
			Country const* first;
			Country const* second;
			Date start, end;

			bool ret = expect_dictionary_keys(
				"first", ONE_EXACTLY, expect_identifier_or_string(country_manager.expect_country_str(assign_variable_callback_pointer(first))),
				"second", ONE_EXACTLY, expect_identifier_or_string(country_manager.expect_country_str(assign_variable_callback_pointer(second))),
				"start_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(start))),
				"end_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(end)))
			)(node);

			alliances.push_back({ first, second, start, end });
			return ret;
		},
		"vassal", ZERO_OR_MORE, [this, &country_manager](ast::NodeCPtr node) -> bool {
			Country const* overlord;
			Country const* subject;
			Date start, end;

			bool ret = expect_dictionary_keys(
				"first", ONE_EXACTLY, expect_identifier_or_string(country_manager.expect_country_str(assign_variable_callback_pointer(overlord))),
				"second", ONE_EXACTLY, expect_identifier_or_string(country_manager.expect_country_str(assign_variable_callback_pointer(subject))),
				"start_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(start))),
				"end_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(end)))
			)(node);

			subjects.push_back({ overlord, subject, SubjectHistory::type_t::VASSAL, start, end });
			return ret;
		},
		"union", ZERO_OR_MORE, [this, &country_manager](ast::NodeCPtr node) -> bool {
			Country const* overlord;
			Country const* subject;
			Date start, end;

			bool ret = expect_dictionary_keys(
				"first", ONE_EXACTLY, country_manager.expect_country_identifier(assign_variable_callback_pointer(overlord)),
				"second", ONE_EXACTLY, country_manager.expect_country_identifier(assign_variable_callback_pointer(subject)),
				"start_date", ONE_EXACTLY, expect_date(assign_variable_callback(start)),
				"end_date", ONE_EXACTLY, expect_date(assign_variable_callback(end))
			)(node);

			subjects.push_back({ overlord, subject, SubjectHistory::type_t::UNION, start, end });
			return ret;
		},
		"substate", ZERO_OR_MORE, [this, &country_manager](ast::NodeCPtr node) -> bool {
			Country const* overlord;
			Country const* subject;
			Date start, end;

			bool ret = expect_dictionary_keys(
				"first", ONE_EXACTLY, country_manager.expect_country_identifier(assign_variable_callback_pointer(overlord)),
				"second", ONE_EXACTLY, country_manager.expect_country_identifier(assign_variable_callback_pointer(subject)),
				"start_date", ONE_EXACTLY, expect_date(assign_variable_callback(start)),
				"end_date", ONE_EXACTLY, expect_date(assign_variable_callback(end))
			)(node);

			subjects.push_back({ overlord, subject, SubjectHistory::type_t::SUBSTATE, start, end });
			return ret;
		}
	)(root);
}

bool DiplomaticHistoryManager::load_war_history_file(GameManager const& game_manager, ast::NodeCPtr root) {
	std::string name = "";
	std::vector<WarHistory::war_participant_t> attackers;
	std::vector<WarHistory::war_participant_t> defenders;
	std::vector<WarHistory::added_wargoal_t> wargoals;
	Date current_date;

	bool ret = expect_dictionary_keys_and_default(
		[&game_manager, &attackers, &defenders, &wargoals, &current_date, &name](std::string_view key, ast::NodeCPtr node) -> bool {
			bool ret = expect_date_str(assign_variable_callback(current_date))(key);
			ret &= expect_dictionary_keys(
				"add_attacker", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&attackers, &current_date, &name](Country const& country) -> bool {
					for (const auto& attacker : attackers) {
						if (attacker.get_country() == &country) {
							Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to add attacking country ", attacker.get_country()->get_identifier(), " which is already present!");
							return false;
						}
					}
					attackers.push_back({ &country, current_date, {} });
					return true;
				}),
				"add_defender", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&defenders, &current_date, &name](Country const& country) -> bool {
					for (const auto& defender : defenders) {
						if (defender.get_country() == &country) {
							Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to add defending country ", defender.get_country()->get_identifier(), " which is already present!");
							return false;
						}
					}
					defenders.push_back({ &country, current_date, {} });
					return true;
				}),
				"rem_attacker", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&attackers, &current_date, &name](Country const& country) -> bool {
					WarHistory::war_participant_t* participant_to_remove = nullptr;

					for (auto& attacker : attackers) {
						if (attacker.country == &country) {
							participant_to_remove = &attacker;
							break;
						}
					}

					if (participant_to_remove == nullptr) {
						Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to remove attacking country ", country.get_identifier(), " which was not present!");
						return false;
					}

					participant_to_remove->exited.emplace(current_date);
					return true;
				}),
				"rem_defender", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&defenders, &current_date, &name](Country const& country) -> bool {
					WarHistory::war_participant_t* participant_to_remove = nullptr;

					for (auto& defender : defenders) {
						if (defender.country == &country) {
							participant_to_remove = &defender;
							break;
						}
					}

					if (participant_to_remove == nullptr) {
						Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to remove attacking country ", country.get_identifier(), " which was not present!");
						return false;
					}

					participant_to_remove->exited.emplace(current_date);
					return true;
				}),
				"war_goal", ZERO_OR_MORE, [&game_manager, &wargoals, &current_date](ast::NodeCPtr value) -> bool {
					Country const* actor;
					Country const* receiver;
					WargoalType const* type;
					std::optional<Country const*> third_party;
					std::optional<Province const*> target;

					bool ret = expect_dictionary_keys(
						"actor", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(actor)),
						"receiver", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(receiver)),
						"casus_belli", ONE_EXACTLY, game_manager.get_military_manager().get_wargoal_manager().expect_wargoal_type_identifier(assign_variable_callback_pointer(type)),
						"country", ZERO_OR_ONE, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(*third_party)),
						"state_province_id", ZERO_OR_ONE, game_manager.get_map().expect_province_identifier(assign_variable_callback_pointer(*target))
					)(value);
					wargoals.push_back({ current_date, actor, receiver, type, third_party, target });
					return ret;
				}
			)(node);
			return ret;
		},
		"name", ZERO_OR_ONE, expect_string(assign_variable_callback_string(name))
	)(root);

	wars.push_back({ name, std::move(attackers), std::move(defenders), std::move(wargoals) });
	return ret;
}