mirror of
https://github.com/LunarVim/LunarVim.git
synced 2024-09-21 02:55:09 +02:00
194 lines
5.7 KiB
Lua
194 lines
5.7 KiB
Lua
local M = {}
|
|
|
|
local tbl = require "lvim.utils.table"
|
|
|
|
function M.is_client_active(name)
|
|
local clients = vim.lsp.get_active_clients()
|
|
return tbl.find_first(clients, function(client)
|
|
return client.name == name
|
|
end)
|
|
end
|
|
|
|
function M.get_active_clients_by_ft(filetype)
|
|
local matches = {}
|
|
local clients = vim.lsp.get_active_clients()
|
|
for _, client in pairs(clients) do
|
|
local supported_filetypes = client.config.filetypes or {}
|
|
if client.name ~= "null-ls" and vim.tbl_contains(supported_filetypes, filetype) then
|
|
table.insert(matches, client)
|
|
end
|
|
end
|
|
return matches
|
|
end
|
|
|
|
function M.get_client_capabilities(client_id)
|
|
local client
|
|
if not client_id then
|
|
local buf_clients = vim.lsp.buf_get_clients()
|
|
for _, buf_client in pairs(buf_clients) do
|
|
if buf_client.name ~= "null-ls" then
|
|
client = buf_client
|
|
break
|
|
end
|
|
end
|
|
else
|
|
client = vim.lsp.get_client_by_id(tonumber(client_id))
|
|
end
|
|
if not client then
|
|
error "Unable to determine client_id"
|
|
return
|
|
end
|
|
|
|
local enabled_caps = {}
|
|
for capability, status in pairs(client.server_capabilities or client.resolved_capabilities) do
|
|
if status == true then
|
|
table.insert(enabled_caps, capability)
|
|
end
|
|
end
|
|
|
|
return enabled_caps
|
|
end
|
|
|
|
---Get supported filetypes per server
|
|
---@param server_name string can be any server supported by nvim-lsp-installer
|
|
---@return table supported filestypes as a list of strings
|
|
function M.get_supported_filetypes(server_name)
|
|
local status_ok, lsp_installer_servers = pcall(require, "nvim-lsp-installer.servers")
|
|
if not status_ok then
|
|
return {}
|
|
end
|
|
|
|
local server_available, requested_server = lsp_installer_servers.get_server(server_name)
|
|
if not server_available then
|
|
return {}
|
|
end
|
|
|
|
return requested_server:get_supported_filetypes()
|
|
end
|
|
|
|
---Get supported servers per filetype
|
|
---@param filetype string
|
|
---@return table list of names of supported servers
|
|
function M.get_supported_servers_per_filetype(filetype)
|
|
local filetype_server_map = require "nvim-lsp-installer._generated.filetype_map"
|
|
return filetype_server_map[filetype]
|
|
end
|
|
|
|
---Get all supported filetypes by nvim-lsp-installer
|
|
---@return table supported filestypes as a list of strings
|
|
function M.get_all_supported_filetypes()
|
|
local status_ok, lsp_installer_filetypes = pcall(require, "nvim-lsp-installer._generated.filetype_map")
|
|
if not status_ok then
|
|
return {}
|
|
end
|
|
return vim.tbl_keys(lsp_installer_filetypes or {})
|
|
end
|
|
|
|
function M.setup_document_highlight(client, bufnr)
|
|
local status_ok, highlight_supported = pcall(function()
|
|
return client.supports_method "textDocument/documentHighlight"
|
|
end)
|
|
if not status_ok or not highlight_supported then
|
|
return
|
|
end
|
|
local augroup_exist, _ = pcall(vim.api.nvim_get_autocmds, {
|
|
group = "lsp_document_highlight",
|
|
})
|
|
if not augroup_exist then
|
|
vim.api.nvim_create_augroup("lsp_document_highlight", {})
|
|
end
|
|
vim.api.nvim_create_autocmd({ "CursorHold", "CursorHoldI" }, {
|
|
group = "lsp_document_highlight",
|
|
buffer = bufnr,
|
|
callback = vim.lsp.buf.document_highlight,
|
|
})
|
|
vim.api.nvim_create_autocmd("CursorMoved", {
|
|
group = "lsp_document_highlight",
|
|
buffer = bufnr,
|
|
callback = vim.lsp.buf.clear_references,
|
|
})
|
|
end
|
|
|
|
function M.setup_codelens_refresh(client, bufnr)
|
|
local status_ok, codelens_supported = pcall(function()
|
|
return client.supports_method "textDocument/codeLens"
|
|
end)
|
|
if not status_ok or not codelens_supported then
|
|
return
|
|
end
|
|
local augroup_exist, _ = pcall(vim.api.nvim_get_autocmds, {
|
|
group = "lsp_code_lens_refresh",
|
|
})
|
|
if not augroup_exist then
|
|
vim.api.nvim_create_augroup("lsp_code_lens_refresh", {})
|
|
end
|
|
vim.api.nvim_create_autocmd({ "BufEnter", "InsertLeave" }, {
|
|
group = "lsp_code_lens_refresh",
|
|
buffer = bufnr,
|
|
callback = vim.lsp.codelens.refresh,
|
|
})
|
|
end
|
|
|
|
---filter passed to vim.lsp.buf.format
|
|
---gives higher priority to null-ls
|
|
---@param clients table clients attached to a buffer
|
|
---@return table chosen clients
|
|
function M.format_filter(clients)
|
|
return vim.tbl_filter(function(client)
|
|
local status_ok, formatting_supported = pcall(function()
|
|
return client.supports_method "textDocument/formatting"
|
|
end)
|
|
-- give higher prio to null-ls
|
|
if status_ok and formatting_supported and client.name == "null-ls" then
|
|
return "null-ls"
|
|
else
|
|
return status_ok and formatting_supported and client.name
|
|
end
|
|
end, clients)
|
|
end
|
|
|
|
---Provide vim.lsp.buf.format for nvim <0.8
|
|
---@param opts table
|
|
function M.format(opts)
|
|
opts = opts or { filter = M.format_filter }
|
|
|
|
if vim.lsp.buf.format then
|
|
return vim.lsp.buf.format(opts)
|
|
end
|
|
|
|
local bufnr = opts.bufnr or vim.api.nvim_get_current_buf()
|
|
local clients = vim.lsp.buf_get_clients(bufnr)
|
|
|
|
if opts.filter then
|
|
clients = opts.filter(clients)
|
|
elseif opts.id then
|
|
clients = vim.tbl_filter(function(client)
|
|
return client.id == opts.id
|
|
end, clients)
|
|
elseif opts.name then
|
|
clients = vim.tbl_filter(function(client)
|
|
return client.name == opts.name
|
|
end, clients)
|
|
end
|
|
|
|
clients = vim.tbl_filter(function(client)
|
|
return client.supports_method "textDocument/formatting"
|
|
end, clients)
|
|
|
|
if #clients == 0 then
|
|
vim.notify_once "[LSP] Format request failed, no matching language servers."
|
|
end
|
|
|
|
local timeout_ms = opts.timeout_ms or 1000
|
|
for _, client in pairs(clients) do
|
|
local params = vim.lsp.util.make_formatting_params(opts.formatting_options)
|
|
local result, err = client.request_sync("textDocument/formatting", params, timeout_ms, bufnr)
|
|
if result and result.result then
|
|
vim.lsp.util.apply_text_edits(result.result, bufnr, client.offset_encoding)
|
|
elseif err then
|
|
vim.notify(string.format("[LSP][%s] %s", client.name, err), vim.log.levels.WARN)
|
|
end
|
|
end
|
|
end
|
|
|
|
return M
|