mroot/lazyvim
1
0
Fork 0
mirror of https://github.com/LazyVim/LazyVim.git synced 2026-07-21 20:11:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(lsp): cleaner support for code_actions

Browse source
This commit is contained in:
Folke Lemaitre 2026-03-20 19:01:02 +01:00
parent 53f4eabd77
commit 76cb567e5c
No known key found for this signature in database
GPG key ID: 9B52594D560070AB
3 changed files with 27 additions and 38 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lua/lazyvim/plugins/lsp/init.lua
View file

@ -103,7 +103,13 @@ return {
"<leader>co",
LazyVim.lsp.action["source.organizeImports"],
desc = "Organize Imports",
code_action = "source.organizeImports",
has = "codeAction",
enabled = function(buf)
local code_actions = vim.tbl_filter(function(action)
return action:find("^source%.organizeImports%.?$")
end, LazyVim.lsp.code_actions({ bufnr = buf }))
return #code_actions > 0
end
},
},
},

40
lua/lazyvim/plugins/lsp/keymaps.lua
View file

@ -3,8 +3,8 @@ local M = {}
---@type LazyKeysLspSpec[]|nil
M._keys = {}
---@alias LazyKeysLspSpec LazyKeysSpec|{has?:string|string[], enabled?:(fun():boolean), code_action?:string}
---@alias LazyKeysLsp LazyKeys|{has?:string|string[], code_action?:string, enabled?:fun():boolean}
---@alias LazyKeysLspSpec LazyKeysSpec|{has?:string|string[], enabled?:(fun(buf:number):boolean)}
---@alias LazyKeysLsp LazyKeys|{has?:string|string[], enabled?:fun(buf:number):boolean}
---@deprecated
---@return LazyKeysLspSpec[]
@ -44,41 +44,7 @@ function M.set(filter, spec)
for _, keys in pairs(Keys.resolve(spec)) do
---@cast keys LazyKeysLsp
local filters = {} ---@type vim.lsp.get_clients.Filter[]
if keys.code_action then
filter = vim.tbl_extend("force", vim.deepcopy(filter), { method = "textDocument/codeAction" })
filters[#filters + 1] = filter
---@param kinds? string[]
local function has(kinds)
for _, k in ipairs(kinds or {}) do
if type(k) == "string" and vim.startswith(k, keys.code_action) then
return true
end
end
end
local enabled = keys.enabled
keys.enabled = function()
if type(enabled) == "function" and not enabled() then
return false
end
local clients = vim.lsp.get_clients(filter)
for _, client in ipairs(clients) do
-- check server cababilities first
if has(vim.tbl_get(client, "server_capabilities", "codeActionProvider", "codeActionKinds")) then
return true
end
-- check dynamic capabilities
local regs = client.dynamic_capabilities:get("codeActionProvider")
for _, reg in ipairs(regs or {}) do
if has(vim.tbl_get(reg, "registerOptions", "codeActionKinds")) then
return true
end
end
end
return false
end
elseif keys.has then
if keys.has then
local methods = type(keys.has) == "string" and { keys.has } or keys.has --[[@as string[] ]]
for _, method in ipairs(methods) do
method = method:find("/") and method or ("textDocument/" .. method)

17
lua/lazyvim/util/lsp.lua
View file

@ -93,4 +93,21 @@ function M.execute(opts)
end
end
---@param filter? vim.lsp.get_clients.Filter
function M.code_actions(filter)
filter = filter or {}
local ret = {} ---@type string[]
local clients = vim.lsp.get_clients(filter)
for _, client in ipairs(clients) do
-- check server cababilities first
vim.list_extend(ret, vim.tbl_get(client, "server_capabilities", "codeActionProvider", "codeActionKinds") or {})
-- check dynamic capabilities
local regs = client.dynamic_capabilities:get("codeActionProvider", filter)
for _, reg in ipairs(regs or {}) do
vim.list_extend(ret, vim.tbl_get(reg, "registerOptions", "codeActionKinds") or {})
end
end
return LazyVim.dedup(ret)
end
return M