@finos/legend-query-builder 4.7.4 → 4.9.0

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

@@ -14,7 +14,14 @@
  * limitations under the License.
  */
 
-import { action, makeObservable, observable, flow, computed } from 'mobx';
+import {
+  action,
+  makeObservable,
+  observable,
+  flow,
+  computed,
+  flowResult,
+} from 'mobx';
 import {
   type GeneratorFn,
   LogEvent,
@@ -24,6 +31,7 @@ import {
   assertNonEmptyString,
   type Hashable,
   hashArray,
+  ActionState,
 } from '@finos/legend-shared';
 import {
   type QueryBuilderExplorerTreePropertyNodeData,
@@ -275,6 +283,7 @@ export class QueryBuilderDerivationProjectionColumnState
   derivationLambdaEditorState: QueryBuilderDerivationProjectionLambdaState;
   lambda: RawLambda;
   returnType: Type | undefined;
+  fetchingLambdaReturnTypeState = ActionState.create();
 
   constructor(tdsState: QueryBuilderTDSState, lambda: RawLambda) {
     super(tdsState, '(derivation)');
@@ -282,8 +291,10 @@ export class QueryBuilderDerivationProjectionColumnState
     makeObservable(this, {
       lambda: observable,
       returnType: observable,
+      fetchingLambdaReturnTypeState: observable,
       setLambda: action,
       fetchDerivationLambdaReturnType: flow,
+      setLambdaReturnType: action,
     });
 
     this.derivationLambdaEditorState =
@@ -307,10 +318,56 @@ export class QueryBuilderDerivationProjectionColumnState
    * Throws error if unable to fetch type or if type is not primitive or an enumeration
    * as expected by a projection column
    */
-  *fetchDerivationLambdaReturnType(): GeneratorFn<void> {
+  *fetchDerivationLambdaReturnType(options?: {
+    forceRefresh?: boolean;
+    forceConversionStringToLambda?: boolean;
+    isBeingDropped?: boolean;
+  }): GeneratorFn<void> {
+    if (!options?.forceRefresh && this.returnType !== undefined) {
+      return;
+    }
+    try {
+      assertTrue(
+        !this.fetchingLambdaReturnTypeState.isInProgress,
+        'Fetching lambda return type already in progress',
+      );
+      this.fetchingLambdaReturnTypeState.inProgress();
+      if (options?.isBeingDropped) {
+        this.tdsState.postFilterState.setDerivedColumnBeingDropped(this);
+      }
+      if (options?.forceConversionStringToLambda) {
+        yield flowResult(
+          this.derivationLambdaEditorState.convertLambdaGrammarStringToObject(),
+        ).catch(
+          this.tdsState.queryBuilderState.applicationStore.alertUnhandledError,
+        );
+      }
+      assertTrue(Array.isArray(this.lambda.parameters));
+      const graph = this.tdsState.queryBuilderState.graphManagerState.graph;
+      const isolatedLambda = this.getIsolatedRawLambda();
+      const type =
+        (yield this.tdsState.queryBuilderState.graphManagerState.graphManager.getLambdaReturnType(
+          isolatedLambda,
+          graph,
+        )) as string;
+      this.setLambdaReturnType(type);
+    } catch (error) {
+      assertErrorThrown(error);
+      this.tdsState.queryBuilderState.applicationStore.logService.info(
+        LogEvent.create(GRAPH_MANAGER_EVENT.COMPILATION_FAILURE),
+        error,
+      );
+    } finally {
+      this.fetchingLambdaReturnTypeState.complete();
+      if (options?.isBeingDropped) {
+        this.tdsState.postFilterState.setDerivedColumnBeingDropped(undefined);
+      }
+    }
+  }
+
+  getIsolatedRawLambda(): RawLambda {
     assertTrue(Array.isArray(this.lambda.parameters));
     const projectionParameter = this.lambda.parameters as object[];
-    const graph = this.tdsState.queryBuilderState.graphManagerState.graph;
     assertTrue(projectionParameter.length === 1);
     const variable = projectionParameter[0] as VariableExpression;
     assertNonEmptyString(variable.name);
@@ -330,12 +387,12 @@ export class QueryBuilderDerivationProjectionColumnState
       [_rawVariableExpression],
       this.lambda.body,
     );
-    const type =
-      (yield this.tdsState.queryBuilderState.graphManagerState.graphManager.getLambdaReturnType(
-        isolatedLambda,
-        graph,
-      )) as string;
-    const resolvedType = graph.getType(type);
+    return isolatedLambda;
+  }
+
+  setLambdaReturnType(type: string): void {
+    const resolvedType =
+      this.tdsState.queryBuilderState.graphManagerState.graph.getType(type);
     assertTrue(
       resolvedType instanceof PrimitiveType ||
         resolvedType instanceof Enumeration,

package/src/stores/milestoning/QueryBuilderBitemporalMilestoningImplementation.ts

@@ -22,6 +22,7 @@ import {
   AbstractPropertyExpression,
   MILESTONING_STEREOTYPE,
   INTERNAL__PropagatedValue,
+  PrimitiveType,
 } from '@finos/legend-graph';
 import {
   assertTrue,
@@ -29,7 +30,9 @@ import {
   guaranteeType,
 } from '@finos/legend-shared';
 import { getParameterValue } from '../../components/QueryBuilderSideBar.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../graph/QueryBuilderMetaModelConst.js';
 import type { QueryBuilderDerivedPropertyExpressionState } from '../QueryBuilderPropertyEditorState.js';
+import { createSupportedFunctionExpression } from '../shared/ValueSpecificationEditorHelper.js';
 import { QueryBuilderMilestoningImplementation } from './QueryBuilderMilestoningImplementation.js';
 
 export class QueryBuilderBitemporalMilestoningImplementation extends QueryBuilderMilestoningImplementation {
@@ -90,6 +93,17 @@ export class QueryBuilderBitemporalMilestoningImplementation extends QueryBuilde
     );
   }
 
+  buildGetAllWithDefaultParameters(
+    getAllFunction: SimpleFunctionExpression,
+  ): void {
+    const parameterValue = createSupportedFunctionExpression(
+      QUERY_BUILDER_SUPPORTED_FUNCTIONS.NOW,
+      PrimitiveType.DATETIME,
+    );
+    getAllFunction.parametersValues.push(parameterValue);
+    getAllFunction.parametersValues.push(parameterValue);
+  }
+
   generateMilestoningDate(
     isDatePropagationSupported: boolean,
     hasDefaultMilestoningDate: boolean,

package/src/stores/milestoning/QueryBuilderBusinessTemporalMilestoningImplementation.ts

@@ -21,9 +21,12 @@ import {
   type AbstractPropertyExpression,
   MILESTONING_STEREOTYPE,
   INTERNAL__PropagatedValue,
+  PrimitiveType,
 } from '@finos/legend-graph';
 import { assertTrue, guaranteeNonNullable } from '@finos/legend-shared';
 import { getParameterValue } from '../../components/QueryBuilderSideBar.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../graph/QueryBuilderMetaModelConst.js';
+import { createSupportedFunctionExpression } from '../shared/ValueSpecificationEditorHelper.js';
 import { QueryBuilderMilestoningImplementation } from './QueryBuilderMilestoningImplementation.js';
 
 export class QueryBuilderBusinessTemporalMilestoningImplementation extends QueryBuilderMilestoningImplementation {
@@ -64,6 +67,16 @@ export class QueryBuilderBusinessTemporalMilestoningImplementation extends Query
     );
   }
 
+  buildGetAllWithDefaultParameters(
+    getAllFunction: SimpleFunctionExpression,
+  ): void {
+    const parameterValue = createSupportedFunctionExpression(
+      QUERY_BUILDER_SUPPORTED_FUNCTIONS.NOW,
+      PrimitiveType.DATETIME,
+    );
+    getAllFunction.parametersValues.push(parameterValue);
+  }
+
   generateMilestoningDate(
     isDatePropagationSupported: boolean,
     hasDefaultMilestoningDate: boolean,

package/src/stores/milestoning/QueryBuilderMilestoningImplementation.ts

@@ -63,6 +63,13 @@ export abstract class QueryBuilderMilestoningImplementation {
     getAllFunction: SimpleFunctionExpression,
   ): void;
 
+  /**
+   * Builds parameters for getAll() function with milestoned class
+   */
+  abstract buildGetAllWithDefaultParameters(
+    getAllFunction: SimpleFunctionExpression,
+  ): void;
+
   /**
    * Generates milestoning date for a propertyexpression based on its source and target stereotype
    */

package/src/stores/milestoning/QueryBuilderProcessingTemporalMilestoningImplementation.ts

@@ -20,9 +20,12 @@ import {
   type SimpleFunctionExpression,
   type AbstractPropertyExpression,
   INTERNAL__PropagatedValue,
+  PrimitiveType,
 } from '@finos/legend-graph';
 import { assertTrue, guaranteeNonNullable } from '@finos/legend-shared';
 import { getParameterValue } from '../../components/QueryBuilderSideBar.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../graph/QueryBuilderMetaModelConst.js';
+import { createSupportedFunctionExpression } from '../shared/ValueSpecificationEditorHelper.js';
 import { QueryBuilderMilestoningImplementation } from './QueryBuilderMilestoningImplementation.js';
 
 export class QueryBuilderProcessingTemporalMilestoningImplementation extends QueryBuilderMilestoningImplementation {
@@ -63,6 +66,16 @@ export class QueryBuilderProcessingTemporalMilestoningImplementation extends Que
     );
   }
 
+  buildGetAllWithDefaultParameters(
+    getAllFunction: SimpleFunctionExpression,
+  ): void {
+    const parameterValue = createSupportedFunctionExpression(
+      QUERY_BUILDER_SUPPORTED_FUNCTIONS.NOW,
+      PrimitiveType.DATETIME,
+    );
+    getAllFunction.parametersValues.push(parameterValue);
+  }
+
   generateMilestoningDate(
     isDatePropagationSupported: boolean,
     hasDefaultMilestoningDate: boolean,