feat(amazonq): support pinned context

Author: avi-alpert

packages/amazonq/src/lsp/chat/activation.ts

@@ -7,7 +7,11 @@ import { window } from 'vscode'
 import { LanguageClient } from 'vscode-languageclient'
 import { AmazonQChatViewProvider } from './webviewProvider'
 import { focusAmazonQPanel, registerCommands } from './commands'
-import { registerLanguageServerEventListener, registerMessageListeners } from './messages'
+import {
+    registerActiveEditorChangeListener,
+    registerLanguageServerEventListener,
+    registerMessageListeners,
+} from './messages'
 import { Commands, getLogger, globals, undefinedIfEmpty } from 'aws-core-vscode/shared'
 import { activate as registerLegacyChatListeners } from '../../app/chat/activation'
 import { DefaultAmazonQAppInitContext } from 'aws-core-vscode/amazonq'
@@ -33,6 +37,7 @@ export async function activate(languageClient: LanguageClient, encryptionKey: Bu
      **/
     registerCommands(provider)
     registerLanguageServerEventListener(languageClient, provider)
+    registerActiveEditorChangeListener(languageClient)
 
     provider.onDidResolveWebview(() => {
         const disposable = DefaultAmazonQAppInitContext.instance.getAppsToWebViewMessageListener().onMessage((msg) => {

packages/amazonq/src/lsp/chat/messages.ts

@@ -62,7 +62,7 @@ import { Disposable, LanguageClient, Position, TextDocumentIdentifier } from 'vs
 import * as jose from 'jose'
 import { AmazonQChatViewProvider } from './webviewProvider'
 import { AuthUtil, ReferenceLogViewProvider } from 'aws-core-vscode/codewhisperer'
-import { amazonQDiffScheme, AmazonQPromptSettings, messages, openUrl } from 'aws-core-vscode/shared'
+import { amazonQDiffScheme, AmazonQPromptSettings, messages, openUrl, isTextEditor } from 'aws-core-vscode/shared'
 import {
     DefaultAmazonQAppInitContext,
     messageDispatcher,
@@ -74,6 +74,30 @@ import { telemetry, TelemetryBase } from 'aws-core-vscode/telemetry'
 import { isValidResponseError } from './error'
 import { focusAmazonQPanel } from './commands'
 
+export function registerActiveEditorChangeListener(languageClient: LanguageClient) {
+    let debounceTimer: NodeJS.Timeout | undefined
+    vscode.window.onDidChangeActiveTextEditor((editor) => {
+        if (debounceTimer) {
+            clearTimeout(debounceTimer)
+        }
+        debounceTimer = setTimeout(() => {
+            let textDocument = undefined
+            let cursorState = undefined
+            if (editor) {
+                textDocument = {
+                    uri: editor.document.uri.toString(),
+                }
+                cursorState = getCursorState(editor.selections)
+            }
+            // todo: replace with message from lsp once consuming latest language-server-runtimes
+            languageClient.sendNotification('aws/chat/activeEditorChanged', {
+                textDocument,
+                cursorState,
+            })
+        }, 100)
+    })
+}
+
 export function registerLanguageServerEventListener(languageClient: LanguageClient, provider: AmazonQChatViewProvider) {
     languageClient.info(
         'Language client received initializeResult from server:',
@@ -316,6 +340,8 @@ export function registerMessageListeners(
                 )
                 break
             }
+            case 'aws/chat/listRules': // todo: switch to imported methods from language-server-runtimes
+            case 'aws/chat/ruleClick':
             case listConversationsRequestType.method:
             case conversationClickRequestType.method:
             case listMcpServersRequestType.method:
@@ -471,6 +497,20 @@ export function registerMessageListeners(
             params: params,
         })
     })
+    languageClient.onNotification(
+        'aws/chat/sendPinnedContext', // todo: switch to type from language-server-runtimes
+        (params: ContextCommandParams & { tabId: string; textDocument?: TextDocumentIdentifier }) => {
+            const editor = vscode.window.activeTextEditor
+            let textDocument = undefined
+            if (editor && isTextEditor(editor)) {
+                textDocument = { uri: vscode.workspace.asRelativePath(editor.document.uri) }
+            }
+            void provider.webview?.postMessage({
+                command: 'aws/chat/sendPinnedContext', // todo: switch to type from language-server-runtimes
+                params: { ...params, textDocument },
+            })
+        }
+    )
 
     languageClient.onNotification(openFileDiffNotificationType.method, async (params: OpenFileDiffParams) => {
         const ecc = new EditorContentController()

packages/amazonq/src/lsp/client.ts

@@ -123,6 +123,7 @@ export async function startLanguageServer(
                 awsClientCapabilities: {
                     q: {
                         developerProfiles: true,
+                        pinnedContextEnabled: true,
                         mcp: true,
                     },
                     window: {

packages/core/src/amazonq/webview/ui/quickActions/handler.ts

@@ -362,7 +362,8 @@ export class QuickActionHandler {
                 cancelButtonWhenLoading: false,
             })
         } else {
-            this.mynahUI.updateStore(affectedTabId, { promptInputOptions: [] })
+            // todo: remove type cast once updated to latest prod mynah that includes promptTopBarTitle
+            this.mynahUI.updateStore(affectedTabId, { promptInputOptions: [], promptTopBarTitle: '' } as any)
         }
 
         if (affectedTabId && this.isHybridChatEnabled) {

packages/core/src/shared/index.ts

@@ -51,6 +51,7 @@ export * from './vscode/commands2'
 export * from './utilities/pathUtils'
 export * from './utilities/zipStream'
 export * from './errors'
+export { isTextEditor } from './utilities/editorUtilities'
 export * as messages from './utilities/messages'
 export * as errors from './errors'
 export * as funcUtil from './utilities/functionUtils'