u.nvim/examples/text-objects.lua
Jonathan Apodaca 4dfadf97e5
All checks were successful
NeoVim tests / plenary-tests (push) Successful in 10s
update lua API to 1-based indices; add renderer
2025-05-04 15:22:19 -06:00

98 lines
3.0 KiB
Lua

local txtobj = require 'u.txtobj'
local Pos = require 'u.pos'
local Range = require 'u.range'
local Buffer = require 'u.buffer'
local M = {}
function M.setup()
-- Select whole file:
txtobj.define('ag', function() return Buffer.current():all() end)
-- Select current line:
txtobj.define('a.', function() return Buffer.current():line(Pos.from_pos('.').lnum) end)
-- Select the nearest quote:
txtobj.define('aq', function() return Range.find_nearest_quotes() end)
txtobj.define('iq', function()
local range = Range.find_nearest_quotes()
if range == nil then return end
return range:shrink(1)
end)
---Selects the next quote object (searches forward)
--- @param q string
local function define_quote_obj(q)
local function select_around() return Range.from_motion('a' .. q) end
txtobj.define('a' .. q, function() return select_around() end)
txtobj.define('i' .. q, function()
local range = select_around()
if range == nil or range:is_empty() then return range end
local start_next = range.start:next(1) or range.start
local stop_prev = range.stop:next(-1)
if start_next > stop_prev then return Range.new(start_next) end
return range:shrink(1) or range
end)
end
define_quote_obj [["]]
define_quote_obj [[']]
define_quote_obj [[`]]
---Selects the "last" quote object (searches backward)
--- @param q string
local function define_last_quote_obj(q)
local function select_around()
local curr = Pos.from_pos('.'):find_next(-1, q)
if not curr then return end
-- Reset visual selection to current context:
curr:save_to_pos '.'
return Range.from_motion('a' .. q)
end
txtobj.define('al' .. q, function() return select_around() end)
txtobj.define('il' .. q, function()
local range = select_around()
if range == nil or range:is_empty() then return range end
local start_next = range.start:next(1) or range.start
local stop_prev = range.stop:next(-1)
if start_next > stop_prev then return Range.new(start_next) end
return range:shrink(1) or range
end)
end
define_last_quote_obj [["]]
define_last_quote_obj [[']]
define_last_quote_obj [[`]]
-- Selects the "last" bracket object (searches backward):
local function define_last_bracket_obj(b, ...)
local function select_around()
local curr = Pos.from_pos('.'):find_next(-1, b)
if not curr then return end
local other = curr:find_match(1000)
if not other then return end
return Range.new(curr, other)
end
local keybinds = { ... }
table.insert(keybinds, b)
for _, k in ipairs(keybinds) do
txtobj.define('al' .. k, function() return select_around() end)
txtobj.define('il' .. k, function()
local range = select_around()
return range and range:shrink(1)
end)
end
end
define_last_bracket_obj('}', 'B')
define_last_bracket_obj ']'
define_last_bracket_obj(')', 'b')
define_last_bracket_obj '>'
end
return M