feat: automated setup works finally
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
local utils = require('utils')
|
||||
local M = {
|
||||
lsp = {},
|
||||
ts = {},
|
||||
lsp = {},
|
||||
lint = {},
|
||||
format = {},
|
||||
}
|
||||
@@ -138,14 +139,20 @@ end
|
||||
|
||||
-- ======== Spec Generation ========
|
||||
|
||||
local function get_mason_registry()
|
||||
vim.cmd.packadd('mason.nvim')
|
||||
require('mason').setup()
|
||||
|
||||
local registry = require('mason-registry')
|
||||
registry.refresh()
|
||||
|
||||
return registry
|
||||
end
|
||||
|
||||
function M.generate_specs(specs_raw)
|
||||
local install_spec = create_spec()
|
||||
local specs = create_spec()
|
||||
|
||||
-- ensure Mason is available
|
||||
vim.cmd.packadd('mason.nvim')
|
||||
require('mason').setup()
|
||||
local registry = require('mason-registry')
|
||||
local registry = get_mason_registry()
|
||||
|
||||
local lsp_map = {}
|
||||
|
||||
@@ -157,17 +164,21 @@ function M.generate_specs(specs_raw)
|
||||
spec.ts = spec.ts or spec.ft
|
||||
specs.add(spec.ts, 'ts_parsers')
|
||||
|
||||
-- resolve LSP name using Mason registry
|
||||
install_spec.add(spec.lsp, 'language_servers')
|
||||
install_spec.add(spec.lsp, 'code_tools')
|
||||
local resolved_lsps = {}
|
||||
for _, lsp in ipairs(wrap(spec.lsp)) do
|
||||
if registry.has_package(lsp) then
|
||||
local pkg = registry.get_package(lsp)
|
||||
local lspconfig = pkg.spec and pkg.spec.neovim and pkg.spec.neovim.lspconfig or lsp
|
||||
table.insert(resolved_lsps, lspconfig)
|
||||
lsp_map[lspconfig] = lsp
|
||||
for _, lsp_name in ipairs(wrap(spec.lsp)) do
|
||||
if registry and registry.has_package(lsp_name) then
|
||||
local pkg = registry.get_package(lsp_name)
|
||||
if pkg.spec and pkg.spec.neovim and pkg.spec.neovim.lspconfig then
|
||||
local lspconfig = pkg.spec and pkg.spec.neovim and pkg.spec.neovim.lspconfig
|
||||
lsp_name = lspconfig
|
||||
lsp_map[lspconfig] = lsp_name
|
||||
table.insert(resolved_lsps, lsp_name)
|
||||
else
|
||||
print('Package found but not lsp name: ' .. lsp_name)
|
||||
end
|
||||
else
|
||||
vim.notify('Unknown LSP: ' .. lsp, vim.log.levels.WARN)
|
||||
print('Package not found: ' .. lsp_name)
|
||||
end
|
||||
end
|
||||
specs.add(resolved_lsps, 'language_servers')
|
||||
@@ -185,19 +196,7 @@ function M.generate_specs(specs_raw)
|
||||
return result, install_spec.get()
|
||||
end
|
||||
|
||||
-- ======== Cache Interface ========
|
||||
|
||||
function M.load_or_generate(specs_raw)
|
||||
local hash = hash_spec(specs_raw)
|
||||
local cache = load_cache()
|
||||
if cache and cache.hash == hash then
|
||||
M.general = cache.spec
|
||||
else
|
||||
M.general, M.mason = M.generate_specs(specs_raw)
|
||||
save_cache({ hash = hash, spec = M.general })
|
||||
end
|
||||
return M.general, M.mason
|
||||
end
|
||||
-- ======== Cache ========
|
||||
|
||||
function M.load_specs()
|
||||
local cache = load_cache()
|
||||
@@ -205,9 +204,8 @@ function M.load_specs()
|
||||
M.general = cache.spec
|
||||
else
|
||||
local specs_raw = require('modules.language-specs').get()
|
||||
local hash = hash_spec(specs_raw)
|
||||
M.general, M.mason = M.generate_specs(specs_raw)
|
||||
save_cache({ hash = hash, spec = M.general })
|
||||
save_cache({ hash = hash_spec(specs_raw), spec = M.general })
|
||||
end
|
||||
return M.general, M.mason
|
||||
end
|
||||
@@ -223,103 +221,53 @@ function M.invalidate_cache()
|
||||
M.mason = nil
|
||||
end
|
||||
|
||||
-- ======== Setup Functions ========
|
||||
|
||||
local function on_ts_installed(parsers, cb)
|
||||
local info = require('nvim-treesitter.info')
|
||||
local timer = vim.loop.new_timer()
|
||||
timer:start(0, 500, vim.schedule_wrap(function()
|
||||
for _, lang in ipairs(parsers) do
|
||||
if not vim.tbl_contains(info.installed_parsers(), lang) then
|
||||
return
|
||||
end
|
||||
end
|
||||
timer:stop()
|
||||
timer:close()
|
||||
cb()
|
||||
end))
|
||||
end
|
||||
|
||||
function M.ts.install()
|
||||
local install = require('nvim-treesitter.install')
|
||||
local parsers = M.general.ts_parsers or {}
|
||||
install.ensure_installed(parsers) -- async
|
||||
return parsers
|
||||
end
|
||||
|
||||
-- ======== Orchestration ========
|
||||
|
||||
function M.install()
|
||||
local parsers = M.ts.install()
|
||||
on_ts_installed(parsers, function()
|
||||
M.mason_install()
|
||||
end)
|
||||
end
|
||||
|
||||
local function ensure_registry_ready()
|
||||
local registry = require('mason-registry')
|
||||
if not registry.is_installed() then
|
||||
registry.update()
|
||||
vim.wait(10000, function()
|
||||
return registry.is_installed()
|
||||
end, 200)
|
||||
end
|
||||
return registry
|
||||
vim.cmd.packadd('nvim-treesitter')
|
||||
local ts_install = require('nvim-treesitter.install').ensure_installed_sync
|
||||
ts_install(M.general.ts_parsers)
|
||||
end
|
||||
|
||||
function M.mason_install()
|
||||
vim.cmd.packadd('mason.nvim')
|
||||
require('mason').setup()
|
||||
local registry = ensure_registry_ready()
|
||||
local list = vim.list_extend(M.mason.language_servers, M.mason.code_tools)
|
||||
print(vim.inspect(list))
|
||||
local packages = M.mason.code_tools
|
||||
local registry = get_mason_registry()
|
||||
|
||||
local function install_and_wait(name)
|
||||
local pkg = registry.get_package(name)
|
||||
if pkg:is_installed() then
|
||||
vim.notify('Already installed ' .. name, vim.log.levels.INFO)
|
||||
return
|
||||
local pending = #packages
|
||||
|
||||
local result = utils.await(function(resolve)
|
||||
for _, name in ipairs(packages) do
|
||||
print('Mason package installing: ' .. name)
|
||||
local pkg = registry.get_package(name)
|
||||
pkg:install({}, function(success, error)
|
||||
if success then
|
||||
print('Mason package installed: ' .. name)
|
||||
else
|
||||
print('Mason package failed: ' .. name)
|
||||
print(' > Error: ' .. vim.inspect(error))
|
||||
end
|
||||
|
||||
pending = pending - 1
|
||||
if pending == 0 then
|
||||
resolve(true)
|
||||
end
|
||||
end)
|
||||
end
|
||||
end, 5 * 60 * 1000, 200)
|
||||
|
||||
vim.notify('Installing ' .. name, vim.log.levels.INFO)
|
||||
local done = false
|
||||
pkg:install():once('closed', function()
|
||||
done = true
|
||||
end)
|
||||
|
||||
local ok = vim.wait(60 * 60 * 1000, function()
|
||||
return done or pkg:is_installed()
|
||||
end, 200)
|
||||
|
||||
if not ok or not pkg:is_installed() then
|
||||
vim.notify('Install failed: ' .. name, vim.log.levels.ERROR)
|
||||
else
|
||||
vim.notify('Installed ' .. name, vim.log.levels.INFO)
|
||||
end
|
||||
if not result.ok or pending ~= 0 then
|
||||
print('\n!! >> Exited timeout, possible clean up needed!')
|
||||
print(' > status: ' .. result.ok)
|
||||
print(' > pending: ' .. pending)
|
||||
end
|
||||
|
||||
for _, name in ipairs(list) do
|
||||
if registry.has_package(name) then
|
||||
install_and_wait(name)
|
||||
else
|
||||
vim.notify('Package not found in registry: ' .. name, vim.log.levels.WARN)
|
||||
end
|
||||
end
|
||||
|
||||
-- force event loop to process any pending Mason async teardown
|
||||
local settled = false
|
||||
vim.defer_fn(function()
|
||||
settled = true
|
||||
end, 3000)
|
||||
vim.wait(10 * 1000, function()
|
||||
return settled
|
||||
end, 100)
|
||||
end
|
||||
|
||||
function M.lsp.setup()
|
||||
for _, lsp_name in ipairs(M.general.language_servers or {}) do
|
||||
vim.lsp.enable(lsp_name)
|
||||
end
|
||||
-- ======== Public API ========
|
||||
|
||||
function M.install()
|
||||
print('\n> Starting ts parsers install')
|
||||
M.ts.install()
|
||||
|
||||
print('\n> Starting mason install: lsp, lint, format')
|
||||
M.mason_install()
|
||||
end
|
||||
|
||||
function M.ts.setup()
|
||||
@@ -329,12 +277,18 @@ function M.ts.setup()
|
||||
})
|
||||
end
|
||||
|
||||
function M.lsp.setup()
|
||||
for _, lsp_name in ipairs((M.general and M.general.language_servers) or {}) do
|
||||
vim.lsp.enable(lsp_name)
|
||||
end
|
||||
end
|
||||
|
||||
function M.lint.setup()
|
||||
vim.api.nvim_create_autocmd({ 'BufReadPre', 'BufNewFile' }, {
|
||||
group = M.group,
|
||||
callback = function()
|
||||
local lint = require('lint')
|
||||
lint.linters_by_ft = M.general.linters_by_ft or {}
|
||||
lint.linters_by_ft = (M.general and M.general.linters_by_ft) or {}
|
||||
vim.api.nvim_create_autocmd({ 'BufEnter', 'BufWritePost', 'InsertLeave' }, {
|
||||
group = vim.api.nvim_create_augroup('language-manager.lint', { clear = true }),
|
||||
callback = function()
|
||||
@@ -353,7 +307,7 @@ function M.format.setup()
|
||||
require('conform').setup({
|
||||
format_on_save = { timeout_ms = 500, lsp_format = 'fallback' },
|
||||
default_format_opts = { stop_after_first = true },
|
||||
formatters_by_ft = M.general.formatters_by_ft or {},
|
||||
formatters_by_ft = (M.general and M.general.formatters_by_ft) or {},
|
||||
})
|
||||
end,
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user