Something went wrong. Try again.
🐻 minimal ui2 fuzzy finder for Neovim codeberg.org/comfysage/artio.nvim
Something went wrong. Try again.
21 kB · 821 lines
Lua
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822local cmdline = require("vim._core.ui2.cmdline")local ui2 = require("vim._core.ui2")
local _log = {}local _loglevel = vim.log.levels.ERROR---@private---@param msgwrapped { [1]: string, [2]: string? }---@param level? integerlocal function logadd(msgwrapped, level) level = level or vim.log.levels.DEBUG if level < _loglevel then return end _log[#_log + 1] = msgwrappedend---@private---@param msg stringlocal function logdebug(msg) logadd({ msg .. "\n" })end---@private---@param msg string---@param v anylocal function logdbg(msg, v) logdebug(string.format("%s: %s\n", msg, (vim.inspect(v))))end---@private---@param msg stringlocal function logerror(msg) logadd({ msg .. "\n", "ErrorMsg" }, vim.log.levels.ERROR)end
local prompt_hl_id = vim.api.nvim_get_hl_id_by_name("ArtioPrompt")
---@class artio.View---@field picker artio.Picker---@field closed boolean---@field opts table<'win'|'buf'|'g',table<string,any>>---@field marks table<string|integer, integer>---@field win artio.View.win---@field preview_win integerlocal View = {}View.__index = View
---@param picker artio.Pickerfunction View:new(picker) ---@diagnostic disable-next-line: undefined-field if picker.log_level then ---@diagnostic disable-next-line: undefined-field _loglevel = picker.log_level end
return setmetatable({ picker = picker, closed = false, opts = {}, marks = {}, win = { height = 1, }, }, View)end
---@class artio.View.win---@field height integer
local prompthl_id = -1
--- gets updated before drawlocal before_draw_tick = 0--- gets updated after changedtick eventlocal last_draw_tick = 0local function get_changedtick() return vim.api.nvim_buf_get_changedtick(ui2.bufs.cmd)end
local cmdbuff = "" ---@type string Stored cmdline used to calculate translation offset.local promptlen = 0 -- Current length of the last line in the prompt.local promptidx = 0--- Concatenate content chunks and set the text for the current row in the cmdline buffer.------@param content CmdContent---@param prompt stringfunction View:setprompttext(content, prompt) local lines = {} ---@type string[] for line in (prompt .. "\n"):gmatch("(.-)\n") do lines[#lines + 1] = vim.fn.strtrans(line) end
local promptstr = lines[#lines] promptlen = #lines[#lines]
cmdbuff = "" for _, chunk in ipairs(content) do cmdbuff = cmdbuff .. chunk[2] end lines[#lines] = ("%s%s"):format(promptstr, vim.fn.strtrans(cmdbuff))
self:promptpos() self:setlines(promptidx, promptidx + 1, lines) if vim.fn.prompt_getprompt(ui2.bufs.cmd) ~= promptstr then vim.fn.prompt_setprompt(ui2.bufs.cmd, promptstr) end vim.schedule(function() local ok, result = pcall(vim.api.nvim_buf_set_mark, ui2.bufs.cmd, ":", promptidx + 1, promptlen, {}) if not ok then logerror(("Failed to set mark %d:%d\n\t%s"):format(promptidx, promptlen, result)) return end end)end
--- Set the cmdline buffer text and cursor position.------@param content CmdContent---@param pos? integer---@param firstc string---@param prompt string---@param indent integer---@param level integer---@param hl_id integerfunction View:show(content, pos, firstc, prompt, indent, level, hl_id) cmdline.level, cmdline.indent = level, indent if cmdline.highlighter and cmdline.highlighter.active then cmdline.highlighter.active[ui2.bufs.cmd] = nil end if ui2.msg.cmd.msg_row ~= -1 then ui2.msg.msg_clear() end ui2.msg.virt.last = { {}, {}, {}, {} }
self:clear() prompthl_id = hl_id
local cmd_text = "" for _, chunk in ipairs(content) do cmd_text = cmd_text .. chunk[2] end
self:showmatches()
self:setprompttext(content, ("%s%s%s"):format(firstc, prompt, (" "):rep(indent))) self:updatecursor(pos)
self:updatewinheight()
self:drawprompt() self:hlselect()end
--- Set the 'cmdheight' and cmdline window height. Reposition message windows.------@param win integer Cmdline window in the current tabpage.---@param hide boolean Whether to hide or show the window.---@param height integer (Text)height of the cmdline window.function View:win_config(win, hide, height) if ui2.cmdheight == 0 and vim.api.nvim_win_get_config(win).hide ~= hide then vim.api.nvim_win_set_config(win, { hide = hide, height = not hide and height or nil }) elseif vim.api.nvim_win_get_height(win) ~= height then vim.api.nvim_win_set_height(win, height) end
if not hide and self.picker.win.hidestatusline then height = 0 end
if vim.o.cmdheight ~= height then -- Avoid moving the cursor with 'splitkeep' = "screen", and altering the user -- configured value with noautocmd. vim._with({ noautocmd = true, o = { splitkeep = "screen" } }, function() vim.o.cmdheight = height end) ui2.msg.set_pos() end
if self.preview_win and vim.api.nvim_win_is_valid(self.preview_win) then vim.api.nvim_win_set_config(self.preview_win, self:previewconfig()) endend
---@param predicted? integer The predicted height of the cmdline windowfunction View:updatewinheight(predicted) local height = math.max(1, predicted or vim.api.nvim_win_text_height(ui2.wins.cmd, {}).all) height = math.min(height, self.win.height) self:win_config(ui2.wins.cmd, false, height)end
function View:saveview() self.save = vim.fn.winsaveview() self.prevwin = vim.api.nvim_get_current_win()end
function View:restoreview() vim.api.nvim_set_current_win(self.prevwin) vim.fn.winrestview(self.save)end
local ext_winhl = "Search:,CurSearch:,IncSearch:"
---@param restore? booleanfunction View:setopts(restore) local opts = { win = { eventignorewin = "all,-FileType,-InsertCharPre,-TextChangedI,-CursorMovedI", winhighlight = "Normal:ArtioNormal," .. ext_winhl, signcolumn = "no", wrap = false, }, buf = { filetype = "artio-picker", buftype = "prompt", autocomplete = false, }, g = { showmode = false, showcmd = false, }, }
for level, o in pairs(opts) do self.opts[level] = self.opts[level] or {} local props = { scope = level == "g" and "global" or "local", buf = level == "buf" and ui2.bufs.cmd or nil, win = level == "win" and ui2.wins.cmd or nil, }
for name, value in pairs(o) do if restore then vim.api.nvim_set_option_value(name, self.opts[level][name], props) else self.opts[level][name] = vim.api.nvim_get_option_value(name, props) vim.api.nvim_set_option_value(name, value, props) end end endend
local maxlistheight = 1 -- Max height of the matches list (`self.win.height - 1`)
function View:on_resized() logdebug("on_resized")
if self.picker.win.height > 1 then self.win.height = self.picker.win.height else self.win.height = vim.o.lines * self.picker.win.height end self.win.height = math.max(math.ceil(self.win.height), 1)
maxlistheight = math.max(self.win.height - 1, 1)end
function View:open() if not self.picker then return end _log = nil _log = {}
vim.schedule(function() self.augroup = vim.api.nvim_create_augroup("@artio.view", { clear = true })
vim.api.nvim_create_autocmd("CmdlineLeave", { group = self.augroup, once = true, callback = function() self:close() end, })
vim.api.nvim_create_autocmd("ModeChanged", { group = self.augroup, callback = function(ev) if string.match(ev.match, "^i:") then self:close() end end, })
vim.api.nvim_create_autocmd({ "VimResized", "WinEnter" }, { group = self.augroup, callback = function() self:on_resized() end, })
vim.api.nvim_create_autocmd("WinEnter", { group = self.augroup, callback = function() self:update(true) end, })
vim.api.nvim_create_autocmd("TextChangedI", { group = self.augroup, buffer = ui2.bufs.cmd, callback = function() self:update() end, })
vim.api.nvim_create_autocmd("CursorMovedI", { group = self.augroup, buffer = ui2.bufs.cmd, callback = function() self:updatecursor() end, }) end)
cmdline.prompt = false cmdline.srow = 0 cmdline.indent = 1 cmdline.level = 1
self:saveview()
-- initial render self:trigger_show()
vim._with({ noautocmd = true }, function() vim.api.nvim_set_current_win(ui2.wins.cmd) end)
self:setopts()
-- start insert *before* registering events self:updatecursor(#cmdbuff) vim._with({ noautocmd = true }, function() vim.cmd.startinsert({ bang = true }) end)
-- trigger after registering events vim.schedule(function() vim._with({ win = ui2.wins.cmd, wo = { eventignorewin = "" } }, function() vim.api.nvim_exec_autocmds("WinEnter", {}) end) end)end
function View:close() if self.closed then return end self:closepreview() vim.schedule(function() pcall(vim.api.nvim_del_augroup_by_id, self.augroup) pcall(vim.api.nvim_buf_detach, ui2.bufs.cmd)
vim.cmd.stopinsert()
-- prepare state self:setopts(true)
-- reset state self:clear() cmdline.srow = 0 cmdline.erow = 0
-- restore ui self:hide() self:restoreview() vim.cmd.redraw()
self.closed = true
self.picker:close()
vim.api.nvim_echo(_log, true, {}) end)end
function View:hide() vim.fn.clearmatches(ui2.wins.cmd) -- Clear matchparen highlights. vim.api.nvim_win_set_cursor(ui2.wins.cmd, { 1, 0 }) vim.api.nvim_buf_set_lines(ui2.bufs.cmd, 0, -1, false, {})
local clear = vim.schedule_wrap(function(was_prompt) -- Avoid clearing prompt window when it is re-entered before the next event -- loop iteration. E.g. when a non-choice confirm button is pressed. if was_prompt and not cmdline.prompt then pcall(function() vim.api.nvim_buf_set_lines(ui2.bufs.cmd, 0, -1, false, {}) vim.api.nvim_buf_set_lines(ui2.bufs.dialog, 0, -1, false, {}) vim.api.nvim_win_set_config(ui2.wins.dialog, { hide = true }) vim.on_key(nil, ui2.msg.dialog_on_key) end) end -- Messages emitted as a result of a typed command are treated specially: -- remember if the cmdline was used this event loop iteration. -- NOTE: Message event callbacks are themselves scheduled, so delay two iterations. vim.schedule(function() cmdline.level = -1 end) end) clear(cmdline.prompt)
cmdline.prompt, cmdline.level = false, 0 self:win_config(ui2.wins.cmd, true, ui2.cmdheight)end
function View:trigger_show() logdebug("trigger_show") local input if self.picker.live then input = self.picker.liveinput else input = self.picker.input end self:show({ { 0, input } }, -1, "", self.picker.prompttext, cmdline.indent, cmdline.level, prompt_hl_id)end
---@param force? booleanfunction View:update(force) if not force and before_draw_tick < last_draw_tick and before_draw_tick == get_changedtick() - 1 then logdebug("update (skip-redraw)") return self:drawprompt() end
logdebug("update")
local text = vim.api.nvim_get_current_line() text = text:sub(promptlen + 1)
if self.picker.live then self.picker.liveinput = text else self.picker.input = text end
vim.schedule(coroutine.wrap(function() logdebug("getmatches") self.picker:getmatches()
if self.closed then return end
vim.schedule_wrap(self.trigger_show)(self) end))end
local curpos = { 0, 0 } -- Last drawn cursor position. absolute---@param pos? integer relative to promptfunction View:updatecursor(pos) logdebug("updatecursor")
self:promptpos()
if not pos or pos < 0 then local cursorpos = vim.api.nvim_win_get_cursor(ui2.wins.cmd) pos = cursorpos[2] - promptlen end
-- set cursor pos to *at least* the prompt length curpos[2] = math.max(curpos[2], promptlen)
if curpos[1] == promptidx + 1 and curpos[2] == promptlen + pos then return end
if pos < 0 then -- reset to last known position pos = curpos[2] - promptlen end
curpos[1], curpos[2] = promptidx + 1, promptlen + pos
vim._with({ noautocmd = true }, function() local ok, _ = pcall(vim.api.nvim_win_set_cursor, ui2.wins.cmd, curpos) if not ok then logerror(("Failed to set cursor %d:%d"):format(curpos[1], curpos[2])) end end)end
local srow = 0
function View:clear() srow = self.picker.opts.bottom and 0 or 1 cmdline.erow = srow self:setlines(0, -1, {})end
function View:promptpos() promptidx = self.picker.opts.bottom and cmdline.erow or 0end
function View:setlines(posstart, posend, lines) -- update winheight to prevent wrong scroll when increasing from 1 local diff = #lines - (posend - posstart) if diff ~= 0 then local height = vim.api.nvim_win_text_height(ui2.wins.cmd, {}).all local predicted = height + diff self:updatewinheight(predicted) end
before_draw_tick = get_changedtick() vim.api.nvim_buf_set_lines(ui2.bufs.cmd, posstart, posend, false, lines) last_draw_tick = get_changedtick()end
local view_ns = vim.api.nvim_create_namespace("@artio.view.ns")local ext_priority = { prompt = 1, info = 2, select = 4, marker = 8, hl = 16, icon = 32, match = 64,}
---@param id? string|integer---@param line integer 0-based---@param col integer 0-based---@param opts vim.api.keyset.set_extmark---@return integerfunction View:mark(id, line, col, opts) if id and self.marks[id] then vim._with({ noautocmd = true }, function() vim.api.nvim_buf_del_extmark(ui2.bufs.cmd, view_ns, self.marks[id]) end) self.marks[id] = nil end
opts.hl_mode = "combine" opts.invalidate = true
local ok, result vim._with({ noautocmd = true }, function() ok, result = pcall(vim.api.nvim_buf_set_extmark, ui2.bufs.cmd, view_ns, line, col, opts) end) if not ok then logerror(("Failed to add extmark %d:%d\n\t%s"):format(line, col, result)) return -1 end
if id and result >= 0 then self.marks[id] = result end
return resultend
---@param p artio.Picker---@param info 'index'|'list'|string---@return stringlocal function getpromptinfo(p, info) if info == "index" then return ("[%d]"):format(p.idx) elseif info == "list" then return ("(%d/%d)"):format(#p.matches, p.items and #p.items or 0) end return ""end
function View:drawprompt() logdebug("drawprompt")
self:promptpos() if promptlen > 0 and prompthl_id > 0 then self:mark("prompthl", promptidx, 0, { hl_group = prompthl_id, end_col = promptlen, priority = ext_priority.prompt }) self:mark("promptinfo", promptidx, 0, { virt_text = { { table.concat( vim .iter(self.picker.opts.infolist) :map(function(info) return getpromptinfo(self.picker, info) end) :totable(), " " ), "InfoText", }, }, virt_text_pos = "eol_right_align", priority = ext_priority.info, }) endend
local offset = 0
function View:updateoffset() self.picker:fix() if self.picker.idx == 0 then offset = 0 return end
local _offset = self.picker.idx - maxlistheight if _offset > offset then offset = _offset elseif self.picker.idx <= offset then offset = self.picker.idx - 1 end
offset = math.min(math.max(0, offset), math.max(0, #self.picker.matches - maxlistheight))end
local icon_pad = 2
function View:showmatches() local indent = vim.fn.strdisplaywidth(self.picker.opts.pointer) + 1 local prefix = (" "):rep(indent) local icon_pad_str = (" "):rep(icon_pad)
self:updateoffset()
local lines = {} ---@type string[] local hls = {} local icons = {} ---@type ([string, string]|false)[] local custom_hls = {} ---@type (artio.Picker.hl[]|false)[] local marks = {} ---@type boolean[] for i = 1 + offset, math.min(#self.picker.matches, maxlistheight + offset) do local match = self.picker.matches[i] local item = self.picker.items[match[1]]
local icon, icon_hl = item.icon, item.icon_hl if not (icon and icon_hl) and vim.is_callable(self.picker.get_icon) then icon, icon_hl = self.picker.get_icon(item) item.icon, item.icon_hl = icon, icon_hl end icons[#icons + 1] = icon and { icon, icon_hl } or false icon = icon and ("%s%s"):format(item.icon, icon_pad_str) or ""
local hl = item.hls if not hl and vim.is_callable(self.picker.hl_item) then hl = self.picker.hl_item(item) item.hls = hl end custom_hls[#custom_hls + 1] = hl or false
marks[#marks + 1] = self.picker.marked[item.id] or false
lines[#lines + 1] = ("%s%s%s"):format(prefix, icon, item.text) hls[#hls + 1] = match[2] end
if not self.picker.opts.shrink then for _ = 1, (maxlistheight - #lines) do lines[#lines + 1] = "" end end self:setlines(srow, cmdline.erow, lines) cmdline.erow = srow + #lines
for i = 1, #lines do local has_icon = icons[i] and icons[i][1] and true local icon_indent = has_icon and (#icons[i][1] + icon_pad) or 0
if has_icon and icons[i][2] then self:mark(nil, srow + i - 1, indent, { end_col = indent + icon_indent, hl_group = icons[i][2], priority = ext_priority.icon, }) end
local line_hls = custom_hls[i] if line_hls then for j = 1, #line_hls do local hl = line_hls[j] self:mark(nil, srow + i - 1, indent + icon_indent + hl[1][1], { end_col = indent + icon_indent + hl[1][2], hl_group = hl[2], priority = ext_priority.hl, }) end end
if marks[i] then self:mark(nil, srow + i - 1, indent - 1, { virt_text = { { self.picker.opts.marker, "ArtioMark" } }, virt_text_pos = "overlay", priority = ext_priority.marker, }) self:mark(nil, srow + i - 1, 0, { hl_group = "ArtioMarkLine", hl_eol = true, end_row = srow + i, end_col = 0,
priority = ext_priority.marker, }) end
if hls[i] then for j = 1, #hls[i] do local col = indent + icon_indent + hls[i][j] self:mark(nil, srow + i - 1, col, { hl_group = "ArtioMatch", end_col = col + 1, priority = ext_priority.match, }) end end endend
function View:hlselect() self:softupdatepreview()
self.picker:fix() local idx = self.picker.idx if idx == 0 then return end
self:updateoffset() local row = math.max(0, math.min(srow + (idx - offset), cmdline.erow) - 1)
self:mark("hlselect", row, 0, { virt_text = { { self.picker.opts.pointer, "ArtioPointer" } }, virt_text_pos = "overlay",
hl_group = "ArtioSel", hl_eol = true, end_row = row + 1, end_col = 0,
priority = ext_priority.select, })end
function View:togglepreview() if self.preview_win then self:closepreview() return end
self:updatepreview()end
---@return {buf?:integer, pos?:[integer,integer], pos_end?:[integer,integer]}?function View:openpreview() if self.picker.idx == 0 then return end
local match = self.picker.matches[self.picker.idx] local item = self.picker.items[match[1]]
if not item or not (self.picker.preview_item and vim.is_callable(self.picker.preview_item)) then return end
return self.picker.preview_item(item.v)end
function View:previewconfig() local previewopts = self.picker.win.preview_opts and vim.is_callable(self.picker.win.preview_opts) and self.picker.win.preview_opts(self) local cmdheight = vim.api.nvim_win_get_height(ui2.wins.cmd)
local winborder = previewopts and previewopts.border or vim.o.winborder return vim.tbl_extend("force", { relative = "editor", focusable = false, width = vim.o.columns, height = self.win.height, col = 0, row = vim.o.lines - (self.win.height + cmdheight) - ((winborder == "none" or winborder == "") and 0 or 2) - (self.picker.win.hidestatusline and 0 or 1), }, previewopts or {})end
function View:updatepreview() local pr = self:openpreview() if not pr or not pr.buf then return end
if not self.preview_win then self.preview_win = vim.api.nvim_open_win(pr.buf, false, self:previewconfig()) else vim.api.nvim_win_set_buf(self.preview_win, pr.buf) end
vim._with({ win = self.preview_win, noautocmd = true }, function() vim.api.nvim_set_option_value("previewwindow", true, { scope = "local" }) vim.api.nvim_set_option_value("eventignorewin", "all,-FileType", { scope = "local" })
local sameline = pr.pos ~= nil and pr.pos_end == nil or pr.pos_end[1] == pr.pos[1] vim.api.nvim_set_option_value("cursorline", sameline, { scope = "local" })
if pr.pos then vim.api.nvim_win_set_cursor(self.preview_win, pr.pos) end end)end
function View:softupdatepreview() if self.picker.idx == 0 then self:closepreview() end
if not self.preview_win then return end
self:updatepreview()end
function View:closepreview() if not self.preview_win then return end
vim.api.nvim_win_close(self.preview_win, true) self.preview_win = nilend
return View