fix(config): load early when executing WhichKey command. Fixes #912

Author: folke

lua/which-key/config.lua

@@ -295,15 +295,16 @@ function M.setup(opts)
 
     M.loaded = true
   end
-  load = vim.schedule_wrap(load)
+  local _load = vim.schedule_wrap(load)
 
   if vim.v.vim_did_enter == 1 then
-    load()
+    _load()
   else
-    vim.api.nvim_create_autocmd("VimEnter", { once = true, callback = load })
+    vim.api.nvim_create_autocmd("VimEnter", { once = true, callback = _load })
   end
 
   vim.api.nvim_create_user_command("WhichKey", function(cmd)
+    load()
     local mode, keys = cmd.args:match("^([nixsotc]?)%s*(.*)$")
     if not mode then
       return require("which-key.util").error("Usage: WhichKey [mode] [keys]")