feat: customer pricing details for amazon q

laileni-aws · laileni-aws · commit 657d64127a92 · 2025-07-22T01:08:07.000-07:00
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/amazonq/package.json b/packages/amazonq/package.json
@@ -388,20 +388,25 @@
                     "when": "view == aws.amazonq.AmazonQChatView",
                     "group": "0_topAmazonQ@1"
                 },
+                {
+                    "command": "aws.amazonq.accountDetails",
+                    "when": "(view == aws.amazonq.AmazonQChatView) && aws.codewhisperer.connected && !aws.isSageMakerUnifiedStudio && aws.amazonq.connectedSsoIdc == true",
+                    "group": "1_amazonQ@1"
+                },
                 {
                     "command": "aws.amazonq.selectRegionProfile",
                     "when": "view == aws.amazonq.AmazonQChatView && aws.amazonq.connectedSsoIdc == true",
-                    "group": "1_amazonQ@1"
+                    "group": "1_amazonQ@2"
                 },
                 {
                     "command": "aws.amazonq.manageSubscription",
                     "when": "(view == aws.amazonq.AmazonQChatView) && aws.codewhisperer.connected",
-                    "group": "1_amazonQ@2"
+                    "group": "1_amazonQ@3"
                 },
                 {
                     "command": "aws.amazonq.learnMore",
                     "when": "view =~ /^aws\\.amazonq/",
-                    "group": "1_amazonQ@3"
+                    "group": "1_amazonQ@4"
                 },
                 {
                     "command": "aws.amazonq.signout",
@@ -709,6 +714,11 @@
                 "title": "%AWS.amazonq.learnMore%",
                 "category": "%AWS.amazonq.title%"
             },
+            {
+                "command": "aws.amazonq.accountDetails",
+                "title": "%AWS.amazonq.accountDetails%",
+                "category": "%AWS.amazonq.title%"
+            },
             {
                 "command": "amazonq.dev.openMenu",
                 "title": "Open Developer Menu",
diff --git a/packages/amazonq/src/lsp/chat/activation.ts b/packages/amazonq/src/lsp/chat/activation.ts
@@ -90,6 +90,22 @@ export async function activate(languageClient: LanguageClient, encryptionKey: Bu
                     getLogger('amazonqLsp').error('failed request: aws/chat/manageSubscription: %O', e)
                 })
         }),
+        /* 
+            Forward the subscription details request to language server directly
+            This triggers the onShowSubscription() method in the language server
+        */
+        Commands.register('aws.amazonq.accountDetails', () => {
+            getLogger('amazonqLsp').info(
+                '[VSCode Client]: Sending subscription details [aws/chat/subscription/show] event notification'
+            )
+            languageClient
+                .sendRequest('workspace/executeCommand', {
+                    command: 'aws/chat/subscription/show',
+                })
+                .catch((e) => {
+                    getLogger('amazonqLsp').error('Failed to send subscription details request: : %O', e)
+                })
+        }),
         globals.logOutputChannel.onDidChangeLogLevel((logLevel) => {
             getLogger('amazonqLsp').info(`Local log level changed to ${logLevel}, notifying LSP`)
             void pushConfigUpdate(languageClient, {
diff --git a/packages/amazonq/src/lsp/chat/messages.ts b/packages/amazonq/src/lsp/chat/messages.ts
@@ -66,6 +66,8 @@ import {
     ShowOpenDialogParams,
     openFileDialogRequestType,
     OpenFileDialogResult,
+    subscriptionDetailsNotificationType,
+    subscriptionUpgradeNotificationType,
 } from '@aws/language-server-runtimes/protocol'
 import { v4 as uuidv4 } from 'uuid'
 import * as vscode from 'vscode'
@@ -94,7 +96,7 @@ import { isValidResponseError } from './error'
 import { decryptResponse, encryptRequest } from '../encryption'
 import { getCursorState } from '../utils'
 import { focusAmazonQPanel } from './commands'
-import { ChatMessage } from '@aws/language-server-runtimes/server-interface'
+import { ChatMessage, SUBSCRIPTION_UPGRADE_NOTIFICATION_METHOD } from '@aws/language-server-runtimes/server-interface'
 
 export function registerActiveEditorChangeListener(languageClient: LanguageClient) {
     let debounceTimer: NodeJS.Timeout | undefined
@@ -126,6 +128,22 @@ export function registerLanguageServerEventListener(languageClient: LanguageClie
     )
 
     const chatOptions = languageClient.initializeResult?.awsServerCapabilities?.chatOptions
+    /*
+        Set subscriptionDetails based on request
+        Set to false if subscriptionDetails is not in the request
+        Set to true if subscriptionDetails is true in the request
+        Set to false otherwise
+    */
+    if (chatOptions) {
+        languageClient.info(
+            'Subscription details capability:',
+            `Server:${languageClient.initializeResult?.awsServerCapabilities?.chatOptions?.subscriptionDetails}`
+        )
+
+        const requestSubscriptionDetails =
+            languageClient.initializeResult?.awsServerCapabilities?.chatOptions?.subscriptionDetails
+        chatOptions.subscriptionDetails = requestSubscriptionDetails === true
+    }
 
     // overide the quick action commands provided by flare server initialization, which doesn't provide the group header
     if (chatOptions?.quickActions?.quickActionsCommandGroups?.[0]) {
@@ -244,6 +262,18 @@ export function registerMessageListeners(
                 }
                 break
             }
+            case SUBSCRIPTION_UPGRADE_NOTIFICATION_METHOD: {
+                languageClient.info('[VSCode Client] Subscription upgrade clicked')
+                try {
+                    // Forward the upgrade request to Flare
+                    languageClient.sendNotification(subscriptionUpgradeNotificationType.method, message.params)
+                } catch (e) {
+                    languageClient.error(
+                        `[VSCode Client] Failed to send subscription upgrade notification: ${(e as Error).message}`
+                    )
+                }
+                break
+            }
             case DISCLAIMER_ACKNOWLEDGED: {
                 void AmazonQPromptSettings.instance.update('amazonQChatDisclaimer', true)
                 break
@@ -653,6 +683,46 @@ export function registerMessageListeners(
             params: params,
         })
     })
+
+    // Handle subscription details response from Flare
+    languageClient.onNotification(subscriptionDetailsNotificationType.method, (params) => {
+        languageClient.info(`[VSCode Client] Received subscription details: ${JSON.stringify(params)}`)
+        // Forward the subscription details to the Chat UI
+        void provider.webview?.postMessage({
+            command: subscriptionDetailsNotificationType.method,
+            params: params,
+        })
+    })
+    // TODO: After implementing the "Upgrade" feature, test its functionality in the Account Details tab.
+    // Handle window/showDocument requests from Flare for subscription upgrade
+    languageClient.onRequest<ShowDocumentParams, ShowDocumentResult>(
+        ShowDocumentRequest.method,
+        async (params: ShowDocumentParams): Promise<ShowDocumentParams | ResponseError<ShowDocumentResult>> => {
+            try {
+                const uri = vscode.Uri.parse(params.uri)
+
+                if (params.external) {
+                    languageClient.info(`[VSCode Client] Opening external URL for subscription upgrade: ${params.uri}`)
+                    // Open the URL in the external browser
+                    await vscode.env.openExternal(uri)
+
+                    // Return success to indicate the URL was opened
+                    return params
+                }
+
+                // If not external, handle as a regular document
+                const doc = await vscode.workspace.openTextDocument(uri)
+                await vscode.window.showTextDocument(doc, { preview: false })
+                return params
+            } catch (e) {
+                languageClient.error(`[VSCode Client] Failed to open subscription upgrade URL: ${(e as Error).message}`)
+                return new ResponseError(
+                    LSPErrorCodes.RequestFailed,
+                    `Failed to open subscription upgrade URL: ${(e as Error).message}`
+                )
+            }
+        }
+    )
 }
 
 function isServerEvent(command: string) {
diff --git a/packages/amazonq/src/lsp/client.ts b/packages/amazonq/src/lsp/client.ts
@@ -168,6 +168,7 @@ export async function startLanguageServer(
                         reroute: true,
                         modelSelection: true,
                         workspaceFilePath: vscode.workspace.workspaceFile?.fsPath,
+                        subscriptionDetails: true,
                     },
                     window: {
                         notifications: true,
diff --git a/packages/core/package.json b/packages/core/package.json
@@ -470,9 +470,9 @@
     "devDependencies": {
         "@aws-sdk/types": "^3.13.1",
         "@aws/chat-client": "^0.1.4",
-        "@aws/chat-client-ui-types": "^0.1.47",
-        "@aws/language-server-runtimes": "^0.2.102",
-        "@aws/language-server-runtimes-types": "^0.1.43",
+        "@aws/chat-client-ui-types": "^0.1.55",
+        "@aws/language-server-runtimes": "^0.2.112",
+        "@aws/language-server-runtimes-types": "^0.1.49",
         "@cspotcode/source-map-support": "^0.8.1",
         "@sinonjs/fake-timers": "^10.0.2",
         "@types/adm-zip": "^0.4.34",
diff --git a/packages/core/package.nls.json b/packages/core/package.nls.json
@@ -353,6 +353,7 @@
     "AWS.amazonq.security": "Code Issues",
     "AWS.amazonq.login": "Login",
     "AWS.amazonq.learnMore": "Learn More About Amazon Q",
+    "AWS.amazonq.accountDetails": "Account details",
     "AWS.amazonq.exploreAgents": "Explore Agent Capabilities",
     "AWS.amazonq.welcomeWalkthrough": "Welcome Walkthrough",
     "AWS.amazonq.codewhisperer.title": "Amazon Q",
