[AXON-494] Implement auto-downloader of Rovo Dev binaries (#712)

Author: marcomura

package-lock.json

@@ -1376,6 +1376,7 @@
         "@speed-highlight/core": "^1.2.7",
         "@vscode/codicons": "^0.0.36",
         "@vscode/webview-ui-toolkit": "^1.4.0",
+        "adm-zip": "^0.5.16",
         "awesome-debounce-promise": "^2.1.0",
         "axios": "1.9.0",
         "axios-curlirize": "^2.0.0",
@@ -1441,6 +1442,7 @@
         "@jest/globals": "^30.0.2",
         "@playwright/test": "1.53.0",
         "@testing-library/react": "^12.1.5",
+        "@types/adm-zip": "^0.5.7",
         "@types/express": "^5.0.2",
         "@types/git-url-parse": "^9.0.1",
         "@types/jest": "^30.0.0",

package.json

@@ -219,7 +219,11 @@ export class Container {
                 }),
             );
             context.subscriptions.push(
-                (this._rovodevWebviewProvider = new RovoDevWebviewProvider(context.extensionPath, context.globalState)),
+                (this._rovodevWebviewProvider = new RovoDevWebviewProvider(
+                    context,
+                    context.extensionPath,
+                    context.globalState,
+                )),
             );
             this.configureRovodevSettingsCommands(context);
         }

src/container.ts

@@ -26,7 +26,7 @@ import {
     BB_PIPELINES_FILENAME,
 } from './pipelines/yaml/pipelinesYamlHelper';
 import { registerResources } from './resources';
-import { deactivateRovoDevProcessManager, initializeRovoDevProcessManager } from './rovo-dev/rovoDevProcessManager';
+import { deactivateRovoDevProcessManager } from './rovo-dev/rovoDevProcessManager';
 import { GitExtension } from './typings/git';
 import { Experiments, FeatureFlagClient, Features } from './util/featureFlags';
 import Performance from './util/perf';
@@ -113,13 +113,7 @@ export async function activate(context: ExtensionContext) {
         // icon to appear in the activity bar
         activateBitbucketFeatures();
         activateYamlFeatures(context);
-
-        if (!!process.env.ROVODEV_ENABLED) {
-            initializeRovoDevProcessManager(context);
-        }
-    }
-
-    if (!!process.env.ROVODEV_BBY && !!process.env.ROVODEV_ENABLED) {
+    } else if (!!process.env.ROVODEV_ENABLED) {
         commands.executeCommand('workbench.view.extension.atlascode-rovo-dev');
     }
 

src/extension.ts

@@ -1,25 +1,18 @@
 import * as React from 'react';
+import { RovoDevInitState, State } from 'src/rovo-dev/rovoDevTypes';
 import { RovoDevProviderMessage, RovoDevProviderMessageType } from 'src/rovo-dev/rovoDevWebviewProviderMessages';
 import { ConnectionTimeout } from 'src/util/time';
 
 import { useMessagingApi } from '../../messagingApi';
 import { ErrorMessageItem, FollowUpActionFooter, OpenFileFunc } from '../common/common';
 import { PullRequestChatItem, PullRequestForm } from '../create-pr/PullRequestForm';
 import { RovoDevLanding } from '../rovoDevLanding';
-import { State } from '../rovoDevView';
 import { RovoDevViewResponse, RovoDevViewResponseType } from '../rovoDevViewMessages';
 import { CodePlanButton } from '../technical-plan/CodePlanButton';
 import { TechnicalPlanComponent } from '../technical-plan/TechnicalPlanComponent';
 import { ToolCallItem } from '../tools/ToolCallItem';
 import { ToolReturnParsedItem } from '../tools/ToolReturnItem';
-import {
-    ChatMessage,
-    DefaultMessage,
-    MessageBlockDetails,
-    parseToolReturnMessage,
-    scrollToEnd,
-    ToolReturnParseResult,
-} from '../utils';
+import { ChatMessage, DefaultMessage, MessageBlockDetails, parseToolReturnMessage, scrollToEnd } from '../utils';
 import { ChatMessageItem } from './ChatMessageItem';
 import { MessageDrawer } from './MessageDrawer';
 
@@ -35,11 +28,12 @@ interface ChatStreamProps {
     messagingApi: ReturnType<
         typeof useMessagingApi<RovoDevViewResponse, RovoDevProviderMessage, RovoDevProviderMessage>
     >;
-    modifiedFiles: ToolReturnParseResult[];
     pendingToolCall: string;
     deepPlanCreated: boolean;
     executeCodePlan: () => void;
     state: State;
+    initState: RovoDevInitState;
+    downloadProgress: [number, number];
     onChangesGitPushed: (msg: DefaultMessage, pullRequestCreated: boolean) => void;
     onCollapsiblePanelExpanded: () => void;
 }
@@ -53,8 +47,9 @@ export const ChatStream: React.FC<ChatStreamProps> = ({
     deepPlanCreated,
     executeCodePlan,
     state,
+    initState,
+    downloadProgress,
     messagingApi,
-    modifiedFiles,
     onChangesGitPushed,
     onCollapsiblePanelExpanded,
 }) => {
@@ -249,7 +244,7 @@ export const ChatStream: React.FC<ChatStreamProps> = ({
 
             {pendingToolCall && (
                 <div style={{ marginBottom: '16px' }}>
-                    <ToolCallItem toolMessage={pendingToolCall} />
+                    <ToolCallItem toolMessage={pendingToolCall} state={initState} downloadProgress={downloadProgress} />
                 </div>
             )}
 

src/react/atlascode/rovo-dev/messaging/ChatStream.tsx

@@ -1,7 +1,7 @@
 import { fireEvent, render, screen } from '@testing-library/react';
 import React from 'react';
+import { State } from 'src/rovo-dev/rovoDevTypes';
 
-import { State } from '../../rovoDevView';
 import { PromptInputBox } from './PromptInput';
 
 describe('PromptInputBox', () => {

src/react/atlascode/rovo-dev/prompt-box/prompt-input/PromptInput.test.tsx

@@ -2,15 +2,17 @@ import { LoadingButton } from '@atlaskit/button';
 import SendIcon from '@atlaskit/icon/core/arrow-up';
 import StopIcon from '@atlaskit/icon/core/video-stop';
 import React from 'react';
+import { State } from 'src/rovo-dev/rovoDevTypes';
 
-import { AiGenerativeTextSummaryIcon, CloseIconDeepPlan, State } from '../../rovoDevView';
+import { AiGenerativeTextSummaryIcon, CloseIconDeepPlan } from '../../rovoDevView';
 import {
     rovoDevDeepPlanStylesSelector,
     rovoDevPromptButtonStyles,
     rovoDevTextareaStyles,
 } from '../../rovoDevViewStyles';
 
 interface PromptInputBoxProps {
+    disabled?: boolean;
     state: State;
     promptText: string;
     onPromptTextChange: (text: string) => void;
@@ -21,7 +23,22 @@ interface PromptInputBoxProps {
     sendButtonDisabled?: boolean;
     onAddContext: () => void;
 }
+
+const TextAreaMessages: Record<State, string> = {
+    [State.Disabled]: 'Rovo Dev is currently disabled. Please, refer to the error message in chat',
+    [State.WaitingForPrompt]: 'Type in a question',
+    [State.NoWorkspaceOpen]: 'Please, open a folder to start a chat session with Rovo Dev',
+    [State.GeneratingResponse]: 'Generating response...',
+    [State.CancellingResponse]: 'Cancelling the response...',
+    [State.ExecutingPlan]: 'Executing the code plan...',
+};
+
+const getTextAreaPlaceholder = (state: State) => {
+    return TextAreaMessages[state];
+};
+
 export const PromptInputBox: React.FC<PromptInputBoxProps> = ({
+    disabled,
     state,
     promptText,
     onPromptTextChange,
@@ -32,13 +49,6 @@ export const PromptInputBox: React.FC<PromptInputBoxProps> = ({
     sendButtonDisabled = false,
     onAddContext,
 }) => {
-    const TextAreaMessages: Record<State, string> = {
-        [State.WaitingForPrompt]: 'Type in a question',
-        [State.GeneratingResponse]: 'Generating response...',
-        [State.CancellingResponse]: 'Cancelling the response...',
-        [State.ExecutingPlan]: 'Executing the code plan...',
-    };
-
     const handleKeyDown = React.useCallback(
         (event: React.KeyboardEvent<HTMLTextAreaElement>) => {
             if (event.key === 'Enter' && !event.shiftKey && state === State.WaitingForPrompt) {
@@ -48,18 +58,16 @@ export const PromptInputBox: React.FC<PromptInputBoxProps> = ({
         },
         [state, onSend, promptText],
     );
-    const getTextAreaPlaceholder = () => {
-        return TextAreaMessages[state] || 'Type in a question';
-    };
 
     return (
         <>
             <textarea
                 style={{ ...{ fieldSizing: 'content' }, ...rovoDevTextareaStyles }}
-                placeholder={getTextAreaPlaceholder()}
+                placeholder={getTextAreaPlaceholder(state)}
                 onChange={(element) => onPromptTextChange(element.target.value)}
                 onKeyDown={handleKeyDown}
                 value={promptText}
+                disabled={disabled}
             />
             <div
                 style={{
@@ -71,54 +79,61 @@ export const PromptInputBox: React.FC<PromptInputBoxProps> = ({
                 }}
             >
                 {/* Left-side Add Context Button */}
-                <LoadingButton
-                    style={{
-                        ...rovoDevPromptButtonStyles,
-                    }}
-                    spacing="compact"
-                    label="Add context"
-                    iconBefore={<i className="codicon codicon-add" />}
-                    onClick={() => onAddContext()}
-                />
-                <div style={{ display: 'flex', gap: 8 }}>
-                    <LoadingButton
-                        style={{
-                            ...rovoDevDeepPlanStylesSelector(isDeepPlanEnabled, state !== State.WaitingForPrompt),
-                        }}
-                        spacing="compact"
-                        label="Enable deep plan"
-                        iconBefore={<AiGenerativeTextSummaryIcon />}
-                        iconAfter={isDeepPlanEnabled ? <CloseIconDeepPlan /> : undefined}
-                        isDisabled={state !== State.WaitingForPrompt}
-                        onClick={() => onDeepPlanToggled()}
-                    >
-                        {isDeepPlanEnabled ? 'Deep plan enabled' : ''}
-                    </LoadingButton>
-                    {state === State.WaitingForPrompt && (
+                {!disabled && (
+                    <>
                         <LoadingButton
                             style={{
                                 ...rovoDevPromptButtonStyles,
-                                color: 'var(--vscode-button-foreground) !important',
-                                backgroundColor: 'var(--vscode-button-background)',
                             }}
                             spacing="compact"
-                            label="Send prompt"
-                            iconBefore={<SendIcon label="Send prompt" />}
-                            isDisabled={sendButtonDisabled}
-                            onClick={() => onSend(promptText)}
-                        />
-                    )}
-                    {state !== State.WaitingForPrompt && (
-                        <LoadingButton
-                            style={rovoDevPromptButtonStyles}
-                            spacing="compact"
-                            label="Stop"
-                            iconBefore={<StopIcon label="Stop" />}
-                            isDisabled={state === State.CancellingResponse}
-                            onClick={() => onCancel()}
+                            label="Add context"
+                            iconBefore={<i className="codicon codicon-add" />}
+                            onClick={() => onAddContext()}
                         />
-                    )}
-                </div>
+                        <div style={{ display: 'flex', gap: 8 }}>
+                            <LoadingButton
+                                style={{
+                                    ...rovoDevDeepPlanStylesSelector(
+                                        isDeepPlanEnabled,
+                                        state !== State.WaitingForPrompt,
+                                    ),
+                                }}
+                                spacing="compact"
+                                label="Enable deep plan"
+                                iconBefore={<AiGenerativeTextSummaryIcon />}
+                                iconAfter={isDeepPlanEnabled ? <CloseIconDeepPlan /> : undefined}
+                                isDisabled={state !== State.WaitingForPrompt}
+                                onClick={() => onDeepPlanToggled()}
+                            >
+                                {isDeepPlanEnabled ? 'Deep plan enabled' : ''}
+                            </LoadingButton>
+                            {state === State.WaitingForPrompt && (
+                                <LoadingButton
+                                    style={{
+                                        ...rovoDevPromptButtonStyles,
+                                        color: 'var(--vscode-button-foreground) !important',
+                                        backgroundColor: 'var(--vscode-button-background)',
+                                    }}
+                                    spacing="compact"
+                                    label="Send prompt"
+                                    iconBefore={<SendIcon label="Send prompt" />}
+                                    isDisabled={sendButtonDisabled}
+                                    onClick={() => onSend(promptText)}
+                                />
+                            )}
+                            {state !== State.WaitingForPrompt && (
+                                <LoadingButton
+                                    style={rovoDevPromptButtonStyles}
+                                    spacing="compact"
+                                    label="Stop"
+                                    iconBefore={<StopIcon label="Stop" />}
+                                    isDisabled={state === State.CancellingResponse}
+                                    onClick={() => onCancel()}
+                                />
+                            )}
+                        </div>
+                    </>
+                )}
             </div>
         </>
     );