fix(codewhisperer): auto-connect on vscode startup #3359

Problem:
CodeWhisperer does not auto-connect on vscode startup. #3348 

Solution:
Call auth.restore()

Also fix some other issues:
1. Clear the inlineCompletionState on focus change so that
they should still be able to invoke CW when they invoked CW and then
switched to another app and switched back.
2. put activateCodeWhipserer() to be before activateAwsExplorer()
since the refreshStatusBar() and showReferenceLog() commands are used
within activateAwsExplorer() when the connection is changed. These
 commands are not registered until activateCodeWhisperer() has been called.
3. Remove unnecessary call to refreshCodeWhisperer(), this only needs to be
called when there's a connection change.

Author: andrewyuq

src/codewhisperer/activation.ts

@@ -202,10 +202,7 @@ export async function activate(context: ExtContext): Promise<void> {
         )
     )
 
-    if (!isCloud9() && !auth.isConnectionValid()) {
-        // this is to proactively check and reflect the state if user's connection is expired
-        auth.refreshCodeWhisperer()
-    }
+    await auth.restore()
 
     function activateSecurityScan() {
         context.extensionContext.subscriptions.push(

src/codewhisperer/explorer/codewhispererNode.ts

@@ -34,10 +34,6 @@ export class CodeWhispererNode implements RootNode {
     constructor() {}
 
     public getTreeItem() {
-        if (!isCloud9()) {
-            AuthUtil.instance.restore()
-        }
-
         const item = new vscode.TreeItem('CodeWhisperer')
         item.description = this.getDescription()
         item.collapsibleState = vscode.TreeItemCollapsibleState.Collapsed

src/codewhisperer/service/inlineCompletionService.ts

@@ -320,6 +320,7 @@ export class InlineCompletionService {
         vsCodeState.isCodeWhispererEditing = false
         ReferenceInlineProvider.instance.removeInlineReference()
         ImportAdderProvider.instance.clear()
+        await InlineCompletionService.instance.clearInlineCompletionStates(vscode.window.activeTextEditor)
     }
 
     async onCursorChange(e: vscode.TextEditorSelectionChangeEvent) {

src/codewhisperer/util/authUtil.ts

@@ -230,7 +230,6 @@ export class AuthUtil {
     }
 
     public async notifyReauthenticate(isAutoTrigger?: boolean) {
-        await this.refreshCodeWhisperer()
         this.showReauthenticatePrompt(isAutoTrigger)
     }
 

src/extension.ts

@@ -195,6 +195,8 @@ export async function activate(context: vscode.ExtensionContext) {
 
         await activateCloudFormationTemplateRegistry(context)
 
+        await activateCodeWhisperer(extContext)
+
         await activateAwsExplorer({
             context: extContext,
             regionProvider,
@@ -217,8 +219,6 @@ export async function activate(context: vscode.ExtensionContext) {
 
         await activateS3(extContext)
 
-        await activateCodeWhisperer(extContext)
-
         await activateEcr(context)
 
         await activateCloudWatchLogs(context, settings)