Merge pull request #418 from mrjones2014/mrj/417/which-key-auto-register

fix(extension): Make which_key extension respect auto_register config
mrjones2014 · Oct 23, 2023 · 05af688 · 05af688
2 parents 6ea46c2 + a9efe38
commit 05af688
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/lua/legendary/extensions/which_key.lua b/lua/legendary/extensions/which_key.lua
@@ -29,14 +29,16 @@ return function(opts)
       end
     end
 
-    local original = wk.register
-    local listener = function(whichkey_tbls, whichkey_opts)
-      Log.trace('Preparing to register items from which-key.nvim automatically')
-      util.bind_whichkey(whichkey_tbls, whichkey_opts, opts.do_binding, opts.use_groups)
-      original(whichkey_tbls, whichkey_opts)
-      Log.trace('Successfully registered items from which-key.nvim')
+    if opts.auto_register then
+      local original = wk.register
+      local listener = function(whichkey_tbls, whichkey_opts)
+        Log.trace('Preparing to register items from which-key.nvim automatically')
+        util.bind_whichkey(whichkey_tbls, whichkey_opts, opts.do_binding, opts.use_groups)
+        original(whichkey_tbls, whichkey_opts)
+        Log.trace('Successfully registered items from which-key.nvim')
+      end
+      wk.register = listener
     end
-    wk.register = listener
   else
     Log.warn(
       'which-key.nvim not available. If you are lazy-loading, be sure that which-key.nvim is added to runtime path '