From 9895ce76810f253eb498d3b714cb1a396553c3c0 Mon Sep 17 00:00:00 2001
From: Enes Kutay SEZEN <eneskutaysezen@gmail.com>
Date: Sun, 5 Jan 2025 12:23:52 +0300
Subject: [PATCH] feat (file_selector): Add multi select (#971)

* Add multi select

* Remove comment
---
 lua/avante/file_selector.lua | 41 +++++++++++++++++++++++++-----------
 1 file changed, 29 insertions(+), 12 deletions(-)

diff --git a/lua/avante/file_selector.lua b/lua/avante/file_selector.lua
index 5e289bf..e27fda2 100644
--- a/lua/avante/file_selector.lua
+++ b/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()