Adding Thumbs UP and Thumbs Down feedback functionality for /review and /test

Author: laileni-aws

packages/amazonq/src/app/amazonqScan/app.ts

@@ -35,6 +35,7 @@ export function init(appContext: AmazonQAppInitContext) {
         processResponseBodyLinkClick: new vscode.EventEmitter<any>(),
         fileClicked: new vscode.EventEmitter<any>(),
         scanCancelled: new vscode.EventEmitter<any>(),
+        processChatItemVotedMessage: new vscode.EventEmitter<any>(),
     }
     const dispatcher = new AppToWebViewMessageDispatcher(appContext.getAppsToWebViewMessagePublisher())
     const messenger = new Messenger(dispatcher)

packages/amazonq/src/app/amazonqScan/chat/controller/controller.ts

@@ -98,6 +98,10 @@ export class ScanController {
         this.chatControllerMessageListeners.scanCancelled.event((data) => {
             return this.handleScanCancelled(data)
         })
+
+        this.chatControllerMessageListeners.processChatItemVotedMessage.event((data) => {
+            // TODO: Emit the telemetry event
+        })
     }
 
     private async tabOpened(message: any) {

packages/amazonq/src/app/amazonqScan/chat/views/actions/uiMessageListener.ts

@@ -55,6 +55,9 @@ export class UIMessageListener {
             case 'file-click':
                 this.processFileClick(msg)
                 break
+            case 'chat-item-voted':
+                this.chatItemVoted(msg)
+                break
         }
     }
 
@@ -112,4 +115,13 @@ export class UIMessageListener {
             filePath: msg.filePath,
         })
     }
+
+    private chatItemVoted(msg: any) {
+        this.scanControllerEventsEmitters?.processChatItemVotedMessage.fire({
+            tabID: msg.tabID,
+            command: msg.command,
+            vote: msg.vote,
+            messageId: msg.messageId,
+        })
+    }
 }

packages/core/src/amazonqScan/controller.ts

@@ -30,4 +30,5 @@ export interface ScanChatControllerEventEmitters {
     readonly processResponseBodyLinkClick: vscode.EventEmitter<any>
     readonly fileClicked: vscode.EventEmitter<any>
     readonly scanCancelled: vscode.EventEmitter<any>
+    readonly processChatItemVotedMessage: vscode.EventEmitter<any>
 }

packages/core/src/amazonqTest/app.ts

@@ -32,6 +32,7 @@ export function init(appContext: AmazonQAppInitContext) {
         errorThrown: new vscode.EventEmitter<any>(),
         insertCodeAtCursorPosition: new vscode.EventEmitter<any>(),
         processResponseBodyLinkClick: new vscode.EventEmitter<any>(),
+        processChatItemVotedMessage: new vscode.EventEmitter<any>(),
     }
     const dispatcher = new AppToWebViewMessageDispatcher(appContext.getAppsToWebViewMessagePublisher())
     const messenger = new Messenger(dispatcher)

packages/core/src/amazonqTest/chat/controller/controller.ts

@@ -58,7 +58,7 @@ import {
 import { UserIntent } from '@amzn/codewhisperer-streaming'
 import { getSelectedCustomization } from '../../../codewhisperer/util/customizationUtil'
 import { createCodeWhispererChatStreamingClient } from '../../../shared/clients/codewhispererChatClient'
-import { ChatTriggerType } from '../../../codewhispererChat/controllers/chat/model'
+import { ChatItemVotedMessage, ChatTriggerType } from '../../../codewhispererChat/controllers/chat/model'
 import { triggerPayloadToChatRequest } from '../../../codewhispererChat/controllers/chat/chatRequest/converter'
 import { EditorContentController } from '../../../amazonq/commons/controllers/contentController'
 import { amazonQTabSuffix } from '../../../shared/constants'
@@ -82,6 +82,7 @@ export interface TestChatControllerEventEmitters {
     readonly errorThrown: vscode.EventEmitter<any>
     readonly insertCodeAtCursorPosition: vscode.EventEmitter<any>
     readonly processResponseBodyLinkClick: vscode.EventEmitter<any>
+    readonly processChatItemVotedMessage: vscode.EventEmitter<any>
 }
 
 type OpenDiffMessage = {
@@ -160,6 +161,12 @@ export class TestController {
             return this.processLink(data)
         })
 
+        this.chatControllerMessageListeners.processChatItemVotedMessage.event((data) => {
+            this.processChatItemVotedMessage(data).catch((e) => {
+                getLogger().error('processChatItemVotedMessage failed: %s', (e as Error).message)
+            })
+        })
+
         this.chatControllerMessageListeners.followUpClicked.event((data) => {
             switch (data.followUp.type) {
                 case FollowUpTypes.ViewDiff:
@@ -209,6 +216,10 @@ export class TestController {
         }
     }
 
+    private async processChatItemVotedMessage(message: ChatItemVotedMessage) {
+        // TODO: Emit the telemetry event
+    }
+
     private async tabClosed(data: any) {
         getLogger().debug('Tab closed with data tab id: %s', data.tabID)
         await this.sessionCleanUp()

packages/core/src/amazonqTest/chat/views/actions/uiMessageListener.ts

@@ -62,6 +62,9 @@ export class UIMessageListener {
             case 'response-body-link-click':
                 this.processResponseBodyLinkClick(msg)
                 break
+            case 'chat-item-voted':
+                this.chatItemVoted(msg)
+                break
         }
     }
 
@@ -140,4 +143,13 @@ export class UIMessageListener {
             link: msg.link,
         })
     }
+
+    private chatItemVoted(msg: any) {
+        this.testControllerEventsEmitters?.processChatItemVotedMessage.fire({
+            tabID: msg.tabID,
+            command: msg.command,
+            vote: msg.vote,
+            messageId: msg.messageId,
+        })
+    }
 }

packages/core/src/codewhisperer/commands/startTestGeneration.ts

@@ -46,7 +46,7 @@ export async function startTestGenerationProcess(
             return
         }
         /**
-         * Zip the project
+         * Step 1: Zip the project
          */
 
         const zipUtil = new ZipUtil()