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

2 files changed, 14 insertions, 0 deletions

diff --git a/lua/pets/commands.lua b/lua/pets/commands.lua
index a240fc0..f521710 100644
--- a/lua/pets/commands.lua
+++ b/lua/pets/commands.lua
@@ -38,6 +38,9 @@ end, { nargs = 1, complete = utils.complete_typestyle })
 vim.api.nvim_create_user_command("PetsKillAll", function()
     pets.kill_all()
 end, {})
+vim.api.nvim_create_user_command("PetsRemoveAll", function()
+    pets.remove_all()
+end, {})
 
 vim.api.nvim_create_user_command("PetsKill", function(input)
     pets.kill_pet(input.args)
@@ -45,6 +48,12 @@ end, {
     nargs = 1,
     complete = utils.complete_name,
 })
+vim.api.nvim_create_user_command("PetsRemove", function(input)
+    pets.remove_pet(input.args)
+end, {
+    nargs = 1,
+    complete = utils.complete_name,
+})
 
 vim.api.nvim_create_user_command("PetsList", function()
     pets.list()
diff --git a/lua/pets/pet.lua b/lua/pets/pet.lua
index 2cc6cfc..6540b1d 100644
--- a/lua/pets/pet.lua
+++ b/lua/pets/pet.lua
@@ -55,6 +55,11 @@ function M.Pet:kill()
     end
 end
 
+function M.Pet:remove()
+    self.animation:stop()
+    self.popup:unmount()
+end
+
 function M.Pet:set_paused(paused)
     self.animation:set_state({
         paused = paused,