259 lines
7.6 KiB
Lua
259 lines
7.6 KiB
Lua
-- Minimal fuzzy finder + content search for Neovim 0.11+
|
|
-- Optional: `fdfind` or `fd` for file listing, `rg` for text search
|
|
|
|
local Fuzzy = {}
|
|
|
|
--------------------------------------------------------------------
|
|
-- 🧩 Helpers
|
|
--------------------------------------------------------------------
|
|
|
|
-- Collect all files (try fdfind/fd first, then globpath)
|
|
local function get_file_list()
|
|
local handle = io.popen('fdfind --type f 2>/dev/null || fd --type f 2>/dev/null')
|
|
if handle then
|
|
local result = handle:read('*a')
|
|
handle:close()
|
|
if result and result ~= '' then
|
|
return vim.split(result, '\n', { trimempty = true })
|
|
end
|
|
end
|
|
return vim.fn.globpath('.', '**/*', false, true)
|
|
end
|
|
|
|
-- Create the input/results floating windows
|
|
local function open_float(prompt)
|
|
local input_buf = vim.api.nvim_create_buf(false, true)
|
|
local result_buf = vim.api.nvim_create_buf(false, true)
|
|
|
|
-- mark both buffers as scratch/unlisted
|
|
for _, b in ipairs({ input_buf, result_buf }) do
|
|
vim.bo[b].bufhidden = 'wipe'
|
|
vim.bo[b].buflisted = false
|
|
vim.bo[b].swapfile = false
|
|
end
|
|
vim.bo[input_buf].buftype = 'prompt'
|
|
vim.bo[result_buf].buftype = 'nofile'
|
|
|
|
local width = math.floor(vim.o.columns * 0.7)
|
|
local height = 20
|
|
local row = math.floor((vim.o.lines - height) / 2)
|
|
local col = math.floor((vim.o.columns - width) / 2)
|
|
|
|
local input_win = vim.api.nvim_open_win(input_buf, true, {
|
|
relative = 'editor',
|
|
row = row,
|
|
col = col,
|
|
width = width,
|
|
height = 1,
|
|
style = 'minimal',
|
|
border = 'rounded',
|
|
})
|
|
vim.fn.prompt_setprompt(input_buf, prompt)
|
|
|
|
local result_win = vim.api.nvim_open_win(result_buf, false, {
|
|
relative = 'editor',
|
|
row = row + 2,
|
|
col = col,
|
|
width = width,
|
|
height = height - 2,
|
|
style = 'minimal',
|
|
border = 'single',
|
|
})
|
|
|
|
return input_buf, result_buf, input_win, result_win
|
|
end
|
|
|
|
--------------------------------------------------------------------
|
|
-- 🔵 Highlight current selection (using extmark, not deprecated API)
|
|
--------------------------------------------------------------------
|
|
function Fuzzy:highlight_selection()
|
|
if not self.result_buf then
|
|
return
|
|
end
|
|
if not self.ns_id then
|
|
self.ns_id = vim.api.nvim_create_namespace('FuzzyHighlight')
|
|
end
|
|
vim.api.nvim_buf_clear_namespace(self.result_buf, self.ns_id, 0, -1)
|
|
if self.matches and self.matches[self.cursor] then
|
|
vim.api.nvim_buf_set_extmark(self.result_buf, self.ns_id, self.cursor - 1, 0, {
|
|
end_line = self.cursor,
|
|
hl_group = 'Visual',
|
|
hl_eol = true,
|
|
})
|
|
end
|
|
end
|
|
|
|
--------------------------------------------------------------------
|
|
-- 🔴 Close all floating windows
|
|
--------------------------------------------------------------------
|
|
function Fuzzy.close()
|
|
local wins = { Fuzzy.input_win, Fuzzy.result_win }
|
|
for _, win in ipairs(wins) do
|
|
if win and vim.api.nvim_win_is_valid(win) then
|
|
vim.api.nvim_win_close(win, true)
|
|
end
|
|
end
|
|
Fuzzy.active = false
|
|
end
|
|
|
|
--------------------------------------------------------------------
|
|
-- 🟢 Fuzzy file finder (using fd or globpath)
|
|
--------------------------------------------------------------------
|
|
function Fuzzy.open()
|
|
if Fuzzy.active then
|
|
Fuzzy.close()
|
|
end
|
|
Fuzzy.active = true
|
|
|
|
Fuzzy.files = get_file_list()
|
|
Fuzzy.matches = Fuzzy.files
|
|
Fuzzy.cursor = 1
|
|
|
|
Fuzzy.input_buf, Fuzzy.result_buf, Fuzzy.input_win, Fuzzy.result_win = open_float('Search: ')
|
|
|
|
local function update_results(text)
|
|
if text == '' then
|
|
Fuzzy.matches = Fuzzy.files
|
|
else
|
|
Fuzzy.matches = vim.fn.matchfuzzy(Fuzzy.files, text)
|
|
end
|
|
if #Fuzzy.matches == 0 then
|
|
vim.api.nvim_buf_set_lines(Fuzzy.result_buf, 0, -1, false, { '-- no matches --' })
|
|
Fuzzy.cursor = 1
|
|
return
|
|
end
|
|
local display = {}
|
|
for i = 1, math.min(#Fuzzy.matches, 50) do
|
|
display[#display + 1] = Fuzzy.matches[i]
|
|
end
|
|
vim.api.nvim_buf_set_lines(Fuzzy.result_buf, 0, -1, false, display)
|
|
Fuzzy.cursor = 1
|
|
Fuzzy:highlight_selection()
|
|
end
|
|
|
|
vim.api.nvim_create_autocmd({ 'TextChangedI', 'TextChangedP' }, {
|
|
buffer = Fuzzy.input_buf,
|
|
callback = function()
|
|
local text = vim.fn.getline('.'):gsub('^Search:%s*', '')
|
|
update_results(text)
|
|
end,
|
|
})
|
|
|
|
vim.keymap.set('i', '<C-n>', function()
|
|
Fuzzy.cursor = math.min(Fuzzy.cursor + 1, #Fuzzy.matches)
|
|
Fuzzy:highlight_selection()
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.keymap.set('i', '<C-p>', function()
|
|
Fuzzy.cursor = math.max(Fuzzy.cursor - 1, 1)
|
|
Fuzzy:highlight_selection()
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.keymap.set('i', '<CR>', function()
|
|
local choice = Fuzzy.matches[Fuzzy.cursor]
|
|
if choice then
|
|
Fuzzy.close()
|
|
vim.cmd.edit(vim.fn.fnameescape(choice))
|
|
end
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.keymap.set('i', '<Esc>', function()
|
|
Fuzzy.close()
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.cmd.startinsert()
|
|
end
|
|
|
|
--------------------------------------------------------------------
|
|
-- 🟣 Ripgrep-based content search
|
|
--------------------------------------------------------------------
|
|
function Fuzzy.open_grep()
|
|
if Fuzzy.active then
|
|
Fuzzy.close()
|
|
end
|
|
Fuzzy.active = true
|
|
|
|
Fuzzy.input_buf, Fuzzy.result_buf, Fuzzy.input_win, Fuzzy.result_win = open_float('Grep: ')
|
|
Fuzzy.matches = {}
|
|
Fuzzy.cursor = 1
|
|
|
|
local function run_grep(query)
|
|
if query == '' then
|
|
vim.api.nvim_buf_set_lines(Fuzzy.result_buf, 0, -1, false, { '-- type to search --' })
|
|
return
|
|
end
|
|
local handle = io.popen('rg --vimgrep --hidden --smart-case ' .. vim.fn.shellescape(query))
|
|
if not handle then
|
|
return
|
|
end
|
|
local result = handle:read('*a')
|
|
handle:close()
|
|
Fuzzy.matches = vim.split(result, '\n', { trimempty = true })
|
|
if #Fuzzy.matches == 0 then
|
|
vim.api.nvim_buf_set_lines(Fuzzy.result_buf, 0, -1, false, { '-- no matches --' })
|
|
return
|
|
end
|
|
local display = {}
|
|
for i = 1, math.min(#Fuzzy.matches, 50) do
|
|
display[#display + 1] = Fuzzy.matches[i]
|
|
end
|
|
vim.api.nvim_buf_set_lines(Fuzzy.result_buf, 0, -1, false, display)
|
|
Fuzzy.cursor = 1
|
|
Fuzzy:highlight_selection()
|
|
end
|
|
|
|
vim.api.nvim_create_autocmd({ 'TextChangedI', 'TextChangedP' }, {
|
|
buffer = Fuzzy.input_buf,
|
|
callback = function()
|
|
local text = vim.fn.getline('.'):gsub('^Grep:%s*', '')
|
|
run_grep(text)
|
|
end,
|
|
})
|
|
|
|
vim.keymap.set('i', '<C-n>', function()
|
|
Fuzzy.cursor = math.min(Fuzzy.cursor + 1, #Fuzzy.matches)
|
|
Fuzzy:highlight_selection()
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.keymap.set('i', '<C-p>', function()
|
|
Fuzzy.cursor = math.max(Fuzzy.cursor - 1, 1)
|
|
Fuzzy:highlight_selection()
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.keymap.set('i', '<CR>', function()
|
|
local line = Fuzzy.matches[Fuzzy.cursor]
|
|
if line then
|
|
local parts = vim.split(line, ':')
|
|
local file, lnum = parts[1], tonumber(parts[2]) or 1
|
|
Fuzzy.close()
|
|
vim.cmd.edit(vim.fn.fnameescape(file))
|
|
vim.api.nvim_win_set_cursor(0, { lnum, 0 })
|
|
end
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.keymap.set('i', '<Esc>', function()
|
|
Fuzzy.close()
|
|
end, { buffer = Fuzzy.input_buf })
|
|
|
|
vim.cmd.startinsert()
|
|
end
|
|
|
|
--------------------------------------------------------------------
|
|
-- 🧩 Commands & Keymaps
|
|
--------------------------------------------------------------------
|
|
vim.api.nvim_create_user_command('FuzzyLive', function()
|
|
Fuzzy.open()
|
|
end, {})
|
|
vim.api.nvim_create_user_command('FuzzyGrep', function()
|
|
Fuzzy.open_grep()
|
|
end, {})
|
|
|
|
vim.keymap.set('n', '<leader>f', function()
|
|
vim.cmd.FuzzyLive()
|
|
end, { desc = 'Open fuzzy file finder' })
|
|
vim.keymap.set('n', '<leader>g', function()
|
|
vim.cmd.FuzzyGrep()
|
|
end, { desc = 'Search file contents with ripgrep' })
|
|
|
|
return Fuzzy
|