fix: make preview (q)uit cancel the whole process

Author: frankroeder

lua/parrot/preview.lua

@@ -20,6 +20,7 @@ function Preview:new(options)
   self.close_fn = nil
   self.apply_fn = nil
   self.reject_fn = nil
+  self.quit_fn = nil
   self.auto_apply_timer = nil
   return self
 end
@@ -122,6 +123,7 @@ end
 ---@param filename string|nil Optional filename to display in header
 ---@param start_line number|nil Optional start line number
 ---@param end_line number|nil Optional end line number
+---@param quit_callback function|nil Optional function to call when quitting (defaults to reject)
 function Preview:show_diff_preview(
   old_content,
   new_content,
@@ -130,7 +132,8 @@ function Preview:show_diff_preview(
   reject_callback,
   filename,
   start_line,
-  end_line
+  end_line,
+  quit_callback
 )
   if not self.options.enable_preview_mode then
     -- Preview disabled, apply changes directly
@@ -206,6 +209,7 @@ function Preview:show_diff_preview(
   -- Store callbacks
   self.apply_fn = apply_callback
   self.reject_fn = reject_callback
+  self.quit_fn = quit_callback
 
   -- Set up keymaps
   self:setup_preview_keymaps()
@@ -308,7 +312,12 @@ end
 
 --- Closes the preview without applying changes
 function Preview:close_preview()
-  self:reject_changes()
+  if self.quit_fn then
+    self.quit_fn()
+  else
+    self:reject_changes()
+  end
+  self:cleanup()
 end
 
 --- Cleans up timers and closes preview window
@@ -331,6 +340,7 @@ function Preview:cleanup()
   self.close_fn = nil
   self.apply_fn = nil
   self.reject_fn = nil
+  self.quit_fn = nil
 end
 
 return Preview

lua/parrot/preview_response_handler.lua

@@ -124,11 +124,23 @@ function PreviewResponseHandler:show_preview()
     end
   end
 
-  self.preview:show_diff_preview(self.original_content, new_content, self.target_type, function()
-    self:apply_changes()
-  end, function()
-    self:reject_changes()
-  end, filename, self.start_line, self.end_line)
+  self.preview:show_diff_preview(
+    self.original_content,
+    new_content,
+    self.target_type,
+    function()
+      self:apply_changes()
+    end,
+    function()
+      self:reject_changes()
+    end,
+    filename,
+    self.start_line,
+    self.end_line,
+    function()
+      self:quit_changes()
+    end
+  )
 end
 
 --- Prepares the new content based on target type
@@ -240,6 +252,12 @@ function PreviewResponseHandler:reject_changes()
   vim.cmd("doautocmd User PrtPreviewRejected")
 end
 
+--- Quits the preview without any prompts or actions
+function PreviewResponseHandler:quit_changes()
+  logger.debug("PreviewResponseHandler: Preview quit")
+  vim.cmd("doautocmd User PrtPreviewQuit")
+end
+
 --- Creates a handler function for the response processing
 ---@return function
 function PreviewResponseHandler:create_handler()