#pragma once

#include <filesystem>
#include <memory>
#include <string>
#include <string_view>
#include <type_traits>
#include <vector>

#include <openvic-dataloader/v2script/AbstractSyntaxTree.hpp>

#include "openvic-simulation/dataloader/Dataloader.hpp"
#include "openvic-simulation/military/Unit.hpp"
#include "openvic-simulation/politics/Government.hpp"
#include "openvic-simulation/politics/Ideology.hpp"
#include "openvic-simulation/politics/Issue.hpp"
#include "openvic-simulation/politics/NationalValue.hpp"
#include "openvic-simulation/politics/PoliticsManager.hpp"
#include "openvic-simulation/pop/Culture.hpp"
#include "openvic-simulation/pop/Religion.hpp"
#include "openvic-simulation/types/Colour.hpp"
#include "openvic-simulation/types/Date.hpp"
#include "openvic-simulation/types/IdentifierRegistry.hpp"

namespace OpenVic {
	struct GameManager;
	struct CountryManager;

	struct CountryParty : HasIdentifier {
		friend struct CountryManager;

		using policy_map_t = std::map<IssueGroup const*, Issue const*>;

	private:
		const Date PROPERTY(start_date);
		const Date PROPERTY(end_date);
		Ideology const& PROPERTY(ideology);
		const policy_map_t PROPERTY(policies);

		CountryParty(
			std::string_view new_identifier, Date new_start_date, Date new_end_date, Ideology const& new_ideology,
			policy_map_t&& new_policies
		);

	public:
		CountryParty(CountryParty&&) = default;
	};

	/* Generic information about a TAG */
	struct Country : HasIdentifierAndColour {
		friend struct CountryManager;

		using unit_names_map_t = std::map<Unit const*, std::vector<std::string>>;
		using government_colour_map_t = std::map<GovernmentType const*, colour_t>;

	private:
		GraphicalCultureType const& PROPERTY(graphical_culture);
		/* Not const to allow elements to be moved, otherwise a copy is forced
		 * which causes a compile error as the copy constructor has been deleted.
		 */
		IdentifierRegistry<CountryParty> IDENTIFIER_REGISTRY_CUSTOM_PLURAL(party, parties);
		const unit_names_map_t PROPERTY(unit_names);
		const bool PROPERTY_CUSTOM_PREFIX(dynamic_tag, is);
		const government_colour_map_t PROPERTY(alternative_colours);

		Country(
			std::string_view new_identifier, colour_t new_colour, GraphicalCultureType const& new_graphical_culture,
			IdentifierRegistry<CountryParty>&& new_parties, unit_names_map_t&& new_unit_names, bool new_dynamic_tag,
			government_colour_map_t&& new_alternative_colours
		);

	public:
		Country(Country&&) = default;

		// TODO - get_colour including alternative colours
	};

	struct CountryManager {
	private:
		IdentifierRegistry<Country> IDENTIFIER_REGISTRY_CUSTOM_PLURAL(country, countries);

		NodeTools::node_callback_t load_country_party(
			PoliticsManager const& politics_manager, IdentifierRegistry<CountryParty>& country_parties
		) const;

	public:
		bool add_country(
			std::string_view identifier, colour_t colour, GraphicalCultureType const* graphical_culture,
			IdentifierRegistry<CountryParty>&& parties, Country::unit_names_map_t&& unit_names, bool dynamic_tag,
			Country::government_colour_map_t&& alternative_colours
		);

		bool load_countries(GameManager const& game_manager, Dataloader const& dataloader, ast::NodeCPtr root);
		bool load_country_data_file(
			GameManager const& game_manager, std::string_view name, bool is_dynamic, ast::NodeCPtr root
		);
	};
}