Files
u.nvim/examples/text-objects.lua
Jonathan Apodaca c1a8af8b7c
Some checks failed
ci / ci (push) Has been cancelled
v3
- range: extmarks/tsquery
- mise for dev env
- get rid of obsolete modules
- implement as single file module
2026-04-08 22:34:42 -06:00

106 lines
3.2 KiB
Lua

--
-- Custom text objects:
-- ag - whole file
-- a. - current line
-- aq/iq - nearest quote (any type)
-- a"/i", a'/i', a`/i` - specific quotes
-- al"/il", al'/il', al`/il` - "last" quote (searches backward)
-- alB/ilB, al]/il], alb/ilb, al>/il> - "last" bracket objects
--
local u = require 'u'
local Pos = u.Pos
local Range = u.Range
local M = {}
function M.setup()
-- Select whole file:
u.define_txtobj('ag', function() return Range.from_buf_text(0) end)
-- Select current line:
u.define_txtobj('a.', function() return Range.from_line(0, Pos.from_pos('.').lnum) end)
-- Select the nearest quote:
u.define_txtobj('aq', function() return Range.find_nearest_quotes() end)
u.define_txtobj('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
u.define_txtobj('a' .. q, function() return select_around() end)
u.define_txtobj('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
u.define_txtobj('al' .. q, function() return select_around() end)
u.define_txtobj('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
u.define_txtobj('al' .. k, function() return select_around() end)
u.define_txtobj('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