1 files changed, 205 insertions, 0 deletions

@@ -0,0 +1,205 @@
+#include "container.h"
+
+namespace com::fourisland::lingo2_archipelago {
+
+uint64_t Container::FindOrAddMap(std::string map_name) {
+  auto it = map_id_by_name_.find(map_name);
+
+  if (it == map_id_by_name_.end()) {
+    uint64_t new_id = all_objects_.maps_size();
+    Map* map = all_objects_.add_maps();
+    map->set_id(new_id);
+    map->set_name(map_name);
+
+    map_id_by_name_[map_name] = new_id;
+
+    return new_id;
+  } else {
+    return it->second;
+  }
+}
+
+uint64_t Container::FindOrAddRoom(std::optional<std::string> map_name,
+                                  std::string room_name,
+                                  std::optional<std::string> map_fallback) {
+  if (!map_name) {
+    if (!map_fallback) {
+      std::cout << "No map name provided for " << room_name << std::endl;
+      map_name = "global";
+    } else {
+      map_name = map_fallback;
+    }
+  }
+
+  auto& map_container = room_id_by_map_room_names_[*map_name];
+  auto it = map_container.find(room_name);
+  if (it == map_container.end()) {
+    uint64_t new_id = all_objects_.rooms_size();
+    Room* room = all_objects_.add_rooms();
+    room->set_id(new_id);
+    room->set_map_id(FindOrAddMap(*map_name));
+    room->set_name(room_name);
+
+    map_container[room_name] = new_id;
+
+    return new_id;
+  } else {
+    return it->second;
+  }
+}
+
+uint64_t Container::FindOrAddPainting(
+    std::optional<std::string> map_name, std::optional<std::string> room_name,
+    std::string painting_name, std::optional<std::string> map_fallback,
+    std::optional<std::string> room_fallback) {
+  if (!map_name) {
+    if (!map_fallback) {
+      std::cout << "No map name provided for " << painting_name << std::endl;
+      map_name = "global";
+    } else {
+      map_name = map_fallback;
+    }
+  }
+
+  if (!room_name) {
+    if (!room_fallback) {
+      std::cout << "No room name provided for " << painting_name << std::endl;
+      room_name = "global";
+    } else {
+      room_name = room_fallback;
+    }
+  }
+
+  auto& room_container =
+      painting_id_by_map_room_painting_names_[*map_name][*room_name];
+  auto it = room_container.find(painting_name);
+  if (it == room_container.end()) {
+    uint64_t new_id = all_objects_.paintings_size();
+    Painting* painting = all_objects_.add_paintings();
+    painting->set_id(new_id);
+    painting->set_room_id(FindOrAddRoom(map_name, *room_name, std::nullopt));
+    painting->set_name(painting_name);
+
+    room_container[painting_name] = new_id;
+
+    return new_id;
+  } else {
+    return it->second;
+  }
+}
+
+uint64_t Container::FindOrAddPort(std::optional<std::string> map_name,
+                                  std::optional<std::string> room_name,
+                                  std::string port_name,
+                                  std::optional<std::string> map_fallback,
+                                  std::optional<std::string> room_fallback) {
+  if (!map_name) {
+    if (!map_fallback) {
+      std::cout << "No map name provided for " << port_name << std::endl;
+      map_name = "global";
+    } else {
+      map_name = map_fallback;
+    }
+  }
+
+  if (!room_name) {
+    if (!room_fallback) {
+      std::cout << "No room name provided for " << port_name << std::endl;
+      room_name = "global";
+    } else {
+      room_name = room_fallback;
+    }
+  }
+
+  auto& room_container = port_id_by_map_room_port_names_[*map_name][*room_name];
+  auto it = room_container.find(port_name);
+  if (it == room_container.end()) {
+    uint64_t new_id = all_objects_.ports_size();
+    Port* port = all_objects_.add_ports();
+    port->set_id(new_id);
+    port->set_room_id(FindOrAddRoom(map_name, *room_name, std::nullopt));
+    port->set_name(port_name);
+
+    room_container[port_name] = new_id;
+
+    return new_id;
+  } else {
+    return it->second;
+  }
+}
+
+uint64_t Container::FindOrAddPanel(std::optional<std::string> map_name,
+                                   std::optional<std::string> room_name,
+                                   std::string panel_name,
+                                   std::optional<std::string> map_fallback,
+                                   std::optional<std::string> room_fallback) {
+  if (!map_name) {
+    if (!map_fallback) {
+      std::cout << "No map name provided for " << panel_name << std::endl;
+      map_name = "global";
+    } else {
+      map_name = map_fallback;
+    }
+  }
+
+  if (!room_name) {
+    if (!room_fallback) {
+      std::cout << "No room name provided for " << panel_name << std::endl;
+      room_name = "global";
+    } else {
+      room_name = room_fallback;
+    }
+  }
+
+  auto& room_container =
+      panel_id_by_map_room_panel_names_[*map_name][*room_name];
+  auto it = room_container.find(panel_name);
+  if (it == room_container.end()) {
+    uint64_t new_id = all_objects_.panels_size();
+    Panel* panel = all_objects_.add_panels();
+    panel->set_id(new_id);
+    panel->set_room_id(FindOrAddRoom(map_name, *room_name, std::nullopt));
+    panel->set_name(panel_name);
+
+    room_container[panel_name] = new_id;
+
+    return new_id;
+  } else {
+    return it->second;
+  }
+}
+
+uint64_t Container::FindOrAddDoor(std::optional<std::string> map_name,
+                                  std::string door_name,
+                                  std::optional<std::string> map_fallback) {
+  if (!map_name) {
+    if (!map_fallback) {
+      std::cout << "No map name provided for " << door_name << std::endl;
+      map_name = "global";
+    } else {
+      map_name = map_fallback;
+    }
+  }
+
+  auto& map_container = door_id_by_map_door_names_[*map_name];
+  auto it = map_container.find(door_name);
+  if (it == map_container.end()) {
+    uint64_t new_id = all_objects_.doors_size();
+    Door* door = all_objects_.add_doors();
+    door->set_id(new_id);
+    door->set_map_id(FindOrAddMap(*map_name));
+    door->set_name(door_name);
+
+    map_container[door_name] = new_id;
+
+    return new_id;
+  } else {
+    return it->second;
+  }
+}
+
+void Container::AddConnection(const Connection& connection) {
+  *all_objects_.add_connections() = connection;
+}
+
+}  // namespace com::fourisland::lingo2_archipelago