Files
LazyVim/lua/lazyvim/util/lsp.lua

113 lines
3.1 KiB
Lua

---@class lazyvim.util.lsp
local M = {}
function M.get_clients(...)
---@diagnostic disable-next-line: deprecated
local fn = vim.lsp.get_clients or vim.lsp.get_active_clients
return fn(...)
end
---@param on_attach fun(client, buffer)
function M.on_attach(on_attach)
vim.api.nvim_create_autocmd("LspAttach", {
callback = function(args)
local buffer = args.buf ---@type number
local client = vim.lsp.get_client_by_id(args.data.client_id)
on_attach(client, buffer)
end,
})
end
---@param from string
---@param to string
function M.on_rename(from, to)
local clients = M.get_clients()
for _, client in ipairs(clients) do
if client.supports_method("workspace/willRenameFiles") then
---@diagnostic disable-next-line: invisible
local resp = client.request_sync("workspace/willRenameFiles", {
files = {
{
oldUri = vim.uri_from_fname(from),
newUri = vim.uri_from_fname(to),
},
},
}, 1000, 0)
if resp and resp.result ~= nil then
vim.lsp.util.apply_workspace_edit(resp.result, client.offset_encoding)
end
end
end
end
---@return _.lspconfig.options
function M.get_config(server)
local configs = require("lspconfig.configs")
return rawget(configs, server)
end
---@param server string
---@param cond fun( root_dir, config): boolean
function M.disable(server, cond)
local util = require("lspconfig.util")
local def = M.get_config(server)
---@diagnostic disable-next-line: undefined-field
def.document_config.on_new_config = util.add_hook_before(def.document_config.on_new_config, function(config, root_dir)
if cond(root_dir, config) then
config.enabled = false
end
end)
end
---@alias lsp.Client.filter fun(client: lsp.Client): boolean
---@param name string
---@return lsp.Client.filter
function M.filter(name)
return function(client)
return client.name == name
end
end
---@param opts? LazyFormatter| {filter?: (string|lsp.Client.filter), bufnr?: number}
function M.formatter(opts)
opts = opts or {}
local filter = opts.filter
filter = type(filter) == "string" and M.filter(filter) or filter
---@cast filter lsp.Client.filter?
---@type LazyFormatter
local ret = {
name = "LSP",
primary = true,
priority = 1,
format = function(buf)
M.format({ bufnr = buf, filter = filter })
end,
sources = function(buf)
local clients = M.get_clients({ bufnr = buf })
---@param client lsp.Client
local ret = vim.tbl_filter(function(client)
return (not filter or filter(client))
and (
client.supports_method("textDocument/formatting")
or client.supports_method("textDocument/rangeFormatting")
)
end, clients)
---@param client lsp.Client
return vim.tbl_map(function(client)
return client.name
end, ret)
end,
}
return vim.tbl_deep_extend("force", ret, opts) --[[@as LazyFormatter]]
end
---@param opts? {filter?: lsp.Client.filter, bufnr?: number}
function M.format(opts)
vim.lsp.buf.format(
vim.tbl_deep_extend("force", opts or {}, require("lazyvim.util").opts("nvim-lspconfig").format or {})
)
end
return M