Adding feedback mechanism for internal amazon users as part of V2 version

Author: laileni-aws

packages/core/src/amazonq/webview/ui/apps/testChatConnector.ts

@@ -162,6 +162,7 @@ export class Connector extends BaseConnector {
                 body: messageData.message,
                 canBeVoted: false,
                 informationCard: messageData.informationCard,
+                buttons: messageData.buttons ?? [],
             }
             this.onChatAnswerReceived(messageData.tabID, answer, messageData)
         }

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

@@ -26,7 +26,7 @@ import {
 import MessengerUtils, { ButtonActions } from './messenger/messengerUtils'
 import { getTelemetryReasonDesc, isAwsError } from '../../../shared/errors'
 import { ChatItemType } from '../../../amazonq/commons/model'
-import { ProgressField } from '@aws/mynah-ui'
+import { ChatItemButton, MynahIcons, ProgressField } from '@aws/mynah-ui'
 import { FollowUpTypes } from '../../../amazonq/commons/types'
 import {
     cancelBuild,
@@ -63,6 +63,9 @@ import {
 } from '../../../codewhisperer/models/constants'
 import { UserWrittenCodeTracker } from '../../../codewhisperer/tracker/userWrittenCodeTracker'
 import { ReferenceLogViewProvider } from '../../../codewhisperer/service/referenceLogViewProvider'
+import { submitFeedback } from '../../../feedback/vue/submitFeedback'
+import { placeholder } from '../../../shared/vscode/commands2'
+import { Auth } from '../../../auth/auth'
 
 export interface TestChatControllerEventEmitters {
     readonly tabOpened: vscode.EventEmitter<any>
@@ -354,6 +357,7 @@ export class TestController {
     // This function handles actions if user clicked on any Button one of these cases will be executed
     private async handleFormActionClicked(data: any) {
         const typedAction = MessengerUtils.stringToEnumValue(ButtonActions, data.action as any)
+        let getFeedbackCommentData = ''
         switch (typedAction) {
             case ButtonActions.STOP_TEST_GEN:
                 testGenState.setToCancelling()
@@ -366,6 +370,16 @@ export class TestController {
                 this.messenger.sendChatInputEnabled(data.tabID, true)
                 await this.sessionCleanUp()
                 break
+            case ButtonActions.PROVIDE_FEEDBACK:
+                getFeedbackCommentData = `Q Test Generation: RequestId: ${this.sessionStorage.getSession().startTestGenerationRequestId}`
+                void submitFeedback(placeholder, 'Amazon Q', getFeedbackCommentData)
+                telemetry.ui_click.emit({ elementId: 'unitTestGeneration_provideFeedback' })
+                this.messenger.sendMessage(
+                    'Unit test generation completed. Thanks for providing feedback.',
+                    data.tabID,
+                    'answer'
+                )
+                break
         }
     }
     // This function handles actions if user gives any input from the chatInput box
@@ -897,7 +911,25 @@ export class TestController {
 
     // TODO: Check if there are more cases to endSession if yes create a enum or type for step
     private async endSession(data: any, step: FollowUpTypes) {
-        this.messenger.sendMessage('Unit test generation completed.', data.tabID, 'answer')
+        const buttons: ChatItemButton[] = []
+        if (Auth.instance.isInternalAmazonUser()) {
+            buttons.push({
+                keepCardAfterClick: false,
+                text: 'How can we make /test better?',
+                id: ButtonActions.PROVIDE_FEEDBACK,
+                disabled: false, // allow button to be re-clicked
+                position: 'outside',
+                icon: 'comment' as MynahIcons,
+            })
+        }
+
+        this.messenger.sendMessage(
+            'Unit test generation completed.',
+            data.tabID,
+            'answer',
+            'testGenEndSessionMessage',
+            buttons
+        )
 
         const session = this.sessionStorage.getSession()
         if (step === FollowUpTypes.RejectCode) {

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

@@ -24,7 +24,7 @@ import {
     UpdatePromptProgressMessage,
 } from '../../views/connector/connector'
 import { ChatItemType } from '../../../../amazonq/commons/model'
-import { ChatItemAction, ProgressField } from '@aws/mynah-ui'
+import { ChatItemAction, ChatItemButton, ProgressField } from '@aws/mynah-ui'
 import * as CodeWhispererConstants from '../../../../codewhisperer/models/constants'
 import { TriggerPayload } from '../../../../codewhispererChat/controllers/chat/model'
 import {
@@ -76,8 +76,16 @@ export class Messenger {
         this.dispatcher.sendChatMessage(new CapabilityCardMessage(params.tabID))
     }
 
-    public sendMessage(message: string, tabID: string, messageType: ChatItemType) {
-        this.dispatcher.sendChatMessage(new ChatMessage({ message, messageType }, tabID))
+    public sendMessage(
+        message: string,
+        tabID: string,
+        messageType: ChatItemType,
+        messageId?: string,
+        buttons?: ChatItemButton[]
+    ) {
+        this.dispatcher.sendChatMessage(
+            new ChatMessage({ message, messageType, messageId: messageId, buttons: buttons }, tabID)
+        )
     }
 
     public sendShortSummary(params: {
@@ -159,16 +167,7 @@ export class Messenger {
                 message = CodeWhispererConstants.invalidFileTypeChatMessage
                 break
         }
-
-        this.dispatcher.sendChatMessage(
-            new ChatMessage(
-                {
-                    message,
-                    messageType: 'answer-stream',
-                },
-                tabID
-            )
-        )
+        this.sendMessage(message, tabID, 'answer-stream')
     }
 
     public sendErrorMessage(errorMessage: string, tabID: string) {

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

@@ -12,6 +12,7 @@ export enum ButtonActions {
     VIEW_DIFF = 'View-Diff',
     STOP_TEST_GEN = 'Stop-Test-Generation',
     STOP_BUILD = 'Stop-Build-Process',
+    PROVIDE_FEEDBACK = 'Provide-Feedback',
 }
 
 // TODO: Refactor the common functionality between Transform, FeatureDev, CWSPRChat, Scan and UTG to a new Folder.

packages/core/src/amazonqTest/chat/views/connector/connector.ts

@@ -80,6 +80,7 @@ export class ChatMessage extends UiMessage {
     readonly messageId?: string | undefined
     readonly messageType: ChatItemType
     readonly canBeVoted?: boolean
+    readonly buttons?: ChatItemButton[]
     readonly informationCard: ChatItemContent['informationCard']
     override type: TestMessageType = 'chatMessage'
 
@@ -90,6 +91,7 @@ export class ChatMessage extends UiMessage {
         this.messageId = props.messageId || undefined
         this.canBeVoted = props.canBeVoted || undefined
         this.informationCard = props.informationCard || undefined
+        this.buttons = props.buttons || undefined
     }
 }