feat(CutsceneLibrary): add setter for the nested message in blocks

GothicKit · Nov 2, 2024 · c2eecf8 · c2eecf8
1 parent aa46ddb
commit c2eecf8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/include/zenkit/CutsceneLibrary.hh b/include/zenkit/CutsceneLibrary.hh
@@ -62,6 +62,7 @@ namespace zenkit {
 		std::variant<std::shared_ptr<CutsceneAtomicBlock>, std::shared_ptr<CutsceneBlock>> block;
 
 		[[nodiscard]] std::shared_ptr<ConversationMessageEvent> get_message() const;
+		void set_message(std::shared_ptr<ConversationMessageEvent> msg);
 
 		ZKAPI void load(ReadArchive& r, GameVersion version) override;
 		ZKAPI void save(WriteArchive& w, GameVersion version) const override;

diff --git a/src/CutsceneLibrary.cc b/src/CutsceneLibrary.cc
@@ -58,6 +58,18 @@ namespace zenkit {
 		return std::get<std::shared_ptr<CutsceneAtomicBlock>>(cur->block)->message;
 	}
 
+	void CutsceneBlock::set_message(std::shared_ptr<ConversationMessageEvent> msg) {
+		auto const* cur = this;
+		while (std::holds_alternative<std::shared_ptr<CutsceneBlock>>(cur->block)) {
+			cur = std::get<std::shared_ptr<CutsceneBlock>>(cur->block).get();
+			if (cur == nullptr) {
+				return;
+			}
+		}
+
+		std::get<std::shared_ptr<CutsceneAtomicBlock>>(cur->block)->message = std::move(msg);
+	}
+
 	void CutsceneBlock::load(ReadArchive& r, GameVersion version) {
 		this->name = r.read_string(); // blockName