Create container on flex app creation and allow flex nodes to be passed into deploy (#4088)

* Create a container at time of flex app creation

* Add flex context for deploy

* Remove extra break

Author: nturinski

src/commands/createFunctionApp/FunctionAppCreateStep.ts

@@ -5,6 +5,7 @@
 
 import { type NameValuePair, type Site, type SiteConfig, type WebSiteManagementClient } from '@azure/arm-appservice';
 import { createHttpHeaders, createPipelineRequest, type RequestBodyType } from '@azure/core-rest-pipeline';
+import { BlobServiceClient } from '@azure/storage-blob';
 import { ParsedSite, WebsiteOS, type CustomLocation, type IAppServiceWizardContext } from '@microsoft/vscode-azext-azureappservice';
 import { LocationListStep, createGenericClient, type AzExtPipelineResponse, type AzExtRequestPrepareOptions } from '@microsoft/vscode-azext-azureutils';
 import { AzureWizardExecuteStep, parseError, randomUtils } from '@microsoft/vscode-azext-utils';
@@ -245,6 +246,9 @@ export class FunctionAppCreateStep extends AzureWizardExecuteStep<IFunctionAppWi
         const client = await createGenericClient(context, context);
         const result = await client.sendRequest(createPipelineRequest(options)) as AzExtPipelineResponse;
         if (result && result.status >= 200 && result.status < 300) {
+            const site = result.parsedBody as Site & { properties: FlexFunctionAppProperties };
+            const storageConnectionString: string = (await getStorageConnectionString(context)).connectionString;
+            await tryCreateStorageContainer(site, storageConnectionString);
             const client: WebSiteManagementClient = await createWebSiteClient(context);
             // the payload for the new API version "2023-12-01" is incompatiable with our current SiteClient so get the old payload
             try {
@@ -275,6 +279,18 @@ function getSiteKind(context: IAppServiceWizardContext): string {
     return kind;
 }
 
+// storage container is needed for flex deployment, but it is not created automatically
+async function tryCreateStorageContainer(site: Site & { properties: FlexFunctionAppProperties }, storageConnectionString: string): Promise<void> {
+    const blobClient = BlobServiceClient.fromConnectionString(storageConnectionString);
+    const containerName = site.properties?.functionAppConfig?.deployment.storage.value.split('/').pop();
+    if (containerName) {
+        const client = blobClient.getContainerClient(containerName);
+        if (!await client.exists()) {
+            await blobClient.createContainer(containerName);
+        }
+    }
+}
+
 type FlexFunctionAppProperties = {
     containerSize?: number,
     sku?: 'FlexConsumption',

src/commands/deploy/deploy.ts

@@ -61,7 +61,8 @@ async function deploy(actionContext: IActionContext, arg1: vscode.Uri | string |
 
     if (treeUtils.isAzExtTreeItem(arg1)) {
         if (!arg1.contextValue.match(ResolvedFunctionAppResource.pickSlotContextValue) &&
-            !arg1.contextValue.match(ResolvedFunctionAppResource.productionContextValue)) {
+            !arg1.contextValue.match(ResolvedFunctionAppResource.productionContextValue) &&
+            !arg1.contextValue.match(ResolvedFunctionAppResource.flexContextValue)) {
             // if the user uses the deploy button, it's possible for the local project node to be passed in, so we should reset it to undefined
             arg1 = undefined;
         }

src/tree/ResolvedFunctionAppResource.ts

@@ -55,6 +55,7 @@ export class ResolvedFunctionAppResource extends ResolvedFunctionAppBase impleme
     public static pickSlotContextValue: RegExp = new RegExp(/azFuncSlot(?!s)/);
     public static productionContextValue: string = 'azFuncProductionSlot';
     public static slotContextValue: string = 'azFuncSlot';
+    public static flexContextValue: string = 'azFuncFlex';
 
     private _isFlex: boolean;
 
@@ -69,7 +70,7 @@ export class ResolvedFunctionAppResource extends ResolvedFunctionAppBase impleme
         this.contextValuesToAdd = [];
         this._isFlex = !!isFlex;
         if (this._isFlex) {
-            this.contextValuesToAdd.push('azFuncFlex');
+            this.contextValuesToAdd.push(ResolvedFunctionAppResource.flexContextValue);
         } else if (this.site.isSlot) {
             this.contextValuesToAdd.push(ResolvedFunctionAppResource.slotContextValue);
         } else {