#pragma once

#include <string>
#include <string_view>
#include <vector>

#include "openvic-simulation/dataloader/Dataloader.hpp"
#include "openvic-simulation/map/Province.hpp"
#include "openvic-simulation/military/Leader.hpp"
#include "openvic-simulation/military/UnitType.hpp"

namespace OpenVic {
	struct RegimentDeployment {
	private:
		std::string PROPERTY(name);
		RegimentType const* PROPERTY(type);
		Province const* PROPERTY(home);

	public:
		RegimentDeployment(std::string_view new_name, RegimentType const* new_type, Province const* new_home);
	};

	struct ShipDeployment {
	private:
		std::string PROPERTY(name);
		ShipType const* PROPERTY(type);

	public:
		ShipDeployment(std::string_view new_name, ShipType const* new_type);
	};

	struct ArmyDeployment {
	private:
		std::string PROPERTY(name);
		Province const* PROPERTY(location);
		std::vector<RegimentDeployment> PROPERTY(regiments);

	public:
		ArmyDeployment(std::string_view new_name, Province const* new_location, std::vector<RegimentDeployment>&& new_regiments);
	};

	struct NavyDeployment {
	private:
		std::string PROPERTY(name);
		Province const* PROPERTY(location);
		std::vector<ShipDeployment> PROPERTY(ships);

	public:
		NavyDeployment(std::string_view new_name, Province const* new_location, std::vector<ShipDeployment>&& new_ships);
	};

	struct Deployment : HasIdentifier {
		friend std::unique_ptr<Deployment> std::make_unique<Deployment>(
			std::string_view&&, std::vector<OpenVic::ArmyDeployment>&&, std::vector<OpenVic::NavyDeployment>&&, std::vector<OpenVic::Leader>&&
		);

	private:
		std::vector<ArmyDeployment> PROPERTY(armies);
		std::vector<NavyDeployment> PROPERTY(navies);
		std::vector<Leader> PROPERTY(leaders);

		Deployment(
			std::string_view new_path, std::vector<ArmyDeployment>&& new_armies, std::vector<NavyDeployment>&& new_navies,
			std::vector<Leader>&& new_leaders
		);

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

	struct DeploymentManager {
	private:
		IdentifierInstanceRegistry<Deployment> IDENTIFIER_REGISTRY(deployment);
		string_set_t missing_oob_files;

	public:
		bool add_deployment(
			std::string_view path, std::vector<ArmyDeployment>&& armies, std::vector<NavyDeployment>&& navies, std::vector<Leader>&& leaders
		);

		bool load_oob_file(
			GameManager const& game_manager, Dataloader const& dataloader, std::string_view history_path,
			Deployment const*& deployment, bool fail_on_missing
		);
		size_t get_missing_oob_file_count() const;
	};
} // namespace OpenVic