From 31502ee2d711bb56b77d41ddb801e513d1d1f252 Mon Sep 17 00:00:00 2001
From: Alex Weininger <alex.weininger@live.com>
Date: Tue, 18 Jul 2023 18:03:40 -0700
Subject: [PATCH] Ensure `treeDataProvider` exists on created web app item
 (#2546)

---
 src/tree/SiteTreeItem.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/tree/SiteTreeItem.ts b/src/tree/SiteTreeItem.ts
index 1cf410a2..3f9994a1 100644
--- a/src/tree/SiteTreeItem.ts
+++ b/src/tree/SiteTreeItem.ts
@@ -6,7 +6,8 @@
 import { Site } from '@azure/arm-appservice';
 import { DeploymentsTreeItem, ParsedSite } from '@microsoft/vscode-azext-azureappservice';
 import { AppSettingsTreeItem } from '@microsoft/vscode-azext-azureappsettings';
-import { AzExtParentTreeItem, AzExtTreeItem, IActionContext, TreeItemIconPath } from '@microsoft/vscode-azext-utils';
+import { AzExtParentTreeItem, AzExtTreeDataProvider, AzExtTreeItem, IActionContext, TreeItemIconPath } from '@microsoft/vscode-azext-utils';
+import { ext } from '../extensionVariables';
 import { ISiteTreeItem } from './ISiteTreeItem';
 import { ResolvedWebAppResource } from './ResolvedWebAppResource';
 
@@ -17,8 +18,10 @@ export class SiteTreeItem extends AzExtParentTreeItem implements ISiteTreeItem {
     public appSettingsNode!: AppSettingsTreeItem;
     public deploymentsNode: DeploymentsTreeItem | undefined;
 
+    public treeDataProvider: AzExtTreeDataProvider;
     public constructor(parent: AzExtParentTreeItem, site: Site) {
         super(parent);
+        this.treeDataProvider = parent.treeDataProvider ?? ext.rgApi.appResourceTree;
         this.resolved = new ResolvedWebAppResource(parent.subscription, site);
     }