-- finder.lua — Minimal, hardened async fuzzy finder + grep for Neovim ≥ 0.11 local M = {} --------------------------------------------------------------------- -- Config --------------------------------------------------------------------- M.config = { file_cmd = nil, -- "fd" | "fdfind" | nil (auto) grep_cmd = 'rg', -- ripgrep binary page_size = 60, -- soft cap; real viewport height is measured debounce_ms = 80, cache_ttl_sec = 20, max_items = 5000, -- safety cap for massive outputs debug = false, -- Exclusion controls (added to tool flags and used in glob fallback) exclude_patterns = { 'node_modules', 'dist', 'build', '.git' }, -- Optional on-disk filelist cache at $ROOT/.devflow/finder_cache.json use_disk_cache = false, } --------------------------------------------------------------------- -- State --------------------------------------------------------------------- local S = { active = false, mode = nil, -- "files" | "grep" root = nil, cache = {}, timer = nil, ns = vim.api.nvim_create_namespace('finder_ns'), aug = nil, gen = 0, -- session generation id -- async jobs job_files = nil, job_rg = nil, -- UI win_inp = nil, buf_inp = nil, win_res = nil, buf_res = nil, -- Data query = '', items = {}, -- full set (files or grep lines) filtered = {}, -- current view positions = {}, -- positions[i] = { {scol, ecol}, ... } for filtered[i] select = 1, -- absolute index in filtered (1-based) scroll = 0, -- top index (0-based) user_moved = false, -- has user navigated during this session? } --------------------------------------------------------------------- -- Utils --------------------------------------------------------------------- local function L(msg, data) if not M.config.debug then return end local s = '[finder] ' .. msg if data ~= nil then s = s .. ' ' .. vim.inspect(data) end vim.schedule(function() pcall(vim.notify, s) end) end local function now_sec() return vim.loop.now() / 1000 end local function clamp(v, lo, hi) return (v < lo) and lo or ((v > hi) and hi or v) end local function cmd_exists(bin) return vim.fn.executable(bin) == 1 end local function debounce(fn, ms) if S.timer then S.timer:stop() S.timer:close() S.timer = nil end S.timer = vim.loop.new_timer() S.timer:start(ms, 0, function() if S.timer then S.timer:stop() S.timer:close() S.timer = nil end vim.schedule(fn) end) end local function read_gitignore(root) local p = root .. '/.gitignore' local lines = {} local f = io.open(p, 'r') if not f then return {} end for line in f:lines() do local l = vim.trim(line) if #l > 0 and not l:match('^#') and not l:match('^!') then -- normalize directory suffix to plain token for our excludes l = l:gsub('/+$', '') table.insert(lines, l) end end f:close() return lines end local function ensure_cache_dir(root) local dir = root .. '/.devflow' local ok = vim.loop.fs_stat(dir) if not ok then pcall(vim.loop.fs_mkdir, dir, 448) end -- 0700 return dir end local function load_cache(root) if not M.config.use_disk_cache then return nil end local file = ensure_cache_dir(root) .. '/finder_cache.json' local f = io.open(file, 'r') if not f then return nil end local content = f:read('*a') f:close() local ok, data = pcall(vim.json.decode, content) if not ok or type(data) ~= 'table' or type(data.files) ~= 'table' then return nil end if now_sec() - (data.timestamp or 0) > M.config.cache_ttl_sec then return nil end return data.files end local function save_cache(root, files) if not M.config.use_disk_cache then return end local file = ensure_cache_dir(root) .. '/finder_cache.json' local payload = { timestamp = now_sec(), files = files } local ok, json = pcall(vim.json.encode, payload) if not ok then return end local f = io.open(file, 'w') if not f then return end f:write(json) f:close() end local function project_root() local obj = vim.system({ 'git', 'rev-parse', '--show-toplevel' }, { text = true }):wait() if obj.code == 0 and obj.stdout and obj.stdout ~= '' then return vim.trim(obj.stdout) end return vim.fn.getcwd(0, 0) end local function normalize_rel(root, p) if not p or p == '' then return '' end -- strip leading ./ and leading root if any p = p:gsub('^%./', '') local root_norm = vim.fs.normalize(root) local abs = vim.fs.normalize(p) if abs:find(root_norm, 1, true) == 1 then local rel = abs:sub(#root_norm + 2) return rel ~= '' and rel or abs end return p end local function to_abs_path(root, rel) if not rel or rel == '' then return root end return vim.fs.normalize(root .. '/' .. rel) end local function page_rows() if S.win_res and vim.api.nvim_win_is_valid(S.win_res) then return math.max(1, vim.api.nvim_win_get_height(S.win_res)) end return math.max(1, math.min(M.config.page_size, vim.o.lines)) end --------------------------------------------------------------------- -- Render helpers --------------------------------------------------------------------- local function ensure_visible() local page = page_rows() local sel = clamp(S.select, 1, #S.filtered) local top = S.scroll + 1 local bot = S.scroll + page if sel < top then S.scroll = sel - 1 elseif sel > bot then S.scroll = sel - page end S.scroll = clamp(S.scroll, 0, math.max(#S.filtered - page, 0)) end --------------------------------------------------------------------- -- Render --------------------------------------------------------------------- local function render() if not (S.buf_res and vim.api.nvim_buf_is_valid(S.buf_res)) then return end ensure_visible() local total = #S.filtered local view = {} if total == 0 then view = { '-- no matches --' } else local start_idx = S.scroll + 1 local end_idx = math.min(start_idx + page_rows() - 1, total) for i = start_idx, end_idx do local line = S.filtered[i] view[#view + 1] = type(line) == 'string' and line or tostring(line or '') end end vim.bo[S.buf_res].modifiable = true vim.bo[S.buf_res].readonly = false vim.api.nvim_buf_set_lines(S.buf_res, 0, -1, false, view) vim.api.nvim_buf_clear_namespace(S.buf_res, S.ns, 0, -1) -- match highlights (visible window only) vim.api.nvim_set_hl(0, 'FinderMatch', { link = 'Search', default = true }) for i = 1, #view do local idx = S.scroll + i local spans = S.positions[idx] if spans then for _, se in ipairs(spans) do local scol, ecol = se[1], se[2] if ecol > scol then vim.api.nvim_buf_set_extmark(S.buf_res, S.ns, i - 1, scol, { end_col = ecol, hl_group = 'FinderMatch', }) end end end end -- selection highlight if total > 0 and #view > 0 then vim.api.nvim_set_hl(0, 'FinderSelection', { link = 'CursorLine', default = true }) local rel = clamp(S.select - S.scroll, 1, #view) vim.api.nvim_buf_set_extmark(S.buf_res, S.ns, rel - 1, 0, { end_line = rel, hl_group = 'FinderSelection', hl_eol = true, }) end vim.bo[S.buf_res].modifiable = false L('render', { select = S.select, scroll = S.scroll, total = total, lines = #view }) end --------------------------------------------------------------------- -- Querying / Filtering --------------------------------------------------------------------- local function compute_positions_files(items, q) -- matchfuzzypos returns {items, positions} local ok, res = pcall(vim.fn.matchfuzzypos, items, q) if not ok or type(res) ~= 'table' then return {}, {} end local out_items = res[1] or {} local pos = res[2] or {} local filtered, positions = {}, {} for i, v in ipairs(out_items) do filtered[i] = (type(v) == 'string') and v or tostring(v or '') local cols = pos[i] or {} local spans = {} for _, c in ipairs(cols) do local start0 = (c > 0) and (c - 1) or 0 -- to 0-based spans[#spans + 1] = { start0, start0 + 1 } end positions[i] = spans end return filtered, positions end local function compute_positions_grep(lines, q) if q == '' then return lines, {} end local qlow = q:lower() local positions = {} for i, line in ipairs(lines) do local sidx = 1 local spans = {} local llow = (type(line) == 'string' and line or tostring(line or '')):lower() while true do local s, e = string.find(llow, qlow, sidx, true) if not s then break end spans[#spans + 1] = { s - 1, e } -- 0-based start, exclusive end sidx = e + 1 if #spans > 64 then break end end positions[i] = (#spans > 0) and spans or nil end return lines, positions end --------------------------------------------------------------------- -- set_items --------------------------------------------------------------------- local function set_items(list) list = list or {} if #list > M.config.max_items then local tmp = {} for i = 1, M.config.max_items do tmp[i] = list[i] end list = tmp end -- sanitize to strings for i, v in ipairs(list) do if type(v) ~= 'string' then list[i] = tostring(v or '') end end S.items = list S.filtered = list S.positions = {} S.select = 1 S.scroll = 0 render() end --------------------------------------------------------------------- -- set_query --------------------------------------------------------------------- local function set_query(q) local prev_val = S.filtered[S.select] S.query = q if S.mode == 'grep' then local filtered, pos = compute_positions_grep(S.items, q) S.filtered, S.positions = filtered, pos else if q == '' then S.filtered = S.items S.positions = {} else local filtered, pos = compute_positions_files(S.items, q) S.filtered, S.positions = filtered, pos end end -- sanitize filtered list for i, v in ipairs(S.filtered) do if type(v) ~= 'string' then S.filtered[i] = tostring(v or '') end end -- selection policy: -- if user hasn't moved, always reset to first on query changes; -- if user moved, preserve previous pick when present, else first. if not S.user_moved then S.select = 1 else local idx = nil if prev_val then for i, v in ipairs(S.filtered) do if v == prev_val then idx = i break end end end S.select = idx or 1 end ensure_visible() render() L('set_query', { query = q, filtered = #S.filtered, user_moved = S.user_moved }) end --------------------------------------------------------------------- -- Move / Accept --------------------------------------------------------------------- local function move_down() if #S.filtered == 0 then return end S.user_moved = true S.select = clamp(S.select + 1, 1, #S.filtered) ensure_visible() render() end local function move_up() if #S.filtered == 0 then return end S.user_moved = true S.select = clamp(S.select - 1, 1, #S.filtered) ensure_visible() render() end local function parse_vimgrep(line) -- Greedy file capture up to last "::" if type(line) ~= 'string' then return nil end local file, lnum = line:match('^(.*):(%d+):') if not file then return nil end return file, tonumber(lnum) or 1 end local function accept_selection_files() local pick = S.filtered[S.select] if type(pick) ~= 'string' or pick == '' then return end local abs = to_abs_path(S.root, pick) M.close() vim.schedule(function() vim.cmd.edit(vim.fn.fnameescape(abs)) end) end local function accept_selection_grep() local pick = S.filtered[S.select] if type(pick) ~= 'string' or pick == '' then return end local file, lnum = parse_vimgrep(pick) if not file then return end local abs = to_abs_path(S.root, file) local ln = tonumber(lnum) or 1 M.close() vim.schedule(function() vim.cmd.edit(vim.fn.fnameescape(abs)) if pcall(vim.api.nvim_win_set_cursor, 0, { ln, 0 }) then end end) end --------------------------------------------------------------------- -- Backends --------------------------------------------------------------------- local function cancel_job(job) if not job then return end pcall(function() job:kill(15) end) -- SIGTERM if available end local function collect_files_async(cb) local root = S.root local gen = S.gen local disk = load_cache(root) if disk and type(disk) == 'table' then cb(disk) return end local file_cmd = M.config.file_cmd if not file_cmd then if cmd_exists('fd') then file_cmd = 'fd' elseif cmd_exists('fdfind') then file_cmd = 'fdfind' end end local excludes = {} for _, p in ipairs(M.config.exclude_patterns or {}) do table.insert(excludes, p) end local gi = read_gitignore(root) for _, p in ipairs(gi) do table.insert(excludes, p) end if file_cmd then local args = { file_cmd, '--type', 'f', '--hidden', '--color', 'never' } -- best-effort: avoid leading ./ in output if file_cmd == 'fd' or file_cmd == 'fdfind' then table.insert(args, '--strip-cwd-prefix') end for _, ex in ipairs(excludes) do table.insert(args, '--exclude') table.insert(args, ex) end cancel_job(S.job_files) S.job_files = vim.system(args, { text = true, cwd = root }, function(obj) if not (S.active and gen == S.gen) then return end local list = {} if obj.code == 0 and obj.stdout then local raw = vim.split(obj.stdout, '\n', { trimempty = true }) for _, p in ipairs(raw) do list[#list + 1] = normalize_rel(root, p) end end if #list > M.config.max_items then local tmp = {} for i = 1, M.config.max_items do tmp[i] = list[i] end list = tmp end save_cache(root, list) vim.schedule(function() if S.active and gen == S.gen then cb(list) end end) end) return end -- Try git ls-files as async fallback if cmd_exists('git') then cancel_job(S.job_files) S.job_files = vim.system( { 'git', 'ls-files', '-co', '--exclude-standard', '-z' }, { text = true, cwd = root }, function(o2) if not (S.active and gen == S.gen) then return end local list = {} if o2.code == 0 and o2.stdout then for p in o2.stdout:gmatch('([^%z]+)') do list[#list + 1] = normalize_rel(root, p) end end if #list > M.config.max_items then local tmp = {} for i = 1, M.config.max_items do tmp[i] = list[i] end list = tmp end save_cache(root, list) vim.schedule(function() if S.active and gen == S.gen then cb(list) end end) end ) return end -- Last resort: blocking glob with simple exclusion checks local list = vim.fn.globpath(root, '**/*', false, true) list = vim.tbl_filter(function(p) if vim.fn.isdirectory(p) == 1 then return false end for _, ex in ipairs(excludes) do if p:find('/' .. ex .. '/', 1, true) or p:find('/' .. ex .. '$', 1, true) then return false end end return true end, list) local rel = {} for i = 1, #list do rel[i] = normalize_rel(root, list[i]) end if #rel > M.config.max_items then local tmp = {} for i = 1, M.config.max_items do tmp[i] = rel[i] end rel = tmp end save_cache(root, rel) cb(rel) end local function grep_async(query, cb) if query == '' then cb({}) return end local gen = S.gen local rg = M.config.grep_cmd if not cmd_exists(rg) then cb({}) return end local root = S.root local args = { rg, '--vimgrep', '--hidden', '--smart-case', '--no-heading', '--no-config', '--color', 'never', '--path-separator', '/', '--', query, } -- Apply excludes as negative globs local excludes = {} for _, p in ipairs(M.config.exclude_patterns or {}) do table.insert(excludes, p) end local gi = read_gitignore(root) for _, p in ipairs(gi) do table.insert(excludes, p) end for _, ex in ipairs(excludes) do table.insert(args, 2, '--glob') -- insert before '--' table.insert(args, 3, '!' .. ex) end cancel_job(S.job_rg) S.job_rg = vim.system(args, { text = true, cwd = root }, function(obj) if not (S.active and gen == S.gen) then return end local list = {} if obj.code == 0 and obj.stdout then list = vim.split(obj.stdout, '\n', { trimempty = true }) end if #list > M.config.max_items then local tmp = {} for i = 1, M.config.max_items do tmp[i] = list[i] end list = tmp end vim.schedule(function() if S.active and gen == S.gen then cb(list) end end) end) end --------------------------------------------------------------------- -- Layout --------------------------------------------------------------------- local function open_layout(prompt) S.buf_inp = vim.api.nvim_create_buf(false, true) S.buf_res = vim.api.nvim_create_buf(false, true) for _, b in ipairs({ S.buf_inp, S.buf_res }) do if b and vim.api.nvim_buf_is_valid(b) then vim.bo[b].buflisted = false vim.bo[b].bufhidden = 'wipe' vim.bo[b].swapfile = false end end if S.buf_inp and vim.api.nvim_buf_is_valid(S.buf_inp) then vim.bo[S.buf_inp].buftype = 'prompt' end if S.buf_res and vim.api.nvim_buf_is_valid(S.buf_res) then vim.bo[S.buf_res].buftype = 'nofile' vim.bo[S.buf_res].modifiable = false vim.bo[S.buf_res].readonly = false end local width = math.floor(vim.o.columns * 0.8) local height = math.min(math.floor(vim.o.lines * 0.5), M.config.page_size + 2) local col = math.floor((vim.o.columns - width) / 2) local row = math.floor((vim.o.lines - height) * 0.7) S.win_inp = vim.api.nvim_open_win(S.buf_inp, true, { relative = 'editor', style = 'minimal', border = 'rounded', width = width, height = 1, col = col, row = row, focusable = true, zindex = 200, }) if S.buf_inp and vim.api.nvim_buf_is_valid(S.buf_inp) then vim.fn.prompt_setprompt(S.buf_inp, prompt or '') end S.win_res = vim.api.nvim_open_win(S.buf_res, false, { relative = 'editor', style = 'minimal', border = 'single', width = width, height = math.max(1, height - 2), col = col, row = row + 2, focusable = false, zindex = 199, }) if S.win_res and vim.api.nvim_win_is_valid(S.win_res) then vim.wo[S.win_res].cursorline = false vim.wo[S.win_res].cursorlineopt = 'line' end if S.aug then pcall(vim.api.nvim_del_augroup_by_id, S.aug) end S.aug = vim.api.nvim_create_augroup('finder_session', { clear = true }) -- Close if focus enters a non-floating window that isn't ours vim.api.nvim_create_autocmd('WinEnter', { group = S.aug, callback = function() if not S.active then return end local w = vim.api.nvim_get_current_win() if w == S.win_inp or w == S.win_res then return end local cfg = vim.api.nvim_win_get_config(w) if cfg and cfg.relative == '' then M.close() end end, }) -- Close if prompt buffer hides or leaves if S.buf_inp and vim.api.nvim_buf_is_valid(S.buf_inp) then vim.api.nvim_create_autocmd({ 'BufHidden', 'BufLeave' }, { group = S.aug, buffer = S.buf_inp, callback = function() if S.active then M.close() end end, }) end -- Re-render on resize to respect new viewport height vim.api.nvim_create_autocmd('VimResized', { group = S.aug, callback = function() if S.active then render() end end, }) vim.cmd.startinsert() L('open_layout', { win_inp = S.win_inp, win_res = S.win_res }) end local function close_layout() for _, win in ipairs({ S.win_inp, S.win_res }) do if win and vim.api.nvim_win_is_valid(win) then pcall(vim.api.nvim_win_close, win, true) end end for _, buf in ipairs({ S.buf_inp, S.buf_res }) do if buf and vim.api.nvim_buf_is_valid(buf) then pcall(vim.api.nvim_buf_delete, buf, { force = true }) end end if S.aug then pcall(vim.api.nvim_del_augroup_by_id, S.aug) S.aug = nil end L('close_layout') end --------------------------------------------------------------------- -- Input handlers --------------------------------------------------------------------- local function attach_handlers() local opts = { buffer = S.buf_inp, nowait = true, silent = true, noremap = true } vim.keymap.set('i', '', move_down, opts) vim.keymap.set('i', '', move_up, opts) vim.keymap.set('i', '', move_down, opts) vim.keymap.set('i', '', move_up, opts) vim.keymap.set('i', '', function() if S.mode == 'grep' then accept_selection_grep() else accept_selection_files() end end, opts) vim.keymap.set('i', '', function() M.close() end, opts) vim.keymap.set('i', '', function() M.close() end, opts) if S.buf_inp and vim.api.nvim_buf_is_valid(S.buf_inp) then vim.api.nvim_create_autocmd({ 'TextChangedI', 'TextChangedP' }, { group = S.aug, buffer = S.buf_inp, callback = function() debounce(function() if not (S.active and S.buf_inp and vim.api.nvim_buf_is_valid(S.buf_inp)) then return end local raw = vim.fn.getline('.') raw = type(raw) == 'string' and raw or '' local prompt_pat = (S.mode == 'files') and '^Search:%s*' or '^Grep:%s*' local q = raw:gsub(prompt_pat, '') if S.mode == 'grep' then grep_async(q, function(list) if not S.active then return end S.items = list or {} set_query(q) end) else set_query(q) end end, M.config.debounce_ms) end, }) end L('keymaps attached', opts) end --------------------------------------------------------------------- -- Public --------------------------------------------------------------------- function M.files() if S.active then M.close() end S.active = true S.gen = S.gen + 1 S.user_moved = false S.mode = 'files' S.root = project_root() open_layout('Search: ') attach_handlers() collect_files_async(function(list) if not S.active then return end set_items(list) -- render initial list (relative paths) end) end function M.grep() if S.active then M.close() end S.active = true S.gen = S.gen + 1 S.user_moved = false S.mode = 'grep' S.root = project_root() open_layout('Grep: ') attach_handlers() S.items = {} set_query('') end function M.close() if not S.active then return end -- stop timers and jobs first if S.timer then S.timer:stop() S.timer:close() S.timer = nil end cancel_job(S.job_rg) S.job_rg = nil cancel_job(S.job_files) S.job_files = nil close_layout() S.active = false S.mode, S.root = nil, nil S.items, S.filtered, S.positions = {}, {}, {} S.query, S.select, S.scroll = '', 1, 0 S.user_moved = false L('session closed') end function M.setup(opts) M.config = vim.tbl_deep_extend('force', M.config, opts or {}) if M.config.file_cmd and not cmd_exists(M.config.file_cmd) then pcall( vim.notify, ("finder: file_cmd '%s' not found"):format(M.config.file_cmd), vim.log.levels.WARN ) end if M.config.grep_cmd and not cmd_exists(M.config.grep_cmd) then pcall( vim.notify, ("finder: grep_cmd '%s' not found, will try 'rg'"):format(M.config.grep_cmd), vim.log.levels.WARN ) end L('setup', M.config) end return M