perf(util): split lazyvim.util in smaller separate modules
This commit is contained in:
112
lua/lazyvim/util/lsp.lua
Normal file
112
lua/lazyvim/util/lsp.lua
Normal file
@ -0,0 +1,112 @@
|
||||
---@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
|
Reference in New Issue
Block a user