Contribute activity children with SchedulerCreationStep

Author: MicroFish91

src/commands/durableTaskScheduler/createScheduler.ts

@@ -5,7 +5,7 @@
 
 import { type ResourceManagementClient } from '@azure/arm-resources';
 import { type AzExtClientContext, createAzureClient, type ILocationWizardContext, type IResourceGroupWizardContext, LocationListStep, parseClientContext, ResourceGroupCreateStep, ResourceGroupListStep, VerifyProvidersStep } from "@microsoft/vscode-azext-azureutils";
-import { AzureWizard, AzureWizardExecuteStep, AzureWizardPromptStep, createSubscriptionContext, type ExecuteActivityContext, type IActionContext, type IAzureQuickPickItem, type ISubscriptionActionContext, subscriptionExperience } from "@microsoft/vscode-azext-utils";
+import { AzureWizard, AzureWizardExecuteStepWithActivityOutput, AzureWizardPromptStep, createSubscriptionContext, type ExecuteActivityContext, type IActionContext, type IAzureQuickPickItem, type ISubscriptionActionContext, subscriptionExperience } from "@microsoft/vscode-azext-utils";
 import { type AzureSubscription } from "@microsoft/vscode-azureresources-api";
 import { type Progress } from "vscode";
 import { DurableTaskProvider, DurableTaskSchedulersResourceType } from "../../constants";
@@ -59,16 +59,22 @@ class SchedulerSkuStep extends AzureWizardPromptStep<ICreateSchedulerContext> {
     }
 }
 
-class SchedulerCreationStep extends AzureWizardExecuteStep<ICreateSchedulerContext> {
+class SchedulerCreationStep extends AzureWizardExecuteStepWithActivityOutput<ICreateSchedulerContext> {
     priority: number = 110;
+    readonly stepName: string = 'schedulerCreationStep';
+
+    protected getTreeItemLabel = (context: ICreateSchedulerContext) => localize('createSchedulerLabel', 'Create Durable Task Scheduler "{0}"', context.schedulerName);
+    protected getOutputLogSuccess = (context: ICreateSchedulerContext) => localize('createSchedulerSuccess', 'Successfully created Durable Task Scheduler "{0}".', context.schedulerName);
+    protected getOutputLogFail = (context: ICreateSchedulerContext) => localize('createSchedulerFail', 'Failed to create Durable Task Scheduler "{0}".', context.schedulerName);
 
     constructor(private readonly schedulerClient: DurableTaskSchedulerClient) {
         super();
     }
 
-    async execute(wizardContext: ICreateSchedulerContext, _: Progress<{ message?: string; increment?: number; }>): Promise<void> {
-        const location = await LocationListStep.getLocation(wizardContext);
+    async execute(wizardContext: ICreateSchedulerContext, progress: Progress<{ message?: string; increment?: number; }>): Promise<void> {
+        progress.report({ message: localize('creatingScheduler', 'Creating Durable Task Scheduler...') });
 
+        const location = await LocationListStep.getLocation(wizardContext);
         const response = await this.schedulerClient.createScheduler(
             wizardContext.subscription as AzureSubscription,
             wizardContext.resourceGroup?.name as string,
@@ -117,7 +123,7 @@ export function createSchedulerCommandFactory(dataBranchProvider: DurableTaskSch
 
             ...actionContext,
             ...createSubscriptionContext(subscription),
-            ...await createActivityContext()
+            ...await createActivityContext({ withChildren: true }),
         };
 
         if (!await isDtsProviderRegistered(wizardContext)) {