perf(tree): small perf optims

Author: folke

lua/which-key/node.lua

@@ -80,7 +80,7 @@ function M:inspect(depth)
 end
 
 function M:count()
-  return #self:children()
+  return not self:can_expand() and vim.tbl_count(self._children) or #self:children()
 end
 
 function M:is_group()
@@ -96,7 +96,7 @@ function M:is_plugin()
 end
 
 function M:can_expand()
-  return self.plugin or self:is_proxy() or (self.mapping and self.mapping.expand)
+  return self.plugin or (self.mapping and (self.mapping.proxy or self.mapping.expand))
 end
 
 ---@return wk.Node[]

lua/which-key/tree.lua

@@ -63,7 +63,7 @@ function M:keep(node)
   if node.hidden or (node.keymap and node.keymap.desc == "which_key_ignore") then
     return false
   end
-  return node:can_expand() or node.keymap or node:is_group() or (node.mapping and not node.group)
+  return node.keymap or (node.mapping and not node.group) or node:is_group()
 end
 
 function M:fix()

lua/which-key/util.lua

@@ -35,7 +35,10 @@ end
 --- Normalizes (and fixes) the lhs of a keymap
 ---@param lhs string
 function M.norm(lhs)
-  M.cache.norm[lhs] = M.cache.norm[lhs] or vim.fn.keytrans(M.t(lhs))
+  if M.cache.norm[lhs] then
+    return M.cache.norm[lhs]
+  end
+  M.cache.norm[lhs] = vim.fn.keytrans(M.t(lhs))
   return M.cache.norm[lhs]
 end