diff options
| -rw-r--r-- | lua/nvim-treesitter/caching.lua | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/lua/nvim-treesitter/caching.lua b/lua/nvim-treesitter/caching.lua index 810d3a408..7733202ed 100644 --- a/lua/nvim-treesitter/caching.lua +++ b/lua/nvim-treesitter/caching.lua @@ -8,7 +8,7 @@ local M = {} function M.create_buffer_cache() local cache = {} - ---@type table<string, table<string, any>> + ---@type table<integer, table<string, any>> local items = setmetatable({}, { __index = function(tbl, key) rawset(tbl, key, {}) @@ -16,7 +16,7 @@ function M.create_buffer_cache() end, }) - ---@type table<integer|string, boolean> + ---@type table<integer, boolean> local loaded_buffers = {} ---@param type_name string @@ -30,21 +30,21 @@ function M.create_buffer_cache() api.nvim_buf_attach(bufnr, false, { on_detach = function() cache.clear_buffer(bufnr) - loaded_buffers[tostring(bufnr)] = nil + loaded_buffers[bufnr] = nil return true end, on_reload = function() end, -- this is needed to prevent on_detach being called on buffer reload }) end - items[tostring(bufnr)][type_name] = value + items[bufnr][type_name] = value end ---@param type_name string ---@param bufnr integer ---@return any function cache.get(type_name, bufnr) - return items[tostring(bufnr)][type_name] + return items[bufnr][type_name] end ---@param type_name string @@ -57,12 +57,12 @@ function M.create_buffer_cache() ---@param type_name string ---@param bufnr integer function cache.remove(type_name, bufnr) - items[tostring(bufnr)][type_name] = nil + items[bufnr][type_name] = nil end ---@param bufnr integer function cache.clear_buffer(bufnr) - items[tostring(bufnr)] = nil + items[bufnr] = nil end return cache |