feat(q-proactive-scan): remove file scans for Builder Id, add timeout limits aws#4712

Problem
Change Timeout limits for project scan and File scans. No Auto Scan status bar options for Free Users(BuilderId) and no file scans enabled for Builder Id users.

Solution
Added Timeout Limits for Project and File Scans.
Disabled File scan for BuilderID users.
Removed status bar Code scan toggle option for Builder Id Users.

Author: ashishrp-aws

packages/core/src/codewhisperer/activation.ts

@@ -282,12 +282,16 @@ export async function activate(context: ExtContext): Promise<void> {
     if (auth.isValidEnterpriseSsoInUse()) {
         await notifyNewCustomizations()
     }
+    if (auth.isBuilderIdInUse()) {
+        await CodeScansState.instance.setScansEnabled(false)
+    }
 
     function setSubscriptionsForAutoScans() {
         // Initial scan when the editor opens for the first time
         const editor = vscode.window.activeTextEditor
         if (
             CodeScansState.instance.isScansEnabled() &&
+            !auth.isBuilderIdInUse() &&
             editor &&
             securityScanLanguageContext.isLanguageSupported(editor.document.languageId) &&
             editor.document.getText().length > 0
@@ -306,6 +310,7 @@ export async function activate(context: ExtContext): Promise<void> {
             vscode.window.onDidChangeActiveTextEditor(editor => {
                 if (
                     CodeScansState.instance.isScansEnabled() &&
+                    !auth.isBuilderIdInUse() &&
                     editor &&
                     securityScanLanguageContext.isLanguageSupported(editor.document.languageId)
                 ) {
@@ -334,6 +339,7 @@ export async function activate(context: ExtContext): Promise<void> {
                 const editor = vscode.window.activeTextEditor
                 if (
                     CodeScansState.instance.isScansEnabled() &&
+                    !auth.isBuilderIdInUse() &&
                     editor &&
                     event.document === editor.document &&
                     securityScanLanguageContext.isLanguageSupported(editor.document.languageId) &&
@@ -355,6 +361,7 @@ export async function activate(context: ExtContext): Promise<void> {
             const editor = vscode.window.activeTextEditor
             if (
                 isScansEnabled &&
+                !auth.isBuilderIdInUse() &&
                 editor &&
                 securityScanLanguageContext.isLanguageSupported(editor.document.languageId) &&
                 editor.document.getText().length > 0

packages/core/src/codewhisperer/models/constants.ts

@@ -233,7 +233,9 @@ export const codeScanZipExt = '.zip'
 
 export const contextTruncationTimeoutSeconds = 10
 
-export const codeScanJobTimeoutSeconds = 50
+export const codeScanJobTimeoutSeconds = 60 * 10 //10 minutes
+
+export const codeFileScanJobTimeoutSeconds = 60 //1 minute
 
 export const projectSizeCalculateTimeoutSeconds = 10
 

packages/core/src/codewhisperer/service/securityScanHandler.ts

@@ -123,7 +123,11 @@ export async function pollScanJobStatus(
         throwIfCancelled(scope)
         await sleep(CodeWhispererConstants.codeScanJobPollingIntervalSeconds * 1000)
         timer += CodeWhispererConstants.codeScanJobPollingIntervalSeconds
-        if (timer > CodeWhispererConstants.codeScanJobTimeoutSeconds) {
+        const timeoutSeconds =
+            scope === CodeWhispererConstants.CodeAnalysisScope.FILE
+                ? CodeWhispererConstants.codeFileScanJobTimeoutSeconds
+                : CodeWhispererConstants.codeScanJobTimeoutSeconds
+        if (timer > timeoutSeconds) {
             getLogger().verbose(`Scan job status: ${status}`)
             getLogger().verbose(`Scan job timeout.`)
             throw new Error('Scan job timeout.')

packages/core/src/codewhisperer/ui/statusBarMenu.ts

@@ -51,7 +51,6 @@ function getAmazonQCodeWhispererNodes() {
             createFreeTierLimitMet('item'),
             createOpenReferenceLog(),
             createSeparator('Other Features'),
-            createAutoScans(autoScansEnabled),
             createSecurityScan(),
         ]
     }
@@ -66,6 +65,26 @@ function getAmazonQCodeWhispererNodes() {
         amazonq = require('../../amazonq/explorer/amazonQChildrenNodes')
     }
 
+    if (AuthUtil.instance.isBuilderIdInUse()) {
+        return [
+            // CodeWhisperer
+            createSeparator('Inline Suggestions'),
+            createAutoSuggestions(autoTriggerEnabled),
+            ...(AuthUtil.instance.isValidEnterpriseSsoInUse() && AuthUtil.instance.isCustomizationFeatureEnabled
+                ? [createSelectCustomization()]
+                : []),
+            createOpenReferenceLog(),
+            createGettingStarted(), // "Learn" node : opens Learn CodeWhisperer page
+
+            // Security scans
+            createSeparator('Security Scans'),
+            createSecurityScan(),
+
+            // Amazon Q + others
+            createSeparator('Other Features'),
+            ...(amazonq ? [amazonq.switchToAmazonQNode('item')] : []),
+        ]
+    }
     return [
         // CodeWhisperer
         createSeparator('Inline Suggestions'),