@finos/legend-query-builder 0.0.1

package/lib/stores/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/fetch-structure/projection/QueryBuilderProjectionValueSpecificationBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderProjectionValueSpecificationBuilder.js","sourceRoot":"","sources":["../../../../src/stores/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,yDAAyD,CAAC;AAC5G,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/fetch-structure/projection/QueryResultSetModifierState.d.ts

@@ -0,0 +1,43 @@
+/**
+ * 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 { QueryBuilderProjectionState } from './QueryBuilderProjectionState.js';
+import type { QueryBuilderProjectionColumnState } from './QueryBuilderProjectionColumnState.js';
+export declare enum COLUMN_SORT_TYPE {
+    ASC = 0,
+    DESC = 1
+}
+export declare class SortColumnState {
+    columnState: QueryBuilderProjectionColumnState;
+    sortType: COLUMN_SORT_TYPE;
+    constructor(columnState: QueryBuilderProjectionColumnState);
+    setColumnState(val: QueryBuilderProjectionColumnState): void;
+    setSortType(val: COLUMN_SORT_TYPE): void;
+}
+export declare class QueryResultSetModifierState {
+    projectionState: QueryBuilderProjectionState;
+    showModal: boolean;
+    limit?: number | undefined;
+    distinct: boolean;
+    sortColumns: SortColumnState[];
+    constructor(projectionState: QueryBuilderProjectionState);
+    setShowModal(val: boolean): void;
+    setLimit(val: number | undefined): void;
+    toggleDistinct(): void;
+    deleteSortColumn(val: SortColumnState): void;
+    addSortColumn(val: SortColumnState): void;
+    updateSortColumns(): void;
+}
+//# sourceMappingURL=QueryResultSetModifierState.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryResultSetModifierState.d.ts","sourceRoot":"","sources":["../../../../src/stores/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/fetch-structure/projection/QueryResultSetModifierState.js

@@ -0,0 +1,77 @@
+/**
+ * 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 { addUniqueEntry, deleteEntry } from '@finos/legend-shared';
+export var COLUMN_SORT_TYPE;
+(function (COLUMN_SORT_TYPE) {
+    COLUMN_SORT_TYPE[COLUMN_SORT_TYPE["ASC"] = 0] = "ASC";
+    COLUMN_SORT_TYPE[COLUMN_SORT_TYPE["DESC"] = 1] = "DESC";
+})(COLUMN_SORT_TYPE = COLUMN_SORT_TYPE || (COLUMN_SORT_TYPE = {}));
+export class SortColumnState {
+    columnState;
+    sortType = COLUMN_SORT_TYPE.ASC;
+    constructor(columnState) {
+        makeAutoObservable(this, {
+            setColumnState: action,
+            setSortType: action,
+        });
+        this.columnState = columnState;
+    }
+    setColumnState(val) {
+        this.columnState = val;
+    }
+    setSortType(val) {
+        this.sortType = val;
+    }
+}
+export class QueryResultSetModifierState {
+    projectionState;
+    showModal = false;
+    limit;
+    distinct = false;
+    sortColumns = [];
+    constructor(projectionState) {
+        makeAutoObservable(this, {
+            projectionState: false,
+            setShowModal: action,
+            setLimit: action,
+            toggleDistinct: action,
+            deleteSortColumn: action,
+            addSortColumn: action,
+            updateSortColumns: action,
+        });
+        this.projectionState = projectionState;
+    }
+    setShowModal(val) {
+        this.showModal = val;
+    }
+    setLimit(val) {
+        this.limit = val === undefined || val <= 0 ? undefined : val;
+    }
+    toggleDistinct() {
+        this.distinct = !this.distinct;
+    }
+    deleteSortColumn(val) {
+        deleteEntry(this.sortColumns, val);
+    }
+    addSortColumn(val) {
+        addUniqueEntry(this.sortColumns, val);
+    }
+    updateSortColumns() {
+        this.sortColumns = this.sortColumns.filter((e) => this.projectionState.columns.includes(e.columnState));
+    }
+}
+//# sourceMappingURL=QueryResultSetModifierState.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryResultSetModifierState.js","sourceRoot":"","sources":["../../../../src/stores/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/fetch-structure/projection/aggregation/QueryBuilderAggregateOperator.d.ts

@@ -0,0 +1,32 @@
+/**
+ * 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 { AbstractPropertyExpression, PureModel, SimpleFunctionExpression, Type, ValueSpecification, VariableExpression } from '@finos/legend-graph';
+import { type QueryBuilderProjectionColumnState } from '../QueryBuilderProjectionColumnState.js';
+import type { QueryBuilderAggregateColumnState } from './QueryBuilderAggregationState.js';
+export declare abstract class QueryBuilderAggregateOperator {
+    readonly uuid: string;
+    abstract getLabel(projectionColumnState: QueryBuilderProjectionColumnState): string;
+    abstract isCompatibleWithColumn(projectionColumnState: QueryBuilderProjectionColumnState): boolean;
+    abstract buildAggregateExpression(propertyExpression: AbstractPropertyExpression | undefined, variableName: string, graph: PureModel): ValueSpecification;
+    buildAggregateExpressionFromState(aggregateColumnState: QueryBuilderAggregateColumnState): ValueSpecification;
+    abstract buildAggregateColumnState(expression: SimpleFunctionExpression, lambdaParam: VariableExpression, projectionColumnState: QueryBuilderProjectionColumnState): QueryBuilderAggregateColumnState | undefined;
+    /**
+     * Returns the expected return type of the operator.
+     * defaults to using the return type of the projection column state which is being aggregated.
+     */
+    getReturnType(aggregateColumnState: QueryBuilderAggregateColumnState): Type;
+}
+//# sourceMappingURL=QueryBuilderAggregateOperator.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator.d.ts","sourceRoot":"","sources":["../../../../../src/stores/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/fetch-structure/projection/aggregation/QueryBuilderAggregateOperator.js

@@ -0,0 +1,36 @@
+/**
+ * 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 { uuid, guaranteeNonNullable } from '@finos/legend-shared';
+import { QueryBuilderSimpleProjectionColumnState, } from '../QueryBuilderProjectionColumnState.js';
+export class QueryBuilderAggregateOperator {
+    uuid = uuid();
+    buildAggregateExpressionFromState(aggregateColumnState) {
+        return this.buildAggregateExpression(aggregateColumnState.projectionColumnState instanceof
+            QueryBuilderSimpleProjectionColumnState
+            ? aggregateColumnState.projectionColumnState.propertyExpressionState
+                .propertyExpression
+            : undefined, aggregateColumnState.lambdaParameterName, aggregateColumnState.aggregationState.projectionState.queryBuilderState
+            .graphManagerState.graph);
+    }
+    /**
+     * Returns the expected return type of the operator.
+     * defaults to using the return type of the projection column state which is being aggregated.
+     */
+    getReturnType(aggregateColumnState) {
+        return guaranteeNonNullable(aggregateColumnState.projectionColumnState.getReturnType());
+    }
+}
+//# sourceMappingURL=QueryBuilderAggregateOperator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperator.js","sourceRoot":"","sources":["../../../../../src/stores/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/fetch-structure/projection/aggregation/QueryBuilderAggregateOperatorLoader.d.ts

@@ -0,0 +1,18 @@
+/**
+ * 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 { QueryBuilderAggregateOperator } from './QueryBuilderAggregateOperator.js';
+export declare const getQueryBuilderCoreAggregrationOperators: () => QueryBuilderAggregateOperator[];
+//# sourceMappingURL=QueryBuilderAggregateOperatorLoader.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperatorLoader.d.ts","sourceRoot":"","sources":["../../../../../src/stores/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/fetch-structure/projection/aggregation/QueryBuilderAggregateOperatorLoader.js

@@ -0,0 +1,38 @@
+/**
+ * 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 { QueryBuilderAggregateOperator_Count } from './operators/QueryBuilderAggregateOperator_Count.js';
+import { QueryBuilderAggregateOperator_Distinct } from './operators/QueryBuilderAggregateOperator_Distinct.js';
+import { QueryBuilderAggregateOperator_Sum } from './operators/QueryBuilderAggregateOperator_Sum.js';
+import { QueryBuilderAggregateOperator_Average } from './operators/QueryBuilderAggregateOperator_Average.js';
+import { QueryBuilderAggregateOperator_StdDev_Population } from './operators/QueryBuilderAggregateOperator_StdDev_Population.js';
+import { QueryBuilderAggregateOperator_StdDev_Sample } from './operators/QueryBuilderAggregateOperator_StdDev_Sample.js';
+import { QueryBuilderAggregateOperator_DistinctCount } from './operators/QueryBuilderAggregateOperator_DistinctCount.js';
+import { QueryBuilderAggregateOperator_Min } from './operators/QueryBuilderAggregateOperator_Min.js';
+import { QueryBuilderAggregateOperator_Max } from './operators/QueryBuilderAggregateOperator_Max.js';
+import { QueryBuilderAggregateOperator_JoinString } from './operators/QueryBuilderAggregateOperator_JoinString.js';
+export const getQueryBuilderCoreAggregrationOperators = () => [
+    new QueryBuilderAggregateOperator_Count(),
+    new QueryBuilderAggregateOperator_DistinctCount(),
+    new QueryBuilderAggregateOperator_Distinct(),
+    new QueryBuilderAggregateOperator_Sum(),
+    new QueryBuilderAggregateOperator_Average(),
+    new QueryBuilderAggregateOperator_Min(),
+    new QueryBuilderAggregateOperator_Max(),
+    new QueryBuilderAggregateOperator_StdDev_Population(),
+    new QueryBuilderAggregateOperator_StdDev_Sample(),
+    new QueryBuilderAggregateOperator_JoinString(),
+];
+//# sourceMappingURL=QueryBuilderAggregateOperatorLoader.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderAggregateOperatorLoader.js","sourceRoot":"","sources":["../../../../../src/stores/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/fetch-structure/projection/aggregation/QueryBuilderAggregationState.d.ts

@@ -0,0 +1,42 @@
+/**
+ * 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 { Type } from '@finos/legend-graph';
+import type { QueryBuilderProjectionState } from '../QueryBuilderProjectionState.js';
+import type { QueryBuilderAggregateOperator } from './QueryBuilderAggregateOperator.js';
+import type { QueryBuilderProjectionColumnState } from '../QueryBuilderProjectionColumnState.js';
+export declare class QueryBuilderAggregateColumnState {
+    readonly uuid: string;
+    aggregationState: QueryBuilderAggregationState;
+    projectionColumnState: QueryBuilderProjectionColumnState;
+    lambdaParameterName: string;
+    operator: QueryBuilderAggregateOperator;
+    constructor(aggregationState: QueryBuilderAggregationState, projectionColumnState: QueryBuilderProjectionColumnState, operator: QueryBuilderAggregateOperator);
+    get columnName(): string;
+    setColumnState(val: QueryBuilderProjectionColumnState): void;
+    setLambdaParameterName(val: string): void;
+    setOperator(val: QueryBuilderAggregateOperator): void;
+    getReturnType(): Type;
+}
+export declare class QueryBuilderAggregationState {
+    projectionState: QueryBuilderProjectionState;
+    operators: QueryBuilderAggregateOperator[];
+    columns: QueryBuilderAggregateColumnState[];
+    constructor(projectionState: QueryBuilderProjectionState, operators: QueryBuilderAggregateOperator[]);
+    removeColumn(val: QueryBuilderAggregateColumnState): void;
+    addColumn(val: QueryBuilderAggregateColumnState): void;
+    changeColumnAggregateOperator(val: QueryBuilderAggregateOperator | undefined, projectionColumnState: QueryBuilderProjectionColumnState): void;
+}
+//# sourceMappingURL=QueryBuilderAggregationState.d.ts.map

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

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