Integrate DTS emulator with workspace view

package.json

@@ -58,6 +58,9 @@
                 "branches": [
                     {
                         "type": "func"
+                    },
+                    {
+                        "type": "DurableTaskSchedulerEmulator"
                     }
                 ],
                 "resources": true
@@ -407,6 +410,11 @@
                 "title": "%azureFunctions.enableSystemIdentity%",
                 "category": "Azure Functions"
             },
+            {
+                "command": "azureFunctions.durableTaskScheduler.copyEmulatorConnectionString",
+                "title": "%azureFunctions.durableTaskScheduler.copyEmulatorConnectionString%",
+                "category": "Azure Functions"
+            },
             {
                 "command": "azureFunctions.durableTaskScheduler.copySchedulerConnectionString",
                 "title": "%azureFunctions.durableTaskScheduler.copySchedulerConnectionString%",
@@ -440,7 +448,20 @@
             {
                 "command": "azureFunctions.durableTaskScheduler.openTaskHubDashboard",
                 "title": "%azureFunctions.durableTaskScheduler.openTaskHubDashboard%",
-                "category": "Azure Functions"
+                "category": "Azure Functions",
+                "icon": "$(dashboard)"
+            },
+            {
+                "command": "azureFunctions.durableTaskScheduler.startEmulator",
+                "title": "%azureFunctions.durableTaskScheduler.startEmulator%",
+                "category": "Azure Functions",
+                "icon": "$(debug-start)"
+            },
+            {
+                "command": "azureFunctions.durableTaskScheduler.stopEmulator",
+                "title": "%azureFunctions.durableTaskScheduler.stopEmulator%",
+                "category": "Azure Functions",
+                "icon": "$(debug-stop)"
             }
         ],
         "submenus": [
@@ -765,14 +786,19 @@
                     "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /userAssignedIdentity/",
                     "group": "1@1"
                 },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.copyEmulatorConnectionString",
+                    "when": "view =~ /(azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.emulatorInstance/",
+                    "group": "3@1"
+                },
                 {
                     "command": "azureFunctions.durableTaskScheduler.copySchedulerConnectionString",
                     "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /azFunc.dts.scheduler/",
                     "group": "3@1"
                 },
                 {
                     "command": "azureFunctions.durableTaskScheduler.copySchedulerEndpoint",
-                    "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /azFunc.dts.scheduler/",
+                    "when": "view =~ /(azureResourceGroups|azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.schedulerEndpoint/",
                     "group": "3@2"
                 },
                 {
@@ -797,8 +823,33 @@
                 },
                 {
                     "command": "azureFunctions.durableTaskScheduler.openTaskHubDashboard",
-                    "when": "view =~ /(azureResourceGroups|azureFocusView)/ && viewItem =~ /azFunc.dts.taskHub/",
+                    "when": "view =~ /(azureResourceGroups|azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.taskHubDashboard/",
+                    "group": "1@1"
+                },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.openTaskHubDashboard",
+                    "when": "view =~ /(azureResourceGroups|azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.taskHubDashboard/",
+                    "group": "inline"
+                },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.startEmulator",
+                    "when": "view == azureWorkspace && viewItem =~ /azFunc.dts.emulators/",
+                    "group": "1@1"
+                },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.startEmulator",
+                    "when": "view =~ /(azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.emulators/",
+                    "group": "inline"
+                },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.stopEmulator",
+                    "when": "view =~ /(azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.emulatorInstance/",
                     "group": "1@1"
+                },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.stopEmulator",
+                    "when": "view =~ /(azureWorkspace|azureFocusView)/ && viewItem =~ /azFunc.dts.emulatorInstance/",
+                    "group": "inline"
                 }
             ],
             "explorer/context": [
@@ -852,6 +903,10 @@
                     "command": "azureFunctions.viewProperties",
                     "when": "never"
                 },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.copyEmulatorConnectionString",
+                    "when": "never"
+                },
                 {
                     "command": "azureFunctions.durableTaskScheduler.copySchedulerConnectionString",
                     "when": "never"
@@ -880,6 +935,10 @@
                     "command": "azureFunctions.durableTaskScheduler.openTaskHubDashboard",
                     "when": "never"
                 },
+                {
+                    "command": "azureFunctions.durableTaskScheduler.stopEmulator",
+                    "when": "never"
+                },
                 {
                     "command": "azureFunctions.unassignManagedIdentity",
                     "when": "never"
@@ -997,6 +1056,21 @@
                         "type": "boolean",
                         "default": false,
                         "description": "%azureFunctions.durableTaskScheduler.enablePreviewFeatures%"
+                    },
+                    "azureFunctions.durableTaskScheduler.emulatorRegistry": {
+                        "type": "string",
+                        "description": "%azureFunctions.durableTaskScheduler.emulatorRegistry%",
+                        "default": "mcr.microsoft.com"
+                    },
+                    "azureFunctions.durableTaskScheduler.emulatorImage": {
+                        "type": "string",
+                        "description": "%azureFunctions.durableTaskScheduler.emulatorImage%",
+                        "default": "dts/dts-emulator"
+                    },
+                    "azureFunctions.durableTaskScheduler.emulatorTag": {
+                        "type": "string",
+                        "description": "%azureFunctions.durableTaskScheduler.emulatorTag%",
+                        "default": "latest"
                     }
                 }
             },
@@ -1395,6 +1469,7 @@
         "vinyl-buffer": "^1.0.1",
         "vinyl-source-stream": "^2.0.0",
         "vscode-azurekudu": "^0.2.0",
+        "vscode-jsonrpc": "^8.2.1",
         "webpack": "^5.96.1",
         "webpack-cli": "^4.6.0"
     },
@@ -1415,6 +1490,7 @@
         "@microsoft/vscode-azext-azureutils": "^3.1.7",
         "@microsoft/vscode-azext-utils": "^2.6.6",
         "@microsoft/vscode-azureresources-api": "^2.0.4",
+        "@microsoft/vscode-container-client": "^0.1.2",
         "cross-fetch": "^4.0.0",
         "escape-string-regexp": "^4.0.0",
         "extract-zip": "^2.0.1",

package.nls.json

@@ -125,6 +125,7 @@
     "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.copyEmulatorConnectionString": "Copy Connection String",
     "azureFunctions.durableTaskScheduler.copySchedulerConnectionString": "Copy Connection String",
     "azureFunctions.durableTaskScheduler.copySchedulerEndpoint": "Copy Endpoint",
     "azureFunctions.durableTaskScheduler.createScheduler": "Create Durable Task Scheduler...",
@@ -133,5 +134,10 @@
     "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"
+    "azureFunctions.durableTaskScheduler.startEmulator": "Start Durable Task Emulator",
+    "azureFunctions.durableTaskScheduler.stopEmulator": "Stop Emulator",
+    "azureFunctions.durableTaskScheduler.enablePreviewFeatures": "Enable Durable Task Scheduler preview features",
+    "azureFunctions.durableTaskScheduler.emulatorRegistry": "The registry of the Durable Task Scheduler emulator image.",
+    "azureFunctions.durableTaskScheduler.emulatorImage": "The name of the Durable Task Scheduler emulator image.",
+    "azureFunctions.durableTaskScheduler.emulatorTag": "The tag of the Durable Task Scheduler emulator image."
 }

src/commands/durableTaskScheduler/copyEmulatorConnectionString.ts

@@ -0,0 +1,58 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { type IActionContext } from "@microsoft/vscode-azext-utils";
+import { localize } from "../../localize";
+import { ext } from "../../extensionVariables";
+import { env, QuickPickItemKind, type QuickPickItem } from "vscode";
+import { type DurableTaskSchedulerEmulatorWorkspaceResourceModel } from "../../tree/durableTaskScheduler/DurableTaskSchedulerEmulatorWorkspaceResourceModel";
+
+export function copyEmulatorConnectionStringCommandFactory() {
+    return async (actionContext: IActionContext, scheduler: DurableTaskSchedulerEmulatorWorkspaceResourceModel | undefined): Promise<void> => {
+        if (!scheduler) {
+            throw new Error(localize('noSchedulerSelectedErrorMessage', 'No scheduler was selected.'));
+        }
+
+        const { endpointUrl } = scheduler;
+
+        let connectionString = `Endpoint=${endpointUrl};Authentication=None`;
+
+        const taskHubs = scheduler.taskHubs;
+
+        if (taskHubs.length > 0) {
+
+            const noTaskHubItem: QuickPickItem = {
+                    detail: localize('noTaskHubDetail', 'Do not connect to a specific task hub.'),
+                    label: localize('noTaskHubLabel', 'None')
+                }
+
+            const taskHubItems: QuickPickItem[] =
+                taskHubs.map(taskHub => ({ label: taskHub }));
+
+            const taskHubResult = await actionContext.ui.showQuickPick(
+                [
+                    noTaskHubItem,
+                    {
+                        kind: QuickPickItemKind.Separator,
+                        label: localize('taskHubSepratorLabel', 'Task Hubs')
+                    },
+                    ...taskHubItems
+                ],
+                {
+                    canPickMany: false,
+                    placeHolder: localize('taskHubSelectionPlaceholder', 'Select a task hub to connect to')
+                });
+
+            if (taskHubResult && taskHubResult !== noTaskHubItem) {
+                connectionString += `;TaskHub=${taskHubResult.label}`;
+            }
+        }
+
+        await env.clipboard.writeText(connectionString);
+
+        ext.outputChannel.show();
+        ext.outputChannel.appendLog(localize('schedulerConnectionStringCopiedMessage', 'Connection string copied to clipboard: {0}', connectionString));
+    }
+}

src/commands/durableTaskScheduler/copySchedulerConnectionString.ts

@@ -16,16 +16,7 @@ export function copySchedulerConnectionStringCommandFactory(schedulerClient: Dur
             throw new Error(localize('noSchedulerSelectedErrorMessage', 'No scheduler was selected.'));
         }
 
-        const schedulerJson = await schedulerClient.getScheduler(
-            scheduler.subscription,
-            scheduler.resourceGroup,
-            scheduler.name);
-
-        if (!schedulerJson) {
-            throw new Error(localize('schedulerNotFoundErrorMessage', 'Scheduler does not exist.'));
-        }
-
-        const { endpoint } = schedulerJson.properties;
+        const { endpointUrl } = scheduler;
 
         const noAuthentication: QuickPickItem = {
             detail: localize('noAuthenticationDetail', 'No credentials will be used.'),
@@ -59,7 +50,7 @@ export function copySchedulerConnectionStringCommandFactory(schedulerClient: Dur
                 placeHolder: localize('authenticationTypePlaceholder', 'Select the credentials to be used to connect to the scheduler')
             });
 
-        let connectionString = `Endpoint=${endpoint};Authentication=`
+        let connectionString = `Endpoint=${endpointUrl};Authentication=`
 
         if (result === noAuthentication) {
             connectionString += 'None';

src/commands/durableTaskScheduler/copySchedulerEndpoint.ts

@@ -4,32 +4,22 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { type IActionContext } from "@microsoft/vscode-azext-utils";
-import { type DurableTaskSchedulerClient } from "../../tree/durableTaskScheduler/DurableTaskSchedulerClient";
-import { type DurableTaskSchedulerResourceModel } from "../../tree/durableTaskScheduler/DurableTaskSchedulerResourceModel";
 import { localize } from "../../localize";
 import { ext } from "../../extensionVariables";
 import { env } from "vscode";
+import { type DurableTaskSchedulerEndpointModel } from "../../tree/durableTaskScheduler/DurableTaskSchedulerEndpointModel";
 
-export function copySchedulerEndpointCommandFactory(schedulerClient: DurableTaskSchedulerClient) {
-    return async (_: IActionContext, scheduler: DurableTaskSchedulerResourceModel | undefined): Promise<void> => {
+export function copySchedulerEndpointCommandFactory() {
+    return async (_: IActionContext, scheduler: DurableTaskSchedulerEndpointModel | undefined): Promise<void> => {
         if (!scheduler) {
             throw new Error(localize('noSchedulerSelectedErrorMessage', 'No scheduler was selected.'));
         }
 
-        const schedulerJson = await schedulerClient.getScheduler(
-            scheduler.subscription,
-            scheduler.resourceGroup,
-            scheduler.name);
+        const { endpointUrl } = scheduler;
 
-        if (!schedulerJson) {
-            throw new Error(localize('schedulerNotFoundErrorMessage', 'Scheduler does not exist.'));
-        }
-
-        const { endpoint } = schedulerJson.properties;
-
-        await env.clipboard.writeText(endpoint);
+        await env.clipboard.writeText(endpointUrl.toString());
 
         ext.outputChannel.show();
-        ext.outputChannel.appendLog(localize('schedulerEndpointCopiedMessage', 'Endpoint copied to clipboard: {0}', endpoint));
+        ext.outputChannel.appendLog(localize('schedulerEndpointCopiedMessage', 'Endpoint copied to clipboard: {0}', endpointUrl.toString()));
     }
 }

src/commands/durableTaskScheduler/openTaskHubDashboard.ts

@@ -4,10 +4,10 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { openUrl, type IActionContext } from "@microsoft/vscode-azext-utils";
-import { type DurableTaskHubResourceModel } from "../../tree/durableTaskScheduler/DurableTaskHubResourceModel";
 import { localize } from '../../localize';
+import { type DurableTaskSchedulerDashboardModel } from "../../tree/durableTaskScheduler/DurableTaskSchedulerDashboardModel";
 
-export async function openTaskHubDashboard(_: IActionContext, taskHub: DurableTaskHubResourceModel | undefined): Promise<void> {
+export async function openTaskHubDashboard(_: IActionContext, taskHub: DurableTaskSchedulerDashboardModel | undefined): Promise<void> {
     if (!taskHub) {
         throw new Error(localize('noTaskHubSelectedErrorMessage', 'No task hub was selected.'));
     }

src/commands/durableTaskScheduler/startEmulator.ts

@@ -0,0 +1,13 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { type IActionContext } from "@microsoft/vscode-azext-utils";
+import { type DurableTaskSchedulerEmulatorClient } from "../../tree/durableTaskScheduler/DurableTaskSchedulerEmulatorClient";
+
+export function startEmulatorCommandFactory(emulatorClient: DurableTaskSchedulerEmulatorClient) {
+    return async (_: IActionContext) => {
+        await emulatorClient.startEmulator();
+    };
+}