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

feat(file_selector): support customized file selector method (#1204)

Browse Source
This commit is contained in:
guanghechen 2025-02-11 13:49:00 +08:00 committed by GitHub
parent 0bfbd39ebc
commit 6d116fac36
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lua/avante/config.lua
View File

@ -5,6 +5,11 @@
local Utils = require("avante.utils") local Utils = require("avante.utils")
---@class avante.file_selector.IParams
---@field public title string
---@field public filepaths string[]
---@field public handler fun(filepaths: string[]|nil): nil
---@class avante.CoreConfig: avante.Config ---@class avante.CoreConfig: avante.Config
local M = {} local M = {}
---@class avante.Config ---@class avante.Config
@ -327,7 +332,7 @@ M._defaults = {
}, },
--- @class AvanteFileSelectorConfig --- @class AvanteFileSelectorConfig
file_selector = { file_selector = {
--- @alias FileSelectorProvider "native" | "fzf" | "mini.pick" | "snacks" | "telescope" | string --- @alias FileSelectorProvider "native" | "fzf" | "mini.pick" | "snacks" | "telescope" | string | fun(params: avante.file_selector.IParams): nil
provider = "native", provider = "native",
-- Options override for custom providers -- Options override for custom providers
provider_opts = {}, provider_opts = {},

5
lua/avante/file_selector.lua
View File

@ -330,6 +330,11 @@ function FileSelector:show_select_ui()
self:snacks_picker_ui(handler) self:snacks_picker_ui(handler)
elseif Config.file_selector.provider == "telescope" then elseif Config.file_selector.provider == "telescope" then
self:telescope_ui(handler) self:telescope_ui(handler)
elseif type(Config.file_selector.provider) == "function" then
local title = string.format("%s:", PROMPT_TITLE) ---@type string
local filepaths = self:get_filepaths() ---@type string[]
local params = { title = title, filepaths = filepaths, handler = handler } ---@type avante.file_selector.IParams
Config.file_selector.provider(params)
else else
Utils.error("Unknown file selector provider: " .. Config.file_selector.provider) Utils.error("Unknown file selector provider: " .. Config.file_selector.provider)
end end