75 lines
2.4 KiB
Lua
75 lines
2.4 KiB
Lua
local Utils = require("avante.utils")
|
|
local M = {}
|
|
|
|
---@param selector avante.ui.Selector
|
|
function M.show(selector)
|
|
local success, fzf_lua = pcall(require, "fzf-lua")
|
|
if not success then
|
|
Utils.error("fzf-lua is not installed. Please install fzf-lua to use it as a file selector.")
|
|
return
|
|
end
|
|
|
|
local title_to_id = {}
|
|
for _, item in ipairs(selector.items) do
|
|
title_to_id[item.title] = item.id
|
|
end
|
|
|
|
local function close_action() selector.on_select(nil) end
|
|
fzf_lua.fzf_live(
|
|
function(args)
|
|
local query = args[1] or ""
|
|
local items = {}
|
|
for _, item in ipairs(vim.iter(selector.items):map(function(item) return item.title end):totable()) do
|
|
if query == "" or item:match(query:gsub("[%(%)%.%%%+%-%*%?%[%]%^%$]", "%%%1")) then
|
|
table.insert(items, item)
|
|
end
|
|
end
|
|
return items
|
|
end,
|
|
vim.tbl_deep_extend("force", {
|
|
prompt = selector.title,
|
|
preview = selector.get_preview_content and function(item)
|
|
local id = title_to_id[item[1]]
|
|
local content = selector.get_preview_content(id)
|
|
return content
|
|
end or nil,
|
|
fzf_opts = { ["--multi"] = true },
|
|
git_icons = false,
|
|
actions = {
|
|
["default"] = function(selected)
|
|
if not selected or #selected == 0 then return close_action() end
|
|
---@type string[]
|
|
local selections = {}
|
|
for _, entry in ipairs(selected) do
|
|
local id = title_to_id[entry]
|
|
if id then table.insert(selections, id) end
|
|
end
|
|
|
|
selector.on_select(selections)
|
|
end,
|
|
["esc"] = close_action,
|
|
["ctrl-c"] = close_action,
|
|
["ctrl-delete"] = {
|
|
fn = function(selected)
|
|
if not selected or #selected == 0 then return close_action() end
|
|
local selections = selected
|
|
vim.ui.input({ prompt = "Remove·selection?·(" .. #selections .. " items) [y/N]" }, function(input)
|
|
if input and input:lower() == "y" then
|
|
for _, selection in ipairs(selections) do
|
|
selector.on_delete_item(title_to_id[selection])
|
|
for i, item in ipairs(selector.items) do
|
|
if item.id == title_to_id[selection] then table.remove(selector.items, i) end
|
|
end
|
|
end
|
|
end
|
|
end)
|
|
end,
|
|
reload = true,
|
|
},
|
|
},
|
|
}, selector.provider_opts)
|
|
)
|
|
end
|
|
|
|
return M
|