#include "MenuSingleton.hpp"

#include <godot_cpp/variant/utility_functions.hpp>

#include <openvic-simulation/economy/GoodDefinition.hpp>
#include <openvic-simulation/GameManager.hpp>
#include <openvic-simulation/modifier/Modifier.hpp>
#include <openvic-simulation/types/fixed_point/FixedPoint.hpp>

#include "openvic-extension/classes/GFXPieChartTexture.hpp"
#include "openvic-extension/classes/GUINode.hpp"
#include "openvic-extension/singletons/GameSingleton.hpp"
#include "openvic-extension/utility/ClassBindings.hpp"
#include "openvic-extension/utility/Utilities.hpp"

using namespace godot;
using namespace OpenVic;

StringName const& MenuSingleton::_signal_population_menu_province_list_changed() {
	static const StringName signal_population_menu_province_list_changed = "population_menu_province_list_changed";
	return signal_population_menu_province_list_changed;
}
StringName const& MenuSingleton::_signal_population_menu_province_list_selected_changed() {
	static const StringName signal_population_menu_province_list_selected_changed =
		"population_menu_province_list_selected_changed";
	return signal_population_menu_province_list_selected_changed;
}
StringName const& MenuSingleton::_signal_population_menu_pops_changed() {
	static const StringName signal_population_menu_pops_changed = "population_menu_pops_changed";
	return signal_population_menu_pops_changed;
}
StringName const& MenuSingleton::_signal_search_cache_changed() {
	static const StringName signal_search_cache_changed = "search_cache_changed";
	return signal_search_cache_changed;
}
StringName const& MenuSingleton::_signal_update_tooltip() {
	static const StringName signal_update_tooltip = "update_tooltip";
	return signal_update_tooltip;
}

String MenuSingleton::get_state_name(State const& state) const {
	StateSet const& state_set = state.get_state_set();

	const String region_identifier = Utilities::std_to_godot_string(state_set.get_region().get_identifier());

	String name = tr(region_identifier);

	const bool named = name != region_identifier;
	const bool owned = state.get_owner() != nullptr;
	const bool split = state_set.get_state_count() > 1;

	if (!named) {
		// Capital province name
		name = tr(GUINode::format_province_name(Utilities::std_to_godot_string(state.get_capital()->get_identifier())));

		if (!owned) {
			static const StringName region_key = "REGION_NAME";
			static const String name_key = "$NAME$";

			String region = tr(region_key);

			if (region != region_key) {
				// CAPITAL Region
				return region.replace(name_key, name);
			}
		}
	}

	if (owned && split) {
		// COUNTRY STATE/CAPITAL
		return get_country_adjective(*state.get_owner()) + " " + name;
	}

	// STATE/CAPITAL
	return name;
}

String MenuSingleton::get_country_name(CountryInstance const& country) const {
	if (country.get_government_type() != nullptr && !country.get_government_type()->get_identifier().empty()) {
		const String government_name_key = Utilities::std_to_godot_string(StringUtils::append_string_views(
			country.get_identifier(), "_", country.get_government_type()->get_identifier()
		));

		String government_name = tr(government_name_key);

		if (government_name != government_name_key) {
			return government_name;
		}
	}

	return tr(Utilities::std_to_godot_string(country.get_identifier()));
}

String MenuSingleton::get_country_adjective(CountryInstance const& country) const {
	static constexpr std::string_view adjective = "_ADJ";

	if (country.get_government_type() != nullptr && !country.get_government_type()->get_identifier().empty()) {
		const String government_adjective_key = Utilities::std_to_godot_string(StringUtils::append_string_views(
			country.get_identifier(), "_", country.get_government_type()->get_identifier(), adjective
		));

		String government_adjective = tr(government_adjective_key);

		if (government_adjective != government_adjective_key) {
			return government_adjective;
		}
	}

	return tr(Utilities::std_to_godot_string(StringUtils::append_string_views(country.get_identifier(), adjective)));
}

String MenuSingleton::make_modifier_effects_tooltip(ModifierValue const& modifier) const {
	if (modifier.empty()) {
		return {};
	}

	String result;

	for (auto const& [effect, value] : modifier.get_values()) {
		if (!result.is_empty()) {
			result += "\n";
		}

		result += tr(Utilities::std_to_godot_string(effect->get_localisation_key()));

		static const String post_name_text = ": " + GUILabel::get_colour_marker();
		result += post_name_text;

		if (value == 0) {
			result += "Y";
		} else if (effect->is_positive_good() == value > 0) {
			result += "G";
		} else {
			result += "R";
		}

		if (value >= 0) {
			result += "+";
		}

		static constexpr int32_t DECIMAL_PLACES = 2;

		using enum ModifierEffect::format_t;

		switch (effect->get_format()) {
		case PROPORTION_DECIMAL:
			result += GUINode::float_to_string_dp((value * 100).to_float(), DECIMAL_PLACES) + "%";
			break;
		case PERCENTAGE_DECIMAL:
			result += GUINode::float_to_string_dp(value.to_float(), DECIMAL_PLACES) + "%";
			break;
		case INT:
			result += String::num_int64(value.to_int64_t());
			break;
		case RAW_DECIMAL: [[fallthrough]];
		default: // Use raw decimal as fallback format
			result += GUINode::float_to_string_dp(value.to_float(), DECIMAL_PLACES);
			break;
		}

		static const String end_text = GUILabel::get_colour_marker() + String { "!" };
		result += end_text;
	}

	return result;
}

String MenuSingleton::make_rules_tooltip(RuleSet const& rules) const {
	if (rules.empty()) {
		return {};
	}

	static const StringName yes_key = "YES";
	static const StringName no_key = "NO";

	static const String start_text = ": " + GUILabel::get_colour_marker();
	static const String end_text = GUILabel::get_colour_marker() + String { "!" };

	const String enabled_text = start_text + String { "G" } + tr(yes_key) + end_text;
	const String disabled_text = start_text + String { "R" } + tr(no_key) + end_text;

	String result;

	for (auto const& [rule_group, rule_map] : rules.get_rule_groups()) {
		for (auto const& [rule, enabled] : rule_map) {
			if (!result.is_empty()) {
				result += "\n";
			}

			result += tr(Utilities::std_to_godot_string(rule->get_localisation_key()))
				+ (enabled ? enabled_text : disabled_text);
		}
	}

	return result;
}

void MenuSingleton::_bind_methods() {
	OV_BIND_SMETHOD(get_tooltip_separator);
	OV_BIND_METHOD(MenuSingleton::get_country_name_from_identifier, { "country_identifier" });
	OV_BIND_METHOD(MenuSingleton::get_country_adjective_from_identifier, { "country_identifier" });

	/* TOOLTIP */
	OV_BIND_METHOD(MenuSingleton::show_tooltip, { "text", "substitution_dict", "position" });
	OV_BIND_METHOD(MenuSingleton::show_control_tooltip, { "text", "substitution_dict", "control" });
	OV_BIND_METHOD(MenuSingleton::hide_tooltip);

	ADD_SIGNAL(MethodInfo(
		_signal_update_tooltip(), PropertyInfo(Variant::STRING, "text"),
		PropertyInfo(Variant::DICTIONARY, "substitution_dict"), PropertyInfo(Variant::VECTOR2, "position")
	));

	/* PROVINCE OVERVIEW PANEL */
	OV_BIND_METHOD(MenuSingleton::get_province_info_from_index, { "index" });
	OV_BIND_METHOD(MenuSingleton::get_province_building_count);
	OV_BIND_METHOD(MenuSingleton::get_province_building_identifier, { "building_index" });
	OV_BIND_METHOD(MenuSingleton::expand_selected_province_building, { "building_index" });
	OV_BIND_METHOD(MenuSingleton::get_slave_pop_icon_index);
	OV_BIND_METHOD(MenuSingleton::get_administrative_pop_icon_index);
	OV_BIND_METHOD(MenuSingleton::get_rgo_owner_pop_icon_index);

	/* TOPBAR */
	OV_BIND_METHOD(MenuSingleton::get_topbar_info);

	/* TIME/SPEED CONTROL PANEL */
	OV_BIND_METHOD(MenuSingleton::set_paused, { "paused" });
	OV_BIND_METHOD(MenuSingleton::toggle_paused);
	OV_BIND_METHOD(MenuSingleton::is_paused);
	OV_BIND_METHOD(MenuSingleton::increase_speed);
	OV_BIND_METHOD(MenuSingleton::decrease_speed);
	OV_BIND_METHOD(MenuSingleton::get_speed);
	OV_BIND_METHOD(MenuSingleton::can_increase_speed);
	OV_BIND_METHOD(MenuSingleton::can_decrease_speed);
	OV_BIND_METHOD(MenuSingleton::get_longform_date);

	/* POPULATION MENU */
	OV_BIND_METHOD(MenuSingleton::get_population_menu_province_list_row_count);
	OV_BIND_METHOD(MenuSingleton::get_population_menu_province_list_rows, { "start", "count" });
	OV_BIND_METHOD(
		MenuSingleton::population_menu_select_province_list_entry, { "select_index", "set_scroll_index" }, DEFVAL(false)
	);
	OV_BIND_METHOD(MenuSingleton::population_menu_select_province, { "province_index" });
	OV_BIND_METHOD(MenuSingleton::population_menu_toggle_expanded, { "toggle_index", "emit_selected_changed" }, DEFVAL(true));

	OV_BIND_METHOD(MenuSingleton::population_menu_update_locale_sort_cache);
	OV_BIND_METHOD(MenuSingleton::population_menu_select_sort_key, { "sort_key" });
	OV_BIND_METHOD(MenuSingleton::get_population_menu_pop_rows, { "start", "count" });
	OV_BIND_METHOD(MenuSingleton::get_population_menu_pop_row_count);

	OV_BIND_METHOD(MenuSingleton::get_population_menu_pop_filter_setup_info);
	OV_BIND_METHOD(MenuSingleton::get_population_menu_pop_filter_info);
	OV_BIND_METHOD(MenuSingleton::population_menu_toggle_pop_filter, { "filter_index" });
	OV_BIND_METHOD(MenuSingleton::population_menu_select_all_pop_filters);
	OV_BIND_METHOD(MenuSingleton::population_menu_deselect_all_pop_filters);

	OV_BIND_METHOD(MenuSingleton::get_population_menu_distribution_setup_info);
	OV_BIND_METHOD(MenuSingleton::get_population_menu_distribution_info);

	ADD_SIGNAL(MethodInfo(_signal_population_menu_province_list_changed()));
	ADD_SIGNAL(
		MethodInfo(_signal_population_menu_province_list_selected_changed(), PropertyInfo(Variant::INT, "scroll_index"))
	);
	ADD_SIGNAL(MethodInfo(_signal_population_menu_pops_changed()));

	using enum population_menu_t::ProvinceListEntry;
	BIND_ENUM_CONSTANT(LIST_ENTRY_NONE);
	BIND_ENUM_CONSTANT(LIST_ENTRY_COUNTRY);
	BIND_ENUM_CONSTANT(LIST_ENTRY_STATE);
	BIND_ENUM_CONSTANT(LIST_ENTRY_PROVINCE);

	using enum population_menu_t::PopSortKey;
	BIND_ENUM_CONSTANT(NONE);
	BIND_ENUM_CONSTANT(SORT_SIZE);
	BIND_ENUM_CONSTANT(SORT_TYPE);
	BIND_ENUM_CONSTANT(SORT_CULTURE);
	BIND_ENUM_CONSTANT(SORT_RELIGION);
	BIND_ENUM_CONSTANT(SORT_LOCATION);
	BIND_ENUM_CONSTANT(SORT_MILITANCY);
	BIND_ENUM_CONSTANT(SORT_CONSCIOUSNESS);
	BIND_ENUM_CONSTANT(SORT_IDEOLOGY);
	BIND_ENUM_CONSTANT(SORT_ISSUES);
	BIND_ENUM_CONSTANT(SORT_UNEMPLOYMENT);
	BIND_ENUM_CONSTANT(SORT_CASH);
	BIND_ENUM_CONSTANT(SORT_LIFE_NEEDS);
	BIND_ENUM_CONSTANT(SORT_EVERYDAY_NEEDS);
	BIND_ENUM_CONSTANT(SORT_LUXURY_NEEDS);
	BIND_ENUM_CONSTANT(SORT_REBEL_FACTION);
	BIND_ENUM_CONSTANT(SORT_SIZE_CHANGE);
	BIND_ENUM_CONSTANT(SORT_LITERACY);

	/* Find/Search Panel */
	OV_BIND_METHOD(MenuSingleton::generate_search_cache);
	OV_BIND_METHOD(MenuSingleton::update_search_results, { "text" });
	OV_BIND_METHOD(MenuSingleton::get_search_result_rows, { "start", "count" });
	OV_BIND_METHOD(MenuSingleton::get_search_result_row_count);
	OV_BIND_METHOD(MenuSingleton::get_search_result_position, { "result_index" });

	ADD_SIGNAL(MethodInfo(_signal_search_cache_changed()));
}

MenuSingleton* MenuSingleton::get_singleton() {
	return singleton;
}

MenuSingleton::MenuSingleton() : population_menu {
	.pop_type_sort_cache { nullptr }, .culture_sort_cache { nullptr }, .religion_sort_cache { nullptr },
	.province_sort_cache { nullptr }, .rebel_type_sort_cache { nullptr }
} {
	ERR_FAIL_COND(singleton != nullptr);
	singleton = this;
}

MenuSingleton::~MenuSingleton() {
	ERR_FAIL_COND(singleton != this);
	singleton = nullptr;
}

String MenuSingleton::get_tooltip_separator() {
	static const String tooltip_separator = "\n" + String { "-" }.repeat(14) + "\n";
	return tooltip_separator;
}

String MenuSingleton::get_country_name_from_identifier(String const& country_identifier) const {
	if (country_identifier.is_empty()) {
		return {};
	}

	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});

	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, {});

	CountryInstance const* country = instance_manager->get_country_instance_manager().get_country_instance_by_identifier(
		Utilities::godot_to_std_string(country_identifier)
	);
	ERR_FAIL_NULL_V(country, {});

	return get_country_name(*country);
}

String MenuSingleton::get_country_adjective_from_identifier(String const& country_identifier) const {
	if (country_identifier.is_empty()) {
		return {};
	}

	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});

	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, {});

	CountryInstance const* country = instance_manager->get_country_instance_manager().get_country_instance_by_identifier(
		Utilities::godot_to_std_string(country_identifier)
	);
	ERR_FAIL_NULL_V(country, {});

	return get_country_adjective(*country);
}

/* TOOLTIP */

void MenuSingleton::show_tooltip(String const& text, Dictionary const& substitution_dict, Vector2 const& position) {
	emit_signal(_signal_update_tooltip(), text, substitution_dict, position);
}

void MenuSingleton::show_control_tooltip(String const& text, Dictionary const& substitution_dict, Control const* control) {
	ERR_FAIL_NULL(control);

	using namespace OpenVic::Utilities::literals;
	static const Vector2 offset { 0.0_real, 64.0_real };

	show_tooltip(text, substitution_dict, control->get_global_position() + offset);
}

void MenuSingleton::hide_tooltip() {
	show_tooltip({}, {}, {});
}

/* PROVINCE OVERVIEW PANEL */

static TypedArray<Dictionary> _make_buildings_dict_array(
	ProvinceInstance const* province
) {
	std::vector<BuildingInstance> const& buildings = province->get_buildings();

	if (buildings.empty()) {
		return {};
	}

	static const StringName building_info_level_key = "level";
	static const StringName building_info_expansion_state_key = "expansion_state";
	static const StringName building_info_start_date_key = "start_date";
	static const StringName building_info_end_date_key = "end_date";
	static const StringName building_info_expansion_progress_key = "expansion_progress";

	/* This system relies on the province buildings all being present in the right order. It will have to
	 * be changed if we want to support variable combinations and permutations of province buildings. */
	TypedArray<Dictionary> buildings_array;

	if (buildings_array.resize(buildings.size()) == OK) {
		for (size_t idx = 0; idx < buildings.size(); ++idx) {
			BuildingInstance const& building = buildings[idx];

			Dictionary building_dict;
			building_dict[building_info_level_key] = static_cast<int32_t>(building.get_level());
			building_dict[building_info_expansion_state_key] = static_cast<int32_t>(building.get_expansion_state());
			building_dict[building_info_start_date_key] = Utilities::std_to_godot_string(building.get_start_date().to_string());
			building_dict[building_info_end_date_key] = Utilities::std_to_godot_string(building.get_end_date().to_string());
			building_dict[building_info_expansion_progress_key] = building.get_expansion_progress();

			buildings_array[idx] = std::move(building_dict);
		}
	} else {
		UtilityFunctions::push_error(
			"Failed to resize buildings array to the correct size (", static_cast<int64_t>(buildings.size()),
			") for province ", Utilities::std_to_godot_string(province->get_identifier())
		);
	}

	return buildings_array;
}

Dictionary MenuSingleton::get_province_info_from_index(int32_t index) const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, {});

	static const StringName province_info_province_key = "province";
	static const StringName province_info_state_key = "state";
	static const StringName province_info_slave_status_key = "slave_status";
	static const StringName province_info_colony_status_key = "colony_status";
	static const StringName province_info_terrain_type_key = "terrain_type";
	static const StringName province_info_life_rating_key = "life_rating";
	static const StringName province_info_controller_key = "controller";
	static const StringName province_info_rgo_name_key = "rgo_name";
	static const StringName province_info_rgo_icon_key = "rgo_icon";
	static const StringName province_info_rgo_total_employees_key = "rgo_total_employees";
	static const StringName province_info_rgo_employment_percentage_key = "rgo_employment_percentage";
	static const StringName province_info_rgo_output_quantity_yesterday_key = "rgo_output_quantity_yesterday";
	static const StringName province_info_rgo_revenue_yesterday_key = "rgo_revenue_yesterday";
	static const StringName province_info_crime_name_key = "crime_name";
	static const StringName province_info_crime_icon_key = "crime_icon";
	static const StringName province_info_total_population_key = "total_population";
	static const StringName province_info_pop_types_key = "pop_types";
	static const StringName province_info_pop_ideologies_key = "pop_ideologies";
	static const StringName province_info_pop_cultures_key = "pop_cultures";
	static const StringName province_info_cores_key = "cores";
	static const StringName province_info_buildings_key = "buildings";

	ProvinceInstance const* province = instance_manager->get_map_instance().get_province_instance_by_index(index);
	if (province == nullptr) {
		return {};
	}
	Dictionary ret;

	ret[province_info_province_key] = Utilities::std_to_godot_string(province->get_identifier());

	State const* state = province->get_state();
	if (state != nullptr) {
		ret[province_info_state_key] = get_state_name(*state);
	}

	ret[province_info_slave_status_key] = province->get_slave();

	ret[province_info_colony_status_key] = static_cast<int32_t>(province->get_colony_status());

	TerrainType const* terrain_type = province->get_terrain_type();
	if (terrain_type != nullptr) {
		ret[province_info_terrain_type_key] = Utilities::std_to_godot_string(terrain_type->get_identifier());
	}

	ret[province_info_life_rating_key] = province->get_life_rating();

	CountryInstance const* controller = province->get_controller();
	if (controller != nullptr) {
		ret[province_info_controller_key] = Utilities::std_to_godot_string(controller->get_identifier());
	}

	ResourceGatheringOperation const& rgo = province->get_rgo();
	ret[province_info_rgo_output_quantity_yesterday_key] = rgo.get_output_quantity_yesterday().to_float();
	ret[province_info_rgo_revenue_yesterday_key] = rgo.get_revenue_yesterday().to_float();
	ret[province_info_rgo_total_employees_key] = rgo.get_total_employees_count_cache();
	const Pop::pop_size_t max_employee_count = rgo.get_max_employee_count_cache();
	if (max_employee_count == 0) {
		ret[province_info_rgo_employment_percentage_key] = 100.0f;
	} else {
		ret[province_info_rgo_employment_percentage_key] = (rgo.get_total_employees_count_cache() * fixed_point_t::_100() / max_employee_count).to_float_rounded();
	}

	GoodDefinition const* const rgo_good = province->get_rgo_good();
	if (rgo_good != nullptr) {
		ret[province_info_rgo_name_key] = Utilities::std_to_godot_string(rgo_good->get_identifier());
		ret[province_info_rgo_icon_key] = static_cast<int32_t>(rgo_good->get_index());
	}

	Crime const* crime = province->get_crime();
	if (crime != nullptr) {
		ret[province_info_crime_name_key] = Utilities::std_to_godot_string(crime->get_identifier());
		ret[province_info_crime_icon_key] = static_cast<int32_t>(crime->get_icon());
	}

	ret[province_info_total_population_key] = province->get_total_population();

	GFXPieChartTexture::godot_pie_chart_data_t pop_types =
		GFXPieChartTexture::distribution_to_slices_array(province->get_pop_type_distribution());
	if (!pop_types.is_empty()) {
		ret[province_info_pop_types_key] = std::move(pop_types);
	}

	GFXPieChartTexture::godot_pie_chart_data_t ideologies =
		GFXPieChartTexture::distribution_to_slices_array(province->get_ideology_distribution());
	if (!ideologies.is_empty()) {
		ret[province_info_pop_ideologies_key] = std::move(ideologies);
	}

	GFXPieChartTexture::godot_pie_chart_data_t cultures =
		GFXPieChartTexture::distribution_to_slices_array(province->get_culture_distribution());
	if (!cultures.is_empty()) {
		ret[province_info_pop_cultures_key] = std::move(cultures);
	}

	ordered_set<CountryInstance*> const& cores = province->get_cores();
	if (!cores.empty()) {
		PackedStringArray cores_array;
		if (cores_array.resize(cores.size()) == OK) {
			for (size_t idx = 0; idx < cores.size(); ++idx) {
				cores_array[idx] = Utilities::std_to_godot_string(cores.data()[idx]->get_identifier());
			}
			ret[province_info_cores_key] = std::move(cores_array);
		} else {
			UtilityFunctions::push_error(
				"Failed to resize cores array to the correct size (", static_cast<int64_t>(cores.size()), ") for province ",
				Utilities::std_to_godot_string(province->get_identifier())
			);
		}
	}

	TypedArray<Dictionary> building_dict_array = _make_buildings_dict_array(province);
	if (!building_dict_array.is_empty()) {
		ret[province_info_buildings_key] = std::move(building_dict_array);
	}

	return ret;
}

int32_t MenuSingleton::get_province_building_count() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, 0);

	return game_singleton->get_definition_manager().get_economy_manager().get_building_type_manager()
		.get_province_building_types().size();
}

String MenuSingleton::get_province_building_identifier(int32_t building_index) const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});

	std::vector<BuildingType const*> const& province_building_types = game_singleton->get_definition_manager()
		.get_economy_manager().get_building_type_manager().get_province_building_types();
	ERR_FAIL_COND_V_MSG(
		building_index < 0 || building_index >= province_building_types.size(), {},
		vformat("Invalid province building index: %d", building_index)
	);
	return Utilities::std_to_godot_string(province_building_types[building_index]->get_identifier());
}

Error MenuSingleton::expand_selected_province_building(int32_t building_index) {
	GameSingleton* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, FAILED);
	InstanceManager* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, FAILED);

	ERR_FAIL_COND_V_MSG(
		!instance_manager->expand_selected_province_building(building_index), FAILED,
		vformat("Failed to expand the currently selected province's building index %d", building_index)
	);
	return OK;
}

int32_t MenuSingleton::get_slave_pop_icon_index() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, 0);

	const PopType::sprite_t sprite = game_singleton->get_definition_manager().get_pop_manager().get_slave_sprite();
	ERR_FAIL_COND_V_MSG(sprite <= 0, 0, "Slave sprite unset!");
	return sprite;
}

int32_t MenuSingleton::get_administrative_pop_icon_index() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, 0);

	const PopType::sprite_t sprite = game_singleton->get_definition_manager().get_pop_manager().get_administrative_sprite();
	ERR_FAIL_COND_V_MSG(sprite <= 0, 0, "Administrative sprite unset!");
	return sprite;
}

int32_t MenuSingleton::get_rgo_owner_pop_icon_index() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, 0);

	const PopType::sprite_t sprite = game_singleton->get_definition_manager().get_economy_manager().get_production_type_manager().get_rgo_owner_sprite();
	ERR_FAIL_COND_V_MSG(sprite <= 0, 0, "RGO owner sprite unset!");
	return sprite;
}

/* TOPBAR */

Dictionary MenuSingleton::get_topbar_info() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});

	CountryInstance const* country = game_singleton->get_viewed_country();
	if (country == nullptr) {
		return {};
	}

	Dictionary ret;

	// Country / Ranking
	static const StringName country_key = "country";
	static const StringName country_status_key = "country_status";
	static const StringName total_rank_key = "total_rank";

	ret[country_key] = Utilities::std_to_godot_string(country->get_identifier());
	ret[country_status_key] = static_cast<int32_t>(country->get_country_status());
	ret[total_rank_key] = static_cast<uint64_t>(country->get_total_rank());

	static const StringName prestige_key = "prestige";
	static const StringName prestige_rank_key = "prestige_rank";
	static const StringName prestige_tooltip_key = "prestige_tooltip";

	ret[prestige_key] = country->get_prestige().to_int32_t();
	ret[prestige_rank_key] = static_cast<uint64_t>(country->get_prestige_rank());
	ret[prestige_tooltip_key] = String {}; // TODO - list prestige sources (e.g. power status)

	static const StringName industrial_power_key = "industrial_power";
	static const StringName industrial_rank_key = "industrial_rank";
	static const StringName industrial_power_tooltip_key = "industrial_power_tooltip";

	ret[industrial_power_key] = country->get_industrial_power().to_int32_t();
	ret[industrial_rank_key] = static_cast<uint64_t>(country->get_industrial_rank());
	{
		String industrial_power_tooltip;

		// Pair: State name / Power
		std::vector<std::pair<String, fixed_point_t>> industrial_power_states;
		for (auto const& [state, power] : country->get_industrial_power_from_states()) {
			industrial_power_states.emplace_back(get_state_name(*state), power);
		}
		std::sort(
			industrial_power_states.begin(), industrial_power_states.end(),
			[](auto const& a, auto const& b) -> bool {
				// Sort by greatest power, then by state name alphabetically
				return a.second != b.second ? a.second > b.second : a.first < b.first;
			}
		);
		for (auto const& [state_name, power] : industrial_power_states) {
			industrial_power_tooltip += "\n" + state_name + ": " + GUILabel::get_colour_marker() + "Y"
				+ GUINode::float_to_string_dp(power, 3) + GUILabel::get_colour_marker() + "!";
		}

		// Tuple: Country identifier / Country name / Power
		std::vector<std::tuple<String, String, fixed_point_t>> industrial_power_from_investments;
		for (auto const& [country, power] : country->get_industrial_power_from_investments()) {
			industrial_power_from_investments.emplace_back(
				Utilities::std_to_godot_string(country->get_identifier()), get_country_name(*country), power
			);
		}
		std::sort(
			industrial_power_from_investments.begin(), industrial_power_from_investments.end(),
			[](auto const& a, auto const& b) -> bool {
				// Sort by greatest power, then by country name alphabetically
				return std::get<2>(a) != std::get<2>(b) ? std::get<2>(a) > std::get<2>(b) : std::get<1>(a) < std::get<1>(b);
			}
		);
		for (auto const& [country_identifier, country_name, power] : industrial_power_from_investments) {
			industrial_power_tooltip += "\n" + GUILabel::get_flag_marker() + country_identifier + country_name + ": "
				+ GUILabel::get_colour_marker() + "Y" + GUINode::float_to_string_dp(power, 3) + GUILabel::get_colour_marker()
				+ "!";
		}

		ret[industrial_power_tooltip_key] = std::move(industrial_power_tooltip);
	}

	static const StringName military_power_key = "military_power";
	static const StringName military_rank_key = "military_rank";
	static const StringName military_power_tooltip_key = "military_power_tooltip";

	ret[military_power_key] = country->get_military_power().to_int32_t();
	ret[military_rank_key] = static_cast<uint64_t>(country->get_military_rank());
	{
		String military_power_tooltip;

		for (auto const& [source, power] : {
			std::pair
			{ "MIL_FROM_TROOPS", country->get_military_power_from_land() },
			{ "MIL_FROM_CAP_SHIPS", country->get_military_power_from_sea() },
			{ "MIL_FROM_LEADERS", country->get_military_power_from_leaders() }
		}) {
			if (power != 0) {
				military_power_tooltip += "\n" + tr(source) + ": " + GUILabel::get_colour_marker() + "Y"
					+ GUINode::float_to_string_dp(power, 3) + GUILabel::get_colour_marker() + "!";
			}
		}

		ret[military_power_tooltip_key] = std::move(military_power_tooltip);
	}

	static const StringName colonial_power_available_key = "colonial_power_available";
	static const StringName colonial_power_max_key = "colonial_power_max";
	static const StringName colonial_power_tooltip_key = "colonial_power_tooltip";
	// TODO - colonial power info
	ret[colonial_power_available_key] = 0;
	ret[colonial_power_max_key] = 0;
	ret[colonial_power_tooltip_key] = String {};

	// Production

	// Budget

	// Technology

	// Politics

	// Population

	// Trade

	// Diplomacy

	// Military
	static const StringName regiment_count_key = "regiment_count";
	static const StringName max_supported_regiments_key = "max_supported_regiments";

	ret[regiment_count_key] = static_cast<uint64_t>(country->get_regiment_count());
	ret[max_supported_regiments_key] = static_cast<uint64_t>(country->get_max_supported_regiment_count());

	static const StringName mobilised_key = "mobilised";
	static const StringName mobilisation_regiments_key = "mobilisation_regiments";
	static const StringName mobilisation_impact_key = "mobilisation_impact";
	static const StringName war_policy_key = "war_policy";
	static const StringName mobilisation_max_regiments_key = "mobilisation_max_regiments";

	if (country->is_mobilised()) {
		ret[mobilised_key] = true;
	} else {
		ret[mobilised_key] = false;
		ret[mobilisation_regiments_key] = static_cast<uint64_t>(country->get_mobilisation_potential_regiment_count());
		ret[mobilisation_impact_key] = country->get_mobilisation_impact().to_float();
		ret[war_policy_key] = String {}; // TODO - get ruling party's war policy
		ret[mobilisation_max_regiments_key] = static_cast<uint64_t>(country->get_mobilisation_max_regiment_count());
	}

	return ret;
}

/* TIME/SPEED CONTROL PANEL */

void MenuSingleton::set_paused(bool paused) {
	GameSingleton* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL(game_singleton);
	InstanceManager* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL(instance_manager);

	instance_manager->get_simulation_clock().set_paused(paused);
}

void MenuSingleton::toggle_paused() {
	GameSingleton* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL(game_singleton);
	InstanceManager* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL(instance_manager);

	instance_manager->get_simulation_clock().toggle_paused();
}

bool MenuSingleton::is_paused() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, true);
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, true);

	return instance_manager->get_simulation_clock().is_paused();
}

void MenuSingleton::increase_speed() {
	GameSingleton* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL(game_singleton);
	InstanceManager* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL(instance_manager);

	instance_manager->get_simulation_clock().increase_simulation_speed();
}

void MenuSingleton::decrease_speed() {
	GameSingleton* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL(game_singleton);
	InstanceManager* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL(instance_manager);

	instance_manager->get_simulation_clock().decrease_simulation_speed();
}

int32_t MenuSingleton::get_speed() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, 0);
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, 0);

	return instance_manager->get_simulation_clock().get_simulation_speed();
}

bool MenuSingleton::can_increase_speed() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, false);
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, false);

	return instance_manager->get_simulation_clock().can_increase_simulation_speed();
}

bool MenuSingleton::can_decrease_speed() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, false);
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, false);

	return instance_manager->get_simulation_clock().can_decrease_simulation_speed();
}

String MenuSingleton::get_longform_date() const {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, {});

	return Utilities::date_to_formatted_string(instance_manager->get_today());
}

Error MenuSingleton::generate_search_cache() {
	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, FAILED);
	InstanceManager const* instance_manager = game_singleton->get_instance_manager();
	ERR_FAIL_NULL_V(instance_manager, FAILED);

	search_panel.entry_cache.clear();

	std::vector<ProvinceInstance> const& provinces = instance_manager->get_map_instance().get_province_instances();
	std::vector<StateSet> const& state_sets = instance_manager->get_map_instance().get_state_manager().get_state_sets();
	std::vector<CountryInstance> const& countries = instance_manager->get_country_instance_manager().get_country_instances();

	// TODO - reserve actual state count rather than state set count (maybe use a vector of pointers to all states?)
	search_panel.entry_cache.reserve(provinces.size() + state_sets.size() + countries.size());

	for (ProvinceInstance const& province : provinces) {
		String identifier = Utilities::std_to_godot_string(province.get_identifier());
		String display_name = tr(GUINode::format_province_name(identifier));
		String search_name = display_name.to_lower();

		search_panel.entry_cache.push_back({
			&province, std::move(display_name), std::move(search_name), identifier.to_lower()
		});
	}

	for (StateSet const& state_set : state_sets) {
		for (State const& state : state_set.get_states()) {
			String display_name = get_state_name(state);
			String search_name = display_name.to_lower();

			search_panel.entry_cache.push_back({
				// TODO - include state identifier? (region and/or split?)
				&state, std::move(display_name), std::move(search_name), {}
			});
		}
	}

	for (CountryInstance const& country : countries) {
		// TODO - replace with a proper "exists" check
		if (country.get_capital() != nullptr) {
			String display_name = get_country_name(country);
			String search_name = display_name.to_lower();

			search_panel.entry_cache.push_back({
				&country, std::move(display_name), std::move(search_name),
				Utilities::std_to_godot_string(country.get_identifier()).to_lower()
			});
		}
	}

	std::sort(search_panel.entry_cache.begin(), search_panel.entry_cache.end(), [](auto const& a, auto const& b) -> bool {
		return a.search_name < b.search_name;
	});

	emit_signal(_signal_search_cache_changed());

	return OK;
}

void MenuSingleton::update_search_results(String const& text) {
	// Sanatise input
	const String search_text = text.strip_edges().to_lower();

	search_panel.result_indices.clear();

	if (!search_text.is_empty()) {
		// Search through cache
		for (size_t idx = 0; idx < search_panel.entry_cache.size(); ++idx) {
			search_panel_t::entry_t const& entry = search_panel.entry_cache[idx];

			if (entry.search_name.begins_with(search_text) || entry.identifier == search_text) {
				search_panel.result_indices.push_back(idx);
			}
		}
	}
}

PackedStringArray MenuSingleton::get_search_result_rows(int32_t start, int32_t count) const {
	if (search_panel.result_indices.empty()) {
		return {};
	}

	ERR_FAIL_INDEX_V_MSG(
		start, search_panel.result_indices.size(), {},
		vformat("Invalid start for search panel result rows: %d", start)
	);
	ERR_FAIL_COND_V_MSG(count <= 0, {}, vformat("Invalid count for search panel result rows: %d", count));

	if (start + count > search_panel.result_indices.size()) {
		UtilityFunctions::push_warning(
			"Requested search result rows beyond the end of the result indices (", start, " + ", count, " > ",
			static_cast<int64_t>(search_panel.result_indices.size()), "), limiting to ",
			static_cast<int64_t>(search_panel.result_indices.size() - start), " rows."
		);
		count = search_panel.result_indices.size() - start;
	}

	PackedStringArray results;
	results.resize(count);

	for (size_t idx = 0; idx < count; ++idx) {
		results[idx] = search_panel.entry_cache[search_panel.result_indices[start + idx]].display_name;
	}

	return results;
}

int32_t MenuSingleton::get_search_result_row_count() const {
	return search_panel.result_indices.size();
}

Vector2 MenuSingleton::get_search_result_position(int32_t result_index) const {
	ERR_FAIL_INDEX_V(result_index, search_panel.result_indices.size(), {});

	GameSingleton const* game_singleton = GameSingleton::get_singleton();
	ERR_FAIL_NULL_V(game_singleton, {});

	struct entry_visitor_t {
		fvec2_t operator()(ProvinceInstance const* province) {
			return province->get_province_definition().get_centre();
		}

		fvec2_t operator()(State const* state) {
			return (*this)(state->get_capital());
		}

		fvec2_t operator()(CountryInstance const* country) {
			return (*this)(country->get_capital());
		}
	} entry_visitor;

	return game_singleton->normalise_map_position(
		std::visit(entry_visitor, search_panel.entry_cache[search_panel.result_indices[result_index]].target)
	);
}