Add shortcut to switching to sovereign cloud within the tenant view

package.json

@@ -254,6 +254,12 @@
                 "title": "%azureTenantsView.signInToTenant%",
                 "category": "Azure",
                 "icon": "$(sign-in)"
+            },
+            {
+                "command": "azureTenantsView.configureSovereignCloud",
+                "title": "%azureTenantsView.configureSovereignCloud%",
+                "category": "Azure",
+                "icon": "$(gear)"
             }
         ],
         "viewsContainers": {
@@ -392,6 +398,11 @@
                     "submenu": "azureWorkspaceCreate",
                     "when": "view == azureWorkspace",
                     "group": "navigation@1"
+                },
+                {
+                    "command": "azureTenantsView.configureSovereignCloud",
+                    "when": "view == azureTenantsView",
+                    "group": "navigation@2"
                 }
             ],
             "view/item/context": [

package.nls.json

@@ -17,6 +17,7 @@
     "azureWorkspace.refresh": "Refresh Workspace",
     "azureTenantsView.refresh": "Refresh Accounts & Tenants",
     "azureTenantsView.signInToTenant": "Sign in to Tenant",
+    "azureTenantsView.configureSovereignCloud": "Configure Sovereign Cloud",
     "azureResourceGroups.selectedSubscriptions": "Selected Subscriptions",
     "azureResourceGroups.revealResource": "Reveal Resource",
     "azureResourceGroups.viewProperties": "View Properties",

src/commands/registerCommands.ts

@@ -29,6 +29,7 @@ import { reviewIssues } from './helpAndFeedback/reviewIssues';
 import { installExtension } from './installExtension';
 import { openInPortal } from './openInPortal';
 import { revealResource } from './revealResource';
+import { configureSovereignCloud } from './sovereignCloud/configureSovereignCloud';
 import { editTags } from './tags/editTags';
 import { viewProperties } from './viewProperties';
 
@@ -116,6 +117,8 @@ export function registerCommands(): void {
     });
 
     registerCommand('azureWorkspace.loadMore', async (context: IActionContext, node: AzExtTreeItem) => await ext.workspaceTree.loadMore(node, context));
+
+    registerCommand('azureTenantsView.configureSovereignCloud', configureSovereignCloud);
 }
 
 async function handleAzExtTreeItemRefresh(context: IActionContext, node?: ResourceGroupsItem): Promise<void> {

src/commands/sovereignCloud/ConfigureSovereignCloudContext.ts

@@ -0,0 +1,10 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { IActionContext } from "@microsoft/vscode-azext-utils";
+
+export interface ConfigureSovereignCloudContext extends IActionContext {
+    sovereignCloud?: string;
+}

src/commands/sovereignCloud/SovereignCloudListStep.ts

@@ -0,0 +1,24 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { AzureWizardPromptStep, IAzureQuickPickItem } from "@microsoft/vscode-azext-utils";
+import { ConfigureSovereignCloudContext } from "./ConfigureSovereignCloudContext";
+
+export class SovereignCloudListStep extends AzureWizardPromptStep<ConfigureSovereignCloudContext> {
+    public async prompt(context: ConfigureSovereignCloudContext): Promise<void> {
+        const picks: IAzureQuickPickItem<string>[] = [
+            { label: 'Azure (Default)', data: '' },
+            { label: 'Azure China', data: 'ChinaCloud' },
+            { label: 'Azure US Government', data: 'USGovernment' },
+            { label: 'A custom Microsoft Sovereign Cloud', data: 'custom' }
+        ]
+
+        context.sovereignCloud = (await context.ui.showQuickPick(picks, { placeHolder: 'Select a sovereign cloud' })).data;
+    }
+
+    public shouldPrompt(context: ConfigureSovereignCloudContext): boolean {
+        return !context.sovereignCloud;
+    }
+}

src/commands/sovereignCloud/SovereignCloudSetStep.ts

@@ -0,0 +1,25 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { AzureWizardExecuteStep } from "@microsoft/vscode-azext-utils";
+import * as vscode from 'vscode';
+import { settingUtils } from "../../utils/settingUtils";
+import { ConfigureSovereignCloudContext } from "./ConfigureSovereignCloudContext";
+
+export class SovereignCloudSetStep extends AzureWizardExecuteStep<ConfigureSovereignCloudContext> {
+    public priority: number = 10;
+
+    public async execute(context: ConfigureSovereignCloudContext): Promise<void> {
+        if (context.sovereignCloud === 'custom') {
+            await vscode.commands.executeCommand('workbench.action.openSettings', 'microsoft-sovereign-cloud');
+        } else {
+            await settingUtils.updateGlobalSetting('environment', context.sovereignCloud, 'microsoft-sovereign-cloud');
+        }
+    }
+
+    public shouldExecute(): boolean {
+        return true;
+    }
+}

src/commands/sovereignCloud/configureSovereignCloud.ts

@@ -0,0 +1,32 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { AzureWizard } from "@microsoft/vscode-azext-utils";
+import { ext } from "../../extensionVariables";
+import { localize } from "../../utils/localize";
+import { ConfigureSovereignCloudContext } from "./ConfigureSovereignCloudContext";
+import { SovereignCloudListStep } from "./SovereignCloudListStep";
+import { SovereignCloudSetStep } from "./SovereignCloudSetStep";
+
+export async function configureSovereignCloud(context: ConfigureSovereignCloudContext): Promise<void> {
+    const wizardContext: ConfigureSovereignCloudContext = {
+        ...context,
+    };
+
+    const title: string = localize('selectSovereignCloud', 'Select Sovereign Cloud');
+
+    const wizard: AzureWizard<ConfigureSovereignCloudContext> = new AzureWizard(wizardContext, {
+        title,
+        promptSteps: [new SovereignCloudListStep()],
+        executeSteps: [new SovereignCloudSetStep()],
+    });
+
+    await wizard.prompt();
+    await wizard.execute();
+
+    // refresh resources and tenant view to accurrately reflect information for the selected sovereign cloud
+    ext.actions.refreshAzureTree();
+    ext.actions.refreshTenantTree();
+}