81 lines
2.1 KiB
Lua
81 lines
2.1 KiB
Lua
local M = {}
|
|
|
|
local function escape_lua_pattern(s)
|
|
local matches = {
|
|
["^"] = "%^",
|
|
["$"] = "%$",
|
|
["("] = "%(",
|
|
[")"] = "%)",
|
|
["%"] = "%%",
|
|
["."] = "%.",
|
|
["["] = "%[",
|
|
["]"] = "%]",
|
|
["*"] = "%*",
|
|
["+"] = "%+",
|
|
["-"] = "%-",
|
|
["?"] = "%?",
|
|
}
|
|
return s:gsub(".", matches)
|
|
end
|
|
|
|
local targets = { '"', "'", "(", ")", "{", "}", "[", "]", "`" }
|
|
|
|
function M.setup(config)
|
|
vim.validate({ config = { config, "table", true } })
|
|
vim.validate({ additional_targets = { config.additional_targets, "table", true } })
|
|
vim.validate({ targets = { config.targets, "table", true } })
|
|
--
|
|
if config and config.targets then
|
|
targets = config.targets
|
|
return
|
|
end
|
|
|
|
if config and config.additional_targets then
|
|
for _, target in ipairs(config.additional_targets) do
|
|
targets[#targets + 1] = target
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.in_and_out()
|
|
local current_row, current_col = unpack(vim.api.nvim_win_get_cursor(0))
|
|
local lines_in_buf = vim.api.nvim_buf_get_lines(0, current_row - 1, -1, false)
|
|
|
|
local target_row = 0
|
|
local target_col = nil
|
|
for i, line in ipairs(lines_in_buf) do
|
|
for _, char in ipairs(targets) do
|
|
local found_col = string.find(line, escape_lua_pattern(char), current_col + 1)
|
|
if found_col and (not target_col or found_col < target_col) then
|
|
-- If char is a multibyte character, we need to take into account the extra bytes.
|
|
target_col = found_col + vim.fn.strlen(char) - 1
|
|
end
|
|
end
|
|
if target_col then
|
|
target_row = current_row + i - 1
|
|
break
|
|
end
|
|
current_col = 0
|
|
end
|
|
if target_col then
|
|
vim.api.nvim_win_set_cursor(0, { target_row, target_col })
|
|
end
|
|
end
|
|
|
|
function M.in_and_out_cmd()
|
|
local current_pos = vim.fn.getcmdpos()
|
|
local target_pos = nil
|
|
local line = vim.fn.getcmdline()
|
|
for _, char in ipairs(targets) do
|
|
local found_pos = string.find(line, escape_lua_pattern(char), current_pos)
|
|
if found_pos and (not target_pos or found_pos < target_pos) then
|
|
-- If char is a multibyte character, we need to take into account the extra bytes.
|
|
target_pos = found_pos + vim.fn.strlen(char)
|
|
end
|
|
end
|
|
if target_pos then
|
|
vim.fn.setcmdline(line, target_pos)
|
|
end
|
|
end
|
|
return M
|