@finos/legend-application-query 5.0.0

package/lib/stores/QueryBuilderSetupState.js

@@ -0,0 +1,163 @@
+/**
+ * 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 { action, makeAutoObservable, observable } from 'mobx';
+import { getNullableFirstElement, uniq } from '@finos/legend-shared';
+import { getMilestoneTemporalStereotype, PackageableElementExplicitReference, RuntimePointer, DEFAULT_BUSINESS_DATE_MILESTONING_PARAMETER_NAME, DEFAULT_PROCESSING_DATE_MILESTONING_PARAMETER_NAME, MILESTONING_STEREOTYPE, observe_ValueSpecification, getAllIncludedMappings, } from '@finos/legend-graph';
+export class QueryBuilderSetupState {
+    queryBuilderState;
+    _class;
+    mapping;
+    runtimeValue;
+    classIsReadOnly = false;
+    mappingIsReadOnly = false;
+    runtimeIsReadOnly = false;
+    showSetupPanel = true;
+    // TODO: Change this when we modify how we deal with milestoning.
+    // See https://github.com/finos/legend-studio/issues/1149
+    businessDate;
+    processingDate;
+    constructor(queryBuilderState) {
+        makeAutoObservable(this, {
+            queryBuilderState: false,
+            processingDate: observable,
+            businessDate: observable,
+            setQueryBuilderState: action,
+            setClass: action,
+            setMapping: action,
+            setRuntimeValue: action,
+            setProcessingDate: action,
+            setBusinessDate: action,
+            setShowSetupPanel: action,
+            setClassIsReadOnly: action,
+            setMappingIsReadOnly: action,
+            setRuntimeIsReadOnly: action,
+        });
+        this.queryBuilderState = queryBuilderState;
+    }
+    initializeQueryMilestoningParameters(stereotype) {
+        switch (stereotype) {
+            case MILESTONING_STEREOTYPE.BUSINESS_TEMPORAL: {
+                this.setBusinessDate(this.queryBuilderState.buildMilestoningParameter(DEFAULT_BUSINESS_DATE_MILESTONING_PARAMETER_NAME));
+                break;
+            }
+            case MILESTONING_STEREOTYPE.PROCESSING_TEMPORAL: {
+                this.setProcessingDate(this.queryBuilderState.buildMilestoningParameter(DEFAULT_PROCESSING_DATE_MILESTONING_PARAMETER_NAME));
+                break;
+            }
+            case MILESTONING_STEREOTYPE.BITEMPORAL: {
+                this.setProcessingDate(this.queryBuilderState.buildMilestoningParameter(DEFAULT_PROCESSING_DATE_MILESTONING_PARAMETER_NAME));
+                this.setBusinessDate(this.queryBuilderState.buildMilestoningParameter(DEFAULT_BUSINESS_DATE_MILESTONING_PARAMETER_NAME));
+                break;
+            }
+            default:
+        }
+    }
+    get classes() {
+        return this.queryBuilderState.graphManagerState.graph.ownClasses
+            .concat(this.queryBuilderState.graphManagerState.filterSystemElementOptions(this.queryBuilderState.graphManagerState.graph.systemModel.ownClasses))
+            .concat(this.queryBuilderState.graphManagerState.graph.dependencyManager
+            .classes);
+    }
+    get mappings() {
+        return this.queryBuilderState.graphManagerState.graph.mappings;
+    }
+    get compatibleMappings() {
+        const mappingsWithClassMapped = this.mappings.filter((mapping) => mapping.classMappings.some((cm) => cm.class.value === this._class));
+        const resolvedMappingIncludes = this.mappings.filter((mapping) => getAllIncludedMappings(mapping).some((e) => mappingsWithClassMapped.includes(e)));
+        return this._class
+            ? uniq([...mappingsWithClassMapped, ...resolvedMappingIncludes])
+            : [];
+    }
+    get compatibleRuntimes() {
+        const mapping = this.mapping;
+        // If the runtime claims to cover some mappings which include the specified mapping,
+        // then we deem the runtime to be compatible with the such mapping
+        return mapping
+            ? this.queryBuilderState.graphManagerState.graph.runtimes.filter((runtime) => runtime.runtimeValue.mappings
+                .map((mappingReference) => [
+                mappingReference.value,
+                ...getAllIncludedMappings(mappingReference.value),
+            ])
+                .flat()
+                .includes(mapping))
+            : [];
+    }
+    setQueryBuilderState(queryBuilderState) {
+        this.queryBuilderState = queryBuilderState;
+    }
+    setRuntimeValue(val) {
+        this.runtimeValue = val;
+    }
+    setShowSetupPanel(val) {
+        this.showSetupPanel = val;
+    }
+    setClassIsReadOnly(val) {
+        this.classIsReadOnly = val;
+    }
+    setMappingIsReadOnly(val) {
+        this.mappingIsReadOnly = val;
+    }
+    setRuntimeIsReadOnly(val) {
+        this.runtimeIsReadOnly = val;
+    }
+    setClass(val, isRebuildingState) {
+        this._class = val;
+        const isMappingEditable = !isRebuildingState && !this.mappingIsReadOnly;
+        const isCurrentMappingCompatible = this.mapping && this.compatibleMappings.includes(this.mapping);
+        if (isMappingEditable && !isCurrentMappingCompatible) {
+            // try to select the first compatible mapping
+            const possibleMapping = getNullableFirstElement(this.compatibleMappings);
+            if (possibleMapping) {
+                this.setMapping(possibleMapping);
+            }
+        }
+        if (val !== undefined) {
+            const stereotype = getMilestoneTemporalStereotype(val, this.queryBuilderState.graphManagerState.graph);
+            this.setBusinessDate(undefined);
+            this.setProcessingDate(undefined);
+            if (stereotype) {
+                this.initializeQueryMilestoningParameters(stereotype);
+                // Show the parameter panel because we populate paramaters state with milestoning parameters
+                this.queryBuilderState.setShowParameterPanel(true);
+            }
+        }
+    }
+    setMapping(val) {
+        this.mapping = val;
+        if (this.runtimeIsReadOnly) {
+            return;
+        }
+        const runtimeToPick = getNullableFirstElement(this.compatibleRuntimes);
+        if (runtimeToPick) {
+            this.setRuntimeValue(new RuntimePointer(PackageableElementExplicitReference.create(runtimeToPick)));
+        }
+        else {
+            // TODO?: we should consider if we allow users to use custom runtime here
+            this.setRuntimeValue(undefined);
+        }
+    }
+    setProcessingDate(val) {
+        this.processingDate = val
+            ? observe_ValueSpecification(val, this.queryBuilderState.observableContext)
+            : val;
+    }
+    setBusinessDate(val) {
+        this.businessDate = val
+            ? observe_ValueSpecification(val, this.queryBuilderState.observableContext)
+            : val;
+    }
+}
+//# sourceMappingURL=QueryBuilderSetupState.js.map

package/lib/stores/QueryBuilderSetupState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderSetupState.js","sourceRoot":"","sources":["../../src/stores/QueryBuilderSetupState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAErE,OAAO,EAML,8BAA8B,EAC9B,mCAAmC,EACnC,cAAc,EACd,gDAAgD,EAChD,kDAAkD,EAClD,sBAAsB,EACtB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,OAAO,sBAAsB;IACjC,iBAAiB,CAAoB;IACrC,MAAM,CAAqB;IAC3B,OAAO,CAAuB;IAC9B,YAAY,CAAuB;IACnC,eAAe,GAAG,KAAK,CAAC;IACxB,iBAAiB,GAAG,KAAK,CAAC;IAC1B,iBAAiB,GAAG,KAAK,CAAC;IAC1B,cAAc,GAAG,IAAI,CAAC;IAEtB,iEAAiE;IACjE,yDAAyD;IACzD,YAAY,CAAkC;IAC9C,cAAc,CAAkC;IAEhD,YAAY,iBAAoC;QAC9C,kBAAkB,CAAC,IAAI,EAAE;YACvB,iBAAiB,EAAE,KAAK;YACxB,cAAc,EAAE,UAAU;YAC1B,YAAY,EAAE,UAAU;YACxB,oBAAoB,EAAE,MAAM;YAC5B,QAAQ,EAAE,MAAM;YAChB,UAAU,EAAE,MAAM;YAClB,eAAe,EAAE,MAAM;YACvB,iBAAiB,EAAE,MAAM;YACzB,eAAe,EAAE,MAAM;YACvB,iBAAiB,EAAE,MAAM;YACzB,kBAAkB,EAAE,MAAM;YAC1B,oBAAoB,EAAE,MAAM;YAC5B,oBAAoB,EAAE,MAAM;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAC7C,CAAC;IAEO,oCAAoC,CAAC,UAAkB;QAC7D,QAAQ,UAAU,EAAE;YAClB,KAAK,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;gBAC7C,IAAI,CAAC,eAAe,CAClB,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAC9C,gDAAgD,CACjD,CACF,CAAC;gBACF,MAAM;aACP;YACD,KAAK,sBAAsB,CAAC,mBAAmB,CAAC,CAAC;gBAC/C,IAAI,CAAC,iBAAiB,CACpB,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAC9C,kDAAkD,CACnD,CACF,CAAC;gBACF,MAAM;aACP;YACD,KAAK,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBACtC,IAAI,CAAC,iBAAiB,CACpB,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAC9C,kDAAkD,CACnD,CACF,CAAC;gBACF,IAAI,CAAC,eAAe,CAClB,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAC9C,gDAAgD,CACjD,CACF,CAAC;gBACF,MAAM;aACP;YACD,QAAQ;SACT;IACH,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU;aAC7D,MAAM,CACL,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,0BAA0B,CACjE,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,CACtE,CACF;aACA,MAAM,CACL,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,iBAAiB;aAC7D,OAAO,CACX,CAAC;IACN,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC;IACjE,CAAC;IAED,IAAY,kBAAkB;QAC5B,MAAM,uBAAuB,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC/D,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,KAAK,IAAI,CAAC,MAAM,CAAC,CACnE,CAAC;QACF,MAAM,uBAAuB,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC/D,sBAAsB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACzC,uBAAuB,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpC,CACF,CAAC;QACF,OAAO,IAAI,CAAC,MAAM;YAChB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,uBAAuB,EAAE,GAAG,uBAAuB,CAAC,CAAC;YAChE,CAAC,CAAC,EAAE,CAAC;IACT,CAAC;IAED,IAAI,kBAAkB;QACpB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,oFAAoF;QACpF,kEAAkE;QAClE,OAAO,OAAO;YACZ,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAC5D,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,YAAY,CAAC,QAAQ;iBAC1B,GAAG,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC;gBACzB,gBAAgB,CAAC,KAAK;gBACtB,GAAG,sBAAsB,CAAC,gBAAgB,CAAC,KAAK,CAAC;aAClD,CAAC;iBACD,IAAI,EAAE;iBACN,QAAQ,CAAC,OAAO,CAAC,CACvB;YACH,CAAC,CAAC,EAAE,CAAC;IACT,CAAC;IAED,oBAAoB,CAAC,iBAAoC;QACvD,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAC7C,CAAC;IACD,eAAe,CAAC,GAAwB;QACtC,IAAI,CAAC,YAAY,GAAG,GAAG,CAAC;IAC1B,CAAC;IACD,iBAAiB,CAAC,GAAY;QAC5B,IAAI,CAAC,cAAc,GAAG,GAAG,CAAC;IAC5B,CAAC;IACD,kBAAkB,CAAC,GAAY;QAC7B,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;IAC7B,CAAC;IACD,oBAAoB,CAAC,GAAY;QAC/B,IAAI,CAAC,iBAAiB,GAAG,GAAG,CAAC;IAC/B,CAAC;IACD,oBAAoB,CAAC,GAAY;QAC/B,IAAI,CAAC,iBAAiB,GAAG,GAAG,CAAC;IAC/B,CAAC;IAED,QAAQ,CAAC,GAAsB,EAAE,iBAA2B;QAC1D,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;QAClB,MAAM,iBAAiB,GAAG,CAAC,iBAAiB,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC;QACxE,MAAM,0BAA0B,GAC9B,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,IAAI,iBAAiB,IAAI,CAAC,0BAA0B,EAAE;YACpD,6CAA6C;YAC7C,MAAM,eAAe,GAAG,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACzE,IAAI,eAAe,EAAE;gBACnB,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;aAClC;SACF;QACD,IAAI,GAAG,KAAK,SAAS,EAAE;YACrB,MAAM,UAAU,GAAG,8BAA8B,CAC/C,GAAG,EACH,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,KAAK,CAC/C,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAChC,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,UAAU,EAAE;gBACd,IAAI,CAAC,oCAAoC,CAAC,UAAU,CAAC,CAAC;gBACtD,4FAA4F;gBAC5F,IAAI,CAAC,iBAAiB,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;aACpD;SACF;IACH,CAAC;IAED,UAAU,CAAC,GAAwB;QACjC,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC;QACnB,IAAI,IAAI,CAAC,iBAAiB,EAAE;YAC1B,OAAO;SACR;QACD,MAAM,aAAa,GAAG,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACvE,IAAI,aAAa,EAAE;YACjB,IAAI,CAAC,eAAe,CAClB,IAAI,cAAc,CAChB,mCAAmC,CAAC,MAAM,CAAC,aAAa,CAAC,CAC1D,CACF,CAAC;SACH;aAAM;YACL,yEAAyE;YACzE,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;SACjC;IACH,CAAC;IAED,iBAAiB,CAAC,GAAmC;QACnD,IAAI,CAAC,cAAc,GAAG,GAAG;YACvB,CAAC,CAAC,0BAA0B,CACxB,GAAG,EACH,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CACzC;YACH,CAAC,CAAC,GAAG,CAAC;IACV,CAAC;IAED,eAAe,CAAC,GAAmC;QACjD,IAAI,CAAC,YAAY,GAAG,GAAG;YACrB,CAAC,CAAC,0BAA0B,CACxB,GAAG,EACH,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CACzC;YACH,CAAC,CAAC,GAAG,CAAC;IACV,CAAC;CACF"}

package/lib/stores/QueryBuilderState.d.ts

@@ -0,0 +1,101 @@
+/**
+ * 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 GeneratorFn } from '@finos/legend-shared';
+import { type QueryBuilderFilterOperator, QueryBuilderFilterState } from './QueryBuilderFilterState.js';
+import { QueryBuilderFetchStructureState } from './QueryBuilderFetchStructureState.js';
+import { QueryResultSetModifierState } from './QueryResultSetModifierState.js';
+import { QueryTextEditorState } from './QueryTextEditorState.js';
+import { QueryBuilderSetupState } from './QueryBuilderSetupState.js';
+import { QueryBuilderExplorerState } from './QueryBuilderExplorerState.js';
+import { QueryBuilderResultState } from './QueryBuilderResultState.js';
+import { QueryBuilderUnsupportedState } from './QueryBuilderUnsupportedState.js';
+import { type Class, type GraphManagerState, type ValueSpecification, RawLambda, ObserverContext } from '@finos/legend-graph';
+import { type GenericLegendApplicationStore } from '@finos/legend-application';
+import { QueryBuilderPostFilterState } from './QueryBuilderPostFilterState.js';
+import type { QueryBuilderPostFilterOperator } from './QueryBuilderPostFilterOperator.js';
+import { QueryFunctionsExplorerState } from './QueryFunctionsExplorerState.js';
+import { QueryParametersState } from './QueryParametersState.js';
+export declare abstract class QueryBuilderMode {
+    abstract get isParametersDisabled(): boolean;
+    abstract get isResultPanelHidden(): boolean;
+}
+export declare class StandardQueryBuilderMode extends QueryBuilderMode {
+    get isParametersDisabled(): boolean;
+    get isResultPanelHidden(): boolean;
+}
+declare class QueryBuilderChangeDetectionState {
+    querybuildState: QueryBuilderState;
+    queryHashCode: string;
+    isEnabled: boolean;
+    constructor(queryBuilderState: QueryBuilderState);
+    setQueryHashCode(val: string): void;
+    setIsEnabled(val: boolean): void;
+}
+export declare class QueryBuilderState {
+    applicationStore: GenericLegendApplicationStore;
+    graphManagerState: GraphManagerState;
+    mode: QueryBuilderMode;
+    querySetupState: QueryBuilderSetupState;
+    explorerState: QueryBuilderExplorerState;
+    queryParametersState: QueryParametersState;
+    queryFunctionsExplorerState: QueryFunctionsExplorerState;
+    fetchStructureState: QueryBuilderFetchStructureState;
+    filterState: QueryBuilderFilterState;
+    postFilterState: QueryBuilderPostFilterState;
+    resultSetModifierState: QueryResultSetModifierState;
+    resultState: QueryBuilderResultState;
+    queryTextEditorState: QueryTextEditorState;
+    queryUnsupportedState: QueryBuilderUnsupportedState;
+    observableContext: ObserverContext;
+    filterOperators: QueryBuilderFilterOperator[];
+    postFilterOperators: QueryBuilderPostFilterOperator[];
+    isCompiling: boolean;
+    backdrop: boolean;
+    showFunctionPanel: boolean;
+    showParameterPanel: boolean;
+    showPostFilterPanel: boolean;
+    changeDetectionState: QueryBuilderChangeDetectionState;
+    constructor(applicationStore: GenericLegendApplicationStore, graphManagerState: GraphManagerState, queryBuilderMode: QueryBuilderMode);
+    setMode(val: QueryBuilderMode): void;
+    setBackdrop(val: boolean): void;
+    setShowFunctionPanel(val: boolean): void;
+    setShowParameterPanel(val: boolean): void;
+    setShowPostFilterPanel(val: boolean): void;
+    getQuery(options?: {
+        keepSourceInformation: boolean;
+    }): RawLambda;
+    resetQueryBuilder(): void;
+    resetQuerySetup(): void;
+    initialize(rawLambda: RawLambda, options?: {
+        notifyError: boolean;
+    }): void;
+    /**
+     * Process the raw lambda, and build the query builder state.
+     *
+     * @throws error if there is an issue building the compiled lambda or rebuilding the state.
+     * consumers of function should handle the errors.
+     */
+    buildStateFromRawLambda(rawLambda: RawLambda): void;
+    buildMilestoningParameter(parameterName: string): ValueSpecification;
+    saveQuery(onSaveQuery: (lambda: RawLambda) => Promise<void>): Promise<void>;
+    isQuerySupported(): boolean;
+    clearCompilationError(): void;
+    compileQuery(): GeneratorFn<void>;
+    changeClass(val: Class | undefined, isRebuildingState?: boolean): void;
+    changeFetchStructure(): void;
+}
+export {};
+//# sourceMappingURL=QueryBuilderState.d.ts.map

package/lib/stores/QueryBuilderState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryBuilderState.d.ts","sourceRoot":"","sources":["../../src/stores/QueryBuilderState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EACL,KAAK,WAAW,EAOjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,KAAK,0BAA0B,EAC/B,uBAAuB,EACxB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,+BAA+B,EAAE,MAAM,sCAAsC,CAAC;AACvF,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAC/E,OAAO,EAEL,oBAAoB,EACrB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAKvE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EACL,KAAK,KAAK,EACV,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EAQvB,SAAS,EAIT,eAAe,EAIhB,MAAM,qBAAqB,CAAC;AA8B7B,OAAO,EAEL,KAAK,6BAA6B,EACnC,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAqB/E,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,qCAAqC,CAAC;AAS1F,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAC/E,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,8BAAsB,gBAAgB;IACpC,QAAQ,KAAK,oBAAoB,IAAI,OAAO,CAAC;IAE7C,QAAQ,KAAK,mBAAmB,IAAI,OAAO,CAAC;CAC7C;AAED,qBAAa,wBAAyB,SAAQ,gBAAgB;IAC5D,IAAI,oBAAoB,IAAI,OAAO,CAElC;IAED,IAAI,mBAAmB,IAAI,OAAO,CAEjC;CACF;AAED,cAAM,gCAAgC;IACpC,eAAe,EAAE,iBAAiB,CAAC;IACnC,aAAa,SAAmD;IAChE,SAAS,UAAS;gBAEN,iBAAiB,EAAE,iBAAiB;IAIhD,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAInC,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;CAGjC;AAED,qBAAa,iBAAiB;IAC5B,gBAAgB,EAAE,6BAA6B,CAAC;IAChD,iBAAiB,EAAE,iBAAiB,CAAC;IAErC,IAAI,EAAE,gBAAgB,CAAC;IACvB,eAAe,EAAE,sBAAsB,CAAC;IACxC,aAAa,EAAE,yBAAyB,CAAC;IACzC,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,2BAA2B,EAAE,2BAA2B,CAAC;IACzD,mBAAmB,EAAE,+BAA+B,CAAC;IACrD,WAAW,EAAE,uBAAuB,CAAC;IACrC,eAAe,EAAE,2BAA2B,CAAC;IAC7C,sBAAsB,EAAE,2BAA2B,CAAC;IACpD,WAAW,EAAE,uBAAuB,CAAC;IACrC,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,qBAAqB,EAAE,4BAA4B,CAAC;IACpD,iBAAiB,EAAE,eAAe,CAAC;IACnC,eAAe,EAAE,0BAA0B,EAAE,CAiB3C;IACF,mBAAmB,EAAE,8BAA8B,EAAE,CAiBnD;IACF,WAAW,UAAS;IACpB,QAAQ,UAAS;IACjB,iBAAiB,UAAS;IAC1B,kBAAkB,UAAS;IAC3B,mBAAmB,UAAS;IAC5B,oBAAoB,EAAE,gCAAgC,CAAC;gBAGrD,gBAAgB,EAAE,6BAA6B,EAC/C,iBAAiB,EAAE,iBAAiB,EACpC,gBAAgB,EAAE,gBAAgB;IA2DpC,OAAO,CAAC,GAAG,EAAE,gBAAgB,GAAG,IAAI;IAIpC,WAAW,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;IAI/B,oBAAoB,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;IAIxC,qBAAqB,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;IAIzC,sBAAsB,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;IAI1C,QAAQ,CAAC,OAAO,CAAC,EAAE;QAAE,qBAAqB,EAAE,OAAO,CAAA;KAAE,GAAG,SAAS;IAoBjE,iBAAiB,IAAI,IAAI;IASzB,eAAe,IAAI,IAAI;IAmBvB,UAAU,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE;QAAE,WAAW,EAAE,OAAO,CAAA;KAAE,GAAG,IAAI;IA4B1E;;;;;OAKG;IACH,uBAAuB,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IAyBnD,yBAAyB,CAAC,aAAa,EAAE,MAAM,GAAG,kBAAkB;IA6B9D,SAAS,CACb,WAAW,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,GAChD,OAAO,CAAC,IAAI,CAAC;IAUhB,gBAAgB,IAAI,OAAO;IAI3B,qBAAqB,IAAI,IAAI;IAI5B,YAAY,IAAI,WAAW,CAAC,IAAI,CAAC;IAoFlC,WAAW,CAAC,GAAG,EAAE,KAAK,GAAG,SAAS,EAAE,iBAAiB,CAAC,EAAE,OAAO,GAAG,IAAI;IAOtE,oBAAoB,IAAI,IAAI;CAO7B"}

package/lib/stores/QueryBuilderState.js

@@ -0,0 +1,371 @@
+/**
+ * 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 { action, flow, observable, makeObservable } from 'mobx';
+import { LogEvent, assertErrorThrown, guaranteeNonNullable, guaranteeType, filterByType, hashObject, } from '@finos/legend-shared';
+import { QueryBuilderFilterState, } from './QueryBuilderFilterState.js';
+import { QueryBuilderFetchStructureState } from './QueryBuilderFetchStructureState.js';
+import { QueryResultSetModifierState } from './QueryResultSetModifierState.js';
+import { QueryTextEditorMode, QueryTextEditorState, } from './QueryTextEditorState.js';
+import { QueryBuilderSetupState } from './QueryBuilderSetupState.js';
+import { QueryBuilderExplorerState } from './QueryBuilderExplorerState.js';
+import { QueryBuilderResultState } from './QueryBuilderResultState.js';
+import { processQueryBuilderLambdaFunction, processQueryParameters, } from './QueryBuilderLambdaProcessor.js';
+import { QueryBuilderUnsupportedState } from './QueryBuilderUnsupportedState.js';
+import { GenericTypeExplicitReference, GenericType, PRIMITIVE_TYPE, GRAPH_MANAGER_EVENT, CompilationError, extractSourceInformationCoordinates, LambdaFunctionInstanceValue, RawLambda, TYPICAL_MULTIPLICITY_TYPE, VariableExpression, observe_ValueSpecification, ObserverContext, isStubbed_RawLambda, buildLambdaVariableExpressions, buildRawLambdaFromLambdaFunction, } from '@finos/legend-graph';
+import { QueryBuilderFilterOperator_Equal, QueryBuilderFilterOperator_NotEqual, } from './filterOperators/QueryBuilderFilterOperator_Equal.js';
+import { QueryBuilderFilterOperator_GreaterThan } from './filterOperators/QueryBuilderFilterOperator_GreaterThan.js';
+import { QueryBuilderFilterOperator_NotStartWith, QueryBuilderFilterOperator_StartWith, } from './filterOperators/QueryBuilderFilterOperator_StartWith.js';
+import { QueryBuilderFilterOperator_GreaterThanEqual } from './filterOperators/QueryBuilderFilterOperator_GreaterThanEqual.js';
+import { QueryBuilderFilterOperator_LessThanEqual } from './filterOperators/QueryBuilderFilterOperator_LessThanEqual.js';
+import { QueryBuilderFilterOperator_LessThan } from './filterOperators/QueryBuilderFilterOperator_LessThan.js';
+import { QueryBuilderFilterOperator_EndWith, QueryBuilderFilterOperator_NotEndWith, } from './filterOperators/QueryBuilderFilterOperator_EndWith.js';
+import { QueryBuilderFilterOperator_Contain, QueryBuilderFilterOperator_NotContain, } from './filterOperators/QueryBuilderFilterOperator_Contain.js';
+import { QueryBuilderFilterOperator_IsEmpty, QueryBuilderFilterOperator_IsNotEmpty, } from './filterOperators/QueryBuilderFilterOperator_IsEmpty.js';
+import { QueryBuilderFilterOperator_In, QueryBuilderFilterOperator_NotIn, } from './filterOperators/QueryBuilderFilterOperator_In.js';
+import { buildLambdaFunction } from './QueryBuilderLambdaBuilder.js';
+import { LambdaParameterState, } from '@finos/legend-application';
+import { QueryBuilderPostFilterState } from './QueryBuilderPostFilterState.js';
+import { QueryBuilderPostFilterOperator_Equal, QueryBuilderPostFilterOperator_NotEqual, } from './postFilterOperators/QueryBuilderPostFilterOperator_Equal.js';
+import { QueryBuilderPostFilterOperator_LessThan } from './postFilterOperators/QueryBuilderPostFilterOperator_LessThan.js';
+import { QueryBuilderPostFilterOperator_LessThanEqual } from './postFilterOperators/QueryBuilderPostFilterOperator_LessThanEqual.js';
+import { QueryBuilderPostFilterOperator_GreaterThan } from './postFilterOperators/QueryBuilderPostFilterOperator_GreaterThan.js';
+import { QueryBuilderPostFilterOperator_GreaterThanEqual } from './postFilterOperators/QueryBuilderPostFilterOperator_GreaterThanEqual.js';
+import { QueryBuilderPostFilterOperator_NotStartWith, QueryBuilderPostFilterOperator_StartWith, } from './postFilterOperators/QueryBuilderPostFilterOperator_StartWith.js';
+import { QueryBuilderPostFilterOperator_Contain, QueryBuilderPostFilterOperator_NotContain, } from './postFilterOperators/QueryBuilderPostFilterOperator_Contain.js';
+import { QueryBuilderPostFilterOperator_EndWith, QueryBuilderPostFilterOperator_NotEndWith, } from './postFilterOperators/QueryBuilderPostFilterOperator_EndWith.js';
+import { QueryBuilderPostFilterOperator_In, QueryBuilderPostFilterOperator_NotIn, } from './postFilterOperators/QueryBuilderPostFilterOperator_In.js';
+import { QueryBuilderPostFilterOperator_IsEmpty, QueryBuilderPostFilterOperator_IsNotEmpty, } from './postFilterOperators/QueryBuilderPostFilterOperator_IsEmpty.js';
+import { QueryFunctionsExplorerState } from './QueryFunctionsExplorerState.js';
+import { QueryParametersState } from './QueryParametersState.js';
+export class QueryBuilderMode {
+}
+export class StandardQueryBuilderMode extends QueryBuilderMode {
+    get isParametersDisabled() {
+        return false;
+    }
+    get isResultPanelHidden() {
+        return false;
+    }
+}
+class QueryBuilderChangeDetectionState {
+    querybuildState;
+    queryHashCode = hashObject(new RawLambda(undefined, undefined));
+    isEnabled = false;
+    constructor(queryBuilderState) {
+        this.querybuildState = queryBuilderState;
+    }
+    setQueryHashCode(val) {
+        this.queryHashCode = val;
+    }
+    setIsEnabled(val) {
+        this.isEnabled = val;
+    }
+}
+export class QueryBuilderState {
+    applicationStore;
+    graphManagerState;
+    mode;
+    querySetupState;
+    explorerState;
+    queryParametersState;
+    queryFunctionsExplorerState;
+    fetchStructureState;
+    filterState;
+    postFilterState;
+    resultSetModifierState;
+    resultState;
+    queryTextEditorState;
+    queryUnsupportedState;
+    observableContext;
+    filterOperators = [
+        new QueryBuilderFilterOperator_Equal(),
+        new QueryBuilderFilterOperator_NotEqual(),
+        new QueryBuilderFilterOperator_LessThan(),
+        new QueryBuilderFilterOperator_LessThanEqual(),
+        new QueryBuilderFilterOperator_GreaterThan(),
+        new QueryBuilderFilterOperator_GreaterThanEqual(),
+        new QueryBuilderFilterOperator_StartWith(),
+        new QueryBuilderFilterOperator_NotStartWith(),
+        new QueryBuilderFilterOperator_Contain(),
+        new QueryBuilderFilterOperator_NotContain(),
+        new QueryBuilderFilterOperator_EndWith(),
+        new QueryBuilderFilterOperator_NotEndWith(),
+        new QueryBuilderFilterOperator_In(),
+        new QueryBuilderFilterOperator_NotIn(),
+        new QueryBuilderFilterOperator_IsEmpty(),
+        new QueryBuilderFilterOperator_IsNotEmpty(),
+    ];
+    postFilterOperators = [
+        new QueryBuilderPostFilterOperator_Equal(),
+        new QueryBuilderPostFilterOperator_NotEqual(),
+        new QueryBuilderPostFilterOperator_LessThan(),
+        new QueryBuilderPostFilterOperator_LessThanEqual(),
+        new QueryBuilderPostFilterOperator_GreaterThan(),
+        new QueryBuilderPostFilterOperator_GreaterThanEqual(),
+        new QueryBuilderPostFilterOperator_StartWith(),
+        new QueryBuilderPostFilterOperator_NotStartWith(),
+        new QueryBuilderPostFilterOperator_Contain(),
+        new QueryBuilderPostFilterOperator_NotContain(),
+        new QueryBuilderPostFilterOperator_EndWith(),
+        new QueryBuilderPostFilterOperator_NotEndWith(),
+        new QueryBuilderPostFilterOperator_In(),
+        new QueryBuilderPostFilterOperator_NotIn(),
+        new QueryBuilderPostFilterOperator_IsEmpty(),
+        new QueryBuilderPostFilterOperator_IsNotEmpty(),
+    ];
+    isCompiling = false;
+    backdrop = false;
+    showFunctionPanel = false;
+    showParameterPanel = false;
+    showPostFilterPanel = false;
+    changeDetectionState;
+    constructor(applicationStore, graphManagerState, queryBuilderMode) {
+        makeObservable(this, {
+            querySetupState: observable,
+            explorerState: observable,
+            queryParametersState: observable,
+            queryFunctionsExplorerState: observable,
+            fetchStructureState: observable,
+            filterState: observable,
+            postFilterState: observable,
+            resultSetModifierState: observable,
+            resultState: observable,
+            queryTextEditorState: observable,
+            queryUnsupportedState: observable,
+            isCompiling: observable,
+            backdrop: observable,
+            mode: observable,
+            showFunctionPanel: observable,
+            showParameterPanel: observable,
+            showPostFilterPanel: observable,
+            changeDetectionState: observable,
+            setMode: action,
+            resetQueryBuilder: action,
+            resetQuerySetup: action,
+            buildStateFromRawLambda: action,
+            saveQuery: action,
+            setBackdrop: action,
+            setShowFunctionPanel: action,
+            setShowParameterPanel: action,
+            setShowPostFilterPanel: action,
+            changeClass: action,
+            changeFetchStructure: action,
+            compileQuery: flow,
+        });
+        this.applicationStore = applicationStore;
+        this.graphManagerState = graphManagerState;
+        this.querySetupState = new QueryBuilderSetupState(this);
+        this.explorerState = new QueryBuilderExplorerState(this);
+        this.queryParametersState = new QueryParametersState(this);
+        this.queryFunctionsExplorerState = new QueryFunctionsExplorerState(this);
+        this.fetchStructureState = new QueryBuilderFetchStructureState(this);
+        this.filterState = new QueryBuilderFilterState(this, this.filterOperators);
+        this.postFilterState = new QueryBuilderPostFilterState(this, this.postFilterOperators);
+        this.resultSetModifierState = new QueryResultSetModifierState(this);
+        this.resultState = new QueryBuilderResultState(this);
+        this.queryTextEditorState = new QueryTextEditorState(this);
+        this.queryUnsupportedState = new QueryBuilderUnsupportedState(this);
+        this.mode = queryBuilderMode;
+        this.observableContext = new ObserverContext(this.graphManagerState.pluginManager.getPureGraphManagerPlugins());
+        this.changeDetectionState = new QueryBuilderChangeDetectionState(this);
+    }
+    setMode(val) {
+        this.mode = val;
+    }
+    setBackdrop(val) {
+        this.backdrop = val;
+    }
+    setShowFunctionPanel(val) {
+        this.showFunctionPanel = val;
+    }
+    setShowParameterPanel(val) {
+        this.showParameterPanel = val;
+    }
+    setShowPostFilterPanel(val) {
+        this.showPostFilterPanel = val;
+    }
+    getQuery(options) {
+        if (!this.isQuerySupported()) {
+            const parameters = this.queryParametersState.parameterStates.map((e) => this.graphManagerState.graphManager.serializeValueSpecification(e.parameter));
+            this.queryUnsupportedState.setRawLambda(new RawLambda(parameters, this.queryUnsupportedState.rawLambda?.body));
+            return guaranteeNonNullable(this.queryUnsupportedState.rawLambda);
+        }
+        return buildRawLambdaFromLambdaFunction(buildLambdaFunction(this, {
+            keepSourceInformation: Boolean(options?.keepSourceInformation),
+        }), this.graphManagerState);
+    }
+    resetQueryBuilder() {
+        const resultState = new QueryBuilderResultState(this);
+        resultState.setPreviewLimit(this.resultState.previewLimit);
+        this.resultState = resultState;
+        this.queryTextEditorState = new QueryTextEditorState(this);
+        this.queryUnsupportedState = new QueryBuilderUnsupportedState(this);
+        this.setShowParameterPanel(false);
+    }
+    resetQuerySetup() {
+        this.explorerState = new QueryBuilderExplorerState(this);
+        this.explorerState.refreshTreeData();
+        this.queryParametersState = new QueryParametersState(this);
+        this.queryFunctionsExplorerState = new QueryFunctionsExplorerState(this);
+        const fetchStructureState = new QueryBuilderFetchStructureState(this);
+        fetchStructureState.setFetchStructureMode(this.fetchStructureState.fetchStructureMode);
+        this.fetchStructureState = fetchStructureState;
+        this.filterState = new QueryBuilderFilterState(this, this.filterOperators);
+        this.postFilterState = new QueryBuilderPostFilterState(this, this.postFilterOperators);
+        this.resultSetModifierState = new QueryResultSetModifierState(this);
+        this.fetchStructureState.graphFetchTreeState.initialize();
+    }
+    initialize(rawLambda, options) {
+        try {
+            this.buildStateFromRawLambda(rawLambda);
+            if (this.queryParametersState.parameterStates.length > 0) {
+                this.setShowParameterPanel(true);
+            }
+        }
+        catch (error) {
+            assertErrorThrown(error);
+            this.changeClass(undefined, true);
+            const parameters = buildLambdaVariableExpressions(rawLambda, this.graphManagerState)
+                .map((param) => observe_ValueSpecification(param, this.observableContext))
+                .filter(filterByType(VariableExpression));
+            processQueryParameters(parameters, this);
+            if (options?.notifyError) {
+                this.applicationStore.notifyError(`Can't initialize query builder: ${error.message}`);
+            }
+            this.queryUnsupportedState.setLambdaError(error);
+            this.queryUnsupportedState.setRawLambda(rawLambda);
+        }
+    }
+    /**
+     * Process the raw lambda, and build the query builder state.
+     *
+     * @throws error if there is an issue building the compiled lambda or rebuilding the state.
+     * consumers of function should handle the errors.
+     */
+    buildStateFromRawLambda(rawLambda) {
+        this.resetQueryBuilder();
+        this.resetQuerySetup();
+        if (!isStubbed_RawLambda(rawLambda)) {
+            const valueSpec = observe_ValueSpecification(this.graphManagerState.graphManager.buildValueSpecification(this.graphManagerState.graphManager.serializeRawValueSpecification(rawLambda), this.graphManagerState.graph), this.observableContext);
+            const compiledValueSpecification = guaranteeType(valueSpec, LambdaFunctionInstanceValue, `Can't build query state: query builder only support lambda`);
+            const compiledLambda = guaranteeNonNullable(compiledValueSpecification.values[0]);
+            processQueryBuilderLambdaFunction(this, compiledLambda);
+        }
+    }
+    buildMilestoningParameter(parameterName) {
+        const milestoningParameter = new VariableExpression(parameterName, this.graphManagerState.graph.getTypicalMultiplicity(TYPICAL_MULTIPLICITY_TYPE.ONE), GenericTypeExplicitReference.create(new GenericType(this.queryParametersState.queryBuilderState.graphManagerState.graph.getPrimitiveType(PRIMITIVE_TYPE.DATE))));
+        if (!this.queryParametersState.parameterStates.find((p) => p.variableName === parameterName)) {
+            const variableState = new LambdaParameterState(milestoningParameter, this.querySetupState.queryBuilderState.observableContext);
+            variableState.mockParameterValue();
+            this.queryParametersState.addParameter(variableState);
+        }
+        return milestoningParameter;
+    }
+    async saveQuery(onSaveQuery) {
+        try {
+            const rawLambda = this.getQuery();
+            await onSaveQuery(rawLambda);
+        }
+        catch (error) {
+            assertErrorThrown(error);
+            this.applicationStore.notifyError(`Can't save query: ${error.message}`);
+        }
+    }
+    isQuerySupported() {
+        return !this.queryUnsupportedState.rawLambda;
+    }
+    clearCompilationError() {
+        this.fetchStructureState.projectionState.clearCompilationError();
+    }
+    *compileQuery() {
+        if (!this.queryTextEditorState.mode) {
+            this.isCompiling = true;
+            this.clearCompilationError();
+            // form mode
+            try {
+                this.queryTextEditorState.setCompilationError(undefined);
+                // NOTE: retain the source information on the lambda in order to be able
+                // to pin-point compilation issue in form mode
+                (yield this.graphManagerState.graphManager.getLambdaReturnType(this.getQuery({ keepSourceInformation: true }), this.graphManagerState.graph, { keepSourceInformation: true }));
+                this.applicationStore.notifySuccess('Compiled successfully');
+            }
+            catch (error) {
+                assertErrorThrown(error);
+                this.applicationStore.log.error(LogEvent.create(GRAPH_MANAGER_EVENT.COMPILATION_FAILURE), error);
+                let fallbackToTextModeForDebugging = true;
+                // if compilation failed, we try to reveal the error in form mode,
+                // if even this fail, we will fall back to show it in text mode
+                if (error instanceof CompilationError && error.sourceInformation) {
+                    fallbackToTextModeForDebugging =
+                        !this.fetchStructureState.projectionState.revealCompilationError(error);
+                }
+                // decide if we need to fall back to text mode for debugging
+                if (fallbackToTextModeForDebugging) {
+                    this.applicationStore.notifyWarning('Compilation failed and error cannot be located in form mode. Redirected to text mode for debugging.');
+                    this.queryTextEditorState.openModal(QueryTextEditorMode.TEXT);
+                    // TODO: trigger another compilation to pin-point the issue
+                    // since we're using the lambda editor right now, we are a little bit limitted
+                    // in terms of the timing to do compilation (since we're using an `useEffect` to
+                    // convert the lambda to grammar text), we might as well wait for the refactor
+                    // of query builder text-mode
+                    // See https://github.com/finos/legend-studio/issues/319
+                }
+                else {
+                    this.applicationStore.notifyWarning(`Compilation failed: ${error.message}`);
+                }
+            }
+            finally {
+                this.isCompiling = false;
+            }
+        }
+        else if (this.queryTextEditorState.mode === QueryTextEditorMode.TEXT) {
+            this.isCompiling = true;
+            try {
+                this.queryTextEditorState.setCompilationError(undefined);
+                (yield this.graphManagerState.graphManager.getLambdaReturnType(this.queryTextEditorState.rawLambdaState.lambda, this.graphManagerState.graph, { keepSourceInformation: true }));
+                this.applicationStore.notifySuccess('Compiled successfully');
+            }
+            catch (error) {
+                assertErrorThrown(error);
+                if (error instanceof CompilationError) {
+                    this.applicationStore.log.error(LogEvent.create(GRAPH_MANAGER_EVENT.COMPILATION_FAILURE), error);
+                    this.applicationStore.notifyWarning(`Compilation failed: ${error.message}`);
+                    const errorElementCoordinates = extractSourceInformationCoordinates(error.sourceInformation);
+                    if (errorElementCoordinates) {
+                        this.queryTextEditorState.setCompilationError(error);
+                    }
+                }
+            }
+            finally {
+                this.isCompiling = false;
+            }
+        }
+    }
+    changeClass(val, isRebuildingState) {
+        this.resetQueryBuilder();
+        this.resetQuerySetup();
+        this.querySetupState.setClass(val, isRebuildingState);
+        this.explorerState.refreshTreeData();
+    }
+    changeFetchStructure() {
+        this.resultSetModifierState = new QueryResultSetModifierState(this);
+        const treeData = this.fetchStructureState.graphFetchTreeState.treeData;
+        if (!treeData) {
+            this.fetchStructureState.graphFetchTreeState.initialize();
+        }
+    }
+}
+//# sourceMappingURL=QueryBuilderState.js.map