@finos/legend-application-query 13.7.2 → 13.7.4

package/src/stores/data-space/DataSpaceTemplateQueryCreatorStore.ts

@@ -21,12 +21,11 @@ import {
   QueryProjectCoordinates,
   extractElementNameFromPath,
 } from '@finos/legend-graph';
-import { type DepotServerClient } from '@finos/legend-server-depot';
 import {
-  IllegalStateError,
-  uuid,
-  type GeneratorFn,
-} from '@finos/legend-shared';
+  type DepotServerClient,
+  StoreProjectData,
+} from '@finos/legend-server-depot';
+import { IllegalStateError, uuid } from '@finos/legend-shared';
 import {
   type QueryBuilderState,
   QueryBuilderDataBrowserWorkflow,
@@ -36,16 +35,18 @@ import {
   parseGACoordinates,
 } from '@finos/legend-storage';
 import {
-  type QueryPersistConfiguration,
   QueryBuilderActionConfig_QueryApplication,
   QueryEditorStore,
+  type QueryPersistConfiguration,
 } from '../QueryEditorStore.js';
 import type { LegendQueryApplicationStore } from '../LegendQueryBaseStore.js';
 import {
-  DataSpacePackageableElementExecutable,
+  DSL_DataSpace_getGraphManagerExtension,
   getDataSpace,
+  retrieveAnalyticsResultCache,
   getExecutionContextFromDataspaceExecutable,
   getQueryFromDataspaceExecutable,
+  DataSpacePackageableElementExecutable,
 } from '@finos/legend-extension-dsl-data-space/graph';
 import {
   type DataSpaceInfo,
@@ -89,84 +90,62 @@ export class DataSpaceTemplateQueryCreatorStore extends QueryEditorStore {
     };
   }
 
-  override *buildGraph(): GeneratorFn<void> {
-    // do nothing
-  }
-
   async initializeQueryBuilderState(): Promise<QueryBuilderState> {
-    const { dataSpaceAnalysisResult, isLightGraphEnabled } =
-      await this.buildGraphAndDataspaceAnalyticsResult(
-        this.groupId,
-        this.artifactId,
-        this.versionId,
-        undefined,
-        this.dataSpacePath,
-        this.templateQueryId,
-      );
     const dataSpace = getDataSpace(
       this.dataSpacePath,
       this.graphManagerState.graph,
     );
-    let query;
-    let executionContext;
-    if (dataSpace.executables && dataSpace.executables.length > 0) {
-      let template = dataSpace.executables.find(
-        (ex) => ex.id === this.templateQueryId,
+    let template = dataSpace.executables?.find(
+      (executable) => executable.id === this.templateQueryId,
+    );
+    if (!template) {
+      template = dataSpace.executables?.find(
+        (executable) =>
+          executable instanceof DataSpacePackageableElementExecutable &&
+          executable.executable.value.path === this.templateQueryId,
       );
-      if (!template) {
-        template = dataSpace.executables.find(
-          (executable) =>
-            executable instanceof DataSpacePackageableElementExecutable &&
-            executable.executable.value.path === this.templateQueryId,
-        );
-      }
-      if (!template) {
-        throw new IllegalStateError(
-          `Can't find template query with id '${this.templateQueryId}'`,
-        );
-      }
-      executionContext = getExecutionContextFromDataspaceExecutable(
-        dataSpace,
-        template,
+    }
+    if (!template) {
+      throw new IllegalStateError(
+        `Can't find template query with id '${this.templateQueryId}'`,
       );
-      query = getQueryFromDataspaceExecutable(template, this.graphManagerState);
-      this.templateQueryTitle = template.title;
-    } else {
-      let template = dataSpaceAnalysisResult?.executables.find(
-        (executable) => executable.info?.id === this.templateQueryId,
+    }
+    const executionContext = getExecutionContextFromDataspaceExecutable(
+      dataSpace,
+      template,
+    );
+    if (!executionContext) {
+      throw new IllegalStateError(
+        `Can't find a correpsonding execution context`,
       );
-      if (!template) {
-        template = dataSpaceAnalysisResult?.executables.find(
-          (executable) => executable.executable === this.templateQueryId,
-        );
-      }
-      if (!template) {
-        throw new IllegalStateError(
-          `Can't find template query with id '${this.templateQueryId}'`,
-        );
-      }
-      executionContext =
-        template.info?.executionContextKey === undefined
-          ? dataSpace.defaultExecutionContext
-          : dataSpace.executionContexts.find(
-              (ex) => ex.name === template.info?.executionContextKey,
-            );
-      if (template.info) {
-        query = await this.graphManagerState.graphManager.pureCodeToLambda(
-          template.info.query,
-        );
-      }
-      this.templateQueryTitle = template.title;
     }
+    const query = getQueryFromDataspaceExecutable(
+      template,
+      this.graphManagerState,
+    );
     if (!query) {
       throw new IllegalStateError(
         `Can't fetch query from dataspace executable`,
       );
     }
-    if (!executionContext) {
-      throw new IllegalStateError(
-        `Can't find a correpsonding execution context`,
+    this.templateQueryTitle = template.title;
+    let dataSpaceAnalysisResult;
+    try {
+      const project = StoreProjectData.serialization.fromJson(
+        await this.depotServerClient.getProject(this.groupId, this.artifactId),
+      );
+      dataSpaceAnalysisResult = await DSL_DataSpace_getGraphManagerExtension(
+        this.graphManagerState.graphManager,
+      ).retrieveDataSpaceAnalysisFromCache(() =>
+        retrieveAnalyticsResultCache(
+          project,
+          this.versionId,
+          dataSpace.path,
+          this.depotServerClient,
+        ),
       );
+    } catch {
+      // do nothing
     }
     const sourceInfo = {
       groupId: this.groupId,
@@ -181,7 +160,6 @@ export class DataSpaceTemplateQueryCreatorStore extends QueryEditorStore {
       new QueryBuilderActionConfig_QueryApplication(this),
       dataSpace,
       executionContext,
-      isLightGraphEnabled,
       createDataSpaceDepoRepo(
         this,
         this.groupId,
@@ -189,7 +167,7 @@ export class DataSpaceTemplateQueryCreatorStore extends QueryEditorStore {
         this.versionId,
         undefined,
       ),
-      async (dataSpaceInfo: DataSpaceInfo) => {
+      (dataSpaceInfo: DataSpaceInfo) => {
         this.applicationStore.notificationService.notifyWarning(
           `Can't switch data space to visit current template query`,
         );
@@ -202,7 +180,7 @@ export class DataSpaceTemplateQueryCreatorStore extends QueryEditorStore {
       sourceInfo,
     );
     queryBuilderState.setExecutionContext(executionContext);
-    await queryBuilderState.propagateExecutionContextChange(true);
+    queryBuilderState.propagateExecutionContextChange(executionContext);
     queryBuilderState.initializeWithQuery(query);
     return queryBuilderState;
   }