perf: store selected tenant

package.json

@@ -299,6 +299,20 @@
             "items": {
               "type": "string"
             }
+          },
+          "azure-api-center.tenant": {
+            "type": "object",
+            "description": "A specific tenant to sign in to",
+            "properties": {
+              "name": {
+                "type": "string",
+                "description": "tenant name"
+              },
+              "id": {
+                "type": "string",
+                "description": "tenant id"
+              }
+            }
           }
         }
       }

src/azure/azureLogin/azureAccount.ts

@@ -2,7 +2,8 @@
 // Licensed under the MIT license.
 import { SubscriptionClient, TenantIdDescription } from "@azure/arm-resources-subscriptions";
 import { TokenCredential } from "@azure/core-auth";
-import { AuthenticationSession, QuickPickItem, Uri, env, window } from "vscode";
+import { AuthenticationSession, ConfigurationTarget, QuickPickItem, Uri, env, window, workspace } from "vscode";
+import { extensionName, tenantSetting } from "../../constants";
 import { UiStrings } from "../../uiStrings";
 import { GeneralUtils } from "../../utils/generalUtils";
 import { SelectionType, SignInStatus, SubscriptionFilter, Tenant } from "./authTypes";
@@ -14,6 +15,14 @@
         await AzureSessionProviderHelper.getSessionProvider().signIn();
     }
 
+    export function getSelectedTenant(): Tenant | undefined {
+        return workspace.getConfiguration(extensionName).get<Tenant>(tenantSetting);
+    }
+
+    export async function updateSelectedTenant(value?: Tenant): Promise<void> {
+        await workspace.getConfiguration(extensionName).update(tenantSetting, value, ConfigurationTarget.Global, true);
+    }
+
     export async function selectTenant(): Promise<void> {
         const sessionProvider = AzureSessionProviderHelper.getSessionProvider();
         if (sessionProvider.signInStatus !== SignInStatus.SignedIn) {
@@ -42,6 +51,7 @@
         }
 
         sessionProvider.selectedTenant = selectedTenant;
+        await updateSelectedTenant(selectedTenant);
     }
 
     type SubscriptionQuickPickItem = QuickPickItem & { subscription: SubscriptionFilter };

src/azure/azureLogin/azureSessionProvider.ts

@@ -143,6 +143,9 @@
             this.tenants = newTenants;
             this.signInStatusValue = newSignInStatus;
             if (signInStatusChanged || tenantsChanged || selectedTenantChanged) {
+                if (newSignInStatus === SignInStatus.SignedOut) {
+                    await AzureAccount.updateSelectedTenant();
+                }
                 this.onSignInStatusChangeEmitter.fire(this.signInStatusValue);
             }
         }
@@ -220,7 +223,11 @@
             );
             const results = await Promise.all(getSessionPromises);
             const accessibleTenants = results.filter(GeneralUtils.succeeded).map((r) => r.result);
-            return accessibleTenants.length === 1 ? AzureAccount.findTenant(tenants, accessibleTenants[0].tenantId) : null;
+            if (accessibleTenants.length === 1) {
+                return AzureAccount.findTenant(tenants, accessibleTenants[0].tenantId);
+            }
+            const lastTenant = AzureAccount.getSelectedTenant();
+            return lastTenant && accessibleTenants.some(item => item.tenantId === lastTenant.id) ? lastTenant : null;
         }
 
         private async getArmSession(

src/constants.ts

@@ -16,6 +16,7 @@ export const azureApiGuidelineRulesetFile = "https://raw.githubusercontent.com/a
 export const spectralOwaspRulesetFile = "https://unpkg.com/@stoplight/spectral-owasp-ruleset/dist/ruleset.mjs";
 export const MODEL_SELECTOR: vscode.LanguageModelChatSelector = { vendor: 'copilot', family: 'gpt-4' };
 export const ExceedTokenLimit = "Message exceeds token limit";
+export const tenantSetting: string = 'tenant';
 
 export const AzureAccountType = {
     createAzureAccount: "azureapicenterCreateAzureAccount",

src/test/unit/azure/azureLogin/azureAccount.test.ts

@@ -0,0 +1,26 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+import * as assert from "assert";
+import * as sinon from "sinon";
+import * as vscode from "vscode";
+import { AzureAccount } from "../../../../azure/azureLogin/azureAccount";
+
+describe("Azure Account test case", () => {
+    let sandbox = null as any;
+    before(() => {
+        sandbox = sinon.createSandbox();
+    });
+    afterEach(() => {
+        sandbox.restore();
+    });
+    it("getSelectedTenant happy path", async () => {
+        let spyConf = sandbox.stub(vscode.workspace, "getConfiguration").returns({
+            get: () => {
+                return "test";
+            },
+        } as any);
+        let res = await AzureAccount.getSelectedTenant();
+        sandbox.assert.calledOnce(spyConf);
+        assert.strictEqual(res, "test");
+    });
+});