From daa2f0bcc9623b2524c877bcd66b76de55dd5773 Mon Sep 17 00:00:00 2001
From: Amatsugu <khamraj@kaisei.app>
Date: Tue, 25 Jun 2024 09:30:24 -0400
Subject: [PATCH] Update buildings_database.rs

---
 game/buildings/src/buildings_database.rs | 35 +++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/game/buildings/src/buildings_database.rs b/game/buildings/src/buildings_database.rs
index 2f584b2..cbf5c0b 100644
--- a/game/buildings/src/buildings_database.rs
+++ b/game/buildings/src/buildings_database.rs
@@ -56,10 +56,43 @@ impl BuildingChunk {
 pub struct BuildingEntry {
 	pub coord: HexCoord,
 	pub entity: Entity,
+	pub is_main: bool,
+	pub main_entity: Option<Entity>,
+	pub has_children: bool,
+	pub child_entities: Option<Vec<Entity>>,
 }
 
 impl BuildingEntry {
 	pub fn new(coord: HexCoord, entity: Entity) -> Self {
-		return BuildingEntry { coord, entity };
+		return BuildingEntry {
+			coord,
+			entity,
+			child_entities: None,
+			has_children: false,
+			main_entity: None,
+			is_main: true,
+		};
+	}
+
+	pub fn new_with_children(coord: HexCoord, entity: Entity, children: Vec<Entity>) -> BuildingEntry {
+		return BuildingEntry {
+			coord,
+			entity,
+			child_entities: Some(children),
+			has_children: true,
+			main_entity: None,
+			is_main: true,
+		};
+	}
+
+	pub fn new_with_parent(coord: HexCoord, entity: Entity, main: Entity) -> BuildingEntry {
+		return BuildingEntry {
+			coord,
+			entity,
+			child_entities: None,
+			has_children: false,
+			main_entity: Some(main),
+			is_main: false,
+		};
 	}
 }