feat: language-manager
This commit is contained in:
128
lua/custom/language-manager.lua
Normal file
128
lua/custom/language-manager.lua
Normal file
@@ -0,0 +1,128 @@
|
||||
--[[
|
||||
local lm = require('language_manager')
|
||||
lm.install() or .install({'ts_parsers','language_servers','linters','formatters'})
|
||||
- treesitter.install specs.ts_parsers
|
||||
- mason.install mason-registry.get_package(name):install()
|
||||
lm.generate_specs() or {use_cache=true, only_installed=true}
|
||||
lm.delete_cache()
|
||||
lm.enable()
|
||||
--]]
|
||||
|
||||
local M = {}
|
||||
|
||||
local function wrap(item)
|
||||
if type(item) == 'string' then
|
||||
return { item }
|
||||
elseif type(item) == 'table' then
|
||||
return item
|
||||
else
|
||||
return {}
|
||||
end
|
||||
end
|
||||
|
||||
local function create_set()
|
||||
local seen = {}
|
||||
return function(ref, item) -- unique()
|
||||
if not seen[ref] then
|
||||
seen[ref] = {}
|
||||
end
|
||||
if item and not seen[ref][item] then
|
||||
seen[ref][item] = true
|
||||
return true
|
||||
end
|
||||
return false
|
||||
end
|
||||
end
|
||||
|
||||
local function create_spec()
|
||||
local M = {}
|
||||
local unique = create_set()
|
||||
local specs = {}
|
||||
|
||||
function M.set(item, group)
|
||||
specs[group] = item
|
||||
end
|
||||
|
||||
function M.get()
|
||||
return specs
|
||||
end
|
||||
|
||||
function M.add(list, ...)
|
||||
local groups = { ... }
|
||||
local ref = table.concat(groups, '.')
|
||||
local pointer = specs
|
||||
local last_i = #groups
|
||||
|
||||
for i = 1, last_i do
|
||||
local group = groups[i]
|
||||
|
||||
if i == last_i then
|
||||
for _, item in ipairs(wrap(list)) do
|
||||
if unique(ref, item) then
|
||||
if not pointer[group] then
|
||||
pointer[group] = {}
|
||||
end
|
||||
table.insert(pointer[group], item)
|
||||
end
|
||||
end
|
||||
else
|
||||
if not pointer[group] then
|
||||
pointer[group] = {}
|
||||
end
|
||||
pointer = pointer[group]
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
end
|
||||
|
||||
function M.generate_specs(specs_raw)
|
||||
local specs = create_spec()
|
||||
|
||||
for _, spec in ipairs(specs_raw) do
|
||||
-- <filetype> -> { ft = <filetype> }
|
||||
if type(spec) == 'string' then
|
||||
spec = { ft = spec }
|
||||
end
|
||||
|
||||
-- Filetype = TS Parser
|
||||
if not spec.ts then
|
||||
spec.ts = spec.ft
|
||||
end
|
||||
|
||||
specs.add(spec.ts, 'ts_parsers')
|
||||
specs.add(spec.lsp, 'language_servers')
|
||||
|
||||
for _, filetype in ipairs(wrap(spec.ft)) do
|
||||
specs.add(spec.lint, 'linters_by_ft', filetype)
|
||||
specs.add(spec.format, 'formatters_by_ft', filetype)
|
||||
end
|
||||
end
|
||||
|
||||
M.general = specs.get()
|
||||
return M.general
|
||||
end
|
||||
|
||||
M.lsp = {}
|
||||
function M.lsp.enable()
|
||||
local registry = require('mason-registry')
|
||||
local lsp_configs = {}
|
||||
|
||||
for _, lsp in ipairs(M.general.language_servers) do
|
||||
if registry.has_package(lsp) then
|
||||
local pkg = registry.get_package(lsp)
|
||||
local spec = pkg.spec and pkg.spec.neovim
|
||||
local lsp_name = (spec and spec.lspconfig) or lsp
|
||||
table.insert(lsp_configs, lsp_name)
|
||||
|
||||
vim.lsp.enable(lsp_name)
|
||||
else
|
||||
vim.notify('Unknown LSP: ' .. lsp, vim.log.levels.WARN)
|
||||
end
|
||||
end
|
||||
|
||||
vim.notify('Enabled LSPs: ' .. table.concat(lsp_configs, ', '), vim.log.levels.INFO)
|
||||
end
|
||||
|
||||
return M
|
||||
98
lua/custom/lm-cmds.lua
Normal file
98
lua/custom/lm-cmds.lua
Normal file
@@ -0,0 +1,98 @@
|
||||
local general = {}
|
||||
---------------------------------------------------------------------
|
||||
-- Mason Install Command
|
||||
---------------------------------------------------------------------
|
||||
vim.api.nvim_create_user_command('MasonInstallAll', function()
|
||||
local registry = require('mason-registry')
|
||||
local list = vim.list_extend(vim.list_extend({}, general.lsps), {})
|
||||
|
||||
for _, ftmap in pairs({ general.linters_by_ft, general.formatters_by_ft }) do
|
||||
for _, tools in pairs(ftmap) do
|
||||
vim.list_extend(list, tools)
|
||||
end
|
||||
end
|
||||
|
||||
list = uniq(list)
|
||||
local installed = {}
|
||||
for _, pkg in ipairs(registry.get_installed_packages()) do
|
||||
installed[pkg.name] = true
|
||||
end
|
||||
|
||||
for _, name in ipairs(list) do
|
||||
if registry.has_package(name) then
|
||||
if not installed[name] then
|
||||
vim.notify('Installing ' .. name, vim.log.levels.INFO)
|
||||
registry.get_package(name):install()
|
||||
else
|
||||
vim.notify('Already installed ' .. name, vim.log.levels.INFO)
|
||||
end
|
||||
else
|
||||
vim.notify('Package not found in registry: ' .. name, vim.log.levels.WARN)
|
||||
end
|
||||
end
|
||||
end, { desc = 'Install all Mason LSPs, linters, and formatters' })
|
||||
|
||||
---------------------------------------------------------------------
|
||||
-- Fetch LSP default configs from nvim-lspconfig
|
||||
---------------------------------------------------------------------
|
||||
vim.api.nvim_create_user_command('FetchLspConfigs', function()
|
||||
local registry = require('mason-registry')
|
||||
local lspconfig_names = {}
|
||||
|
||||
for _, lsp in ipairs(general.lsps) do
|
||||
if registry.has_package(lsp) then
|
||||
local pkg = registry.get_package(lsp)
|
||||
local spec = pkg.spec and pkg.spec.neovim
|
||||
if spec and spec.lspconfig then
|
||||
table.insert(lspconfig_names, spec.lspconfig)
|
||||
else
|
||||
table.insert(lspconfig_names, lsp)
|
||||
end
|
||||
else
|
||||
table.insert(lspconfig_names, lsp)
|
||||
end
|
||||
end
|
||||
|
||||
lspconfig_names = uniq(lspconfig_names)
|
||||
|
||||
-- base URL same as your original
|
||||
local base_url = 'https://raw.githubusercontent.com/neovim/nvim-lspconfig/master/lsp/'
|
||||
-- write to current working directory
|
||||
local lsp_dir = vim.fs.joinpath(vim.fn.getcwd(), 'lsp')
|
||||
vim.fn.mkdir(lsp_dir, 'p')
|
||||
|
||||
for _, name in ipairs(lspconfig_names) do
|
||||
local file = vim.fs.joinpath(lsp_dir, name .. '.lua')
|
||||
if vim.fn.filereadable(file) == 0 then
|
||||
local url = base_url .. name .. '.lua'
|
||||
local cmd = string.format('curl -fsSL -o %q %q', file, url)
|
||||
vim.fn.system(cmd)
|
||||
if vim.v.shell_error ~= 0 then
|
||||
vim.notify('Failed to fetch ' .. name .. '.lua', vim.log.levels.ERROR)
|
||||
vim.fn.delete(file)
|
||||
else
|
||||
vim.notify('Fetched ' .. name .. '.lua', vim.log.levels.INFO)
|
||||
end
|
||||
else
|
||||
vim.notify('Skipped existing ' .. name .. '.lua', vim.log.levels.INFO)
|
||||
end
|
||||
end
|
||||
end, { desc = 'Fetch default LSP configs into ./lsp in cwd' })
|
||||
|
||||
vim.api.nvim_create_user_command('TreesitterInstallAll', function()
|
||||
local parsers = require('nvim-treesitter.parsers')
|
||||
local configs = require('nvim-treesitter.configs')
|
||||
local langs = configs.get_module('ensure_installed') or {}
|
||||
if type(langs) == 'string' and langs == 'all' then
|
||||
vim.notify('Treesitter ensure_installed = "all" not supported here', vim.log.levels.WARN)
|
||||
return
|
||||
end
|
||||
|
||||
for _, lang in ipairs(langs) do
|
||||
if not parsers.has_parser(lang) then
|
||||
vim.cmd('TSInstall ' .. lang)
|
||||
else
|
||||
vim.notify('Parser already installed: ' .. lang, vim.log.levels.INFO)
|
||||
end
|
||||
end
|
||||
end, { desc = 'Install all Treesitter parsers defined in ensure_installed' })
|
||||
Reference in New Issue
Block a user