feat(options): added the random options for the PetsNew command

2 files changed, 9 insertions, 2 deletions

diff --git a/lua/pets.lua b/lua/pets.lua
index fd71b2b..7e4d4c2 100644
--- a/lua/pets.lua
+++ b/lua/pets.lua
@@ -6,6 +6,7 @@ M.options = {
     speed_multiplier = 1,
     default_pet = "cat",
     default_style = "brown",
+    random = true,
 }
 
 M.pets = {}
diff --git a/lua/pets/commands.lua b/lua/pets/commands.lua
index 3274aba..ab64fdb 100644
--- a/lua/pets/commands.lua
+++ b/lua/pets/commands.lua
@@ -1,13 +1,19 @@
 local pets = require("pets")
+local wd = debug.getinfo(1).source:sub(2):match("(.*nvim/)") .. "media/"
 
 vim.api.nvim_create_user_command("PetsNew", function(input)
-    pets.create_pet(input.args, pets.options.default_pet, pets.options.default_style)
+    --TODO: validate style and type options
+    local pet, style = pets.options.default_pet, pets.options.default_style
+    if pets.options.random then
+        local styles = require("pets.utils").listdir(wd .. "cat")
+        pet, style = "cat", styles[math.random(#styles)]
+    end
+    pets.create_pet(input.args, pet, style)
 end, { nargs = 1 })
 
 local function autocomplete(_, cmdline)
     local matches = {}
     local words = vim.split(cmdline, " ", { trimempty = true })
-    local wd = debug.getinfo(1).source:sub(2):match("(.*nvim/)") .. "media/"
 
     if not vim.endswith(cmdline, " ") then
         table.remove(words, #words)