@finos/legend-application-query 5.2.8 → 6.0.0

package/lib/stores/query-builder/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/query-builder/QueryBuilderStateBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateBuilder.js","sourceRoot":"","sources":["../../../src/stores/query-builder/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,sDAAsD,CAAC;AAEzG,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/query-builder/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/query-builder/QueryBuilderStateTestUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateTestUtils.d.ts","sourceRoot":"","sources":["../../../src/stores/query-builder/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,EAEL,KAAK,iBAAiB,EACvB,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,CA2C3B,CAAC"}

package/lib/stores/query-builder/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__getTestApplicationStore } from '@finos/legend-application';
+import { TEST__buildGraphWithEntities, TEST__getTestGraphManagerState, } from '@finos/legend-graph';
+import { jest } from '@jest/globals';
+import { flowResult } from 'mobx';
+import { LegendQueryPluginManager } from '../../application/LegendQueryPluginManager.js';
+import { QueryBuilder_GraphManagerPreset } from '../../graphManager/QueryBuilder_GraphManagerPreset.js';
+import { INTERNAL__BasicQueryBuilderState, } from './QueryBuilderState.js';
+import { TEST__getTestLegendQueryApplicationConfig } from '../QueryEditorStoreTestUtils.js';
+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());
+    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')
+            .mockResolvedValue(queryBuilderState.graphManagerState.graphManager.buildMappingModelCoverageAnalysisResult(rawMappingModelCoverageAnalysisResult));
+        await flowResult(queryBuilderState.explorerState.analyzeMappingModelCoverage());
+    }
+    return queryBuilderState;
+};
+//# sourceMappingURL=QueryBuilderStateTestUtils.js.map

package/lib/stores/query-builder/QueryBuilderStateTestUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderStateTestUtils.js","sourceRoot":"","sources":["../../../src/stores/query-builder/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,+CAA+C,CAAC;AACzF,OAAO,EAAE,+BAA+B,EAAE,MAAM,uDAAuD,CAAC;AACxG,OAAO,EACL,gCAAgC,GAEjC,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,yCAAyC,EAAE,MAAM,iCAAiC,CAAC;AAE5F,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,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;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/query-builder/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/query-builder/QueryBuilderTextEditorState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderTextEditorState.d.ts","sourceRoot":"","sources":["../../../src/stores/query-builder/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"}

package/lib/stores/query-builder/QueryBuilderTextEditorState.js

@@ -0,0 +1,145 @@
+/**
+ * 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 { GRAPH_MANAGER_EVENT, buildSourceInformationSourceId, ParserError, RawLambda, stub_RawLambda, pruneSourceInformation, } from '@finos/legend-graph';
+import { assertErrorThrown, LogEvent, } from '@finos/legend-shared';
+import { observable, action, flow, makeObservable, flowResult } from 'mobx';
+import { LambdaEditorState, TAB_SIZE } from '@finos/legend-application';
+export class QueryBuilderRawLambdaState {
+    lambda;
+    constructor(lambda) {
+        this.lambda = lambda;
+        makeObservable(this, {
+            lambda: observable,
+            setLambda: action,
+        });
+    }
+    setLambda(lambda) {
+        this.lambda = lambda;
+    }
+}
+export var QueryBuilderTextEditorMode;
+(function (QueryBuilderTextEditorMode) {
+    QueryBuilderTextEditorMode["TEXT"] = "TEXT";
+    QueryBuilderTextEditorMode["JSON"] = "JSON";
+})(QueryBuilderTextEditorMode = QueryBuilderTextEditorMode || (QueryBuilderTextEditorMode = {}));
+export class QueryBuilderTextEditorState extends LambdaEditorState {
+    queryBuilderState;
+    rawLambdaState;
+    isConvertingLambdaToString = false;
+    mode;
+    /**
+     * 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) {
+        super('', '');
+        makeObservable(this, {
+            rawLambdaState: observable,
+            isConvertingLambdaToString: observable,
+            mode: observable,
+            setQueryRawLambdaState: action,
+            setMode: action,
+            openModal: action,
+            closeModal: flow,
+        });
+        this.queryBuilderState = queryBuilderState;
+        this.rawLambdaState = new QueryBuilderRawLambdaState(stub_RawLambda());
+    }
+    get lambdaId() {
+        return buildSourceInformationSourceId(['query-builder']);
+    }
+    setQueryRawLambdaState(rawLambdaState) {
+        this.rawLambdaState = rawLambdaState;
+    }
+    setMode(openModal) {
+        this.mode = openModal;
+    }
+    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, { pruneSourceInformation: true }));
+                this.setParserError(undefined);
+                this.rawLambdaState.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.rawLambdaState.setLambda(emptyLambda);
+        }
+    }
+    *convertLambdaObjectToGrammarString(pretty) {
+        if (this.rawLambdaState.lambda.body) {
+            this.isConvertingLambdaToString = true;
+            try {
+                const lambdas = new Map();
+                lambdas.set(this.lambdaId, new RawLambda(this.rawLambdaState.lambda.parameters, this.rawLambdaState.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();
+                this.isConvertingLambdaToString = false;
+            }
+            catch (error) {
+                assertErrorThrown(error);
+                this.queryBuilderState.applicationStore.log.error(LogEvent.create(GRAPH_MANAGER_EVENT.PARSING_FAILURE), error);
+                this.isConvertingLambdaToString = false;
+            }
+        }
+        else {
+            this.clearErrors();
+            this.setLambdaString('');
+        }
+    }
+    openModal(mode) {
+        const rawLambda = this.queryBuilderState.buildQuery();
+        if (mode === QueryBuilderTextEditorMode.TEXT) {
+            this.setQueryRawLambdaState(new QueryBuilderRawLambdaState(rawLambda));
+        }
+        if (mode === QueryBuilderTextEditorMode.JSON) {
+            this.setLambdaJson(JSON.stringify(pruneSourceInformation(this.queryBuilderState.graphManagerState.graphManager.serializeRawValueSpecification(rawLambda)), null, TAB_SIZE));
+        }
+        this.setMode(mode);
+    }
+    *closeModal() {
+        if (this.mode === QueryBuilderTextEditorMode.TEXT) {
+            yield flowResult(this.convertLambdaGrammarStringToObject());
+            if (this.parserError) {
+                this.queryBuilderState.applicationStore.notifyError(`Can't parse query. Please fix error before closing: ${this.parserError.message}`);
+            }
+            else {
+                this.queryBuilderState.initialize(this.rawLambdaState.lambda);
+                this.setMode(undefined);
+            }
+            return;
+        }
+        this.setMode(undefined);
+    }
+}
+//# sourceMappingURL=QueryBuilderTextEditorState.js.map

package/lib/stores/query-builder/QueryBuilderTextEditorState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderTextEditorState.js","sourceRoot":"","sources":["../../../src/stores/query-builder/QueryBuilderTextEditorState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,mBAAmB,EACnB,8BAA8B,EAC9B,WAAW,EACX,SAAS,EACT,cAAc,EACd,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,iBAAiB,EACjB,QAAQ,GACT,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAE5E,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAExE,MAAM,OAAO,0BAA0B;IACrC,MAAM,CAAY;IAElB,YAAY,MAAiB;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,cAAc,CAAC,IAAI,EAAE;YACnB,MAAM,EAAE,UAAU;YAClB,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,MAAiB;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,CAAN,IAAY,0BAGX;AAHD,WAAY,0BAA0B;IACpC,2CAAa,CAAA;IACb,2CAAa,CAAA;AACf,CAAC,EAHW,0BAA0B,GAA1B,0BAA0B,KAA1B,0BAA0B,QAGrC;AAED,MAAM,OAAO,2BAA4B,SAAQ,iBAAiB;IAChE,iBAAiB,CAAoB;IACrC,cAAc,CAA6B;IAC3C,0BAA0B,GAAG,KAAK,CAAC;IACnC,IAAI,CAAyC;IAC7C;;;OAGG;IACH,kBAAkB,GAAG,EAAE,CAAC;IAExB,YAAY,iBAAoC;QAC9C,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAEd,cAAc,CAAC,IAAI,EAAE;YACnB,cAAc,EAAE,UAAU;YAC1B,0BAA0B,EAAE,UAAU;YACtC,IAAI,EAAE,UAAU;YAChB,sBAAsB,EAAE,MAAM;YAC9B,OAAO,EAAE,MAAM;YACf,SAAS,EAAE,MAAM;YACjB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,cAAc,GAAG,IAAI,0BAA0B,CAAC,cAAc,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,8BAA8B,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,sBAAsB,CAAC,cAA0C;QAC/D,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,OAAO,CAAC,SAAiD;QACvD,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;IACxB,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,EACb,EAAE,sBAAsB,EAAE,IAAI,EAAE,CACjC,CAAc,CAAC;gBAClB,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;gBAC/B,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;aACvC;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,cAAc,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;SAC5C;IACH,CAAC;IAED,CAAC,kCAAkC,CAAC,MAAe;QACjD,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,EAAE;YACnC,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC;YACvC,IAAI;gBACF,MAAM,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;gBAC7C,OAAO,CAAC,GAAG,CACT,IAAI,CAAC,QAAQ,EACb,IAAI,SAAS,CACX,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,EACrC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAChC,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;gBACnB,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;aACzC;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;gBACF,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;aACzC;SACF;aAAM;YACL,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;SAC1B;IACH,CAAC;IAED,SAAS,CAAC,IAAgC;QACxC,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;QACtD,IAAI,IAAI,KAAK,0BAA0B,CAAC,IAAI,EAAE;YAC5C,IAAI,CAAC,sBAAsB,CAAC,IAAI,0BAA0B,CAAC,SAAS,CAAC,CAAC,CAAC;SACxE;QACD,IAAI,IAAI,KAAK,0BAA0B,CAAC,IAAI,EAAE;YAC5C,IAAI,CAAC,aAAa,CAChB,IAAI,CAAC,SAAS,CACZ,sBAAsB,CACpB,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,YAAY,CAAC,8BAA8B,CAClF,SAAS,CACV,CACF,EACD,IAAI,EACJ,QAAQ,CACT,CACF,CAAC;SACH;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IAED,CAAC,UAAU;QACT,IAAI,IAAI,CAAC,IAAI,KAAK,0BAA0B,CAAC,IAAI,EAAE;YACjD,MAAM,UAAU,CAAC,IAAI,CAAC,kCAAkC,EAAE,CAAC,CAAC;YAC5D,IAAI,IAAI,CAAC,WAAW,EAAE;gBACpB,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,WAAW,CACjD,uDAAuD,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAClF,CAAC;aACH;iBAAM;gBACL,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBAC9D,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;aACzB;YACD,OAAO;SACR;QACD,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;CACF"}

package/lib/stores/query-builder/QueryBuilderTypeaheadHelper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderTypeaheadHelper.d.ts","sourceRoot":"","sources":["../../../src/stores/query-builder/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;AAqDhE,eAAO,MAAM,2BAA2B,sBACnB,iBAAiB,sBAChB,0BAA0B,SACvC,kBAAkB,GAAG,SAAS,KACpC,SAYF,CAAC;AAEF,eAAO,MAAM,mCAAmC,sBAC3B,iBAAiB,eAEhC,iCAAiC,GACjC,gCAAgC,SAC7B,kBAAkB,GAAG,SAAS,KACpC,SAGF,CAAC;AAEF,eAAO,MAAM,qBAAqB,WAAY,eAAe,KAAG,MAAM,EAgBrE,CAAC;AAEF,eAAO,MAAM,gBAAgB,QACtB,kBAAkB,GAAG,SAAS,KAClC,OAaF,CAAC"}