feat(commands): added `PetsRemove` and `PetsRemoveAll`

1 files changed, 16 insertions, 0 deletions

diff --git a/lua/pets.lua b/lua/pets.lua
index 6f04798..9ebc71a 100644
--- a/lua/pets.lua
+++ b/lua/pets.lua
@@ -80,6 +80,15 @@ function M.kill_pet(name)
     end
 end
 
+function M.remove_pet(name)
+    if M.pets[name] ~= nil then
+        M.pets[name]:remove()
+        M.pets[name] = nil
+    else
+        utils.warning("Couldn't find a pet named \"" .. name .. '"')
+    end
+end
+
 function M.kill_all()
     for _, pet in pairs(M.pets) do
         pet:kill()
@@ -87,6 +96,13 @@ function M.kill_all()
     M.pets = {}
 end
 
+function M.remove_all()
+    for _, pet in pairs(M.pets) do
+        pet:remove()
+    end
+    M.pets = {}
+end
+
 function M.list()
     local empty = true
     for pet in pairs(M.pets) do