@finos/legend-query-builder 4.14.29 → 4.14.30

package/src/components/__test-utils__/QueryBuilderComponentTestUtils.tsx

@@ -51,6 +51,7 @@ import {
 } from '../../stores/__test-utils__/QueryBuilderStateTestUtils.js';
 import { STYLE_PREFIX, STYLE_PREFIX__DARK } from '@finos/legend-art';
 import { expect } from '@jest/globals';
+import { QueryBuilderAdvancedWorkflowState } from '../../stores/workflow/QueryBuilderWorkFlowState.js';
 
 const getSelectorContainerClassName = (lightMode?: boolean): string =>
   '.' + `${lightMode ? STYLE_PREFIX : STYLE_PREFIX__DARK}__value-container`;
@@ -198,6 +199,7 @@ export const TEST__setUpQueryBuilder = async (
   const queryBuilderState = new INTERNAL__BasicQueryBuilderState(
     MOCK__applicationStore,
     graphManagerState,
+    QueryBuilderAdvancedWorkflowState.INSTANCE,
     undefined,
   );
   const mapping = graphManagerState.graph.getMapping(mappingPath);

package/src/index.ts

@@ -93,3 +93,4 @@ export * from './stores/QueryBuilder_LegendApplicationPlugin_Extension.js';
 
 export * from './stores/data-access/DataAccessState.js';
 export * from './components/data-access/DataAccessOverview.js';
+export * from './stores/workflow/QueryBuilderWorkFlowState.js';

package/src/stores/QueryBuilderState.ts

@@ -109,6 +109,7 @@ import {
 import type { QueryBuilderConfig } from '../graph-manager/QueryBuilderConfig.js';
 import { QUERY_BUILDER_EVENT } from '../__lib__/QueryBuilderEvent.js';
 import { QueryBuilderChangeHistoryState } from './QueryBuilderChangeHistoryState.js';
+import { type QueryBuilderWorkflowState } from './workflow/QueryBuilderWorkFlowState.js';
 
 export interface QuerySDLC {}
 
@@ -118,6 +119,11 @@ export type QueryStateInfo = QuerySDLC & {
   runtime: string;
 };
 
+export enum QUERY_BUILDER_LAMBDA_WRITER_MODE {
+  STANDARD = 'STANDARD',
+  TYPED_FETCH_STRUCTURE = 'TYPED_FETCH_STRUCTURE',
+}
+
 export abstract class QueryBuilderState implements CommandRegistrar {
   readonly applicationStore: GenericLegendApplicationStore;
   readonly graphManagerState: GraphManagerState;
@@ -126,6 +132,7 @@ export abstract class QueryBuilderState implements CommandRegistrar {
   readonly queryCompileState = ActionState.create();
   readonly observerContext: ObserverContext;
   readonly config: QueryBuilderConfig | undefined;
+  readonly workflowState: QueryBuilderWorkflowState;
 
   explorerState: QueryBuilderExplorerState;
   functionsExplorerState: QueryFunctionsExplorerState;
@@ -168,6 +175,7 @@ export abstract class QueryBuilderState implements CommandRegistrar {
   constructor(
     applicationStore: GenericLegendApplicationStore,
     graphManagerState: GraphManagerState,
+    workflowState: QueryBuilderWorkflowState,
     config: QueryBuilderConfig | undefined,
     sourceInfo?: QuerySDLC | undefined,
   ) {
@@ -245,6 +253,8 @@ export abstract class QueryBuilderState implements CommandRegistrar {
     this.changeDetectionState = new QueryBuilderChangeDetectionState(this);
     this.changeHistoryState = new QueryBuilderChangeHistoryState(this);
     this.config = config;
+
+    this.workflowState = workflowState;
     this.sourceInfo = sourceInfo;
   }
 
@@ -789,6 +799,7 @@ export abstract class QueryBuilderState implements CommandRegistrar {
     const basicState = new INTERNAL__BasicQueryBuilderState(
       this.applicationStore,
       this.graphManagerState,
+      this.workflowState,
       undefined,
     );
     basicState.class = this.class;

package/src/stores/__test-utils__/QueryBuilderStateTestUtils.ts

@@ -41,6 +41,7 @@ import {
   INTERNAL__BasicQueryBuilderState,
   type QueryBuilderState,
 } from '../QueryBuilderState.js';
+import { QueryBuilderAdvancedWorkflowState } from '../workflow/QueryBuilderWorkFlowState.js';
 
 export class TEST__LegendApplicationPluginManager
   extends LegendApplicationPluginManager<LegendApplicationPlugin>
@@ -131,6 +132,7 @@ export const TEST__setUpQueryBuilderState = async (
   const queryBuilderState = new INTERNAL__BasicQueryBuilderState(
     applicationStore,
     graphManagerState,
+    QueryBuilderAdvancedWorkflowState.INSTANCE,
     undefined,
   );
   if (rawLambda) {

package/src/stores/workflow/QueryBuilderWorkFlowState.ts

@@ -0,0 +1,27 @@
+/**
+ * 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.
+ */
+
+export abstract class QueryBuilderWorkflowState {
+  abstract get showStatusBar(): boolean;
+}
+
+export class QueryBuilderAdvancedWorkflowState extends QueryBuilderWorkflowState {
+  get showStatusBar(): boolean {
+    return true;
+  }
+
+  static INSTANCE = new QueryBuilderAdvancedWorkflowState();
+}

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 '../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 '../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 '../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

@@ -177,6 +177,7 @@
     "./src/stores/watermark/QueryBuilderWatermarkState.ts",
     "./src/stores/watermark/QueryBuilderWatermarkStateBuilder.ts",
     "./src/stores/watermark/QueryBuilderWatermarkValueSpecificationBuilder.ts",
+    "./src/stores/workflow/QueryBuilderWorkFlowState.ts",
     "./src/stores/workflows/ClassQueryBuilderState.ts",
     "./src/stores/workflows/FunctionQueryBuilderState.ts",
     "./src/stores/workflows/MappingQueryBuilderState.ts",