mirror of
https://github.com/tiyn/action-hints.nvim.git
synced 2025-11-13 13:59:45 +01:00
Switch to cursor movement update approach
This commit is contained in:
Rob Wilson
@@ -1,178 +1,248 @@
|
||||
local M = {}
|
||||
|
||||
local default_definition_color
|
||||
local default_references_color
|
||||
|
||||
if vim.o.termguicolors then
|
||||
default_definition_color = "#add8e6"
|
||||
default_references_color = "#ff6666"
|
||||
else
|
||||
default_definition_color = "blue"
|
||||
default_references_color = "red"
|
||||
end
|
||||
|
||||
M.config = {
|
||||
template = {
|
||||
{ " ⊛", "ActionHintsDefinition" },
|
||||
{ " ↱%s", "ActionHintsReferences" },
|
||||
},
|
||||
use_virtual_text = false,
|
||||
definition_color = "#add8e6",
|
||||
reference_color = "#ff6666",
|
||||
template = {
|
||||
definition = { text = " ⊛", color = default_definition_color },
|
||||
references = { text = " ↱%s", color = default_references_color },
|
||||
},
|
||||
use_virtual_text = false,
|
||||
}
|
||||
|
||||
M.references_available = false
|
||||
M.reference_count = 0
|
||||
M.definition_available = false
|
||||
|
||||
local references_namespace = vim.api.nvim_create_namespace("references")
|
||||
local references_namespace = vim.api.nvim_create_namespace("action_hints_references")
|
||||
local last_virtual_text_line = nil
|
||||
|
||||
local function debounce(func, delay)
|
||||
local timer_id = nil
|
||||
return function(...)
|
||||
if timer_id then
|
||||
vim.fn.timer_stop(timer_id)
|
||||
end
|
||||
local args = { ... }
|
||||
timer_id = vim.fn.timer_start(delay, function()
|
||||
func(unpack(args))
|
||||
end)
|
||||
end
|
||||
end
|
||||
|
||||
local function set_virtual_text(bufnr, line, chunks)
|
||||
if last_virtual_text_line then
|
||||
vim.api.nvim_buf_clear_namespace(
|
||||
bufnr,
|
||||
references_namespace,
|
||||
last_virtual_text_line,
|
||||
last_virtual_text_line + 1
|
||||
)
|
||||
end
|
||||
|
||||
if
|
||||
vim.api.nvim_buf_get_option(bufnr, "buftype") ~= ""
|
||||
or vim.api.nvim_buf_get_option(bufnr, "filetype") == "help"
|
||||
or vim.fn.bufname(bufnr) == ""
|
||||
then
|
||||
return
|
||||
end
|
||||
|
||||
vim.api.nvim_buf_set_virtual_text(bufnr, references_namespace, line, chunks, {})
|
||||
last_virtual_text_line = line
|
||||
end
|
||||
|
||||
local function references()
|
||||
if not M.supports_method("referencesProvider") then
|
||||
return
|
||||
end
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
local bufname = vim.api.nvim_buf_get_name(bufnr)
|
||||
|
||||
local params = {
|
||||
textDocument = { uri = bufname },
|
||||
position = { line = cursor[1] - 1, character = cursor[2] },
|
||||
context = { includeDeclaration = true },
|
||||
}
|
||||
|
||||
vim.lsp.buf_request(bufnr, "textDocument/references", params, function(err, result, _, _)
|
||||
if err or not result then
|
||||
M.references_available = false
|
||||
M.reference_count = 0
|
||||
return false
|
||||
end
|
||||
|
||||
if vim.tbl_count(result) > 0 then
|
||||
M.references_available = true
|
||||
M.reference_count = vim.tbl_count(result) - 1
|
||||
return true
|
||||
end
|
||||
|
||||
M.references_available = false
|
||||
M.reference_count = 0
|
||||
return false
|
||||
end)
|
||||
end
|
||||
|
||||
local function definition()
|
||||
if not M.supports_method("definitionProvider") then
|
||||
return
|
||||
end
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
local bufname = vim.api.nvim_buf_get_name(bufnr)
|
||||
|
||||
local params = {
|
||||
textDocument = { uri = bufname },
|
||||
position = { line = cursor[1] - 1, character = cursor[2] },
|
||||
}
|
||||
|
||||
vim.lsp.buf_request(bufnr, "textDocument/definition", params, function(err, result, _, _)
|
||||
if err or not result then
|
||||
M.definition_available = false
|
||||
return false
|
||||
end
|
||||
|
||||
if vim.tbl_count(result) > 0 then
|
||||
M.definition_available = true
|
||||
return true
|
||||
end
|
||||
|
||||
M.definition_available = false
|
||||
return false
|
||||
end)
|
||||
local timer_id = nil
|
||||
return function(...)
|
||||
if timer_id then
|
||||
vim.fn.timer_stop(timer_id)
|
||||
end
|
||||
local args = { ... }
|
||||
timer_id = vim.fn.timer_start(delay, function()
|
||||
func(unpack(args))
|
||||
end)
|
||||
end
|
||||
end
|
||||
|
||||
M.supports_method = function(method)
|
||||
local clients = vim.lsp.buf_get_clients()
|
||||
for _, client in pairs(clients) do
|
||||
if client.server_capabilities[method] then
|
||||
return true
|
||||
end
|
||||
end
|
||||
return false
|
||||
local clients = vim.lsp.buf_get_clients()
|
||||
for _, client in pairs(clients) do
|
||||
if client.server_capabilities[method] then
|
||||
return true
|
||||
end
|
||||
end
|
||||
return false
|
||||
end
|
||||
|
||||
local function set_virtual_text(bufnr, line, chunks)
|
||||
-- Clear the virtual text from the previous line
|
||||
if last_virtual_text_line then
|
||||
vim.api.nvim_buf_clear_namespace(bufnr, references_namespace, last_virtual_text_line, last_virtual_text_line + 1)
|
||||
end
|
||||
|
||||
-- Check for conditions where you might want to exit early
|
||||
if
|
||||
vim.api.nvim_buf_get_option(bufnr, "buftype") ~= ""
|
||||
or vim.api.nvim_buf_get_option(bufnr, "filetype") == "help"
|
||||
or vim.fn.bufname(bufnr) == ""
|
||||
then
|
||||
-- Reset the last virtual text line to ensure it gets cleared next time
|
||||
last_virtual_text_line = nil
|
||||
return
|
||||
end
|
||||
|
||||
-- Prepare the virtual text with proper highlight groups
|
||||
local virtual_text_chunks = {}
|
||||
for _, chunk in ipairs(chunks) do
|
||||
table.insert(virtual_text_chunks, { chunk[1], chunk[2] })
|
||||
end
|
||||
|
||||
-- Set the virtual text for the current line
|
||||
vim.api.nvim_buf_set_virtual_text(bufnr, references_namespace, line, virtual_text_chunks, {})
|
||||
last_virtual_text_line = line
|
||||
end
|
||||
|
||||
local function update_virtual_text()
|
||||
if M.config.use_virtual_text then
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
local definition_status = M.definition_available and M.config.template.definition.text or ""
|
||||
local reference_status = M.reference_count > 0
|
||||
and string.format(M.config.template.references.text, tostring(M.reference_count))
|
||||
or ""
|
||||
local chunks = {
|
||||
{ definition_status, "ActionHintsDefinition" },
|
||||
{ reference_status, "ActionHintsReferences" },
|
||||
}
|
||||
|
||||
set_virtual_text(bufnr, cursor[1] - 1, chunks)
|
||||
end
|
||||
end
|
||||
|
||||
local function is_cursor_on_whitespace()
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
local line = vim.api.nvim_buf_get_lines(bufnr, cursor[1] - 1, cursor[1], false)[1]
|
||||
local char = line:sub(cursor[2] + 1, cursor[2] + 1)
|
||||
return char:match("^%s$") ~= nil
|
||||
end
|
||||
|
||||
local function references()
|
||||
if not M.supports_method("referencesProvider") then
|
||||
return
|
||||
end
|
||||
|
||||
if is_cursor_on_whitespace() then
|
||||
M.clear_virtual_text()
|
||||
end
|
||||
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
local bufname = vim.uri_from_bufnr(bufnr)
|
||||
|
||||
local params = {
|
||||
textDocument = { uri = bufname },
|
||||
position = { line = cursor[1] - 1, character = cursor[2] },
|
||||
context = { includeDeclaration = true },
|
||||
}
|
||||
|
||||
vim.lsp.buf_request(bufnr, "textDocument/references", params, function(err, result, _, _)
|
||||
if err or not result then
|
||||
M.references_available = false
|
||||
M.reference_count = 0
|
||||
M.clear_virtual_text()
|
||||
return
|
||||
end
|
||||
|
||||
if vim.tbl_count(result) > 0 then
|
||||
M.references_available = true
|
||||
M.reference_count = vim.tbl_count(result) - 1
|
||||
update_virtual_text()
|
||||
return
|
||||
end
|
||||
|
||||
M.references_available = false
|
||||
M.reference_count = 0
|
||||
M.clear_virtual_text()
|
||||
end)
|
||||
end
|
||||
|
||||
local function definition()
|
||||
if not M.supports_method("definitionProvider") then
|
||||
return
|
||||
end
|
||||
|
||||
if is_cursor_on_whitespace() then
|
||||
M.clear_virtual_text()
|
||||
end
|
||||
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
local bufname = vim.uri_from_bufnr(bufnr)
|
||||
|
||||
local params = {
|
||||
textDocument = { uri = bufname },
|
||||
position = { line = cursor[1] - 1, character = cursor[2] },
|
||||
}
|
||||
|
||||
vim.lsp.buf_request(bufnr, "textDocument/definition", params, function(err, result, _, _)
|
||||
if err or not result then
|
||||
M.definition_available = false
|
||||
M.clear_virtual_text()
|
||||
return
|
||||
end
|
||||
|
||||
if vim.tbl_count(result) > 0 then
|
||||
M.definition_available = true
|
||||
update_virtual_text()
|
||||
return
|
||||
end
|
||||
|
||||
M.clear_virtual_text()
|
||||
M.definition_available = false
|
||||
end)
|
||||
end
|
||||
|
||||
M.clear_virtual_text = function()
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
if last_virtual_text_line then
|
||||
vim.api.nvim_buf_clear_namespace(bufnr, references_namespace, last_virtual_text_line, last_virtual_text_line + 1)
|
||||
last_virtual_text_line = nil
|
||||
end
|
||||
end
|
||||
|
||||
local debounced_references = debounce(references, 100)
|
||||
local debounced_definition = debounce(definition, 100)
|
||||
|
||||
M.update = function()
|
||||
debounced_references()
|
||||
debounced_definition()
|
||||
debounced_references()
|
||||
debounced_definition()
|
||||
end
|
||||
|
||||
M.statusline = function()
|
||||
local definition_status = M.definition_available and string.format(M.config.template[1][1], "") or ""
|
||||
local reference_status = M.reference_count > 0
|
||||
and string.format(M.config.template[2][1], tostring(M.reference_count))
|
||||
or ""
|
||||
local chunks = {
|
||||
{ definition_status, M.config.template[1][2] },
|
||||
{ reference_status, M.config.template[2][2] },
|
||||
}
|
||||
local definition_status = M.definition_available and M.config.template.definition.text or ""
|
||||
local reference_status = M.reference_count > 0
|
||||
and string.format(M.config.template.references.text, tostring(M.reference_count))
|
||||
or ""
|
||||
local chunks = {
|
||||
{ definition_status, "ActionHintsDefinition" },
|
||||
{ reference_status, "ActionHintsReferences" },
|
||||
}
|
||||
|
||||
if M.config.use_virtual_text then
|
||||
local bufnr = vim.api.nvim_get_current_buf()
|
||||
local cursor = vim.api.nvim_win_get_cursor(0)
|
||||
set_virtual_text(bufnr, cursor[1] - 1, chunks)
|
||||
end
|
||||
local text = ""
|
||||
for _, chunk in ipairs(chunks) do
|
||||
text = text .. chunk[1]
|
||||
end
|
||||
|
||||
local text = ""
|
||||
for i, chunk in ipairs(chunks) do
|
||||
text = text .. chunk[1]
|
||||
end
|
||||
return text
|
||||
end
|
||||
|
||||
return text
|
||||
M.set_highlight = function()
|
||||
if vim.o.termguicolors then
|
||||
vim.api.nvim_command("highlight ActionHintsDefinition ctermfg=NONE guifg=" .. M.config.template.definition.color)
|
||||
vim.api.nvim_command("highlight ActionHintsReferences ctermfg=NONE guifg=" .. M.config.template.references.color)
|
||||
else
|
||||
vim.api.nvim_command("highlight ActionHintsDefinition guifg=NONE ctermfg=" .. M.config.template.definition.color)
|
||||
vim.api.nvim_command("highlight ActionHintsReferences guifg=NONE ctermfg=" .. M.config.template.references.color)
|
||||
end
|
||||
end
|
||||
|
||||
M.setup = function(options)
|
||||
if options == nil then
|
||||
options = {}
|
||||
end
|
||||
if options == nil then
|
||||
options = {}
|
||||
end
|
||||
|
||||
for k, v in pairs(options) do
|
||||
M.config[k] = v
|
||||
end
|
||||
-- Merge keys
|
||||
for k, v in pairs(options) do
|
||||
if k == "template" and type(v) == "table" then
|
||||
for tk, tv in pairs(v) do
|
||||
M.config.template[tk] = tv
|
||||
end
|
||||
else
|
||||
M.config[k] = v
|
||||
end
|
||||
end
|
||||
|
||||
local defColor = M.config.definition_color
|
||||
local refColor = M.config.reference_color
|
||||
M.set_highlight()
|
||||
|
||||
vim.api.nvim_command("highlight ActionHintsDefinition guifg=" .. defColor)
|
||||
vim.api.nvim_command("highlight ActionHintsReferences guifg=" .. refColor)
|
||||
|
||||
vim.api.nvim_command([[autocmd CursorMoved * lua require("action-hints").update()]])
|
||||
vim.api.nvim_command([[autocmd CursorMovedI * lua require("action-hints").update()]])
|
||||
vim.api.nvim_command([[autocmd OptionSet termguicolors lua require("action-hints").set_highlight()]])
|
||||
vim.api.nvim_command([[autocmd CursorMoved * lua require("action-hints").update()]])
|
||||
vim.api.nvim_command([[autocmd CursorMovedI * lua require("action-hints").update()]])
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
Reference in New Issue
Block a user