feat: Allow inline buttons and popup confirmation for both ACP and normal Providers (#2760)
Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com>
This commit is contained in:
65
lua/avante/ui/acp_confirm_adapter.lua
Normal file
65
lua/avante/ui/acp_confirm_adapter.lua
Normal file
@@ -0,0 +1,65 @@
|
||||
local Highlights = require("avante.highlights")
|
||||
|
||||
---@class avante.ui.ConfirmAdapter
|
||||
local M = {}
|
||||
|
||||
---@class avante.ui.ACPConfirmAdapter.ACPMappedOptions
|
||||
---@field yes? string
|
||||
---@field all? string
|
||||
---@field no? string
|
||||
|
||||
---Converts the ACP permission options to confirmation popup-compatible format (yes/all/no)
|
||||
---@param options avante.acp.PermissionOption[]
|
||||
---@return avante.ui.ACPConfirmAdapter.ACPMappedOptions
|
||||
function M.map_acp_options(options)
|
||||
local option_map = { yes = nil, all = nil, no = nil }
|
||||
|
||||
for _, opt in ipairs(options) do
|
||||
if opt.kind == "allow_once" then
|
||||
option_map.yes = opt.optionId
|
||||
elseif opt.kind == "allow_always" then
|
||||
option_map.all = opt.optionId
|
||||
elseif opt.kind == "reject_once" then
|
||||
option_map.no = opt.optionId
|
||||
|
||||
-- elseif opt.kind == "reject_always" then
|
||||
-- ignore, no 4th option in the confirm popup yet
|
||||
end
|
||||
end
|
||||
|
||||
return option_map
|
||||
end
|
||||
|
||||
---@class avante.ui.ACPConfirmAdapter.ButtonOption
|
||||
---@field id string
|
||||
---@field icon string
|
||||
---@field name string
|
||||
---@field hl? string
|
||||
|
||||
---@param options avante.acp.PermissionOption[]
|
||||
---@return avante.ui.ACPConfirmAdapter.ButtonOption[]
|
||||
function M.generate_buttons_for_acp_options(options)
|
||||
local items = vim
|
||||
.iter(options)
|
||||
:map(function(item)
|
||||
---@cast item avante.acp.PermissionOption
|
||||
local icon = item.kind == "allow_once" and "" or ""
|
||||
if item.kind == "allow_always" then icon = "" end
|
||||
local hl = nil
|
||||
if item.kind == "reject_once" or item.kind == "reject_always" then hl = Highlights.BUTTON_DANGER_HOVER end
|
||||
---@type avante.ui.ACPConfirmAdapter.ButtonOption
|
||||
local button = {
|
||||
id = item.optionId,
|
||||
name = item.name,
|
||||
icon = icon,
|
||||
hl = hl,
|
||||
}
|
||||
return button
|
||||
end)
|
||||
:totable()
|
||||
-- Sort to have "allow" first, then "allow always", then "reject"
|
||||
table.sort(items, function(a, b) return a.name < b.name end)
|
||||
return items
|
||||
end
|
||||
|
||||
return M
|
||||
Reference in New Issue
Block a user