fix: cache for search

tris203 · Dec 18, 2023 · bb133c5 · bb133c5
1 parent 6394c21
commit bb133c5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/lua/hawtkeys/score.lua b/lua/hawtkeys/score.lua
@@ -2,6 +2,7 @@ local config = require("hawtkeys")
 local keyboardLayouts = require("hawtkeys.keyboards")
 local tsSearch = require("hawtkeys.ts")
 local utils = require("hawtkeys.utils")
+local already_used_keys = {}
 
 ---@param key1 string
 ---@param key2 string
@@ -121,8 +122,9 @@ local function process_string(str)
         table.insert(sortedScores, { combo = combo, score = score })
     end
     table.sort(sortedScores, utils.score_sort)
-
-    local already_used_keys = tsSearch.get_all_keymaps()
+    if already_used_keys == nil then
+        already_used_keys = tsSearch.get_all_keymaps()
+    end
 
     local find_mapping = function(maps, lhs)
         for _, value in ipairs(maps) do
@@ -205,6 +207,11 @@ local function scoreTable(str)
     return resultTable
 end
 
+local function reset_already_used_keys()
+    already_used_keys = nil
+end
+
 return {
     ScoreTable = scoreTable,
+    ResetAlreadyUsedKeys = reset_already_used_keys,
 }
diff --git a/lua/hawtkeys/ui.lua b/lua/hawtkeys/ui.lua
@@ -154,6 +154,7 @@ M.hide = function()
     if SearchWin and vim.api.nvim_win_is_valid(SearchWin) then
         vim.api.nvim_win_close(SearchWin, true)
     end
+    Hawtkeys.ResetAlreadyUsedKeys()
     SearchWin = nil
     ResultWin = nil
     vim.api.nvim_command("stopinsert")