phyer/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat (file_selector): Add multi select ()

Browse Source 
* Add multi select

* Remove comment
This commit is contained in:
Enes Kutay SEZEN 2025-01-05 12:23:52 +03:00 committed by GitHub
parent 42c1d292cf
commit 9895ce7681
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 29 additions and 12 deletions

41
lua/avante/file_selector.lua

@ -163,6 +163,7 @@ function FileSelector:telescope_ui(handler)
local conf = require("telescope.config").values
local actions = require("telescope.actions")
local action_state = require("telescope.actions.state")
local action_utils = require("telescope.actions.utils")
pickers
.new(
@ -176,9 +177,21 @@ function FileSelector:telescope_ui(handler)
map("i", "<esc>", require("telescope.actions").close)
actions.select_default:replace(function()
local picker = action_state.get_current_picker(prompt_bufnr)
if #picker:get_multi_selection() ~= 0 then
local selections = {}
action_utils.map_selections(prompt_bufnr, function(selection) table.insert(selections, selection[1]) end)
if #selections > 0 then handler(selections) end
else
local selection = action_state.get_selected_entry()
handler(selection[1])
end
actions.close(prompt_bufnr)
local selection = action_state.get_selected_entry()
handler(selection[1])
end)
return true
end,
@ -202,19 +215,23 @@ end
---@return nil
function FileSelector:show_select_ui()
local handler = function(filepath)
if not filepath then return end
local uniform_path = Utils.uniform_path(filepath)
if Config.file_selector.provider == "native" then
-- Native handler filters out already selected files
table.insert(self.selected_filepaths, uniform_path)
self:emit("update")
else
if not vim.tbl_contains(self.selected_filepaths, uniform_path) then
local handler = function(filepaths)
if not filepaths then return end
-- Convert single filepath to array for unified handling
local paths = type(filepaths) == "string" and { filepaths } or filepaths
for _, filepath in ipairs(paths) do
local uniform_path = Utils.uniform_path(filepath)
if Config.file_selector.provider == "native" then
table.insert(self.selected_filepaths, uniform_path)
self:emit("update")
else
if not vim.tbl_contains(self.selected_filepaths, uniform_path) then
table.insert(self.selected_filepaths, uniform_path)
end
end
end
self:emit("update")
end
vim.schedule(function()