From f766b42d85d394cb9d3a5d75b60fd17eb75e2785 Mon Sep 17 00:00:00 2001
From: Brewinski <cbrewin93@gmail.com>
Date: Fri, 13 Jun 2025 14:26:25 +1000
Subject: [PATCH] fix(ui): prevent confirm dialog from closing prematurely on
 rejection (#2218)

---
 lua/avante/ui/confirm.lua | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/lua/avante/ui/confirm.lua b/lua/avante/ui/confirm.lua
index 7245aef..cf92fab 100644
--- a/lua/avante/ui/confirm.lua
+++ b/lua/avante/ui/confirm.lua
@@ -151,18 +151,22 @@ function M:open()
   end
 
   local function click_button()
-    self:close()
     if focus_index == 1 then
+      self:close()
       callback("yes")
       return
     end
     if focus_index == 2 then
+      self:close()
       Utils.notify("Accept all")
       callback("all")
       return
     end
     local prompt_input = PromptInput:new({
-      submit_callback = function(input) callback("no", input ~= "" and input or nil) end,
+      submit_callback = function(input)
+        self:close()
+        callback("no", input ~= "" and input or nil)
+      end,
       close_on_submit = true,
       win_opts = {
         relative = "win",