@finos/legend-application-query 5.2.6 → 6.0.0

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);
-      }
-    }
-  }
-}

package/src/stores/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Average.ts

@@ -1,98 +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 {
-  PRIMITIVE_TYPE,
-  type ValueSpecification,
-  type SimpleFunctionExpression,
-  type VariableExpression,
-  type PureModel,
-  type AbstractPropertyExpression,
-  type Type,
-} from '@finos/legend-graph';
-import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../QueryBuilder_Const.js';
-import type { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
-import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
-import {
-  type QueryBuilderProjectionColumnState,
-  QueryBuilderSimpleProjectionColumnState,
-} from '../../QueryBuilderProjectionColumnState.js';
-import {
-  buildAggregateColumnState,
-  buildAggregateExpression,
-} from './QueryBuilderAggregateOperatorHelper.js';
-
-export class QueryBuilderAggregateOperator_Average extends QueryBuilderAggregateOperator {
-  getLabel(projectionColumnState: QueryBuilderProjectionColumnState): string {
-    return 'average';
-  }
-
-  isCompatibleWithColumn(
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): boolean {
-    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,
-        ] as string[]
-      ).includes(propertyType.path);
-    }
-    return true;
-  }
-
-  buildAggregateExpression(
-    propertyExpression: AbstractPropertyExpression | undefined,
-    variableName: string,
-    graph: PureModel,
-  ): ValueSpecification {
-    return buildAggregateExpression(
-      QUERY_BUILDER_SUPPORTED_FUNCTIONS.AVERAGE,
-      graph,
-      variableName,
-    );
-  }
-
-  buildAggregateColumnState(
-    expression: SimpleFunctionExpression,
-    lambdaParam: VariableExpression,
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): QueryBuilderAggregateColumnState | undefined {
-    return buildAggregateColumnState(
-      projectionColumnState,
-      lambdaParam,
-      expression,
-      QUERY_BUILDER_SUPPORTED_FUNCTIONS.AVERAGE,
-      this,
-    );
-  }
-
-  override getReturnType(
-    aggregateColumnState: QueryBuilderAggregateColumnState,
-  ): Type {
-    const graph =
-      aggregateColumnState.aggregationState.projectionState.queryBuilderState
-        .graphManagerState.graph;
-    return graph.getType(PRIMITIVE_TYPE.FLOAT);
-  }
-}

package/src/stores/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Count.ts

@@ -1,106 +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 {
-  PRIMITIVE_TYPE,
-  type Type,
-  type ValueSpecification,
-  type SimpleFunctionExpression,
-  type VariableExpression,
-  type PureModel,
-  type AbstractPropertyExpression,
-  Enumeration,
-} from '@finos/legend-graph';
-import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../QueryBuilder_Const.js';
-import type { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
-import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
-import {
-  type QueryBuilderProjectionColumnState,
-  QueryBuilderSimpleProjectionColumnState,
-} from '../../QueryBuilderProjectionColumnState.js';
-import {
-  buildAggregateColumnState,
-  buildAggregateExpression,
-} from './QueryBuilderAggregateOperatorHelper.js';
-
-export class QueryBuilderAggregateOperator_Count extends QueryBuilderAggregateOperator {
-  getLabel(projectionColumnState: QueryBuilderProjectionColumnState): string {
-    return 'count';
-  }
-
-  isCompatibleWithColumn(
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): boolean {
-    if (
-      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
-    ) {
-      const propertyType =
-        projectionColumnState.propertyExpressionState.propertyExpression.func
-          .genericType.value.rawType;
-      return (
-        (
-          [
-            PRIMITIVE_TYPE.STRING,
-            PRIMITIVE_TYPE.BOOLEAN,
-            PRIMITIVE_TYPE.NUMBER,
-            PRIMITIVE_TYPE.INTEGER,
-            PRIMITIVE_TYPE.DECIMAL,
-            PRIMITIVE_TYPE.FLOAT,
-            PRIMITIVE_TYPE.DATE,
-            PRIMITIVE_TYPE.STRICTDATE,
-            PRIMITIVE_TYPE.DATETIME,
-          ] as string[]
-        ).includes(propertyType.path) || propertyType instanceof Enumeration
-      );
-    }
-    return true;
-  }
-
-  buildAggregateExpression(
-    propertyExpression: AbstractPropertyExpression | undefined,
-    variableName: string,
-    graph: PureModel,
-  ): ValueSpecification {
-    return buildAggregateExpression(
-      QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT,
-      graph,
-      variableName,
-    );
-  }
-
-  buildAggregateColumnState(
-    expression: SimpleFunctionExpression,
-    lambdaParam: VariableExpression,
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): QueryBuilderAggregateColumnState | undefined {
-    return buildAggregateColumnState(
-      projectionColumnState,
-      lambdaParam,
-      expression,
-      QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT,
-      this,
-    );
-  }
-
-  override getReturnType(
-    aggregateColumnState: QueryBuilderAggregateColumnState,
-  ): Type {
-    const graph =
-      aggregateColumnState.aggregationState.projectionState.queryBuilderState
-        .graphManagerState.graph;
-    return graph.getType(PRIMITIVE_TYPE.INTEGER);
-  }
-}

package/src/stores/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_Distinct.ts

@@ -1,96 +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 {
-  PRIMITIVE_TYPE,
-  type ValueSpecification,
-  type SimpleFunctionExpression,
-  type VariableExpression,
-  type PureModel,
-  type AbstractPropertyExpression,
-  Enumeration,
-} from '@finos/legend-graph';
-import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../QueryBuilder_Const.js';
-import type { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
-import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
-import {
-  type QueryBuilderProjectionColumnState,
-  QueryBuilderSimpleProjectionColumnState,
-} from '../../QueryBuilderProjectionColumnState.js';
-import {
-  buildAggregateColumnState,
-  buildAggregateExpression,
-} from './QueryBuilderAggregateOperatorHelper.js';
-
-export class QueryBuilderAggregateOperator_Distinct extends QueryBuilderAggregateOperator {
-  getLabel(projectionColumnState: QueryBuilderProjectionColumnState): string {
-    return 'distinct';
-  }
-
-  isCompatibleWithColumn(
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): boolean {
-    if (
-      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
-    ) {
-      const propertyType =
-        projectionColumnState.propertyExpressionState.propertyExpression.func
-          .genericType.value.rawType;
-      return (
-        (
-          [
-            PRIMITIVE_TYPE.STRING,
-            PRIMITIVE_TYPE.BOOLEAN,
-            PRIMITIVE_TYPE.NUMBER,
-            PRIMITIVE_TYPE.INTEGER,
-            PRIMITIVE_TYPE.DECIMAL,
-            PRIMITIVE_TYPE.FLOAT,
-            PRIMITIVE_TYPE.DATE,
-            PRIMITIVE_TYPE.STRICTDATE,
-            PRIMITIVE_TYPE.DATETIME,
-          ] as string[]
-        ).includes(propertyType.path) || propertyType instanceof Enumeration
-      );
-    }
-    return true;
-  }
-
-  buildAggregateExpression(
-    propertyExpression: AbstractPropertyExpression | undefined,
-    variableName: string,
-    graph: PureModel,
-  ): ValueSpecification {
-    return buildAggregateExpression(
-      QUERY_BUILDER_SUPPORTED_FUNCTIONS.UNIQUE_VALUE_ONLY,
-      graph,
-      variableName,
-    );
-  }
-
-  buildAggregateColumnState(
-    expression: SimpleFunctionExpression,
-    lambdaParam: VariableExpression,
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): QueryBuilderAggregateColumnState | undefined {
-    return buildAggregateColumnState(
-      projectionColumnState,
-      lambdaParam,
-      expression,
-      QUERY_BUILDER_SUPPORTED_FUNCTIONS.UNIQUE_VALUE_ONLY,
-      this,
-    );
-  }
-}

package/src/stores/fetch-structure/projection/aggregation/operators/QueryBuilderAggregateOperator_DistinctCount.ts

@@ -1,173 +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 AbstractPropertyExpression,
-  type PureModel,
-  type ValueSpecification,
-  SimpleFunctionExpression,
-  VariableExpression,
-  TYPICAL_MULTIPLICITY_TYPE,
-  extractElementNameFromPath,
-  matchFunctionName,
-  PRIMITIVE_TYPE,
-  type Type,
-  Enumeration,
-} from '@finos/legend-graph';
-import { assertTrue, guaranteeType } from '@finos/legend-shared';
-import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../QueryBuilder_Const.js';
-import { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
-import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
-import {
-  type QueryBuilderProjectionColumnState,
-  QueryBuilderSimpleProjectionColumnState,
-} from '../../QueryBuilderProjectionColumnState.js';
-
-export class QueryBuilderAggregateOperator_DistinctCount extends QueryBuilderAggregateOperator {
-  getLabel(projectionColumnState: QueryBuilderProjectionColumnState): string {
-    return 'distinct count';
-  }
-
-  isCompatibleWithColumn(
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): boolean {
-    if (
-      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
-    ) {
-      const propertyType =
-        projectionColumnState.propertyExpressionState.propertyExpression.func
-          .genericType.value.rawType;
-      return (
-        (
-          [
-            PRIMITIVE_TYPE.STRING,
-            PRIMITIVE_TYPE.BOOLEAN,
-            PRIMITIVE_TYPE.NUMBER,
-            PRIMITIVE_TYPE.INTEGER,
-            PRIMITIVE_TYPE.DECIMAL,
-            PRIMITIVE_TYPE.FLOAT,
-            PRIMITIVE_TYPE.DATE,
-            PRIMITIVE_TYPE.STRICTDATE,
-            PRIMITIVE_TYPE.DATETIME,
-          ] as string[]
-        ).includes(propertyType.path) || propertyType instanceof Enumeration
-      );
-    }
-    return true;
-  }
-
-  buildAggregateExpression(
-    propertyExpression: AbstractPropertyExpression | undefined,
-    variableName: string,
-    graph: PureModel,
-  ): ValueSpecification {
-    const multiplicityOne = graph.getTypicalMultiplicity(
-      TYPICAL_MULTIPLICITY_TYPE.ONE,
-    );
-    const distinctExpression = new SimpleFunctionExpression(
-      extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.DISTINCT),
-      multiplicityOne,
-    );
-    distinctExpression.parametersValues.push(
-      new VariableExpression(variableName, multiplicityOne),
-    );
-    const distinctCountExpression = new SimpleFunctionExpression(
-      extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT),
-      multiplicityOne,
-    );
-    distinctCountExpression.parametersValues.push(distinctExpression);
-    return distinctCountExpression;
-  }
-
-  buildAggregateColumnState(
-    expression: SimpleFunctionExpression,
-    lambdaParam: VariableExpression,
-    projectionColumnState: QueryBuilderProjectionColumnState,
-  ): QueryBuilderAggregateColumnState | undefined {
-    if (
-      matchFunctionName(
-        expression.functionName,
-        QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT,
-      )
-    ) {
-      const aggregateColumnState = new QueryBuilderAggregateColumnState(
-        projectionColumnState.projectionState.aggregationState,
-        projectionColumnState,
-        this,
-      );
-      aggregateColumnState.setLambdaParameterName(lambdaParam.name);
-
-      // count expression
-      assertTrue(
-        expression.parametersValues.length === 1,
-        `Can't process count() expression: count() expects no argument`,
-      );
-
-      // distinct expression
-      const distinctExpression = guaranteeType(
-        expression.parametersValues[0],
-        SimpleFunctionExpression,
-        `Can't process '${this.getLabel(
-          projectionColumnState,
-        )}' aggregate lambda: only support count() immediately following an expression`,
-      );
-      assertTrue(
-        matchFunctionName(
-          distinctExpression.functionName,
-          QUERY_BUILDER_SUPPORTED_FUNCTIONS.DISTINCT,
-        ),
-        `Can't process '${this.getLabel(
-          projectionColumnState,
-        )}' aggregate lambda: only support count() immediately following distinct() expression`,
-      );
-      assertTrue(
-        distinctExpression.parametersValues.length === 1,
-        `Can't process distinct() expression: distinct() expects no argument`,
-      );
-
-      // variable
-      const variableExpression = guaranteeType(
-        distinctExpression.parametersValues[0],
-        VariableExpression,
-        `Can't process distinct() expression: only support distinct() immediately following a variable expression`,
-      );
-      assertTrue(
-        aggregateColumnState.lambdaParameterName === variableExpression.name,
-        `Can't process distinct() expression: expects variable used in lambda body '${variableExpression.name}' to match lambda parameter '${aggregateColumnState.lambdaParameterName}'`,
-      );
-
-      // operator
-      assertTrue(
-        this.isCompatibleWithColumn(aggregateColumnState.projectionColumnState),
-        `Can't process disc expression: property is not compatible with operator`,
-      );
-      aggregateColumnState.setOperator(this);
-
-      return aggregateColumnState;
-    }
-
-    return undefined;
-  }
-
-  override getReturnType(
-    aggregateColumnState: QueryBuilderAggregateColumnState,
-  ): Type {
-    const graph =
-      aggregateColumnState.aggregationState.projectionState.queryBuilderState
-        .graphManagerState.graph;
-    return graph.getType(PRIMITIVE_TYPE.INTEGER);
-  }
-}