@finos/legend-query-builder 4.14.29 → 4.14.31

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

@@ -0,0 +1,108 @@
+/**
+ * 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 {
+  ColSpecArray,
+  type ValueSpecification,
+  extractElementNameFromPath,
+  SimpleFunctionExpression,
+  ColSpecArrayInstance,
+  Multiplicity,
+  ColSpec,
+  INTERNAL__UnknownValueSpecification,
+  V1_serializeRawValueSpecification,
+  V1_transformRawLambda,
+  V1_GraphTransformerContextBuilder,
+} from '@finos/legend-graph';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../graph/QueryBuilderMetaModelConst.js';
+import {
+  QueryBuilderDerivationProjectionColumnState,
+  QueryBuilderSimpleProjectionColumnState,
+} from './QueryBuilderProjectionColumnState.js';
+import type { QueryBuilderTDSState } from '../QueryBuilderTDSState.js';
+import { buildGenericLambdaFunctionInstanceValue } from '../../../QueryBuilderValueSpecificationHelper.js';
+import {
+  buildPropertyExpressionChain,
+  type LambdaFunctionBuilderOption,
+} from '../../../QueryBuilderValueSpecificationBuilderHelper.js';
+import { UnsupportedOperationError } from '@finos/legend-shared';
+
+export const buildRelationProjection = (
+  precedingExpression: ValueSpecification,
+  tdsState: QueryBuilderTDSState,
+  options?: LambdaFunctionBuilderOption,
+): SimpleFunctionExpression => {
+  const projectFunction = new SimpleFunctionExpression(
+    extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_PROJECT),
+  );
+  const queryBuilderState = tdsState.queryBuilderState;
+
+  const instanceVal = new ColSpecArrayInstance(Multiplicity.ONE, undefined);
+
+  const colSepcArray = new ColSpecArray();
+  instanceVal.values = [colSepcArray];
+
+  tdsState.projectionColumns.forEach((projectionColumnState) => {
+    const colSpec = new ColSpec();
+    colSepcArray.colSpecs.push(colSpec);
+    colSpec.name = projectionColumnState.columnName;
+    // column projection
+    let columnLambda: ValueSpecification;
+    if (
+      projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState
+    ) {
+      columnLambda = buildGenericLambdaFunctionInstanceValue(
+        projectionColumnState.lambdaParameterName,
+        [
+          buildPropertyExpressionChain(
+            projectionColumnState.propertyExpressionState.propertyExpression,
+            projectionColumnState.propertyExpressionState.queryBuilderState,
+            projectionColumnState.lambdaParameterName,
+            options,
+          ),
+        ],
+        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 relation project() column expression: unsupported projection column state`,
+        projectionColumnState,
+      );
+    }
+    colSpec.function1 = columnLambda;
+  });
+  projectFunction.parametersValues = [precedingExpression, instanceVal];
+  return projectFunction;
+};

package/src/stores/query-workflow/QueryBuilderWorkFlowState.ts

@@ -0,0 +1,64 @@
+/**
+ * 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 { QueryBuilderState } from '../QueryBuilderState.js';
+
+export interface FetchStructureLayoutConfig {
+  label: string;
+  showInFetchPanel: boolean;
+}
+
+export abstract class QueryBuilderWorkflowState {
+  abstract get showStatusBar(): boolean;
+
+  abstract getFetchStructureLayoutConfig(
+    state: QueryBuilderState,
+  ): FetchStructureLayoutConfig;
+}
+
+export class QueryBuilderAdvancedWorkflowState extends QueryBuilderWorkflowState {
+  get showStatusBar(): boolean {
+    return true;
+  }
+
+  override getFetchStructureLayoutConfig(
+    state: QueryBuilderState,
+  ): FetchStructureLayoutConfig {
+    return {
+      label: 'fetch structure',
+      showInFetchPanel: true,
+    };
+  }
+
+  static INSTANCE = new QueryBuilderAdvancedWorkflowState();
+}
+
+export class QueryBuilderDataBrowserWorkflow extends QueryBuilderWorkflowState {
+  override get showStatusBar(): boolean {
+    return false;
+  }
+
+  override getFetchStructureLayoutConfig(
+    state: QueryBuilderState,
+  ): FetchStructureLayoutConfig {
+    return {
+      label: state.fetchStructureState.implementation.fetchLabel,
+      showInFetchPanel: false,
+    };
+  }
+
+  static readonly INSTANCE = new QueryBuilderDataBrowserWorkflow();
+}

package/src/stores/workflows/FunctionQueryBuilderState.ts

@@ -21,6 +21,7 @@ import type {
 } from '@finos/legend-graph';
 import { ClassQueryBuilderState } from './ClassQueryBuilderState.js';
 import type { QueryBuilderConfig } from '../../graph-manager/QueryBuilderConfig.js';
+import type { QueryBuilderWorkflowState } from '../query-workflow/QueryBuilderWorkFlowState.js';
 
 // Note: We may want to move it to extend QueryBuilderState directly
 // but for now we will use the same setup as class as class, mapping, runtime are editable
@@ -30,10 +31,11 @@ export class FunctionQueryBuilderState extends ClassQueryBuilderState {
   constructor(
     applicationStore: GenericLegendApplicationStore,
     graphManagerState: GraphManagerState,
+    workflowState: QueryBuilderWorkflowState,
     functionElemenet: ConcreteFunctionDefinition,
     config: QueryBuilderConfig | undefined,
   ) {
-    super(applicationStore, graphManagerState, config);
+    super(applicationStore, graphManagerState, workflowState, config);
     this.functionElement = functionElemenet;
     this.showParametersPanel = true;
   }

package/src/stores/workflows/MappingQueryBuilderState.ts

@@ -28,6 +28,7 @@ import { getNullableFirstEntry } from '@finos/legend-shared';
 import { renderMappingQueryBuilderSetupPanelContent } from '../../components/workflows/MappingQueryBuilder.js';
 import { QueryBuilderState } from '../QueryBuilderState.js';
 import type { QueryBuilderConfig } from '../../graph-manager/QueryBuilderConfig.js';
+import type { QueryBuilderWorkflowState } from '../query-workflow/QueryBuilderWorkFlowState.js';
 
 export class MappingQueryBuilderState extends QueryBuilderState {
   readonly onMappingChange?: ((val: Mapping) => void) | undefined;
@@ -39,12 +40,19 @@ export class MappingQueryBuilderState extends QueryBuilderState {
   constructor(
     applicationStore: GenericLegendApplicationStore,
     graphManagerState: GraphManagerState,
+    workflowState: QueryBuilderWorkflowState,
     onMappingChange?: ((val: Mapping) => void) | undefined,
     onRuntimeChange?: ((val: Runtime) => void) | undefined,
     config?: QueryBuilderConfig | undefined,
     sourceInfo?: object | undefined,
   ) {
-    super(applicationStore, graphManagerState, config, sourceInfo);
+    super(
+      applicationStore,
+      graphManagerState,
+      workflowState,
+      config,
+      sourceInfo,
+    );
 
     this.onMappingChange = onMappingChange;
     this.onRuntimeChange = onRuntimeChange;

package/src/stores/workflows/ServiceQueryBuilderState.ts

@@ -34,6 +34,7 @@ import { action, makeObservable, observable } from 'mobx';
 import { renderServiceQueryBuilderSetupPanelContent } from '../../components/workflows/ServiceQueryBuilder.js';
 import { QueryBuilderState } from '../QueryBuilderState.js';
 import type { QueryBuilderConfig } from '../../graph-manager/QueryBuilderConfig.js';
+import type { QueryBuilderWorkflowState } from '../query-workflow/QueryBuilderWorkFlowState.js';
 
 export type ServiceExecutionContext = {
   key: string;
@@ -58,6 +59,7 @@ export class ServiceQueryBuilderState extends QueryBuilderState {
   constructor(
     applicationStore: GenericLegendApplicationStore,
     graphManagerState: GraphManagerState,
+    workflowState: QueryBuilderWorkflowState,
     service: Service,
     usableServices: Service[] | undefined,
     executionContextKey?: string | undefined,
@@ -68,7 +70,13 @@ export class ServiceQueryBuilderState extends QueryBuilderState {
     config?: QueryBuilderConfig | undefined,
     sourceInfo?: object | undefined,
   ) {
-    super(applicationStore, graphManagerState, config, sourceInfo);
+    super(
+      applicationStore,
+      graphManagerState,
+      workflowState,
+      config,
+      sourceInfo,
+    );
 
     makeObservable(this, {
       selectedExecutionContext: observable,

package/tsconfig.json

@@ -92,6 +92,7 @@
     "./src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeStateBuilder.ts",
     "./src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeUtil.ts",
     "./src/stores/fetch-structure/graph-fetch/QueryBuilderGraphFetchTreeValueSpecificationBuilder.ts",
+    "./src/stores/fetch-structure/tds/QueryBuilderAdvancedWorkflowState.ts",
     "./src/stores/fetch-structure/tds/QueryBuilderTDSColumnState.ts",
     "./src/stores/fetch-structure/tds/QueryBuilderTDSHelper.ts",
     "./src/stores/fetch-structure/tds/QueryBuilderTDSState.ts",
@@ -134,6 +135,7 @@
     "./src/stores/fetch-structure/tds/projection/QueryBuilderProjectionColumnState.ts",
     "./src/stores/fetch-structure/tds/projection/QueryBuilderProjectionStateBuilder.ts",
     "./src/stores/fetch-structure/tds/projection/QueryBuilderProjectionValueSpecificationBuilder.ts",
+    "./src/stores/fetch-structure/tds/projection/QueryBuilderRelationProjectValueSpecBuidler.ts",
     "./src/stores/fetch-structure/tds/window/QueryBuilderWindowGroupByOperatorLoader.ts",
     "./src/stores/fetch-structure/tds/window/QueryBuilderWindowState.ts",
     "./src/stores/fetch-structure/tds/window/QueryBuilderWindowStateBuilder.ts",
@@ -170,6 +172,7 @@
     "./src/stores/milestoning/QueryBuilderMilestoningImplementation.ts",
     "./src/stores/milestoning/QueryBuilderMilestoningState.ts",
     "./src/stores/milestoning/QueryBuilderProcessingTemporalMilestoningImplementation.ts",
+    "./src/stores/query-workflow/QueryBuilderWorkFlowState.ts",
     "./src/stores/shared/LambdaEditorState.ts",
     "./src/stores/shared/LambdaParameterState.ts",
     "./src/stores/shared/ValueSpecificationEditorHelper.ts",