@finos/legend-query-builder 3.0.7 → 3.1.0

package/src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_ReportEndDay.ts

@@ -0,0 +1,98 @@
+/**
+ * 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 AbstractPropertyExpression,
+} from '@finos/legend-graph';
+import type { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
+import {
+  type QueryBuilderProjectionColumnState,
+  QueryBuilderSimpleProjectionColumnState,
+} from '../../projection/QueryBuilderProjectionColumnState.js';
+import { QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS } from '../../../../../graph/QueryBuilderMetaModelConst.js';
+import { hashArray, type Hashable } from '@finos/legend-shared';
+import { QUERY_BUILDER_STATE_HASH_STRUCTURE } from '../../../../QueryBuilderStateHashUtils.js';
+import { QueryBuilderAggregateCalendarFunction } from '../QueryBuilderAggregateCalendarFunction.js';
+import {
+  updateAggregateColumnState,
+  buildCalendarFunctionExpression,
+} from './QueryBuilderAggregateCalendarFunctionValueSpecificationBuilder.js';
+
+export class QueryBuilderAggregateCalendarFunction_ReportEndDay
+  extends QueryBuilderAggregateCalendarFunction
+  implements Hashable
+{
+  getLabel(): string {
+    return 'Report End Day';
+  }
+
+  isCompatibleWithColumn(
+    projectionColumnState: QueryBuilderProjectionColumnState,
+  ): boolean {
+    if (
+      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
+    ) {
+      const propertyType =
+        projectionColumnState.propertyExpressionState.propertyExpression.func
+          .value.genericType.value.rawType;
+      return (
+        [
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+        ] as string[]
+      ).includes(propertyType.path);
+    }
+    return true;
+  }
+
+  buildCalendarFunctionExpression(
+    p: AbstractPropertyExpression,
+  ): ValueSpecification {
+    return buildCalendarFunctionExpression(
+      QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS.CALENDAR_REPORT_END_DAY,
+      this.dateColumn,
+      this.calendarType,
+      this.endDate,
+      p,
+    );
+  }
+
+  updateAggregateColumnState(
+    expression: SimpleFunctionExpression,
+    aggregationColumnState: QueryBuilderAggregateColumnState,
+  ): void {
+    updateAggregateColumnState(
+      expression,
+      QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS.CALENDAR_REPORT_END_DAY,
+      this,
+      aggregationColumnState,
+    );
+  }
+
+  get hashCode(): string {
+    return hashArray([
+      QUERY_BUILDER_STATE_HASH_STRUCTURE.AGGREGATE_CALENDAR_FUNCTION_REPORT_END_DAY,
+      this.dateColumn ?? '',
+      this.calendarType,
+      this.endDate,
+    ]);
+  }
+}

package/src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Wtd.ts

@@ -0,0 +1,98 @@
+/**
+ * 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 AbstractPropertyExpression,
+} from '@finos/legend-graph';
+import type { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
+import {
+  type QueryBuilderProjectionColumnState,
+  QueryBuilderSimpleProjectionColumnState,
+} from '../../projection/QueryBuilderProjectionColumnState.js';
+import { QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS } from '../../../../../graph/QueryBuilderMetaModelConst.js';
+import { hashArray, type Hashable } from '@finos/legend-shared';
+import { QUERY_BUILDER_STATE_HASH_STRUCTURE } from '../../../../QueryBuilderStateHashUtils.js';
+import { QueryBuilderAggregateCalendarFunction } from '../QueryBuilderAggregateCalendarFunction.js';
+import {
+  updateAggregateColumnState,
+  buildCalendarFunctionExpression,
+} from './QueryBuilderAggregateCalendarFunctionValueSpecificationBuilder.js';
+
+export class QueryBuilderAggregateCalendarFunction_Wtd
+  extends QueryBuilderAggregateCalendarFunction
+  implements Hashable
+{
+  getLabel(): string {
+    return 'Week to Date';
+  }
+
+  isCompatibleWithColumn(
+    projectionColumnState: QueryBuilderProjectionColumnState,
+  ): boolean {
+    if (
+      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
+    ) {
+      const propertyType =
+        projectionColumnState.propertyExpressionState.propertyExpression.func
+          .value.genericType.value.rawType;
+      return (
+        [
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+        ] as string[]
+      ).includes(propertyType.path);
+    }
+    return true;
+  }
+
+  buildCalendarFunctionExpression(
+    p: AbstractPropertyExpression,
+  ): ValueSpecification {
+    return buildCalendarFunctionExpression(
+      QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS.CALENDAR_WTD,
+      this.dateColumn,
+      this.calendarType,
+      this.endDate,
+      p,
+    );
+  }
+
+  updateAggregateColumnState(
+    expression: SimpleFunctionExpression,
+    aggregationColumnState: QueryBuilderAggregateColumnState,
+  ): void {
+    updateAggregateColumnState(
+      expression,
+      QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS.CALENDAR_WTD,
+      this,
+      aggregationColumnState,
+    );
+  }
+
+  get hashCode(): string {
+    return hashArray([
+      QUERY_BUILDER_STATE_HASH_STRUCTURE.AGGREGATE_CALENDAR_FUNCTION_WTD,
+      this.dateColumn ?? '',
+      this.calendarType,
+      this.endDate,
+    ]);
+  }
+}

package/src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Ytd.ts

@@ -0,0 +1,98 @@
+/**
+ * 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 AbstractPropertyExpression,
+} from '@finos/legend-graph';
+import type { QueryBuilderAggregateColumnState } from '../QueryBuilderAggregationState.js';
+import {
+  type QueryBuilderProjectionColumnState,
+  QueryBuilderSimpleProjectionColumnState,
+} from '../../projection/QueryBuilderProjectionColumnState.js';
+import { QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS } from '../../../../../graph/QueryBuilderMetaModelConst.js';
+import { hashArray, type Hashable } from '@finos/legend-shared';
+import { QUERY_BUILDER_STATE_HASH_STRUCTURE } from '../../../../QueryBuilderStateHashUtils.js';
+import { QueryBuilderAggregateCalendarFunction } from '../QueryBuilderAggregateCalendarFunction.js';
+import {
+  updateAggregateColumnState,
+  buildCalendarFunctionExpression,
+} from './QueryBuilderAggregateCalendarFunctionValueSpecificationBuilder.js';
+
+export class QueryBuilderAggregateCalendarFunction_Ytd
+  extends QueryBuilderAggregateCalendarFunction
+  implements Hashable
+{
+  getLabel(): string {
+    return 'Year to Date';
+  }
+
+  isCompatibleWithColumn(
+    projectionColumnState: QueryBuilderProjectionColumnState,
+  ): boolean {
+    if (
+      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
+    ) {
+      const propertyType =
+        projectionColumnState.propertyExpressionState.propertyExpression.func
+          .value.genericType.value.rawType;
+      return (
+        [
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+        ] as string[]
+      ).includes(propertyType.path);
+    }
+    return true;
+  }
+
+  buildCalendarFunctionExpression(
+    p: AbstractPropertyExpression,
+  ): ValueSpecification {
+    return buildCalendarFunctionExpression(
+      QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS.CALENDAR_YTD,
+      this.dateColumn,
+      this.calendarType,
+      this.endDate,
+      p,
+    );
+  }
+
+  updateAggregateColumnState(
+    expression: SimpleFunctionExpression,
+    aggregationColumnState: QueryBuilderAggregateColumnState,
+  ): void {
+    updateAggregateColumnState(
+      expression,
+      QUERY_BUILDER_SUPPORTED_CALENDAR_AGGREGATION_FUNCTIONS.CALENDAR_YTD,
+      this,
+      aggregationColumnState,
+    );
+  }
+
+  get hashCode(): string {
+    return hashArray([
+      QUERY_BUILDER_STATE_HASH_STRUCTURE.AGGREGATE_CALENDAR_FUNCTION_YTD,
+      this.dateColumn ?? '',
+      this.calendarType,
+      this.endDate,
+    ]);
+  }
+}

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

@@ -274,8 +274,25 @@ export const appendProjection = (
           aggregateColumnState.aggregationState.tdsState.queryBuilderState
             .graphManagerState.graph,
         );
+        let aggregateCalendarLambda;
+        const aggregateCalendarLambdaState =
+          aggregateColumnState.calendarFunction?.buildCalendarFunctionExpressionFromState(
+            aggregateColumnState,
+          );
+        if (
+          queryBuilderState.isCalendarEnabled &&
+          aggregateCalendarLambdaState !== undefined
+        ) {
+          aggregateCalendarLambda = buildGenericLambdaFunctionInstanceValue(
+            guaranteeNonNullable(aggregateColumnState.calendarFunction)
+              .lambdaParameterName,
+            [aggregateCalendarLambdaState],
+            aggregateColumnState.aggregationState.tdsState.queryBuilderState
+              .graphManagerState.graph,
+          );
+        }
         aggregateFunctionExpression.parametersValues = [
-          columnLambda,
+          aggregateCalendarLambda ?? columnLambda,
           aggregateLambda,
         ];
         aggregateLambdas.values.push(aggregateFunctionExpression);

package/tsconfig.json

@@ -43,6 +43,7 @@
     "./src/components/QueryBuilder_LegendApplicationPlugin.ts",
     "./src/components/QuerySetupUtils.ts",
     "./src/graph/QueryBuilderMetaModelConst.ts",
+    "./src/graph-manager/QueryBuilderConst.ts",
     "./src/graph-manager/QueryBuilder_GraphManagerPreset.ts",
     "./src/graph-manager/QueryBuilder_PureGraphManagerPlugin.ts",
     "./src/graph-manager/action/analytics/MappingRuntimeCompatibilityAnalysis.ts",
@@ -87,10 +88,44 @@
     "./src/stores/fetch-structure/tds/QueryBuilderTDSHelper.ts",
     "./src/stores/fetch-structure/tds/QueryBuilderTDSState.ts",
     "./src/stores/fetch-structure/tds/QueryResultSetModifierState.ts",
+    "./src/stores/fetch-structure/tds/aggregation/QueryBuilderAggregateCalendarFunction.ts",
+    "./src/stores/fetch-structure/tds/aggregation/QueryBuilderAggregateCalendarFunctionLoader.ts",
     "./src/stores/fetch-structure/tds/aggregation/QueryBuilderAggregateOperator.ts",
     "./src/stores/fetch-structure/tds/aggregation/QueryBuilderAggregateOperatorLoader.ts",
     "./src/stores/fetch-structure/tds/aggregation/QueryBuilderAggregationState.ts",
     "./src/stores/fetch-structure/tds/aggregation/QueryBuilderAggregationStateBuilder.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunctionValueSpecificationBuilder.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Annualized.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Cme.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Cw.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Cw_Fm.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Cy_Minus2.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Cy_Minus3.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Mtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_P12Wa.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_P12Wtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_P4Wa.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_P4Wtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_P52Wa.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_P52Wtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pma.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pmtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pqtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_PriorDay.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_PriorYear.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pw.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pw_Fm.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pwa.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pwtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pymtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pyqtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pytd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pywa.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Pywtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Qtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_ReportEndDay.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Wtd.ts",
+    "./src/stores/fetch-structure/tds/aggregation/calendarFunctions/QueryBuilderAggregateCalendarFunction_Ytd.ts",
     "./src/stores/fetch-structure/tds/aggregation/operators/QueryBuilderAggregateOperatorValueSpecificationBuilder.ts",
     "./src/stores/fetch-structure/tds/aggregation/operators/QueryBuilderAggregateOperator_Average.ts",
     "./src/stores/fetch-structure/tds/aggregation/operators/QueryBuilderAggregateOperator_Count.ts",