Files
neovim/test/functional/editor/atom_testutil.lua
Justin M. Keyes f6bf814378 fixfix(cmdatom): repeat Visual <Cmd>; CmdFrame + stage #41355
Problem:
- Dot-repeat of a Visual selection prepared by a `<Cmd>` mapping,
  results in E1255 and leaves Visual mode active.
- Visual-mode capture is implemented as a second, parallel "capture
  engine": it re-composes a CmdSpec from cmdarg_T per key and decides
  replayability from its own table of "void" key classes.

Solution:
- Model nested `normal_execute()` as a `CmdFrame` stack,
  instead of a single module-scoped `stage`.
  - Capture `<Cmd>` ":norm …" commands as subatoms from its nested frames.
- Produce every command exactly once; atom_push_raw() routes the atom to
  the Visual composite while a selection is open, like it already does
  for mapping composites.
  - `v/pat<CR>d` is now repeatable and cascades.
2026-08-18 07:50:27 -04:00

93 lines
2.5 KiB
Lua

-- Helpers shared by the atom-capture specs (mcursor_spec.lua, cmdatom_spec.lua).
local n = require('test.functional.testnvim')()
local m = {}
function m.get_lines()
return n.buf_lines(0)
end
--- vim.keycode(): |key-notation| => the raw bytes of the CmdAtom event's keys/lhs.
function m.k(s)
return n.api.nvim_replace_termcodes(s, true, true, true)
end
--- Starts collecting CmdAtom event-data.
function m.atoms_start()
n.exec_lua([[
_G.atoms = {}
vim.api.nvim_create_autocmd('CmdAtom', {
callback = function(ev)
table.insert(_G.atoms, ev.data)
end,
})
]])
end
--- Gets the collected CmdAtom event-data.
function m.atoms()
n.poke_eventloop() -- CmdAtom is deferred, so drain the event loop first.
return n.exec_lua('return _G.atoms')
end
--- Gets the last collected CmdAtom event.
function m.atom_last()
local evs = m.atoms()
return evs[#evs]
end
--- Projects only the named fields of `ev` (a nil field stays absent, so `eq` still asserts
--- omission when the expected table lacks it).
function m.pick(ev, ...)
local r = {}
for _, f in ipairs({ ... }) do
r[f] = ev[f]
end
return r
end
--- Gets the last `count` collected CmdAtom events: bare `keys` strings by default, or
--- projections of the named `fields`.
function m.atoms_tail(count, ...)
local evs = m.atoms()
local fields = select('#', ...) > 0 and { ... } or nil
local tail = {}
for i = #evs - count + 1, #evs do
table.insert(tail, fields and m.pick(evs[i], unpack(fields)) or evs[i].keys)
end
return tail
end
--- Minimal vim-surround "ys": an <expr> mapping sets 'operatorfunc' and returns "g@"; the opfunc
--- reads the wrap char with getchar() and wraps the motion region (yank, modify register, paste
--- back).
m.minisurround_vim = [[
function! MiniSurroundSetup() abort
set operatorfunc=MiniSurround
return 'g@'
endfunction
function! MiniSurround(type) abort
let char = nr2char(getchar())
let save = getreg('"')
silent exe "norm! v`[o`]y"
call setreg('"', char .. getreg('"') .. char, 'v')
silent exe "norm! gvp`["
call setreg('"', save)
endfunction
nnoremap <expr> ys MiniSurroundSetup()
]]
--- Minimal vim-surround "ds": a ":call" mapping whose edit runs through :normal inside a
--- function, with a getchar() payload naming the surround to delete.
m.delsurround_vim = [[
function! DelSurround() abort
call getchar()
" cursor is on the "("; delete it and its matching ")".
normal! mz%x`zx
endfunction
nnoremap <silent> ds :<C-U>call DelSurround()<CR>
]]
return m