#pragma once

#include <cassert>

#include "openvic-simulation/economy/BuildingInstance.hpp"
#include "openvic-simulation/politics/Ideology.hpp"
#include "openvic-simulation/pop/Pop.hpp"
#include "openvic-simulation/country/Country.hpp"

namespace OpenVic {
	struct Map;
	struct Region;
	struct State;
	struct Crime;
	struct Good;
	struct TerrainType;
	struct TerrainTypeMapping;
	struct ProvinceHistoryEntry;

	/* REQUIREMENTS:
	 * MAP-5, MAP-7, MAP-8, MAP-43, MAP-47
	 * POP-22
	 */
	struct Province : HasIdentifierAndColour {
		friend struct Map;

		using index_t = uint16_t;
		using life_rating_t = int8_t;
		using distance_t = fixed_point_t;
		using flags_t = uint16_t;

		enum struct colony_status_t : uint8_t { STATE, PROTECTORATE, COLONY };

		struct adjacency_t {
			friend struct Province;

		private:
			Province const* const province;
			const distance_t PROPERTY(distance);
			flags_t PROPERTY(flags);

			adjacency_t(Province const* province, distance_t distance, flags_t flags);
		};

		struct province_positions_t {
			fvec2_t center;
			fvec2_t text;
			fixed_point_t text_rotation;
			fixed_point_t text_scale;
			std::optional<fvec2_t> unit;
			fvec2_t city;
			fvec2_t factory;
			fvec2_t building_construction;
			fvec2_t military_construction;
			std::map<BuildingType const*, fvec2_t> building_position;
			fixed_point_map_t<BuildingType const*> building_rotation;
		};

		static constexpr index_t NULL_INDEX = 0, MAX_INDEX = std::numeric_limits<index_t>::max();

	private:
		const index_t PROPERTY(index);
		Region* PROPERTY(region);
		State const* PROPERTY_RW(state);
		bool PROPERTY(on_map);
		bool PROPERTY(has_region);
		bool PROPERTY(water);
		/* Terrain type calculated from terrain image */
		TerrainType const* PROPERTY(default_terrain_type);

		std::vector<adjacency_t> PROPERTY(adjacencies);
		province_positions_t PROPERTY(positions);

		TerrainType const* PROPERTY(terrain_type);
		life_rating_t PROPERTY(life_rating);
		colony_status_t PROPERTY(colony_status);
		Country const* PROPERTY(owner);
		Country const* PROPERTY(controller);
		std::vector<Country const*> PROPERTY(cores);
		bool PROPERTY(slave);
		Crime const* PROPERTY_RW(crime);
		// TODO - change this into a factory-like structure
		Good const* PROPERTY(rgo);
		IdentifierRegistry<BuildingInstance> IDENTIFIER_REGISTRY(building);

		std::vector<Pop> PROPERTY(pops);
		Pop::pop_size_t PROPERTY(total_population);
		fixed_point_map_t<PopType const*> PROPERTY(pop_type_distribution);
		fixed_point_map_t<Ideology const*> PROPERTY(ideology_distribution);
		fixed_point_map_t<Culture const*> PROPERTY(culture_distribution);
		fixed_point_map_t<Religion const*> PROPERTY(religion_distribution);

		fvec2_t get_unit_position() const;

		Province(std::string_view new_identifier, colour_t new_colour, index_t new_index);

	public:
		Province(Province&&) = default;

		std::string to_string() const;

		bool load_positions(BuildingTypeManager const& building_type_manager, ast::NodeCPtr root);

		bool expand_building(std::string_view building_type_identifier);

		bool add_pop(Pop&& pop);
		bool add_pop_vec(std::vector<Pop> const& pop_vec);
		size_t get_pop_count() const;
		void update_pops();

		void update_state(Date today);
		void tick(Date today);

		bool is_adjacent_to(Province const* province) const;
		bool add_adjacency(Province const* province, distance_t distance, flags_t flags);

		distance_t calculate_distance_to(Province const* province) const;

		bool reset(BuildingTypeManager const& building_type_manager);
		bool apply_history_to_province(ProvinceHistoryEntry const* entry);
	};
}