From df3597f7dc0f379bda865e3c9dd6303fa6e4c959 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Mon, 17 Apr 2023 12:13:39 +0200
Subject: [PATCH] fix(util): clear cache when leader changes

---
 lua/which-key/util.lua | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/lua/which-key/util.lua b/lua/which-key/util.lua
index 8fbe423..f736928 100644
--- a/lua/which-key/util.lua
+++ b/lua/which-key/util.lua
@@ -6,6 +6,17 @@ local strsub = string.sub
 local cache = {}
 ---@type table<string,string>
 local tcache = {}
+local cache_leaders = ""
+
+function M.check_cache()
+  ---@type string
+  local leaders = vim.g.mapleader .. ":" .. vim.g.maplocalleader
+  if leaders ~= cache_leaders then
+    cache = {}
+    tcache = {}
+    cache_leaders = leaders
+  end
+end
 
 function M.count(tab)
   local ret = 0
@@ -27,6 +38,7 @@ function M.is_empty(tab)
 end
 
 function M.t(str)
+  M.check_cache()
   if not tcache[str] then
     -- https://github.com/neovim/neovim/issues/17369
     tcache[str] = vim.api.nvim_replace_termcodes(str, false, true, true):gsub("\128\254X", "\128")
@@ -63,6 +75,7 @@ local Tokens = {
 }
 ---@return KeyCodes
 function M.parse_keys(keystr)
+  M.check_cache()
   if cache[keystr] then
     return cache[keystr]
   end