37 lines
1.3 KiB
Lua
37 lines
1.3 KiB
Lua
local ns = vim.api.nvim_create_namespace('highlight_undo')
|
|
|
|
vim.api.nvim_set_hl(0, 'HighlightUndo', { link = 'IncSearch', default = true })
|
|
|
|
for _, key in ipairs({ 'u', '<C-r>', 'U' }) do
|
|
vim.keymap.set('n', key, function()
|
|
vim.api.nvim_buf_attach(0, false, {
|
|
on_bytes = function(_, buf, _, sr, sc, _, _, _, _, ner, nec)
|
|
local er, ec = sr + ner, sc + nec
|
|
if er >= vim.api.nvim_buf_line_count(buf) then
|
|
ec = #(
|
|
vim.api.nvim_buf_get_lines(buf, -2, -1, false)[1] or ''
|
|
)
|
|
end
|
|
vim.schedule(function()
|
|
if not vim.api.nvim_buf_is_valid(buf) then
|
|
return
|
|
end
|
|
vim.hl.range(
|
|
buf,
|
|
ns,
|
|
'HighlightUndo',
|
|
{ sr, sc },
|
|
{ er, ec }
|
|
)
|
|
vim.defer_fn(function()
|
|
if vim.api.nvim_buf_is_valid(buf) then
|
|
vim.api.nvim_buf_clear_namespace(buf, ns, 0, -1)
|
|
end
|
|
end, 300)
|
|
end)
|
|
return true
|
|
end,
|
|
})
|
|
return key
|
|
end, { expr = true })
|
|
end
|