mirror of
https://github.com/cap153/nvim.git
synced 2026-04-17 09:45:29 +08:00
327 lines
11 KiB
Lua
327 lines
11 KiB
Lua
-- ==============================================================
|
||
-- 插件花名册
|
||
-- ==============================================================
|
||
local specs = {
|
||
-- 公共依赖
|
||
'https://github.com/nvim-lua/plenary.nvim',
|
||
'https://github.com/williamboman/mason.nvim',
|
||
'https://github.com/mason-org/mason-registry',
|
||
'https://github.com/nvim-tree/nvim-web-devicons',
|
||
-- blinkcmp.lua 自动补全、代码片段
|
||
'https://github.com/saghen/blink.cmp',
|
||
'https://github.com/rafamadriz/friendly-snippets',
|
||
-- lspconfig.lua
|
||
'https://github.com/neovim/nvim-lspconfig',
|
||
'https://github.com/williamboman/mason-lspconfig.nvim',
|
||
'https://github.com/MysticalDevil/inlay-hints.nvim',
|
||
-- treesitter.lua 需要安装tree-sitter-cli工具包
|
||
'https://github.com/nvim-treesitter/nvim-treesitter',
|
||
-- indentblankline.lua 彩虹缩进
|
||
"https://github.com/lukas-reineke/indent-blankline.nvim",
|
||
-- ufo.lua 折叠插件
|
||
"https://github.com/kevinhwang91/nvim-ufo",
|
||
"https://github.com/kevinhwang91/promise-async",
|
||
-- conform.lua 格式化工具formatter
|
||
"https://github.com/stevearc/conform.nvim",
|
||
-- noice.lua 取代消息、命令行和弹出菜单的 UI
|
||
"https://github.com/folke/noice.nvim",
|
||
"https://github.com/MunifTanjim/nui.nvim",
|
||
-- snacks.lua 图片预览、lazygit、lsp_references模糊查找
|
||
"https://github.com/folke/snacks.nvim",
|
||
-- mini.lua 各种对字符的surround包裹{} [] ''
|
||
"https://github.com/echasnovski/mini.nvim",
|
||
-- autopairs.lua 对字符自动补全另一半
|
||
"https://github.com/windwp/nvim-autopairs",
|
||
-- tv.lua 模糊查找television
|
||
"https://github.com/alexpasmantier/tv.nvim",
|
||
-- coderunner.lua 运行代码
|
||
'https://github.com/CRAG666/code_runner.nvim',
|
||
-- peek.lua 预览markdown
|
||
'https://github.com/cap153/peek.nvim',
|
||
-- bufferline.lua 顶部状态栏
|
||
'https://github.com/akinsho/bufferline.nvim',
|
||
-- yazi.lua 文件管理器
|
||
'https://github.com/mikavilpas/yazi.nvim',
|
||
-- sudo权限保存文件
|
||
"https://github.com/lambdalisue/vim-suda",
|
||
-- 查看可用键位
|
||
"https://github.com/folke/which-key.nvim",
|
||
}
|
||
-- 禁用插件：不会加载，不会下载(如果是新添加的)，已在硬盘上不会被删除
|
||
local disabled = {
|
||
-- { src = 'https://github.com/saghen/blink.cmp', version = 'v1.10.2' }, -- 指定版本，暂时不想用，但想留着源码
|
||
}
|
||
|
||
-- ==============================================================
|
||
-- 快捷管理命令
|
||
-- ==============================================================
|
||
|
||
-- 获取所有已安装插件的名称列表（用于 Tab 补全）
|
||
local function get_plugin_names(arg_lead)
|
||
local installed = vim.pack.get(nil, { info = false })
|
||
local names = {}
|
||
for _, p in ipairs(installed) do
|
||
local name = p.spec.name
|
||
-- 只添加匹配开头字符串的插件
|
||
if name:lower():find(arg_lead:lower(), 1, true) == 1 then
|
||
table.insert(names, name)
|
||
end
|
||
end
|
||
-- 排序让补全列表更整洁
|
||
table.sort(names)
|
||
return names
|
||
end
|
||
|
||
-- :PackUpdate 命令更新插件，不带参数更新全部
|
||
vim.api.nvim_create_user_command("PackUpdate", function(opts)
|
||
local targets = #opts.fargs > 0 and opts.fargs or nil
|
||
if targets then
|
||
vim.notify("Checking updates for: " .. table.concat(targets, ", "), vim.log.levels.INFO)
|
||
else
|
||
vim.notify("Checking updates for all plugins...", vim.log.levels.INFO)
|
||
end
|
||
vim.pack.update(targets)
|
||
end, {
|
||
nargs = "*", -- 支持 0 到多个参数
|
||
complete = get_plugin_names, -- 绑定补全函数
|
||
desc = "Update specified or all plugins",
|
||
})
|
||
|
||
-- :PackStatus 命令查看插件当前状态和版本
|
||
vim.api.nvim_create_user_command("PackStatus", function(opts)
|
||
local targets = #opts.fargs > 0 and opts.fargs or nil
|
||
vim.pack.update(targets, { offline = true })
|
||
end, {
|
||
nargs = "*",
|
||
complete = get_plugin_names,
|
||
desc = "Check plugin status without downloading",
|
||
})
|
||
|
||
-- ==============================================================
|
||
-- 插件管理引擎 (PackUtils) (暴露给全局，供 configs/*.lua 调用)
|
||
-- ==============================================================
|
||
_G.PackUtils = {
|
||
is_building = {}, -- 记录各插件的构建状态，防止重复构建
|
||
is_initialized = {}, -- 统一在这里管理所有插件的初始化状态
|
||
disabled_plugins = {}, -- 专门记录被禁用的插件，供 load 拦截使用
|
||
}
|
||
|
||
-- [解析插件名]
|
||
function PackUtils.get_name(spec)
|
||
local url = type(spec) == "table" and spec.src or spec
|
||
return type(spec) == "table" and spec.name or url:match("([^/]+)$"):gsub("%.git$", "")
|
||
end
|
||
|
||
-- [同步清理] 自动删除孤儿，并注册禁用名单
|
||
function PackUtils.sync(active_specs, disabled_specs)
|
||
disabled_specs = disabled_specs or {}
|
||
local protected_names = {}
|
||
|
||
-- 将插件加入受保护名单
|
||
for _, spec in ipairs(active_specs) do
|
||
protected_names[PackUtils.get_name(spec)] = true
|
||
end
|
||
for _, spec in ipairs(disabled_specs) do
|
||
local name = PackUtils.get_name(spec)
|
||
protected_names[name] = true
|
||
PackUtils.disabled_plugins[name] = true -- 写入字典，供 load 拦截
|
||
end
|
||
|
||
-- 扫描磁盘
|
||
local pack_dir = vim.fn.stdpath("data") .. "/site/pack"
|
||
local installed_plugins = {}
|
||
local packages = vim.fn.expand(pack_dir .. "/*", false, true)
|
||
for _, pkg in ipairs(packages) do
|
||
for _, type_dir in ipairs({ "start", "opt" }) do
|
||
local path = pkg .. "/" .. type_dir
|
||
if vim.fn.isdirectory(path) == 1 then
|
||
local dirs = vim.fn.expand(path .. "/*", false, true)
|
||
for _, dir in ipairs(dirs) do
|
||
local name = dir:match("([^/]+)$")
|
||
if name ~= "README.md" and name ~= "doc" then
|
||
table.insert(installed_plugins, name)
|
||
end
|
||
end
|
||
end
|
||
end
|
||
end
|
||
-- 找出既不在 active 也不在 disabled 里的孤儿
|
||
local to_delete = {}
|
||
for _, installed in ipairs(installed_plugins) do
|
||
if not protected_names[installed] then
|
||
table.insert(to_delete, installed)
|
||
end
|
||
end
|
||
|
||
if #to_delete > 0 then
|
||
vim.schedule(function()
|
||
vim.notify("🧹 Clean Up Orphaned Plugins: " .. table.concat(to_delete, ", "), vim.log.levels.INFO)
|
||
vim.pack.del(to_delete)
|
||
end)
|
||
end
|
||
end
|
||
|
||
-- [动态路径] 获取插件根目录
|
||
function PackUtils.get_root(name)
|
||
name = PackUtils.get_name(name)
|
||
local paths = vim.api.nvim_get_runtime_file("pack/*/*/" .. name, true)
|
||
if #paths > 0 then return paths[1] end
|
||
local glob = vim.fn.globpath(vim.o.packpath, "pack/*/*/" .. name, 0, 1)
|
||
return glob[1] or nil
|
||
end
|
||
|
||
-- [构建执行] 执行编译任务
|
||
function PackUtils.run_build(name, build_cmd)
|
||
name = PackUtils.get_name(name)
|
||
if PackUtils.disabled_plugins[name] then return end
|
||
if not build_cmd or PackUtils.is_building[name] then return end
|
||
local path = PackUtils.get_root(name)
|
||
if not path then return end
|
||
local stamp = path .. "/.build_done"
|
||
PackUtils.is_building[name] = true
|
||
|
||
-- 判断是否为 Neovim 内部命令 (以 : 开头)
|
||
local is_vim_cmd = false
|
||
local vim_cmd_str = ""
|
||
|
||
if type(build_cmd) == "string" and build_cmd:sub(1, 1) == ":" then
|
||
is_vim_cmd = true
|
||
vim_cmd_str = build_cmd:sub(2)
|
||
elseif type(build_cmd) == "table" and type(build_cmd[1]) == "string" and build_cmd[1]:sub(1, 1) == ":" then
|
||
is_vim_cmd = true
|
||
vim_cmd_str = table.concat(build_cmd, " "):sub(2)
|
||
end
|
||
|
||
if is_vim_cmd then
|
||
-- 在当前实例的空闲时执行 vim.cmd
|
||
vim.schedule(function()
|
||
vim.notify("⚙️ Running " .. name .. " setup command...", vim.log.levels.INFO)
|
||
-- 确保插件在当前实例已经被加载
|
||
pcall(vim.cmd.packadd, name)
|
||
-- 保护执行，防止命令错误导致编辑器崩溃
|
||
local ok, err = pcall(vim.cmd, vim_cmd_str)
|
||
PackUtils.is_building[name] = false
|
||
if ok then
|
||
local f = io.open(stamp, "w")
|
||
if f then f:close() end
|
||
vim.notify("✅ " .. name .. " setup success.", vim.log.levels.INFO)
|
||
else
|
||
vim.notify("❌ " .. name .. " setup failed: " .. tostring(err), vim.log.levels.ERROR)
|
||
end
|
||
end)
|
||
else
|
||
local final_cmd = {}
|
||
if type(build_cmd) == "string" then
|
||
for word in build_cmd:gmatch("%S+") do
|
||
table.insert(final_cmd, word)
|
||
end
|
||
else
|
||
final_cmd = build_cmd
|
||
end
|
||
vim.schedule(function() vim.notify("⚙️ Building " .. name .. " (Background)...", vim.log.levels.INFO) end)
|
||
vim.system(final_cmd, { cwd = path }, function(out)
|
||
PackUtils.is_building[name] = false
|
||
if out.code == 0 then
|
||
local f = io.open(stamp, "w")
|
||
if f then f:close() end
|
||
vim.schedule(function() vim.notify("✅ " .. name .. " build success.", vim.log.levels.INFO) end)
|
||
else
|
||
vim.schedule(function()
|
||
vim.notify("❌ " .. name .. " build failed: " .. (out.stderr or "Unknown Error"), vim.log.levels.ERROR)
|
||
end)
|
||
end
|
||
end)
|
||
end
|
||
end
|
||
|
||
-- [监听器] 注册安装/更新监听
|
||
function PackUtils.setup_listener(name, build_cmd)
|
||
name = PackUtils.get_name(name)
|
||
if PackUtils.disabled_plugins[name] then return end
|
||
if not build_cmd then return end
|
||
vim.api.nvim_create_autocmd('PackChanged', {
|
||
pattern = '*',
|
||
callback = function(ev)
|
||
if ev.data.spec.name == name and (ev.data.kind == "update" or ev.data.kind == "install") then
|
||
local stamp = ev.data.path .. "/.build_done"
|
||
os.remove(stamp) -- 自动删除.build_done文件触发构建
|
||
PackUtils.run_build(name, build_cmd)
|
||
end
|
||
end
|
||
})
|
||
end
|
||
|
||
-- [健康检查] 如果没标记且有构建命令，则触发构建
|
||
function PackUtils.check_health(name, build_cmd)
|
||
name = PackUtils.get_name(name)
|
||
if PackUtils.disabled_plugins[name] then return end
|
||
if not build_cmd then return end
|
||
local path = PackUtils.get_root(name)
|
||
if path then
|
||
local stamp = path .. "/.build_done"
|
||
if vim.fn.filereadable(stamp) == 0 then
|
||
PackUtils.run_build(name, build_cmd)
|
||
end
|
||
end
|
||
end
|
||
|
||
-- 全方位防崩加载引擎
|
||
function PackUtils.load(P, config_fn)
|
||
-- 自动纠错插件名和依赖名
|
||
P.name = PackUtils.get_name(P.name)
|
||
if P.deps then
|
||
for i, dep in ipairs(P.deps) do
|
||
P.deps[i] = PackUtils.get_name(dep)
|
||
end
|
||
end
|
||
|
||
if PackUtils.disabled_plugins[P.name] then return end
|
||
if PackUtils.is_initialized[P.name] then return end
|
||
|
||
-- 检查插件是否存在于磁盘，如果找不到，说明它正在后台被 vim.pack 异步克隆下载，直接静默退出
|
||
if not PackUtils.get_root(P.name) then return end
|
||
|
||
-- 走到这里，说明插件绝对在硬盘上了，执行常规准备工作
|
||
PackUtils.check_health(P.name, P.build_cmd)
|
||
pcall(vim.cmd.packadd, P.name)
|
||
if P.deps then
|
||
for _, dep in ipairs(P.deps) do
|
||
local dep_ok = pcall(vim.cmd.packadd, dep)
|
||
if not dep_ok then
|
||
vim.notify("Warning: " .. P.name .. " dependency[" .. dep .. "] missing", vim.log.levels.WARN)
|
||
end
|
||
end
|
||
end
|
||
|
||
-- 保护 Setup 执行：自由地 require，如有拼写错误，这里的 pcall 会完美捕获并报错
|
||
if config_fn then
|
||
local setup_ok, err = pcall(config_fn)
|
||
if not setup_ok then
|
||
vim.notify("Error: " .. P.name .. " setup failed: \n" .. tostring(err), vim.log.levels.ERROR)
|
||
return
|
||
end
|
||
end
|
||
|
||
-- 只有全部流程走通，才标记为已初始化
|
||
PackUtils.is_initialized[P.name] = true
|
||
end
|
||
|
||
-- ==============================================================
|
||
-- 执行启动流程
|
||
-- ==============================================================
|
||
|
||
-- 同步清理孤儿插件并注册禁用名单
|
||
PackUtils.sync(specs, disabled)
|
||
|
||
-- 正式下载/更新插件
|
||
vim.pack.add(specs)
|
||
|
||
-- 加载 configs/ 注册所有监听器
|
||
local config_path = vim.fn.stdpath("config") .. "/lua/pack/configs"
|
||
if vim.fn.isdirectory(config_path) == 1 then
|
||
for name, type in vim.fs.dir(config_path) do
|
||
if type == "file" and name:match("%.lua$") then
|
||
pcall(require, "pack.configs." .. name:gsub("%.lua$", ""))
|
||
end
|
||
end
|
||
end
|