Update lua/hawtkeys/init.lua

Co-authored-by: Will Hopkins <willothyh@gmail.com>
tris203 · Dec 30, 2023 · b6bd79c · b6bd79c
1 parent eb98828
commit b6bd79c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lua/hawtkeys/init.lua b/lua/hawtkeys/init.lua
@@ -121,9 +121,10 @@ function M.setup(config)
     })
 
     vim.api.nvim_create_user_command("Hawtkeys", function(args)
-        if args.fargs[#args.fargs]:lower() == "all" then
+        local cmd = args.fargs[1] and args.fargs[1]:gsub("^%s+", ""):gsub("%s$", ""):lower()
+        if cmd == "all" then
             require("hawtkeys.ui").show_all()
-        elseif args.fargs[#args.fargs]:lower() == "dupes" then
+        elseif cmd == "dupes" then
             require("hawtkeys.ui").show_dupes()
         else
             require("hawtkeys.ui").show()