@finos/legend-query-builder 0.0.1

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.js

@@ -0,0 +1,173 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { assertType, addUniqueEntry, deleteEntry } from '@finos/legend-shared';
+import { PropertyExplicitReference, Class, PropertyGraphFetchTree, PackageableElementExplicitReference, TYPE_CAST_TOKEN, } from '@finos/legend-graph';
+import { QueryBuilderExplorerTreePropertyNodeData, QueryBuilderExplorerTreeSubTypeNodeData, } from '../../explorer/QueryBuilderExplorerState.js';
+import { graphFetchTree_addSubTree, graphFetchTree_removeSubTree, } from '@finos/legend-application';
+export class QueryBuilderGraphFetchTreeNodeData {
+    isSelected;
+    isOpen;
+    id;
+    label;
+    childrenIds = [];
+    parentId;
+    tree;
+    constructor(id, label, parentId, graphFetchTreeNode) {
+        this.id = id;
+        this.label = label;
+        this.parentId = parentId;
+        this.tree = graphFetchTreeNode;
+        // NOTE: in this tree, every node is open
+        this.isOpen = true;
+    }
+    get type() {
+        return this.tree.property.value.genericType.value.rawType;
+    }
+}
+export const generateGraphFetchTreeNodeID = (property, parentNodeId, subType) => `${parentNodeId ? `${parentNodeId}.` : ''}${property.name}${subType ? `${TYPE_CAST_TOKEN}${subType.path}` : ''}`;
+const buildGraphFetchSubTree = (tree, parentNode, nodes) => {
+    assertType(tree, PropertyGraphFetchTree, 'Graph fetch sub-tree must be a property graph fetch tree');
+    const property = tree.property.value;
+    const subType = tree.subType?.value;
+    const parentNodeId = parentNode?.id;
+    const node = new QueryBuilderGraphFetchTreeNodeData(generateGraphFetchTreeNodeID(property, parentNodeId, subType), property.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) => {
+    const rootIds = [];
+    const nodes = new Map();
+    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 isGraphFetchTreeDataEmpty = (treeData) => treeData.tree.subTrees.length === 0;
+const getPrunableNodes = (treeData) => Array.from(treeData.nodes.values()).filter((node) => 
+// childless class nodes
+(node.tree instanceof PropertyGraphFetchTree &&
+    node.type instanceof Class &&
+    node.childrenIds.length === 0) ||
+    // orphan node
+    (node.tree instanceof PropertyGraphFetchTree &&
+        !(node.type instanceof Class) &&
+        node.parentId &&
+        !treeData.nodes.has(node.parentId)));
+const removeNode = (treeData, node) => {
+    const parentNode = node.parentId
+        ? treeData.nodes.get(node.parentId)
+        : undefined;
+    if (parentNode) {
+        deleteEntry(parentNode.childrenIds, node.id);
+        graphFetchTree_removeSubTree(parentNode.tree, node.tree);
+    }
+    else {
+        deleteEntry(treeData.rootIds, node.id);
+        graphFetchTree_removeSubTree(treeData.tree, node.tree);
+    }
+    treeData.nodes.delete(node.id);
+};
+const pruneTreeData = (treeData) => {
+    let prunableNodes = getPrunableNodes(treeData);
+    while (prunableNodes.length) {
+        prunableNodes.forEach((node) => {
+            removeNode(treeData, node);
+        });
+        prunableNodes = getPrunableNodes(treeData);
+    }
+};
+export const removeNodeRecursively = (treeData, node) => {
+    removeNode(treeData, node);
+    pruneTreeData(treeData);
+};
+export const addQueryBuilderPropertyNode = (treeData, explorerTreeData, node, queryBuilderState) => {
+    // traverse the property node all the way to the root and resolve the
+    // chain of property that leads to this property node
+    const propertyGraphFetchTrees = [
+        new PropertyGraphFetchTree(PropertyExplicitReference.create(node.property), undefined),
+    ];
+    let parentExplorerTreeNode = explorerTreeData.nodes.get(node.parentId);
+    while (parentExplorerTreeNode instanceof
+        QueryBuilderExplorerTreePropertyNodeData ||
+        parentExplorerTreeNode instanceof QueryBuilderExplorerTreeSubTypeNodeData) {
+        let subType = undefined;
+        let subtypeAssigned = false;
+        while (parentExplorerTreeNode instanceof QueryBuilderExplorerTreeSubTypeNodeData) {
+            if (!subtypeAssigned) {
+                subType = PackageableElementExplicitReference.create(parentExplorerTreeNode.subclass);
+                subtypeAssigned = true;
+            }
+            parentExplorerTreeNode = explorerTreeData.nodes.get(parentExplorerTreeNode.parentId);
+        }
+        if (parentExplorerTreeNode instanceof QueryBuilderExplorerTreePropertyNodeData) {
+            const propertyGraphFetchTree = new PropertyGraphFetchTree(PropertyExplicitReference.create(parentExplorerTreeNode.property), subType);
+            propertyGraphFetchTree.subTrees.push(propertyGraphFetchTrees[0]);
+            propertyGraphFetchTrees.unshift(propertyGraphFetchTree);
+            parentExplorerTreeNode = explorerTreeData.nodes.get(parentExplorerTreeNode.parentId);
+        }
+        else {
+            queryBuilderState.applicationStore.notifyError(`Can't cast the root class of graph fetch structure to its subtype`);
+            return;
+        }
+    }
+    // traverse the chain of property from the root class to find a node in the
+    // current graph fetch tree that matches any of this property, consider
+    // this the starting point
+    //
+    // If we found a match, use that as the starting point, otherwise, use the root
+    // of the graph fetch tree as the starting point
+    //
+    // NOTE: here we assume that we don't allow specifying duplicated nodes.
+    // perhaps we need to allow that behavior in the future so people can customize the
+    // shape of the object (e.g. same field but under different aliases)
+    let currentNodeId = undefined;
+    let parentNode = undefined;
+    let newSubTree;
+    for (const propertyGraphFetchTree of propertyGraphFetchTrees) {
+        currentNodeId = generateGraphFetchTreeNodeID(propertyGraphFetchTree.property.value, currentNodeId, propertyGraphFetchTree.subType?.value);
+        const existingGraphFetchNode = treeData.nodes.get(currentNodeId);
+        if (existingGraphFetchNode) {
+            parentNode = existingGraphFetchNode;
+        }
+        else {
+            newSubTree = propertyGraphFetchTree;
+            break;
+        }
+    }
+    // construct the query builder graph fetch subtree from the starting point
+    if (newSubTree) {
+        const childNode = buildGraphFetchSubTree(newSubTree, parentNode, treeData.nodes);
+        treeData.nodes.set(childNode.id, childNode);
+        if (parentNode) {
+            addUniqueEntry(parentNode.childrenIds, childNode.id);
+            graphFetchTree_addSubTree(parentNode.tree, childNode.tree);
+        }
+        else {
+            addUniqueEntry(treeData.rootIds, childNode.id);
+            graphFetchTree_addSubTree(treeData.tree, childNode.tree);
+        }
+    }
+};
+//# sourceMappingURL=QueryBuilderGraphFetchTreeUtil.js.map

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchTreeUtil.js","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAE/E,OAAO,EAKL,yBAAyB,EACzB,KAAK,EACL,sBAAsB,EACtB,mCAAmC,EACnC,eAAe,GAChB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,wCAAwC,EACxC,uCAAuC,GACxC,MAAM,6CAA6C,CAAC;AAErD,OAAO,EACL,yBAAyB,EACzB,4BAA4B,GAC7B,MAAM,2BAA2B,CAAC;AAEnC,MAAM,OAAO,kCAAkC;IAC7C,UAAU,CAAuB;IACjC,MAAM,CAAuB;IAC7B,EAAE,CAAS;IACX,KAAK,CAAS;IACd,WAAW,GAAa,EAAE,CAAC;IAC3B,QAAQ,CAAsB;IAC9B,IAAI,CAAyB;IAE7B,YACE,EAAU,EACV,KAAa,EACb,QAA4B,EAC5B,kBAA0C;QAE1C,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC;QAC/B,yCAAyC;QACzC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;IAC5D,CAAC;CACF;AAOD,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAC1C,QAA0B,EAC1B,YAAgC,EAChC,OAAyB,EACjB,EAAE,CACV,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,GACvD,OAAO,CAAC,CAAC,CAAC,GAAG,eAAe,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAClD,EAAE,CAAC;AAEL,MAAM,sBAAsB,GAAG,CAC7B,IAAoB,EACpB,UAA0D,EAC1D,KAAsD,EAClB,EAAE;IACtC,UAAU,CACR,IAAI,EACJ,sBAAsB,EACtB,0DAA0D,CAC3D,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC;IACpC,MAAM,YAAY,GAAG,UAAU,EAAE,EAAE,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,kCAAkC,CACjD,4BAA4B,CAAC,QAAQ,EAAE,YAAY,EAAE,OAAO,CAAC,EAC7D,QAAQ,CAAC,IAAI,EACb,YAAY,EACZ,IAAI,CACL,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAChC,MAAM,WAAW,GAAG,sBAAsB,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACjE,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QACjD,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,CACrC,IAAwB,EACQ,EAAE;IAClC,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAI,GAAG,EAA8C,CAAC;IACpE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAChC,MAAM,WAAW,GAAG,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACtE,cAAc,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QACxC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AAClC,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,QAAwC,EAC/B,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC;AAElD,MAAM,gBAAgB,GAAG,CACvB,QAAwC,EACF,EAAE,CACxC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACxC,CAAC,IAAI,EAAE,EAAE;AACP,wBAAwB;AACxB,CAAC,IAAI,CAAC,IAAI,YAAY,sBAAsB;IAC1C,IAAI,CAAC,IAAI,YAAY,KAAK;IAC1B,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,CAAC;IAChC,cAAc;IACd,CAAC,IAAI,CAAC,IAAI,YAAY,sBAAsB;QAC1C,CAAC,CAAC,IAAI,CAAC,IAAI,YAAY,KAAK,CAAC;QAC7B,IAAI,CAAC,QAAQ;QACb,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CACxC,CAAC;AAEJ,MAAM,UAAU,GAAG,CACjB,QAAwC,EACxC,IAAwC,EAClC,EAAE;IACR,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ;QAC9B,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC;QACnC,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,UAAU,EAAE;QACd,WAAW,CAAC,UAAU,CAAC,WAAW,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7C,4BAA4B,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;KAC1D;SAAM;QACL,WAAW,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,4BAA4B,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;KACxD;IACD,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACjC,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CAAC,QAAwC,EAAQ,EAAE;IACvE,IAAI,aAAa,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC/C,OAAO,aAAa,CAAC,MAAM,EAAE;QAC3B,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YAC7B,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,aAAa,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;KAC5C;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,QAAwC,EACxC,IAAwC,EAClC,EAAE;IACR,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC3B,aAAa,CAAC,QAAQ,CAAC,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,QAAwC,EACxC,gBAA4D,EAC5D,IAA8C,EAC9C,iBAAoC,EAC9B,EAAE;IACR,qEAAqE;IACrE,qDAAqD;IACrD,MAAM,uBAAuB,GAA6B;QACxD,IAAI,sBAAsB,CACxB,yBAAyB,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAC/C,SAAS,CACV;KACF,CAAC;IACF,IAAI,sBAAsB,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvE,OACE,sBAAsB;QACpB,wCAAwC;QAC1C,sBAAsB,YAAY,uCAAuC,EACzE;QACA,IAAI,OAAO,GAAG,SAAS,CAAC;QACxB,IAAI,eAAe,GAAG,KAAK,CAAC;QAC5B,OACE,sBAAsB,YAAY,uCAAuC,EACzE;YACA,IAAI,CAAC,eAAe,EAAE;gBACpB,OAAO,GAAG,mCAAmC,CAAC,MAAM,CAClD,sBAAsB,CAAC,QAAQ,CAChC,CAAC;gBACF,eAAe,GAAG,IAAI,CAAC;aACxB;YACD,sBAAsB,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CACjD,sBAAsB,CAAC,QAAQ,CAChC,CAAC;SACH;QACD,IACE,sBAAsB,YAAY,wCAAwC,EAC1E;YACA,MAAM,sBAAsB,GAAG,IAAI,sBAAsB,CACvD,yBAAyB,CAAC,MAAM,CAAC,sBAAsB,CAAC,QAAQ,CAAC,EACjE,OAAO,CACR,CAAC;YACF,sBAAsB,CAAC,QAAQ,CAAC,IAAI,CAClC,uBAAuB,CAAC,CAAC,CAA2B,CACrD,CAAC;YACF,uBAAuB,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;YACxD,sBAAsB,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CACjD,sBAAsB,CAAC,QAAQ,CAChC,CAAC;SACH;aAAM;YACL,iBAAiB,CAAC,gBAAgB,CAAC,WAAW,CAC5C,mEAAmE,CACpE,CAAC;YACF,OAAO;SACR;KACF;IAED,2EAA2E;IAC3E,uEAAuE;IACvE,0BAA0B;IAC1B,EAAE;IACF,+EAA+E;IAC/E,gDAAgD;IAChD,EAAE;IACF,wEAAwE;IACxE,mFAAmF;IACnF,oEAAoE;IACpE,IAAI,aAAa,GAAuB,SAAS,CAAC;IAClD,IAAI,UAAU,GAAmD,SAAS,CAAC;IAC3E,IAAI,UAA8C,CAAC;IACnD,KAAK,MAAM,sBAAsB,IAAI,uBAAuB,EAAE;QAC5D,aAAa,GAAG,4BAA4B,CAC1C,sBAAsB,CAAC,QAAQ,CAAC,KAAK,EACrC,aAAa,EACb,sBAAsB,CAAC,OAAO,EAAE,KAAK,CACtC,CAAC;QACF,MAAM,sBAAsB,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACjE,IAAI,sBAAsB,EAAE;YAC1B,UAAU,GAAG,sBAAsB,CAAC;SACrC;aAAM;YACL,UAAU,GAAG,sBAAsB,CAAC;YACpC,MAAM;SACP;KACF;IAED,0EAA0E;IAC1E,IAAI,UAAU,EAAE;QACd,MAAM,SAAS,GAAG,sBAAsB,CACtC,UAAU,EACV,UAAU,EACV,QAAQ,CAAC,KAAK,CACf,CAAC;QACF,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC5C,IAAI,UAAU,EAAE;YACd,cAAc,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YACrD,yBAAyB,CAAC,UAAU,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;SAC5D;aAAM;YACL,cAAc,CAAC,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YAC/C,yBAAyB,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;SAC1D;KACF;AACH,CAAC,CAAC"}

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.d.ts

@@ -0,0 +1,26 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { type LambdaFunction } from '@finos/legend-graph';
+import type { QueryBuilderGraphFetchTreeState } from './QueryBuilderGraphFetchTreeState.js';
+export declare const appendGraphFetch: (graphFetchTreeState: QueryBuilderGraphFetchTreeState, lambdaFunction: LambdaFunction, options?: {
+    /**
+     * Set queryBuilderState to `true` when we construct query for execution within the app.
+     * queryBuilderState will make the lambda function building process overrides several query values, such as the row limit.
+     */
+    isBuildingExecutionQuery?: boolean | undefined;
+    keepSourceInformation?: boolean | undefined;
+}) => void;
+//# sourceMappingURL=QueryBuilderGraphFetchValueSpecificationBuilder.d.ts.map

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchValueSpecificationBuilder.d.ts","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAUL,KAAK,cAAc,EAEpB,MAAM,qBAAqB,CAAC;AAI7B,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,sCAAsC,CAAC;AA8D5F,eAAO,MAAM,gBAAgB,wBACN,+BAA+B,kBACpC,cAAc,YACpB;IACR;;;OAGG;IACH,wBAAwB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC/C,qBAAqB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAC7C,KACA,IAgDF,CAAC"}

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.js

@@ -0,0 +1,72 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { extractElementNameFromPath, GenericType, GenericTypeExplicitReference, matchFunctionName, PrimitiveInstanceValue, PRIMITIVE_TYPE, SimpleFunctionExpression, TYPICAL_MULTIPLICITY_TYPE, RootGraphFetchTreeInstanceValue, } from '@finos/legend-graph';
+import { guaranteeNonNullable, guaranteeType } from '@finos/legend-shared';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { isGraphFetchTreeDataEmpty } from './QueryBuilderGraphFetchTreeUtil.js';
+const appendTakeLimit = (queryBuilderState, lambda, previewLimit) => {
+    if (!previewLimit) {
+        return lambda;
+    }
+    const multiplicityOne = queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+    if (lambda.expressionSequence.length === 1) {
+        const func = lambda.expressionSequence[0];
+        if (func instanceof SimpleFunctionExpression) {
+            if (matchFunctionName(func.functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.SERIALIZE)) {
+                const limit = new PrimitiveInstanceValue(GenericTypeExplicitReference.create(new GenericType(queryBuilderState.graphManagerState.graph.getPrimitiveType(PRIMITIVE_TYPE.INTEGER))), multiplicityOne);
+                limit.values = [previewLimit];
+                const takeFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TAKE), multiplicityOne);
+                // NOTE: `take()` does not work on `graphFetch()` or `serialize()` so we need to
+                // put it next to `all()`
+                const serializeFunction = func;
+                const graphFetchFunc = guaranteeType(serializeFunction.parametersValues[0], SimpleFunctionExpression);
+                const getAllFunc = graphFetchFunc
+                    .parametersValues[0];
+                takeFunction.parametersValues[0] = getAllFunc;
+                takeFunction.parametersValues[1] = limit;
+                graphFetchFunc.parametersValues = [
+                    takeFunction,
+                    graphFetchFunc.parametersValues[1],
+                ];
+                return lambda;
+            }
+        }
+    }
+    return lambda;
+};
+export const appendGraphFetch = (graphFetchTreeState, lambdaFunction, options) => {
+    const queryBuilderState = graphFetchTreeState.queryBuilderState;
+    const precedingExpression = guaranteeNonNullable(lambdaFunction.expressionSequence[0], `Can't build graph-fetch tree expression: preceding expression is not defined`);
+    const multiplicityOne = queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+    // build graph-fetch tree
+    if (graphFetchTreeState.treeData &&
+        !isGraphFetchTreeDataEmpty(graphFetchTreeState.treeData)) {
+        const graphFetchInstance = new RootGraphFetchTreeInstanceValue(multiplicityOne);
+        graphFetchInstance.values = [graphFetchTreeState.treeData.tree];
+        const serializeFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.SERIALIZE), multiplicityOne);
+        const graphFetchFunc = new SimpleFunctionExpression(graphFetchTreeState.isChecked
+            ? extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.GRAPH_FETCH_CHECKED)
+            : extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.GRAPH_FETCH), multiplicityOne);
+        graphFetchFunc.parametersValues = [precedingExpression, graphFetchInstance];
+        serializeFunction.parametersValues = [graphFetchFunc, graphFetchInstance];
+        lambdaFunction.expressionSequence[0] = serializeFunction;
+    }
+    // build result set modifier: i.e. preview limit
+    if (options?.isBuildingExecutionQuery) {
+        appendTakeLimit(queryBuilderState, lambdaFunction, queryBuilderState.resultState.previewLimit);
+    }
+};
+//# sourceMappingURL=QueryBuilderGraphFetchValueSpecificationBuilder.js.map

package/lib/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderGraphFetchValueSpecificationBuilder.js","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,0BAA0B,EAC1B,WAAW,EACX,4BAA4B,EAC5B,iBAAiB,EACjB,sBAAsB,EACtB,cAAc,EACd,wBAAwB,EACxB,yBAAyB,EAGzB,+BAA+B,GAChC,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC3E,OAAO,EAAE,iCAAiC,EAAE,MAAM,yDAAyD,CAAC;AAG5G,OAAO,EAAE,yBAAyB,EAAE,MAAM,qCAAqC,CAAC;AAEhF,MAAM,eAAe,GAAG,CACtB,iBAAoC,EACpC,MAAsB,EACtB,YAAiC,EACjB,EAAE;IAClB,IAAI,CAAC,YAAY,EAAE;QACjB,OAAO,MAAM,CAAC;KACf;IACD,MAAM,eAAe,GACnB,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,sBAAsB,CAC9D,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IACJ,IAAI,MAAM,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAI,IAAI,YAAY,wBAAwB,EAAE;YAC5C,IACE,iBAAiB,CACf,IAAI,CAAC,YAAY,EACjB,iCAAiC,CAAC,SAAS,CAC5C,EACD;gBACA,MAAM,KAAK,GAAG,IAAI,sBAAsB,CACtC,4BAA4B,CAAC,MAAM,CACjC,IAAI,WAAW,CACb,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,gBAAgB,CACxD,cAAc,CAAC,OAAO,CACvB,CACF,CACF,EACD,eAAe,CAChB,CAAC;gBACF,KAAK,CAAC,MAAM,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC9B,MAAM,YAAY,GAAG,IAAI,wBAAwB,CAC/C,0BAA0B,CAAC,iCAAiC,CAAC,IAAI,CAAC,EAClE,eAAe,CAChB,CAAC;gBAEF,gFAAgF;gBAChF,yBAAyB;gBACzB,MAAM,iBAAiB,GAAG,IAAI,CAAC;gBAC/B,MAAM,cAAc,GAAG,aAAa,CAClC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC,CAAC,EACrC,wBAAwB,CACzB,CAAC;gBACF,MAAM,UAAU,GAAG,cAAc;qBAC9B,gBAAgB,CAAC,CAAC,CAAuB,CAAC;gBAC7C,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC;gBAC9C,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;gBACzC,cAAc,CAAC,gBAAgB,GAAG;oBAChC,YAAY;oBACZ,cAAc,CAAC,gBAAgB,CAAC,CAAC,CAAuB;iBACzD,CAAC;gBACF,OAAO,MAAM,CAAC;aACf;SACF;KACF;IACD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,mBAAoD,EACpD,cAA8B,EAC9B,OAOC,EACK,EAAE;IACR,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,iBAAiB,CAAC;IAChE,MAAM,mBAAmB,GAAG,oBAAoB,CAC9C,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,EACpC,8EAA8E,CAC/E,CAAC;IACF,MAAM,eAAe,GACnB,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,sBAAsB,CAC9D,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IAEJ,yBAAyB;IACzB,IACE,mBAAmB,CAAC,QAAQ;QAC5B,CAAC,yBAAyB,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EACxD;QACA,MAAM,kBAAkB,GAAG,IAAI,+BAA+B,CAC5D,eAAe,CAChB,CAAC;QACF,kBAAkB,CAAC,MAAM,GAAG,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,iBAAiB,GAAG,IAAI,wBAAwB,CACpD,0BAA0B,CAAC,iCAAiC,CAAC,SAAS,CAAC,EACvE,eAAe,CAChB,CAAC;QACF,MAAM,cAAc,GAAG,IAAI,wBAAwB,CACjD,mBAAmB,CAAC,SAAS;YAC3B,CAAC,CAAC,0BAA0B,CACxB,iCAAiC,CAAC,mBAAmB,CACtD;YACH,CAAC,CAAC,0BAA0B,CACxB,iCAAiC,CAAC,WAAW,CAC9C,EACL,eAAe,CAChB,CAAC;QAEF,cAAc,CAAC,gBAAgB,GAAG,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,CAAC;QAC5E,iBAAiB,CAAC,gBAAgB,GAAG,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;QAC1E,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC;KAC1D;IAED,gDAAgD;IAChD,IAAI,OAAO,EAAE,wBAAwB,EAAE;QACrC,eAAe,CACb,iBAAiB,EACjB,cAAc,EACd,iBAAiB,CAAC,WAAW,CAAC,YAAY,CAC3C,CAAC;KACH;AACH,CAAC,CAAC"}

package/lib/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.d.ts

@@ -0,0 +1,73 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { type GeneratorFn } from '@finos/legend-shared';
+import { type QueryBuilderExplorerTreePropertyNodeData } from '../../explorer/QueryBuilderExplorerState.js';
+import { QueryBuilderPropertyExpressionState } from '../../QueryBuilderPropertyEditorState.js';
+import type { QueryBuilderState } from '../../QueryBuilderState.js';
+import { type AbstractPropertyExpression, type Type, RawLambda } from '@finos/legend-graph';
+import { LambdaEditorState } from '@finos/legend-application';
+import type { QueryBuilderProjectionState } from './QueryBuilderProjectionState.js';
+export declare const QUERY_BUILDER_PROJECTION_COLUMN_DND_TYPE = "PROJECTION_COLUMN";
+export interface QueryBuilderProjectionColumnDragSource {
+    columnState: QueryBuilderProjectionColumnState;
+}
+export declare abstract class QueryBuilderProjectionColumnState {
+    readonly uuid: string;
+    projectionState: QueryBuilderProjectionState;
+    columnName: string;
+    constructor(projectionState: QueryBuilderProjectionState, columnName: string);
+    setColumnName(val: string): void;
+    abstract getReturnType(): Type | undefined;
+}
+export declare class QueryBuilderSimpleProjectionColumnState extends QueryBuilderProjectionColumnState {
+    lambdaParameterName: string;
+    propertyExpressionState: QueryBuilderPropertyExpressionState;
+    constructor(projectionState: QueryBuilderProjectionState, propertyExpression: AbstractPropertyExpression, humanizePropertyName: boolean);
+    setLambdaParameterName(val: string): void;
+    changeProperty(node: QueryBuilderExplorerTreePropertyNodeData, humanizePropertyName: boolean): void;
+    getReturnType(): Type | undefined;
+}
+declare class QueryBuilderDerivationProjectionLambdaState extends LambdaEditorState {
+    queryBuilderState: QueryBuilderState;
+    derivationProjectionColumnState: QueryBuilderDerivationProjectionColumnState;
+    /**
+     * This is used to store the JSON string when viewing the query in JSON mode
+     * TODO: consider moving this to another state if we need to simplify the logic of text-mode
+     */
+    readOnlylambdaJson: string;
+    constructor(queryBuilderState: QueryBuilderState, derivationProjectionColumnState: QueryBuilderDerivationProjectionColumnState);
+    get lambdaId(): string;
+    setLambdaJson(lambdaJson: string): void;
+    convertLambdaGrammarStringToObject(): GeneratorFn<void>;
+    convertLambdaObjectToGrammarString(pretty: boolean): GeneratorFn<void>;
+}
+export declare class QueryBuilderDerivationProjectionColumnState extends QueryBuilderProjectionColumnState {
+    derivationLambdaEditorState: QueryBuilderDerivationProjectionLambdaState;
+    lambda: RawLambda;
+    returnType: Type | undefined;
+    constructor(projectionState: QueryBuilderProjectionState, lambda: RawLambda);
+    setLambda(val: RawLambda): void;
+    setReturnType(val: Type | undefined): void;
+    /**
+     * Fetches lambda return type for derivation column.
+     * Throws error if unable to fetch type or if type is not primitive or an enumeration
+     * as expected by a projection column
+     */
+    fetchDerivationLambdaReturnType(): GeneratorFn<void>;
+    getReturnType(): Type | undefined;
+}
+export {};
+//# sourceMappingURL=QueryBuilderProjectionColumnState.d.ts.map

package/lib/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderProjectionColumnState.d.ts","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EACL,KAAK,WAAW,EAOjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,KAAK,wCAAwC,EAE9C,MAAM,6CAA6C,CAAC;AACrD,OAAO,EAEL,mCAAmC,EACpC,MAAM,0CAA0C,CAAC;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EACL,KAAK,0BAA0B,EAC/B,KAAK,IAAI,EAUT,SAAS,EAEV,MAAM,qBAAqB,CAAC;AAK7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAEpF,eAAO,MAAM,wCAAwC,sBAAsB,CAAC;AAE5E,MAAM,WAAW,sCAAsC;IACrD,WAAW,EAAE,iCAAiC,CAAC;CAChD;AAED,8BAAsB,iCAAiC;IACrD,QAAQ,CAAC,IAAI,SAAU;IACvB,eAAe,EAAE,2BAA2B,CAAC;IAC7C,UAAU,EAAE,MAAM,CAAC;gBAGjB,eAAe,EAAE,2BAA2B,EAC5C,UAAU,EAAE,MAAM;IAapB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAIhC,QAAQ,CAAC,aAAa,IAAI,IAAI,GAAG,SAAS;CAC3C;AAED,qBAAa,uCAAwC,SAAQ,iCAAiC;IAC5F,mBAAmB,EAAE,MAAM,CAAgC;IAC3D,uBAAuB,EAAE,mCAAmC,CAAC;gBAG3D,eAAe,EAAE,2BAA2B,EAC5C,kBAAkB,EAAE,0BAA0B,EAC9C,oBAAoB,EAAE,OAAO;IAqB/B,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAIzC,cAAc,CACZ,IAAI,EAAE,wCAAwC,EAC9C,oBAAoB,EAAE,OAAO,GAC5B,IAAI;IAkBE,aAAa,IAAI,IAAI,GAAG,SAAS;CAI3C;AAED,cAAM,2CAA4C,SAAQ,iBAAiB;IACzE,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,+BAA+B,EAAE,2CAA2C,CAAC;IAC7E;;;OAGG;IACH,kBAAkB,SAAM;gBAGtB,iBAAiB,EAAE,iBAAiB,EACpC,+BAA+B,EAAE,2CAA2C;IAO9E,IAAI,QAAQ,IAAI,MAAM,CAQrB;IAED,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAItC,kCAAkC,IAAI,WAAW,CAAC,IAAI,CAAC;IA2BvD,kCAAkC,CAAC,MAAM,EAAE,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC;CAmCxE;AAED,qBAAa,2CAA4C,SAAQ,iCAAiC;IAChG,2BAA2B,EAAE,2CAA2C,CAAC;IACzE,MAAM,EAAE,SAAS,CAAC;IAClB,UAAU,EAAE,IAAI,GAAG,SAAS,CAAC;gBAEjB,eAAe,EAAE,2BAA2B,EAAE,MAAM,EAAE,SAAS;IAkB3E,SAAS,CAAC,GAAG,EAAE,SAAS,GAAG,IAAI;IAI/B,aAAa,CAAC,GAAG,EAAE,IAAI,GAAG,SAAS,GAAG,IAAI;IAI1C;;;;OAIG;IACF,+BAA+B,IAAI,WAAW,CAAC,IAAI,CAAC;IA0C5C,aAAa,IAAI,IAAI,GAAG,SAAS;CAG3C"}

package/lib/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.js

@@ -0,0 +1,187 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { action, makeObservable, observable, flow } from 'mobx';
+import { LogEvent, assertErrorThrown, uuid, guaranteeNonNullable, assertTrue, assertNonEmptyString, } from '@finos/legend-shared';
+import { buildPropertyExpressionFromExplorerTreeNodeData, } from '../../explorer/QueryBuilderExplorerState.js';
+import { getPropertyChainName, QueryBuilderPropertyExpressionState, } from '../../QueryBuilderPropertyEditorState.js';
+import { PackageableElementExplicitReference, GRAPH_MANAGER_EVENT, PRIMITIVE_TYPE, buildSourceInformationSourceId, ParserError, TYPICAL_MULTIPLICITY_TYPE, RawVariableExpression, Enumeration, RawLambda, stub_RawLambda, } from '@finos/legend-graph';
+import { DEFAULT_LAMBDA_VARIABLE_NAME, QUERY_BUILDER_SOURCE_ID_LABEL, } from '../../QueryBuilderConfig.js';
+import { LambdaEditorState } from '@finos/legend-application';
+export const QUERY_BUILDER_PROJECTION_COLUMN_DND_TYPE = 'PROJECTION_COLUMN';
+export class QueryBuilderProjectionColumnState {
+    uuid = uuid();
+    projectionState;
+    columnName;
+    constructor(projectionState, columnName) {
+        makeObservable(this, {
+            uuid: false,
+            projectionState: false,
+            columnName: observable,
+            setColumnName: action,
+        });
+        this.projectionState = projectionState;
+        this.columnName = columnName;
+    }
+    setColumnName(val) {
+        this.columnName = val;
+    }
+}
+export class QueryBuilderSimpleProjectionColumnState extends QueryBuilderProjectionColumnState {
+    lambdaParameterName = DEFAULT_LAMBDA_VARIABLE_NAME;
+    propertyExpressionState;
+    constructor(projectionState, propertyExpression, humanizePropertyName) {
+        super(projectionState, '');
+        makeObservable(this, {
+            lambdaParameterName: observable,
+            propertyExpressionState: observable,
+            setLambdaParameterName: action,
+            changeProperty: action,
+        });
+        this.propertyExpressionState = new QueryBuilderPropertyExpressionState(projectionState.queryBuilderState, propertyExpression);
+        this.columnName = getPropertyChainName(this.propertyExpressionState.propertyExpression, humanizePropertyName);
+    }
+    setLambdaParameterName(val) {
+        this.lambdaParameterName = val;
+    }
+    changeProperty(node, humanizePropertyName) {
+        this.propertyExpressionState = new QueryBuilderPropertyExpressionState(this.projectionState.queryBuilderState, buildPropertyExpressionFromExplorerTreeNodeData(this.projectionState.queryBuilderState.explorerState
+            .nonNullableTreeData, node, this.projectionState.queryBuilderState.graphManagerState.graph, this.projectionState.queryBuilderState.explorerState.propertySearchState
+            .allMappedPropertyNodes));
+        this.columnName = getPropertyChainName(this.propertyExpressionState.propertyExpression, humanizePropertyName);
+    }
+    getReturnType() {
+        return this.propertyExpressionState.propertyExpression.func.genericType
+            .value.rawType;
+    }
+}
+class QueryBuilderDerivationProjectionLambdaState extends LambdaEditorState {
+    queryBuilderState;
+    derivationProjectionColumnState;
+    /**
+     * This is used to store the JSON string when viewing the query in JSON mode
+     * TODO: consider moving this to another state if we need to simplify the logic of text-mode
+     */
+    readOnlylambdaJson = '';
+    constructor(queryBuilderState, derivationProjectionColumnState) {
+        super('', '');
+        this.queryBuilderState = queryBuilderState;
+        this.derivationProjectionColumnState = derivationProjectionColumnState;
+    }
+    get lambdaId() {
+        return buildSourceInformationSourceId([
+            // TODO: to be reworked
+            // See https://github.com/finos/legend-studio/issues/1168
+            QUERY_BUILDER_SOURCE_ID_LABEL.QUERY_BUILDER,
+            QUERY_BUILDER_SOURCE_ID_LABEL.PROJECTION,
+            this.derivationProjectionColumnState.uuid,
+        ]);
+    }
+    setLambdaJson(lambdaJson) {
+        this.readOnlylambdaJson = lambdaJson;
+    }
+    *convertLambdaGrammarStringToObject() {
+        const emptyLambda = stub_RawLambda();
+        if (this.lambdaString) {
+            try {
+                const lambda = (yield this.queryBuilderState.graphManagerState.graphManager.pureCodeToLambda(this.fullLambdaString, this.lambdaId));
+                this.setParserError(undefined);
+                this.derivationProjectionColumnState.setLambda(lambda);
+            }
+            catch (error) {
+                assertErrorThrown(error);
+                if (error instanceof ParserError) {
+                    this.setParserError(error);
+                }
+                this.queryBuilderState.applicationStore.log.error(LogEvent.create(GRAPH_MANAGER_EVENT.PARSING_FAILURE), error);
+            }
+        }
+        else {
+            this.clearErrors();
+            this.derivationProjectionColumnState.setLambda(emptyLambda);
+        }
+    }
+    *convertLambdaObjectToGrammarString(pretty) {
+        if (this.derivationProjectionColumnState.lambda.body) {
+            try {
+                const lambdas = new Map();
+                lambdas.set(this.lambdaId, new RawLambda(this.derivationProjectionColumnState.lambda.parameters, this.derivationProjectionColumnState.lambda.body));
+                const isolatedLambdas = (yield this.queryBuilderState.graphManagerState.graphManager.lambdasToPureCode(lambdas, pretty));
+                const grammarText = isolatedLambdas.get(this.lambdaId);
+                this.setLambdaString(grammarText !== undefined
+                    ? this.extractLambdaString(grammarText)
+                    : '');
+                this.clearErrors();
+            }
+            catch (error) {
+                assertErrorThrown(error);
+                this.queryBuilderState.applicationStore.log.error(LogEvent.create(GRAPH_MANAGER_EVENT.PARSING_FAILURE), error);
+            }
+        }
+        else {
+            this.clearErrors();
+            this.setLambdaString('');
+        }
+    }
+}
+export class QueryBuilderDerivationProjectionColumnState extends QueryBuilderProjectionColumnState {
+    derivationLambdaEditorState;
+    lambda;
+    returnType;
+    constructor(projectionState, lambda) {
+        super(projectionState, '(derivation)');
+        makeObservable(this, {
+            lambda: observable,
+            returnType: observable,
+            setLambda: action,
+            fetchDerivationLambdaReturnType: flow,
+        });
+        this.derivationLambdaEditorState =
+            new QueryBuilderDerivationProjectionLambdaState(projectionState.queryBuilderState, this);
+        this.lambda = lambda;
+    }
+    setLambda(val) {
+        this.lambda = val;
+    }
+    setReturnType(val) {
+        this.returnType = val;
+    }
+    /**
+     * Fetches lambda return type for derivation column.
+     * Throws error if unable to fetch type or if type is not primitive or an enumeration
+     * as expected by a projection column
+     */
+    *fetchDerivationLambdaReturnType() {
+        assertTrue(Array.isArray(this.lambda.parameters));
+        const projectionParameter = this.lambda.parameters;
+        const graph = this.projectionState.queryBuilderState.graphManagerState.graph;
+        const multiplicityOne = graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+        assertTrue(projectionParameter.length === 1);
+        const variable = projectionParameter[0];
+        assertNonEmptyString(variable.name);
+        // assign variable to query class
+        const rawVariableExpression = new RawVariableExpression(variable.name, multiplicityOne, PackageableElementExplicitReference.create(guaranteeNonNullable(this.projectionState.queryBuilderState.class)));
+        const _rawVariableExpression = this.projectionState.queryBuilderState.graphManagerState.graphManager.serializeRawValueSpecification(rawVariableExpression);
+        const isolatedLambda = new RawLambda([_rawVariableExpression], this.lambda.body);
+        const type = (yield this.projectionState.queryBuilderState.graphManagerState.graphManager.getLambdaReturnType(isolatedLambda, graph));
+        const resolvedType = graph.getType(type);
+        assertTrue(Object.values(PRIMITIVE_TYPE).includes(resolvedType.path) || resolvedType instanceof Enumeration, 'projection column must have primitive return type');
+        this.setReturnType(resolvedType);
+    }
+    getReturnType() {
+        return this.returnType;
+    }
+}
+//# sourceMappingURL=QueryBuilderProjectionColumnState.js.map

package/lib/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderProjectionColumnState.js","sourceRoot":"","sources":["../../../../src/stores/fetch-structure/projection/QueryBuilderProjectionColumnState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAChE,OAAO,EAEL,QAAQ,EACR,iBAAiB,EACjB,IAAI,EACJ,oBAAoB,EACpB,UAAU,EACV,oBAAoB,GACrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,+CAA+C,GAChD,MAAM,6CAA6C,CAAC;AACrD,OAAO,EACL,oBAAoB,EACpB,mCAAmC,GACpC,MAAM,0CAA0C,CAAC;AAElD,OAAO,EAIL,mCAAmC,EACnC,mBAAmB,EACnB,cAAc,EACd,8BAA8B,EAC9B,WAAW,EACX,yBAAyB,EACzB,qBAAqB,EACrB,WAAW,EACX,SAAS,EACT,cAAc,GACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,4BAA4B,EAC5B,6BAA6B,GAC9B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAG9D,MAAM,CAAC,MAAM,wCAAwC,GAAG,mBAAmB,CAAC;AAM5E,MAAM,OAAgB,iCAAiC;IAC5C,IAAI,GAAG,IAAI,EAAE,CAAC;IACvB,eAAe,CAA8B;IAC7C,UAAU,CAAS;IAEnB,YACE,eAA4C,EAC5C,UAAkB;QAElB,cAAc,CAAC,IAAI,EAAE;YACnB,IAAI,EAAE,KAAK;YACX,eAAe,EAAE,KAAK;YACtB,UAAU,EAAE,UAAU;YACtB,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,aAAa,CAAC,GAAW;QACvB,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;IACxB,CAAC;CAGF;AAED,MAAM,OAAO,uCAAwC,SAAQ,iCAAiC;IAC5F,mBAAmB,GAAW,4BAA4B,CAAC;IAC3D,uBAAuB,CAAsC;IAE7D,YACE,eAA4C,EAC5C,kBAA8C,EAC9C,oBAA6B;QAE7B,KAAK,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAE3B,cAAc,CAAC,IAAI,EAAE;YACnB,mBAAmB,EAAE,UAAU;YAC/B,uBAAuB,EAAE,UAAU;YACnC,sBAAsB,EAAE,MAAM;YAC9B,cAAc,EAAE,MAAM;SACvB,CAAC,CAAC;QAEH,IAAI,CAAC,uBAAuB,GAAG,IAAI,mCAAmC,CACpE,eAAe,CAAC,iBAAiB,EACjC,kBAAkB,CACnB,CAAC;QACF,IAAI,CAAC,UAAU,GAAG,oBAAoB,CACpC,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,EAC/C,oBAAoB,CACrB,CAAC;IACJ,CAAC;IAED,sBAAsB,CAAC,GAAW;QAChC,IAAI,CAAC,mBAAmB,GAAG,GAAG,CAAC;IACjC,CAAC;IAED,cAAc,CACZ,IAA8C,EAC9C,oBAA6B;QAE7B,IAAI,CAAC,uBAAuB,GAAG,IAAI,mCAAmC,CACpE,IAAI,CAAC,eAAe,CAAC,iBAAiB,EACtC,+CAA+C,CAC7C,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,aAAa;aACjD,mBAAmB,EACtB,IAAI,EACJ,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,EAC9D,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,aAAa,CAAC,mBAAmB;aACrE,sBAAsB,CAC1B,CACF,CAAC;QACF,IAAI,CAAC,UAAU,GAAG,oBAAoB,CACpC,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,EAC/C,oBAAoB,CACrB,CAAC;IACJ,CAAC;IAEQ,aAAa;QACpB,OAAO,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,CAAC,IAAI,CAAC,WAAW;aACpE,KAAK,CAAC,OAAO,CAAC;IACnB,CAAC;CACF;AAED,MAAM,2CAA4C,SAAQ,iBAAiB;IACzE,iBAAiB,CAAoB;IACrC,+BAA+B,CAA8C;IAC7E;;;OAGG;IACH,kBAAkB,GAAG,EAAE,CAAC;IAExB,YACE,iBAAoC,EACpC,+BAA4E;QAE5E,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACd,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,+BAA+B,GAAG,+BAA+B,CAAC;IACzE,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,8BAA8B,CAAC;YACpC,uBAAuB;YACvB,yDAAyD;YACzD,6BAA6B,CAAC,aAAa;YAC3C,6BAA6B,CAAC,UAAU;YACxC,IAAI,CAAC,+BAA+B,CAAC,IAAI;SAC1C,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,UAAkB;QAC9B,IAAI,CAAC,kBAAkB,GAAG,UAAU,CAAC;IACvC,CAAC;IAED,CAAC,kCAAkC;QACjC,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;QACrC,IAAI,IAAI,CAAC,YAAY,EAAE;YACrB,IAAI;gBACF,MAAM,MAAM,GACV,CAAC,MAAM,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,gBAAgB,CAC3E,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,QAAQ,CACd,CAAc,CAAC;gBAClB,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;gBAC/B,IAAI,CAAC,+BAA+B,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;aACxD;YAAC,OAAO,KAAK,EAAE;gBACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACzB,IAAI,KAAK,YAAY,WAAW,EAAE;oBAChC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;iBAC5B;gBACD,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAC/C,QAAQ,CAAC,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,EACpD,KAAK,CACN,CAAC;aACH;SACF;aAAM;YACL,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,CAAC,+BAA+B,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;SAC7D;IACH,CAAC;IAED,CAAC,kCAAkC,CAAC,MAAe;QACjD,IAAI,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,IAAI,EAAE;YACpD,IAAI;gBACF,MAAM,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;gBAC7C,OAAO,CAAC,GAAG,CACT,IAAI,CAAC,QAAQ,EACb,IAAI,SAAS,CACX,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,UAAU,EACtD,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,IAAI,CACjD,CACF,CAAC;gBACF,MAAM,eAAe,GACnB,CAAC,MAAM,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,iBAAiB,CAC5E,OAAO,EACP,MAAM,CACP,CAAwB,CAAC;gBAC5B,MAAM,WAAW,GAAG,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvD,IAAI,CAAC,eAAe,CAClB,WAAW,KAAK,SAAS;oBACvB,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC;oBACvC,CAAC,CAAC,EAAE,CACP,CAAC;gBACF,IAAI,CAAC,WAAW,EAAE,CAAC;aACpB;YAAC,OAAO,KAAK,EAAE;gBACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACzB,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAC/C,QAAQ,CAAC,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,EACpD,KAAK,CACN,CAAC;aACH;SACF;aAAM;YACL,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;SAC1B;IACH,CAAC;CACF;AAED,MAAM,OAAO,2CAA4C,SAAQ,iCAAiC;IAChG,2BAA2B,CAA8C;IACzE,MAAM,CAAY;IAClB,UAAU,CAAmB;IAE7B,YAAY,eAA4C,EAAE,MAAiB;QACzE,KAAK,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC;QAEvC,cAAc,CAAC,IAAI,EAAE;YACnB,MAAM,EAAE,UAAU;YAClB,UAAU,EAAE,UAAU;YACtB,SAAS,EAAE,MAAM;YACjB,+BAA+B,EAAE,IAAI;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,2BAA2B;YAC9B,IAAI,2CAA2C,CAC7C,eAAe,CAAC,iBAAiB,EACjC,IAAI,CACL,CAAC;QACJ,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,SAAS,CAAC,GAAc;QACtB,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;IACpB,CAAC;IAED,aAAa,CAAC,GAAqB;QACjC,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;IACxB,CAAC;IAED;;;;OAIG;IACH,CAAC,+BAA+B;QAC9B,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,MAAM,mBAAmB,GAAG,IAAI,CAAC,MAAM,CAAC,UAAsB,CAAC;QAC/D,MAAM,KAAK,GACT,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC;QACjE,MAAM,eAAe,GAAG,KAAK,CAAC,sBAAsB,CAClD,yBAAyB,CAAC,GAAG,CAC9B,CAAC;QACF,UAAU,CAAC,mBAAmB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,mBAAmB,CAAC,CAAC,CAAuB,CAAC;QAC9D,oBAAoB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpC,iCAAiC;QACjC,MAAM,qBAAqB,GAAG,IAAI,qBAAqB,CACrD,QAAQ,CAAC,IAAI,EACb,eAAe,EACf,mCAAmC,CAAC,MAAM,CACxC,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,KAAK,CAAC,CACnE,CACF,CAAC;QACF,MAAM,sBAAsB,GAC1B,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,8BAA8B,CAClG,qBAAqB,CACtB,CAAC;QACJ,MAAM,cAAc,GAAG,IAAI,SAAS,CAClC,CAAC,sBAAsB,CAAC,EACxB,IAAI,CAAC,MAAM,CAAC,IAAI,CACjB,CAAC;QACF,MAAM,IAAI,GACR,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,mBAAmB,CAC9F,cAAc,EACd,KAAK,CACN,CAAW,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,UAAU,CACR,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,QAAQ,CACpC,YAAY,CAAC,IAAsB,CACpC,IAAI,YAAY,YAAY,WAAW,EACxC,mDAAmD,CACpD,CAAC;QACF,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;IACnC,CAAC;IAEQ,aAAa;QACpB,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF"}

package/lib/stores/fetch-structure/projection/QueryBuilderProjectionState.d.ts

@@ -0,0 +1,68 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { type GeneratorFn } from '@finos/legend-shared';
+import type { QueryBuilderState } from '../../QueryBuilderState.js';
+import { type CompilationError, Class, type LambdaFunction } from '@finos/legend-graph';
+import { QueryBuilderAggregationState } from './aggregation/QueryBuilderAggregationState.js';
+import { QueryBuilderFetchStructureImplementationState } from '../QueryBuilderFetchStructureImplementationState.js';
+import { QueryResultSetModifierState } from './QueryResultSetModifierState.js';
+import { QueryBuilderPostFilterState } from './post-filter/QueryBuilderPostFilterState.js';
+import type { QueryBuilderPostFilterOperator } from './post-filter/QueryBuilderPostFilterOperator.js';
+import type { QueryBuilderAggregateOperator } from './aggregation/QueryBuilderAggregateOperator.js';
+import { QueryBuilderDerivationProjectionColumnState, QueryBuilderSimpleProjectionColumnState, type QueryBuilderProjectionColumnState } from './QueryBuilderProjectionColumnState.js';
+import type { QueryBuilderFetchStructureState } from '../QueryBuilderFetchStructureState.js';
+import { type QueryBuilderExplorerTreePropertyNodeData } from '../../explorer/QueryBuilderExplorerState.js';
+import type { LambdaFunctionBuilderOption } from '../../QueryBuilderValueSpecificationBuilderHelper.js';
+export declare class QueryBuilderProjectionState extends QueryBuilderFetchStructureImplementationState {
+    columns: QueryBuilderProjectionColumnState[];
+    aggregationState: QueryBuilderAggregationState;
+    postFilterState: QueryBuilderPostFilterState;
+    resultSetModifierState: QueryResultSetModifierState;
+    isConvertDerivationProjectionObjects: boolean;
+    showPostFilterPanel: boolean;
+    postFilterOperators: QueryBuilderPostFilterOperator[];
+    aggregationOperators: QueryBuilderAggregateOperator[];
+    constructor(queryBuilderState: QueryBuilderState, fetchStructureState: QueryBuilderFetchStructureState);
+    get type(): string;
+    get derivations(): QueryBuilderDerivationProjectionColumnState[];
+    get hasParserError(): boolean;
+    get usedExplorerTreePropertyNodeIDs(): string[];
+    get validationIssues(): string[] | undefined;
+    onClassChange(_class: Class | undefined): void;
+    appendFetchStructure(lambdaFunction: LambdaFunction, options?: LambdaFunctionBuilderOption): void;
+    setShowPostFilterPanel(val: boolean): void;
+    convertDerivationProjectionObjects(): GeneratorFn<void>;
+    transformSimpleProjectionToDerivation(simpleProjectionColumnState: QueryBuilderSimpleProjectionColumnState): void;
+    replaceColumn(oldVal: QueryBuilderProjectionColumnState, newVal: QueryBuilderProjectionColumnState): void;
+    removeColumn(val: QueryBuilderProjectionColumnState): void;
+    addColumn(val: QueryBuilderProjectionColumnState, options?: {
+        /**
+         * Often time, we would want to enforce doing a sort when adding new column
+         * to ensure aggregate columns stay at the bottom of the list of projections
+         * But sometimes, we can opt in to use this flag to disable this sorting behavior,
+         * such as when we build/process.
+         */
+        skipSorting?: boolean | undefined;
+    }): void;
+    moveColumn(sourceIndex: number, targetIndex: number): void;
+    addNewBlankDerivation(): void;
+    revealCompilationError(compilationError: CompilationError): boolean;
+    clearCompilationError(): void;
+    fetchProperty(node: QueryBuilderExplorerTreePropertyNodeData): void;
+    fetchProperties(nodes: QueryBuilderExplorerTreePropertyNodeData[]): void;
+    checkBeforeChangingImplementation(onChange: () => void): void;
+}
+//# sourceMappingURL=QueryBuilderProjectionState.d.ts.map