#pragma once

#include "openvic-simulation/misc/Modifier.hpp"
#include "openvic-simulation/types/Date.hpp"
#include "openvic-simulation/military/Unit.hpp"
#include "openvic-simulation/economy/BuildingType.hpp"
#include <cstdint>

namespace OpenVic {
	struct TechnologyFolder : HasIdentifier {
		friend struct TechnologyManager;

	private:
		TechnologyFolder(std::string_view new_identifier);

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

	struct TechnologyArea : HasIdentifier {
		friend struct TechnologyManager;

	private:
		TechnologyFolder const& PROPERTY(folder);

		TechnologyArea(std::string_view new_identifier, TechnologyFolder const& new_folder);

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

	struct Technology : Modifier {
		friend struct TechnologyManager;

		using unit_set_t = std::unordered_set<Unit const*>;
		using building_set_t = std::unordered_set<BuildingType const*>;

	private:
		TechnologyArea const& PROPERTY(area);
		const Date::year_t PROPERTY(year);
		const fixed_point_t PROPERTY(cost);
		const bool PROPERTY(unciv_military);
		const uint8_t PROPERTY(unit);
		const unit_set_t PROPERTY(activated_buildings);
		const building_set_t PROPERTY(activated_units);

		//TODO: implement rules/modifiers and ai_chance

		Technology(
			std::string_view new_identifier, TechnologyArea const& new_area, Date::year_t new_year, fixed_point_t new_cost,
			bool new_unciv_military, uint8_t new_unit, unit_set_t&& new_activated_units,
			building_set_t&& new_activated_buildings, ModifierValue&& new_values
		);

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

	struct TechnologySchool : Modifier {
		friend struct TechnologyManager;

	private:
		TechnologySchool(std::string_view new_identifier, ModifierValue&& new_values);
	};

	struct TechnologyManager {
		IdentifierRegistry<TechnologyFolder> technology_folders;
		IdentifierRegistry<TechnologyArea> technology_areas;
		IdentifierRegistry<Technology> technologies;
		IdentifierRegistry<TechnologySchool> technology_schools;

	public:
		TechnologyManager();

		bool add_technology_folder(std::string_view identifier);
		IDENTIFIER_REGISTRY_ACCESSORS(technology_folder)

		bool add_technology_area(std::string_view identifier, TechnologyFolder const* folder);
		IDENTIFIER_REGISTRY_ACCESSORS(technology_area)

		bool add_technology(
			std::string_view identifier, TechnologyArea const* area, Date::year_t year, fixed_point_t cost,
			bool unciv_military, uint8_t unit, Technology::unit_set_t&& activated_units,
			Technology::building_set_t&& activated_buildings, ModifierValue&& values);
		IDENTIFIER_REGISTRY_ACCESSORS_CUSTOM_PLURAL(technology, technologies)

		bool add_technology_school(std::string_view identifier, ModifierValue&& values);
		IDENTIFIER_REGISTRY_ACCESSORS(technology_school);

		bool load_technology_file_areas(ast::NodeCPtr root); // common/technology.txt
		bool load_technology_file_schools(ModifierManager const& modifier_manager, ast::NodeCPtr root); // also common/technology.txt
		bool load_technologies_file(
			ModifierManager const& modifier_manager, UnitManager const& unit_manager, BuildingManager const& building_manager,
			ast::NodeCPtr root
		); // technologies/*.txt
		bool generate_modifiers(ModifierManager& modifier_manager);
	};
}