mirror of
https://github.com/neovim/neovim.git
synced 2025-10-22 17:11:49 +00:00
refactor(loader): format annotations
This commit is contained in:
@@ -10,8 +10,8 @@ local VERSION = 4
|
|||||||
|
|
||||||
local M = {}
|
local M = {}
|
||||||
|
|
||||||
---@alias vim.loader.CacheHash {mtime: {nsec: integer, sec: integer}, size: integer, type?: string}
|
--- @alias vim.loader.CacheHash {mtime: {nsec: integer, sec: integer}, size: integer, type?: string}
|
||||||
---@alias vim.loader.CacheEntry {hash:vim.loader.CacheHash, chunk:string}
|
--- @alias vim.loader.CacheEntry {hash:vim.loader.CacheHash, chunk:string}
|
||||||
|
|
||||||
--- @class vim.loader.find.Opts
|
--- @class vim.loader.find.Opts
|
||||||
--- @inlinedoc
|
--- @inlinedoc
|
||||||
@@ -45,21 +45,21 @@ local M = {}
|
|||||||
--- The fs_stat of the module path. Won't be returned for `modname="*"`
|
--- The fs_stat of the module path. Won't be returned for `modname="*"`
|
||||||
--- @field stat? uv.fs_stat.result
|
--- @field stat? uv.fs_stat.result
|
||||||
|
|
||||||
---@alias vim.loader.Stats table<string, {total:number, time:number, [string]:number?}?>
|
--- @alias vim.loader.Stats table<string, {total:number, time:number, [string]:number?}?>
|
||||||
|
|
||||||
---@nodoc
|
--- @private
|
||||||
M.path = vim.fn.stdpath('cache') .. '/luac'
|
M.path = vim.fn.stdpath('cache') .. '/luac'
|
||||||
|
|
||||||
---@nodoc
|
--- @private
|
||||||
M.enabled = false
|
M.enabled = false
|
||||||
|
|
||||||
---@type vim.loader.Stats
|
--- @type vim.loader.Stats
|
||||||
local stats = { find = { total = 0, time = 0, not_found = 0 } }
|
local stats = { find = { total = 0, time = 0, not_found = 0 } }
|
||||||
|
|
||||||
--- @type table<string, uv.fs_stat.result>?
|
--- @type table<string, uv.fs_stat.result>?
|
||||||
local fs_stat_cache
|
local fs_stat_cache
|
||||||
|
|
||||||
---@type table<string, table<string,vim.loader.ModuleInfo>>
|
--- @type table<string, table<string,vim.loader.ModuleInfo>>
|
||||||
local indexed = {}
|
local indexed = {}
|
||||||
|
|
||||||
--- @param path string
|
--- @param path string
|
||||||
@@ -113,17 +113,17 @@ local function get_rtp()
|
|||||||
end
|
end
|
||||||
|
|
||||||
--- Returns the cache file name
|
--- Returns the cache file name
|
||||||
---@param name string can be a module name, or a file name
|
--- @param name string can be a module name, or a file name
|
||||||
---@return string file_name
|
--- @return string file_name
|
||||||
local function cache_filename(name)
|
local function cache_filename(name)
|
||||||
local ret = ('%s/%s'):format(M.path, uri_encode(name, 'rfc2396'))
|
local ret = ('%s/%s'):format(M.path, uri_encode(name, 'rfc2396'))
|
||||||
return ret:sub(-4) == '.lua' and (ret .. 'c') or (ret .. '.luac')
|
return ret:sub(-4) == '.lua' and (ret .. 'c') or (ret .. '.luac')
|
||||||
end
|
end
|
||||||
|
|
||||||
--- Saves the cache entry for a given module or file
|
--- Saves the cache entry for a given module or file
|
||||||
---@param cname string cache filename
|
--- @param cname string cache filename
|
||||||
---@param hash vim.loader.CacheHash
|
--- @param hash vim.loader.CacheHash
|
||||||
---@param chunk function
|
--- @param chunk function
|
||||||
local function write_cachefile(cname, hash, chunk)
|
local function write_cachefile(cname, hash, chunk)
|
||||||
local f = assert(uv.fs_open(cname, 'w', 438))
|
local f = assert(uv.fs_open(cname, 'w', 438))
|
||||||
local header = {
|
local header = {
|
||||||
@@ -144,16 +144,16 @@ local function readfile(path, mode)
|
|||||||
local f = uv.fs_open(path, 'r', mode)
|
local f = uv.fs_open(path, 'r', mode)
|
||||||
if f then
|
if f then
|
||||||
local size = assert(uv.fs_fstat(f)).size
|
local size = assert(uv.fs_fstat(f)).size
|
||||||
local data = uv.fs_read(f, size, 0) --[[@as string?]]
|
local data = uv.fs_read(f, size, 0)
|
||||||
uv.fs_close(f)
|
uv.fs_close(f)
|
||||||
return data
|
return data
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
--- Loads the cache entry for a given module or file
|
--- Loads the cache entry for a given module or file
|
||||||
---@param cname string cache filename
|
--- @param cname string cache filename
|
||||||
---@return vim.loader.CacheHash? hash
|
--- @return vim.loader.CacheHash? hash
|
||||||
---@return string? chunk
|
--- @return string? chunk
|
||||||
local function read_cachefile(cname)
|
local function read_cachefile(cname)
|
||||||
local data = readfile(cname, 438)
|
local data = readfile(cname, 438)
|
||||||
if not data then
|
if not data then
|
||||||
@@ -165,7 +165,7 @@ local function read_cachefile(cname)
|
|||||||
return
|
return
|
||||||
end
|
end
|
||||||
|
|
||||||
---@type integer[]|{[0]:integer}
|
--- @type integer[]|{[0]:integer}
|
||||||
local header = vim.split(data:sub(1, zero - 1), ',')
|
local header = vim.split(data:sub(1, zero - 1), ',')
|
||||||
if tonumber(header[1]) ~= VERSION then
|
if tonumber(header[1]) ~= VERSION then
|
||||||
return
|
return
|
||||||
@@ -182,8 +182,8 @@ local function read_cachefile(cname)
|
|||||||
end
|
end
|
||||||
|
|
||||||
--- The `package.loaders` loader for Lua files using the cache.
|
--- The `package.loaders` loader for Lua files using the cache.
|
||||||
---@param modname string module name
|
--- @param modname string module name
|
||||||
---@return string|function
|
--- @return string|function
|
||||||
local function loader_cached(modname)
|
local function loader_cached(modname)
|
||||||
fs_stat_cache = {}
|
fs_stat_cache = {}
|
||||||
local ret = M.find(modname)[1]
|
local ret = M.find(modname)[1]
|
||||||
@@ -201,8 +201,8 @@ end
|
|||||||
local is_win = vim.fn.has('win32') == 1
|
local is_win = vim.fn.has('win32') == 1
|
||||||
|
|
||||||
--- The `package.loaders` loader for libs
|
--- The `package.loaders` loader for libs
|
||||||
---@param modname string module name
|
--- @param modname string module name
|
||||||
---@return string|function
|
--- @return string|function
|
||||||
local function loader_lib_cached(modname)
|
local function loader_lib_cached(modname)
|
||||||
local ret = M.find(modname, { patterns = { is_win and '.dll' or '.so' } })[1]
|
local ret = M.find(modname, { patterns = { is_win and '.dll' or '.so' } })[1]
|
||||||
if not ret then
|
if not ret then
|
||||||
@@ -224,8 +224,8 @@ end
|
|||||||
--- * file size
|
--- * file size
|
||||||
--- * mtime in seconds
|
--- * mtime in seconds
|
||||||
--- * mtime in nanoseconds
|
--- * mtime in nanoseconds
|
||||||
---@param a? vim.loader.CacheHash
|
--- @param a? vim.loader.CacheHash
|
||||||
---@param b? vim.loader.CacheHash
|
--- @param b? vim.loader.CacheHash
|
||||||
local function hash_eq(a, b)
|
local function hash_eq(a, b)
|
||||||
return a
|
return a
|
||||||
and b
|
and b
|
||||||
@@ -236,10 +236,10 @@ end
|
|||||||
|
|
||||||
--- `loadfile` using the cache
|
--- `loadfile` using the cache
|
||||||
--- Note this has the mode and env arguments which is supported by LuaJIT and is 5.1 compatible.
|
--- Note this has the mode and env arguments which is supported by LuaJIT and is 5.1 compatible.
|
||||||
---@param filename? string
|
--- @param filename? string
|
||||||
---@param mode? "b"|"t"|"bt"
|
--- @param mode? "b"|"t"|"bt"
|
||||||
---@param env? table
|
--- @param env? table
|
||||||
---@return function?, string? error_message
|
--- @return function?, string? error_message
|
||||||
local function loadfile_cached(filename, mode, env)
|
local function loadfile_cached(filename, mode, env)
|
||||||
local modpath = normalize(filename)
|
local modpath = normalize(filename)
|
||||||
local stat = fs_stat_cached(modpath)
|
local stat = fs_stat_cached(modpath)
|
||||||
@@ -263,7 +263,7 @@ local function loadfile_cached(filename, mode, env)
|
|||||||
end
|
end
|
||||||
|
|
||||||
--- Return the top-level \`/lua/*` modules for this path
|
--- Return the top-level \`/lua/*` modules for this path
|
||||||
---@param path string path to check for top-level Lua modules
|
--- @param path string path to check for top-level Lua modules
|
||||||
local function lsmod(path)
|
local function lsmod(path)
|
||||||
if not indexed[path] then
|
if not indexed[path] then
|
||||||
indexed[path] = {}
|
indexed[path] = {}
|
||||||
@@ -271,7 +271,7 @@ local function lsmod(path)
|
|||||||
local modpath = path .. '/lua/' .. name
|
local modpath = path .. '/lua/' .. name
|
||||||
-- HACK: type is not always returned due to a bug in luv
|
-- HACK: type is not always returned due to a bug in luv
|
||||||
t = t or fs_stat_cached(modpath).type
|
t = t or fs_stat_cached(modpath).type
|
||||||
---@type string
|
--- @type string
|
||||||
local topname
|
local topname
|
||||||
local ext = name:sub(-4)
|
local ext = name:sub(-4)
|
||||||
if ext == '.lua' or ext == '.dll' then
|
if ext == '.lua' or ext == '.dll' then
|
||||||
@@ -293,9 +293,9 @@ end
|
|||||||
---
|
---
|
||||||
--- @since 0
|
--- @since 0
|
||||||
---
|
---
|
||||||
---@param modname string Module name, or `"*"` to find the top-level modules instead
|
--- @param modname string Module name, or `"*"` to find the top-level modules instead
|
||||||
---@param opts? vim.loader.find.Opts Options for finding a module:
|
--- @param opts? vim.loader.find.Opts Options for finding a module:
|
||||||
---@return vim.loader.ModuleInfo[]
|
--- @return vim.loader.ModuleInfo[]
|
||||||
function M.find(modname, opts)
|
function M.find(modname, opts)
|
||||||
opts = opts or {}
|
opts = opts or {}
|
||||||
|
|
||||||
@@ -321,7 +321,7 @@ function M.find(modname, opts)
|
|||||||
patterns[p] = '/lua/' .. basename .. pattern
|
patterns[p] = '/lua/' .. basename .. pattern
|
||||||
end
|
end
|
||||||
|
|
||||||
---@type vim.loader.ModuleInfo[]
|
--- @type vim.loader.ModuleInfo[]
|
||||||
local results = {}
|
local results = {}
|
||||||
|
|
||||||
-- Only continue if we haven't found anything yet or we want to find all
|
-- Only continue if we haven't found anything yet or we want to find all
|
||||||
@@ -331,7 +331,7 @@ function M.find(modname, opts)
|
|||||||
|
|
||||||
-- Checks if the given paths contain the top-level module.
|
-- Checks if the given paths contain the top-level module.
|
||||||
-- If so, it tries to find the module path for the given module name.
|
-- If so, it tries to find the module path for the given module name.
|
||||||
---@param paths string[]
|
--- @param paths string[]
|
||||||
local function _find(paths)
|
local function _find(paths)
|
||||||
for _, path in ipairs(paths) do
|
for _, path in ipairs(paths) do
|
||||||
if topmod == '*' then
|
if topmod == '*' then
|
||||||
@@ -385,7 +385,7 @@ end
|
|||||||
---
|
---
|
||||||
--- @since 0
|
--- @since 0
|
||||||
---
|
---
|
||||||
---@param path string? path to reset
|
--- @param path string? path to reset
|
||||||
function M.reset(path)
|
function M.reset(path)
|
||||||
if path then
|
if path then
|
||||||
indexed[normalize(path)] = nil
|
indexed[normalize(path)] = nil
|
||||||
@@ -460,13 +460,13 @@ local function track(stat, f)
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
---@class (private) vim.loader._profile.Opts
|
--- @class (private) vim.loader._profile.Opts
|
||||||
---@field loaders? boolean Add profiling to the loaders
|
--- @field loaders? boolean Add profiling to the loaders
|
||||||
|
|
||||||
--- Debug function that wraps all loaders and tracks stats
|
--- Debug function that wraps all loaders and tracks stats
|
||||||
--- Must be called before vim.loader.enable()
|
--- Must be called before vim.loader.enable()
|
||||||
---@private
|
--- @private
|
||||||
---@param opts vim.loader._profile.Opts?
|
--- @param opts vim.loader._profile.Opts?
|
||||||
function M._profile(opts)
|
function M._profile(opts)
|
||||||
get_rtp = track('get_rtp', get_rtp)
|
get_rtp = track('get_rtp', get_rtp)
|
||||||
read_cachefile = track('read', read_cachefile)
|
read_cachefile = track('read', read_cachefile)
|
||||||
@@ -485,15 +485,15 @@ function M._profile(opts)
|
|||||||
end
|
end
|
||||||
|
|
||||||
--- Prints all cache stats
|
--- Prints all cache stats
|
||||||
---@param opts? {print?:boolean}
|
--- @param opts? {print?:boolean}
|
||||||
---@return vim.loader.Stats
|
--- @return vim.loader.Stats
|
||||||
---@private
|
--- @private
|
||||||
function M._inspect(opts)
|
function M._inspect(opts)
|
||||||
if opts and opts.print then
|
if opts and opts.print then
|
||||||
local function ms(nsec)
|
local function ms(nsec)
|
||||||
return math.floor(nsec / 1e6 * 1000 + 0.5) / 1000 .. 'ms'
|
return math.floor(nsec / 1e6 * 1000 + 0.5) / 1000 .. 'ms'
|
||||||
end
|
end
|
||||||
local chunks = {} ---@type string[][]
|
local chunks = {} --- @type string[][]
|
||||||
for _, stat in vim.spairs(stats) do
|
for _, stat in vim.spairs(stats) do
|
||||||
vim.list_extend(chunks, {
|
vim.list_extend(chunks, {
|
||||||
{ '\n' .. stat .. '\n', 'Title' },
|
{ '\n' .. stat .. '\n', 'Title' },
|
||||||
|
Reference in New Issue
Block a user