#include "Culture.hpp"

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

using namespace OpenVic;
using namespace OpenVic::NodeTools;

GraphicalCultureType::GraphicalCultureType(const std::string_view new_identifier) : HasIdentifier { new_identifier } {}

CultureGroup::CultureGroup(const std::string_view new_identifier, const std::string_view new_leader,
	GraphicalCultureType const& new_unit_graphical_culture_type, bool new_is_overseas)
	: HasIdentifier { new_identifier }, leader { new_leader },
	  unit_graphical_culture_type { new_unit_graphical_culture_type },
	  is_overseas { new_is_overseas } {}


std::string const& CultureGroup::get_leader() const {
	return leader;
}

GraphicalCultureType const& CultureGroup::get_unit_graphical_culture_type() const {
	return unit_graphical_culture_type;
}

bool CultureGroup::get_is_overseas() const {
	return is_overseas;
}

Culture::Culture(const std::string_view new_identifier, colour_t new_colour, CultureGroup const& new_group,
	std::vector<std::string> const& new_first_names, std::vector<std::string> const& new_last_names)
	: HasIdentifierAndColour { new_identifier, new_colour, true },
	  group { new_group },
	  first_names { new_first_names },
	  last_names { new_last_names } {}

CultureGroup const& Culture::get_group() const {
	return group;
}

std::vector<std::string> const& Culture::get_first_names() const {
	return first_names;
}

std::vector<std::string> const& Culture::get_last_names() const {
	return last_names;
}

CultureManager::CultureManager()
	: graphical_culture_types { "graphical culture types" },
	  culture_groups { "culture groups" },
	  cultures { "cultures" } {}

bool CultureManager::add_graphical_culture_type(const std::string_view identifier) {
	if (identifier.empty()) {
		Logger::error("Invalid culture group identifier - empty!");
		return false;
	}
	return graphical_culture_types.add_item({ identifier });
}

bool CultureManager::add_culture_group(const std::string_view identifier, const std::string_view leader, GraphicalCultureType const* graphical_culture_type, bool is_overseas) {
	if (!graphical_culture_types.is_locked()) {
		Logger::error("Cannot register culture groups until graphical culture types are locked!");
		return false;
	}
	if (identifier.empty()) {
		Logger::error("Invalid culture group identifier - empty!");
		return false;
	}
	if (leader.empty()) {
		Logger::error("Invalid culture group leader - empty!");
		return false;
	}
	if (graphical_culture_type == nullptr) {
		Logger::error("Null graphical culture type for ", identifier);
		return false;
	}
	return culture_groups.add_item({ identifier, leader, *graphical_culture_type, is_overseas });
}

bool CultureManager::add_culture(const std::string_view identifier, colour_t colour, CultureGroup const* group, std::vector<std::string> const& first_names, std::vector<std::string> const& last_names) {
	if (!culture_groups.is_locked()) {
		Logger::error("Cannot register cultures until culture groups are locked!");
		return false;
	}
	if (identifier.empty()) {
		Logger::error("Invalid culture identifier - empty!");
		return false;
	}
	if (group == nullptr) {
		Logger::error("Null culture group for ", identifier);
		return false;
	}
	if (colour > MAX_COLOUR_RGB) {
		Logger::error("Invalid culture colour for ", identifier, ": ", colour_to_hex_string(colour));
		return false;
	}
	return cultures.add_item({ identifier, colour, *group, first_names, last_names });
}

bool CultureManager::load_graphical_culture_type_file(ast::NodeCPtr root) {
	const bool ret = expect_list_reserve_length(
		graphical_culture_types,
		expect_identifier(
			std::bind(&CultureManager::add_graphical_culture_type, this, std::placeholders::_1)
		)
	)(root);
	lock_graphical_culture_types();
	return ret;
}

bool CultureManager::_load_culture_group(size_t& total_expected_cultures,
	GraphicalCultureType const* default_unit_graphical_culture_type,
	const std::string_view culture_group_key, ast::NodeCPtr culture_group_node) {

	std::string_view leader;
	GraphicalCultureType const* unit_graphical_culture_type = default_unit_graphical_culture_type;
	bool is_overseas = true;

	bool ret = expect_dictionary_keys_and_length(
		[&total_expected_cultures](size_t size) -> size_t {
			total_expected_cultures += size;
			return size;
		},
		ALLOW_OTHER_KEYS,
		"leader", ONE_EXACTLY, [&total_expected_cultures, &leader](ast::NodeCPtr node) -> bool {
			total_expected_cultures--;
			return expect_identifier(assign_variable_callback(leader))(node);
		},
		"unit", ZERO_OR_ONE, [this, &total_expected_cultures, &unit_graphical_culture_type](ast::NodeCPtr node) -> bool {
			total_expected_cultures--;
			return expect_graphical_culture_type(unit_graphical_culture_type)(node);
		},
		"union", ZERO_OR_ONE, [&total_expected_cultures](ast::NodeCPtr) -> bool {
			total_expected_cultures--;
			return true;
		},
		"is_overseas", ZERO_OR_ONE, [&total_expected_cultures, &is_overseas](ast::NodeCPtr node) -> bool {
			total_expected_cultures--;
			return expect_bool(assign_variable_callback(is_overseas))(node);
		}
	)(culture_group_node);
	ret &= add_culture_group(culture_group_key, leader, unit_graphical_culture_type, is_overseas);
	return ret;
}

bool CultureManager::_load_culture(CultureGroup const* culture_group,
	const std::string_view culture_key, ast::NodeCPtr culture_node) {

	colour_t colour = NULL_COLOUR;
	std::vector<std::string> first_names, last_names;

	bool ret = expect_dictionary_keys(
		"color", ONE_EXACTLY, expect_colour(assign_variable_callback(colour)),
		"first_names", ONE_EXACTLY, name_list_callback(first_names),
		"last_names", ONE_EXACTLY, name_list_callback(last_names),
		"radicalism", ZERO_OR_ONE, success_callback,
		"primary", ZERO_OR_ONE, success_callback
	)(culture_node);
	ret &= add_culture(culture_key, colour, culture_group, first_names, last_names);
	return ret;
}

bool CultureManager::load_culture_file(ast::NodeCPtr root) {
	if (!graphical_culture_types.is_locked()) {
		Logger::error("Cannot load culture groups until graphical culture types are locked!");
		return false;
	}

	static const std::string default_unit_graphical_culture_type_identifier = "Generic";
	GraphicalCultureType const* const default_unit_graphical_culture_type = get_graphical_culture_type_by_identifier(default_unit_graphical_culture_type_identifier);
	if (default_unit_graphical_culture_type == nullptr) {
		Logger::error("Failed to find default unit graphical culture type: ", default_unit_graphical_culture_type_identifier);
	}

	size_t total_expected_cultures = 0;
	bool ret = expect_dictionary_reserve_length(
		culture_groups,
		[this, default_unit_graphical_culture_type, &total_expected_cultures](std::string_view key, ast::NodeCPtr value) -> bool {
			return _load_culture_group(total_expected_cultures, default_unit_graphical_culture_type, key, value);
		}
	)(root);
	lock_culture_groups();
	cultures.reserve(cultures.size() + total_expected_cultures);

	ret &= expect_dictionary(
		[this](std::string_view culture_group_key, ast::NodeCPtr culture_group_value) -> bool {
			CultureGroup const* culture_group = get_culture_group_by_identifier(culture_group_key);
			return expect_dictionary(
				[this, culture_group](std::string_view key, ast::NodeCPtr value) -> bool {
					if (key == "leader" || key == "unit" || key == "union" || key == "is_overseas") return true;
					return _load_culture(culture_group, key, value);
				}
			)(culture_group_value);
		}
	)(root);
	lock_cultures();
	return ret;
}