Add Local and Remote Settings conversion

package.json

@@ -156,6 +156,16 @@
                 "title": "%azureFunctions.connectToGitHub%",
                 "category": "Azure Functions"
             },
+            {
+                "command": "azureFunctions.addLocalMIConnections",
+                "title": "%azureFunctions.addLocalMIConnections%",
+                "category": "Azure Functions"
+            },
+            {
+                "command": "azureFunctions.addRemoteMIConnections",
+                "title": "%azureFunctions.addRemoteMIConnections%",
+                "category": "Azure Functions"
+            },
             {
                 "command": "azureFunctions.copyFunctionUrl",
                 "title": "%azureFunctions.copyFunctionUrl%",
@@ -484,6 +494,16 @@
                     "when": "view == azureResourceGroups && viewItem =~ /functionapp/i && viewItem =~ /azureResourceTypeGroup/i",
                     "group": "1@2"
                 },
+                {
+                    "command": "azureFunctions.addLocalMIConnections",
+                    "when": "view == azureWorkspace && viewItem =~ /(azFuncLocalProject.*convert|applicationSettingItem.*localSettings)/i",
+                    "group": "1@3"
+                },
+                {
+                    "command": "azureFunctions.addRemoteMIConnections",
+                    "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /(applicationSettings.*azFunc.*convert)|(applicationSettingItem.*azFunc.*convertSetting)/i",
+                    "group": "1@4"
+                },
                 {
                     "command": "azureFunctions.browseWebsite",
                     "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /azFunc(Production|)(Slot|Flex)(?!s)/",
@@ -661,7 +681,7 @@
                 },
                 {
                     "command": "azureFunctions.toggleAppSettingVisibility",
-                    "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /applicationSettingItem.*azFunc/",
+                    "when": "view =~ /(azureResourceGroups|azureFocusView|azureWorkspace)/ && viewItem =~ /applicationSettingItem.*(azFunc|localSettings)/",
                     "group": "inline"
                 },
                 {
@@ -1345,7 +1365,7 @@
         "@azure/core-rest-pipeline": "^1.11.0",
         "@azure/storage-blob": "^12.5.0",
         "@microsoft/vscode-azext-azureappservice": "^3.3.1",
-        "@microsoft/vscode-azext-azureappsettings": "^0.2.2",
+        "@microsoft/vscode-azext-azureappsettings": "^0.2.4",
         "@microsoft/vscode-azext-azureutils": "^3.1.5",
         "@microsoft/vscode-azext-utils": "^2.6.3",
         "@microsoft/vscode-azureresources-api": "^2.0.4",

package.nls.json

@@ -21,6 +21,8 @@
     "azureFunctions.createNewProjectWithDockerfile": "Create New Containerized Project...",
     "azureFunctions.createPythonVenv": "Create a virtual environment when creating a new Python project.",
     "azureFunctions.createSlot": "Create Slot...",
+    "azureFunctions.addLocalMIConnections": "Add Local Project Identity Connections...",
+    "azureFunctions.addRemoteMIConnections": "Add Function App Identity Connections...",
     "azureFunctions.deleteFunction": "Delete Function...",
     "azureFunctions.deleteFunctionApp": "Delete Function App...",
     "azureFunctions.deleteSlot": "Delete Slot...",
@@ -119,7 +121,6 @@
     "azureFunctions.walkthrough.functionsStart.scenarios.description": "Learn how you can use Azure Functions to build event-driven systems.\n\nIf you're just getting started with Azure Functions, you can [learn about the anatomy of an Azure Functions application](https://aka.ms/functions-getstarted-devguide).",
     "azureFunctions.walkthrough.functionsStart.scenarios.title": "Explore common scenarios",
     "azureFunctions.walkthrough.functionsStart.title": "Get Started with Azure Functions",
-
     "azureFunctions.durableTaskScheduler.copySchedulerConnectionString": "Copy Connection String",
     "azureFunctions.durableTaskScheduler.copySchedulerEndpoint": "Copy Endpoint",
     "azureFunctions.durableTaskScheduler.createScheduler": "Create Durable Task Scheduler...",
@@ -128,6 +129,5 @@
     "azureFunctions.durableTaskScheduler.deleteScheduler": "Delete Scheduler...",
     "azureFunctions.durableTaskScheduler.deleteTaskHub": "Delete Task Hub...",
     "azureFunctions.durableTaskScheduler.openTaskHubDashboard": "Open in Dashboard",
-
     "azureFunctions.durableTaskScheduler.enablePreviewFeatures": "Enable Durable Task Scheduler preview features"
 }

src/LocalResourceProvider.ts

@@ -18,15 +18,14 @@ export class FunctionsLocalResourceProvider implements WorkspaceResourceProvider
 
     public async provideResources(parent: AzExtParentTreeItem): Promise<AzExtTreeItem[] | null | undefined> {
         const children: AzExtTreeItem[] = [];
-
         Disposable.from(...this._projectDisposables).dispose();
         this._projectDisposables = [];
 
         const localProjects = await listLocalProjects();
         let hasLocalProject = false;
 
         for (const project of localProjects.initializedProjects) {
-            const treeItem: LocalProjectTreeItem = new LocalProjectTreeItem(parent, project as LocalProjectInternal);
+            const treeItem: LocalProjectTreeItem = await LocalProjectTreeItem.createLocalProjectTreeItem(parent, project as LocalProjectInternal);
             this._projectDisposables.push(treeItem);
             children.push(treeItem);
         }

src/commands/addMIConnections/AddMIConnectionsContext.ts

@@ -0,0 +1,17 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { type IResourceGroupWizardContext, type Role } from "@microsoft/vscode-azext-azureutils";
+import { type ExecuteActivityContext, type ISubscriptionActionContext } from "@microsoft/vscode-azext-utils";
+import { type SlotTreeItem } from "../../tree/SlotTreeItem";
+import { type Connection } from "./ConnectionsListStep";
+
+export interface AddMIConnectionsContext extends ExecuteActivityContext, IResourceGroupWizardContext, ISubscriptionActionContext {
+    functionapp?: SlotTreeItem;
+    connections?: Connection[];
+    connectionsToAdd?: Connection[];
+    roles?: Role[];
+    localSettingsPath?: string;
+}

src/commands/addMIConnections/ConnectionsListStep.ts

@@ -0,0 +1,120 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { type StringDictionary } from "@azure/arm-appservice";
+import { isSettingConvertible } from "@microsoft/vscode-azext-azureappsettings";
+import { AzExtFsExtra, AzureWizardPromptStep, nonNullValue, type IAzureQuickPickItem } from "@microsoft/vscode-azext-utils";
+import * as vscode from 'vscode';
+import { type ILocalSettingsJson } from "../../funcConfig/local.settings";
+import { localize } from "../../localize";
+import { decryptLocalSettings } from "../appSettings/localSettings/decryptLocalSettings";
+import { encryptLocalSettings } from "../appSettings/localSettings/encryptLocalSettings";
+import { getLocalSettingsFile } from "../appSettings/localSettings/getLocalSettingsFile";
+import { type AddMIConnectionsContext } from "./AddMIConnectionsContext";
+
+export interface Connection {
+    name: string;
+    value: string;
+}
+
+export class ConnectionsListStep extends AzureWizardPromptStep<AddMIConnectionsContext> {
+    public async prompt(context: AddMIConnectionsContext): Promise<void> {
+        const picks = await this.getPicks(context);
+
+        if (picks.length === 0) {
+            const noItemFoundMessage: string = localize('noConnectionsFound', 'No connections found in local settings');
+            (await context.ui.showQuickPick(picks, {
+                placeHolder: localize('selectConnections', 'Select the connections you want to add managed identity support for'),
+                suppressPersistence: true,
+                noPicksMessage: noItemFoundMessage
+            }));
+        } else {
+            context.connections = (await context.ui.showQuickPick(picks, {
+                placeHolder: localize('selectConnections', 'Select the connections you want to add managed identity support for'),
+                suppressPersistence: true,
+                canPickMany: true,
+            })).map(item => item.data);
+        }
+    }
+
+    public shouldPrompt(context: AddMIConnectionsContext): boolean {
+        return !context.connections || context.connections.length === 0;
+    }
+
+    private async getPicks(context: AddMIConnectionsContext): Promise<IAzureQuickPickItem<Connection>[]> {
+        if (context.functionapp) {
+            return this.getRemoteQuickPicks(context);
+        } else {
+            return this.getLocalQuickPicks(context);
+        }
+    }
+
+    private async getLocalQuickPicks(context: AddMIConnectionsContext, workspaceFolder?: vscode.WorkspaceFolder): Promise<IAzureQuickPickItem<Connection>[]> {
+        const picks: IAzureQuickPickItem<Connection>[] = [];
+        const message: string = localize('selectLocalSettings', 'Select the local settings to add identity settings for.');
+        const localSettingsPath: string = await getLocalSettingsFile(context, message, workspaceFolder);
+        context.localSettingsPath = localSettingsPath;
+
+        if (await AzExtFsExtra.pathExists(localSettingsPath)) {
+            const localSettings = await getLocalSettingsJson(context, localSettingsPath);
+            if (localSettings.Values) {
+                for (const [key, value] of Object.entries(localSettings.Values)) {
+                    if (!isSettingConvertible(key, value)) {
+                        continue;
+                    }
+
+                    picks.push({
+                        label: key,
+                        data: {
+                            name: key,
+                            value: value
+                        }
+                    });
+                }
+            }
+        }
+
+        return picks;
+    }
+
+    private async getRemoteQuickPicks(context: AddMIConnectionsContext): Promise<IAzureQuickPickItem<Connection>[]> {
+        const picks: IAzureQuickPickItem<Connection>[] = [];
+
+        const client = await nonNullValue(context.functionapp?.site.createClient(context));
+        const appSettings: StringDictionary = await client.listApplicationSettings();
+        if (appSettings.properties) {
+            for (const [key, value] of Object.entries(appSettings.properties)) {
+                if (!isSettingConvertible(key, value)) {
+                    continue;
+                }
+
+                picks.push({
+                    label: key,
+                    data: {
+                        name: key,
+                        value: value
+                    }
+                });
+            }
+        }
+
+        return picks;
+    }
+}
+
+export async function getLocalSettingsJson(context: AddMIConnectionsContext, localSettingsPath: string): Promise<ILocalSettingsJson> {
+    let localSettings: ILocalSettingsJson = <ILocalSettingsJson>await AzExtFsExtra.readJSON(localSettingsPath);
+    const localSettingsUri: vscode.Uri = vscode.Uri.file(localSettingsPath);
+    if (localSettings.IsEncrypted) {
+        await decryptLocalSettings(context, localSettingsUri);
+        try {
+            localSettings = await AzExtFsExtra.readJSON<ILocalSettingsJson>(localSettingsPath);
+        } finally {
+            await encryptLocalSettings(context, localSettingsUri);
+        }
+    }
+
+    return localSettings;
+}

src/commands/addMIConnections/LocalSettingsAddStep.ts

@@ -0,0 +1,42 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { activitySuccessContext, activitySuccessIcon, AzExtFsExtra, AzureWizardExecuteStep, createUniversallyUniqueContextValue, GenericTreeItem, nonNullProp } from "@microsoft/vscode-azext-utils";
+import { ext } from "../../extensionVariables";
+import { localize } from "../../localize";
+import { getLocalSettingsFile } from "../appSettings/localSettings/getLocalSettingsFile";
+import { type AddMIConnectionsContext } from "./AddMIConnectionsContext";
+import { getLocalSettingsJson } from "./ConnectionsListStep";
+
+export class LocalSettingsAddStep extends AzureWizardExecuteStep<AddMIConnectionsContext> {
+    public priority: number = 125;
+
+    public async execute(context: AddMIConnectionsContext): Promise<void> {
+        // If right clicking on a connection we will have the connections to convert but not the local settings path
+        if (!context.localSettingsPath) {
+            const message: string = localize('selectLocalSettings', 'Select the local settings file to add connections to.');
+            context.localSettingsPath = await getLocalSettingsFile(context, message);
+        }
+
+        const localSettings = await getLocalSettingsJson(context, context.localSettingsPath);
+        if (localSettings.Values) {
+            for (const connection of nonNullProp(context, 'connectionsToAdd')) {
+                localSettings.Values[connection.name] = connection.value;
+                context.activityChildren?.push(
+                    new GenericTreeItem(undefined, {
+                        contextValue: createUniversallyUniqueContextValue(['useExistingResourceGroupInfoItem', activitySuccessContext]),
+                        label: localize('addedLocalSetting', 'Add Local setting "{0}"', connection.name),
+                        iconPath: activitySuccessIcon
+                    })
+                );
+            }
+            await AzExtFsExtra.writeJSON(nonNullProp(context, 'localSettingsPath'), localSettings);
+            await ext.rgApi.workspaceResourceTree.refresh(context);
+        }
+    }
+    public shouldExecute(context: AddMIConnectionsContext): boolean {
+        return !context.functionapp && !!context.connections
+    }
+}

src/commands/addMIConnections/RemoteSettingsAddStep.ts

@@ -0,0 +1,38 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.md in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { activitySuccessContext, activitySuccessIcon, AzureWizardExecuteStep, createUniversallyUniqueContextValue, GenericTreeItem, nonNullProp } from "@microsoft/vscode-azext-utils";
+import { ext } from "../../extensionVariables";
+import { localize } from "../../localize";
+import { type AddMIConnectionsContext } from "./AddMIConnectionsContext";
+
+export class RemoteSettingsAddStep extends AzureWizardExecuteStep<AddMIConnectionsContext> {
+    public priority: number = 110;
+
+    public async execute(context: AddMIConnectionsContext): Promise<void> {
+        const client = await nonNullProp(context, 'functionapp').site.createClient(context);
+        const remoteSettings = await client.listApplicationSettings();
+        const properties = remoteSettings.properties || {};
+        for (const connection of nonNullProp(context, 'connectionsToAdd')) {
+            properties[connection.name] = connection.value;
+        }
+
+        await client.updateApplicationSettings({ properties });
+        for (const connection of nonNullProp(context, 'connectionsToAdd')) {
+            context.activityChildren?.push(
+                new GenericTreeItem(undefined, {
+                    contextValue: createUniversallyUniqueContextValue(['useExistingResourceGroupInfoItem', activitySuccessContext]),
+                    label: localize('addedAppSetting', 'Add app setting "{0}"', connection.name),
+                    iconPath: activitySuccessIcon
+                })
+            );
+        }
+        await ext.rgApi.tree.refresh(context);
+    }
+
+    public shouldExecute(context: AddMIConnectionsContext): boolean {
+        return !!context.functionapp && !!context.connections;
+    }
+}