lvim/lua/user/codelens.lua at main · abzcoding/lvim · GitHub

History

158 lines (143 loc) · 4.97 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

local M = {}

local SIGN_GROUP = "nvim-lightbulblens"

local SIGN_NAME = "LightBulbSign"

-- Set default sign

if vim.tbl_isempty(vim.fn.sign_getdefined(SIGN_NAME)) then

vim.fn.sign_define(SIGN_NAME, { text = require("user.lsp_kind").icons.code_action, texthl = "MoreMsg" })

end

M.show_line_sign = function()

-- Check for code action capability

local code_action_cap_found = false

for _, client in pairs(vim.lsp.get_clients()) do

if client then

if client.supports_method "textDocument/codeAction" then

code_action_cap_found = true

end

if not code_action_cap_found then

return

end

local params = require("vim.lsp.util").make_range_params()

params.context = { diagnostics = vim.diagnostic.get(0, { lnum = params.range.start.line }) }

local bufnr = vim.api.nvim_get_current_buf()

vim.lsp.buf_request_all(0, "textDocument/codeAction", params, M.handler_factory(params.range.start.line, bufnr))

end

--- Check if the current line contains a codelens action

---@param cursor {number, number} The cursor position

---@return boolean True if the line contains a codelens action

M.code_lens_available = function(cursor)

local codelens_actions = {}

for _, l in ipairs(vim.lsp.codelens.get(0)) do

table.insert(codelens_actions, { start = l.range.start, finish = l.range["end"] })

end

for _, action in ipairs(codelens_actions) do

action.start.line <= cursor[1]

and cursor[1] <= action.finish.line

and action.start.character <= cursor[2]

and cursor[2] <= action.finish.character

then

return true

end

return false

end

--- Handler factory to keep track of current lightbulb line.

---

--- @param line number The line when the the code action request is called

--- @param bufnr number|nil Buffer handle

M.handler_factory = function(line, bufnr)

--- Handler for textDocument/codeAction.

---

--- See lsp-handler for more information.

---

--- @private

local function code_action_handler(responses)

local ignored_messages = {

"Extract Block",

"Extract Block To File",

"Extract Function",

"Extract Function To File",

"Extract Variable",

"Inline Variable",

}

-- Check for available code actions from all LSP server responses

local has_actions = false

for _, resp in pairs(responses) do

if resp.result and not vim.tbl_isempty(resp.result) then

for _, action in ipairs(resp.result) do

if not vim.tbl_contains(ignored_messages, action.title) then

has_actions = true

goto continue

end

::continue::

break

end

local cursor = vim.api.nvim_win_get_cursor(0)

cursor[1] = cursor[1] - 1

if M.code_lens_available(cursor) then

M.update_sign(10, vim.b.lightbulb_line, cursor[1] + 1, bufnr, "code_lens_action")

return

end

-- No available code actions

if not has_actions then

M.update_sign(10, vim.b.lightbulb_line, nil, bufnr, "code_action")

else

M.update_sign(10, vim.b.lightbulb_line, line + 1, bufnr, "code_action")

end

return M.mk_handler(code_action_handler)

end

--- Patch for breaking neovim master update to LSP handlers

--- See: https://github.com/neovim/neovim/issues/14090#issuecomment-913198455

M.mk_handler = function(fn)

return function(...)

local config_or_client_id = select(4, ...)

local is_new = type(config_or_client_id) ~= "number"

if is_new then

fn(...)

else

local err = select(1, ...)

local method = select(2, ...)

local result = select(3, ...)

local client_id = select(4, ...)

local bufnr = select(5, ...)

local config = select(6, ...)

fn(err, result, { method = method, client_id = client_id, bufnr = bufnr }, config)

end

--- Update sign position from `old_line` to `new_line`.

---

--- Either line can be optional, and will result in just adding/removing

--- the sign on the given line.

---

--- @param priority number The priority of the sign to add

--- @param old_line number|nil The line to remove the sign on

--- @param new_line number|nil The line to add the sign on

--- @param bufnr number|nil Buffer handle

--- @param text string the sign icon

---

M.update_sign = function(priority, old_line, new_line, bufnr, text)

bufnr = bufnr or 0

if old_line then

vim.fn.sign_unplace(SIGN_GROUP, { id = old_line, buffer = bufnr })

-- Update current lightbulb line

vim.b.lightbulb_line = nil

end

-- Avoid redrawing lightbulb if code action line did not change

if new_line and (vim.b.lightbulb_line ~= new_line) then

vim.fn.sign_place(new_line, SIGN_GROUP, SIGN_NAME, bufnr, { lnum = new_line, priority = priority })

-- Update current lightbulb line

vim.b.lightbulb_line = new_line

end

local icon = require("user.lsp_kind").icons.code_action

if text == "code_lens_action" then

icon = require("user.lsp_kind").icons.code_lens_action

end

vim.fn.sign_define(SIGN_NAME, { text = icon, texthl = "MoreMsg" })

end

return M

lvim/lua/user/codelens.lua at main · abzcoding/lvim · GitHub

Expand file tree

Breadcrumbs

codelens.lua

Latest commit

History

Breadcrumbs

codelens.lua

File metadata and controls