93 lines
2.8 KiB
Lua
93 lines
2.8 KiB
Lua
-- Custom yank functions for copying file paths and visual selections with file paths.
|
|
-- From: https://richardgill.org/blog/configuring-neovim-coding-agents
|
|
-- Source: https://github.com/richardgill/nix/blob/ebdd8260f770d242db7b6163158cfe9ad9784c41/modules/home-manager/dot-files/nvim/lua/custom/yank.lua
|
|
|
|
local M = {}
|
|
|
|
M.get_buffer_absolute = function()
|
|
return vim.fn.expand("%:p")
|
|
end
|
|
|
|
M.get_buffer_cwd_relative = function()
|
|
return vim.fn.expand("%:.")
|
|
end
|
|
|
|
M.get_visual_bounds = function()
|
|
local mode = vim.fn.mode()
|
|
if mode ~= "v" and mode ~= "V" then
|
|
error("get_visual_bounds must be called in visual or visual-line mode (current mode: " .. vim.inspect(mode) .. ")")
|
|
end
|
|
local is_visual_line_mode = mode == "V"
|
|
local start_pos = vim.fn.getpos("v")
|
|
local end_pos = vim.fn.getpos(".")
|
|
|
|
return {
|
|
start_line = math.min(start_pos[2], end_pos[2]),
|
|
end_line = math.max(start_pos[2], end_pos[2]),
|
|
start_col = is_visual_line_mode and 0 or math.min(start_pos[3], end_pos[3]) - 1,
|
|
end_col = is_visual_line_mode and -1 or math.max(start_pos[3], end_pos[3]),
|
|
mode = mode,
|
|
start_pos = start_pos,
|
|
end_pos = end_pos,
|
|
}
|
|
end
|
|
|
|
M.format_line_range = function(start_line, end_line)
|
|
return start_line == end_line and tostring(start_line) or start_line .. "-" .. end_line
|
|
end
|
|
|
|
M.simulate_yank_highlight = function()
|
|
local bounds = M.get_visual_bounds()
|
|
|
|
local ns = vim.api.nvim_create_namespace("simulate_yank_highlight")
|
|
vim.highlight.range(
|
|
0,
|
|
ns,
|
|
"IncSearch",
|
|
{ bounds.start_line - 1, bounds.start_col },
|
|
{ bounds.end_line - 1, bounds.end_col },
|
|
{ priority = 200 }
|
|
)
|
|
vim.defer_fn(function()
|
|
vim.api.nvim_buf_clear_namespace(0, ns, 0, -1)
|
|
end, 150)
|
|
end
|
|
|
|
M.exit_visual_mode = function()
|
|
vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes("<Esc>", true, false, true), "n", false)
|
|
end
|
|
|
|
M.yank_path = function(path, label)
|
|
vim.fn.setreg("+", path)
|
|
print("Yanked " .. label .. " path: " .. path)
|
|
end
|
|
|
|
M.yank_visual_with_path = function(path, label)
|
|
local bounds = M.get_visual_bounds()
|
|
|
|
local selected_lines = vim.fn.getregion(bounds.start_pos, bounds.end_pos, { type = bounds.mode })
|
|
local selected_text = table.concat(selected_lines, "\n")
|
|
|
|
local line_range = M.format_line_range(bounds.start_line, bounds.end_line)
|
|
local path_with_lines = path .. ":" .. line_range
|
|
|
|
local result = path_with_lines .. "\n\n" .. selected_text
|
|
vim.fn.setreg("+", result)
|
|
|
|
M.simulate_yank_highlight()
|
|
|
|
M.exit_visual_mode()
|
|
|
|
print("Yanked " .. label .. " with lines " .. line_range)
|
|
end
|
|
|
|
vim.keymap.set("v", "<leader>ya", function()
|
|
M.yank_visual_with_path(M.get_buffer_absolute(), "absolute")
|
|
end, { desc = "[Y]ank selection with [A]bsolute path" })
|
|
|
|
vim.keymap.set("v", "<leader>yr", function()
|
|
M.yank_visual_with_path(M.get_buffer_cwd_relative(), "relative")
|
|
end, { desc = "[Y]ank selection with [R]elative path" })
|
|
|
|
return M
|