@finos/legend-application-query 5.2.8 → 6.0.0

package/src/stores/fetch-structure/projection/QueryBuilderProjectionValueSpecificationBuilder.ts

@@ -1,416 +0,0 @@
-/**
- * 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 LambdaFunction,
-  CollectionInstanceValue,
-  extractElementNameFromPath,
-  GenericType,
-  GenericTypeExplicitReference,
-  Multiplicity,
-  PrimitiveInstanceValue,
-  PRIMITIVE_TYPE,
-  SimpleFunctionExpression,
-  TYPICAL_MULTIPLICITY_TYPE,
-  type ValueSpecification,
-  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 '../../../QueryBuilder_Const.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 type { QueryBuilderProjectionState } from './QueryBuilderProjectionState.js';
-import {
-  COLUMN_SORT_TYPE,
-  type QueryResultSetModifierState,
-  type SortColumnState,
-} from './QueryResultSetModifierState.js';
-
-const buildSortExpression = (
-  sortColumnState: SortColumnState,
-): SimpleFunctionExpression => {
-  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: QueryResultSetModifierState,
-  lambda: LambdaFunction,
-  options?:
-    | {
-        overridingLimit?: number | undefined;
-      }
-    | undefined,
-): LambdaFunction => {
-  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: QueryBuilderProjectionState,
-  lambdaFunction: LambdaFunction,
-  options?: {
-    /**
-     * Set queryBuilderState to `true` when we construct query for execution within the app.
-     * queryBuilderState will make the lambda function building process overrides several query values, such as the row limit.
-     */
-    isBuildingExecutionQuery?: boolean | undefined;
-    keepSourceInformation?: boolean | undefined;
-  },
-): void => {
-  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: ValueSpecification;
-      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: ValueSpecification;
-      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,
-    },
-  );
-};

package/src/stores/fetch-structure/projection/aggregation/QueryBuilderAggregationState.ts

@@ -1,145 +0,0 @@
-/**
- * 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 type { Type } from '@finos/legend-graph';
-import { DEFAULT_LAMBDA_VARIABLE_NAME } from '../../../../QueryBuilder_Const.js';
-import type { QueryBuilderProjectionState } from '../QueryBuilderProjectionState.js';
-import type { QueryBuilderAggregateOperator } from './QueryBuilderAggregateOperator.js';
-import type { QueryBuilderProjectionColumnState } from '../QueryBuilderProjectionColumnState.js';
-
-export class QueryBuilderAggregateColumnState {
-  readonly uuid = uuid();
-  aggregationState: QueryBuilderAggregationState;
-  projectionColumnState: QueryBuilderProjectionColumnState;
-  lambdaParameterName: string = DEFAULT_LAMBDA_VARIABLE_NAME;
-  operator: QueryBuilderAggregateOperator;
-
-  constructor(
-    aggregationState: QueryBuilderAggregationState,
-    projectionColumnState: QueryBuilderProjectionColumnState,
-    operator: QueryBuilderAggregateOperator,
-  ) {
-    makeAutoObservable(this, {
-      uuid: false,
-      aggregationState: false,
-      setColumnState: action,
-      setLambdaParameterName: action,
-      setOperator: action,
-    });
-
-    this.aggregationState = aggregationState;
-    this.projectionColumnState = projectionColumnState;
-    this.operator = operator;
-  }
-
-  get columnName(): string {
-    return this.projectionColumnState.columnName;
-  }
-
-  setColumnState(val: QueryBuilderProjectionColumnState): void {
-    this.projectionColumnState = val;
-  }
-
-  setLambdaParameterName(val: string): void {
-    this.lambdaParameterName = val;
-  }
-
-  setOperator(val: QueryBuilderAggregateOperator): void {
-    this.operator = val;
-  }
-
-  getReturnType(): Type {
-    return this.operator.getReturnType(this);
-  }
-}
-
-export class QueryBuilderAggregationState {
-  projectionState: QueryBuilderProjectionState;
-  operators: QueryBuilderAggregateOperator[] = [];
-  columns: QueryBuilderAggregateColumnState[] = [];
-
-  constructor(
-    projectionState: QueryBuilderProjectionState,
-    operators: QueryBuilderAggregateOperator[],
-  ) {
-    makeAutoObservable(this, {
-      projectionState: false,
-      removeColumn: action,
-      addColumn: action,
-    });
-
-    this.projectionState = projectionState;
-    this.operators = operators;
-  }
-
-  removeColumn(val: QueryBuilderAggregateColumnState): void {
-    deleteEntry(this.columns, val);
-  }
-
-  addColumn(val: QueryBuilderAggregateColumnState): void {
-    addUniqueEntry(this.columns, val);
-  }
-
-  changeColumnAggregateOperator(
-    val: QueryBuilderAggregateOperator | undefined,
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): void {
-    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);
-      }
-    }
-  }
-}