[AXON-1316] Update Rovo Dev to 0.12.14 and handle returned cta links (#1256)

Author: marcomura

CHANGELOG.md

@@ -5,6 +5,7 @@
 ### Features
 
 - Added edit and delete work log support
+- Added Call-to-Action links for Rovo Dev enablement errors and credit limit errors
 
 ### Bug Fixes
 

package.json

@@ -46,7 +46,7 @@
     ],
     "main": "./build/extension/extension",
     "rovoDev": {
-        "version": "0.12.10"
+        "version": "0.12.14"
     },
     "scripts": {
         "vscode:uninstall": "node ./build/extension/uninstall.js",

src/react/atlascode/rovo-dev/common/DialogMessage.tsx

@@ -20,7 +20,8 @@ export const DialogMessageItem: React.FC<{
     isRetryAfterErrorButtonEnabled?: (uid: string) => boolean;
     retryAfterError?: () => void;
     onToolPermissionChoice?: (toolCallId: string, choice: ToolPermissionChoice) => void;
-}> = ({ msg, isRetryAfterErrorButtonEnabled, retryAfterError, onToolPermissionChoice }) => {
+    customButton?: { text: string; onClick: () => void };
+}> = ({ msg, isRetryAfterErrorButtonEnabled, retryAfterError, onToolPermissionChoice, customButton }) => {
     const [title, icon] = React.useMemo(() => {
         let title: string;
         let icon: React.JSX.Element;
@@ -89,6 +90,7 @@ export const DialogMessageItem: React.FC<{
                                 </button>
                             </div>
                         )}
+
                     {msg.type === 'toolPermissionRequest' && onToolPermissionChoice && (
                         <div
                             style={{
@@ -113,6 +115,15 @@ export const DialogMessageItem: React.FC<{
                             </button>
                         </div>
                     )}
+
+                    {customButton && (
+                        <div style={{ display: 'flex', justifyContent: 'flex-end', width: '100%', marginTop: '8px' }}>
+                            <button style={inChatButtonStyles} onClick={customButton.onClick}>
+                                {customButton.text}
+                            </button>
+                        </div>
+                    )}
+
                     {msg.statusCode && <div style={{ fontSize: 'smaller', textAlign: 'right' }}>{msg.statusCode}</div>}
                 </div>
             </div>

src/react/atlascode/rovo-dev/common/common.tsx

@@ -134,6 +134,7 @@ export const renderChatHistory = (
     msg: ChatMessage,
     openFile: OpenFileFunc,
     openJira: OpenJiraFunc,
+    onLinkClick: (link: string) => void,
     checkFileExists: CheckFileExistsFunc,
     isRetryAfterErrorButtonEnabled: (uid: string) => boolean,
     retryAfterError: () => void,
@@ -154,6 +155,15 @@ export const renderChatHistory = (
                 return <ToolReturnParsedItem msg={message} openFile={openFile} />;
             });
         case '_RovoDevDialog':
+            let customButton: { text: string; onClick: () => void } | undefined = undefined;
+            if (msg.ctaLink) {
+                const { text, link } = msg.ctaLink;
+                customButton = {
+                    text,
+                    onClick: () => onLinkClick(link),
+                };
+            }
+
             return (
                 <DialogMessageItem
                     msg={msg}
@@ -162,6 +172,7 @@ export const renderChatHistory = (
                     onToolPermissionChoice={
                         () => {} /* this codepath is not supposed to have tool permissions requests */
                     }
+                    customButton={customButton}
                 />
             );
         case 'text':

src/react/atlascode/rovo-dev/landing-page/RovoDevLanding.tsx

@@ -36,6 +36,7 @@ export const RovoDevLanding: React.FC<{
     setPromptText: (context: string) => void;
     jiraWorkItems: MinimalIssue<DetailedSiteInfo>[] | undefined;
     onJiraItemClick: (issue: MinimalIssue<DetailedSiteInfo>) => void;
+    onLinkClick: (url: string) => void;
 }> = ({
     currentState,
     isHistoryEmpty,
@@ -45,6 +46,7 @@ export const RovoDevLanding: React.FC<{
     setPromptText,
     jiraWorkItems,
     onJiraItemClick,
+    onLinkClick,
 }) => {
     const shouldHideSuggestions = React.useMemo(
         () =>
@@ -87,6 +89,7 @@ export const RovoDevLanding: React.FC<{
                 onLoginClick={onLoginClick}
                 onMcpChoice={onMcpChoice}
                 onOpenFolder={onOpenFolder}
+                onLinkClick={onLinkClick}
             />
         </div>
     );

src/react/atlascode/rovo-dev/landing-page/disabled-messages/DisabledMessage.tsx

@@ -13,8 +13,9 @@ export const DisabledMessage: React.FC<{
     currentState: State;
     onLoginClick: (openApiTokenLogin: boolean) => void;
     onOpenFolder: () => void;
+    onLinkClick: (url: string) => void;
     onMcpChoice: (choice: McpConsentChoice, serverName?: string) => void;
-}> = ({ currentState, onLoginClick, onOpenFolder, onMcpChoice }) => {
+}> = ({ currentState, onLoginClick, onOpenFolder, onLinkClick, onMcpChoice }) => {
     if (currentState.state === 'Disabled' && currentState.subState === 'NeedAuth') {
         return (
             <div style={messageOuterStyles}>
@@ -53,17 +54,29 @@ export const DisabledMessage: React.FC<{
     }
 
     if (currentState.state === 'Disabled' && currentState.subState === 'EntitlementCheckFailed') {
+        let customButton: { text: string; onClick: () => void } | undefined = undefined;
+        if (currentState.detail.payload.ctaLink) {
+            const { text, link } = currentState.detail.payload.ctaLink;
+            customButton = {
+                text,
+                onClick: () => onLinkClick(link),
+            };
+        }
+
+        const message = currentState.detail.payload.message.replace('{ctaLink}', '').trim();
+
         return (
             <div style={{ ...messageOuterStyles, width: '100%' }}>
                 <DialogMessageItem
                     msg={{
                         event_kind: '_RovoDevDialog',
                         type: 'error',
                         title: currentState.detail.payload.title,
-                        text: currentState.detail.payload.message,
+                        text: message,
                         statusCode: `Failure code: ${currentState.detail.payload.status}`,
                         uid: '',
                     }}
+                    customButton={customButton}
                 />
             </div>
         );

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

@@ -49,6 +49,7 @@ export const ChatItem = React.memo<ChatItemProps>(
                     opened={drawerOpen}
                     renderProps={renderProps}
                     onCollapsiblePanelExpanded={onCollapsiblePanelExpanded}
+                    onLinkClick={onLinkClick}
                 />
             );
         } else if (block.event_kind === '_RovoDevUserPrompt' || block.event_kind === 'text') {
@@ -79,12 +80,22 @@ export const ChatItem = React.memo<ChatItemProps>(
                 return <ToolReturnParsedItem msg={message} openFile={renderProps.openFile} />;
             });
         } else if (block.event_kind === '_RovoDevDialog') {
+            let customButton: { text: string; onClick: () => void } | undefined = undefined;
+            if (block.ctaLink) {
+                const { text, link } = block.ctaLink;
+                customButton = {
+                    text,
+                    onClick: () => onLinkClick(link),
+                };
+            }
+
             return (
                 <DialogMessageItem
                     msg={block}
                     isRetryAfterErrorButtonEnabled={renderProps.isRetryAfterErrorButtonEnabled}
                     retryAfterError={renderProps.retryPromptAfterError}
                     onToolPermissionChoice={onToolPermissionChoice}
+                    customButton={customButton}
                 />
             );
         } else if (block.event_kind === '_RovoDevPullRequest') {

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

@@ -252,6 +252,7 @@ export const ChatStream: React.FC<ChatStreamProps> = ({
                     setPromptText={setPromptText}
                     jiraWorkItems={jiraWorkItems}
                     onJiraItemClick={onJiraItemClick}
+                    onLinkClick={onLinkClick}
                 />
             )}
             {!isChatHistoryDisabled && (

src/react/atlascode/rovo-dev/messaging/MessageDrawer.test.tsx

@@ -51,6 +51,7 @@ describe('MessageDrawer', () => {
                 messages={mockMessages}
                 renderProps={mockRenderProps}
                 onCollapsiblePanelExpanded={() => {}}
+                onLinkClick={() => {}}
             />,
         );
 
@@ -67,6 +68,7 @@ describe('MessageDrawer', () => {
                 messages={mockMessages}
                 renderProps={mockRenderProps}
                 onCollapsiblePanelExpanded={() => {}}
+                onLinkClick={() => {}}
             />,
         );
 
@@ -83,6 +85,7 @@ describe('MessageDrawer', () => {
                 messages={mockMessages}
                 renderProps={mockRenderProps}
                 onCollapsiblePanelExpanded={() => {}}
+                onLinkClick={() => {}}
             />,
         );
 
@@ -109,6 +112,7 @@ describe('MessageDrawer', () => {
                 messages={emptyMessages}
                 renderProps={mockRenderProps}
                 onCollapsiblePanelExpanded={() => {}}
+                onLinkClick={() => {}}
             />,
         );
 

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

@@ -16,13 +16,15 @@ interface MessageDrawerProps {
     };
     opened: boolean;
     onCollapsiblePanelExpanded: () => void;
+    onLinkClick: (link: string) => void;
 }
 
 export const MessageDrawer: React.FC<MessageDrawerProps> = ({
     messages,
     renderProps: { openFile, openJira, checkFileExists, isRetryAfterErrorButtonEnabled, retryPromptAfterError },
     onCollapsiblePanelExpanded,
     opened,
+    onLinkClick,
 }) => {
     const [isOpen, setIsOpen] = React.useState(opened);
 
@@ -62,6 +64,7 @@ export const MessageDrawer: React.FC<MessageDrawerProps> = ({
                         msg,
                         openFile,
                         openJira,
+                        onLinkClick,
                         checkFileExists,
                         isRetryAfterErrorButtonEnabled,
                         retryPromptAfterError,