Diplomacy History Loading

5 files changed, 479 insertions, 18 deletions

diff --git a/src/openvic-simulation/dataloader/Dataloader.cpp b/src/openvic-simulation/dataloader/Dataloader.cpp
index 330005f..aefb6fc 100644
--- a/src/openvic-simulation/dataloader/Dataloader.cpp
+++ b/src/openvic-simulation/dataloader/Dataloader.cpp
@@ -639,6 +639,22 @@ bool Dataloader::_load_history(GameManager& game_manager, bool unused_history_fi
 	);
 	game_manager.get_history_manager().get_province_manager().lock_province_histories(game_manager.get_map(), false);
 
+	static constexpr std::string_view diplomacy_history_directory = "history/diplomacy";
+	ret &= apply_to_files(
+		lookup_files_in_dir(diplomacy_history_directory, ".txt"),
+		[this, &game_manager](fs::path const& file) -> bool {
+			return game_manager.get_history_manager().get_diplomacy_manager().load_diplomacy_history_file(game_manager, parse_defines(file).get_file_node());
+		}
+	);
+	static constexpr std::string_view war_history_directory = "history/wars";
+	ret &= apply_to_files(
+		lookup_files_in_dir(war_history_directory, ".txt"),
+		[this, &game_manager](fs::path const& file) -> bool {
+			return game_manager.get_history_manager().get_diplomacy_manager().load_war_history_file(game_manager, parse_defines(file).get_file_node());
+		}
+	);
+	game_manager.get_history_manager().get_diplomacy_manager().lock_diplomatic_history();
+
 	return ret;
 }
 
diff --git a/src/openvic-simulation/history/DiplomaticHistory.cpp b/src/openvic-simulation/history/DiplomaticHistory.cpp
new file mode 100644
index 0000000..6f9d73e
--- /dev/null
+++ b/src/openvic-simulation/history/DiplomaticHistory.cpp
@@ -0,0 +1,311 @@
+#include "DiplomaticHistory.hpp"
+
+#include "openvic-simulation/GameManager.hpp"
+#include "openvic-simulation/dataloader/NodeTools.hpp"
+
+using namespace OpenVic;
+using namespace OpenVic::NodeTools;
+
+WarHistory::added_wargoal_t::added_wargoal_t(
+	Date new_added,
+	Country const* new_actor,
+	Country const* new_receiver,
+	WargoalType const* new_wargoal,
+	std::optional<Country const*> new_third_party,
+	std::optional<Province const*> new_target
+) : added { new_added }, actor { new_actor }, receiver { new_receiver }, wargoal { new_wargoal }, third_party { new_third_party }, target { new_target } {}
+
+Country const* WarHistory::added_wargoal_t::get_actor() const {
+	return actor;
+}
+
+Country const* WarHistory::added_wargoal_t::get_receiver() const {
+	return receiver;
+}
+
+WargoalType const* WarHistory::added_wargoal_t::get_wargoal_type() const {
+	return wargoal;
+}
+
+std::optional<Country const*> const& WarHistory::added_wargoal_t::get_third_party() const {
+	return third_party;
+}
+
+std::optional<Province const*> const& WarHistory::added_wargoal_t::get_target() const {
+	return target;
+}
+
+Date WarHistory::added_wargoal_t::get_date_added() const {
+	return added;
+}
+
+WarHistory::war_participant_t::war_participant_t(
+	Country const* new_country,
+	Date new_joined,
+	std::optional<Date> new_exited
+) : country { new_country }, joined { new_joined }, exited { new_exited } {}
+
+Country const* WarHistory::war_participant_t::get_country() const {
+	return country;
+}
+
+Date WarHistory::war_participant_t::get_date_joined() const {
+	return joined;
+}
+
+std::optional<Date> WarHistory::war_participant_t::get_date_exited() const {
+	return exited;
+}
+
+WarHistory::WarHistory(
+	std::string_view new_war_name,
+	std::vector<war_participant_t>&& new_attackers,
+	std::vector<war_participant_t>&& new_defenders,
+	std::vector<added_wargoal_t>&& new_wargoals
+) : war_name { new_war_name }, attackers { std::move(new_attackers) }, defenders { std::move(new_defenders) }, wargoals { std::move(new_wargoals) } {}
+
+std::string_view WarHistory::get_war_name() const {
+	return war_name;
+}
+
+std::vector<WarHistory::war_participant_t> const& WarHistory::get_attackers() const {
+	return attackers;
+}
+
+std::vector<WarHistory::war_participant_t> const& WarHistory::get_defenders() const {
+	return defenders;
+}
+
+std::vector<WarHistory::added_wargoal_t> const& WarHistory::get_wargoals() const {
+	return wargoals;
+}
+
+AllianceHistory::AllianceHistory(
+	Country const* new_first, Country const* new_second,
+	Date new_start, Date new_end
+) : first { new_first }, second { new_second }, start { new_start }, end { new_end } {}
+
+Country const* AllianceHistory::get_first() const {
+	return first;
+}
+
+Country const* AllianceHistory::get_second() const {
+	return second;
+}
+
+SubjectHistory::SubjectHistory(
+	Country const* new_overlord, Country const* new_subject,
+	const type_t new_type, const Date new_start, const Date new_end
+) : overlord { new_overlord }, subject { new_subject }, type { new_type }, start { new_start }, end { new_end } {}
+
+Country const* SubjectHistory::get_overlord() const {
+	return overlord;
+}
+
+Country const* SubjectHistory::get_subject() const {
+	return subject;
+}
+
+const SubjectHistory::type_t SubjectHistory::get_subject_type() const {
+	return type;
+}
+
+void DiplomaticHistoryManager::lock_diplomatic_history() {
+	Logger::info("Locked diplomacy history registry after registering ", alliances.size() + subjects.size() + wars.size(), " items");
+	locked = true;
+}
+
+bool DiplomaticHistoryManager::is_locked() const {
+	return locked;
+}
+
+std::vector<AllianceHistory const*> DiplomaticHistoryManager::get_alliances(Date date) const {
+	std::vector<AllianceHistory const*> ret;
+	for (const auto& alliance : alliances) {
+		if (alliance.start <= date && alliance.end >= date) {
+			ret.push_back(&alliance);
+		}
+	}
+	return ret;
+}
+
+std::vector<SubjectHistory const*> DiplomaticHistoryManager::get_subjects(Date date) const {
+	std::vector<SubjectHistory const*> ret;
+	for (const auto& subject : subjects) {
+		if (subject.start <= date && subject.end >= date) {
+			ret.push_back(&subject);
+		}
+	}
+	return ret;
+}
+
+std::vector<WarHistory const*> DiplomaticHistoryManager::get_wars(Date date) const {
+	std::vector<WarHistory const*> ret;
+	for (const auto& war : wars) {
+		Date start;
+		for (const auto& wargoal : war.wargoals) {
+			if (wargoal.added < start) start = wargoal.added;
+		}
+		if (start >= date) ret.push_back(&war);
+	}
+	return ret;
+}
+
+bool DiplomaticHistoryManager::load_diplomacy_history_file(GameManager& game_manager, ast::NodeCPtr root) {
+	return expect_dictionary_keys(
+		"alliance", ZERO_OR_MORE, [this, &game_manager](ast::NodeCPtr node) -> bool {
+			Country const* first;
+			Country const* second;
+			Date start, end;
+
+			bool ret = expect_dictionary_keys(
+				"first", ONE_EXACTLY, expect_identifier_or_string(game_manager.get_country_manager().expect_country_str(assign_variable_callback_pointer(first))),
+				"second", ONE_EXACTLY, expect_identifier_or_string(game_manager.get_country_manager().expect_country_str(assign_variable_callback_pointer(second))),
+				"start_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(start))),
+				"end_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(end)))
+			)(node);
+			
+			alliances.push_back({ first, second, start, end });
+			return ret;
+		},
+		"vassal", ZERO_OR_MORE, [this, &game_manager](ast::NodeCPtr node) -> bool {
+			Country const* overlord;
+			Country const* subject;
+			Date start, end;
+
+			bool ret = expect_dictionary_keys(
+				"first", ONE_EXACTLY, expect_identifier_or_string(game_manager.get_country_manager().expect_country_str(assign_variable_callback_pointer(overlord))),
+				"second", ONE_EXACTLY, expect_identifier_or_string(game_manager.get_country_manager().expect_country_str(assign_variable_callback_pointer(subject))),
+				"start_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(start))),
+				"end_date", ONE_EXACTLY, expect_identifier_or_string(expect_date_str(assign_variable_callback(end)))
+			)(node);
+
+			subjects.push_back({ overlord, subject, SubjectHistory::type_t::VASSAL, start, end });
+			return ret;
+		},
+		"union", ZERO_OR_MORE, [this, &game_manager](ast::NodeCPtr node) -> bool {
+			Country const* overlord;
+			Country const* subject;
+			Date start, end;
+
+			bool ret = expect_dictionary_keys(
+				"first", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(overlord)),
+				"second", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(subject)),
+				"start_date", ONE_EXACTLY, expect_date(assign_variable_callback(start)),
+				"end_date", ONE_EXACTLY, expect_date(assign_variable_callback(end))
+			)(node);
+
+			subjects.push_back({ overlord, subject, SubjectHistory::type_t::UNION, start, end });
+			return ret;
+		},
+		"substate", ZERO_OR_MORE, [this, &game_manager](ast::NodeCPtr node) -> bool {
+			Country const* overlord;
+			Country const* subject;
+			Date start, end;
+
+			bool ret = expect_dictionary_keys(
+				"first", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(overlord)),
+				"second", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(subject)),
+				"start_date", ONE_EXACTLY, expect_date(assign_variable_callback(start)),
+				"end_date", ONE_EXACTLY, expect_date(assign_variable_callback(end))
+			)(node);
+
+			subjects.push_back({ overlord, subject, SubjectHistory::type_t::SUBSTATE, start, end });
+			return ret;
+		}
+	)(root);
+}
+
+bool DiplomaticHistoryManager::load_war_history_file(GameManager& game_manager, ast::NodeCPtr root) {
+	std::string name = "";
+	std::vector<WarHistory::war_participant_t> attackers;
+	std::vector<WarHistory::war_participant_t> defenders;
+	std::vector<WarHistory::added_wargoal_t> wargoals;
+	Date current_date;
+
+	bool ret = expect_dictionary_keys_and_default(
+		[&game_manager, &attackers, &defenders, &wargoals, &current_date, &name](std::string_view key, ast::NodeCPtr node) -> bool {
+			bool ret = expect_date_str(assign_variable_callback(current_date))(key);
+			ret &= expect_dictionary_keys(
+				"add_attacker", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&attackers, &current_date, &name](Country const& country) -> bool {
+					for (const auto& attacker : attackers) {
+						if (attacker.get_country() == &country) {
+							Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to add attacking country ", attacker.get_country()->get_identifier(), " which is already present!");
+							return false;
+						}
+					}
+					attackers.push_back({ &country, current_date, {} });
+					return true;
+				}),
+				"add_defender", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&defenders, &current_date, &name](Country const& country) -> bool {
+					for (const auto& defender : defenders) {
+						if (defender.get_country() == &country) {
+							Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to add defending country ", defender.get_country()->get_identifier(), " which is already present!");
+							return false;
+						}
+					}
+					defenders.push_back({ &country, current_date, {} });
+					return true;
+				}),
+				"rem_attacker", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&attackers, &current_date, &name](Country const& country) -> bool {
+					WarHistory::war_participant_t* participant_to_remove = nullptr;
+
+					for (auto& attacker : attackers) {
+						if (attacker.country == &country) {
+							participant_to_remove = &attacker;
+							break;
+						}
+					}
+
+					if (participant_to_remove == nullptr) {
+						Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to remove attacking country ",    country.get_identifier(), " which was not present!");
+						return false;
+					}
+
+					participant_to_remove->exited.emplace(current_date);
+					return true;
+				}),
+				"rem_defender", ZERO_OR_MORE, game_manager.get_country_manager().expect_country_identifier([&defenders, &current_date, &name](Country const& country) -> bool {
+					WarHistory::war_participant_t* participant_to_remove = nullptr;
+
+					for (auto& defender : defenders) {
+						if (defender.country == &country) {
+							participant_to_remove = &defender;
+							break;
+						}
+					}
+
+					if (participant_to_remove == nullptr) {
+						Logger::error("In history of war ", name, " at date ", current_date.to_string(), ": Attempted to remove attacking country ",    country.get_identifier(), " which was not present!");
+						return false;
+					}
+
+					participant_to_remove->exited.emplace(current_date);
+					return true;
+				}),
+				"war_goal", ZERO_OR_MORE, [&game_manager, &wargoals, &current_date](ast::NodeCPtr value) -> bool {
+					Country const* actor;
+					Country const* receiver;
+					WargoalType const* type;
+					std::optional<Country const*> third_party;
+					std::optional<Province const*> target;
+
+					bool ret = expect_dictionary_keys(
+						"actor", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(actor)),
+						"receiver", ONE_EXACTLY, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(receiver)),
+						"casus_belli", ONE_EXACTLY, game_manager.get_military_manager().get_wargoal_manager().expect_wargoal_type_identifier(assign_variable_callback_pointer(type)),
+						"country", ZERO_OR_ONE, game_manager.get_country_manager().expect_country_identifier(assign_variable_callback_pointer(*third_party)),
+						"state_province_id", ZERO_OR_ONE, game_manager.get_map().expect_province_identifier(assign_variable_callback_pointer(*target))
+					)(value);
+					wargoals.push_back({ current_date, actor, receiver, type, third_party, target });
+					return ret;
+				}
+			)(node);
+			return ret;
+		},
+		"name", ZERO_OR_ONE, expect_string(assign_variable_callback_string(name))
+	)(root);
+
+	wars.push_back({ name, std::move(attackers), std::move(defenders), std::move(wargoals) });
+	return ret;
+}
\ No newline at end of file
diff --git a/src/openvic-simulation/history/DiplomaticHistory.hpp b/src/openvic-simulation/history/DiplomaticHistory.hpp
new file mode 100644
index 0000000..84c2bd7
--- /dev/null
+++ b/src/openvic-simulation/history/DiplomaticHistory.hpp
@@ -0,0 +1,131 @@
+#pragma once
+
+#include <bitset>
+#include <vector>
+#include <map>
+#include <optional>
+
+#include "openvic-simulation/country/Country.hpp"
+#include "openvic-simulation/military/Wargoal.hpp"
+
+namespace OpenVic {
+	struct DiplomaticHistoryManager;
+
+	struct WarHistory {
+		friend struct DiplomaticHistoryManager;
+
+		struct added_wargoal_t {
+			friend struct DiplomaticHistoryManager;
+
+		private:
+			Date added;
+			Country const* actor;
+			Country const* receiver;
+			WargoalType const* wargoal;
+			std::optional<Country const*> third_party;
+			std::optional<Province const*> target;
+
+			added_wargoal_t(Date new_added, Country const* new_actor, Country const* new_receiver, WargoalType const* new_wargoal, std::optional<Country const*> new_third_party, std::optional<Province const*> new_target);
+
+		public:
+			Date get_date_added() const;
+			Country const* get_actor() const;
+			Country const* get_receiver() const;
+			WargoalType const* get_wargoal_type() const;
+			std::optional<Country const*> const& get_third_party() const;
+			std::optional<Province const*> const& get_target() const;
+		};
+
+		struct war_participant_t {
+			friend struct DiplomaticHistoryManager;
+
+		private:
+			Country const* country;
+			Date joined;
+			std::optional<Date> exited;
+
+			war_participant_t(Country const* new_country, Date new_joined, std::optional<Date> new_exited);
+		
+		public:
+			Country const* get_country() const;
+			Date get_date_joined() const;
+			std::optional<Date> get_date_exited() const;
+		};
+
+	private:
+		std::string war_name; // edge cases where this is empty/undef for some reason, probably need to just generate war names like usual for that.
+		std::vector<war_participant_t> attackers;
+		std::vector<war_participant_t> defenders;
+		std::vector<added_wargoal_t> wargoals;
+
+		WarHistory(std::string_view new_war_name, std::vector<war_participant_t>&& new_attackers, std::vector<war_participant_t>&& new_defenders, std::vector<added_wargoal_t>&& new_wargoals);
+
+	public:
+		std::string_view get_war_name() const;
+		std::vector<war_participant_t> const& get_attackers() const;
+		std::vector<war_participant_t> const& get_defenders() const;
+		std::vector<added_wargoal_t> const& get_wargoals() const;
+	};
+
+	struct AllianceHistory {
+		friend struct DiplomaticHistoryManager;
+
+	private:
+		Country const* first;
+		Country const* second;
+		const Date start;
+		const Date end;
+
+		AllianceHistory(Country const* new_first, Country const* new_second, const Date new_start, const Date new_end);
+
+	public:
+		Country const* get_first() const;
+		Country const* get_second() const;
+	};
+
+	struct SubjectHistory {
+		friend struct DiplomaticHistoryManager;
+
+		enum class type_t {
+			VASSAL,
+			UNION,
+			SUBSTATE
+		};
+
+	private:
+		Country const* overlord;
+		Country const* subject;
+		const type_t type;
+		const Date start;
+		const Date end;
+
+		SubjectHistory(Country const* new_overlord, Country const* new_subject, const type_t new_type, const Date new_start, const Date new_end);
+
+	public:
+		Country const* get_overlord() const;
+		Country const* get_subject() const;
+		const type_t get_subject_type() const; 
+	};
+
+	struct DiplomaticHistoryManager {
+	private:
+		std::vector<AllianceHistory> alliances;
+		std::vector<SubjectHistory> subjects;
+		std::vector<WarHistory> wars;
+		bool locked = false;
+
+	public:
+		DiplomaticHistoryManager() {}
+
+		void lock_diplomatic_history();
+		bool is_locked() const;
+
+		std::vector<AllianceHistory const*> get_alliances(Date date) const;
+		std::vector<SubjectHistory const*> get_subjects(Date date) const;
+		/* Returns all wars that begin before date. NOTE: Some wargoals may be added or countries may join after date, should be checked for by functions that use get_wars() */
+		std::vector<WarHistory const*> get_wars(Date date) const;
+
+		bool load_diplomacy_history_file(GameManager& game_manager, ast::NodeCPtr root);
+		bool load_war_history_file(GameManager& game_manager, ast::NodeCPtr root);
+	};
+} // namespace OpenVic
\ No newline at end of file
diff --git a/src/openvic-simulation/history/HistoryManager.hpp b/src/openvic-simulation/history/HistoryManager.hpp
index 5782e97..ec6d1c5 100644
--- a/src/openvic-simulation/history/HistoryManager.hpp
+++ b/src/openvic-simulation/history/HistoryManager.hpp
@@ -3,6 +3,7 @@
 #include "openvic-simulation/history/Bookmark.hpp"
 #include "openvic-simulation/history/CountryHistory.hpp"
 #include "openvic-simulation/history/ProvinceHistory.hpp"
+#include "openvic-simulation/history/DiplomaticHistory.hpp"
 #include "openvic-simulation/types/IdentifierRegistry.hpp"
 
 namespace OpenVic {
@@ -11,11 +12,13 @@ namespace OpenVic {
 		BookmarkManager bookmark_manager;
 		CountryHistoryManager country_manager;
 		ProvinceHistoryManager province_manager;
+		DiplomaticHistoryManager diplomacy_manager;
 
 	public:
 		REF_GETTERS(bookmark_manager)
 		REF_GETTERS(country_manager)
 		REF_GETTERS(province_manager)
+		REF_GETTERS(diplomacy_manager)
 
 		inline bool load_bookmark_file(ast::NodeCPtr root) {
 			return bookmark_manager.load_bookmark_file(root);
diff --git a/src/openvic-simulation/military/Wargoal.hpp b/src/openvic-simulation/military/Wargoal.hpp
index fc9e3af..dbb8d67 100644
--- a/src/openvic-simulation/military/Wargoal.hpp
+++ b/src/openvic-simulation/military/Wargoal.hpp
@@ -8,24 +8,24 @@ namespace OpenVic {
 	struct WargoalTypeManager;
 
 	enum class peace_options_t : uint32_t {
-		PO_ANNEX =                 0b1000000000000000,
-		PO_DEMAND_STATE =          0b0100000000000000,
-		PO_COLONY =                0b0010000000000000,
-		PO_ADD_TO_SPHERE =         0b0001000000000000,
-		PO_DISARMAMENT =           0b0000100000000000,
-		PO_REMOVE_FORTS =           0b0000100000000000,
-		PO_REMOVE_NAVAL_BASES =           0b0000100000000000,
-		PO_REPARATIONS =           0b0000010000000000,
-		PO_REPAY_DEBT =            0b0000001000000000,
-		PO_REMOVE_PRESTIGE =       0b0000000100000000,
-		PO_MAKE_PUPPET =           0b0000000010000000,
-		PO_RELEASE_PUPPET =        0b0000000001000000,
-		PO_STATUS_QUO =            0b0000000000100000,
-		PO_INSTALL_COMMUNISM =     0b0000000000010000,
-		PO_REMOVE_COMMUNISM =      0b0000000000001000,
-		PO_REMOVE_CORES =          0b0000000000000100, // only usable with ANNEX, DEMAND_STATE, or TRANSFER_PROVINCES
-		PO_TRANSFER_PROVINCES =    0b0000000000000010,
-		PO_CLEAR_UNION_SPHERE =    0b0000000000000001
+		PO_ANNEX = 					0b100000000000000000,
+		PO_DEMAND_STATE = 			0b010000000000000000,
+		PO_COLONY = 				0b001000000000000000,
+		PO_ADD_TO_SPHERE = 			0b000100000000000000,
+		PO_DISARMAMENT = 			0b000010000000000000,
+		PO_REMOVE_FORTS = 			0b000001000000000000,
+		PO_REMOVE_NAVAL_BASES = 	0b000000100000000000,
+		PO_REPARATIONS = 			0b000000010000000000,
+		PO_REPAY_DEBT = 			0b000000001000000000,
+		PO_REMOVE_PRESTIGE = 		0b000000000100000000,
+		PO_MAKE_PUPPET = 			0b000000000010000000,
+		PO_RELEASE_PUPPET = 		0b000000000001000000,
+		PO_STATUS_QUO = 			0b000000000000100000,
+		PO_INSTALL_COMMUNISM = 		0b000000000000010000,
+		PO_REMOVE_COMMUNISM = 		0b000000000000001000,
+		PO_REMOVE_CORES = 			0b000000000000000100, // only usable with ANNEX, DEMAND_STATE, or TRANSFER_PROVINCES
+		PO_TRANSFER_PROVINCES = 	0b000000000000000010,
+		PO_CLEAR_UNION_SPHERE = 	0b000000000000000001
 	};
 	template<> struct enable_bitfield<peace_options_t> : std::true_type{};