@finos/legend-query-builder 0.0.1

package/src/stores/fetch-structure/projection/post-filter/operators/QueryBuilderPostFilterOperator_EndWith.ts

@@ -0,0 +1,135 @@
+/**
+ * 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 SimpleFunctionExpression,
+  type Type,
+  type ValueSpecification,
+  type FunctionExpression,
+  PRIMITIVE_TYPE,
+  buildPrimitiveInstanceValue,
+} from '@finos/legend-graph';
+import { UnsupportedOperationError } from '@finos/legend-shared';
+import { QueryBuilderPostFilterOperator } from '../QueryBuilderPostFilterOperator.js';
+import { buildPostFilterConditionState } from '../QueryBuilderPostFilterStateBuilder.js';
+import type {
+  PostFilterConditionState,
+  QueryBuilderPostFilterState,
+} from '../QueryBuilderPostFilterState.js';
+import {
+  buildNotExpression,
+  generateDefaultValueForPrimitiveType,
+  getNonCollectionValueSpecificationType,
+  unwrapNotExpression,
+} from '../../../../QueryBuilderValueSpecificationHelper.js';
+import { buildPostFilterConditionExpression } from './QueryBuilderPostFilterOperatorHelper.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderPostFilterOperator_EndWith extends QueryBuilderPostFilterOperator {
+  getLabel(): string {
+    return 'ends with';
+  }
+
+  isCompatibleWithType(type: Type): boolean {
+    if (type.path === PRIMITIVE_TYPE.STRING) {
+      return true;
+    }
+    return false;
+  }
+
+  isCompatibleWithConditionValue(
+    postFilterConditionState: PostFilterConditionState,
+  ): boolean {
+    const type = postFilterConditionState.value
+      ? getNonCollectionValueSpecificationType(postFilterConditionState.value)
+      : undefined;
+    return PRIMITIVE_TYPE.STRING === type?.path;
+  }
+
+  getDefaultFilterConditionValue(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification {
+    const propertyType = postFilterConditionState.columnState.getReturnType();
+    switch (propertyType?.path) {
+      case PRIMITIVE_TYPE.STRING: {
+        return buildPrimitiveInstanceValue(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          propertyType.path,
+          generateDefaultValueForPrimitiveType(propertyType.path),
+        );
+      }
+      default:
+        throw new UnsupportedOperationError(
+          `Can't get default value for post-filter operator '${this.getLabel()}' when the LHS property is of type '${
+            propertyType?.path
+          }'`,
+        );
+    }
+  }
+
+  buildPostFilterConditionExpression(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification | undefined {
+    return buildPostFilterConditionExpression(
+      postFilterConditionState,
+      this,
+      QUERY_BUILDER_SUPPORTED_FUNCTIONS.ENDS_WITH,
+    );
+  }
+
+  buildPostFilterConditionState(
+    postFilterState: QueryBuilderPostFilterState,
+    expression: FunctionExpression,
+  ): PostFilterConditionState | undefined {
+    return buildPostFilterConditionState(
+      postFilterState,
+      expression,
+      QUERY_BUILDER_SUPPORTED_FUNCTIONS.ENDS_WITH,
+      this,
+    );
+  }
+}
+
+export class QueryBuilderPostFilterOperator_NotEndWith extends QueryBuilderPostFilterOperator_EndWith {
+  override getLabel(): string {
+    return `doesn't end with`;
+  }
+
+  override buildPostFilterConditionExpression(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification | undefined {
+    const expression = super.buildPostFilterConditionExpression(
+      postFilterConditionState,
+    );
+    return expression
+      ? buildNotExpression(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          expression,
+        )
+      : undefined;
+  }
+
+  override buildPostFilterConditionState(
+    postFilterState: QueryBuilderPostFilterState,
+    expression: SimpleFunctionExpression,
+  ): PostFilterConditionState | undefined {
+    const innerExpression = unwrapNotExpression(expression);
+    return innerExpression
+      ? super.buildPostFilterConditionState(postFilterState, innerExpression)
+      : undefined;
+  }
+}

package/src/stores/fetch-structure/projection/post-filter/operators/QueryBuilderPostFilterOperator_Equal.ts

@@ -0,0 +1,218 @@
+/**
+ * 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,
+  type ValueSpecification,
+  type Enum,
+  type SimpleFunctionExpression,
+  type FunctionExpression,
+  AbstractPropertyExpression,
+  Enumeration,
+  EnumValueExplicitReference,
+  EnumValueInstanceValue,
+  GenericType,
+  GenericTypeExplicitReference,
+  PRIMITIVE_TYPE,
+  TYPICAL_MULTIPLICITY_TYPE,
+  SUPPORTED_FUNCTIONS,
+  buildPrimitiveInstanceValue,
+} from '@finos/legend-graph';
+import {
+  guaranteeNonNullable,
+  UnsupportedOperationError,
+} from '@finos/legend-shared';
+import { QueryBuilderPostFilterOperator } from '../QueryBuilderPostFilterOperator.js';
+import { buildPostFilterConditionState } from '../QueryBuilderPostFilterStateBuilder.js';
+import type {
+  PostFilterConditionState,
+  QueryBuilderPostFilterState,
+} from '../QueryBuilderPostFilterState.js';
+import {
+  buildNotExpression,
+  generateDefaultValueForPrimitiveType,
+  getNonCollectionValueSpecificationType,
+  isTypeCompatibleForAssignment,
+  unwrapNotExpression,
+} from '../../../../QueryBuilderValueSpecificationHelper.js';
+import { buildPostFilterConditionExpression } from './QueryBuilderPostFilterOperatorHelper.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderPostFilterOperator_Equal extends QueryBuilderPostFilterOperator {
+  getLabel(): string {
+    return 'is';
+  }
+
+  isCompatibleWithType(type: Type): boolean {
+    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(type.path) ||
+      // if the type is enumeration, make sure the enumeration has some value
+      (type instanceof Enumeration && type.values.length > 0)
+    );
+  }
+
+  isCompatibleWithConditionValue(
+    postFilterConditionState: PostFilterConditionState,
+  ): boolean {
+    const valueSpecification = postFilterConditionState.value;
+    if (valueSpecification) {
+      return isTypeCompatibleForAssignment(
+        getNonCollectionValueSpecificationType(valueSpecification),
+        guaranteeNonNullable(
+          postFilterConditionState.columnState.getReturnType(),
+        ),
+      );
+    }
+    return false;
+  }
+
+  getDefaultFilterConditionValue(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification {
+    const propertyType = guaranteeNonNullable(
+      postFilterConditionState.columnState.getReturnType(),
+    );
+    switch (propertyType.path) {
+      case PRIMITIVE_TYPE.STRING:
+      case PRIMITIVE_TYPE.BOOLEAN:
+      case PRIMITIVE_TYPE.STRICTDATE:
+      case PRIMITIVE_TYPE.DATETIME:
+      case PRIMITIVE_TYPE.NUMBER:
+      case PRIMITIVE_TYPE.DECIMAL:
+      case PRIMITIVE_TYPE.FLOAT:
+      case PRIMITIVE_TYPE.INTEGER: {
+        return buildPrimitiveInstanceValue(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          propertyType.path,
+          generateDefaultValueForPrimitiveType(propertyType.path),
+        );
+      }
+      case PRIMITIVE_TYPE.DATE: {
+        return buildPrimitiveInstanceValue(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          PRIMITIVE_TYPE.STRICTDATE,
+          generateDefaultValueForPrimitiveType(propertyType.path),
+        );
+      }
+      default:
+        if (propertyType instanceof Enumeration) {
+          if (propertyType.values.length > 0) {
+            const multiplicityOne =
+              postFilterConditionState.postFilterState.projectionState.queryBuilderState.graphManagerState.graph.getTypicalMultiplicity(
+                TYPICAL_MULTIPLICITY_TYPE.ONE,
+              );
+            const enumValueInstanceValue = new EnumValueInstanceValue(
+              GenericTypeExplicitReference.create(
+                new GenericType(propertyType),
+              ),
+              multiplicityOne,
+            );
+            enumValueInstanceValue.values = [
+              EnumValueExplicitReference.create(propertyType.values[0] as Enum),
+            ];
+            return enumValueInstanceValue;
+          }
+          throw new UnsupportedOperationError(
+            `Can't get default value for post-filter operator '${this.getLabel()}' since enumeration '${
+              propertyType.path
+            }' has no value`,
+          );
+        }
+        throw new UnsupportedOperationError(
+          `Can't get default value for post-filter operator '${this.getLabel()}' when the LHS property is of type '${
+            propertyType.path
+          }'`,
+        );
+    }
+  }
+
+  buildPostFilterConditionExpression(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification | undefined {
+    return buildPostFilterConditionExpression(
+      postFilterConditionState,
+      this,
+      postFilterConditionState.columnState.getReturnType()?.path ===
+        PRIMITIVE_TYPE.DATETIME &&
+        postFilterConditionState.value?.genericType?.value.rawType.path !==
+          PRIMITIVE_TYPE.DATETIME
+        ? SUPPORTED_FUNCTIONS.IS_ON_DAY
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.EQUAL,
+    );
+  }
+
+  buildPostFilterConditionState(
+    postFilterState: QueryBuilderPostFilterState,
+    expression: FunctionExpression,
+  ): PostFilterConditionState | undefined {
+    return buildPostFilterConditionState(
+      postFilterState,
+      expression,
+      expression.parametersValues[0] instanceof AbstractPropertyExpression &&
+        expression.parametersValues[0].func.genericType.value.rawType.path ===
+          PRIMITIVE_TYPE.DATETIME &&
+        expression.parametersValues[1]?.genericType?.value.rawType.path !==
+          PRIMITIVE_TYPE.DATETIME
+        ? SUPPORTED_FUNCTIONS.IS_ON_DAY
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.EQUAL,
+      this,
+    );
+  }
+}
+export class QueryBuilderPostFilterOperator_NotEqual extends QueryBuilderPostFilterOperator_Equal {
+  override getLabel(): string {
+    return `is not`;
+  }
+
+  override buildPostFilterConditionExpression(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification | undefined {
+    const expression = super.buildPostFilterConditionExpression(
+      postFilterConditionState,
+    );
+    return expression
+      ? buildNotExpression(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          expression,
+        )
+      : undefined;
+  }
+
+  override buildPostFilterConditionState(
+    postFilterState: QueryBuilderPostFilterState,
+    expression: SimpleFunctionExpression,
+  ): PostFilterConditionState | undefined {
+    const innerExpression = unwrapNotExpression(expression);
+    return innerExpression
+      ? super.buildPostFilterConditionState(postFilterState, innerExpression)
+      : undefined;
+  }
+}

package/src/stores/fetch-structure/projection/post-filter/operators/QueryBuilderPostFilterOperator_GreaterThan.ts

@@ -0,0 +1,143 @@
+/**
+ * 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,
+  type ValueSpecification,
+  type FunctionExpression,
+  AbstractPropertyExpression,
+  PRIMITIVE_TYPE,
+  SUPPORTED_FUNCTIONS,
+  buildPrimitiveInstanceValue,
+} from '@finos/legend-graph';
+import {
+  guaranteeNonNullable,
+  UnsupportedOperationError,
+} from '@finos/legend-shared';
+import { QueryBuilderPostFilterOperator } from '../QueryBuilderPostFilterOperator.js';
+import { buildPostFilterConditionState } from '../QueryBuilderPostFilterStateBuilder.js';
+import type {
+  PostFilterConditionState,
+  QueryBuilderPostFilterState,
+} from '../QueryBuilderPostFilterState.js';
+import {
+  generateDefaultValueForPrimitiveType,
+  getNonCollectionValueSpecificationType,
+  isTypeCompatibleForAssignment,
+} from '../../../../QueryBuilderValueSpecificationHelper.js';
+import { buildPostFilterConditionExpression } from './QueryBuilderPostFilterOperatorHelper.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderPostFilterOperator_GreaterThan extends QueryBuilderPostFilterOperator {
+  getLabel(): string {
+    return '>';
+  }
+
+  isCompatibleWithType(type: Type): boolean {
+    return (
+      [
+        PRIMITIVE_TYPE.NUMBER,
+        PRIMITIVE_TYPE.INTEGER,
+        PRIMITIVE_TYPE.DECIMAL,
+        PRIMITIVE_TYPE.FLOAT,
+        PRIMITIVE_TYPE.DATE,
+        PRIMITIVE_TYPE.STRICTDATE,
+        PRIMITIVE_TYPE.DATETIME,
+      ] as string[]
+    ).includes(type.path);
+  }
+
+  isCompatibleWithConditionValue(
+    postFilterConditionState: PostFilterConditionState,
+  ): boolean {
+    return isTypeCompatibleForAssignment(
+      postFilterConditionState.value
+        ? getNonCollectionValueSpecificationType(postFilterConditionState.value)
+        : undefined,
+      guaranteeNonNullable(
+        postFilterConditionState.columnState.getReturnType(),
+      ),
+    );
+  }
+
+  getDefaultFilterConditionValue(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification {
+    const propertyType = postFilterConditionState.columnState.getReturnType();
+    switch (propertyType?.path) {
+      case PRIMITIVE_TYPE.NUMBER:
+      case PRIMITIVE_TYPE.DECIMAL:
+      case PRIMITIVE_TYPE.FLOAT:
+      case PRIMITIVE_TYPE.INTEGER:
+      case PRIMITIVE_TYPE.STRICTDATE:
+      case PRIMITIVE_TYPE.DATETIME: {
+        return buildPrimitiveInstanceValue(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          propertyType.path,
+          generateDefaultValueForPrimitiveType(propertyType.path),
+        );
+      }
+      case PRIMITIVE_TYPE.DATE: {
+        return buildPrimitiveInstanceValue(
+          postFilterConditionState.postFilterState.projectionState
+            .queryBuilderState.graphManagerState.graph,
+          PRIMITIVE_TYPE.STRICTDATE,
+          generateDefaultValueForPrimitiveType(propertyType.path),
+        );
+      }
+      default:
+        throw new UnsupportedOperationError(
+          `Can't get default value for post-filter operator '${this.getLabel()}' when the LHS property is of type '${
+            propertyType?.path
+          }'`,
+        );
+    }
+  }
+
+  buildPostFilterConditionExpression(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification | undefined {
+    return buildPostFilterConditionExpression(
+      postFilterConditionState,
+      this,
+      postFilterConditionState.columnState.getReturnType()?.path ===
+        PRIMITIVE_TYPE.DATETIME &&
+        postFilterConditionState.value?.genericType?.value.rawType.path !==
+          PRIMITIVE_TYPE.DATETIME
+        ? SUPPORTED_FUNCTIONS.IS_AFTER_DAY
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.GREATER_THAN,
+    );
+  }
+
+  buildPostFilterConditionState(
+    postFilterState: QueryBuilderPostFilterState,
+    expression: FunctionExpression,
+  ): PostFilterConditionState | undefined {
+    return buildPostFilterConditionState(
+      postFilterState,
+      expression,
+      expression.parametersValues[0] instanceof AbstractPropertyExpression &&
+        expression.parametersValues[0].func.genericType.value.rawType.path ===
+          PRIMITIVE_TYPE.DATETIME &&
+        expression.parametersValues[1]?.genericType?.value.rawType.path !==
+          PRIMITIVE_TYPE.DATETIME
+        ? SUPPORTED_FUNCTIONS.IS_AFTER_DAY
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.GREATER_THAN,
+      this,
+    );
+  }
+}

package/src/stores/fetch-structure/projection/post-filter/operators/QueryBuilderPostFilterOperator_GreaterThanEqual.ts

@@ -0,0 +1,70 @@
+/**
+ * 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 FunctionExpression,
+  type ValueSpecification,
+  AbstractPropertyExpression,
+  PRIMITIVE_TYPE,
+  SUPPORTED_FUNCTIONS,
+} from '@finos/legend-graph';
+import { buildPostFilterConditionState } from '../QueryBuilderPostFilterStateBuilder.js';
+import type {
+  PostFilterConditionState,
+  QueryBuilderPostFilterState,
+} from '../QueryBuilderPostFilterState.js';
+import { buildPostFilterConditionExpression } from './QueryBuilderPostFilterOperatorHelper.js';
+import { QueryBuilderPostFilterOperator_GreaterThan } from './QueryBuilderPostFilterOperator_GreaterThan.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderPostFilterOperator_GreaterThanEqual extends QueryBuilderPostFilterOperator_GreaterThan {
+  override getLabel(): string {
+    return '>=';
+  }
+
+  override buildPostFilterConditionExpression(
+    postFilterConditionState: PostFilterConditionState,
+  ): ValueSpecification | undefined {
+    return buildPostFilterConditionExpression(
+      postFilterConditionState,
+      this,
+      postFilterConditionState.columnState.getReturnType()?.path ===
+        PRIMITIVE_TYPE.DATETIME &&
+        postFilterConditionState.value?.genericType?.value.rawType.path !==
+          PRIMITIVE_TYPE.DATETIME
+        ? SUPPORTED_FUNCTIONS.IS_AFTER_DAY
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.GREATER_THAN,
+    );
+  }
+
+  override buildPostFilterConditionState(
+    postFilterState: QueryBuilderPostFilterState,
+    expression: FunctionExpression,
+  ): PostFilterConditionState | undefined {
+    return buildPostFilterConditionState(
+      postFilterState,
+      expression,
+      expression.parametersValues[0] instanceof AbstractPropertyExpression &&
+        expression.parametersValues[0].func.genericType.value.rawType.path ===
+          PRIMITIVE_TYPE.DATETIME &&
+        expression.parametersValues[1]?.genericType?.value.rawType.path !==
+          PRIMITIVE_TYPE.DATETIME
+        ? SUPPORTED_FUNCTIONS.IS_AFTER_DAY
+        : QUERY_BUILDER_SUPPORTED_FUNCTIONS.GREATER_THAN,
+      this,
+    );
+  }
+}