Decouple editTargetPort from createContainerApp (#371)

src/commands/createContainerApp/EnableIngressStep.ts

@@ -5,9 +5,9 @@
 
 import { AzureWizardPromptStep, IWizardOptions } from "@microsoft/vscode-azext-utils";
 import { localize } from "../../utils/localize";
+import { TargetPortInputStep } from "../ingress/editTargetPort/TargetPortInputStep";
 import { ICreateContainerAppContext } from "./ICreateContainerAppContext";
 import { IngressVisibilityStep } from "./IngressVisibilityStep";
-import { TargetPortStep } from "./TargetPortStep";
 
 export class EnableIngressStep extends AzureWizardPromptStep<ICreateContainerAppContext> {
     public async prompt(context: ICreateContainerAppContext): Promise<void> {
@@ -23,7 +23,9 @@ export class EnableIngressStep extends AzureWizardPromptStep<ICreateContainerApp
     public async getSubWizard(context: ICreateContainerAppContext): Promise<IWizardOptions<ICreateContainerAppContext> | undefined> {
         if (context.enableIngress) {
             return {
-                promptSteps: [new IngressVisibilityStep(), new TargetPortStep()]
+                // Context types will be temporarily incompatible until all the ingress commands are decoupled from the create command
+                // @ts-ignore
+                promptSteps: [new IngressVisibilityStep(), new TargetPortInputStep()]
             }
         }
 

src/commands/ingress/IngressContext.ts

@@ -0,0 +1,14 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.txt in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import type { ExecuteActivityContext } from "@microsoft/vscode-azext-utils";
+import type { IContainerAppContext } from "../IContainerAppContext";
+
+export interface IngressContext extends IContainerAppContext, ExecuteActivityContext {
+    enableIngress?: boolean;
+    enableExternal?: boolean;
+
+    targetPort?: number;
+}

src/commands/ingress/IngressUpdateBaseStep.ts

@@ -0,0 +1,30 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import type { Ingress } from "@azure/arm-appcontainers";
+import { AzureWizardExecuteStep, nonNullProp } from "@microsoft/vscode-azext-utils";
+import type { Progress } from "vscode";
+import { ext } from "../../extensionVariables";
+import type { IContainerAppContext } from "../IContainerAppContext";
+import { updateContainerApp } from "../deployContainerApp/updateContainerApp";
+
+type IngressOptions = {
+    ingress: Ingress | null,
+    working: string,
+    workCompleted: string
+}
+
+export abstract class IngressUpdateBaseStep<T extends IContainerAppContext> extends AzureWizardExecuteStep<T> {
+    protected async updateIngressSettings(context: T, progress: Progress<{ message?: string | undefined}>, options: IngressOptions): Promise<void> {
+        const containerApp = nonNullProp(context, 'containerApp');
+        const { ingress, working, workCompleted } = options;
+
+        progress.report({ message: working });
+        await updateContainerApp(context, context.subscription, containerApp, { configuration: { ingress: ingress as Ingress | undefined } });
+
+        ext.outputChannel.appendLog(workCompleted);
+        ext.state.notifyChildrenChanged(containerApp.managedEnvironmentId);
+    }
+}

src/commands/createContainerApp/TargetPortStep.ts → src/commands/ingress/editTargetPort/TargetPortInputStep.ts

@@ -4,19 +4,20 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { AzureWizardPromptStep } from "@microsoft/vscode-azext-utils";
-import { localize } from "../../utils/localize";
-import { ICreateContainerAppContext } from "./ICreateContainerAppContext";
+import { localize } from "../../../utils/localize";
+import type { IngressContext } from "../IngressContext";
+import { getDefaultPort } from "./getDefaultPort";
 
-export class TargetPortStep extends AzureWizardPromptStep<ICreateContainerAppContext> {
-    public async prompt(context: ICreateContainerAppContext): Promise<void> {
+export class TargetPortInputStep extends AzureWizardPromptStep<IngressContext> {
+    public async prompt(context: IngressContext): Promise<void> {
         context.targetPort = Number(await context.ui.showInputBox({
             prompt: localize('targetPort', 'This is the port your container is listening on that will receive traffic. Set this value to the port number that your container uses.'),
-            value: String(context.defaultPort ?? 80),
+            value: String(getDefaultPort(context)),
             validateInput: this.validateInput
         }));
     }
 
-    public shouldPrompt(context: ICreateContainerAppContext): boolean {
+    public shouldPrompt(context: IngressContext): boolean {
         return !context.targetPort;
     }
 

src/commands/ingress/editTargetPort/TargetPortUpdateStep.ts

@@ -0,0 +1,30 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { nonNullProp, nonNullValueAndProp } from "@microsoft/vscode-azext-utils";
+import type { Progress } from "vscode";
+import { localize } from "../../../utils/localize";
+import type { IngressContext } from "../IngressContext";
+import { IngressUpdateBaseStep } from "../IngressUpdateBaseStep";
+
+export class TargetPortUpdateStep extends IngressUpdateBaseStep<IngressContext> {
+    public priority: number = 280;
+
+    public async execute(context: IngressContext, progress: Progress<{ message?: string | undefined; increment?: number | undefined }>): Promise<void> {
+        const containerApp = nonNullProp(context, 'containerApp');
+        const ingress = nonNullValueAndProp(containerApp.configuration, 'ingress');
+        ingress.targetPort = context.targetPort;
+
+        const working: string = localize('updatingTargetPort', 'Updating target port...');
+        const workCompleted: string = localize('updatedTargetPort', 'Updated target port to {0} for container app "{1}"', context.targetPort, containerApp.name);
+
+        context.activityTitle = localize('updateTargetPort', 'Update target port to {0} for container app "{1}"', context.targetPort, containerApp.name);
+        await this.updateIngressSettings(context, progress, { ingress, working, workCompleted });
+    }
+
+    public shouldExecute(): boolean {
+        return true;
+    }
+}

src/commands/ingress/editTargetPort/editTargetPort.ts

@@ -0,0 +1,46 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the MIT License. See License.txt in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+
+import { AzureWizard, AzureWizardExecuteStep, AzureWizardPromptStep, createSubscriptionContext, IActionContext } from "@microsoft/vscode-azext-utils";
+import type { ContainerAppItem } from "../../../tree/ContainerAppItem";
+import { IngressItem } from "../../../tree/IngressItem";
+import { createActivityContext } from "../../../utils/activityUtils";
+import { localize } from "../../../utils/localize";
+import { pickContainerApp } from "../../../utils/pickContainerApp";
+import type { IngressContext } from "../IngressContext";
+import { TargetPortInputStep } from "./TargetPortInputStep";
+import { TargetPortUpdateStep } from "./TargetPortUpdateStep";
+
+export async function editTargetPort(context: IActionContext, node?: IngressItem): Promise<void> {
+    const { subscription, containerApp }: ContainerAppItem | IngressItem = node ?? await pickContainerApp(context);
+
+    const wizardContext: IngressContext = {
+        ...context,
+        ...createSubscriptionContext(subscription),
+        ...(await createActivityContext()),
+        subscription,
+        containerApp
+    };
+
+    const title: string = localize('updateTargetPort', 'Update target port for container app "{0}"', containerApp.name);
+
+    const promptSteps: AzureWizardPromptStep<IngressContext>[] = [
+        new TargetPortInputStep()
+    ];
+
+    const executeSteps: AzureWizardExecuteStep<IngressContext>[] = [
+        new TargetPortUpdateStep()
+    ];
+
+    const wizard: AzureWizard<IngressContext> = new AzureWizard(wizardContext, {
+        title,
+        promptSteps,
+        executeSteps,
+        showLoadingPrompt: true
+    });
+
+    await wizard.prompt();
+    await wizard.execute();
+}

src/commands/ingress/editTargetPort/getDefaultPort.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 type { IngressContext } from "../IngressContext";
+
+export function getDefaultPort(context: IngressContext, fallbackPort: number = 80): number {
+    return context.containerApp?.configuration?.ingress?.targetPort || fallbackPort;
+}

src/commands/registerCommands.ts

@@ -16,7 +16,7 @@ import { connectToGitHub } from './gitHub/connectToGitHub/connectToGitHub';
 import { disconnectRepo } from './gitHub/disconnectRepo/disconnectRepo';
 import { openGitHubRepo } from './gitHub/openGitHubRepo';
 import { disableIngress } from './ingress/disableIngress';
-import { editTargetPort } from './ingress/editTargetPort';
+import { editTargetPort } from './ingress/editTargetPort/editTargetPort';
 import { enableIngress } from './ingress/enableIngress';
 import { toggleIngressVisibility } from './ingress/toggleIngressVisibility';
 import { startStreamingLogs } from './logStream/startStreamingLogs';