mirror of https://github.com/folke/lazy.nvim.git
97 lines
2.0 KiB
Lua
97 lines
2.0 KiB
Lua
local Config = require("lazy.core.config")
|
|
local Util = require("lazy.util")
|
|
|
|
local M = {}
|
|
|
|
---@alias LazyPkgSpec LazySpec | fun():LazySpec
|
|
|
|
---@class LazyPkg
|
|
---@field source string
|
|
---@field file string
|
|
---@field spec? LazySpec
|
|
---@field chunk? string|fun():LazySpec
|
|
|
|
---@class LazyPkgSource
|
|
---@field get fun(plugin:LazyPlugin):LazyPkg
|
|
|
|
---@type table<string, LazyPkg>?
|
|
M.cache = nil
|
|
|
|
function M.update()
|
|
---@type LazyPkgSource[]
|
|
local sources = {}
|
|
for _, s in ipairs(Config.options.pkg.sources) do
|
|
sources[#sources + 1] = require("lazy.pkg." .. s)
|
|
end
|
|
|
|
---@type table<string, LazyPkg>
|
|
local ret = {}
|
|
for _, plugin in pairs(Config.plugins) do
|
|
for _, source in ipairs(sources) do
|
|
local spec = source.get(plugin)
|
|
if spec then
|
|
if type(spec.chunk) == "function" then
|
|
spec.chunk = string.dump(spec.chunk)
|
|
end
|
|
ret[plugin.dir] = spec
|
|
break
|
|
end
|
|
end
|
|
end
|
|
local code = "return " .. Util.dump(ret)
|
|
Util.write_file(Config.options.pkg.cache, code)
|
|
M.cache = nil
|
|
end
|
|
|
|
local function _load()
|
|
Util.track("pkg")
|
|
M.cache = {}
|
|
if vim.uv.fs_stat(Config.options.pkg.cache) then
|
|
Util.try(function()
|
|
local chunk, err = loadfile(Config.options.pkg.cache)
|
|
if not chunk then
|
|
error(err)
|
|
end
|
|
M.cache = chunk()
|
|
end, "Error loading pkg:")
|
|
end
|
|
Util.track()
|
|
end
|
|
|
|
---@param plugin LazyPlugin
|
|
---@return LazyPkg?
|
|
function M.get(plugin)
|
|
if not M.cache then
|
|
_load()
|
|
end
|
|
|
|
local ret = M.cache[plugin.dir]
|
|
if not ret then
|
|
return
|
|
end
|
|
|
|
if ret.chunk and not ret.spec then
|
|
if type(ret.chunk) == "string" then
|
|
ret.chunk = load(ret.chunk, "@" .. plugin.dir)
|
|
end
|
|
ret.spec = ret.chunk()
|
|
ret.chunk = nil
|
|
end
|
|
|
|
return ret
|
|
end
|
|
|
|
---@param plugin LazyPlugin
|
|
---@return LazySpec?
|
|
function M.get_spec(plugin)
|
|
local pkg = M.get(plugin)
|
|
local spec = pkg and pkg.spec
|
|
if not spec then
|
|
return
|
|
end
|
|
spec = type(spec) == "table" and vim.deepcopy(spec) or spec
|
|
return { plugin.name, specs = spec }
|
|
end
|
|
|
|
return M
|