@finos/legend-application-query 5.2.6 → 6.0.0

package/lib/stores/query-builder/fetch-structure/projection/QueryBuilderProjectionValueSpecificationBuilder.js

@@ -0,0 +1,189 @@
+/**
+ * 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 { CollectionInstanceValue, extractElementNameFromPath, GenericType, GenericTypeExplicitReference, Multiplicity, PrimitiveInstanceValue, PRIMITIVE_TYPE, SimpleFunctionExpression, TYPICAL_MULTIPLICITY_TYPE, INTERNAL__UnknownValueSpecification, V1_serializeRawValueSpecification, V1_transformRawLambda, V1_GraphTransformerContextBuilder, matchFunctionName, } from '@finos/legend-graph';
+import { guaranteeNonNullable, UnsupportedOperationError, } from '@finos/legend-shared';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { buildPropertyExpressionChain } from '../../QueryBuilderValueSpecificationBuilderHelper.js';
+import { buildGenericLambdaFunctionInstanceValue } from '../../QueryBuilderValueSpecificationHelper.js';
+import { appendPostFilter } from './post-filter/QueryBuilderPostFilterValueSpecificationBuilder.js';
+import { QueryBuilderDerivationProjectionColumnState, QueryBuilderSimpleProjectionColumnState, } from './QueryBuilderProjectionColumnState.js';
+import { COLUMN_SORT_TYPE, } from './QueryResultSetModifierState.js';
+const buildSortExpression = (sortColumnState) => {
+    const multiplicityOne = sortColumnState.columnState.projectionState.queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+    const sortColumnFunction = new SimpleFunctionExpression(extractElementNameFromPath(sortColumnState.sortType === COLUMN_SORT_TYPE.ASC
+        ? QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_ASC
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_DESC), multiplicityOne);
+    const sortColumnName = new PrimitiveInstanceValue(GenericTypeExplicitReference.create(new GenericType(sortColumnState.columnState.projectionState.queryBuilderState.graphManagerState.graph.getPrimitiveType(PRIMITIVE_TYPE.STRING))), multiplicityOne);
+    sortColumnName.values = [sortColumnState.columnState.columnName];
+    sortColumnFunction.parametersValues[0] = sortColumnName;
+    return sortColumnFunction;
+};
+const appendResultSetModifier = (resultModifierState, lambda, options) => {
+    const multiplicityOne = resultModifierState.projectionState.queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+    if (lambda.expressionSequence.length === 1) {
+        const func = lambda.expressionSequence[0];
+        if (func instanceof SimpleFunctionExpression) {
+            if (matchFunctionName(func.functionName, [
+                QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT,
+                QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY,
+                QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_FILTER,
+            ])) {
+                let currentExpression = func;
+                // build distinct()
+                if (resultModifierState.distinct) {
+                    const distinctFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_DISTINCT), multiplicityOne);
+                    distinctFunction.parametersValues[0] = currentExpression;
+                    currentExpression = distinctFunction;
+                }
+                // build sort()
+                if (resultModifierState.sortColumns.length) {
+                    const sortFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_SORT), multiplicityOne);
+                    const multiplicity = new Multiplicity(resultModifierState.sortColumns.length, resultModifierState.sortColumns.length);
+                    const collection = new CollectionInstanceValue(multiplicity, undefined);
+                    collection.values =
+                        resultModifierState.sortColumns.map(buildSortExpression);
+                    sortFunction.parametersValues[0] = currentExpression;
+                    sortFunction.parametersValues[1] = collection;
+                    currentExpression = sortFunction;
+                }
+                // build take()
+                if (resultModifierState.limit || options?.overridingLimit) {
+                    const limit = new PrimitiveInstanceValue(GenericTypeExplicitReference.create(new GenericType(resultModifierState.projectionState.queryBuilderState.graphManagerState.graph.getPrimitiveType(PRIMITIVE_TYPE.INTEGER))), multiplicityOne);
+                    limit.values = [
+                        Math.min(resultModifierState.limit ?? Number.MAX_SAFE_INTEGER, options?.overridingLimit ?? Number.MAX_SAFE_INTEGER),
+                    ];
+                    const takeFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_TAKE), multiplicityOne);
+                    takeFunction.parametersValues[0] = currentExpression;
+                    takeFunction.parametersValues[1] = limit;
+                    currentExpression = takeFunction;
+                }
+                lambda.expressionSequence[0] = currentExpression;
+                return lambda;
+            }
+        }
+    }
+    return lambda;
+};
+export const appendProjection = (projectionState, lambdaFunction, options) => {
+    const queryBuilderState = projectionState.queryBuilderState;
+    const precedingExpression = guaranteeNonNullable(lambdaFunction.expressionSequence[0], `Can't build projection expression: preceding expression is not defined`);
+    const multiplicityOne = queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE);
+    const typeString = queryBuilderState.graphManagerState.graph.getPrimitiveType(PRIMITIVE_TYPE.STRING);
+    // build projection
+    if (projectionState.aggregationState.columns.length) {
+        // aggregation
+        const groupByFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY), multiplicityOne);
+        const colLambdas = new CollectionInstanceValue(new Multiplicity(projectionState.columns.length -
+            projectionState.aggregationState.columns.length, projectionState.columns.length -
+            projectionState.aggregationState.columns.length));
+        const aggregateLambdas = new CollectionInstanceValue(new Multiplicity(projectionState.aggregationState.columns.length, projectionState.aggregationState.columns.length));
+        const colAliases = new CollectionInstanceValue(new Multiplicity(projectionState.columns.length, projectionState.columns.length));
+        projectionState.columns.forEach((projectionColumnState) => {
+            // column alias
+            const colAlias = new PrimitiveInstanceValue(GenericTypeExplicitReference.create(new GenericType(typeString)), multiplicityOne);
+            colAlias.values.push(projectionColumnState.columnName);
+            colAliases.values.push(colAlias);
+            const aggregateColumnState = projectionState.aggregationState.columns.find((column) => column.projectionColumnState === projectionColumnState);
+            // column projection
+            let columnLambda;
+            if (projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState) {
+                columnLambda = buildGenericLambdaFunctionInstanceValue(projectionColumnState.lambdaParameterName, [
+                    buildPropertyExpressionChain(projectionColumnState.propertyExpressionState.propertyExpression, projectionColumnState.propertyExpressionState.queryBuilderState, true),
+                ], queryBuilderState.graphManagerState.graph);
+            }
+            else if (projectionColumnState instanceof
+                QueryBuilderDerivationProjectionColumnState) {
+                columnLambda = new INTERNAL__UnknownValueSpecification(V1_serializeRawValueSpecification(V1_transformRawLambda(projectionColumnState.lambda, new V1_GraphTransformerContextBuilder(
+                // TODO?: do we need to include the plugins here?
+                [])
+                    .withKeepSourceInformationFlag(Boolean(options?.keepSourceInformation))
+                    .build())));
+            }
+            else {
+                throw new UnsupportedOperationError(`Can't build project() column expression: unsupported projection column state`, projectionColumnState);
+            }
+            // column aggregation
+            if (aggregateColumnState) {
+                const aggregateFunctionExpression = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_AGG), multiplicityOne);
+                const aggregateLambda = buildGenericLambdaFunctionInstanceValue(aggregateColumnState.lambdaParameterName, [
+                    aggregateColumnState.operator.buildAggregateExpressionFromState(aggregateColumnState),
+                ], aggregateColumnState.aggregationState.projectionState
+                    .queryBuilderState.graphManagerState.graph);
+                aggregateFunctionExpression.parametersValues = [
+                    columnLambda,
+                    aggregateLambda,
+                ];
+                aggregateLambdas.values.push(aggregateFunctionExpression);
+            }
+            else {
+                colLambdas.values.push(columnLambda);
+            }
+        });
+        groupByFunction.parametersValues = [
+            precedingExpression,
+            colLambdas,
+            aggregateLambdas,
+            colAliases,
+        ];
+        lambdaFunction.expressionSequence[0] = groupByFunction;
+    }
+    else if (projectionState.columns.length) {
+        // projection
+        const projectFunction = new SimpleFunctionExpression(extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT), multiplicityOne);
+        const colLambdas = new CollectionInstanceValue(new Multiplicity(projectionState.columns.length, projectionState.columns.length));
+        const colAliases = new CollectionInstanceValue(new Multiplicity(projectionState.columns.length, projectionState.columns.length));
+        projectionState.columns.forEach((projectionColumnState) => {
+            // column alias
+            const colAlias = new PrimitiveInstanceValue(GenericTypeExplicitReference.create(new GenericType(typeString)), multiplicityOne);
+            colAlias.values.push(projectionColumnState.columnName);
+            colAliases.values.push(colAlias);
+            // column projection
+            let columnLambda;
+            if (projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState) {
+                columnLambda = buildGenericLambdaFunctionInstanceValue(projectionColumnState.lambdaParameterName, [
+                    buildPropertyExpressionChain(projectionColumnState.propertyExpressionState.propertyExpression, projectionColumnState.propertyExpressionState.queryBuilderState),
+                ], queryBuilderState.graphManagerState.graph);
+            }
+            else if (projectionColumnState instanceof
+                QueryBuilderDerivationProjectionColumnState) {
+                columnLambda = new INTERNAL__UnknownValueSpecification(V1_serializeRawValueSpecification(V1_transformRawLambda(projectionColumnState.lambda, new V1_GraphTransformerContextBuilder(
+                // TODO?: do we need to include the plugins here?
+                [])
+                    .withKeepSourceInformationFlag(Boolean(options?.keepSourceInformation))
+                    .build())));
+            }
+            else {
+                throw new UnsupportedOperationError(`Can't build project() column expression: unsupported projection column state`, projectionColumnState);
+            }
+            colLambdas.values.push(columnLambda);
+        });
+        projectFunction.parametersValues = [
+            precedingExpression,
+            colLambdas,
+            colAliases,
+        ];
+        lambdaFunction.expressionSequence[0] = projectFunction;
+    }
+    // build post-filter
+    appendPostFilter(projectionState.postFilterState, lambdaFunction);
+    // build result set modifiers
+    appendResultSetModifier(projectionState.resultSetModifierState, lambdaFunction, {
+        overridingLimit: options?.isBuildingExecutionQuery
+            ? queryBuilderState.resultState.previewLimit
+            : undefined,
+    });
+};
+//# sourceMappingURL=QueryBuilderProjectionValueSpecificationBuilder.js.map

package/lib/stores/query-builder/fetch-structure/projection/QueryBuilderProjectionValueSpecificationBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderProjectionValueSpecificationBuilder.js","sourceRoot":"","sources":["../../../../../src/stores/query-builder/fetch-structure/projection/QueryBuilderProjectionValueSpecificationBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAEL,uBAAuB,EACvB,0BAA0B,EAC1B,WAAW,EACX,4BAA4B,EAC5B,YAAY,EACZ,sBAAsB,EACtB,cAAc,EACd,wBAAwB,EACxB,yBAAyB,EAEzB,mCAAmC,EACnC,iCAAiC,EACjC,qBAAqB,EACrB,iCAAiC,EACjC,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,iCAAiC,EAAE,MAAM,4DAA4D,CAAC;AAC/G,OAAO,EAAE,4BAA4B,EAAE,MAAM,sDAAsD,CAAC;AACpG,OAAO,EAAE,uCAAuC,EAAE,MAAM,+CAA+C,CAAC;AACxG,OAAO,EAAE,gBAAgB,EAAE,MAAM,kEAAkE,CAAC;AACpG,OAAO,EACL,2CAA2C,EAC3C,uCAAuC,GACxC,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EACL,gBAAgB,GAGjB,MAAM,kCAAkC,CAAC;AAE1C,MAAM,mBAAmB,GAAG,CAC1B,eAAgC,EACN,EAAE;IAC5B,MAAM,eAAe,GACnB,eAAe,CAAC,WAAW,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,sBAAsB,CAC1G,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IACJ,MAAM,kBAAkB,GAAG,IAAI,wBAAwB,CACrD,0BAA0B,CACxB,eAAe,CAAC,QAAQ,KAAK,gBAAgB,CAAC,GAAG;QAC/C,CAAC,CAAC,iCAAiC,CAAC,OAAO;QAC3C,CAAC,CAAC,iCAAiC,CAAC,QAAQ,CAC/C,EACD,eAAe,CAChB,CAAC;IACF,MAAM,cAAc,GAAG,IAAI,sBAAsB,CAC/C,4BAA4B,CAAC,MAAM,CACjC,IAAI,WAAW,CACb,eAAe,CAAC,WAAW,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,gBAAgB,CACpG,cAAc,CAAC,MAAM,CACtB,CACF,CACF,EACD,eAAe,CAChB,CAAC;IACF,cAAc,CAAC,MAAM,GAAG,CAAC,eAAe,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IACjE,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC;IACxD,OAAO,kBAAkB,CAAC;AAC5B,CAAC,CAAC;AAEF,MAAM,uBAAuB,GAAG,CAC9B,mBAAgD,EAChD,MAAsB,EACtB,OAIa,EACG,EAAE;IAClB,MAAM,eAAe,GACnB,mBAAmB,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,sBAAsB,CAClG,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IACJ,IAAI,MAAM,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAI,IAAI,YAAY,wBAAwB,EAAE;YAC5C,IACE,iBAAiB,CAAC,IAAI,CAAC,YAAY,EAAE;gBACnC,iCAAiC,CAAC,WAAW;gBAC7C,iCAAiC,CAAC,YAAY;gBAC9C,iCAAiC,CAAC,UAAU;aAC7C,CAAC,EACF;gBACA,IAAI,iBAAiB,GAAG,IAAI,CAAC;gBAE7B,mBAAmB;gBACnB,IAAI,mBAAmB,CAAC,QAAQ,EAAE;oBAChC,MAAM,gBAAgB,GAAG,IAAI,wBAAwB,CACnD,0BAA0B,CACxB,iCAAiC,CAAC,YAAY,CAC/C,EACD,eAAe,CAChB,CAAC;oBACF,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC;oBACzD,iBAAiB,GAAG,gBAAgB,CAAC;iBACtC;gBAED,eAAe;gBACf,IAAI,mBAAmB,CAAC,WAAW,CAAC,MAAM,EAAE;oBAC1C,MAAM,YAAY,GAAG,IAAI,wBAAwB,CAC/C,0BAA0B,CACxB,iCAAiC,CAAC,QAAQ,CAC3C,EACD,eAAe,CAChB,CAAC;oBACF,MAAM,YAAY,GAAG,IAAI,YAAY,CACnC,mBAAmB,CAAC,WAAW,CAAC,MAAM,EACtC,mBAAmB,CAAC,WAAW,CAAC,MAAM,CACvC,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,YAAY,EACZ,SAAS,CACV,CAAC;oBACF,UAAU,CAAC,MAAM;wBACf,mBAAmB,CAAC,WAAW,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;oBAC3D,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC;oBACrD,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC;oBAC9C,iBAAiB,GAAG,YAAY,CAAC;iBAClC;gBAED,eAAe;gBACf,IAAI,mBAAmB,CAAC,KAAK,IAAI,OAAO,EAAE,eAAe,EAAE;oBACzD,MAAM,KAAK,GAAG,IAAI,sBAAsB,CACtC,4BAA4B,CAAC,MAAM,CACjC,IAAI,WAAW,CACb,mBAAmB,CAAC,eAAe,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,gBAAgB,CAC5F,cAAc,CAAC,OAAO,CACvB,CACF,CACF,EACD,eAAe,CAChB,CAAC;oBACF,KAAK,CAAC,MAAM,GAAG;wBACb,IAAI,CAAC,GAAG,CACN,mBAAmB,CAAC,KAAK,IAAI,MAAM,CAAC,gBAAgB,EACpD,OAAO,EAAE,eAAe,IAAI,MAAM,CAAC,gBAAgB,CACpD;qBACF,CAAC;oBACF,MAAM,YAAY,GAAG,IAAI,wBAAwB,CAC/C,0BAA0B,CACxB,iCAAiC,CAAC,QAAQ,CAC3C,EACD,eAAe,CAChB,CAAC;oBAEF,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC;oBACrD,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;oBACzC,iBAAiB,GAAG,YAAY,CAAC;iBAClC;gBAED,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC;gBACjD,OAAO,MAAM,CAAC;aACf;SACF;KACF;IACD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,eAA4C,EAC5C,cAA8B,EAC9B,OAOC,EACK,EAAE;IACR,MAAM,iBAAiB,GAAG,eAAe,CAAC,iBAAiB,CAAC;IAC5D,MAAM,mBAAmB,GAAG,oBAAoB,CAC9C,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,EACpC,wEAAwE,CACzE,CAAC;IACF,MAAM,eAAe,GACnB,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,sBAAsB,CAC9D,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IACJ,MAAM,UAAU,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,gBAAgB,CAC3E,cAAc,CAAC,MAAM,CACtB,CAAC;IAEF,mBAAmB;IACnB,IAAI,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,EAAE;QACnD,cAAc;QACd,MAAM,eAAe,GAAG,IAAI,wBAAwB,CAClD,0BAA0B,CACxB,iCAAiC,CAAC,YAAY,CAC/C,EACD,eAAe,CAChB,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,IAAI,YAAY,CACd,eAAe,CAAC,OAAO,CAAC,MAAM;YAC5B,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,EACjD,eAAe,CAAC,OAAO,CAAC,MAAM;YAC5B,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAClD,CACF,CAAC;QACF,MAAM,gBAAgB,GAAG,IAAI,uBAAuB,CAClD,IAAI,YAAY,CACd,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,EAC/C,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAChD,CACF,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,IAAI,YAAY,CACd,eAAe,CAAC,OAAO,CAAC,MAAM,EAC9B,eAAe,CAAC,OAAO,CAAC,MAAM,CAC/B,CACF,CAAC;QACF,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,EAAE,EAAE;YACxD,eAAe;YACf,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CACzC,4BAA4B,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC,UAAU,CAAC,CAAC,EAChE,eAAe,CAChB,CAAC;YACF,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;YACvD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAEjC,MAAM,oBAAoB,GACxB,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAC3C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,qBAAqB,KAAK,qBAAqB,CACnE,CAAC;YAEJ,oBAAoB;YACpB,IAAI,YAAgC,CAAC;YACrC,IACE,qBAAqB,YAAY,uCAAuC,EACxE;gBACA,YAAY,GAAG,uCAAuC,CACpD,qBAAqB,CAAC,mBAAmB,EACzC;oBACE,4BAA4B,CAC1B,qBAAqB,CAAC,uBAAuB,CAAC,kBAAkB,EAChE,qBAAqB,CAAC,uBAAuB,CAAC,iBAAiB,EAC/D,IAAI,CACL;iBACF,EACD,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAC1C,CAAC;aACH;iBAAM,IACL,qBAAqB;gBACrB,2CAA2C,EAC3C;gBACA,YAAY,GAAG,IAAI,mCAAmC,CACpD,iCAAiC,CAC/B,qBAAqB,CACnB,qBAAqB,CAAC,MAAM,EAC5B,IAAI,iCAAiC;gBACnC,iDAAiD;gBACjD,EAAE,CACH;qBACE,6BAA6B,CAC5B,OAAO,CAAC,OAAO,EAAE,qBAAqB,CAAC,CACxC;qBACA,KAAK,EAAE,CACX,CACF,CACF,CAAC;aACH;iBAAM;gBACL,MAAM,IAAI,yBAAyB,CACjC,8EAA8E,EAC9E,qBAAqB,CACtB,CAAC;aACH;YAED,qBAAqB;YACrB,IAAI,oBAAoB,EAAE;gBACxB,MAAM,2BAA2B,GAAG,IAAI,wBAAwB,CAC9D,0BAA0B,CAAC,iCAAiC,CAAC,OAAO,CAAC,EACrE,eAAe,CAChB,CAAC;gBACF,MAAM,eAAe,GAAG,uCAAuC,CAC7D,oBAAoB,CAAC,mBAAmB,EACxC;oBACE,oBAAoB,CAAC,QAAQ,CAAC,iCAAiC,CAC7D,oBAAoB,CACrB;iBACF,EACD,oBAAoB,CAAC,gBAAgB,CAAC,eAAe;qBAClD,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAC7C,CAAC;gBACF,2BAA2B,CAAC,gBAAgB,GAAG;oBAC7C,YAAY;oBACZ,eAAe;iBAChB,CAAC;gBAEF,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;aAC3D;iBAAM;gBACL,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;aACtC;QACH,CAAC,CAAC,CAAC;QACH,eAAe,CAAC,gBAAgB,GAAG;YACjC,mBAAmB;YACnB,UAAU;YACV,gBAAgB;YAChB,UAAU;SACX,CAAC;QACF,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,GAAG,eAAe,CAAC;KACxD;SAAM,IAAI,eAAe,CAAC,OAAO,CAAC,MAAM,EAAE;QACzC,aAAa;QACb,MAAM,eAAe,GAAG,IAAI,wBAAwB,CAClD,0BAA0B,CAAC,iCAAiC,CAAC,WAAW,CAAC,EACzE,eAAe,CAChB,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,IAAI,YAAY,CACd,eAAe,CAAC,OAAO,CAAC,MAAM,EAC9B,eAAe,CAAC,OAAO,CAAC,MAAM,CAC/B,CACF,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,IAAI,YAAY,CACd,eAAe,CAAC,OAAO,CAAC,MAAM,EAC9B,eAAe,CAAC,OAAO,CAAC,MAAM,CAC/B,CACF,CAAC;QACF,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,qBAAqB,EAAE,EAAE;YACxD,eAAe;YACf,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CACzC,4BAA4B,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC,UAAU,CAAC,CAAC,EAChE,eAAe,CAChB,CAAC;YACF,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;YACvD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAEjC,oBAAoB;YACpB,IAAI,YAAgC,CAAC;YACrC,IACE,qBAAqB,YAAY,uCAAuC,EACxE;gBACA,YAAY,GAAG,uCAAuC,CACpD,qBAAqB,CAAC,mBAAmB,EACzC;oBACE,4BAA4B,CAC1B,qBAAqB,CAAC,uBAAuB,CAAC,kBAAkB,EAChE,qBAAqB,CAAC,uBAAuB,CAAC,iBAAiB,CAChE;iBACF,EACD,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAC1C,CAAC;aACH;iBAAM,IACL,qBAAqB;gBACrB,2CAA2C,EAC3C;gBACA,YAAY,GAAG,IAAI,mCAAmC,CACpD,iCAAiC,CAC/B,qBAAqB,CACnB,qBAAqB,CAAC,MAAM,EAC5B,IAAI,iCAAiC;gBACnC,iDAAiD;gBACjD,EAAE,CACH;qBACE,6BAA6B,CAC5B,OAAO,CAAC,OAAO,EAAE,qBAAqB,CAAC,CACxC;qBACA,KAAK,EAAE,CACX,CACF,CACF,CAAC;aACH;iBAAM;gBACL,MAAM,IAAI,yBAAyB,CACjC,8EAA8E,EAC9E,qBAAqB,CACtB,CAAC;aACH;YACD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,eAAe,CAAC,gBAAgB,GAAG;YACjC,mBAAmB;YACnB,UAAU;YACV,UAAU;SACX,CAAC;QACF,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,GAAG,eAAe,CAAC;KACxD;IAED,oBAAoB;IACpB,gBAAgB,CAAC,eAAe,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC;IAElE,6BAA6B;IAC7B,uBAAuB,CACrB,eAAe,CAAC,sBAAsB,EACtC,cAAc,EACd;QACE,eAAe,EAAE,OAAO,EAAE,wBAAwB;YAChD,CAAC,CAAC,iBAAiB,CAAC,WAAW,CAAC,YAAY;YAC5C,CAAC,CAAC,SAAS;KACd,CACF,CAAC;AACJ,CAAC,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/QueryResultSetModifierState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryResultSetModifierState.d.ts","sourceRoot":"","sources":["../../../../../src/stores/query-builder/fetch-structure/projection/QueryResultSetModifierState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AACpF,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,wCAAwC,CAAC;AAGhG,oBAAY,gBAAgB;IAC1B,GAAG,IAAA;IACH,IAAI,IAAA;CACL;AAED,qBAAa,eAAe;IAC1B,WAAW,EAAE,iCAAiC,CAAC;IAC/C,QAAQ,mBAAwB;gBAEpB,WAAW,EAAE,iCAAiC;IAS1D,cAAc,CAAC,GAAG,EAAE,iCAAiC,GAAG,IAAI;IAI5D,WAAW,CAAC,GAAG,EAAE,gBAAgB,GAAG,IAAI;CAGzC;AAED,qBAAa,2BAA2B;IACtC,eAAe,EAAE,2BAA2B,CAAC;IAC7C,SAAS,UAAS;IAClB,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,UAAS;IACjB,WAAW,EAAE,eAAe,EAAE,CAAM;gBAExB,eAAe,EAAE,2BAA2B;IAcxD,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;IAIhC,QAAQ,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI;IAIvC,cAAc,IAAI,IAAI;IAItB,gBAAgB,CAAC,GAAG,EAAE,eAAe,GAAG,IAAI;IAI5C,aAAa,CAAC,GAAG,EAAE,eAAe,GAAG,IAAI;IAIzC,iBAAiB,IAAI,IAAI;CAK1B"}

package/lib/stores/query-builder/fetch-structure/projection/QueryResultSetModifierState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryResultSetModifierState.js","sourceRoot":"","sources":["../../../../../src/stores/query-builder/fetch-structure/projection/QueryResultSetModifierState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,MAAM,CAAC;AAGlD,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAEnE,MAAM,CAAN,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IAC1B,qDAAG,CAAA;IACH,uDAAI,CAAA;AACN,CAAC,EAHW,gBAAgB,GAAhB,gBAAgB,KAAhB,gBAAgB,QAG3B;AAED,MAAM,OAAO,eAAe;IAC1B,WAAW,CAAoC;IAC/C,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC;IAEhC,YAAY,WAA8C;QACxD,kBAAkB,CAAC,IAAI,EAAE;YACvB,cAAc,EAAE,MAAM;YACtB,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,GAAsC;QACnD,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;IACzB,CAAC;IAED,WAAW,CAAC,GAAqB;QAC/B,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC;IACtB,CAAC;CACF;AAED,MAAM,OAAO,2BAA2B;IACtC,eAAe,CAA8B;IAC7C,SAAS,GAAG,KAAK,CAAC;IAClB,KAAK,CAAsB;IAC3B,QAAQ,GAAG,KAAK,CAAC;IACjB,WAAW,GAAsB,EAAE,CAAC;IAEpC,YAAY,eAA4C;QACtD,kBAAkB,CAAC,IAAI,EAAE;YACvB,eAAe,EAAE,KAAK;YACtB,YAAY,EAAE,MAAM;YACpB,QAAQ,EAAE,MAAM;YAChB,cAAc,EAAE,MAAM;YACtB,gBAAgB,EAAE,MAAM;YACxB,aAAa,EAAE,MAAM;YACrB,iBAAiB,EAAE,MAAM;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IACzC,CAAC;IAED,YAAY,CAAC,GAAY;QACvB,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC;IACvB,CAAC;IAED,QAAQ,CAAC,GAAuB;QAC9B,IAAI,CAAC,KAAK,GAAG,GAAG,KAAK,SAAS,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC;IAC/D,CAAC;IAED,cAAc;QACZ,IAAI,CAAC,QAAQ,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC;IACjC,CAAC;IAED,gBAAgB,CAAC,GAAoB;QACnC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;IACrC,CAAC;IAED,aAAa,CAAC,GAAoB;QAChC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;IACxC,CAAC;IAED,iBAAiB;QACf,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAC/C,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CACrD,CAAC;IACJ,CAAC;CACF"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperator.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,EACV,0BAA0B,EAC1B,SAAS,EACT,wBAAwB,EACxB,IAAI,EACJ,kBAAkB,EAClB,kBAAkB,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,iCAAiC,EAEvC,MAAM,yCAAyC,CAAC;AACjD,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,mCAAmC,CAAC;AAE1F,8BAAsB,6BAA6B;IACjD,QAAQ,CAAC,IAAI,SAAU;IAEvB,QAAQ,CAAC,QAAQ,CACf,qBAAqB,EAAE,iCAAiC,GACvD,MAAM;IAET,QAAQ,CAAC,sBAAsB,CAC7B,qBAAqB,EAAE,iCAAiC,GACvD,OAAO;IAEV,QAAQ,CAAC,wBAAwB,CAC/B,kBAAkB,EAAE,0BAA0B,GAAG,SAAS,EAC1D,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,SAAS,GACf,kBAAkB;IAErB,iCAAiC,CAC/B,oBAAoB,EAAE,gCAAgC,GACrD,kBAAkB;IAarB,QAAQ,CAAC,yBAAyB,CAChC,UAAU,EAAE,wBAAwB,EACpC,WAAW,EAAE,kBAAkB,EAC/B,qBAAqB,EAAE,iCAAiC,GACvD,gCAAgC,GAAG,SAAS;IAE/C;;;OAGG;IACH,aAAa,CAAC,oBAAoB,EAAE,gCAAgC,GAAG,IAAI;CAK5E"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator.js","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperator.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AASlE,OAAO,EAEL,uCAAuC,GACxC,MAAM,yCAAyC,CAAC;AAGjD,MAAM,OAAgB,6BAA6B;IACxC,IAAI,GAAG,IAAI,EAAE,CAAC;IAgBvB,iCAAiC,CAC/B,oBAAsD;QAEtD,OAAO,IAAI,CAAC,wBAAwB,CAClC,oBAAoB,CAAC,qBAAqB;YACxC,uCAAuC;YACvC,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,uBAAuB;iBAC/D,kBAAkB;YACvB,CAAC,CAAC,SAAS,EACb,oBAAoB,CAAC,mBAAmB,EACxC,oBAAoB,CAAC,gBAAgB,CAAC,eAAe,CAAC,iBAAiB;aACpE,iBAAiB,CAAC,KAAK,CAC3B,CAAC;IACJ,CAAC;IAQD;;;OAGG;IACH,aAAa,CAAC,oBAAsD;QAClE,OAAO,oBAAoB,CACzB,oBAAoB,CAAC,qBAAqB,CAAC,aAAa,EAAE,CAC3D,CAAC;IACJ,CAAC;CACF"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperatorLoader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperatorLoader.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperatorLoader.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAYH,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAExF,eAAO,MAAM,wCAAwC,QAC/C,6BAA6B,EAWhC,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperatorLoader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperatorLoader.js","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregateOperatorLoader.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,mCAAmC,EAAE,MAAM,oDAAoD,CAAC;AACzG,OAAO,EAAE,sCAAsC,EAAE,MAAM,uDAAuD,CAAC;AAC/G,OAAO,EAAE,iCAAiC,EAAE,MAAM,kDAAkD,CAAC;AACrG,OAAO,EAAE,qCAAqC,EAAE,MAAM,sDAAsD,CAAC;AAC7G,OAAO,EAAE,+CAA+C,EAAE,MAAM,gEAAgE,CAAC;AACjI,OAAO,EAAE,2CAA2C,EAAE,MAAM,4DAA4D,CAAC;AACzH,OAAO,EAAE,2CAA2C,EAAE,MAAM,4DAA4D,CAAC;AACzH,OAAO,EAAE,iCAAiC,EAAE,MAAM,kDAAkD,CAAC;AACrG,OAAO,EAAE,iCAAiC,EAAE,MAAM,kDAAkD,CAAC;AACrG,OAAO,EAAE,wCAAwC,EAAE,MAAM,yDAAyD,CAAC;AAGnH,MAAM,CAAC,MAAM,wCAAwC,GACnD,GAAoC,EAAE,CAAC;IACrC,IAAI,mCAAmC,EAAE;IACzC,IAAI,2CAA2C,EAAE;IACjD,IAAI,sCAAsC,EAAE;IAC5C,IAAI,iCAAiC,EAAE;IACvC,IAAI,qCAAqC,EAAE;IAC3C,IAAI,iCAAiC,EAAE;IACvC,IAAI,iCAAiC,EAAE;IACvC,IAAI,+CAA+C,EAAE;IACrD,IAAI,2CAA2C,EAAE;IACjD,IAAI,wCAAwC,EAAE;CAC/C,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregationState.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAEhD,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AACrF,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AACxF,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,yCAAyC,CAAC;AAEjG,qBAAa,gCAAgC;IAC3C,QAAQ,CAAC,IAAI,SAAU;IACvB,gBAAgB,EAAE,4BAA4B,CAAC;IAC/C,qBAAqB,EAAE,iCAAiC,CAAC;IACzD,mBAAmB,EAAE,MAAM,CAAgC;IAC3D,QAAQ,EAAE,6BAA6B,CAAC;gBAGtC,gBAAgB,EAAE,4BAA4B,EAC9C,qBAAqB,EAAE,iCAAiC,EACxD,QAAQ,EAAE,6BAA6B;IAezC,IAAI,UAAU,IAAI,MAAM,CAEvB;IAED,cAAc,CAAC,GAAG,EAAE,iCAAiC,GAAG,IAAI;IAI5D,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAIzC,WAAW,CAAC,GAAG,EAAE,6BAA6B,GAAG,IAAI;IAIrD,aAAa,IAAI,IAAI;CAGtB;AAED,qBAAa,4BAA4B;IACvC,eAAe,EAAE,2BAA2B,CAAC;IAC7C,SAAS,EAAE,6BAA6B,EAAE,CAAM;IAChD,OAAO,EAAE,gCAAgC,EAAE,CAAM;gBAG/C,eAAe,EAAE,2BAA2B,EAC5C,SAAS,EAAE,6BAA6B,EAAE;IAY5C,YAAY,CAAC,GAAG,EAAE,gCAAgC,GAAG,IAAI;IAIzD,SAAS,CAAC,GAAG,EAAE,gCAAgC,GAAG,IAAI;IAItD,6BAA6B,CAC3B,GAAG,EAAE,6BAA6B,GAAG,SAAS,EAC9C,qBAAqB,EAAE,iCAAiC,GACvD,IAAI;CA4CR"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationState.js

@@ -0,0 +1,103 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { action, makeAutoObservable } from 'mobx';
+import { uuid, deleteEntry, addUniqueEntry } from '@finos/legend-shared';
+import { DEFAULT_LAMBDA_VARIABLE_NAME } from '../../../QueryBuilderConfig.js';
+export class QueryBuilderAggregateColumnState {
+    uuid = uuid();
+    aggregationState;
+    projectionColumnState;
+    lambdaParameterName = DEFAULT_LAMBDA_VARIABLE_NAME;
+    operator;
+    constructor(aggregationState, projectionColumnState, operator) {
+        makeAutoObservable(this, {
+            uuid: false,
+            aggregationState: false,
+            setColumnState: action,
+            setLambdaParameterName: action,
+            setOperator: action,
+        });
+        this.aggregationState = aggregationState;
+        this.projectionColumnState = projectionColumnState;
+        this.operator = operator;
+    }
+    get columnName() {
+        return this.projectionColumnState.columnName;
+    }
+    setColumnState(val) {
+        this.projectionColumnState = val;
+    }
+    setLambdaParameterName(val) {
+        this.lambdaParameterName = val;
+    }
+    setOperator(val) {
+        this.operator = val;
+    }
+    getReturnType() {
+        return this.operator.getReturnType(this);
+    }
+}
+export class QueryBuilderAggregationState {
+    projectionState;
+    operators = [];
+    columns = [];
+    constructor(projectionState, operators) {
+        makeAutoObservable(this, {
+            projectionState: false,
+            removeColumn: action,
+            addColumn: action,
+        });
+        this.projectionState = projectionState;
+        this.operators = operators;
+    }
+    removeColumn(val) {
+        deleteEntry(this.columns, val);
+    }
+    addColumn(val) {
+        addUniqueEntry(this.columns, val);
+    }
+    changeColumnAggregateOperator(val, projectionColumnState) {
+        const aggregateColumnState = this.columns.find((column) => column.projectionColumnState === projectionColumnState);
+        const aggreateOperators = this.operators.filter((op) => op.isCompatibleWithColumn(projectionColumnState));
+        if (val) {
+            if (!aggreateOperators.includes(val)) {
+                return;
+            }
+            if (aggregateColumnState) {
+                aggregateColumnState.setOperator(val);
+            }
+            else {
+                const newAggregateColumnState = new QueryBuilderAggregateColumnState(this, projectionColumnState, val);
+                newAggregateColumnState.setOperator(val);
+                this.addColumn(newAggregateColumnState);
+                // automatically move the column to the end
+                // as aggregate column should always be the last one
+                // NOTE: unless we do `olap` aggregation
+                // See https://github.com/finos/legend-studio/issues/253
+                this.projectionState.moveColumn(this.projectionState.columns.indexOf(projectionColumnState), this.projectionState.columns.length - 1);
+            }
+        }
+        else {
+            if (aggregateColumnState) {
+                // automatically move the column to the last position before the aggregate columns
+                // NOTE: `moveColumn` will take care of this placement calculation
+                this.projectionState.moveColumn(this.projectionState.columns.indexOf(projectionColumnState), 0);
+                this.removeColumn(aggregateColumnState);
+            }
+        }
+    }
+}
+//# sourceMappingURL=QueryBuilderAggregationState.js.map

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregationState.js","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,MAAM,CAAC;AAClD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEzE,OAAO,EAAE,4BAA4B,EAAE,MAAM,gCAAgC,CAAC;AAK9E,MAAM,OAAO,gCAAgC;IAClC,IAAI,GAAG,IAAI,EAAE,CAAC;IACvB,gBAAgB,CAA+B;IAC/C,qBAAqB,CAAoC;IACzD,mBAAmB,GAAW,4BAA4B,CAAC;IAC3D,QAAQ,CAAgC;IAExC,YACE,gBAA8C,EAC9C,qBAAwD,EACxD,QAAuC;QAEvC,kBAAkB,CAAC,IAAI,EAAE;YACvB,IAAI,EAAE,KAAK;YACX,gBAAgB,EAAE,KAAK;YACvB,cAAc,EAAE,MAAM;YACtB,sBAAsB,EAAE,MAAM;YAC9B,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QAEH,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,CAAC;QACnD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC;IAC/C,CAAC;IAED,cAAc,CAAC,GAAsC;QACnD,IAAI,CAAC,qBAAqB,GAAG,GAAG,CAAC;IACnC,CAAC;IAED,sBAAsB,CAAC,GAAW;QAChC,IAAI,CAAC,mBAAmB,GAAG,GAAG,CAAC;IACjC,CAAC;IAED,WAAW,CAAC,GAAkC;QAC5C,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC;IACtB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,OAAO,4BAA4B;IACvC,eAAe,CAA8B;IAC7C,SAAS,GAAoC,EAAE,CAAC;IAChD,OAAO,GAAuC,EAAE,CAAC;IAEjD,YACE,eAA4C,EAC5C,SAA0C;QAE1C,kBAAkB,CAAC,IAAI,EAAE;YACvB,eAAe,EAAE,KAAK;YACtB,YAAY,EAAE,MAAM;YACpB,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,YAAY,CAAC,GAAqC;QAChD,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACjC,CAAC;IAED,SAAS,CAAC,GAAqC;QAC7C,cAAc,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACpC,CAAC;IAED,6BAA6B,CAC3B,GAA8C,EAC9C,qBAAwD;QAExD,MAAM,oBAAoB,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAC5C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,qBAAqB,KAAK,qBAAqB,CACnE,CAAC;QACF,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CACrD,EAAE,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CACjD,CAAC;QACF,IAAI,GAAG,EAAE;YACP,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;gBACpC,OAAO;aACR;YACD,IAAI,oBAAoB,EAAE;gBACxB,oBAAoB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;aACvC;iBAAM;gBACL,MAAM,uBAAuB,GAAG,IAAI,gCAAgC,CAClE,IAAI,EACJ,qBAAqB,EACrB,GAAG,CACJ,CAAC;gBACF,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;gBACzC,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;gBAExC,2CAA2C;gBAC3C,oDAAoD;gBACpD,wCAAwC;gBACxC,wDAAwD;gBACxD,IAAI,CAAC,eAAe,CAAC,UAAU,CAC7B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAC3D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CACxC,CAAC;aACH;SACF;aAAM;YACL,IAAI,oBAAoB,EAAE;gBACxB,kFAAkF;gBAClF,kEAAkE;gBAClE,IAAI,CAAC,eAAe,CAAC,UAAU,CAC7B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAC3D,CAAC,CACF,CAAC;gBAEF,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC;aACzC;SACF;IACH,CAAC;CACF"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.d.ts

@@ -0,0 +1,20 @@
+/**
+ * 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 { SimpleFunctionExpression } from '@finos/legend-graph';
+import type { QueryBuilderState } from '../../../QueryBuilderState.js';
+export declare const processTDSAggregateExpression: (expression: SimpleFunctionExpression, parentExpression: SimpleFunctionExpression | undefined, queryBuilderState: QueryBuilderState) => void;
+export declare const processTDSGroupByExpression: (expression: SimpleFunctionExpression, queryBuilderState: QueryBuilderState) => void;
+//# sourceMappingURL=QueryBuilderAggregationStateBuilder.d.ts.map

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregationStateBuilder.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAIL,wBAAwB,EAEzB,MAAM,qBAAqB,CAAC;AAW7B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAMvE,eAAO,MAAM,6BAA6B,eAC5B,wBAAwB,oBAClB,wBAAwB,GAAG,SAAS,qBACnC,iBAAiB,KACnC,IA4FF,CAAC;AAEF,eAAO,MAAM,2BAA2B,eAC1B,wBAAwB,qBACjB,iBAAiB,KACnC,IAuFF,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.js

@@ -0,0 +1,96 @@
+/**
+ * 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 { CollectionInstanceValue, LambdaFunctionInstanceValue, matchFunctionName, SimpleFunctionExpression, VariableExpression, } from '@finos/legend-graph';
+import { assertTrue, assertType, guaranteeNonNullable, guaranteeType, isNonNullable, returnUndefOnError, UnsupportedOperationError, } from '@finos/legend-shared';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { QueryBuilderValueSpecificationProcessor } from '../../../QueryBuilderStateBuilder.js';
+import { extractNullableStringFromInstanceValue } from '../../../QueryBuilderValueSpecificationHelper.js';
+import { FETCH_STRUCTURE_IMPLEMENTATION } from '../../QueryBuilderFetchStructureImplementationState.js';
+import { QueryBuilderProjectionState } from '../QueryBuilderProjectionState.js';
+export const processTDSAggregateExpression = (expression, parentExpression, queryBuilderState) => {
+    // check parent expression
+    assertTrue(Boolean(parentExpression &&
+        matchFunctionName(parentExpression.functionName, QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_GROUP_BY)), `Can't process agg() expression: only support agg() used within a groupBy() expression`);
+    // check parameters
+    assertTrue(expression.parametersValues.length === 2, `Can't process agg() expression: agg() expects 2 arguments`);
+    // process column lambda
+    QueryBuilderValueSpecificationProcessor.processChild(
+    // TODO?: do we want to do more validation here for the shape of the column lambda?
+    guaranteeNonNullable(expression.parametersValues[0]), expression, queryBuilderState);
+    // build state
+    if (queryBuilderState.fetchStructureState.implementation instanceof
+        QueryBuilderProjectionState) {
+        const projectionState = queryBuilderState.fetchStructureState.implementation;
+        const aggregationState = projectionState.aggregationState;
+        // NOTE: since we process agg() expressions one by one, we know that the current agg()
+        // always correspond to the last column projection state, based on our processing procedure
+        const projectionColumnState = guaranteeNonNullable(projectionState.columns[projectionState.columns.length - 1]);
+        const aggregateLambda = expression.parametersValues[1];
+        assertType(aggregateLambda, LambdaFunctionInstanceValue, `Can't process agg() expression: agg() expects argument #1 to be a lambda function`);
+        const lambdaFunc = guaranteeNonNullable(aggregateLambda.values[0], `Can't process agg() lambda: agg() lambda function is missing`);
+        assertTrue(lambdaFunc.expressionSequence.length === 1, `Can't process agg() lambda: only support agg() lambda body with 1 expression`);
+        const aggregateColumnExpression = guaranteeType(lambdaFunc.expressionSequence[0], SimpleFunctionExpression, `Can't process agg() lambda: only support agg() lambda body with 1 expression`);
+        assertTrue(lambdaFunc.functionType.parameters.length === 1, `Can't process agg() lambda: only support agg() lambda with 1 parameter`);
+        const lambdaParam = guaranteeType(lambdaFunc.functionType.parameters[0], VariableExpression, `Can't process agg() lambda: only support agg() lambda with 1 parameter`);
+        for (const operator of aggregationState.operators) {
+            // NOTE: this allow plugin author to either return `undefined` or throw error
+            // if there is a problem with building the lambda. Either case, the plugin is
+            // considered as not supporting the lambda.
+            const aggregateColumnState = returnUndefOnError(() => operator.buildAggregateColumnState(aggregateColumnExpression, lambdaParam, projectionColumnState));
+            if (aggregateColumnState) {
+                aggregationState.addColumn(aggregateColumnState);
+                return;
+            }
+        }
+        throw new UnsupportedOperationError(`Can't process aggregate expression function: no compatible aggregate operator processer available from plugins`);
+    }
+};
+export const processTDSGroupByExpression = (expression, queryBuilderState) => {
+    // update fetch-structure
+    queryBuilderState.fetchStructureState.changeImplementation(FETCH_STRUCTURE_IMPLEMENTATION.PROJECTION);
+    // check parameters
+    assertTrue(expression.parametersValues.length === 4, `Can't process groupBy() expression: groupBy() expects 3 arguments`);
+    // check preceding expression
+    const precedingExpression = guaranteeType(expression.parametersValues[0], SimpleFunctionExpression, `Can't process groupBy() expression: only support groupBy() immediately following an expression`);
+    assertTrue(matchFunctionName(precedingExpression.functionName, [
+        QUERY_BUILDER_SUPPORTED_FUNCTIONS.GET_ALL,
+        QUERY_BUILDER_SUPPORTED_FUNCTIONS.FILTER,
+    ]), `Can't process groupBy() expression: only support groupBy() immediately following either getAll() or filter()`);
+    QueryBuilderValueSpecificationProcessor.process(precedingExpression, queryBuilderState);
+    // process columns
+    const columnExpressions = expression.parametersValues[1];
+    assertType(columnExpressions, CollectionInstanceValue, `Can't process groupBy() expression: groupBy() expects argument #1 to be a collection`);
+    columnExpressions.values.map((value) => QueryBuilderValueSpecificationProcessor.processChild(value, expression, queryBuilderState));
+    // process aggregations
+    const aggregateLambdas = expression.parametersValues[2];
+    assertType(aggregateLambdas, CollectionInstanceValue, `Can't process groupBy() expression: groupBy() expects argument #2 to be a collection`);
+    aggregateLambdas.values.map((value) => QueryBuilderValueSpecificationProcessor.processChild(value, expression, queryBuilderState));
+    // process column aliases
+    const columnAliases = expression.parametersValues[3];
+    assertType(columnAliases, CollectionInstanceValue, `Can't process groupBy() expression: groupBy() expects argument #3 to be a collection`);
+    assertTrue(columnAliases.values.length ===
+        columnExpressions.values.length + aggregateLambdas.values.length, `Can't process groupBy() expression: number of aliases does not match the number of columns`);
+    const aliases = columnAliases.values
+        .map(extractNullableStringFromInstanceValue)
+        .filter(isNonNullable);
+    // build state
+    if (queryBuilderState.fetchStructureState.implementation instanceof
+        QueryBuilderProjectionState) {
+        const projectionState = queryBuilderState.fetchStructureState.implementation;
+        projectionState.columns.forEach((column, idx) => column.setColumnName(aliases[idx]));
+    }
+};
+//# sourceMappingURL=QueryBuilderAggregationStateBuilder.js.map

package/lib/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregationStateBuilder.js","sourceRoot":"","sources":["../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/QueryBuilderAggregationStateBuilder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,EACvB,2BAA2B,EAC3B,iBAAiB,EACjB,wBAAwB,EACxB,kBAAkB,GACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,UAAU,EACV,UAAU,EACV,oBAAoB,EACpB,aAAa,EACb,aAAa,EACb,kBAAkB,EAClB,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,iCAAiC,EAAE,MAAM,+DAA+D,CAAC;AAElH,OAAO,EAAE,uCAAuC,EAAE,MAAM,sCAAsC,CAAC;AAC/F,OAAO,EAAE,sCAAsC,EAAE,MAAM,kDAAkD,CAAC;AAC1G,OAAO,EAAE,8BAA8B,EAAE,MAAM,wDAAwD,CAAC;AACxG,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAEhF,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAC3C,UAAoC,EACpC,gBAAsD,EACtD,iBAAoC,EAC9B,EAAE;IACR,0BAA0B;IAC1B,UAAU,CACR,OAAO,CACL,gBAAgB;QACd,iBAAiB,CACf,gBAAgB,CAAC,YAAY,EAC7B,iCAAiC,CAAC,YAAY,CAC/C,CACJ,EACD,uFAAuF,CACxF,CAAC;IAEF,mBAAmB;IACnB,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,EACxC,2DAA2D,CAC5D,CAAC;IAEF,wBAAwB;IACxB,uCAAuC,CAAC,YAAY;IAClD,mFAAmF;IACnF,oBAAoB,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,EACpD,UAAU,EACV,iBAAiB,CAClB,CAAC;IAEF,cAAc;IACd,IACE,iBAAiB,CAAC,mBAAmB,CAAC,cAAc;QACpD,2BAA2B,EAC3B;QACA,MAAM,eAAe,GACnB,iBAAiB,CAAC,mBAAmB,CAAC,cAAc,CAAC;QACvD,MAAM,gBAAgB,GAAG,eAAe,CAAC,gBAAgB,CAAC;QAC1D,sFAAsF;QACtF,2FAA2F;QAC3F,MAAM,qBAAqB,GAAG,oBAAoB,CAChD,eAAe,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAC5D,CAAC;QACF,MAAM,eAAe,GAAG,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;QACvD,UAAU,CACR,eAAe,EACf,2BAA2B,EAC3B,mFAAmF,CACpF,CAAC;QAEF,MAAM,UAAU,GAAG,oBAAoB,CACrC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EACzB,8DAA8D,CAC/D,CAAC;QACF,UAAU,CACR,UAAU,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAC1C,8EAA8E,CAC/E,CAAC;QACF,MAAM,yBAAyB,GAAG,aAAa,CAC7C,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAChC,wBAAwB,EACxB,8EAA8E,CAC/E,CAAC;QAEF,UAAU,CACR,UAAU,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAC/C,wEAAwE,CACzE,CAAC;QAEF,MAAM,WAAW,GAAG,aAAa,CAC/B,UAAU,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,EACrC,kBAAkB,EAClB,wEAAwE,CACzE,CAAC;QAEF,KAAK,MAAM,QAAQ,IAAI,gBAAgB,CAAC,SAAS,EAAE;YACjD,6EAA6E;YAC7E,6EAA6E;YAC7E,2CAA2C;YAC3C,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,GAAG,EAAE,CACnD,QAAQ,CAAC,yBAAyB,CAChC,yBAAyB,EACzB,WAAW,EACX,qBAAqB,CACtB,CACF,CAAC;YACF,IAAI,oBAAoB,EAAE;gBACxB,gBAAgB,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;gBACjD,OAAO;aACR;SACF;QACD,MAAM,IAAI,yBAAyB,CACjC,gHAAgH,CACjH,CAAC;KACH;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,UAAoC,EACpC,iBAAoC,EAC9B,EAAE;IACR,yBAAyB;IACzB,iBAAiB,CAAC,mBAAmB,CAAC,oBAAoB,CACxD,8BAA8B,CAAC,UAAU,CAC1C,CAAC;IAEF,mBAAmB;IACnB,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,EACxC,mEAAmE,CACpE,CAAC;IAEF,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,aAAa,CACvC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAC9B,wBAAwB,EACxB,gGAAgG,CACjG,CAAC;IACF,UAAU,CACR,iBAAiB,CAAC,mBAAmB,CAAC,YAAY,EAAE;QAClD,iCAAiC,CAAC,OAAO;QACzC,iCAAiC,CAAC,MAAM;KACzC,CAAC,EACF,8GAA8G,CAC/G,CAAC;IACF,uCAAuC,CAAC,OAAO,CAC7C,mBAAmB,EACnB,iBAAiB,CAClB,CAAC;IAEF,kBAAkB;IAClB,MAAM,iBAAiB,GAAG,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;IACzD,UAAU,CACR,iBAAiB,EACjB,uBAAuB,EACvB,sFAAsF,CACvF,CAAC;IACF,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACrC,uCAAuC,CAAC,YAAY,CAClD,KAAK,EACL,UAAU,EACV,iBAAiB,CAClB,CACF,CAAC;IAEF,uBAAuB;IACvB,MAAM,gBAAgB,GAAG,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;IACxD,UAAU,CACR,gBAAgB,EAChB,uBAAuB,EACvB,sFAAsF,CACvF,CAAC;IACF,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACpC,uCAAuC,CAAC,YAAY,CAClD,KAAK,EACL,UAAU,EACV,iBAAiB,CAClB,CACF,CAAC;IAEF,yBAAyB;IACzB,MAAM,aAAa,GAAG,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;IACrD,UAAU,CACR,aAAa,EACb,uBAAuB,EACvB,sFAAsF,CACvF,CAAC;IACF,UAAU,CACR,aAAa,CAAC,MAAM,CAAC,MAAM;QACzB,iBAAiB,CAAC,MAAM,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC,MAAM,EAClE,4FAA4F,CAC7F,CAAC;IACF,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM;SACjC,GAAG,CAAC,sCAAsC,CAAC;SAC3C,MAAM,CAAC,aAAa,CAAC,CAAC;IAEzB,cAAc;IACd,IACE,iBAAiB,CAAC,mBAAmB,CAAC,cAAc;QACpD,2BAA2B,EAC3B;QACA,MAAM,eAAe,GACnB,iBAAiB,CAAC,mBAAmB,CAAC,cAAc,CAAC;QACvD,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAC9C,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,CAAW,CAAC,CAC7C,CAAC;KACH;AACH,CAAC,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperatorHelper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperatorHelper.d.ts","sourceRoot":"","sources":["../../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperatorHelper.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,SAAS,EACd,KAAK,kBAAkB,EACvB,wBAAwB,EAGxB,kBAAkB,EAEnB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,gCAAgC,EAAE,MAAM,oCAAoC,CAAC;AACtF,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,4CAA4C,CAAC;AACpG,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AAEzF,eAAO,MAAM,wBAAwB,6BACT,MAAM,SACzB,SAAS,gBACF,MAAM,KACnB,kBAYF,CAAC;AAEF,eAAO,MAAM,yBAAyB,0BACb,iCAAiC,eAC3C,kBAAkB,cACnB,wBAAwB,4BACV,MAAM,YACtB,6BAA6B,KACtC,gCAAgC,GAAG,SAsDrC,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperatorHelper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperatorHelper.js","sourceRoot":"","sources":["../../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperatorHelper.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAGL,wBAAwB,EACxB,0BAA0B,EAC1B,iBAAiB,EACjB,kBAAkB,EAClB,yBAAyB,GAC1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,gCAAgC,EAAE,MAAM,oCAAoC,CAAC;AAItF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACtC,wBAAgC,EAChC,KAAgB,EAChB,YAAoB,EACA,EAAE;IACtB,MAAM,eAAe,GAAG,KAAK,CAAC,sBAAsB,CAClD,yBAAyB,CAAC,GAAG,CAC9B,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,wBAAwB,CAC7C,0BAA0B,CAAC,wBAAwB,CAAC,EACpD,eAAe,CAChB,CAAC;IACF,UAAU,CAAC,gBAAgB,CAAC,IAAI,CAC9B,IAAI,kBAAkB,CAAC,YAAY,EAAE,eAAe,CAAC,CACtD,CAAC;IACF,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,qBAAwD,EACxD,WAA+B,EAC/B,UAAoC,EACpC,wBAAgC,EAChC,QAAuC,EACO,EAAE;IAChD,IAAI,iBAAiB,CAAC,UAAU,CAAC,YAAY,EAAE,wBAAwB,CAAC,EAAE;QACxE,MAAM,oBAAoB,GAAG,IAAI,gCAAgC,CAC/D,qBAAqB,CAAC,eAAe,CAAC,gBAAgB,EACtD,qBAAqB,EACrB,QAAQ,CACT,CAAC;QACF,oBAAoB,CAAC,sBAAsB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAE9D,UAAU,CACR,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,EACxC,iBAAiB,0BAA0B,CACzC,wBAAwB,CACzB,kBAAkB,0BAA0B,CAC3C,wBAAwB,CACzB,wBAAwB,CAC1B,CAAC;QAEF,WAAW;QACX,MAAM,kBAAkB,GAAG,aAAa,CACtC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAC9B,kBAAkB,EAClB,iBAAiB,0BAA0B,CACzC,wBAAwB,CACzB,+BAA+B,0BAA0B,CACxD,wBAAwB,CACzB,gDAAgD,CAClD,CAAC;QACF,UAAU,CACR,oBAAoB,CAAC,mBAAmB,KAAK,kBAAkB,CAAC,IAAI,EACpE,iBAAiB,0BAA0B,CACzC,wBAAwB,CACzB,wDACC,kBAAkB,CAAC,IACrB,gCACE,oBAAoB,CAAC,mBACvB,GAAG,CACJ,CAAC;QAEF,WAAW;QACX,UAAU,CACR,QAAQ,CAAC,sBAAsB,CAC7B,oBAAoB,CAAC,qBAAqB,CAC3C,EACD,iBAAiB,0BAA0B,CACzC,wBAAwB,CACzB,yDAAyD,CAC3D,CAAC;QACF,oBAAoB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE3C,OAAO,oBAAoB,CAAC;KAC7B;IAED,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Average.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator_Average.d.ts","sourceRoot":"","sources":["../../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Average.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,wBAAwB,EAC7B,KAAK,kBAAkB,EACvB,KAAK,SAAS,EACd,KAAK,0BAA0B,EAC/B,KAAK,IAAI,EACV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,oCAAoC,CAAC;AAC3F,OAAO,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AACpF,OAAO,EACL,KAAK,iCAAiC,EAEvC,MAAM,4CAA4C,CAAC;AAOpD,qBAAa,qCAAsC,SAAQ,6BAA6B;IACtF,QAAQ,CAAC,qBAAqB,EAAE,iCAAiC,GAAG,MAAM;IAI1E,sBAAsB,CACpB,qBAAqB,EAAE,iCAAiC,GACvD,OAAO;IAmBV,wBAAwB,CACtB,kBAAkB,EAAE,0BAA0B,GAAG,SAAS,EAC1D,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,SAAS,GACf,kBAAkB;IAQrB,yBAAyB,CACvB,UAAU,EAAE,wBAAwB,EACpC,WAAW,EAAE,kBAAkB,EAC/B,qBAAqB,EAAE,iCAAiC,GACvD,gCAAgC,GAAG,SAAS;IAUtC,aAAa,CACpB,oBAAoB,EAAE,gCAAgC,GACrD,IAAI;CAMR"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Average.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 { PRIMITIVE_TYPE, } from '@finos/legend-graph';
+import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
+import { QueryBuilderSimpleProjectionColumnState, } from '../../QueryBuilderProjectionColumnState.js';
+import { buildAggregateColumnState, buildAggregateExpression, } from './QueryBuilderAggregateOperatorHelper.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+export class QueryBuilderAggregateOperator_Average extends QueryBuilderAggregateOperator {
+    getLabel(projectionColumnState) {
+        return 'average';
+    }
+    isCompatibleWithColumn(projectionColumnState) {
+        if (projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState) {
+            const propertyType = projectionColumnState.propertyExpressionState.propertyExpression.func
+                .genericType.value.rawType;
+            return [
+                PRIMITIVE_TYPE.NUMBER,
+                PRIMITIVE_TYPE.INTEGER,
+                PRIMITIVE_TYPE.DECIMAL,
+                PRIMITIVE_TYPE.FLOAT,
+            ].includes(propertyType.path);
+        }
+        return true;
+    }
+    buildAggregateExpression(propertyExpression, variableName, graph) {
+        return buildAggregateExpression(QUERY_BUILDER_SUPPORTED_FUNCTIONS.AVERAGE, graph, variableName);
+    }
+    buildAggregateColumnState(expression, lambdaParam, projectionColumnState) {
+        return buildAggregateColumnState(projectionColumnState, lambdaParam, expression, QUERY_BUILDER_SUPPORTED_FUNCTIONS.AVERAGE, this);
+    }
+    getReturnType(aggregateColumnState) {
+        const graph = aggregateColumnState.aggregationState.projectionState.queryBuilderState
+            .graphManagerState.graph;
+        return graph.getType(PRIMITIVE_TYPE.FLOAT);
+    }
+}
+//# sourceMappingURL=QueryBuilderAggregateOperator_Average.js.map

package/lib/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Average.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator_Average.js","sourceRoot":"","sources":["../../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Average.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,cAAc,GAOf,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AACpF,OAAO,EAEL,uCAAuC,GACxC,MAAM,4CAA4C,CAAC;AACpD,OAAO,EACL,yBAAyB,EACzB,wBAAwB,GACzB,MAAM,0CAA0C,CAAC;AAClD,OAAO,EAAE,iCAAiC,EAAE,MAAM,kEAAkE,CAAC;AAErH,MAAM,OAAO,qCAAsC,SAAQ,6BAA6B;IACtF,QAAQ,CAAC,qBAAwD;QAC/D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,sBAAsB,CACpB,qBAAwD;QAExD,IACE,qBAAqB,YAAY,uCAAuC,EACxE;YACA,MAAM,YAAY,GAChB,qBAAqB,CAAC,uBAAuB,CAAC,kBAAkB,CAAC,IAAI;iBAClE,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;YAC/B,OACE;gBACE,cAAc,CAAC,MAAM;gBACrB,cAAc,CAAC,OAAO;gBACtB,cAAc,CAAC,OAAO;gBACtB,cAAc,CAAC,KAAK;aAEvB,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;SAC/B;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,wBAAwB,CACtB,kBAA0D,EAC1D,YAAoB,EACpB,KAAgB;QAEhB,OAAO,wBAAwB,CAC7B,iCAAiC,CAAC,OAAO,EACzC,KAAK,EACL,YAAY,CACb,CAAC;IACJ,CAAC;IAED,yBAAyB,CACvB,UAAoC,EACpC,WAA+B,EAC/B,qBAAwD;QAExD,OAAO,yBAAyB,CAC9B,qBAAqB,EACrB,WAAW,EACX,UAAU,EACV,iCAAiC,CAAC,OAAO,EACzC,IAAI,CACL,CAAC;IACJ,CAAC;IAEQ,aAAa,CACpB,oBAAsD;QAEtD,MAAM,KAAK,GACT,oBAAoB,CAAC,gBAAgB,CAAC,eAAe,CAAC,iBAAiB;aACpE,iBAAiB,CAAC,KAAK,CAAC;QAC7B,OAAO,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC;CACF"}

package/lib/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Count.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator_Count.d.ts","sourceRoot":"","sources":["../../../../../../../src/stores/query-builder/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Count.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAEL,KAAK,IAAI,EACT,KAAK,kBAAkB,EACvB,KAAK,wBAAwB,EAC7B,KAAK,kBAAkB,EACvB,KAAK,SAAS,EACd,KAAK,0BAA0B,EAEhC,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,oCAAoC,CAAC;AAC3F,OAAO,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AACpF,OAAO,EACL,KAAK,iCAAiC,EAEvC,MAAM,4CAA4C,CAAC;AAOpD,qBAAa,mCAAoC,SAAQ,6BAA6B;IACpF,QAAQ,CAAC,qBAAqB,EAAE,iCAAiC,GAAG,MAAM;IAI1E,sBAAsB,CACpB,qBAAqB,EAAE,iCAAiC,GACvD,OAAO;IA0BV,wBAAwB,CACtB,kBAAkB,EAAE,0BAA0B,GAAG,SAAS,EAC1D,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,SAAS,GACf,kBAAkB;IAQrB,yBAAyB,CACvB,UAAU,EAAE,wBAAwB,EACpC,WAAW,EAAE,kBAAkB,EAC/B,qBAAqB,EAAE,iCAAiC,GACvD,gCAAgC,GAAG,SAAS;IAUtC,aAAa,CACpB,oBAAoB,EAAE,gCAAgC,GACrD,IAAI;CAMR"}