Use progress notification for checkpoint restoration on accepted responses and clear progress when response preview starts

Author: robertpiosik

apps/editor/src/commands/checkpoints-command/actions/restore-checkpoint.ts

@@ -545,12 +545,18 @@ export const restore_checkpoint = async (params: {
       ? 'Successfully reverted changes.'
       : 'Checkpoint has been restored.'
 
-    if (params.options?.show_auto_closing_modal_on_success) {
+    if (params.checkpoint.trigger == 'response-accepted') {
+      params.panel_provider.send_message({
+        command: 'SHOW_PROGRESS',
+        title: message.endsWith('.') ? message.slice(0, -1) : message,
+        cancellable: false,
+        delay_visibility: true
+      })
+    } else if (params.options?.show_auto_closing_modal_on_success) {
       params.panel_provider.send_message({
         command: 'SHOW_AUTO_CLOSING_MODAL',
         title: message.slice(0, -1),
-        type: 'success',
-        non_dismissable: true
+        type: 'success'
       })
     } else if (temp_checkpoint) {
       const action = await vscode.window.showInformationMessage(

apps/editor/src/views/panel/frontend/hooks/use-modal-manager.ts

@@ -88,6 +88,8 @@ export const use_modal_manager = () => {
         set_is_preview_ongoing_modal_visible(true)
       } else if (message.command == 'SHOW_RELEVANT_FILES_MODAL') {
         set_relevant_files_modal_data({ files: message.files })
+      } else if (message.command == 'RESPONSE_PREVIEW_STARTED') {
+        set_progress_state(undefined)
       }
     }
 

apps/editor/src/views/panel/types/messages.ts

@@ -878,6 +878,12 @@ export interface ShowAutoClosingModalMessage extends BaseMessage {
   non_dismissable?: boolean
 }
 
+export interface ShowNeverClosingModalMessage extends BaseMessage {
+  command: 'SHOW_NEVER_CLOSING_MODAL'
+  title: string
+  type: 'success' | 'warning' | 'error' | 'info'
+}
+
 export interface UpdateFileInPreviewMessage extends BaseMessage {
   command: 'UPDATE_FILE_IN_PREVIEW'
   file: FileInPreview
@@ -974,6 +980,7 @@ export type BackendMessage =
   | ShowApiManagerProgressMessage
   | HideApiManagerProgressMessage
   | ShowAutoClosingModalMessage
+  | ShowNeverClosingModalMessage
   | UpdateFileInPreviewMessage
   | CollapsedStatesMessage
   | CheckpointsMessage