@finos/legend-query-builder 0.0.1

package/lib/stores/QueryBuilderStateBuilder.d.ts

@@ -0,0 +1,124 @@
+/**
+ * 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 { QueryBuilderState } from './QueryBuilderState.js';
+import { type AlloySerializationConfigInstanceValue, type EnumValueInstanceValue, type FunctionExpression, type MappingInstanceValue, type PairInstanceValue, type PropertyGraphFetchTreeInstanceValue, type PureListInstanceValue, type RootGraphFetchTreeInstanceValue, type RuntimeInstanceValue, type ValueSpecificationVisitor, type InstanceValue, type INTERNAL__UnknownValueSpecification, type LambdaFunction, type CollectionInstanceValue, type LambdaFunctionInstanceValue, type PrimitiveInstanceValue, SimpleFunctionExpression, type VariableExpression, type AbstractPropertyExpression, type INTERNAL__PropagatedValue, type ValueSpecification } from '@finos/legend-graph';
+/**
+ * This is the value specification processor (a.k.a state builder) for query builder.
+ *
+ * Unlike value specification builder which takes care of transforming the value specification
+ * from `protocol` to `metamodel`, and doing some (naive) type-inferencing, the processor takes care
+ * of traversing the query, analyzing it to build the query builder state. The processor represents
+ * a particular way we want to look at the query, if the query is not like what the processor
+ * expects, or understands, query builder state could not be built, i.e. the query will be deemed
+ * unsupported by query builder
+ *
+ * NOTE: While traversing the expression, this processor also does a fair amount of
+ * validations and assertions--enough to properly build the state. Unlike the builder,
+ * the processor should NEVER modify the metamodel, just traversing it.
+ */
+export declare class QueryBuilderValueSpecificationProcessor implements ValueSpecificationVisitor<void> {
+    readonly queryBuilderState: QueryBuilderState;
+    /**
+     * A value specification is a tree of value specifications.
+     *
+     * This structure may not intuitively correspond to how we write it in Pure
+     * For example, consider the following expression in Pure:
+     *
+     * | Person.all()->filter(x|$x.age > 0)->project([x|$x.name], ['Name'])->sort([desc('Name')]);
+     *
+     * which is equivalent to the more cryptic version:
+     *
+     * | sort(project(filter(all(Person), x|$x.age > 0), [x|$x.name], ['Name']), [desc('Name')]);
+     *
+     * Hence, the metamodel looks something like this (with some parts redacted)
+     * {
+     *   "function": "sort",
+     *   "parameters": [
+     *     {
+     *       "function": "project",
+     *       "parameters": [
+     *         {
+     *           "function": "filter",
+     *           "parameters": [
+     *             {
+     *               "function": "getAll",
+     *               ...
+     *             },
+     *             ... // filter() expression
+     *           ]
+     *         },
+     *         ... // project() expression column and alias
+     *       ]
+     *     },
+     *     {
+     *       "_type": "collection",
+     *       "values": [
+     *         {
+     *           "function": "desc",
+     *           ... // desc() expression
+     *         }
+     *       ]
+     *     }
+     *   ]
+     * }
+     *
+     * The first grammar form is farily declarative in nature and perhaps more intuitive.
+     * We tend to think, in terms of order of running:
+     * all() --> filter() --> project() --> sort()
+     *
+     * However, in reality, when the expression is used or read, the order of traversal is reversed:
+     * sort() --> project() --> filter() --> all()
+     *
+     * Now, in the context of the value specification processor, we need information about preceding
+     * expression in order to validate the usage context of functions, (for example, we want to only
+     * support project() after all() or filter()). In the case of function expression chain,
+     * we have to look at the function expression first parameter. There are some other context,
+     * such as what happens within the second parameter of the sort() expression
+     * in this case, we have the desc() expression, but it's not part of the main expression chain;
+     * however, we need to verify desc() is used within sort(), in this case, we have to record
+     * the parent expression. This is when the following attribute comes into use when processing
+     * value specification.
+     */
+    readonly parentExpression?: SimpleFunctionExpression | undefined;
+    private constructor();
+    static process(valueSpecification: ValueSpecification, queryBuilderState: QueryBuilderState): void;
+    /**
+     * Process value specification with information about parent function expression
+     * in order to be used in some validation/assertion
+     */
+    static processChild(valueSpecification: ValueSpecification, parentExpression: SimpleFunctionExpression, queryBuilderState: QueryBuilderState): void;
+    visit_INTERNAL__UnknownValueSpecification(valueSpecification: INTERNAL__UnknownValueSpecification): void;
+    visit_RootGraphFetchTreeInstanceValue(valueSpecification: RootGraphFetchTreeInstanceValue): void;
+    visit_PropertyGraphFetchTreeInstanceValue(valueSpecification: PropertyGraphFetchTreeInstanceValue): void;
+    visit_AlloySerializationConfigInstanceValue(valueSpecification: AlloySerializationConfigInstanceValue): void;
+    visit_PrimitiveInstanceValue(valueSpecification: PrimitiveInstanceValue): void;
+    visit_EnumValueInstanceValue(valueSpecification: EnumValueInstanceValue): void;
+    visit_RuntimeInstanceValue(valueSpecification: RuntimeInstanceValue): void;
+    visit_PairInstanceValue(valueSpecification: PairInstanceValue): void;
+    visit_MappingInstanceValue(valueSpecification: MappingInstanceValue): void;
+    visit_PureListInstanceValue(valueSpecification: PureListInstanceValue): void;
+    visit_CollectionInstanceValue(valueSpecification: CollectionInstanceValue): void;
+    visit_FunctionExpression(valueSpecification: FunctionExpression): void;
+    visit_INTERNAL__PropagatedValue(valueSpecification: INTERNAL__PropagatedValue): void;
+    visit_SimpleFunctionExpression(valueSpecification: SimpleFunctionExpression): void;
+    visit_VariableExpression(valueSpecification: VariableExpression): void;
+    visit_LambdaFunctionInstanceValue(valueSpecification: LambdaFunctionInstanceValue): void;
+    visit_AbstractPropertyExpression(valueSpecification: AbstractPropertyExpression): void;
+    visit_InstanceValue(valueSpecification: InstanceValue): void;
+}
+export declare const processParameters: (parameters: VariableExpression[], queryBuilderState: QueryBuilderState) => void;
+export declare const processQueryLambdaFunction: (lambdaFunction: LambdaFunction, queryBuilderState: QueryBuilderState) => void;
+//# sourceMappingURL=QueryBuilderStateBuilder.d.ts.map

package/lib/stores/QueryBuilderStateBuilder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateBuilder.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderStateBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AASH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EACL,KAAK,qCAAqC,EAC1C,KAAK,sBAAsB,EAC3B,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,mCAAmC,EACxC,KAAK,qBAAqB,EAC1B,KAAK,+BAA+B,EACpC,KAAK,oBAAoB,EACzB,KAAK,yBAAyB,EAC9B,KAAK,aAAa,EAClB,KAAK,mCAAmC,EACxC,KAAK,cAAc,EAInB,KAAK,uBAAuB,EAC5B,KAAK,2BAA2B,EAChC,KAAK,sBAAsB,EAC3B,wBAAwB,EACxB,KAAK,kBAAkB,EACvB,KAAK,0BAA0B,EAE/B,KAAK,yBAAyB,EAC9B,KAAK,kBAAkB,EACxB,MAAM,qBAAqB,CAAC;AAqF7B;;;;;;;;;;;;;GAaG;AACH,qBAAa,uCACX,YAAW,yBAAyB,CAAC,IAAI,CAAC;IAE1C,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAC9C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4DG;IACH,QAAQ,CAAC,gBAAgB,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;IAEjE,OAAO;IAQP,MAAM,CAAC,OAAO,CACZ,kBAAkB,EAAE,kBAAkB,EACtC,iBAAiB,EAAE,iBAAiB,GACnC,IAAI;IAMP;;;OAGG;IACH,MAAM,CAAC,YAAY,CACjB,kBAAkB,EAAE,kBAAkB,EACtC,gBAAgB,EAAE,wBAAwB,EAC1C,iBAAiB,EAAE,iBAAiB,GACnC,IAAI;IASP,yCAAyC,CACvC,kBAAkB,EAAE,mCAAmC,GACtD,IAAI;IA0BP,qCAAqC,CACnC,kBAAkB,EAAE,+BAA+B,GAClD,IAAI;IAIP,yCAAyC,CACvC,kBAAkB,EAAE,mCAAmC,GACtD,IAAI;IAIP,2CAA2C,CACzC,kBAAkB,EAAE,qCAAqC,GACxD,IAAI;IAIP,4BAA4B,CAC1B,kBAAkB,EAAE,sBAAsB,GACzC,IAAI;IAIP,4BAA4B,CAC1B,kBAAkB,EAAE,sBAAsB,GACzC,IAAI;IAIP,0BAA0B,CAAC,kBAAkB,EAAE,oBAAoB,GAAG,IAAI;IAI1E,uBAAuB,CAAC,kBAAkB,EAAE,iBAAiB,GAAG,IAAI;IAIpE,0BAA0B,CAAC,kBAAkB,EAAE,oBAAoB,GAAG,IAAI;IAI1E,2BAA2B,CAAC,kBAAkB,EAAE,qBAAqB,GAAG,IAAI;IAI5E,6BAA6B,CAC3B,kBAAkB,EAAE,uBAAuB,GAC1C,IAAI;IAIP,wBAAwB,CAAC,kBAAkB,EAAE,kBAAkB,GAAG,IAAI;IAItE,+BAA+B,CAC7B,kBAAkB,EAAE,yBAAyB,GAC5C,IAAI;IAIP,8BAA8B,CAC5B,kBAAkB,EAAE,wBAAwB,GAC3C,IAAI;IA+JP,wBAAwB,CAAC,kBAAkB,EAAE,kBAAkB,GAAG,IAAI;IAItE,iCAAiC,CAC/B,kBAAkB,EAAE,2BAA2B,GAC9C,IAAI;IAaP,gCAAgC,CAC9B,kBAAkB,EAAE,0BAA0B,GAC7C,IAAI;IAyBP,mBAAmB,CAAC,kBAAkB,EAAE,aAAa,GAAG,IAAI;CAG7D;AAED,eAAO,MAAM,iBAAiB,eAChB,kBAAkB,EAAE,qBACb,iBAAiB,KACnC,IAWF,CAAC;AAEF,eAAO,MAAM,0BAA0B,mBACrB,cAAc,qBACX,iBAAiB,KACnC,IAaF,CAAC"}

package/lib/stores/QueryBuilderStateBuilder.js

@@ -0,0 +1,311 @@
+/**
+ * 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 { assertNonNullable, assertTrue, assertType, guaranteeType, UnsupportedOperationError, } from '@finos/legend-shared';
+import { MILESTONING_STEREOTYPE, matchFunctionName, Class, SimpleFunctionExpression, getMilestoneTemporalStereotype, } from '@finos/legend-graph';
+import { processTDSPostFilterExpression } from './fetch-structure/projection/post-filter/QueryBuilderPostFilterStateBuilder.js';
+import { LambdaParameterState } from '@finos/legend-application';
+import { processFilterExpression } from './filter/QueryBuilderFilterStateBuilder.js';
+import { processTDSAggregateExpression, processTDSGroupByExpression, } from './fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.js';
+import { processGraphFetchExpression, processGraphFetchSerializeExpression, } from './fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeStateBuilder.js';
+import { processTDSDistinctExpression, processTDSProjectExpression, processTDSProjectionColumnPropertyExpression, processTDSProjectionDerivationExpression, processTDSSortDirectionExpression, processTDSSortExpression, processTDSTakeExpression, } from './fetch-structure/projection/QueryBuilderProjectionStateBuilder.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../graphManager/QueryBuilderSupportedFunctions.js';
+const processGetAllExpression = (expression, queryBuilderState) => {
+    const _class = expression.genericType?.value.rawType;
+    assertType(_class, Class, `Can't process getAll() expression: getAll() return type is missing`);
+    queryBuilderState.setClass(_class);
+    queryBuilderState.milestoningState.updateMilestoningConfiguration();
+    queryBuilderState.explorerState.refreshTreeData();
+    // check parameters (milestoning) and build state
+    let acceptedNoOfParameters = 1;
+    const stereotype = getMilestoneTemporalStereotype(_class, queryBuilderState.graphManagerState.graph);
+    switch (stereotype) {
+        case MILESTONING_STEREOTYPE.BITEMPORAL:
+            acceptedNoOfParameters = 3;
+            assertTrue(expression.parametersValues.length === acceptedNoOfParameters, `Can't process getAll() expression: when used with a bitemporal milestoned class getAll() expects two parameters`);
+            queryBuilderState.milestoningState.setProcessingDate(expression.parametersValues[1]);
+            queryBuilderState.milestoningState.setBusinessDate(expression.parametersValues[2]);
+            break;
+        case MILESTONING_STEREOTYPE.BUSINESS_TEMPORAL:
+            acceptedNoOfParameters = 2;
+            assertTrue(expression.parametersValues.length === acceptedNoOfParameters, `Can't process getAll() expression: when used with a milestoned class getAll() expects a parameter`);
+            queryBuilderState.milestoningState.setBusinessDate(expression.parametersValues[1]);
+            break;
+        case MILESTONING_STEREOTYPE.PROCESSING_TEMPORAL:
+            acceptedNoOfParameters = 2;
+            assertTrue(expression.parametersValues.length === acceptedNoOfParameters, `Can't process getAll() expression: when used with a milestoned class getAll() expects a parameter`);
+            queryBuilderState.milestoningState.setProcessingDate(expression.parametersValues[1]);
+            break;
+        default:
+            assertTrue(expression.parametersValues.length === acceptedNoOfParameters, `Can't process getAll() expression: getAll() expects no arguments`);
+    }
+};
+/**
+ * This is the value specification processor (a.k.a state builder) for query builder.
+ *
+ * Unlike value specification builder which takes care of transforming the value specification
+ * from `protocol` to `metamodel`, and doing some (naive) type-inferencing, the processor takes care
+ * of traversing the query, analyzing it to build the query builder state. The processor represents
+ * a particular way we want to look at the query, if the query is not like what the processor
+ * expects, or understands, query builder state could not be built, i.e. the query will be deemed
+ * unsupported by query builder
+ *
+ * NOTE: While traversing the expression, this processor also does a fair amount of
+ * validations and assertions--enough to properly build the state. Unlike the builder,
+ * the processor should NEVER modify the metamodel, just traversing it.
+ */
+export class QueryBuilderValueSpecificationProcessor {
+    queryBuilderState;
+    /**
+     * A value specification is a tree of value specifications.
+     *
+     * This structure may not intuitively correspond to how we write it in Pure
+     * For example, consider the following expression in Pure:
+     *
+     * | Person.all()->filter(x|$x.age > 0)->project([x|$x.name], ['Name'])->sort([desc('Name')]);
+     *
+     * which is equivalent to the more cryptic version:
+     *
+     * | sort(project(filter(all(Person), x|$x.age > 0), [x|$x.name], ['Name']), [desc('Name')]);
+     *
+     * Hence, the metamodel looks something like this (with some parts redacted)
+     * {
+     *   "function": "sort",
+     *   "parameters": [
+     *     {
+     *       "function": "project",
+     *       "parameters": [
+     *         {
+     *           "function": "filter",
+     *           "parameters": [
+     *             {
+     *               "function": "getAll",
+     *               ...
+     *             },
+     *             ... // filter() expression
+     *           ]
+     *         },
+     *         ... // project() expression column and alias
+     *       ]
+     *     },
+     *     {
+     *       "_type": "collection",
+     *       "values": [
+     *         {
+     *           "function": "desc",
+     *           ... // desc() expression
+     *         }
+     *       ]
+     *     }
+     *   ]
+     * }
+     *
+     * The first grammar form is farily declarative in nature and perhaps more intuitive.
+     * We tend to think, in terms of order of running:
+     * all() --> filter() --> project() --> sort()
+     *
+     * However, in reality, when the expression is used or read, the order of traversal is reversed:
+     * sort() --> project() --> filter() --> all()
+     *
+     * Now, in the context of the value specification processor, we need information about preceding
+     * expression in order to validate the usage context of functions, (for example, we want to only
+     * support project() after all() or filter()). In the case of function expression chain,
+     * we have to look at the function expression first parameter. There are some other context,
+     * such as what happens within the second parameter of the sort() expression
+     * in this case, we have the desc() expression, but it's not part of the main expression chain;
+     * however, we need to verify desc() is used within sort(), in this case, we have to record
+     * the parent expression. This is when the following attribute comes into use when processing
+     * value specification.
+     */
+    parentExpression;
+    constructor(queryBuilderState, parentExpression) {
+        this.queryBuilderState = queryBuilderState;
+        this.parentExpression = parentExpression;
+    }
+    static process(valueSpecification, queryBuilderState) {
+        valueSpecification.accept_ValueSpecificationVisitor(new QueryBuilderValueSpecificationProcessor(queryBuilderState, undefined));
+    }
+    /**
+     * Process value specification with information about parent function expression
+     * in order to be used in some validation/assertion
+     */
+    static processChild(valueSpecification, parentExpression, queryBuilderState) {
+        valueSpecification.accept_ValueSpecificationVisitor(new QueryBuilderValueSpecificationProcessor(queryBuilderState, parentExpression));
+    }
+    visit_INTERNAL__UnknownValueSpecification(valueSpecification) {
+        assertNonNullable(this.parentExpression, `Can't process unknown value: parent expression cannot be retrieved`);
+        if (matchFunctionName(this.parentExpression.functionName, [
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_AGG,
+        ])) {
+            processTDSProjectionDerivationExpression(valueSpecification, this.parentExpression, this.queryBuilderState);
+            return;
+        }
+        throw new UnsupportedOperationError(`Can't process unknown value with parent expression of function ${this.parentExpression.functionName}()`);
+    }
+    visit_RootGraphFetchTreeInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_PropertyGraphFetchTreeInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_AlloySerializationConfigInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_PrimitiveInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_EnumValueInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_RuntimeInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_PairInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_MappingInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_PureListInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_CollectionInstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_FunctionExpression(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_INTERNAL__PropagatedValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_SimpleFunctionExpression(valueSpecification) {
+        const functionName = valueSpecification.functionName;
+        if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.GET_ALL)) {
+            processGetAllExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, [
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.FILTER,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_FILTER,
+        ])) {
+            // NOTE: for filter, since it sometimes can be ambiguous
+            // whether meta::pure::functions::collection::filter() was used
+            // or meta::pure::tds::filter() was used, we have to have custom logic
+            // to determine
+            // check parameters
+            assertTrue(valueSpecification.parametersValues.length === 2, `Can't process filter() expression: filter() expects 1 argument`);
+            // check preceding expression
+            const precedingExpression = guaranteeType(valueSpecification.parametersValues[0], SimpleFunctionExpression, `Can't process filter() expression: only support filter() immediately following an expression`);
+            QueryBuilderValueSpecificationProcessor.process(precedingExpression, this.queryBuilderState);
+            if (matchFunctionName(precedingExpression.functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.GET_ALL)) {
+                assertTrue(matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.FILTER), `Can't process filter() expression: only supports ${QUERY_BUILDER_SUPPORTED_FUNCTIONS.FILTER}() immediately following getAll() (got '${functionName}')`);
+                processFilterExpression(valueSpecification, this.queryBuilderState);
+                return;
+            }
+            else if (matchFunctionName(precedingExpression.functionName, [
+                QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT,
+                QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY,
+            ])) {
+                assertTrue(matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_FILTER), `Can't process post-filter expression: only supports ${QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_FILTER}() immediately following project()/groupBy() (got '${functionName}')`);
+                processTDSPostFilterExpression(valueSpecification, this.queryBuilderState);
+                return;
+            }
+            else {
+                throw new UnsupportedOperationError(`Can't process filter() expression: only support filter() immediately following getAll() or project()/groupBy()`);
+            }
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT)) {
+            processTDSProjectExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_TAKE)) {
+            processTDSTakeExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_DISTINCT)) {
+            processTDSDistinctExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_SORT)) {
+            processTDSSortExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, [
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_ASC,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_DESC,
+        ])) {
+            processTDSSortDirectionExpression(valueSpecification, this.parentExpression, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY)) {
+            processTDSGroupByExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_AGG)) {
+            processTDSAggregateExpression(valueSpecification, this.parentExpression, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.SERIALIZE)) {
+            processGraphFetchSerializeExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        else if (matchFunctionName(functionName, [
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.GRAPH_FETCH_CHECKED,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.GRAPH_FETCH,
+        ])) {
+            processGraphFetchExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        throw new UnsupportedOperationError(`Can't process expression of function ${functionName}()`);
+    }
+    visit_VariableExpression(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+    visit_LambdaFunctionInstanceValue(valueSpecification) {
+        valueSpecification.values.forEach((value) => value.expressionSequence.forEach((expression) => expression.accept_ValueSpecificationVisitor(new QueryBuilderValueSpecificationProcessor(this.queryBuilderState, this.parentExpression))));
+    }
+    visit_AbstractPropertyExpression(valueSpecification) {
+        assertNonNullable(this.parentExpression, `Can't process property expression: parent expression cannot be retrieved`);
+        if (matchFunctionName(this.parentExpression.functionName, [
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY,
+            QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_AGG,
+        ])) {
+            processTDSProjectionColumnPropertyExpression(valueSpecification, this.queryBuilderState);
+            return;
+        }
+        throw new UnsupportedOperationError(`Can't process property expression with parent expression of function ${this.parentExpression.functionName}()`);
+    }
+    visit_InstanceValue(valueSpecification) {
+        throw new UnsupportedOperationError();
+    }
+}
+export const processParameters = (parameters, queryBuilderState) => {
+    const queryParameterState = queryBuilderState.parametersState;
+    parameters.forEach((parameter) => {
+        const parameterState = new LambdaParameterState(parameter, queryBuilderState.observableContext, queryBuilderState.graphManagerState.graph);
+        parameterState.mockParameterValue();
+        queryParameterState.addParameter(parameterState);
+    });
+};
+export const processQueryLambdaFunction = (lambdaFunction, queryBuilderState) => {
+    if (lambdaFunction.functionType.parameters.length) {
+        processParameters(lambdaFunction.functionType.parameters, queryBuilderState);
+    }
+    lambdaFunction.expressionSequence.map((expression) => QueryBuilderValueSpecificationProcessor.process(expression, queryBuilderState));
+};
+//# sourceMappingURL=QueryBuilderStateBuilder.js.map

package/lib/stores/QueryBuilderStateBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateBuilder.js","sourceRoot":"","sources":["../../src/stores/QueryBuilderStateBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,iBAAiB,EACjB,UAAU,EACV,UAAU,EACV,aAAa,EACb,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAcL,sBAAsB,EACtB,iBAAiB,EACjB,KAAK,EAIL,wBAAwB,EAGxB,8BAA8B,GAG/B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,8BAA8B,EAAE,MAAM,gFAAgF,CAAC;AAChI,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4CAA4C,CAAC;AACrF,OAAO,EACL,6BAA6B,EAC7B,2BAA2B,GAC5B,MAAM,iFAAiF,CAAC;AACzF,OAAO,EACL,2BAA2B,EAC3B,oCAAoC,GACrC,MAAM,yEAAyE,CAAC;AACjF,OAAO,EACL,4BAA4B,EAC5B,2BAA2B,EAC3B,4CAA4C,EAC5C,wCAAwC,EACxC,iCAAiC,EACjC,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,oEAAoE,CAAC;AAC5E,OAAO,EAAE,iCAAiC,EAAE,MAAM,mDAAmD,CAAC;AAEtG,MAAM,uBAAuB,GAAG,CAC9B,UAAoC,EACpC,iBAAoC,EAC9B,EAAE;IACR,MAAM,MAAM,GAAG,UAAU,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC;IACrD,UAAU,CACR,MAAM,EACN,KAAK,EACL,oEAAoE,CACrE,CAAC;IACF,iBAAiB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACnC,iBAAiB,CAAC,gBAAgB,CAAC,8BAA8B,EAAE,CAAC;IACpE,iBAAiB,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;IAElD,iDAAiD;IACjD,IAAI,sBAAsB,GAAG,CAAC,CAAC;IAC/B,MAAM,UAAU,GAAG,8BAA8B,CAC/C,MAAM,EACN,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAC1C,CAAC;IACF,QAAQ,UAAU,EAAE;QAClB,KAAK,sBAAsB,CAAC,UAAU;YACpC,sBAAsB,GAAG,CAAC,CAAC;YAC3B,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,sBAAsB,EAC7D,iHAAiH,CAClH,CAAC;YACF,iBAAiB,CAAC,gBAAgB,CAAC,iBAAiB,CAClD,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAC/B,CAAC;YACF,iBAAiB,CAAC,gBAAgB,CAAC,eAAe,CAChD,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAC/B,CAAC;YACF,MAAM;QACR,KAAK,sBAAsB,CAAC,iBAAiB;YAC3C,sBAAsB,GAAG,CAAC,CAAC;YAC3B,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,sBAAsB,EAC7D,mGAAmG,CACpG,CAAC;YACF,iBAAiB,CAAC,gBAAgB,CAAC,eAAe,CAChD,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAC/B,CAAC;YACF,MAAM;QACR,KAAK,sBAAsB,CAAC,mBAAmB;YAC7C,sBAAsB,GAAG,CAAC,CAAC;YAC3B,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,sBAAsB,EAC7D,mGAAmG,CACpG,CAAC;YACF,iBAAiB,CAAC,gBAAgB,CAAC,iBAAiB,CAClD,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAC/B,CAAC;YACF,MAAM;QACR;YACE,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,sBAAsB,EAC7D,kEAAkE,CACnE,CAAC;KACL;AACH,CAAC,CAAC;AAEF;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAO,uCAAuC;IAGzC,iBAAiB,CAAoB;IAC9C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4DG;IACM,gBAAgB,CAAwC;IAEjE,YACE,iBAAoC,EACpC,gBAAsD;QAEtD,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,OAAO,CACZ,kBAAsC,EACtC,iBAAoC;QAEpC,kBAAkB,CAAC,gCAAgC,CACjD,IAAI,uCAAuC,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAC1E,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,YAAY,CACjB,kBAAsC,EACtC,gBAA0C,EAC1C,iBAAoC;QAEpC,kBAAkB,CAAC,gCAAgC,CACjD,IAAI,uCAAuC,CACzC,iBAAiB,EACjB,gBAAgB,CACjB,CACF,CAAC;IACJ,CAAC;IAED,yCAAyC,CACvC,kBAAuD;QAEvD,iBAAiB,CACf,IAAI,CAAC,gBAAgB,EACrB,oEAAoE,CACrE,CAAC;QAEF,IACE,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE;YACpD,iCAAiC,CAAC,WAAW;YAC7C,iCAAiC,CAAC,YAAY;YAC9C,iCAAiC,CAAC,OAAO;SAC1C,CAAC,EACF;YACA,wCAAwC,CACtC,kBAAkB,EAClB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,iBAAiB,CACvB,CAAC;YACF,OAAO;SACR;QAED,MAAM,IAAI,yBAAyB,CACjC,kEAAkE,IAAI,CAAC,gBAAgB,CAAC,YAAY,IAAI,CACzG,CAAC;IACJ,CAAC;IAED,qCAAqC,CACnC,kBAAmD;QAEnD,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,yCAAyC,CACvC,kBAAuD;QAEvD,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,2CAA2C,CACzC,kBAAyD;QAEzD,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,4BAA4B,CAC1B,kBAA0C;QAE1C,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,4BAA4B,CAC1B,kBAA0C;QAE1C,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,0BAA0B,CAAC,kBAAwC;QACjE,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,uBAAuB,CAAC,kBAAqC;QAC3D,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,0BAA0B,CAAC,kBAAwC;QACjE,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,2BAA2B,CAAC,kBAAyC;QACnE,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,6BAA6B,CAC3B,kBAA2C;QAE3C,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,wBAAwB,CAAC,kBAAsC;QAC7D,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,+BAA+B,CAC7B,kBAA6C;QAE7C,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,8BAA8B,CAC5B,kBAA4C;QAE5C,MAAM,YAAY,GAAG,kBAAkB,CAAC,YAAY,CAAC;QACrD,IACE,iBAAiB,CAAC,YAAY,EAAE,iCAAiC,CAAC,OAAO,CAAC,EAC1E;YACA,uBAAuB,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACpE,OAAO;SACR;aAAM,IACL,iBAAiB,CAAC,YAAY,EAAE;YAC9B,iCAAiC,CAAC,MAAM;YACxC,iCAAiC,CAAC,UAAU;SAC7C,CAAC,EACF;YACA,wDAAwD;YACxD,+DAA+D;YAC/D,sEAAsE;YACtE,eAAe;YAEf,mBAAmB;YACnB,UAAU,CACR,kBAAkB,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAChD,gEAAgE,CACjE,CAAC;YAEF,6BAA6B;YAC7B,MAAM,mBAAmB,GAAG,aAAa,CACvC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,EACtC,wBAAwB,EACxB,8FAA8F,CAC/F,CAAC;YACF,uCAAuC,CAAC,OAAO,CAC7C,mBAAmB,EACnB,IAAI,CAAC,iBAAiB,CACvB,CAAC;YAEF,IACE,iBAAiB,CACf,mBAAmB,CAAC,YAAY,EAChC,iCAAiC,CAAC,OAAO,CAC1C,EACD;gBACA,UAAU,CACR,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,MAAM,CACzC,EACD,oDAAoD,iCAAiC,CAAC,MAAM,2CAA2C,YAAY,IAAI,CACxJ,CAAC;gBACF,uBAAuB,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBACpE,OAAO;aACR;iBAAM,IACL,iBAAiB,CAAC,mBAAmB,CAAC,YAAY,EAAE;gBAClD,iCAAiC,CAAC,WAAW;gBAC7C,iCAAiC,CAAC,YAAY;aAC/C,CAAC,EACF;gBACA,UAAU,CACR,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,UAAU,CAC7C,EACD,uDAAuD,iCAAiC,CAAC,UAAU,sDAAsD,YAAY,IAAI,CAC1K,CAAC;gBACF,8BAA8B,CAC5B,kBAAkB,EAClB,IAAI,CAAC,iBAAiB,CACvB,CAAC;gBACF,OAAO;aACR;iBAAM;gBACL,MAAM,IAAI,yBAAyB,CACjC,gHAAgH,CACjH,CAAC;aACH;SACF;aAAM,IACL,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,WAAW,CAC9C,EACD;YACA,2BAA2B,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACxE,OAAO;SACR;aAAM,IACL,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,QAAQ,CAC3C,EACD;YACA,wBAAwB,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACrE,OAAO;SACR;aAAM,IACL,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,YAAY,CAC/C,EACD;YACA,4BAA4B,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACzE,OAAO;SACR;aAAM,IACL,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,QAAQ,CAC3C,EACD;YACA,wBAAwB,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACrE,OAAO;SACR;aAAM,IACL,iBAAiB,CAAC,YAAY,EAAE;YAC9B,iCAAiC,CAAC,OAAO;YACzC,iCAAiC,CAAC,QAAQ;SAC3C,CAAC,EACF;YACA,iCAAiC,CAC/B,kBAAkB,EAClB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,iBAAiB,CACvB,CAAC;YACF,OAAO;SACR;aAAM,IACL,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,YAAY,CAC/C,EACD;YACA,2BAA2B,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACxE,OAAO;SACR;aAAM,IACL,iBAAiB,CAAC,YAAY,EAAE,iCAAiC,CAAC,OAAO,CAAC,EAC1E;YACA,6BAA6B,CAC3B,kBAAkB,EAClB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,iBAAiB,CACvB,CAAC;YACF,OAAO;SACR;aAAM,IACL,iBAAiB,CACf,YAAY,EACZ,iCAAiC,CAAC,SAAS,CAC5C,EACD;YACA,oCAAoC,CAClC,kBAAkB,EAClB,IAAI,CAAC,iBAAiB,CACvB,CAAC;YACF,OAAO;SACR;aAAM,IACL,iBAAiB,CAAC,YAAY,EAAE;YAC9B,iCAAiC,CAAC,mBAAmB;YACrD,iCAAiC,CAAC,WAAW;SAC9C,CAAC,EACF;YACA,2BAA2B,CAAC,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACxE,OAAO;SACR;QACD,MAAM,IAAI,yBAAyB,CACjC,wCAAwC,YAAY,IAAI,CACzD,CAAC;IACJ,CAAC;IAED,wBAAwB,CAAC,kBAAsC;QAC7D,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IAED,iCAAiC,CAC/B,kBAA+C;QAE/C,kBAAkB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAC1C,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAC9C,UAAU,CAAC,gCAAgC,CACzC,IAAI,uCAAuC,CACzC,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,gBAAgB,CACtB,CACF,CACF,CACF,CAAC;IACJ,CAAC;IAED,gCAAgC,CAC9B,kBAA8C;QAE9C,iBAAiB,CACf,IAAI,CAAC,gBAAgB,EACrB,0EAA0E,CAC3E,CAAC;QAEF,IACE,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE;YACpD,iCAAiC,CAAC,WAAW;YAC7C,iCAAiC,CAAC,YAAY;YAC9C,iCAAiC,CAAC,OAAO;SAC1C,CAAC,EACF;YACA,4CAA4C,CAC1C,kBAAkB,EAClB,IAAI,CAAC,iBAAiB,CACvB,CAAC;YACF,OAAO;SACR;QAED,MAAM,IAAI,yBAAyB,CACjC,wEAAwE,IAAI,CAAC,gBAAgB,CAAC,YAAY,IAAI,CAC/G,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,kBAAiC;QACnD,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;CACF;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,UAAgC,EAChC,iBAAoC,EAC9B,EAAE;IACR,MAAM,mBAAmB,GAAG,iBAAiB,CAAC,eAAe,CAAC;IAC9D,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;QAC/B,MAAM,cAAc,GAAG,IAAI,oBAAoB,CAC7C,SAAS,EACT,iBAAiB,CAAC,iBAAiB,EACnC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAC1C,CAAC;QACF,cAAc,CAAC,kBAAkB,EAAE,CAAC;QACpC,mBAAmB,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CACxC,cAA8B,EAC9B,iBAAoC,EAC9B,EAAE;IACR,IAAI,cAAc,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,EAAE;QACjD,iBAAiB,CACf,cAAc,CAAC,YAAY,CAAC,UAAU,EACtC,iBAAiB,CAClB,CAAC;KACH;IACD,cAAc,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACnD,uCAAuC,CAAC,OAAO,CAC7C,UAAU,EACV,iBAAiB,CAClB,CACF,CAAC;AACJ,CAAC,CAAC"}

package/lib/stores/QueryBuilderStateTestUtils.d.ts

@@ -0,0 +1,24 @@
+/**
+ * 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 RawLambda, type RawMappingModelCoverageAnalysisResult } from '@finos/legend-graph';
+import type { Entity } from '@finos/legend-storage';
+import { type QueryBuilderState } from './QueryBuilderState.js';
+export declare const TEST__setUpQueryBuilderState: (entities: Entity[], rawLambda?: RawLambda | undefined, executionContext?: {
+    _class: string;
+    mapping: string;
+    runtime?: string | undefined;
+} | undefined, rawMappingModelCoverageAnalysisResult?: RawMappingModelCoverageAnalysisResult | undefined) => Promise<QueryBuilderState>;
+//# sourceMappingURL=QueryBuilderStateTestUtils.d.ts.map

package/lib/stores/QueryBuilderStateTestUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateTestUtils.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderStateTestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAOH,OAAO,EACL,KAAK,SAAS,EACd,KAAK,qCAAqC,EAG3C,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAIpD,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,wBAAwB,CAAC;AAEhC,eAAO,MAAM,4BAA4B,aAC7B,MAAM,EAAE,cACN,SAAS,GAAG,SAAS,qBAE7B;IACE,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B,GACD,SAAS,0CAET,qCAAqC,GACrC,SAAS,KACZ,QAAQ,iBAAiB,CAiD3B,CAAC"}

package/lib/stores/QueryBuilderStateTestUtils.js

@@ -0,0 +1,50 @@
+/**
+ * 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 { TEST__getGenericApplicationConfig, TEST__getTestApplicationStore, TEST__LegendApplicationPluginManager, } from '@finos/legend-application';
+import { TEST__buildGraphWithEntities, TEST__getTestGraphManagerState, } from '@finos/legend-graph';
+import { jest } from '@jest/globals';
+import { flowResult } from 'mobx';
+import { QueryBuilder_GraphManagerPreset } from '../graphManager/QueryBuilder_GraphManagerPreset.js';
+import { INTERNAL__BasicQueryBuilderState, } from './QueryBuilderState.js';
+export const TEST__setUpQueryBuilderState = async (entities, rawLambda, executionContext, rawMappingModelCoverageAnalysisResult) => {
+    const pluginManager = TEST__LegendApplicationPluginManager.create();
+    pluginManager.usePresets([new QueryBuilder_GraphManagerPreset()]).install();
+    const applicationStore = TEST__getTestApplicationStore(TEST__getGenericApplicationConfig(), pluginManager);
+    const graphManagerState = TEST__getTestGraphManagerState(pluginManager);
+    await TEST__buildGraphWithEntities(graphManagerState, entities);
+    const queryBuilderState = new INTERNAL__BasicQueryBuilderState(applicationStore, graphManagerState);
+    if (rawLambda) {
+        queryBuilderState.rebuildWithQuery(rawLambda);
+    }
+    if (executionContext) {
+        const graph = queryBuilderState.graphManagerState.graph;
+        queryBuilderState.class = graph.getClass(executionContext._class);
+        queryBuilderState.mapping = graph.getMapping(executionContext.mapping);
+        if (executionContext.runtime) {
+            queryBuilderState.runtimeValue = graph.getRuntime(executionContext.runtime);
+        }
+    }
+    if (rawMappingModelCoverageAnalysisResult) {
+        jest
+            .spyOn(queryBuilderState.graphManagerState.graphManager, 'analyzeMappingModelCoverage')
+            .mockReturnValue(Promise.resolve(queryBuilderState.graphManagerState.graphManager.buildMappingModelCoverageAnalysisResult(rawMappingModelCoverageAnalysisResult)));
+    }
+    if (rawMappingModelCoverageAnalysisResult) {
+        await flowResult(queryBuilderState.explorerState.analyzeMappingModelCoverage());
+    }
+    return queryBuilderState;
+};
+//# sourceMappingURL=QueryBuilderStateTestUtils.js.map

package/lib/stores/QueryBuilderStateTestUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateTestUtils.js","sourceRoot":"","sources":["../../src/stores/QueryBuilderStateTestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,iCAAiC,EACjC,6BAA6B,EAC7B,oCAAoC,GACrC,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAGL,4BAA4B,EAC5B,8BAA8B,GAC/B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAClC,OAAO,EAAE,+BAA+B,EAAE,MAAM,oDAAoD,CAAC;AACrG,OAAO,EACL,gCAAgC,GAEjC,MAAM,wBAAwB,CAAC;AAEhC,MAAM,CAAC,MAAM,4BAA4B,GAAG,KAAK,EAC/C,QAAkB,EAClB,SAAiC,EACjC,gBAMa,EACb,qCAEa,EACe,EAAE;IAC9B,MAAM,aAAa,GAAG,oCAAoC,CAAC,MAAM,EAAE,CAAC;IACpE,aAAa,CAAC,UAAU,CAAC,CAAC,IAAI,+BAA+B,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAC5E,MAAM,gBAAgB,GAAG,6BAA6B,CACpD,iCAAiC,EAAE,EACnC,aAAa,CACd,CAAC;IACF,MAAM,iBAAiB,GAAG,8BAA8B,CAAC,aAAa,CAAC,CAAC;IACxE,MAAM,4BAA4B,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;IAChE,MAAM,iBAAiB,GAAG,IAAI,gCAAgC,CAC5D,gBAAgB,EAChB,iBAAiB,CAClB,CAAC;IACF,IAAI,SAAS,EAAE;QACb,iBAAiB,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;KAC/C;IACD,IAAI,gBAAgB,EAAE;QACpB,MAAM,KAAK,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC;QACxD,iBAAiB,CAAC,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAClE,iBAAiB,CAAC,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACvE,IAAI,gBAAgB,CAAC,OAAO,EAAE;YAC5B,iBAAiB,CAAC,YAAY,GAAG,KAAK,CAAC,UAAU,CAC/C,gBAAgB,CAAC,OAAO,CACzB,CAAC;SACH;KACF;IAED,IAAI,qCAAqC,EAAE;QACzC,IAAI;aACD,KAAK,CACJ,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,EAChD,6BAA6B,CAC9B;aACA,eAAe,CACd,OAAO,CAAC,OAAO,CACb,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,uCAAuC,CACtF,qCAAqC,CACtC,CACF,CACF,CAAC;KACL;IAED,IAAI,qCAAqC,EAAE;QACzC,MAAM,UAAU,CACd,iBAAiB,CAAC,aAAa,CAAC,2BAA2B,EAAE,CAC9D,CAAC;KACH;IAED,OAAO,iBAAiB,CAAC;AAC3B,CAAC,CAAC"}

package/lib/stores/QueryBuilderTextEditorState.d.ts

@@ -0,0 +1,49 @@
+/**
+ * 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 { RawLambda } from '@finos/legend-graph';
+import { type GeneratorFn } from '@finos/legend-shared';
+import type { QueryBuilderState } from './QueryBuilderState.js';
+import { LambdaEditorState } from '@finos/legend-application';
+export declare class QueryBuilderRawLambdaState {
+    lambda: RawLambda;
+    constructor(lambda: RawLambda);
+    setLambda(lambda: RawLambda): void;
+}
+export declare enum QueryBuilderTextEditorMode {
+    TEXT = "TEXT",
+    JSON = "JSON"
+}
+export declare class QueryBuilderTextEditorState extends LambdaEditorState {
+    queryBuilderState: QueryBuilderState;
+    rawLambdaState: QueryBuilderRawLambdaState;
+    isConvertingLambdaToString: boolean;
+    mode: QueryBuilderTextEditorMode | undefined;
+    /**
+     * 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);
+    get lambdaId(): string;
+    setQueryRawLambdaState(rawLambdaState: QueryBuilderRawLambdaState): void;
+    setMode(openModal: QueryBuilderTextEditorMode | undefined): void;
+    setLambdaJson(lambdaJson: string): void;
+    convertLambdaGrammarStringToObject(): GeneratorFn<void>;
+    convertLambdaObjectToGrammarString(pretty: boolean): GeneratorFn<void>;
+    openModal(mode: QueryBuilderTextEditorMode): void;
+    closeModal(): GeneratorFn<void>;
+}
+//# sourceMappingURL=QueryBuilderTextEditorState.d.ts.map

package/lib/stores/QueryBuilderTextEditorState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderTextEditorState.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderTextEditorState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAIL,SAAS,EAGV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,WAAW,EAGjB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAY,MAAM,2BAA2B,CAAC;AAExE,qBAAa,0BAA0B;IACrC,MAAM,EAAE,SAAS,CAAC;gBAEN,MAAM,EAAE,SAAS;IAS7B,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI;CAGnC;AAED,oBAAY,0BAA0B;IACpC,IAAI,SAAS;IACb,IAAI,SAAS;CACd;AAED,qBAAa,2BAA4B,SAAQ,iBAAiB;IAChE,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,cAAc,EAAE,0BAA0B,CAAC;IAC3C,0BAA0B,UAAS;IACnC,IAAI,EAAE,0BAA0B,GAAG,SAAS,CAAC;IAC7C;;;OAGG;IACH,kBAAkB,SAAM;gBAEZ,iBAAiB,EAAE,iBAAiB;IAiBhD,IAAI,QAAQ,IAAI,MAAM,CAErB;IAED,sBAAsB,CAAC,cAAc,EAAE,0BAA0B,GAAG,IAAI;IAIxE,OAAO,CAAC,SAAS,EAAE,0BAA0B,GAAG,SAAS,GAAG,IAAI;IAIhE,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAItC,kCAAkC,IAAI,WAAW,CAAC,IAAI,CAAC;IA4BvD,kCAAkC,CAAC,MAAM,EAAE,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC;IAuCvE,SAAS,CAAC,IAAI,EAAE,0BAA0B,GAAG,IAAI;IAqBhD,UAAU,IAAI,WAAW,CAAC,IAAI,CAAC;CAejC"}