From 8ee3958d1a0958917ecb3f7481e369bb62e8ddb3 Mon Sep 17 00:00:00 2001 From: Giuseppe Gadola Date: Fri, 10 Feb 2023 17:47:42 +0100 Subject: feat(pets): ability to kill pets by name --- lua/pets.lua | 9 ++++++++- lua/pets/commands.lua | 4 ++++ 2 files changed, 12 insertions(+), 1 deletion(-) (limited to 'lua') diff --git a/lua/pets.lua b/lua/pets.lua index 1c09146..363f049 100644 --- a/lua/pets.lua +++ b/lua/pets.lua @@ -27,7 +27,14 @@ function M.create_pet(name, type, style) -- TODO: don't allow duplicate names M.pets[pet.name] = pet end --- function M.kill_pet(name) end +function M.kill_pet(name) + if M.pets[name] ~= nil then + M.pets[name]:kill() + M.pets[name] = nil + else + print("Pet name incorrect") + end +end function M.kill_all() for _, pet in pairs(M.pets) do diff --git a/lua/pets/commands.lua b/lua/pets/commands.lua index 71eb4f4..7bf9d8c 100644 --- a/lua/pets/commands.lua +++ b/lua/pets/commands.lua @@ -8,6 +8,10 @@ vim.api.nvim_create_user_command("PetsKillAll", function() pets.kill_all() end, {}) +vim.api.nvim_create_user_command("PetsKill", function(input) + pets.kill_pet(input.args) +end, { nargs = 1 }) + vim.api.nvim_create_user_command("PetsList", function() pets.list() end, {}) -- cgit v1.3-2-g11bf