@finos/legend-query-builder 4.3.9 → 4.5.0

package/src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.ts

@@ -20,13 +20,14 @@ import {
   deleteEntry,
   hashArray,
   type Hashable,
+  UnsupportedOperationError,
 } from '@finos/legend-shared';
 import type { TreeNodeData, TreeData } from '@finos/legend-art';
 import {
   type AbstractProperty,
   type GraphFetchTree,
   type Type,
-  type RootGraphFetchTree,
+  RootGraphFetchTree,
   PropertyExplicitReference,
   Class,
   PropertyGraphFetchTree,
@@ -51,7 +52,7 @@ export class QueryBuilderGraphFetchTreeNodeData
 {
   readonly id: string;
   readonly label: string;
-  readonly tree: PropertyGraphFetchTree;
+  readonly tree: PropertyGraphFetchTree | RootGraphFetchTree;
   readonly parentId?: string | undefined;
   isSelected?: boolean | undefined;
   isOpen?: boolean | undefined;
@@ -61,7 +62,7 @@ export class QueryBuilderGraphFetchTreeNodeData
     id: string,
     label: string,
     parentId: string | undefined,
-    graphFetchTreeNode: PropertyGraphFetchTree,
+    graphFetchTreeNode: PropertyGraphFetchTree | RootGraphFetchTree,
   ) {
     makeObservable(this, {
       hashCode: computed,
@@ -76,7 +77,16 @@ export class QueryBuilderGraphFetchTreeNodeData
   }
 
   get type(): Type {
-    return this.tree.property.value.genericType.value.rawType;
+    if (this.tree instanceof PropertyGraphFetchTree) {
+      return this.tree.property.value.genericType.value.rawType;
+    }
+    if (this.tree instanceof RootGraphFetchTree) {
+      return this.tree.class.value;
+    }
+    throw new UnsupportedOperationError(
+      `Can't get type of Graph Fetch Tree`,
+      this.tree,
+    );
   }
 
   get hashCode(): string {
@@ -105,6 +115,14 @@ export const generateGraphFetchTreeNodeID = (
     subType ? `${TYPE_CAST_TOKEN}${subType.path}` : ''
   }`;
 
+export const generateRootGraphFetchTreeNodeID = (
+  parentNodeId: string | undefined,
+  classValue: string | undefined,
+): string =>
+  `${parentNodeId ? `${parentNodeId}.` : ''}${
+    classValue ? `${TYPE_CAST_TOKEN}${classValue}` : ''
+  }`;
+
 const buildGraphFetchSubTree = (
   tree: GraphFetchTree,
   parentNode: QueryBuilderGraphFetchTreeNodeData | undefined,
@@ -132,19 +150,65 @@ const buildGraphFetchSubTree = (
   return node;
 };
 
+const buildRootGraphFetchSubTree = (
+  tree: RootGraphFetchTree,
+  parentNode: QueryBuilderGraphFetchTreeNodeData | undefined,
+  nodes: Map<string, QueryBuilderGraphFetchTreeNodeData>,
+): QueryBuilderGraphFetchTreeNodeData => {
+  const parentNodeId = parentNode?.id;
+  const node = new QueryBuilderGraphFetchTreeNodeData(
+    generateRootGraphFetchTreeNodeID(
+      parentNodeId,
+      tree.class.valueForSerialization ?? '',
+    ),
+    tree.class.value.name,
+    parentNodeId,
+    tree,
+  );
+  tree.subTrees.forEach((subTree) => {
+    const subTreeNode = buildGraphFetchSubTree(subTree, node, nodes);
+    addUniqueEntry(node.childrenIds, subTreeNode.id);
+    nodes.set(subTreeNode.id, subTreeNode);
+  });
+  return node;
+};
+
 /**
  * Build graph fetch tree data from an existing graph fetch tree root
  */
 export const buildGraphFetchTreeData = (
   tree: RootGraphFetchTree,
+  displayRoot: boolean = false,
 ): QueryBuilderGraphFetchTreeData => {
   const rootIds: string[] = [];
   const nodes = new Map<string, QueryBuilderGraphFetchTreeNodeData>();
-  tree.subTrees.forEach((subTree) => {
-    const subTreeNode = buildGraphFetchSubTree(subTree, undefined, nodes);
-    addUniqueEntry(rootIds, subTreeNode.id);
-    nodes.set(subTreeNode.id, subTreeNode);
-  });
+  if (displayRoot) {
+    const rootNode = buildRootGraphFetchSubTree(tree, undefined, nodes);
+    addUniqueEntry(rootIds, rootNode.id);
+    nodes.set(rootNode.id, rootNode);
+  } else {
+    tree.subTrees.forEach((subTree) => {
+      const subTreeNode = buildGraphFetchSubTree(subTree, undefined, nodes);
+      addUniqueEntry(rootIds, subTreeNode.id);
+      nodes.set(subTreeNode.id, subTreeNode);
+    });
+  }
+
+  return { rootIds, nodes, tree };
+};
+
+export const buildPropertyGraphFetchTreeData = (
+  propertyTree: PropertyGraphFetchTree,
+): QueryBuilderGraphFetchTreeData => {
+  const rootIds: string[] = [];
+  const nodes = new Map<string, QueryBuilderGraphFetchTreeNodeData>();
+  const propertyNode = buildGraphFetchSubTree(propertyTree, undefined, nodes);
+  addUniqueEntry(rootIds, propertyNode.id);
+  nodes.set(propertyNode.id, propertyNode);
+  const tree = new RootGraphFetchTree(
+    PackageableElementExplicitReference.create(new Class('root')),
+  );
+  tree.subTrees.push(propertyTree);
   return { rootIds, nodes, tree };
 };
 

package/src/stores/workflows/FunctionQueryBuilderState.ts

@@ -20,6 +20,7 @@ import type {
   GraphManagerState,
 } from '@finos/legend-graph';
 import { ClassQueryBuilderState } from './ClassQueryBuilderState.js';
+import type { QueryBuilderConfig } from '../../graph-manager/QueryBuilderConfig.js';
 
 // Note: We may want to move it to extend QueryBuilderState directly
 // but for now we will use the same setup as class as class, mapping, runtime are editable
@@ -30,8 +31,9 @@ export class FunctionQueryBuilderState extends ClassQueryBuilderState {
     applicationStore: GenericLegendApplicationStore,
     graphManagerState: GraphManagerState,
     functionElemenet: ConcreteFunctionDefinition,
+    config: QueryBuilderConfig | undefined,
   ) {
-    super(applicationStore, graphManagerState);
+    super(applicationStore, graphManagerState, config);
     this.functionElement = functionElemenet;
     this.showParametersPanel = true;
   }

package/src/stores/workflows/MappingQueryBuilderState.ts

@@ -27,6 +27,7 @@ import {
 import { getNullableFirstEntry } from '@finos/legend-shared';
 import { renderMappingQueryBuilderSetupPanelContent } from '../../components/workflows/MappingQueryBuilder.js';
 import { QueryBuilderState } from '../QueryBuilderState.js';
+import type { QueryBuilderConfig } from '../../graph-manager/QueryBuilderConfig.js';
 
 export class MappingQueryBuilderState extends QueryBuilderState {
   readonly onMappingChange?: ((val: Mapping) => void) | undefined;
@@ -40,8 +41,9 @@ export class MappingQueryBuilderState extends QueryBuilderState {
     graphManagerState: GraphManagerState,
     onMappingChange?: ((val: Mapping) => void) | undefined,
     onRuntimeChange?: ((val: Runtime) => void) | undefined,
+    config?: QueryBuilderConfig | undefined,
   ) {
-    super(applicationStore, graphManagerState);
+    super(applicationStore, graphManagerState, config);
 
     this.onMappingChange = onMappingChange;
     this.onRuntimeChange = onRuntimeChange;

package/src/stores/workflows/ServiceQueryBuilderState.ts

@@ -33,6 +33,7 @@ import {
 import { action, makeObservable, observable } from 'mobx';
 import { renderServiceQueryBuilderSetupPanelContent } from '../../components/workflows/ServiceQueryBuilder.js';
 import { QueryBuilderState } from '../QueryBuilderState.js';
+import type { QueryBuilderConfig } from '../../graph-manager/QueryBuilderConfig.js';
 
 export type ServiceExecutionContext = {
   key: string;
@@ -64,8 +65,9 @@ export class ServiceQueryBuilderState extends QueryBuilderState {
     onExecutionContextChange?:
       | ((val: ServiceExecutionContext) => void)
       | undefined,
+    config?: QueryBuilderConfig | undefined,
   ) {
-    super(applicationStore, graphManagerState);
+    super(applicationStore, graphManagerState, config);
 
     makeObservable(this, {
       selectedExecutionContext: observable,

package/tsconfig.json

@@ -44,6 +44,7 @@
     "./src/components/QueryBuilder_LegendApplicationPlugin.ts",
     "./src/components/QuerySetupUtils.ts",
     "./src/graph/QueryBuilderMetaModelConst.ts",
+    "./src/graph-manager/QueryBuilderConfig.ts",
     "./src/graph-manager/QueryBuilderConst.ts",
     "./src/graph-manager/QueryBuilder_GraphManagerPreset.ts",
     "./src/graph-manager/QueryBuilder_PureGraphManagerPlugin.ts",
@@ -231,11 +232,18 @@
     "./src/components/execution-plan/DataTypeResultTypeViewer.tsx",
     "./src/components/execution-plan/ExecutionPlanViewer.tsx",
     "./src/components/execution-plan/FunctionParametersValidationNodeViewer.tsx",
+    "./src/components/execution-plan/ImplementationViewer.tsx",
+    "./src/components/execution-plan/PartialClassResultViewer.tsx",
+    "./src/components/execution-plan/RelationalClassQueryTempTableGraphFetchExecutionNodeViewer.tsx",
+    "./src/components/execution-plan/RelationalCrossRootQueryTempTableGraphFetchExecutionNodeViewer.tsx",
+    "./src/components/execution-plan/RelationalRootQueryTempTableGraphFetchExecutionNodeViewer.tsx",
     "./src/components/execution-plan/RelationalTDSInstantiationExecutionNodeViewer.tsx",
     "./src/components/execution-plan/ResultTypeViewer.tsx",
     "./src/components/execution-plan/SQLExecutionNodeViewer.tsx",
     "./src/components/execution-plan/SequenceExecutionNodeViewer.tsx",
+    "./src/components/execution-plan/StoreMappingGlobalGraphFetchExecutionNodeViewer.tsx",
     "./src/components/execution-plan/TDSResultTypeViewer.tsx",
+    "./src/components/execution-plan/TempTableStrategyViewer.tsx",
     "./src/components/explorer/QueryBuilderExplorerPanel.tsx",
     "./src/components/explorer/QueryBuilderFunctionsExplorerPanel.tsx",
     "./src/components/explorer/QueryBuilderMilestoningEditor.tsx",