Removing of creatures from a list

Author: Kade-github

CreatureEdit.cpp

@@ -178,6 +178,39 @@ void checkImage(ToolState* state)
   }
 }
 
+void removeCreature(void* s)
+{
+  ToolState* state = (ToolState*)s;
+  std::vector<dnd::Creature> creatures = {state->current_creature};
+  if (std::filesystem::exists(
+          "creatures/" + state->current_creature.original_list + ".json")) {
+    creatures = parse_creatures_from_file(
+        "creatures/" + state->current_creature.original_list + ".json");
+
+    auto it = std::remove_if(creatures.begin(), creatures.end(),
+                             [state](const dnd::Creature& c) {
+                               return c.get_name() == state->current_creature.get_name();
+        });
+    if (it != creatures.end()) {
+      creatures.erase(it, creatures.end());
+    }
+    save_creatures_to_file(
+        creatures,
+        "creatures/" + state->current_creature.original_list + ".json");
+  }
+
+  // remove from state list
+  auto it = std::remove_if(state->creatures.begin(), state->creatures.end(),
+                           [state](const dnd::Creature& c) {
+                             return c.get_name() == state->current_creature.get_name() &&
+                                    c.original_list == state->current_creature.original_list;
+      });
+  if (it != state->creatures.end()) {
+    state->creatures.erase(it, state->creatures.end());
+    state->reload_creatures = true;
+  }
+}
+
 void saveCreature(void* s) {
   ToolState* state = (ToolState*)s;
   std::vector<dnd::Creature> creatures = {state->current_creature};
@@ -290,6 +323,10 @@ void displayCreatureEdit(void* s) {
   ImGui::InputText("List Name", listNameBuffer, sizeof(listNameBuffer));
 
   state->current_creature.original_list = std::string(listNameBuffer);
+  if (ImGui::Button("Remove from"))
+  {
+
+  }
   ImGui::SameLine();
   if (ImGui::Button("Save to")) {
     saveCreature(s);

CreatureList.cpp

@@ -60,17 +60,16 @@ void displayCreatures(void* s) {
                 creature.get_max_hit_points(),
                 creature.get_hit_dice().value().c_str());
     if (state->encounter_planner) {
+      if (ImGui::Button("Add")) {
+        state->current_encounter.add_creature(creature);
+      }
+      ImGui::SameLine();
       if (ImGui::Button("Edit")) {
         state->clear_image = true;
         state->current_creature = creature;
         // Focus on the edit tab
         ImGui::SetWindowFocus("Creature Edit");
       }
-      ImGui::SameLine();
-      if (ImGui::Button("Add"))
-      {
-        state->current_encounter.add_creature(creature);
-      }
     } else if (state->encounter_battler) {
       if (ImGui::Button("View")) {
         state->clear_image = true;