refactor: simplify plugin loader
This commit is contained in:
+269
@@ -0,0 +1,269 @@
|
||||
local log = require("log")
|
||||
|
||||
local config_dir = vim.fn.stdpath("config")
|
||||
|
||||
---@param path string
|
||||
---@return boolean success
|
||||
---@return string? err
|
||||
local function exec(path)
|
||||
local chunk, load_err = loadfile(path)
|
||||
if not chunk then
|
||||
return false, load_err
|
||||
end
|
||||
|
||||
local ok, call_err = pcall(chunk)
|
||||
if not ok then
|
||||
return false, call_err
|
||||
end
|
||||
|
||||
return true
|
||||
end
|
||||
|
||||
---@class ow.Pack.PluginSpec
|
||||
---@field [1] string
|
||||
---@field name? string
|
||||
---@field version? string | vim.VersionRange
|
||||
---@field build? string[] | fun(self: ow.Pack.Plugin)
|
||||
|
||||
---@class ow.Pack.Plugin : ow.Pack.PluginSpec
|
||||
---@field name string
|
||||
---@field path string
|
||||
|
||||
---@param plugin ow.Pack.Plugin
|
||||
local function load(plugin)
|
||||
local name = plugin.name:match("[^.]+"):lower()
|
||||
local path = string.format("%s/lua/plugins/%s.lua", config_dir, name)
|
||||
|
||||
if vim.uv.fs_stat(path) then
|
||||
local ok, err = exec(path)
|
||||
if not ok then
|
||||
log.error("Failed to load %s: %s", name, err)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
---@param spec string | ow.Pack.PluginSpec
|
||||
---@return vim.pack.Spec
|
||||
local function to_pack_spec(spec)
|
||||
if type(spec) == "string" then
|
||||
return { src = spec }
|
||||
end
|
||||
|
||||
return {
|
||||
src = spec[1],
|
||||
name = spec.name,
|
||||
version = spec.version,
|
||||
data = {
|
||||
build = spec.build,
|
||||
},
|
||||
}
|
||||
end
|
||||
|
||||
---@param plugin ow.Pack.Plugin
|
||||
local function run_build(plugin)
|
||||
if type(plugin.build) == "function" then
|
||||
plugin.build(plugin)
|
||||
return
|
||||
elseif type(plugin.build) == "table" then
|
||||
local ret = vim.system(
|
||||
plugin.build --[[@as table]],
|
||||
{ cwd = plugin.path }
|
||||
):wait()
|
||||
if ret.code ~= 0 then
|
||||
log.error("Build failed for %s: %s", plugin.name, ret.stderr or "")
|
||||
end
|
||||
return
|
||||
end
|
||||
|
||||
log.error("invalid build parameter for %s", plugin.name)
|
||||
end
|
||||
|
||||
---@class ow.Pack.Event.Data
|
||||
---@field active boolean
|
||||
---@field kind "install" | "update" | "delete"
|
||||
---@field spec vim.pack.Spec
|
||||
---@field path string
|
||||
|
||||
---@class ow.Pack.Event : vim.api.keyset.create_autocmd.callback_args
|
||||
---@field data ow.Pack.Event.Data
|
||||
|
||||
---@param plugin ow.Pack.Plugin
|
||||
---@param events ow.Pack.Event[]
|
||||
local function process_events(plugin, events)
|
||||
if not plugin.build then
|
||||
return
|
||||
end
|
||||
|
||||
for _, ev in ipairs(events) do
|
||||
if ev.data.spec.name == plugin.name
|
||||
and ev.event == "PackChanged"
|
||||
and (ev.data.kind == "install" or ev.data.kind == "update")
|
||||
then
|
||||
run_build(plugin)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
---@type uv.uv_fs_event_t?
|
||||
local watcher = nil
|
||||
|
||||
---@type table<string, uv.uv_timer_t>
|
||||
local timers = {}
|
||||
|
||||
---@class ow.Pack
|
||||
---@field names string[]
|
||||
---@field paths string[]
|
||||
---@field plugins ow.Pack.Plugin[]
|
||||
local M = {
|
||||
names = {},
|
||||
paths = {},
|
||||
plugins = {},
|
||||
}
|
||||
|
||||
---@param name string
|
||||
function M.reload(name)
|
||||
local path = string.format("%s/lua/plugins/%s.lua", config_dir, name)
|
||||
|
||||
if not vim.uv.fs_stat(path) then
|
||||
log.error("No config file found for %s", name)
|
||||
return
|
||||
end
|
||||
|
||||
local ok, err = exec(path)
|
||||
if ok then
|
||||
log.info("Reloaded %s", name)
|
||||
else
|
||||
log.error("Failed to reload %s: %s", name, err)
|
||||
end
|
||||
end
|
||||
|
||||
function M.watch()
|
||||
if watcher then
|
||||
return
|
||||
end
|
||||
|
||||
local plugins_dir = config_dir .. "/lua/plugins"
|
||||
|
||||
watcher = vim.uv.new_fs_event()
|
||||
if not watcher then
|
||||
log.error("Failed to create fs_event watcher")
|
||||
return
|
||||
end
|
||||
|
||||
watcher:start(
|
||||
plugins_dir,
|
||||
{},
|
||||
vim.schedule_wrap(function(err, filename)
|
||||
if err then
|
||||
log.error("Watch error: %s", err)
|
||||
return
|
||||
end
|
||||
|
||||
if not filename or not filename:match("%.lua$") then
|
||||
return
|
||||
end
|
||||
|
||||
if timers[filename] then
|
||||
timers[filename]:stop()
|
||||
else
|
||||
timers[filename] = vim.uv.new_timer()
|
||||
end
|
||||
|
||||
timers[filename]:start(
|
||||
100,
|
||||
0,
|
||||
vim.schedule_wrap(function()
|
||||
timers[filename]:stop()
|
||||
timers[filename]:close()
|
||||
timers[filename] = nil
|
||||
|
||||
local path = plugins_dir .. "/" .. filename
|
||||
if not vim.uv.fs_stat(path) then
|
||||
return
|
||||
end
|
||||
|
||||
local ok, load_err = exec(path)
|
||||
if ok then
|
||||
log.info("Reloaded %s", filename)
|
||||
else
|
||||
log.error(
|
||||
"Failed to reload %s: %s",
|
||||
filename,
|
||||
load_err
|
||||
)
|
||||
end
|
||||
end)
|
||||
)
|
||||
end)
|
||||
)
|
||||
end
|
||||
|
||||
function M.unwatch()
|
||||
if not watcher then
|
||||
return
|
||||
end
|
||||
|
||||
for key, timer in pairs(timers) do
|
||||
timer:stop()
|
||||
timer:close()
|
||||
timers[key] = nil
|
||||
end
|
||||
|
||||
watcher:stop()
|
||||
watcher:close()
|
||||
watcher = nil
|
||||
end
|
||||
|
||||
---@param specs (string | ow.Pack.PluginSpec)[]
|
||||
function M.setup(specs)
|
||||
---@type table<string, ow.Pack.Event[]?>
|
||||
local events = {}
|
||||
local group = vim.api.nvim_create_augroup("ow.Pack", { clear = true })
|
||||
local id = vim.api.nvim_create_autocmd(
|
||||
{ "PackChangedPre", "PackChanged" },
|
||||
{
|
||||
group = group,
|
||||
---@param ev ow.Pack.Event
|
||||
callback = function(ev)
|
||||
local name = ev.data.spec.name
|
||||
if not name then return end
|
||||
if not events[name] then
|
||||
events[name] = {}
|
||||
end
|
||||
table.insert(events[name], ev)
|
||||
end
|
||||
}
|
||||
)
|
||||
vim.pack.add(vim.tbl_map(to_pack_spec, specs), {
|
||||
load = function(data)
|
||||
if not data.spec.name then
|
||||
log.error("Missing name for plugin: %s", data.spec.src)
|
||||
return
|
||||
end
|
||||
local d = data.spec.data or {}
|
||||
---@type ow.Pack.Plugin
|
||||
local plugin = {
|
||||
[1] = data.spec.src,
|
||||
name = data.spec.name,
|
||||
version = data.spec.version,
|
||||
build = d.build,
|
||||
path = data.path,
|
||||
}
|
||||
table.insert(M.plugins, plugin)
|
||||
table.insert(M.names, plugin.name)
|
||||
table.insert(M.paths, plugin.path)
|
||||
vim.cmd.packadd(plugin.name)
|
||||
end
|
||||
})
|
||||
vim.api.nvim_del_autocmd(id)
|
||||
|
||||
for _, plugin in ipairs(M.plugins) do
|
||||
local ev = events[plugin.name]
|
||||
if ev then
|
||||
process_events(plugin, ev)
|
||||
end
|
||||
load(plugin)
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
Reference in New Issue
Block a user