Display text after /scan in Q chat panel

Author: laileni-aws

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

@@ -213,8 +213,8 @@ export class Connector {
     onRunSecurityScans = (tabID: string): void => {
         this.sendMessageToExtension({
             tabID: tabID,
-            command: 'runProjectScan',
-            chatMessage: 'Security scan started and you can see results in terminal',
+            command: 'scan',
+            chatMessage: '',
             tabType: 'cwc',
         })
     }

packages/core/src/amazonq/webview/ui/commands.ts

@@ -31,6 +31,6 @@ type MessageCommand =
     | 'file-click'
     | 'form-action-click'
     | 'open-settings'
-    | 'runProjectScan'
+    | 'scan'
 
 export type ExtensionMessage = Record<string, any> & { command: MessageCommand }

packages/core/src/amazonq/webview/ui/tabs/constants.ts

@@ -21,7 +21,6 @@ const commonTabData: TabTypeData = {
 export const TabTypeDataMap: Record<TabType, TabTypeData> = {
     unknown: commonTabData,
     cwc: commonTabData,
-    // scan:commonTabData,
     featuredev: {
         title: 'Q - Dev',
         placeholder: 'Describe your task or issue in as much detail as possible',
@@ -38,9 +37,4 @@ To get started, describe the task you are trying to accomplish.`,
 
 I can help you upgrade your Java 8 and 11 codebases to Java 17.`,
     },
-    // scan: {
-    //     title: 'Q Security Scans',
-    //     placeholder: 'Open a new Tab to run security scans',
-    //     welcome: `Welcome to Security scans!`,
-    // }
 }

packages/core/src/codewhispererChat/controllers/chat/controller.ts

@@ -204,7 +204,6 @@ export class ChatController {
             .extractContextForTrigger('QuickAction')
             .then((context) => {
                 const triggerID = randomUUID()
-
                 this.messenger.sendQuickActionMessage(quickActionCommand, triggerID)
 
                 this.triggerEventsStorage.addTriggerEvent({
@@ -215,7 +214,11 @@ export class ChatController {
                     quickAction: quickActionCommand,
                     context,
                 })
-
+                if (quickActionCommand === 'scan') {
+                    void this.generateStaticTextResponse('quick-action-scan', triggerID)
+                    //TODO: Do telemetry
+                    return
+                }
                 if (quickActionCommand === 'help') {
                     void this.generateStaticTextResponse('quick-action-help', triggerID)
                     recordTelemetryChatRunCommand('help')

packages/core/src/codewhispererChat/controllers/chat/messenger/messenger.ts

@@ -34,7 +34,12 @@ import { marked } from 'marked'
 import { JSDOM } from 'jsdom'
 import { LspController } from '../../../../amazonq/lsp/lspController'
 
-export type StaticTextResponseType = 'quick-action-help' | 'onboarding-help' | 'transform' | 'help'
+export type StaticTextResponseType =
+    | 'quick-action-help'
+    | 'onboarding-help'
+    | 'transform'
+    | 'help'
+    | 'quick-action-scan'
 
 export class Messenger {
     public constructor(
@@ -376,6 +381,9 @@ export class Messenger {
                 \n\n- Do not enter any confidential, sensitive, or personal information.
                 \n\n*For additional help, visit the [Amazon Q User Guide](${userGuideURL}).*`
                 break
+            case 'quick-action-scan':
+                message = `This is security scans message shown in the Q chat panel` //TODO: modify the code acc to UX
+                break
             case 'onboarding-help':
                 message = `### What I can do:
                 \n\n- Answer questions about AWS
@@ -427,6 +435,11 @@ export class Messenger {
             case 'help':
                 message = 'What can Amazon Q help me with?'
                 break
+            /* Optional
+            case 'scan':
+                message = 'What can Amazon QSA help me with?'
+                break
+            */
         }
 
         this.dispatcher.sendQuickActionMessage(

packages/core/src/codewhispererChat/controllers/chat/model.ts

@@ -62,6 +62,7 @@ export interface CopyCodeToClipboard {
 export type ChatPromptCommandType =
     | 'help'
     | 'clear'
+    | 'scan'
     | 'follow-up-was-clicked'
     | 'onboarding-page-cwc-button-clicked'
     | 'chat-prompt'

packages/core/src/codewhispererChat/view/messages/messageListener.ts

@@ -98,10 +98,9 @@ export class UIMessageListener {
             case 'footer-info-link-click':
                 this.processFooterInfoLinkClick(msg)
                 break
-            case 'runProjectScan':
+            case 'scan':
                 showSecurityScan.execute(placeholder, cwQuickPickSource)
-                // void vscode.window.setStatusBarMessage('Got Message from Chat')
-                // this.processChatMessage(msg)
+                this.processChatMessage(msg)
                 break
             case 'open-settings':
                 this.processOpenSettings(msg)