Improve logic apps loading in tree

apps/vs-code-designer/src/LogicAppResolver.ts

@@ -1,24 +1,46 @@
 import { LogicAppResourceTree } from './app/tree/LogicAppResourceTree';
 import { logicAppFilter } from './constants';
+import { ext } from './extensionVariables';
+import type { Site } from '@azure/arm-appservice';
 import { createWebSiteClient } from '@microsoft/vscode-azext-azureappservice';
-import { getResourceGroupFromId } from '@microsoft/vscode-azext-azureutils';
+import { uiUtils } from '@microsoft/vscode-azext-azureutils';
 import type { IActionContext, ISubscriptionContext } from '@microsoft/vscode-azext-utils';
-import { callWithTelemetryAndErrorHandling, nonNullProp } from '@microsoft/vscode-azext-utils';
+import { callWithTelemetryAndErrorHandling } from '@microsoft/vscode-azext-utils';
 import type { AppResource, AppResourceResolver } from '@microsoft/vscode-azext-utils/hostapi';
 
 export class LogicAppResolver implements AppResourceResolver {
   public async resolveResource(subContext: ISubscriptionContext, resource: AppResource): Promise<LogicAppResourceTree | undefined> {
     return await callWithTelemetryAndErrorHandling('resolveResource', async (context: IActionContext) => {
-      const client = await createWebSiteClient({ ...context, ...subContext });
-      const rg = getResourceGroupFromId(nonNullProp(resource, 'id'));
-      const name = nonNullProp(resource, 'name');
-
-      const site = await client.webApps.get(rg, name);
+      const site: Site = await this.getAppResourceSite(context, subContext, resource);
+      if (!site) {
+        return;
+      }
       return LogicAppResourceTree.createLogicAppResourceTree(context, subContext, site);
     });
   }
 
   public matchesResource(resource: AppResource): boolean {
     return resource.type.toLowerCase() === logicAppFilter.type && resource.kind?.toLowerCase() === logicAppFilter.kind;
   }
+
+  private async getAppResourceSite(context: IActionContext, subContext: ISubscriptionContext, resource: AppResource): Promise<Site> {
+    const logicAppsSites = ext.logicAppSitesMap.get(subContext.subscriptionId);
+    let site: Site;
+
+    if (!logicAppsSites) {
+      const client = await createWebSiteClient({ ...context, ...subContext });
+
+      const listOfSites: Site[] = await uiUtils.listAllIterator(client.webApps.list());
+      const subscriptionSites = new Map<string, Site>();
+      listOfSites.forEach((item: Site) => {
+        subscriptionSites.set(item.id, item);
+      });
+
+      ext.logicAppSitesMap.set(subContext.subscriptionId, subscriptionSites);
+      site = subscriptionSites.get(resource.id);
+    } else {
+      site = ext.logicAppSitesMap.get(subContext.subscriptionId).get(resource.id);
+    }
+    return site;
+  }
 }

apps/vs-code-designer/src/extensionVariables.ts

@@ -4,6 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import type { AzureAccountTreeItemWithProjects } from './app/tree/AzureAccountTreeItemWithProjects';
 import { func } from './constants';
+import type { Site } from '@azure/arm-appservice';
 import type { IAzExtOutputChannel } from '@microsoft/vscode-azext-utils';
 import type { AzureHostExtensionApi } from '@microsoft/vscode-azext-utils/hostapi';
 import type * as cp from 'child_process';
@@ -26,6 +27,7 @@ export namespace ext {
   export let azureAccountTreeItem: AzureAccountTreeItemWithProjects;
   export const treeViewName = 'azLogicApps';
   export let deploymentFolderPath: string;
+  export const logicAppSitesMap: Map<string, Map<string, Site>> = new Map();
 
   // Resource group API
   export let rgApi: AzureHostExtensionApi;