feat: universal selector (#1877)

2025-04-15 16:40:47 +08:00
parent 0d26590389
commit 756d1f1e24
14 changed files with 452 additions and 316 deletions
--- a/lua/avante/ui/selector/providers/telescope.lua
+++ b/lua/avante/ui/selector/providers/telescope.lua
@@ -0,0 +1,92 @@
+local Utils = require("avante.utils")
+
+local M = {}
+
+---@param selector avante.ui.Selector
+function M.show(selector)
+  local success, _ = pcall(require, "telescope")
+  if not success then
+    Utils.error("telescope is not installed. Please install telescope to use it as a file selector.")
+    return
+  end
+
+  local pickers = require("telescope.pickers")
+  local finders = require("telescope.finders")
+  local conf = require("telescope.config").values
+  local actions = require("telescope.actions")
+  local action_state = require("telescope.actions.state")
+  local previewers = require("telescope.previewers")
+
+  local items = {}
+  for _, item in ipairs(selector.items) do
+    if not vim.list_contains(selector.selected_item_ids, item.id) then table.insert(items, item) end
+  end
+
+  pickers
+    .new(
+      {},
+      vim.tbl_extend("force", {
+        prompt_title = selector.title,
+        finder = finders.new_table({
+          results = items,
+          entry_maker = function(entry)
+            return {
+              value = entry.id,
+              display = entry.title,
+              ordinal = entry.title,
+            }
+          end,
+        }),
+        sorter = conf.file_sorter(),
+        previewer = selector.get_preview_content and previewers.new_buffer_previewer({
+          title = "Preview",
+          define_preview = function(self, entry)
+            if not entry then return end
+            local content, filetype = selector.get_preview_content(entry.value)
+            local lines = vim.split(content or "", "\n")
+            -- Ensure the buffer exists and is valid before setting lines
+            if vim.api.nvim_buf_is_valid(self.state.bufnr) then
+              vim.api.nvim_buf_set_lines(self.state.bufnr, 0, -1, false, lines)
+              -- Set filetype after content is loaded
+              vim.api.nvim_set_option_value("filetype", filetype, { buf = self.state.bufnr })
+              -- Ensure cursor is within bounds
+              vim.schedule(function()
+                if vim.api.nvim_buf_is_valid(self.state.bufnr) then
+                  local row = math.min(vim.api.nvim_buf_line_count(self.state.bufnr), 1)
+                  pcall(vim.api.nvim_win_set_cursor, self.state.winnr, { row, 0 })
+                end
+              end)
+            end
+          end,
+        }),
+        attach_mappings = function(prompt_bufnr, map)
+          map("i", "<esc>", require("telescope.actions").close)
+          actions.select_default:replace(function()
+            local picker = action_state.get_current_picker(prompt_bufnr)
+
+            local selections
+            local multi_selection = picker:get_multi_selection()
+            if #multi_selection ~= 0 then
+              selections = multi_selection
+            else
+              selections = action_state.get_selected_entry()
+              selections = vim.islist(selections) and selections or { selections }
+            end
+
+            local selected_item_ids = vim
+              .iter(selections)
+              :map(function(selection) return selection.value end)
+              :totable()
+
+            selector.on_select(selected_item_ids)
+
+            actions.close(prompt_bufnr)
+          end)
+          return true
+        end,
+      }, selector.provider_opts)
+    )
+    :find()
+end
+
+return M