feat: automated setup works finally

This commit is contained in:
2025-10-26 02:53:13 +03:00
parent 2692b95109
commit cd5025859c
7 changed files with 135 additions and 168 deletions

View File

@@ -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,
})