Adding /scans functionality to Q chat which runs Project scans

Author: laileni-aws

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

@@ -210,6 +210,15 @@ export class Connector {
         })
     }
 
+    onRunSecurityScans = (tabID: string): void => {
+        this.sendMessageToExtension({
+            tabID: tabID,
+            command: 'runProjectScan',
+            chatMessage: 'Security scan started and you can see results in terminal',
+            tabType: 'cwc',
+        })
+    }
+
     private sendTriggerMessageProcessed = async (requestID: any): Promise<void> => {
         this.sendMessageToExtension({
             command: 'trigger-message-processed',

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

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

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

@@ -151,6 +151,15 @@ export class Connector {
         this.gumbyChatConnector.transform(tabID)
     }
 
+    scan = (tabID: string): void => {
+        switch (this.tabsStorage.getTab(tabID)?.type) {
+            case 'cwc':
+                this.cwChatConnector.onRunSecurityScans(tabID)
+                break
+        }
+        // this.cwChatConnector.onRunSecurityScans(tabID)
+    }
+
     handleMessageReceive = async (message: MessageEvent): Promise<void> => {
         if (message.data === undefined) {
             return

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

@@ -42,6 +42,10 @@ export class QuickActionGenerator {
                               },
                           ]
                         : []),
+                    {
+                        command: '/scan',
+                        description: 'Run a project scan',
+                    },
                 ],
             },
             {
@@ -71,11 +75,11 @@ export class QuickActionGenerator {
             },
             featuredev: {
                 description: "This command isn't available in /dev",
-                unavailableItems: ['/dev', '/transform', '/help', '/clear'],
+                unavailableItems: ['/dev', '/transform', '/help', '/clear', '/scan'],
             },
             gumby: {
                 description: "This command isn't available in /transform",
-                unavailableItems: ['/dev', '/transform'],
+                unavailableItems: ['/dev', '/transform', '/scan'],
             },
             unknown: {
                 description: '',

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

@@ -52,6 +52,9 @@ export class QuickActionHandler {
             case '/clear':
                 this.handleClearCommand(tabID)
                 break
+            case '/scan':
+                this.handleSecurityScansCommand(tabID)
+                break
         }
     }
 
@@ -169,4 +172,61 @@ export class QuickActionHandler {
             }
         }
     }
+
+    private handleSecurityScansCommand(tabID: string) {
+        if (this.tabsStorage.getTab(tabID)?.type === 'unknown') {
+            this.tabsStorage.updateTabTypeFromUnknown(tabID, 'cwc')
+        }
+        // this.connector.scan(tabID)
+        this.connector.scan(tabID)
+
+        /*
+        let scanTabId: string | undefined = undefined
+
+        this.tabsStorage.getTabs().forEach((tab) => {
+            if (tab.type === 'scan') {
+                scanTabId = tab.id
+            }
+        })
+
+        if (scanTabId !== undefined) {
+            this.mynahUI.selectTab(scanTabId, eventId || '')
+            this.connector.onTabChange(scanTabId)
+            return
+        }
+
+        let affectedTabId: string | undefined = tabID
+        // if there is no scan tab, open a new one
+        if (this.tabsStorage.getTab(affectedTabId)?.type !== 'unknown') {
+            affectedTabId = this.mynahUI.updateStore('', {
+                loadingChat: true,
+            })
+        }
+
+        if (affectedTabId === undefined) {
+            this.mynahUI.notify({
+                content: uiComponentsTexts.noMoreTabsTooltip,
+                type: NotificationType.WARNING,
+            })
+            return
+        } else {
+            this.tabsStorage.updateTabTypeFromUnknown(affectedTabId, 'scan')
+            this.connector.onKnownTabOpen(affectedTabId)
+            this.connector.onUpdateTabType(affectedTabId)
+
+            // reset chat history
+            this.mynahUI.updateStore(affectedTabId, {
+                chatItems: [],
+            })
+
+            this.mynahUI.updateStore(affectedTabId, this.tabDataGenerator.getTabData('scan', true, undefined))
+
+            // disable chat prompt
+            this.mynahUI.updateStore(affectedTabId, {
+                loadingChat: true,
+            })
+
+            this.connector.scan(affectedTabId)
+        }*/
+    }
 }

packages/core/src/amazonq/webview/ui/storages/tabsStorage.ts

@@ -4,6 +4,7 @@
  */
 
 export type TabStatus = 'free' | 'busy' | 'dead'
+// export type TabType = 'cwc' | 'featuredev' | 'gumby' | 'unknown' | 'scan'
 export type TabType = 'cwc' | 'featuredev' | 'gumby' | 'unknown'
 export type TabOpenType = 'click' | 'contextMenu' | 'hotkeys'
 

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

@@ -21,6 +21,7 @@ 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',
@@ -37,4 +38,9 @@ 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/view/messages/messageListener.ts

@@ -9,6 +9,9 @@ import { ChatControllerMessagePublishers } from '../../controllers/chat/controll
 import { ReferenceLogController } from './referenceLogController'
 import { getLogger } from '../../../shared/logger'
 import { openSettingsId } from '../../../shared/settings'
+import { showSecurityScan } from '../../../codewhisperer/commands/basicCommands'
+import { placeholder } from '../../../shared/vscode/commands2'
+import { cwQuickPickSource } from '../../../codewhisperer/commands/types'
 
 export interface UIMessageListenerProps {
     readonly chatControllerMessagePublishers: ChatControllerMessagePublishers
@@ -95,8 +98,15 @@ export class UIMessageListener {
             case 'footer-info-link-click':
                 this.processFooterInfoLinkClick(msg)
                 break
+            case 'runProjectScan':
+                showSecurityScan.execute(placeholder, cwQuickPickSource)
+                // void vscode.window.setStatusBarMessage('Got Message from Chat')
+                // this.processChatMessage(msg)
+                break
             case 'open-settings':
                 this.processOpenSettings(msg)
+            // this.processOpenSettings(msg)
+            // void vscode.window.setStatusBarMessage('Got Message from Chat')
         }
     }