fix: plugin actions & spelling

lua/which-key/plugins/init.lua

@@ -66,7 +66,8 @@ function PluginNode:__index(k)
         parent = self,
         desc = item.desc,
         order = i,
-        keymap = item,
+        value = item.value,
+        action = item.action,
       }
       ret[item.key] = child
     end

lua/which-key/plugins/spelling.lua

@@ -29,16 +29,16 @@ function M.expand()
   local suggestions = vim.fn.spellsuggest(word, M.opts.suggestions or 20, bad[2] == "caps" and 1 or 0)
 
   local items = {} ---@type wk.Plugin.item[]
-  local keys = "1234567890abcdefghijklmnopqrstuvwxyz"
+  local keys = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
 
   for i, label in ipairs(suggestions) do
     local key = keys:sub(i, i)
 
     table.insert(items, {
       key = key,
       desc = label,
-      fn = function()
-        vim.cmd('norm! "_ciw' .. label)
+      action = function()
+        vim.cmd("norm! " .. i .. "z=")
       end,
     })
   end

lua/which-key/state.lua

@@ -72,15 +72,20 @@ function M.step(state)
   if node then
     local is_group = Tree.is_group(node)
     local is_nowait = node.keymap and node.keymap.nowait == 1
+    local is_action = node.action ~= nil
     local is_keymap = node.keymap ~= nil
 
-    if is_group and not is_nowait then
+    if is_group and not is_nowait and not is_action then
       return node
     end
   end
 
   state.mode:reattach(node or state.node)
 
+  if node and node.action then
+    return node.action()
+  end
+
   local keys = vim.deepcopy(state.node.path)
   keys[#keys + 1] = key
 

lua/which-key/tree.lua

@@ -9,6 +9,8 @@ local Util = require("which-key.util")
 ---@field keymap? wk.Keymap
 ---@field order? number
 ---@field children? table<string, wk.Node>
+---@field value? string
+---@field action? fun()
 
 ---@class wk.Tree
 ---@field root wk.Node

lua/which-key/types.lua

@@ -34,8 +34,8 @@
 ---@field key string
 ---@field value string
 ---@field desc string
----@field cmd string
----@field highlights table
+---@field highlights? table
+---@field action? fun()
 
 ---@class wk.Plugin.action
 ---@field trigger string