@finos/legend-query-builder 4.17.82 → 4.17.84

package/lib/stores/workflows/dataProduct/DataProductQueryBuilderState.js

@@ -0,0 +1,131 @@
+/**
+ * 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 { DataProduct, ModelAccessPointGroup, CORE_PURE_PATH, resolveUsableDataProductClasses, } from '@finos/legend-graph';
+import { QueryBuilderState } from '../../QueryBuilderState.js';
+import {} from '@finos/legend-application';
+import { renderDataProductQueryBuilderSetupPanelContent } from '../../../components/workflows/DataProductQueryBuilder.js';
+import { ActionState, filterByType, } from '@finos/legend-shared';
+import { action, computed, flow, makeObservable, observable } from 'mobx';
+import { DepotEntityWithOrigin, } from '@finos/legend-storage';
+import { compareLabelFn } from '@finos/legend-art';
+export const buildDataProductOption = (value) => ({
+    label: value.name,
+    value,
+});
+export const buildExecOptions = (val) => {
+    return {
+        label: val.key,
+        value: val,
+    };
+};
+export class DataProductQueryBuilderState extends QueryBuilderState {
+    onClassChange;
+    onDataProductChange;
+    onExecutionContextChange;
+    loadDataProductModelState = ActionState.create();
+    nativeModelAccess;
+    dataProduct;
+    dataProductArtifact;
+    selectedExecContext;
+    entities;
+    prioritizeEntityFunc;
+    TEMPORARY__setupPanelContentRenderer = () => renderDataProductQueryBuilderSetupPanelContent(this);
+    constructor(applicationStore, graphManagerState, workflow, dataProduct, artifact, actionConfig, nativeNativeModelAccess, nativeModelExecContext, prioritizeEntityFunc, onDataProductChange, onExecutionContextChange, onClassChange, config, sourceInfo) {
+        super(applicationStore, graphManagerState, workflow, config, sourceInfo);
+        makeObservable(this, {
+            selectedExecContext: observable,
+            dataProduct: observable,
+            setExecOptions: action,
+            selectedDataProductOption: computed,
+            isProductLinkable: computed,
+            loadEntities: flow,
+            entities: observable,
+        });
+        this.workflowState.updateActionConfig(actionConfig);
+        this.dataProduct = dataProduct;
+        this.dataProductArtifact = artifact;
+        this.nativeModelAccess = nativeNativeModelAccess;
+        this.selectedExecContext = nativeModelExecContext;
+        this.prioritizeEntityFunc = prioritizeEntityFunc;
+        this.onDataProductChange = onDataProductChange;
+        this.onExecutionContextChange = onExecutionContextChange;
+        this.onClassChange = onClassChange;
+    }
+    get isProductLinkable() {
+        return false;
+    }
+    copyDataProductLinkToClipBoard() {
+        if (!this.isProductLinkable) {
+            this.applicationStore.notificationService.notifyError('Data Product link is not available.');
+        }
+    }
+    getElementType() {
+        return DataProduct;
+    }
+    setExecOptions(exec) {
+        this.selectedExecContext = exec;
+    }
+    *loadEntities() {
+        this.loadDataProductModelState.inProgress();
+        this.entities = this.graphManagerState.graph.allOwnElements
+            .filter(filterByType(this.getElementType()))
+            .map((element) => this.transformElement(element));
+        this.loadDataProductModelState.complete();
+    }
+    transformElement(element) {
+        return new DepotEntityWithOrigin(undefined, element.name, element.path, CORE_PURE_PATH.DATA_PRODUCT);
+    }
+    get dataProductOptions() {
+        const sortedAllOptions = (this.entities ?? [])
+            .map(buildDataProductOption)
+            .sort(compareLabelFn);
+        return this.prioritizeEntityFunc
+            ? [
+                ...sortedAllOptions.filter((val) => this.prioritizeEntityFunc?.(val.value)),
+                ...sortedAllOptions.filter((val) => !this.prioritizeEntityFunc?.(val.value)),
+            ]
+            : sortedAllOptions;
+    }
+    get selectedDataProductOption() {
+        return {
+            label: this.dataProduct.title ?? this.dataProduct.name,
+            value: new DepotEntityWithOrigin(undefined, this.dataProduct.name, this.dataProduct.path, CORE_PURE_PATH.DATA_PRODUCT),
+        };
+    }
+    get isSupported() {
+        return (this.dataProduct.nativeModelAccess !== undefined ||
+            // contains model access point group
+            this.dataProduct.accessPointGroups.filter(filterByType(ModelAccessPointGroup)).length > 0);
+    }
+    // includes model access point group if more than one group
+    get execOptions() {
+        return (this.dataProduct.nativeModelAccess?.nativeModelExecutionContexts.map(buildExecOptions) ?? []).sort(compareLabelFn);
+    }
+    async propagateExecutionContextChange(requireReBuildingGraph) {
+        const currentMapping = this.executionContextState.mapping;
+        const execMapping = this.selectedExecContext.mapping.value;
+        if (execMapping !== currentMapping) {
+            this.changeMapping(execMapping, {
+                keepQueryContent: true,
+            });
+            const classes = resolveUsableDataProductClasses(this.nativeModelAccess.featuredElements, this.selectedExecContext.mapping.value, this.graphManagerState, undefined);
+            if (this.class && !classes.includes(this.class)) {
+                this.setClass(classes[0]);
+            }
+        }
+    }
+}
+//# sourceMappingURL=DataProductQueryBuilderState.js.map

package/lib/stores/workflows/dataProduct/DataProductQueryBuilderState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DataProductQueryBuilderState.js","sourceRoot":"","sources":["../../../../src/stores/workflows/dataProduct/DataProductQueryBuilderState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAGL,WAAW,EAEX,qBAAqB,EAErB,cAAc,EAEd,+BAA+B,GAChC,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAsC,MAAM,2BAA2B,CAAC;AAM/E,OAAO,EAAE,8CAA8C,EAAE,MAAM,0DAA0D,CAAC;AAC1H,OAAO,EACL,WAAW,EACX,YAAY,GAEb,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAC1E,OAAO,EACL,qBAAqB,GAEtB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAOnD,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,KAA4B,EACT,EAAE,CAAC,CAAC;IACvB,KAAK,EAAE,KAAK,CAAC,IAAI;IACjB,KAAK;CACN,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,GAAgC,EAIhC,EAAE;IACF,OAAO;QACL,KAAK,EAAE,GAAG,CAAC,GAAG;QACd,KAAK,EAAE,GAAG;KACX,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,OAAO,4BAA6B,SAAQ,iBAAiB;IACxD,aAAa,CAAsC;IACnD,mBAAmB,CAAiD;IACpE,wBAAwB,CAEnB;IAEd,yBAAyB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IACjD,iBAAiB,CAAoB;IACrC,WAAW,CAAc;IACzB,mBAAmB,CAAqC;IACxD,mBAAmB,CAA8B;IACjD,QAAQ,CAAsC;IAE9C,oBAAoB,CAAyD;IAEpE,oCAAoC,GAAG,GAAoB,EAAE,CACpE,8CAA8C,CAAC,IAAI,CAAC,CAAC;IAEvD,YACE,gBAA+C,EAC/C,iBAAoC,EACpC,QAAmC,EACnC,WAAwB,EACxB,QAA4C,EAC5C,YAAsC,EACtC,uBAA0C,EAC1C,sBAAmD,EACnD,oBAA2E,EAC3E,mBAAkE,EAClE,wBAEa,EACb,aAAkD,EAClD,MAAuC,EACvC,UAA4C;QAE5C,KAAK,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACzE,cAAc,CAAC,IAAI,EAAE;YACnB,mBAAmB,EAAE,UAAU;YAC/B,WAAW,EAAE,UAAU;YACvB,cAAc,EAAE,MAAM;YACtB,yBAAyB,EAAE,QAAQ;YACnC,iBAAiB,EAAE,QAAQ;YAC3B,YAAY,EAAE,IAAI;YAClB,QAAQ,EAAE,UAAU;SACrB,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACpD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,mBAAmB,GAAG,QAAQ,CAAC;QACpC,IAAI,CAAC,iBAAiB,GAAG,uBAAuB,CAAC;QACjD,IAAI,CAAC,mBAAmB,GAAG,sBAAsB,CAAC;QAClD,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;QACjD,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,wBAAwB,GAAG,wBAAwB,CAAC;QACzD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACrC,CAAC;IAED,IAAI,iBAAiB;QACnB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,8BAA8B;QAC5B,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CACnD,qCAAqC,CACtC,CAAC;QACJ,CAAC;IACH,CAAC;IAES,cAAc;QACtB,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,cAAc,CAAC,IAAiC;QAC9C,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAClC,CAAC;IAED,CAAC,YAAY;QACX,IAAI,CAAC,yBAAyB,CAAC,UAAU,EAAE,CAAC;QAC5C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,cAAc;aACxD,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;aAC3C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QACpD,IAAI,CAAC,yBAAyB,CAAC,QAAQ,EAAE,CAAC;IAC5C,CAAC;IAES,gBAAgB,CAAC,OAAoB;QAC7C,OAAO,IAAI,qBAAqB,CAC9B,SAAS,EACT,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,IAAI,EACZ,cAAc,CAAC,YAAY,CAC5B,CAAC;IACJ,CAAC;IAED,IAAI,kBAAkB;QACpB,MAAM,gBAAgB,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;aAC3C,GAAG,CAAC,sBAAsB,CAAC;aAC3B,IAAI,CAAC,cAAc,CAAC,CAAC;QAExB,OAAO,IAAI,CAAC,oBAAoB;YAC9B,CAAC,CAAC;gBACE,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CACjC,IAAI,CAAC,oBAAoB,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC;gBACD,GAAG,gBAAgB,CAAC,MAAM,CACxB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CACjD;aACF;YACH,CAAC,CAAC,gBAAgB,CAAC;IACvB,CAAC;IAED,IAAI,yBAAyB;QAC3B,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI;YACtD,KAAK,EAAE,IAAI,qBAAqB,CAC9B,SAAS,EACT,IAAI,CAAC,WAAW,CAAC,IAAI,EACrB,IAAI,CAAC,WAAW,CAAC,IAAI,EACrB,cAAc,CAAC,YAAY,CAC5B;SACF,CAAC;IACJ,CAAC;IAED,IAAI,WAAW;QACb,OAAO,CACL,IAAI,CAAC,WAAW,CAAC,iBAAiB,KAAK,SAAS;YAChD,oCAAoC;YACpC,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,MAAM,CACvC,YAAY,CAAC,qBAAqB,CAAC,CACpC,CAAC,MAAM,GAAG,CAAC,CACb,CAAC;IACJ,CAAC;IAED,2DAA2D;IAC3D,IAAI,WAAW;QACb,OAAO,CACL,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,4BAA4B,CAAC,GAAG,CAClE,gBAAgB,CACjB,IAAI,EAAE,CACR,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACzB,CAAC;IAEQ,KAAK,CAAC,+BAA+B,CAC5C,sBAA4C;QAE5C,MAAM,cAAc,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC;QAC1D,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,KAAK,CAAC;QAC3D,IAAI,WAAW,KAAK,cAAc,EAAE,CAAC;YACnC,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE;gBAC9B,gBAAgB,EAAE,IAAI;aACvB,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,+BAA+B,CAC7C,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,EACvC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,KAAK,EACtC,IAAI,CAAC,iBAAiB,EACtB,SAAS,CACV,CAAC;YACF,IAAI,IAAI,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChD,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@finos/legend-query-builder",
-  "version": "4.17.82",
+  "version": "4.17.84",
   "description": "Legend query builder core",
   "keywords": [
     "legend",
@@ -43,15 +43,15 @@
     "test:watch": "jest --watch"
   },
   "dependencies": {
-    "@finos/legend-application": "16.0.93",
+    "@finos/legend-application": "16.0.95",
     "@finos/legend-art": "7.1.141",
-    "@finos/legend-code-editor": "2.0.149",
-    "@finos/legend-data-cube": "0.3.64",
-    "@finos/legend-graph": "32.3.29",
-    "@finos/legend-lego": "2.0.166",
-    "@finos/legend-server-depot": "6.1.7",
+    "@finos/legend-code-editor": "2.0.151",
+    "@finos/legend-data-cube": "0.3.66",
+    "@finos/legend-graph": "32.3.31",
+    "@finos/legend-lego": "2.0.168",
+    "@finos/legend-server-depot": "6.1.9",
     "@finos/legend-shared": "11.0.22",
-    "@finos/legend-storage": "3.0.141",
+    "@finos/legend-storage": "3.0.143",
     "@testing-library/dom": "10.4.0",
     "@testing-library/react": "16.2.0",
     "@types/react": "19.0.10",

package/src/components/workflows/DataProductQueryBuilder.tsx

@@ -0,0 +1,198 @@
+/**
+ * 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 { useApplicationStore } from '@finos/legend-application';
+import {
+  buildExecOptions,
+  type DataProductOption,
+  type DataProductQueryBuilderState,
+} from '../../stores/workflows/dataProduct/DataProductQueryBuilderState.js';
+import { CustomSelectorInput } from '@finos/legend-art';
+import { QueryBuilderClassSelector } from '../QueryBuilderSideBar.js';
+import { observer } from 'mobx-react-lite';
+import {
+  resolveUsableDataProductClasses,
+  type NativeModelExecutionContext,
+} from '@finos/legend-graph';
+import type { DepotEntityWithOrigin } from '@finos/legend-storage';
+import { useEffect } from 'react';
+import { flowResult } from 'mobx';
+
+export type NativeExecutionContextOption = {
+  label: string;
+  value: NativeModelExecutionContext;
+};
+
+export const buildNativeExecutionContextOption = (
+  value: NativeModelExecutionContext,
+): NativeExecutionContextOption => ({
+  label: value.key,
+  value,
+});
+
+export type DataProductRuntimeInfoOption = {
+  label: string;
+  value: NativeModelExecutionContext;
+};
+export const buildDataProductRuntimeInfoOption = (
+  value: NativeModelExecutionContext,
+): DataProductRuntimeInfoOption => ({
+  label: value.key,
+  value,
+});
+
+export const buildDataProductOption = (
+  value: DepotEntityWithOrigin,
+): DataProductOption => ({
+  label: value.name,
+  value,
+});
+
+const DataProductQueryBuilderSetupPanelContent = observer(
+  (props: { queryBuilderState: DataProductQueryBuilderState }) => {
+    const { queryBuilderState } = props;
+    const applicationStore = useApplicationStore();
+    // data product
+    const nativeAccessGroup = queryBuilderState.nativeModelAccess;
+    const selectedExecContext = queryBuilderState.selectedExecContext;
+    const dataProductOptions = queryBuilderState.dataProductOptions;
+    const selectedDataProductOption =
+      queryBuilderState.selectedDataProductOption;
+    const onDataProductOptionChange = (
+      option: DataProductOption | null,
+    ): void => {
+      if (option?.value) {
+        queryBuilderState
+          .onDataProductChange?.(option.value)
+          .catch(queryBuilderState.applicationStore.alertUnhandledError);
+      }
+    };
+    // execution context
+    const executionContextOptions = queryBuilderState.execOptions;
+    const showExecutionContextOptions = executionContextOptions.length > 1;
+    const selectedExecOptions = buildExecOptions(selectedExecContext);
+    // class
+    const classes = resolveUsableDataProductClasses(
+      nativeAccessGroup.featuredElements,
+      selectedExecContext.mapping.value,
+      queryBuilderState.graphManagerState,
+      undefined,
+    );
+
+    useEffect(() => {
+      flowResult(queryBuilderState.loadEntities()).catch(
+        applicationStore.alertUnhandledError,
+      );
+    }, [queryBuilderState, applicationStore]);
+
+    const onExecutionContextOptionChange = async (
+      option: NativeExecutionContextOption,
+    ): Promise<void> => {
+      if (option.value === queryBuilderState.selectedExecContext) {
+        return;
+      }
+      const currentMapping =
+        queryBuilderState.selectedExecContext.mapping.value.path;
+      queryBuilderState.setExecOptions(option.value);
+      await queryBuilderState.propagateExecutionContextChange(
+        currentMapping === option.value.mapping.value.path,
+      );
+      queryBuilderState.onExecutionContextChange?.(option.value);
+    };
+
+    const handleExecutionContextOptionChange = (
+      option: NativeExecutionContextOption,
+    ): void => {
+      flowResult(onExecutionContextOptionChange(option));
+    };
+
+    return (
+      <div className="query-builder__setup__config-group">
+        <div className="query-builder__setup__config-group__content">
+          <div className="query-builder__setup__config-group__item">
+            <label
+              className="btn--sm query-builder__setup__config-group__data-product"
+              title="data product"
+              htmlFor="query-builder__setup__data-product-selector"
+            >
+              Data Product
+            </label>
+            <CustomSelectorInput
+              inputId="query-builder__setup__data-product-selector"
+              className="panel__content__form__section__dropdown query-builder__setup__config-group__item__selector"
+              options={dataProductOptions}
+              isLoading={
+                queryBuilderState.loadDataProductModelState.isInProgress
+              }
+              onChange={onDataProductOptionChange}
+              value={selectedDataProductOption}
+              placeholder="Search for data product..."
+              escapeClearsValue={true}
+              darkMode={
+                !applicationStore.layoutService
+                  .TEMPORARY__isLightColorThemeEnabled
+              }
+            />
+          </div>
+          {Boolean(showExecutionContextOptions) && (
+            <div className="query-builder__setup__config-group__item">
+              <label
+                className="btn--sm query-builder__setup__config-group__item__label"
+                title="execution context"
+                htmlFor="query-builder__setup__context-selector"
+              >
+                Context
+              </label>
+              <CustomSelectorInput
+                inputId="query-builder__setup__context-selector"
+                className="panel__content__form__section__dropdown query-builder__setup__config-group__item__selector"
+                placeholder="Choose an execution context..."
+                options={executionContextOptions}
+                disabled={
+                  executionContextOptions.length < 1 ||
+                  (executionContextOptions.length === 1 &&
+                    Boolean(selectedExecOptions))
+                }
+                onChange={handleExecutionContextOptionChange}
+                value={selectedExecOptions}
+                darkMode={
+                  !applicationStore.layoutService
+                    .TEMPORARY__isLightColorThemeEnabled
+                }
+              />
+            </div>
+          )}
+          <div className="query-builder__setup__config-group__item">
+            <QueryBuilderClassSelector
+              queryBuilderState={queryBuilderState}
+              classes={classes}
+              onClassChange={queryBuilderState.onClassChange}
+              noMatchMessage="No compatible entity found for specified execution context"
+            />
+          </div>
+        </div>
+      </div>
+    );
+  },
+);
+
+export const renderDataProductQueryBuilderSetupPanelContent = (
+  queryBuilderState: DataProductQueryBuilderState,
+): React.ReactNode => (
+  <DataProductQueryBuilderSetupPanelContent
+    queryBuilderState={queryBuilderState}
+  />
+);

package/src/index.ts

@@ -34,9 +34,11 @@ export { QueryDataCubeViewer } from './components/data-cube/QueryBuilderDataCube
 export { QueryBuilder } from './components/QueryBuilder.js';
 export { QUERY_BUILDER_COMPONENT_ELEMENT_ID } from './components/QueryBuilderComponentElement.js';
 export {
-  type QueryableSourceInfo as QuerySDLC,
   type QueryBuilderExtraFunctionAnalysisInfo,
+  type EntityWithOriginOption,
   QueryBuilderState,
+  ExtraOptionsConfig,
+  BaseQueryBuilderState,
   QUERY_BUILDER_LAMBDA_WRITER_MODE,
   INTERNAL__BasicQueryBuilderState,
 } from './stores/QueryBuilderState.js';
@@ -58,6 +60,7 @@ export {
 } from './components/QueryBuilderSideBar.js';
 export { ClassQueryBuilderState } from './stores/workflows/ClassQueryBuilderState.js';
 export { FunctionQueryBuilderState } from './stores/workflows/FunctionQueryBuilderState.js';
+export { DataProductQueryBuilderState } from './stores/workflows/dataProduct/DataProductQueryBuilderState.js';
 export { MappingQueryBuilderState } from './stores/workflows/MappingQueryBuilderState.js';
 export {
   type ServiceExecutionContext,

package/src/stores/QueryBuilderState.ts

@@ -116,9 +116,10 @@ import type { QueryBuilder_LegendApplicationPlugin_Extension } from './QueryBuil
 import { createDataCubeViewerStateFromQueryBuilder } from './data-cube/QueryBuilderDataCubeHelper.js';
 import type { QueryBuilderDataCubeViewerState } from './data-cube/QueryBuilderDataCubeViewerState.js';
 import { QueryBuilderTelemetryHelper } from '../__lib__/QueryBuilderTelemetryHelper.js';
-
-// eslint-disable-next-line @typescript-eslint/no-empty-object-type
-export interface QueryableSourceInfo {}
+import type {
+  DepotEntityWithOrigin,
+  QueryableSourceInfo,
+} from '@finos/legend-storage';
 
 export type QueryableClassMappingRuntimeInfo = QueryableSourceInfo & {
   class: string;
@@ -136,6 +137,53 @@ export enum QUERY_BUILDER_LAMBDA_WRITER_MODE {
   TYPED_FETCH_STRUCTURE = 'TYPED_FETCH_STRUCTURE',
 }
 
+export type EntityWithOriginOption = {
+  label: string;
+  value: DepotEntityWithOrigin;
+};
+
+export class ExtraOptionsConfig<T> {
+  label: string;
+  type: string;
+  options:
+    | {
+        label: string;
+        value: T;
+      }[]
+    | undefined;
+  selectedValue: T | undefined;
+  onChange: (val: T) => void;
+  placeholder?: string | undefined;
+  disabled?: boolean | undefined;
+
+  constructor(
+    label: string,
+    type: string,
+    options: { label: string; value: T }[] | undefined,
+    selectedValue: T | undefined,
+    onChange: (val: T) => void,
+    placeholder?: string | undefined,
+    disabled?: boolean | undefined,
+  ) {
+    this.label = label;
+    this.type = type;
+    this.options = options;
+    this.selectedValue = selectedValue;
+    this.onChange = onChange;
+    this.placeholder = placeholder;
+    this.disabled = disabled;
+
+    makeObservable(this, {
+      options: observable,
+      setOptions: action,
+    });
+  }
+
+  setOptions(options: { label: string; value: T }[] | undefined): void {
+    this.options = options;
+  }
+}
+
 export abstract class QueryBuilderState implements CommandRegistrar {
   readonly applicationStore: GenericLegendApplicationStore;
   readonly graphManagerState: GraphManagerState;
@@ -281,7 +329,6 @@ 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;
     this.isQueryChatOpened =
@@ -960,6 +1007,17 @@ export abstract class QueryBuilderState implements CommandRegistrar {
   }
 }
 
+/**
+ * Base query builder state with result panel hidden.
+ * This is useful for scenarios where the query builder is used
+ * only for query construction without execution/result display.
+ */
+export abstract class BaseQueryBuilderState extends QueryBuilderState {
+  override get isResultPanelHidden(): boolean {
+    return true;
+  }
+}
+
 /**
  * This type is used for testing and analytics operation in query builder.
  * For example, we use this to build the preview data lambda, or to build the auto-complete lambda