feat(amazonq): support context commands in agentic chat (#948)

* feat(amazonq): chat context commands for prompts

feat(amazonq): integrate with local context server

fix(amazonq): fixes and refactor

* fix(amazonq): pr comments

* fix(amazonq): dispose contextCommandProvider

* fix(amazonq): fix test

Author: zixlin7

package-lock.json

@@ -38,6 +38,7 @@
         "adm-zip": "^0.5.10",
         "archiver": "^7.0.1",
         "aws-sdk": "^2.1403.0",
+        "chokidar": "^4.0.3",
         "deepmerge": "^4.3.1",
         "diff": "^7.0.0",
         "fastest-levenshtein": "^1.0.16",

server/aws-lsp-codewhisperer/package.json

@@ -3,6 +3,7 @@
  * Will be deleted or merged.
  */
 
+import * as path from 'path'
 import {
     ChatResponseStream,
     CodeWhispererStreaming,
@@ -36,6 +37,8 @@ import { DEFAULT_HELP_FOLLOW_UP_PROMPT, HELP_MESSAGE } from '../chat/constants'
 import { TelemetryService } from '../../shared/telemetry/telemetryService'
 import { AmazonQTokenServiceManager } from '../../shared/amazonQServiceManager/AmazonQTokenServiceManager'
 import { TabBarController } from './tabBarController'
+import { getUserPromptsDirectory } from './context/contextUtils'
+import { AdditionalContextProvider } from './context/addtionalContextProvider'
 
 describe('AgenticChatController', () => {
     const mockTabId = 'tab-1'
@@ -97,11 +100,13 @@ describe('AgenticChatController', () => {
 
     let sendMessageStub: sinon.SinonStub
     let generateAssistantResponseStub: sinon.SinonStub
+    let additionalContextProviderStub: sinon.SinonStub
     let disposeStub: sinon.SinonStub
     let activeTabSpy: {
         get: sinon.SinonSpy<[], string | undefined>
         set: sinon.SinonSpy<[string | undefined], void>
     }
+    let fsWriteFileStub: sinon.SinonStub
     let removeConversationSpy: sinon.SinonSpy
     let emitConversationMetricStub: sinon.SinonStub
 
@@ -144,13 +149,14 @@ describe('AgenticChatController', () => {
             })
 
         testFeatures = new TestFeatures()
+        fsWriteFileStub = sinon.stub()
 
         testFeatures.workspace.fs = {
             ...testFeatures.workspace.fs,
             getServerDataDirPath: sinon.stub().returns('/mock/server/data/path'),
             mkdir: sinon.stub().resolves(),
             readFile: sinon.stub().resolves(),
-            writeFile: sinon.stub().resolves(),
+            writeFile: fsWriteFileStub.resolves(),
             rm: sinon.stub().resolves(),
         }
 
@@ -161,6 +167,8 @@ describe('AgenticChatController', () => {
             addTool: sinon.stub().resolves(),
         }
 
+        additionalContextProviderStub = sinon.stub(AdditionalContextProvider.prototype, 'getAdditionalContext')
+        additionalContextProviderStub.resolves([])
         // @ts-ignore
         const cachedInitializeParams: InitializeParams = {
             initializationOptions: {
@@ -173,6 +181,7 @@ describe('AgenticChatController', () => {
                 },
             },
         }
+        testFeatures.lsp.window.showDocument = sinon.stub()
         testFeatures.lsp.getClientInitializeParams.returns(cachedInitializeParams)
         setCredentials('builderId')
 
@@ -1001,6 +1010,25 @@ describe('AgenticChatController', () => {
         })
     })
 
+    describe('onCreatePrompt', () => {
+        it('should create prompt file with given name', async () => {
+            const promptName = 'testPrompt'
+            const expectedPath = path.join(getUserPromptsDirectory(), 'testPrompt.prompt.md')
+
+            await chatController.onCreatePrompt({ promptName })
+
+            sinon.assert.calledOnceWithExactly(fsWriteFileStub, expectedPath, '', { mode: 0o600 })
+        })
+
+        it('should create default prompt file when no name provided', async () => {
+            const expectedPath = path.join(getUserPromptsDirectory(), 'default.prompt.md')
+
+            await chatController.onCreatePrompt({ promptName: '' })
+
+            sinon.assert.calledOnceWithExactly(fsWriteFileStub, expectedPath, '', { mode: 0o600 })
+        })
+    })
+
     describe('onInlineChatPrompt', () => {
         it('read all the response streams and return compiled results', async () => {
             const chatResultPromise = chatController.onInlineChatPrompt(