1 files changed, 40 insertions, 0 deletions

@@ -246,6 +246,46 @@ uint64_t Container::FindOrAddMastery(std::optional<std::string> map_name,
   }
 }
 
+uint64_t Container::FindOrAddKeyholder(
+    std::optional<std::string> map_name, std::optional<std::string> room_name,
+    std::string keyholder_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 " << keyholder_name << std::endl;
+      map_name = "global";
+    } else {
+      map_name = map_fallback;
+    }
+  }
+
+  if (!room_name) {
+    if (!room_fallback) {
+      std::cout << "No room name provided for " << keyholder_name << std::endl;
+      room_name = "global";
+    } else {
+      room_name = room_fallback;
+    }
+  }
+
+  auto& room_container =
+      keyholder_id_by_map_room_keyholder_names_[*map_name][*room_name];
+  auto it = room_container.find(keyholder_name);
+  if (it == room_container.end()) {
+    uint64_t new_id = all_objects_.keyholders_size();
+    Keyholder* keyholder = all_objects_.add_keyholders();
+    keyholder->set_id(new_id);
+    keyholder->set_room_id(FindOrAddRoom(map_name, *room_name, std::nullopt));
+    keyholder->set_name(keyholder_name);
+
+    room_container[keyholder_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) {