fix(codewhisperer): "Don't show again" access token popup #3022

* Do Not Show Again btn and show once a week

Co-authored-by: Rachel Cross <[email protected]>

Author: leigaol

src/codewhisperer/activation.ts

@@ -247,16 +247,44 @@ export async function activate(context: ExtContext): Promise<void> {
 
             await vscode.commands.executeCommand('aws.codeWhisperer.refreshRootNode')
             const t = new Date()
+            const doNotShowAgain =
+                context.extensionContext.globalState.get<boolean>(
+                    CodeWhispererConstants.accessTokenMigrationDoNotShowAgainKey
+                ) || false
+            const notificationLastShown =
+                context.extensionContext.globalState.get<Date>(
+                    CodeWhispererConstants.accessTokenMigrationDoNotShowAgainLastShown
+                ) || t
+
+            //Add 7 days to notificationLastShown to determine whether warn message should show
+            notificationLastShown.setDate(notificationLastShown.getDate() + 7)
+            if (doNotShowAgain || notificationLastShown <= t) {
+                return
+            }
             if (t <= CodeWhispererConstants.accessTokenCutOffDate) {
                 vscode.window
                     .showWarningMessage(
                         CodeWhispererConstants.accessTokenMigrationWarningMessage,
-                        CodeWhispererConstants.accessTokenMigrationWarningButtonMessage
+                        CodeWhispererConstants.accessTokenMigrationWarningButtonMessage,
+                        CodeWhispererConstants.accessTokenMigrationDoNotShowAgain
                     )
                     .then(async resp => {
                         if (resp === CodeWhispererConstants.accessTokenMigrationWarningButtonMessage) {
                             await vscode.commands.executeCommand('aws.codeWhisperer.refresh')
                             await showSsoSignIn.execute()
+                        } else if (resp === CodeWhispererConstants.accessTokenMigrationDoNotShowAgain) {
+                            await vscode.window.showInformationMessage(
+                                CodeWhispererConstants.accessTokenMigrationDoNotShowAgainInfo,
+                                'OK'
+                            )
+                            await context.extensionContext.globalState.update(
+                                CodeWhispererConstants.accessTokenMigrationDoNotShowAgainKey,
+                                true
+                            )
+                            await context.extensionContext.globalState.update(
+                                CodeWhispererConstants.accessTokenMigrationDoNotShowAgainLastShown,
+                                t
+                            )
                         }
                     })
             } else {
@@ -271,6 +299,11 @@ export async function activate(context: ExtContext): Promise<void> {
                         if (resp === CodeWhispererConstants.accessTokenMigrationErrorButtonMessage) {
                             await vscode.commands.executeCommand('aws.codeWhisperer.refresh')
                             await showSsoSignIn.execute()
+                        } else if (resp === CodeWhispererConstants.accessTokenMigrationDoNotShowAgain) {
+                            await context.extensionContext.globalState.update(
+                                CodeWhispererConstants.accessTokenMigrationDoNotShowAgainKey,
+                                true
+                            )
                         }
                     })
             }

src/codewhisperer/models/constants.ts

@@ -235,3 +235,12 @@ export const failedToConnectAwsBuilderId = `Failed to connect to AWS Builder ID`
 export const failedToConnectIamIdentityCenter = `Failed to connect to IAM Identity Center`
 
 export const connectionExpired = `AWS Toolkit: Connection expired. Reauthenticate to continue.`
+
+export const accessTokenMigrationDoNotShowAgain = `Don\'t Show Again`
+
+export const accessTokenMigrationDoNotShowAgainKey = 'CODEWHISPERER_ACCESS_TOKEN_MIGRATION_DO_NOT_SHOW_AGAIN'
+
+export const accessTokenMigrationDoNotShowAgainLastShown =
+    'CODEWHISPERER_ACCESS_TOKEN_MIGRATION_DO_NOT_SHOW_AGAIN_LAST_SHOWN'
+
+export const accessTokenMigrationDoNotShowAgainInfo = `You will not receive this notification again. If you would like to continue using CodeWhisperer after January 31, 2023, you can still connect with AWS. [Learn More](https://docs.aws.amazon.com/toolkit-for-vscode/latest/userguide/codewhisper-setup-general.html).`