Files
neovim/runtime/lua/vim/lsp/_dynamic.lua
Lewis Russell 9f8c96240d refactor(lsp): resolve the config-client entanglement
Previously the LSP-Client object contained some fields that are also
in the client config, but for a lot of other fields, the config was used
directly making the two objects vaguely entangled with either not having
a clear role.

Now the config object is treated purely as config (read-only) from the
client, and any fields the client needs from the config are now copied
in as additional fields.

This means:
- the config object is no longet normalised and is left as the user
  provided it.
- the client only reads the config on creation of the client and all
  other implementations now read the clients version of the fields.

In addition, internal support for multiple callbacks has been added to
the client so the client tracking logic (done in lua.lsp) can be done
more robustly instead of wrapping the user callbacks which may error.
2024-02-13 14:49:20 +00:00

115 lines
3.0 KiB
Lua

local glob = vim.glob
--- @class lsp.DynamicCapabilities
--- @field capabilities table<string, lsp.Registration[]>
--- @field client_id number
local M = {}
--- @param client_id number
--- @return lsp.DynamicCapabilities
function M.new(client_id)
return setmetatable({
capabilities = {},
client_id = client_id,
}, { __index = M })
end
function M:supports_registration(method)
local client = vim.lsp.get_client_by_id(self.client_id)
if not client then
return false
end
local capability = vim.tbl_get(client.capabilities, unpack(vim.split(method, '/')))
return type(capability) == 'table' and capability.dynamicRegistration
end
--- @param registrations lsp.Registration[]
--- @package
function M:register(registrations)
-- remove duplicates
self:unregister(registrations)
for _, reg in ipairs(registrations) do
local method = reg.method
if not self.capabilities[method] then
self.capabilities[method] = {}
end
table.insert(self.capabilities[method], reg)
end
end
--- @param unregisterations lsp.Unregistration[]
--- @package
function M:unregister(unregisterations)
for _, unreg in ipairs(unregisterations) do
local method = unreg.method
if not self.capabilities[method] then
return
end
local id = unreg.id
for i, reg in ipairs(self.capabilities[method]) do
if reg.id == id then
table.remove(self.capabilities[method], i)
break
end
end
end
end
--- @param method string
--- @param opts? {bufnr: integer?}
--- @return lsp.Registration? (table|nil) the registration if found
--- @private
function M:get(method, opts)
opts = opts or {}
opts.bufnr = opts.bufnr or vim.api.nvim_get_current_buf()
for _, reg in ipairs(self.capabilities[method] or {}) do
if not reg.registerOptions then
return reg
end
local documentSelector = reg.registerOptions.documentSelector
if not documentSelector then
return reg
end
if self:match(opts.bufnr, documentSelector) then
return reg
end
end
end
--- @param method string
--- @param opts? {bufnr: integer?}
--- @package
function M:supports(method, opts)
return self:get(method, opts) ~= nil
end
--- @param bufnr number
--- @param documentSelector lsp.DocumentSelector
--- @private
function M:match(bufnr, documentSelector)
local client = vim.lsp.get_client_by_id(self.client_id)
if not client then
return false
end
local language = client.get_language_id(bufnr, vim.bo[bufnr].filetype)
local uri = vim.uri_from_bufnr(bufnr)
local fname = vim.uri_to_fname(uri)
for _, filter in ipairs(documentSelector) do
local matches = true
if filter.language and language ~= filter.language then
matches = false
end
if matches and filter.scheme and not vim.startswith(uri, filter.scheme .. ':') then
matches = false
end
if matches and filter.pattern and not glob.to_lpeg(filter.pattern):match(fname) then
matches = false
end
if matches then
return true
end
end
end
return M