mirror of
https://github.com/neovim/neovim.git
synced 2026-03-28 03:12:00 +00:00
ce1154048b
refactor(lua): add integer coercion helpers Add vim._tointeger() and vim._ensure_integer(), including optional base support, and switch integer-only tonumber()/assert call sites in the Lua runtime to use them. This also cleans up related integer parsing in LSP, health, loader, URI, tohtml, and Treesitter code. supported by AI
88 lines
2.1 KiB
Lua
88 lines
2.1 KiB
Lua
--- Module for private utility functions
|
|
|
|
--- @alias vim.func.MemoObj { _hash: (fun(...): any), _weak: boolean?, _cache: table<any> }
|
|
|
|
--- @param argc integer?
|
|
--- @return fun(...): any
|
|
local function concat_hash(argc)
|
|
return function(...)
|
|
return table.concat({ ... }, '%%', 1, argc)
|
|
end
|
|
end
|
|
|
|
--- @param idx integer
|
|
--- @return fun(...): any
|
|
local function idx_hash(idx)
|
|
return function(...)
|
|
return select(idx, ...)
|
|
end
|
|
end
|
|
|
|
--- @param hash integer|string|fun(...): any
|
|
--- @return fun(...): any
|
|
local function resolve_hash(hash)
|
|
if type(hash) == 'number' then
|
|
hash = idx_hash(hash)
|
|
elseif type(hash) == 'string' then
|
|
if hash == 'concat' then
|
|
hash = concat_hash()
|
|
else
|
|
local c = hash:match('^concat%-(%d+)')
|
|
if c then
|
|
hash = concat_hash(vim._ensure_integer(c))
|
|
else
|
|
error('invalid value for hash: ' .. hash)
|
|
end
|
|
end
|
|
end
|
|
--- @cast hash -integer
|
|
return hash
|
|
end
|
|
|
|
--- @param weak boolean?
|
|
--- @return table
|
|
local create_cache = function(weak)
|
|
return setmetatable({}, {
|
|
__mode = weak ~= false and 'kv',
|
|
})
|
|
end
|
|
|
|
--- @generic F: function
|
|
--- @param hash integer|string|fun(...): any
|
|
--- @param fn F
|
|
--- @param weak? boolean
|
|
--- @return F
|
|
return function(hash, fn, weak)
|
|
vim.validate('hash', hash, { 'number', 'string', 'function' })
|
|
vim.validate('fn', fn, 'function')
|
|
vim.validate('weak', weak, 'boolean', true)
|
|
|
|
--- @type vim.func.MemoObj
|
|
local obj = {
|
|
_cache = create_cache(weak),
|
|
_hash = resolve_hash(hash),
|
|
_weak = weak,
|
|
--- @param self vim.func.MemoObj
|
|
clear = function(self, ...)
|
|
if select('#', ...) == 0 then
|
|
self._cache = create_cache(self._weak)
|
|
return
|
|
end
|
|
local key = self._hash(...)
|
|
self._cache[key] = nil
|
|
end,
|
|
}
|
|
|
|
return setmetatable(obj, {
|
|
--- @param self vim.func.MemoObj
|
|
__call = function(self, ...)
|
|
local key = self._hash(...)
|
|
local cache = self._cache
|
|
if cache[key] == nil then
|
|
cache[key] = vim.F.pack_len(fn(...))
|
|
end
|
|
return vim.F.unpack_len(cache[key])
|
|
end,
|
|
})
|
|
end
|