AXON-1778: rovo dev show restart process action for terminating rovo dev exceptions

CHANGELOG.md

@@ -4,6 +4,7 @@
 ### Features
 
 - Added Draft pull requests
+- Rovo Dev: Added "Restart Process" button to error dialogs when the Rovo Dev process terminates unexpectedly
 
 ### Improvements
 

src/rovo-dev/rovoDevWebviewProvider.ts

@@ -559,6 +559,10 @@ export class RovoDevWebviewProvider extends Disposable implements WebviewViewPro
                         await this.executeNewSession();
                         break;
 
+                    case RovoDevViewResponseType.RestartProcess:
+                        await this.executeRestartProcess();
+                        break;
+
                     case RovoDevViewResponseType.MessageRendered:
                         this._chatProvider.signalMessageRendered(e.promptId);
                         break;

src/rovo-dev/ui/common/DialogMessage.tsx

@@ -28,6 +28,7 @@ export const DialogMessageItem: React.FC<{
     customButton?: { text: string; onClick?: () => void };
     onOpenLogFile?: () => void;
     onLinkClick: (href: string) => void;
+    onRestartProcess?: () => void;
 }> = ({
     msg,
     isRetryAfterErrorButtonEnabled,
@@ -36,6 +37,7 @@ export const DialogMessageItem: React.FC<{
     customButton,
     onOpenLogFile,
     onLinkClick,
+    onRestartProcess,
 }) => {
     const [isDetailsExpanded, setIsDetailsExpanded] = React.useState(false);
     const [isStackTraceExpanded, setIsStackTraceExpanded] = React.useState(false);
@@ -140,6 +142,21 @@ export const DialogMessageItem: React.FC<{
                             </div>
                         )}
 
+                    {msg.type === 'error' && msg.isProcessTerminated && onRestartProcess && (
+                        <div
+                            style={{
+                                display: 'flex',
+                                justifyContent: 'flex-start',
+                                width: '100%',
+                                marginTop: '8px',
+                            }}
+                        >
+                            <button style={inChatButtonStyles} onClick={onRestartProcess}>
+                                Restart Process
+                            </button>
+                        </div>
+                    )}
+
                     {msg.type === 'toolPermissionRequest' && onToolPermissionChoice && (
                         <div
                             style={{

src/rovo-dev/ui/messaging/ChatItem.test.tsx

@@ -47,6 +47,7 @@ describe('ChatItem', () => {
             checkFileExists: jest.fn(),
             isRetryAfterErrorButtonEnabled: jest.fn(),
             retryPromptAfterError: jest.fn(),
+            onRestartProcess: jest.fn(),
             onOpenLogFile: jest.fn(),
             onError: jest.fn(),
         },

src/rovo-dev/ui/messaging/ChatItem.tsx

@@ -23,6 +23,7 @@ interface ChatItemProps {
         checkFileExists: CheckFileExistsFunc;
         isRetryAfterErrorButtonEnabled: (uid: string) => boolean;
         retryPromptAfterError: () => void;
+        onRestartProcess: () => void;
         onOpenLogFile: () => void;
         onError: (error: Error, errorMessage: string) => void;
     };
@@ -98,6 +99,7 @@ export const ChatItem = React.memo<ChatItemProps>(
                     msg={block}
                     isRetryAfterErrorButtonEnabled={renderProps.isRetryAfterErrorButtonEnabled}
                     retryAfterError={renderProps.retryPromptAfterError}
+                    onRestartProcess={renderProps.onRestartProcess}
                     onToolPermissionChoice={onToolPermissionChoice}
                     customButton={customButton}
                     onOpenLogFile={renderProps.onOpenLogFile}

src/rovo-dev/ui/messaging/ChatStream.tsx

@@ -30,6 +30,7 @@ interface ChatStreamProps {
         checkFileExists: CheckFileExistsFunc;
         isRetryAfterErrorButtonEnabled: (uid: string) => boolean;
         retryPromptAfterError: () => void;
+        onRestartProcess: () => void;
         onOpenLogFile: () => void;
         onError: (error: Error, errorMessage: string) => void;
     };
@@ -274,6 +275,7 @@ export const ChatStream: React.FC<ChatStreamProps> = ({
                             msg={dialog}
                             isRetryAfterErrorButtonEnabled={renderProps.isRetryAfterErrorButtonEnabled}
                             retryAfterError={renderProps.retryPromptAfterError}
+                            onRestartProcess={renderProps.onRestartProcess}
                             onToolPermissionChoice={onToolPermissionChoice}
                             onOpenLogFile={renderProps.onOpenLogFile}
                             onLinkClick={onLinkClick}

src/rovo-dev/ui/messaging/ChatStreamMessageRenderer.test.tsx

@@ -20,6 +20,7 @@ const mockRenderProps = {
     checkFileExists: jest.fn(),
     isRetryAfterErrorButtonEnabled: jest.fn(),
     retryPromptAfterError: jest.fn(),
+    onRestartProcess: jest.fn(),
     onOpenLogFile: jest.fn(),
     onError: jest.fn(),
 };

src/rovo-dev/ui/messaging/ChatStreamMessageRenderer.tsx

@@ -17,6 +17,7 @@ interface ChatStreamMessageRendererProps {
         checkFileExists: CheckFileExistsFunc;
         isRetryAfterErrorButtonEnabled: (uid: string) => boolean;
         retryPromptAfterError: () => void;
+        onRestartProcess: () => void;
         onOpenLogFile: () => void;
         onError: (error: Error, errorMessage: string) => void;
     };

src/rovo-dev/ui/rovoDevView.tsx

@@ -714,6 +714,12 @@ const RovoDevView: React.FC = () => {
         [retryAfterErrorEnabled],
     );
 
+    const handleRestartProcess = useCallback(() => {
+        postMessage({
+            type: RovoDevViewResponseType.RestartProcess,
+        });
+    }, [postMessage]);
+
     const onChangesGitPushed = useCallback(
         (msg: PullRequestMessage, pullRequestCreated: boolean) => {
             if (totalModifiedFiles.length > 0) {
@@ -1054,6 +1060,7 @@ const RovoDevView: React.FC = () => {
                         checkFileExists,
                         isRetryAfterErrorButtonEnabled,
                         retryPromptAfterError,
+                        onRestartProcess: handleRestartProcess,
                         onOpenLogFile: () => postMessage({ type: RovoDevViewResponseType.OpenRovoDevLogFile }),
                         onError,
                     }}

src/rovo-dev/ui/rovoDevViewMessages.tsx

@@ -45,6 +45,7 @@ export const enum RovoDevViewResponseType {
     MessageRendered = 'messageRendered',
     ReportRenderError = 'reportRenderError',
     StartNewSession = 'startNewSession',
+    RestartProcess = 'restartProcess',
     ShowSessionHistory = 'showSessionHistory',
     FetchSavedPrompts = 'fetchSavedPrompts',
 }
@@ -113,5 +114,6 @@ export type RovoDevViewResponse =
           }
       >
     | ReducerAction<RovoDevViewResponseType.StartNewSession>
+    | ReducerAction<RovoDevViewResponseType.RestartProcess>
     | ReducerAction<RovoDevViewResponseType.ShowSessionHistory>
     | ReducerAction<RovoDevViewResponseType.FetchSavedPrompts>;