@finos/legend-application-query 5.2.6 → 5.2.7

package/lib/stores/QueryBuilderStateBuilder.js

@@ -0,0 +1,310 @@
+/**
+ * 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 { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../QueryBuilder_Const.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';
+const processGetAllExpression = (expression, queryBuilderState) => {
+    const _class = expression.genericType?.value.rawType;
+    assertType(_class, Class, `Can't process getAll() expression: getAll() return type is missing`);
+    queryBuilderState.querySetupState.setClass(_class, true);
+    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.querySetupState.setProcessingDate(expression.parametersValues[1]);
+            queryBuilderState.querySetupState.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.querySetupState.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.querySetupState.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.queryParametersState;
+    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,iCAAiC,EAAE,MAAM,0BAA0B,CAAC;AAC7E,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;AAE5E,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,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACzD,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,eAAe,CAAC,iBAAiB,CACjD,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAC/B,CAAC;YACF,iBAAiB,CAAC,eAAe,CAAC,eAAe,CAC/C,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,eAAe,CAAC,eAAe,CAC/C,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,eAAe,CAAC,iBAAiB,CACjD,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,oBAAoB,CAAC;IACnE,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/{QueryBuilderTestUtils.d.ts → QueryBuilderStateTestUtils.d.ts}

@@ -16,9 +16,9 @@
 import { type RawLambda, type RawMappingModelCoverageAnalysisResult } from '@finos/legend-graph';
 import type { Entity } from '@finos/legend-storage';
 import { QueryBuilderState } from './QueryBuilderState.js';
-export declare const TEST_setUpQueryBuilderState: (entities: Entity[], rawLambda?: RawLambda | undefined, executionContext?: {
+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=QueryBuilderTestUtils.d.ts.map
+//# 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;AAGH,OAAO,EACL,KAAK,SAAS,EACd,KAAK,qCAAqC,EAG3C,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAKpD,OAAO,EACL,iBAAiB,EAElB,MAAM,wBAAwB,CAAC;AAGhC,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,CAgD3B,CAAC"}

package/lib/stores/{QueryBuilderTestUtils.js → QueryBuilderStateTestUtils.js}

@@ -21,7 +21,7 @@ import { LegendQueryPluginManager } from '../application/LegendQueryPluginManage
 import { QueryBuilder_GraphManagerPreset } from '../graphManager/QueryBuilder_GraphManagerPreset.js';
 import { QueryBuilderState, StandardQueryBuilderMode, } from './QueryBuilderState.js';
 import { TEST__getTestLegendQueryApplicationConfig } from './QueryEditorStoreTestUtils.js';
-export const TEST_setUpQueryBuilderState = async (entities, rawLambda, executionContext, rawMappingModelCoverageAnalysisResult) => {
+export const TEST__setUpQueryBuilderState = async (entities, rawLambda, executionContext, rawMappingModelCoverageAnalysisResult) => {
     const pluginManager = LegendQueryPluginManager.create();
     pluginManager.usePresets([new QueryBuilder_GraphManagerPreset()]).install();
     const applicationStore = TEST__getTestApplicationStore(TEST__getTestLegendQueryApplicationConfig(), LegendQueryPluginManager.create());
@@ -47,4 +47,4 @@ export const TEST_setUpQueryBuilderState = async (entities, rawLambda, execution
     }
     return queryBuilderState;
 };
-//# sourceMappingURL=QueryBuilderTestUtils.js.map
+//# 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,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAC1E,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,wBAAwB,EAAE,MAAM,4CAA4C,CAAC;AACtF,OAAO,EAAE,+BAA+B,EAAE,MAAM,oDAAoD,CAAC;AACrG,OAAO,EACL,iBAAiB,EACjB,wBAAwB,GACzB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,yCAAyC,EAAE,MAAM,gCAAgC,CAAC;AAE3F,MAAM,CAAC,MAAM,4BAA4B,GAAG,KAAK,EAC/C,QAAkB,EAClB,SAAiC,EACjC,gBAMa,EACb,qCAEa,EACe,EAAE;IAC9B,MAAM,aAAa,GAAG,wBAAwB,CAAC,MAAM,EAAE,CAAC;IACxD,aAAa,CAAC,UAAU,CAAC,CAAC,IAAI,+BAA+B,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAC5E,MAAM,gBAAgB,GAAG,6BAA6B,CACpD,yCAAyC,EAAE,EAC3C,wBAAwB,CAAC,MAAM,EAAE,CAClC,CAAC;IACF,MAAM,iBAAiB,GAAG,8BAA8B,CAAC,aAAa,CAAC,CAAC;IACxE,MAAM,4BAA4B,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;IAChE,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,CAC7C,gBAAgB,EAChB,iBAAiB,EACjB,IAAI,wBAAwB,EAAE,CAC/B,CAAC;IACF,IAAI,SAAS,EAAE;QACb,iBAAiB,CAAC,uBAAuB,CAAC,SAAS,CAAC,CAAC;KACtD;IACD,IAAI,gBAAgB,EAAE;QACpB,MAAM,KAAK,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC;QACxD,iBAAiB,CAAC,eAAe,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ,CACvD,gBAAgB,CAAC,MAAM,CACxB,CAAC;QACF,iBAAiB,CAAC,eAAe,CAAC,OAAO,GAAG,KAAK,CAAC,UAAU,CAC1D,gBAAgB,CAAC,OAAO,CACzB,CAAC;QACF,IAAI,gBAAgB,CAAC,OAAO,EAAE;YAC5B,iBAAiB,CAAC,eAAe,CAAC,YAAY,GAAG,KAAK,CAAC,UAAU,CAC/D,gBAAgB,CAAC,OAAO,CACzB,CAAC;SACH;KACF;IACD,IAAI,qCAAqC,EAAE;QACzC,IAAI;aACD,KAAK,CACJ,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,EAChD,6BAA6B,CAC9B;aACA,iBAAiB,CAChB,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,uCAAuC,CACtF,qCAAqC,CACtC,CACF,CAAC;QAEJ,MAAM,UAAU,CACd,iBAAiB,CAAC,aAAa,CAAC,2BAA2B,EAAE,CAC9D,CAAC;KACH;IACD,OAAO,iBAAiB,CAAC;AAC3B,CAAC,CAAC"}

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

@@ -1 +1 @@
-{"version":3,"file":"QueryBuilderTypeaheadHelper.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderTypeaheadHelper.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,0BAA0B,EAC/B,KAAK,eAAe,EACpB,KAAK,kBAAkB,EAIvB,KAAK,SAAS,EACf,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,0EAA0E,CAAC;AAKjI,OAAO,EACL,iCAAiC,EAElC,MAAM,mEAAmE,CAAC;AAC3E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAmDhE,eAAO,MAAM,2BAA2B,sBACnB,iBAAiB,sBAChB,0BAA0B,SACvC,kBAAkB,GAAG,SAAS,KACpC,SASF,CAAC;AAEF,eAAO,MAAM,mCAAmC,sBAC3B,iBAAiB,eAEhC,iCAAiC,GACjC,gCAAgC,SAC7B,kBAAkB,GAAG,SAAS,KACpC,SAIF,CAAC;AAEF,eAAO,MAAM,qBAAqB,WAAY,eAAe,KAAG,MAAM,EAgBrE,CAAC;AAEF,eAAO,MAAM,gBAAgB,QACtB,kBAAkB,GAAG,SAAS,KAClC,OAaF,CAAC"}
+{"version":3,"file":"QueryBuilderTypeaheadHelper.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderTypeaheadHelper.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,0BAA0B,EAC/B,KAAK,eAAe,EACpB,KAAK,kBAAkB,EAIvB,KAAK,SAAS,EACf,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,0EAA0E,CAAC;AAKjI,OAAO,EACL,iCAAiC,EAElC,MAAM,mEAAmE,CAAC;AAC3E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAsDhE,eAAO,MAAM,2BAA2B,sBACnB,iBAAiB,sBAChB,0BAA0B,SACvC,kBAAkB,GAAG,SAAS,KACpC,SAaF,CAAC;AAEF,eAAO,MAAM,mCAAmC,sBAC3B,iBAAiB,eAEhC,iCAAiC,GACjC,gCAAgC,SAC7B,kBAAkB,GAAG,SAAS,KACpC,SAIF,CAAC;AAEF,eAAO,MAAM,qBAAqB,WAAY,eAAe,KAAG,MAAM,EAgBrE,CAAC;AAEF,eAAO,MAAM,gBAAgB,QACtB,kBAAkB,GAAG,SAAS,KAClC,OAaF,CAAC"}

package/lib/stores/QueryBuilderTypeaheadHelper.js

@@ -18,36 +18,38 @@ import { guaranteeType, isNonNullable, isString } from '@finos/legend-shared';
 import { QueryBuilderPostFilterOperator_StartWith } from './fetch-structure/projection/post-filter/operators/QueryBuilderPostFilterOperator_StartWith.js';
 import { PostFilterConditionState, QueryBuilderPostFilterTreeConditionNodeData, } from './fetch-structure/projection/post-filter/QueryBuilderPostFilterState.js';
 import { QueryBuilderProjectionColumnState, QueryBuilderSimpleProjectionColumnState, } from './fetch-structure/projection/QueryBuilderProjectionColumnState.js';
+import { QueryBuilderProjectionState } from './fetch-structure/projection/QueryBuilderProjectionState.js';
 const TYPEAHEAD_TAKE_LIMIT = 10;
 const START_LENGTH = 3;
 const createAndSetupQueryBuilderStateForTypeahead = (queryBuilderState) => {
     const builderState = queryBuilderState.cloneQueryBuilderState();
     builderState.querySetupState = queryBuilderState.querySetupState;
-    builderState.fetchStructureState.projectionState.resultSetModifierState.distinct =
-        true;
-    builderState.fetchStructureState.projectionState.resultSetModifierState.limit =
-        TYPEAHEAD_TAKE_LIMIT;
+    const projectionState = guaranteeType(builderState.fetchStructureState.implementation, QueryBuilderProjectionState);
+    projectionState.resultSetModifierState.distinct = true;
+    projectionState.resultSetModifierState.limit = TYPEAHEAD_TAKE_LIMIT;
     return builderState;
 };
 const buildColumnTypeaheadQuery = (builderState, columnState, value) => {
-    let projectionState;
+    const projectionState = guaranteeType(builderState.fetchStructureState.implementation, QueryBuilderProjectionState);
+    let projectionColumnState;
     if (columnState instanceof QueryBuilderProjectionColumnState) {
-        projectionState = columnState;
+        projectionColumnState = columnState;
     }
     else {
-        projectionState = columnState.projectionColumnState;
-        const aggregationState = builderState.fetchStructureState.projectionState.aggregationState;
+        projectionColumnState = columnState.projectionColumnState;
+        const aggregationState = projectionState.aggregationState;
         aggregationState.columns = [columnState];
     }
-    builderState.fetchStructureState.projectionState.columns = [projectionState];
-    const postConditionState = new PostFilterConditionState(builderState.fetchStructureState.projectionState.postFilterState, columnState, value, new QueryBuilderPostFilterOperator_StartWith());
+    projectionState.columns = [projectionColumnState];
+    const postConditionState = new PostFilterConditionState(projectionState.postFilterState, columnState, value, new QueryBuilderPostFilterOperator_StartWith());
     const postFilterNode = new QueryBuilderPostFilterTreeConditionNodeData(undefined, postConditionState);
-    builderState.fetchStructureState.projectionState.postFilterState.addNodeFromNode(postFilterNode, undefined);
+    projectionState.postFilterState.addNodeFromNode(postFilterNode, undefined);
     return builderState.resultState.buildExecutionRawLambda();
 };
 export const buildPropertyTypeaheadQuery = (queryBuilderState, propertyExpression, value) => {
     const builderState = createAndSetupQueryBuilderStateForTypeahead(queryBuilderState);
-    const columnState = new QueryBuilderSimpleProjectionColumnState(builderState.fetchStructureState.projectionState, propertyExpression, false);
+    const projectionState = guaranteeType(builderState.fetchStructureState.implementation, QueryBuilderProjectionState);
+    const columnState = new QueryBuilderSimpleProjectionColumnState(projectionState, propertyExpression, false);
     return buildColumnTypeaheadQuery(builderState, columnState, value);
 };
 export const buildProjectionColumnTypeaheadQuery = (queryBuilderState, columnState, value) => {

package/lib/stores/QueryBuilderTypeaheadHelper.js.map

@@ -1 +1 @@
-{"version":3,"file":"QueryBuilderTypeaheadHelper.js","sourceRoot":"","sources":["../../src/stores/QueryBuilderTypeaheadHelper.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAIL,sBAAsB,EACtB,cAAc,EACd,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EAAE,wCAAwC,EAAE,MAAM,gGAAgG,CAAC;AAE1J,OAAO,EACL,wBAAwB,EACxB,2CAA2C,GAC5C,MAAM,yEAAyE,CAAC;AACjF,OAAO,EACL,iCAAiC,EACjC,uCAAuC,GACxC,MAAM,mEAAmE,CAAC;AAG3E,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAChC,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,MAAM,2CAA2C,GAAG,CAClD,iBAAoC,EACjB,EAAE;IACrB,MAAM,YAAY,GAAG,iBAAiB,CAAC,sBAAsB,EAAE,CAAC;IAChE,YAAY,CAAC,eAAe,GAAG,iBAAiB,CAAC,eAAe,CAAC;IACjE,YAAY,CAAC,mBAAmB,CAAC,eAAe,CAAC,sBAAsB,CAAC,QAAQ;QAC9E,IAAI,CAAC;IACP,YAAY,CAAC,mBAAmB,CAAC,eAAe,CAAC,sBAAsB,CAAC,KAAK;QAC3E,oBAAoB,CAAC;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC;AAEF,MAAM,yBAAyB,GAAG,CAChC,YAA+B,EAC/B,WAEoC,EACpC,KAAqC,EAC1B,EAAE;IACb,IAAI,eAAe,CAAC;IACpB,IAAI,WAAW,YAAY,iCAAiC,EAAE;QAC5D,eAAe,GAAG,WAAW,CAAC;KAC/B;SAAM;QACL,eAAe,GAAG,WAAW,CAAC,qBAAqB,CAAC;QACpD,MAAM,gBAAgB,GACpB,YAAY,CAAC,mBAAmB,CAAC,eAAe,CAAC,gBAAgB,CAAC;QACpE,gBAAgB,CAAC,OAAO,GAAG,CAAC,WAAW,CAAC,CAAC;KAC1C;IACD,YAAY,CAAC,mBAAmB,CAAC,eAAe,CAAC,OAAO,GAAG,CAAC,eAAe,CAAC,CAAC;IAC7E,MAAM,kBAAkB,GAAG,IAAI,wBAAwB,CACrD,YAAY,CAAC,mBAAmB,CAAC,eAAe,CAAC,eAAe,EAChE,WAAW,EACX,KAAK,EACL,IAAI,wCAAwC,EAAE,CAC/C,CAAC;IACF,MAAM,cAAc,GAAG,IAAI,2CAA2C,CACpE,SAAS,EACT,kBAAkB,CACnB,CAAC;IACF,YAAY,CAAC,mBAAmB,CAAC,eAAe,CAAC,eAAe,CAAC,eAAe,CAC9E,cAAc,EACd,SAAS,CACV,CAAC;IACF,OAAO,YAAY,CAAC,WAAW,CAAC,uBAAuB,EAAE,CAAC;AAC5D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,iBAAoC,EACpC,kBAA8C,EAC9C,KAAqC,EAC1B,EAAE;IACb,MAAM,YAAY,GAChB,2CAA2C,CAAC,iBAAiB,CAAC,CAAC;IACjE,MAAM,WAAW,GAAG,IAAI,uCAAuC,CAC7D,YAAY,CAAC,mBAAmB,CAAC,eAAe,EAChD,kBAAkB,EAClB,KAAK,CACN,CAAC;IACF,OAAO,yBAAyB,CAAC,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;AACrE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,mCAAmC,GAAG,CACjD,iBAAoC,EACpC,WAEoC,EACpC,KAAqC,EAC1B,EAAE;IACb,MAAM,YAAY,GAChB,2CAA2C,CAAC,iBAAiB,CAAC,CAAC;IACjE,OAAO,yBAAyB,CAAC,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;AACrE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAuB,EAAY,EAAE;IACzE,MAAM,SAAS,GAAG,aAAa,CAC7B,MAAM,EACN,kBAAkB,EAClB,wDAAwD,CACzD,CAAC;IACF,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,SAAS,CAAC,MAAM,CAAC,IAAI;SAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACvB,MAAM,CAAC,aAAa,CAAC;SACrB,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;QACb,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE;YACf,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACjB;IACH,CAAC,CAAC,CAAC;IACL,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,GAAmC,EAC1B,EAAE;IACX,IAAI,GAAG,YAAY,sBAAsB,EAAE;QACzC,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5C,QAAQ,KAAK,CAAC,IAAI,EAAE;YAClB,KAAK,cAAc,CAAC,MAAM,CAAC,CAAC;gBAC1B,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAW,CAAC;gBACtC,OAAO,KAAK,CAAC,MAAM,IAAI,YAAY,CAAC;aACrC;YACD;gBACE,OAAO,KAAK,CAAC;SAChB;KACF;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC"}
+{"version":3,"file":"QueryBuilderTypeaheadHelper.js","sourceRoot":"","sources":["../../src/stores/QueryBuilderTypeaheadHelper.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAIL,sBAAsB,EACtB,cAAc,EACd,kBAAkB,GAEnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EAAE,wCAAwC,EAAE,MAAM,gGAAgG,CAAC;AAE1J,OAAO,EACL,wBAAwB,EACxB,2CAA2C,GAC5C,MAAM,yEAAyE,CAAC;AACjF,OAAO,EACL,iCAAiC,EACjC,uCAAuC,GACxC,MAAM,mEAAmE,CAAC;AAE3E,OAAO,EAAE,2BAA2B,EAAE,MAAM,6DAA6D,CAAC;AAE1G,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAChC,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,MAAM,2CAA2C,GAAG,CAClD,iBAAoC,EACjB,EAAE;IACrB,MAAM,YAAY,GAAG,iBAAiB,CAAC,sBAAsB,EAAE,CAAC;IAChE,YAAY,CAAC,eAAe,GAAG,iBAAiB,CAAC,eAAe,CAAC;IACjE,MAAM,eAAe,GAAG,aAAa,CACnC,YAAY,CAAC,mBAAmB,CAAC,cAAc,EAC/C,2BAA2B,CAC5B,CAAC;IACF,eAAe,CAAC,sBAAsB,CAAC,QAAQ,GAAG,IAAI,CAAC;IACvD,eAAe,CAAC,sBAAsB,CAAC,KAAK,GAAG,oBAAoB,CAAC;IACpE,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC;AAEF,MAAM,yBAAyB,GAAG,CAChC,YAA+B,EAC/B,WAEoC,EACpC,KAAqC,EAC1B,EAAE;IACb,MAAM,eAAe,GAAG,aAAa,CACnC,YAAY,CAAC,mBAAmB,CAAC,cAAc,EAC/C,2BAA2B,CAC5B,CAAC;IACF,IAAI,qBAAqB,CAAC;IAC1B,IAAI,WAAW,YAAY,iCAAiC,EAAE;QAC5D,qBAAqB,GAAG,WAAW,CAAC;KACrC;SAAM;QACL,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC;QAC1D,MAAM,gBAAgB,GAAG,eAAe,CAAC,gBAAgB,CAAC;QAC1D,gBAAgB,CAAC,OAAO,GAAG,CAAC,WAAW,CAAC,CAAC;KAC1C;IACD,eAAe,CAAC,OAAO,GAAG,CAAC,qBAAqB,CAAC,CAAC;IAClD,MAAM,kBAAkB,GAAG,IAAI,wBAAwB,CACrD,eAAe,CAAC,eAAe,EAC/B,WAAW,EACX,KAAK,EACL,IAAI,wCAAwC,EAAE,CAC/C,CAAC;IACF,MAAM,cAAc,GAAG,IAAI,2CAA2C,CACpE,SAAS,EACT,kBAAkB,CACnB,CAAC;IACF,eAAe,CAAC,eAAe,CAAC,eAAe,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;IAC3E,OAAO,YAAY,CAAC,WAAW,CAAC,uBAAuB,EAAE,CAAC;AAC5D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,iBAAoC,EACpC,kBAA8C,EAC9C,KAAqC,EAC1B,EAAE;IACb,MAAM,YAAY,GAChB,2CAA2C,CAAC,iBAAiB,CAAC,CAAC;IACjE,MAAM,eAAe,GAAG,aAAa,CACnC,YAAY,CAAC,mBAAmB,CAAC,cAAc,EAC/C,2BAA2B,CAC5B,CAAC;IACF,MAAM,WAAW,GAAG,IAAI,uCAAuC,CAC7D,eAAe,EACf,kBAAkB,EAClB,KAAK,CACN,CAAC;IACF,OAAO,yBAAyB,CAAC,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;AACrE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,mCAAmC,GAAG,CACjD,iBAAoC,EACpC,WAEoC,EACpC,KAAqC,EAC1B,EAAE;IACb,MAAM,YAAY,GAChB,2CAA2C,CAAC,iBAAiB,CAAC,CAAC;IACjE,OAAO,yBAAyB,CAAC,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;AACrE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAuB,EAAY,EAAE;IACzE,MAAM,SAAS,GAAG,aAAa,CAC7B,MAAM,EACN,kBAAkB,EAClB,wDAAwD,CACzD,CAAC;IACF,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,SAAS,CAAC,MAAM,CAAC,IAAI;SAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACvB,MAAM,CAAC,aAAa,CAAC;SACrB,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;QACb,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE;YACf,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACjB;IACH,CAAC,CAAC,CAAC;IACL,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,GAAmC,EAC1B,EAAE;IACX,IAAI,GAAG,YAAY,sBAAsB,EAAE;QACzC,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5C,QAAQ,KAAK,CAAC,IAAI,EAAE;YAClB,KAAK,cAAc,CAAC,MAAM,CAAC,CAAC;gBAC1B,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAW,CAAC;gBACtC,OAAO,KAAK,CAAC,MAAM,IAAI,YAAY,CAAC;aACrC;YACD;gBACE,OAAO,KAAK,CAAC;SAChB;KACF;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC"}

package/lib/stores/QueryBuilderValueSpecificationBuilder.d.ts

@@ -13,16 +13,8 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { type ValueSpecification, Class, Multiplicity, LambdaFunction, SimpleFunctionExpression, AbstractPropertyExpression } from '@finos/legend-graph';
+import { LambdaFunction } from '@finos/legend-graph';
 import type { QueryBuilderState } from './QueryBuilderState.js';
-export declare const buildGetAllFunction: (_class: Class, multiplicity: Multiplicity) => SimpleFunctionExpression;
-export declare const buildPropertyExpressionChain: (propertyExpression: AbstractPropertyExpression, queryBuilderState: QueryBuilderState) => ValueSpecification;
-export declare const buildLambdaFunction: (queryBuilderState: QueryBuilderState, 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;
-}) => LambdaFunction;
+import type { LambdaFunctionBuilderOption } from './QueryBuilderValueSpecificationBuilderHelper.js';
+export declare const buildLambdaFunction: (queryBuilderState: QueryBuilderState, options?: LambdaFunctionBuilderOption) => LambdaFunction;
 //# sourceMappingURL=QueryBuilderValueSpecificationBuilder.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"QueryBuilderValueSpecificationBuilder.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAQH,OAAO,EACL,KAAK,kBAAkB,EACvB,KAAK,EACL,YAAY,EAcZ,cAAc,EAId,wBAAwB,EAGxB,0BAA0B,EAI3B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAkHhE,eAAO,MAAM,mBAAmB,WACtB,KAAK,gBACC,YAAY,KACzB,wBAYF,CAAC;AAEF,eAAO,MAAM,4BAA4B,uBACnB,0BAA0B,qBAC3B,iBAAiB,KACnC,kBAkGF,CAAC;AAEF,eAAO,MAAM,mBAAmB,sBACX,iBAAiB,YAC1B;IACR;;;OAGG;IACH,wBAAwB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC/C,qBAAqB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAC7C,KACA,cA0bF,CAAC"}
+{"version":3,"file":"QueryBuilderValueSpecificationBuilder.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAWL,cAAc,EAIf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAIhE,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,kDAAkD,CAAC;AA+BpG,eAAO,MAAM,mBAAmB,sBACX,iBAAiB,YAC1B,2BAA2B,KACpC,cA2GF,CAAC"}