@finos/legend-application-studio 28.21.23 → 28.21.24

package/lib/stores/editor/editor-state/element-editor-state/ingest/IngestTestableState.js

@@ -0,0 +1,420 @@
+/**
+ * Copyright (c) 2026-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 { BaseDataResolver, DataProduct, EqualToRelation, IngestMatViewTest, IngestTestSuite, PackageableElementExplicitReference, RelationElement, RelationElementsData, RelationRowTestData, ReferenceDataResolver, observe_RelationElement, observe_RelationElementsData, observe_RelationRowTestData, } from '@finos/legend-graph';
+import { addUniqueEntry, deleteEntry, noop, uuid, } from '@finos/legend-shared';
+import { action, flow, flowResult, makeObservable, observable, runInAction, } from 'mobx';
+import { testSuite_addTest } from '../../../../graph-modifier/Testable_GraphModifierHelper.js';
+import { TestableTestEditorState, TestableTestSuiteEditorState, } from '../testable/TestableEditorState.js';
+import { RelationElementState } from '../data/EmbeddedDataState.js';
+import { buildRelationElementsDataWithColumns, createEmptyRelationElement, isIngestOrDataProductAccessor, LakehouseElementTestDataState, } from '../testable/LakehouseTestableUtils.js';
+// ─── Ingest-specific helpers ──────────────────────────────────────────────────
+const createExpectedRelationElement = (datasetName, columns) => {
+    const expected = new RelationElement();
+    const row = observe_RelationRowTestData(new RelationRowTestData());
+    row.values = columns.map(() => '');
+    expected.paths = [datasetName];
+    expected.columns = columns;
+    expected.rows = [row];
+    return observe_RelationElement(expected);
+};
+const inferIngestDatasetColumns = async (editorStore, ingest, datasetName) => {
+    const dataset = ingest.TEMPORARY_MATVIEW_FUNCTION_DATA_SETS?.find((candidate) => candidate.name === datasetName);
+    if (!dataset) {
+        return undefined;
+    }
+    const relationType = await editorStore.graphManagerState.graphManager.getLambdaRelationType(dataset.source.function, editorStore.graphManagerState.graph);
+    return relationType.columns.map((column) => column.name);
+};
+// ─── Per-test state ───────────────────────────────────────────────────────────
+export class IngestTestState extends TestableTestEditorState {
+    suiteState;
+    test;
+    uuid = uuid();
+    testDataRelationState;
+    constructor(suiteState, test) {
+        super(suiteState.testableState.ingest, test, suiteState.testableState.ingestDefinitionEditorState.isReadOnly, suiteState.editorStore);
+        makeObservable(this, {
+            selectedAsertionState: observable,
+            selectedTab: observable,
+            assertionToRename: observable,
+            assertionEditorStates: observable,
+            testResultState: observable,
+            runningTestAction: observable,
+            testDataRelationState: observable,
+            setSelectedTab: action,
+            setAssertionToRename: action,
+            addAssertion: action,
+            deleteAssertion: action,
+            openAssertion: action,
+            resetResult: action,
+            handleTestResult: action,
+            runTest: flow,
+            setDatasetId: action,
+        });
+        this.suiteState = suiteState;
+        this.test = test;
+        this.backfillDatasetIdIfMissing();
+        this.buildTestDataRelationState().catch(noop());
+    }
+    setDatasetId(val) {
+        this.test.datasetId = val;
+        this.resetResult();
+    }
+    get datasetLabel() {
+        if (this.test.datasetId) {
+            return this.test.datasetId;
+        }
+        const assertion = this.test.assertions.find((candidate) => candidate instanceof EqualToRelation);
+        const inferred = assertion?.expected.paths[0];
+        if (inferred) {
+            return inferred;
+        }
+        const availableDatasets = this.suiteState.testableState.ingestDefinitionEditorState.getMatviewFuncNames();
+        return availableDatasets[0] ?? 'N/A';
+    }
+    backfillDatasetIdIfMissing() {
+        if (this.test.datasetId) {
+            return;
+        }
+        const assertion = this.test.assertions.find((candidate) => candidate instanceof EqualToRelation);
+        const fallbackDatasetId = assertion?.expected.paths[0];
+        if (fallbackDatasetId) {
+            this.test.datasetId = fallbackDatasetId;
+            return;
+        }
+        const availableDatasets = this.suiteState.testableState.ingestDefinitionEditorState.getMatviewFuncNames();
+        if (availableDatasets[0]) {
+            this.test.datasetId = availableDatasets[0];
+        }
+    }
+    async buildTestDataRelationState() {
+        const assertion = this.test.assertions.find((candidate) => candidate instanceof EqualToRelation);
+        if (!assertion) {
+            return;
+        }
+        if (assertion.expected.columns.length === 0) {
+            try {
+                const engineColumns = await inferIngestDatasetColumns(this.editorStore, this.suiteState.testableState.ingest, this.datasetLabel);
+                if (engineColumns?.length) {
+                    runInAction(() => {
+                        assertion.expected.columns = engineColumns;
+                    });
+                }
+            }
+            catch {
+                // best-effort only
+            }
+        }
+        runInAction(() => {
+            this.testDataRelationState = new RelationElementState(assertion.expected);
+        });
+    }
+}
+// ─── Per-element test data state ─────────────────────────────────────────────
+// Replaced by the shared LakehouseElementTestDataState from LakehouseTestableUtils.
+// The type alias above maintains backward compatibility for any existing imports.
+// ─── Test data state for a suite ─────────────────────────────────────────────
+export class IngestTestDataState {
+    editorStore;
+    suiteState;
+    elementTestDataStates = [];
+    selectedElementTestDataState;
+    showAddElementModal = false;
+    constructor(suiteState, options) {
+        makeObservable(this, {
+            elementTestDataStates: observable,
+            selectedElementTestDataState: observable,
+            showAddElementModal: observable,
+            setSelectedElementTestDataState: action,
+            setShowAddElementModal: action,
+            addElement: action,
+            deleteElement: action,
+            refreshElementTestDataStates: action,
+        });
+        this.editorStore = suiteState.editorStore;
+        this.suiteState = suiteState;
+        this.refreshElementTestDataStates(options);
+    }
+    get availableElementsToAdd() {
+        const suite = this.suiteState.suite;
+        const existingPaths = new Set(suite.testData
+            .filter((testData) => testData instanceof BaseDataResolver ||
+            testData instanceof ReferenceDataResolver)
+            .map((testData) => testData.element.value.path));
+        const graph = this.editorStore.graphManagerState.graph;
+        const currentIngestPath = this.suiteState.testableState.ingest.path;
+        const candidates = [
+            ...graph.ingests.filter((ingest) => ingest.path !== currentIngestPath),
+            ...graph.allElements.filter((e) => e instanceof DataProduct),
+        ];
+        return candidates.filter((element) => !existingPaths.has(element.path));
+    }
+    setShowAddElementModal(val) {
+        this.showAddElementModal = val;
+    }
+    addElement(path) {
+        const element = this.editorStore.graphManagerState.graph.getNullableElement(path);
+        if (!element) {
+            return;
+        }
+        const resolver = new BaseDataResolver();
+        resolver.element = PackageableElementExplicitReference.create(element);
+        const relationElementsData = new RelationElementsData();
+        relationElementsData.relationElements = [];
+        observe_RelationElementsData(relationElementsData);
+        resolver.data = relationElementsData;
+        const suite = this.suiteState.suite;
+        suite.testData = [...suite.testData, resolver];
+        this.refreshElementTestDataStates({ selectedElementPath: path });
+    }
+    deleteElement(elementState) {
+        const suite = this.suiteState.suite;
+        const index = suite.testData.indexOf(elementState.testData);
+        suite.testData.splice(index, 1);
+        this.refreshElementTestDataStates();
+    }
+    refreshElementTestDataStates(options) {
+        const previouslySelectedElementPath = options?.selectedElementPath ??
+            this.selectedElementTestDataState?.element.path;
+        const suite = this.suiteState.suite;
+        this.elementTestDataStates = suite.testData
+            .filter((testData) => testData instanceof BaseDataResolver)
+            .map((testData) => new LakehouseElementTestDataState(testData, this.editorStore));
+        this.selectedElementTestDataState =
+            this.elementTestDataStates.find((state) => state.element.path === previouslySelectedElementPath) ?? this.elementTestDataStates[0];
+    }
+    setSelectedElementTestDataState(val) {
+        this.selectedElementTestDataState = val;
+    }
+}
+// ─── Per-suite state ──────────────────────────────────────────────────────────
+export class IngestTestSuiteState extends TestableTestSuiteEditorState {
+    testableState;
+    suite;
+    testStates = [];
+    testDataState;
+    uuid = uuid();
+    constructor(editorStore, testableState, suite) {
+        super(testableState.ingest, suite, testableState.ingestDefinitionEditorState.isReadOnly, editorStore);
+        makeObservable(this, {
+            testStates: observable,
+            runningSuiteState: observable,
+            selectTestState: observable,
+            testDataState: observable,
+            changeTest: action,
+            deleteTest: action,
+            removeTestState: action,
+            addNewTest: flow,
+            runSuite: flow,
+            runFailingTests: flow,
+            buildTestStates: action,
+        });
+        this.testableState = testableState;
+        this.suite = suite;
+        this.testDataState = new IngestTestDataState(this);
+        this.buildTestStates();
+    }
+    refreshTestDataState() {
+        this.testDataState = new IngestTestDataState(this, {
+            selectedElementPath: this.testDataState.selectedElementTestDataState?.element.path,
+        });
+    }
+    buildTestStates() {
+        this.suite.tests.forEach((test) => {
+            test.__parent = this.suite;
+            if (test instanceof IngestMatViewTest && !test.datasetId) {
+                const relationAssertion = test.assertions.find((candidate) => candidate instanceof EqualToRelation);
+                test.datasetId =
+                    relationAssertion?.expected.paths[0] ??
+                        this.testableState.datasetNames[0] ??
+                        '';
+            }
+        });
+        this.testStates = this.suite.tests.map((test) => new IngestTestState(this, test));
+        this.selectTestState = this.testStates[0];
+    }
+    deleteTest(test) {
+        super.deleteTest(test);
+        if (this.suite.tests.length === 0) {
+            // Ingest requires at least one test per suite; remove emptied suites.
+            this.testableState.deleteSuite(this);
+        }
+    }
+    *addNewTest(testName, datasetName) {
+        if (this.suite.tests.find((test) => test.id === testName)) {
+            return 'Duplicated test name';
+        }
+        let inferredColumns = [];
+        try {
+            const columns = (yield inferIngestDatasetColumns(this.editorStore, this.testableState.ingest, datasetName));
+            if (columns) {
+                inferredColumns = columns;
+            }
+        }
+        catch {
+            // Best effort only; continue with an empty expected relation.
+        }
+        const dataset = this.testableState.ingest.TEMPORARY_MATVIEW_FUNCTION_DATA_SETS?.find((candidate) => candidate.name === datasetName);
+        const rawLambda = dataset?.source.function;
+        if (rawLambda) {
+            const all = (yield this.editorStore.graphManagerState.graphManager.collectAccessorsInRawLambda(rawLambda, this.editorStore.graphManagerState.graph));
+            const externalAccessors = all.filter((accessor) => isIngestOrDataProductAccessor(accessor));
+            const byElement = new Map();
+            for (const accessor of externalAccessors) {
+                const group = byElement.get(accessor.parentElement.path) ?? [];
+                group.push(accessor);
+                byElement.set(accessor.parentElement.path, group);
+            }
+            for (const [elementPath, accessors] of byElement) {
+                const element = this.editorStore.graphManagerState.graph.getNullableElement(elementPath);
+                if (!element) {
+                    continue;
+                }
+                const existingResolver = this.suite.testData.find((testData) => testData instanceof BaseDataResolver &&
+                    testData.element.value === element);
+                const relationElementsData = existingResolver?.data instanceof RelationElementsData
+                    ? existingResolver.data
+                    : undefined;
+                if (!existingResolver || !relationElementsData) {
+                    const resolver = new BaseDataResolver();
+                    resolver.element =
+                        PackageableElementExplicitReference.create(element);
+                    const relationData = buildRelationElementsDataWithColumns(accessors);
+                    observe_RelationElementsData(relationData);
+                    resolver.data = relationData;
+                    this.suite.testData = [...this.suite.testData, resolver];
+                }
+                else {
+                    for (const accessor of accessors) {
+                        const itemId = accessor.accessor;
+                        if (!relationElementsData.relationElements.find((relationElement) => relationElement.paths[0] === itemId)) {
+                            const columns = accessor.relationType.columns.map((column) => column.name);
+                            relationElementsData.relationElements.push(createEmptyRelationElement(itemId, columns));
+                        }
+                    }
+                }
+            }
+        }
+        else {
+            this.editorStore.applicationStore.notificationService.notifyWarning('Dataset accessors cannot be resolved; using default ingest test data');
+        }
+        if (this.suite.testData.length === 0) {
+            // Ingest test suites require testData >= 1; seed a minimal editable resolver.
+            const resolver = new BaseDataResolver();
+            resolver.element = PackageableElementExplicitReference.create(this.testableState.ingest);
+            const relationData = new RelationElementsData();
+            relationData.relationElements = [
+                createEmptyRelationElement(datasetName, inferredColumns),
+            ];
+            observe_RelationElementsData(relationData);
+            resolver.data = relationData;
+            this.suite.testData = [resolver];
+        }
+        const test = new IngestMatViewTest();
+        test.id = testName;
+        test.datasetId = datasetName;
+        test.__parent = this.suite;
+        const assertion = new EqualToRelation();
+        assertion.id = 'assert_1';
+        assertion.expected = createExpectedRelationElement(datasetName, inferredColumns);
+        test.assertions = [assertion];
+        testSuite_addTest(this.suite, test, this.editorStore.changeDetectionState.observerContext);
+        this.refreshTestDataState();
+        const testState = new IngestTestState(this, test);
+        this.testStates.push(testState);
+        this.selectTestState = testState;
+        return undefined;
+    }
+}
+// ─── Top-level testable state ─────────────────────────────────────────────────
+export class IngestTestableState {
+    editorStore;
+    ingestDefinitionEditorState;
+    suiteStates = [];
+    selectedSuiteState;
+    showCreateSuiteModal = false;
+    showCreateTestModal = false;
+    suiteToRename;
+    constructor(ingestDefinitionEditorState) {
+        makeObservable(this, {
+            suiteStates: observable,
+            selectedSuiteState: observable,
+            showCreateSuiteModal: observable,
+            showCreateTestModal: observable,
+            suiteToRename: observable,
+            setSelectedSuiteState: action,
+            setShowCreateSuiteModal: action,
+            setShowCreateTestModal: action,
+            setSuiteToRename: action,
+            changeSuite: action,
+            init: action,
+            createSuite: flow,
+            deleteSuite: action,
+        });
+        this.editorStore = ingestDefinitionEditorState.editorStore;
+        this.ingestDefinitionEditorState = ingestDefinitionEditorState;
+    }
+    get ingest() {
+        return this.ingestDefinitionEditorState.ingest;
+    }
+    get datasetNames() {
+        return this.ingestDefinitionEditorState.getMatviewFuncNames();
+    }
+    setSelectedSuiteState(val) {
+        this.selectedSuiteState = val;
+    }
+    setShowCreateSuiteModal(val) {
+        this.showCreateSuiteModal = val;
+    }
+    setShowCreateTestModal(val) {
+        this.showCreateTestModal = val;
+    }
+    setSuiteToRename(val) {
+        this.suiteToRename = val;
+    }
+    changeSuite(suite) {
+        this.selectedSuiteState = this.suiteStates.find((s) => s.suite === suite);
+    }
+    init() {
+        this.suiteStates = this.ingest.tests.map((suite) => new IngestTestSuiteState(this.editorStore, this, suite));
+        this.selectedSuiteState = this.suiteStates[0];
+    }
+    *createSuite(suiteName, testName, datasetName) {
+        if (this.ingest.tests.find((suite) => suite.id === suiteName)) {
+            return 'Duplicated suite name';
+        }
+        const suite = new IngestTestSuite();
+        suite.id = suiteName;
+        suite.tests = [];
+        const suiteState = new IngestTestSuiteState(this.editorStore, this, suite);
+        const error = (yield flowResult(suiteState.addNewTest(testName, datasetName)));
+        if (error) {
+            return error;
+        }
+        addUniqueEntry(this.ingest.tests, suite);
+        this.suiteStates.push(suiteState);
+        this.selectedSuiteState = suiteState;
+        return undefined;
+    }
+    deleteSuite(suiteState) {
+        deleteEntry(this.ingest.tests, suiteState.suite);
+        deleteEntry(this.suiteStates, suiteState);
+        if (this.selectedSuiteState === suiteState) {
+            this.selectedSuiteState = this.suiteStates[0];
+        }
+    }
+}
+//# sourceMappingURL=IngestTestableState.js.map

package/lib/stores/editor/editor-state/element-editor-state/ingest/IngestTestableState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IngestTestableState.js","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/ingest/IngestTestableState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAEL,gBAAgB,EAChB,WAAW,EACX,eAAe,EAEf,iBAAiB,EACjB,eAAe,EACf,mCAAmC,EAEnC,eAAe,EACf,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EAErB,uBAAuB,EACvB,4BAA4B,EAC5B,2BAA2B,GAC5B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,cAAc,EACd,WAAW,EACX,IAAI,EACJ,IAAI,GACL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,MAAM,EACN,IAAI,EACJ,UAAU,EACV,cAAc,EACd,UAAU,EACV,WAAW,GACZ,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,iBAAiB,EAAE,MAAM,4DAA4D,CAAC;AAC/F,OAAO,EACL,uBAAuB,EACvB,4BAA4B,GAC7B,MAAM,oCAAoC,CAAC;AAG5C,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EACL,oCAAoC,EACpC,0BAA0B,EAC1B,6BAA6B,EAC7B,6BAA6B,GAC9B,MAAM,uCAAuC,CAAC;AAI/C,iFAAiF;AAEjF,MAAM,6BAA6B,GAAG,CACpC,WAAmB,EACnB,OAAiB,EACA,EAAE;IACnB,MAAM,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAC;IACvC,MAAM,GAAG,GAAG,2BAA2B,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC;IACnE,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;IACnC,QAAQ,CAAC,KAAK,GAAG,CAAC,WAAW,CAAC,CAAC;IAC/B,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;IAC3B,QAAQ,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACtB,OAAO,uBAAuB,CAAC,QAAQ,CAAC,CAAC;AAC3C,CAAC,CAAC;AAEF,MAAM,yBAAyB,GAAG,KAAK,EACrC,WAAwB,EACxB,MAAwB,EACxB,WAAmB,EACY,EAAE;IACjC,MAAM,OAAO,GAAG,MAAM,CAAC,oCAAoC,EAAE,IAAI,CAC/D,CAAC,SAA2B,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,WAAW,CAChE,CAAC;IACF,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,YAAY,GAChB,MAAM,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,qBAAqB,CACpE,OAAO,CAAC,MAAM,CAAC,QAAQ,EACvB,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACpC,CAAC;IACJ,OAAO,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC3D,CAAC,CAAC;AAEF,iFAAiF;AAEjF,MAAM,OAAO,eAAgB,SAAQ,uBAAuB;IACjD,UAAU,CAAuB;IACjC,IAAI,CAAoB;IACxB,IAAI,GAAG,IAAI,EAAE,CAAC;IACvB,qBAAqB,CAAmC;IAExD,YAAY,UAAgC,EAAE,IAAuB;QACnE,KAAK,CACH,UAAU,CAAC,aAAa,CAAC,MAAM,EAC/B,IAAI,EACJ,UAAU,CAAC,aAAa,CAAC,2BAA2B,CAAC,UAAU,EAC/D,UAAU,CAAC,WAAW,CACvB,CAAC;QAEF,cAAc,CAAC,IAAI,EAAE;YACnB,qBAAqB,EAAE,UAAU;YACjC,WAAW,EAAE,UAAU;YACvB,iBAAiB,EAAE,UAAU;YAC7B,qBAAqB,EAAE,UAAU;YACjC,eAAe,EAAE,UAAU;YAC3B,iBAAiB,EAAE,UAAU;YAC7B,qBAAqB,EAAE,UAAU;YACjC,cAAc,EAAE,MAAM;YACtB,oBAAoB,EAAE,MAAM;YAC5B,YAAY,EAAE,MAAM;YACpB,eAAe,EAAE,MAAM;YACvB,aAAa,EAAE,MAAM;YACrB,WAAW,EAAE,MAAM;YACnB,gBAAgB,EAAE,MAAM;YACxB,OAAO,EAAE,IAAI;YACb,YAAY,EAAE,MAAM;SACrB,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAClC,IAAI,CAAC,0BAA0B,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,YAAY,CAAC,GAAW;QACtB,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC;QAC1B,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED,IAAI,YAAY;QACd,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;QAC7B,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CACzC,CAAC,SAAS,EAAgC,EAAE,CAC1C,SAAS,YAAY,eAAe,CACvC,CAAC;QACF,MAAM,QAAQ,GAAG,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,MAAM,iBAAiB,GACrB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,2BAA2B,CAAC,mBAAmB,EAAE,CAAC;QAClF,OAAO,iBAAiB,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;IACvC,CAAC;IAEO,0BAA0B;QAChC,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CACzC,CAAC,SAAS,EAAgC,EAAE,CAC1C,SAAS,YAAY,eAAe,CACvC,CAAC;QACF,MAAM,iBAAiB,GAAG,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvD,IAAI,iBAAiB,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,iBAAiB,CAAC;YACxC,OAAO;QACT,CAAC;QACD,MAAM,iBAAiB,GACrB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,2BAA2B,CAAC,mBAAmB,EAAE,CAAC;QAClF,IAAI,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CACzC,CAAC,SAAS,EAAgC,EAAE,CAC1C,SAAS,YAAY,eAAe,CACvC,CAAC;QACF,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,yBAAyB,CACnD,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,MAAM,EACpC,IAAI,CAAC,YAAY,CAClB,CAAC;gBACF,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;oBAC1B,WAAW,CAAC,GAAG,EAAE;wBACf,SAAS,CAAC,QAAQ,CAAC,OAAO,GAAG,aAAa,CAAC;oBAC7C,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,mBAAmB;YACrB,CAAC;QACH,CAAC;QAED,WAAW,CAAC,GAAG,EAAE;YACf,IAAI,CAAC,qBAAqB,GAAG,IAAI,oBAAoB,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,gFAAgF;AAChF,oFAAoF;AACpF,kFAAkF;AAElF,gFAAgF;AAEhF,MAAM,OAAO,mBAAmB;IACrB,WAAW,CAAc;IACzB,UAAU,CAAuB;IAE1C,qBAAqB,GAAoC,EAAE,CAAC;IAC5D,4BAA4B,CAA4C;IACxE,mBAAmB,GAAG,KAAK,CAAC;IAE5B,YACE,UAAgC,EAChC,OAEC;QAED,cAAc,CAAC,IAAI,EAAE;YACnB,qBAAqB,EAAE,UAAU;YACjC,4BAA4B,EAAE,UAAU;YACxC,mBAAmB,EAAE,UAAU;YAC/B,+BAA+B,EAAE,MAAM;YACvC,sBAAsB,EAAE,MAAM;YAC9B,UAAU,EAAE,MAAM;YAClB,aAAa,EAAE,MAAM;YACrB,4BAA4B,EAAE,MAAM;SACrC,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC1C,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED,IAAI,sBAAsB;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QACpC,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,KAAK,CAAC,QAAQ;aACX,MAAM,CACL,CAAC,QAAQ,EAAwD,EAAE,CACjE,QAAQ,YAAY,gBAAgB;YACpC,QAAQ,YAAY,qBAAqB,CAC5C;aACA,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAClD,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CAAC;QACvD,MAAM,iBAAiB,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC;QACpE,MAAM,UAAU,GAAyB;YACvC,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,iBAAiB,CAAC;YACtE,GAAG,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,WAAW,CAAC;SAC7D,CAAC;QACF,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1E,CAAC;IAED,sBAAsB,CAAC,GAAY;QACjC,IAAI,CAAC,mBAAmB,GAAG,GAAG,CAAC;IACjC,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,MAAM,OAAO,GACX,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACpE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;QACxC,QAAQ,CAAC,OAAO,GAAG,mCAAmC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACvE,MAAM,oBAAoB,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACxD,oBAAoB,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3C,4BAA4B,CAAC,oBAAoB,CAAC,CAAC;QACnD,QAAQ,CAAC,IAAI,GAAG,oBAAoB,CAAC;QAErC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QACpC,KAAK,CAAC,QAAQ,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,4BAA4B,CAAC,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,aAAa,CAAC,YAA2C;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QACpC,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC5D,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAChC,IAAI,CAAC,4BAA4B,EAAE,CAAC;IACtC,CAAC;IAED,4BAA4B,CAAC,OAE5B;QACC,MAAM,6BAA6B,GACjC,OAAO,EAAE,mBAAmB;YAC5B,IAAI,CAAC,4BAA4B,EAAE,OAAO,CAAC,IAAI,CAAC;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QAEpC,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC,QAAQ;aACxC,MAAM,CACL,CAAC,QAAQ,EAAgC,EAAE,CACzC,QAAQ,YAAY,gBAAgB,CACvC;aACA,GAAG,CACF,CAAC,QAAQ,EAAE,EAAE,CACX,IAAI,6BAA6B,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAChE,CAAC;QAEJ,IAAI,CAAC,4BAA4B;YAC/B,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAC7B,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,6BAA6B,CAChE,IAAI,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,+BAA+B,CAC7B,GAA8C;QAE9C,IAAI,CAAC,4BAA4B,GAAG,GAAG,CAAC;IAC1C,CAAC;CACF;AAED,iFAAiF;AAEjF,MAAM,OAAO,oBAAqB,SAAQ,4BAA4B;IAC3D,aAAa,CAAsB;IACnC,KAAK,CAAkB;IACvB,UAAU,GAAsB,EAAE,CAAC;IAE5C,aAAa,CAAsB;IAC1B,IAAI,GAAG,IAAI,EAAE,CAAC;IAEvB,YACE,WAAwB,EACxB,aAAkC,EAClC,KAAsB;QAEtB,KAAK,CACH,aAAa,CAAC,MAAM,EACpB,KAAK,EACL,aAAa,CAAC,2BAA2B,CAAC,UAAU,EACpD,WAAW,CACZ,CAAC;QACF,cAAc,CAAC,IAAI,EAAE;YACnB,UAAU,EAAE,UAAU;YACtB,iBAAiB,EAAE,UAAU;YAC7B,eAAe,EAAE,UAAU;YAC3B,aAAa,EAAE,UAAU;YACzB,UAAU,EAAE,MAAM;YAClB,UAAU,EAAE,MAAM;YAClB,eAAe,EAAE,MAAM;YACvB,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,IAAI;YACd,eAAe,EAAE,IAAI;YACrB,eAAe,EAAE,MAAM;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;IAED,oBAAoB;QAClB,IAAI,CAAC,aAAa,GAAG,IAAI,mBAAmB,CAAC,IAAI,EAAE;YACjD,mBAAmB,EACjB,IAAI,CAAC,aAAa,CAAC,4BAA4B,EAAE,OAAO,CAAC,IAAI;SAChE,CAAC,CAAC;IACL,CAAC;IAED,eAAe;QACb,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YAChC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;YAC3B,IAAI,IAAI,YAAY,iBAAiB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACzD,MAAM,iBAAiB,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAC5C,CAAC,SAAS,EAAgC,EAAE,CAC1C,SAAS,YAAY,eAAe,CACvC,CAAC;gBACF,IAAI,CAAC,SAAS;oBACZ,iBAAiB,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;wBACpC,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC,CAAC;wBAClC,EAAE,CAAC;YACP,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CACpC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,eAAe,CAAC,IAAI,EAAE,IAAyB,CAAC,CAC/D,CAAC;QACF,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAEQ,UAAU,CAAC,IAAuB;QACzC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,sEAAsE;YACtE,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,CAAC,UAAU,CACT,QAAgB,EAChB,WAAmB;QAEnB,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,CAAC;YAC1D,OAAO,sBAAsB,CAAC;QAChC,CAAC;QAED,IAAI,eAAe,GAAa,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,CAAC,MAAM,yBAAyB,CAC9C,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,CAAC,MAAM,EACzB,WAAW,CACZ,CAAyB,CAAC;YAC3B,IAAI,OAAO,EAAE,CAAC;gBACZ,eAAe,GAAG,OAAO,CAAC;YAC5B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,8DAA8D;QAChE,CAAC;QAED,MAAM,OAAO,GACX,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,oCAAoC,EAAE,IAAI,CAClE,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,WAAW,CAC9C,CAAC;QACJ,MAAM,SAAS,GAAG,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC;QAE3C,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,GAAG,GACP,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,2BAA2B,CAChF,SAAS,EACT,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAe,CAAC;YACnB,MAAM,iBAAiB,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAChD,6BAA6B,CAAC,QAAQ,CAAC,CACxC,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,GAAG,EAAsB,CAAC;YAChD,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC/D,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACrB,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACpD,CAAC;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,SAAS,CAAC,IAAI,SAAS,EAAE,CAAC;gBACjD,MAAM,OAAO,GACX,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CAAC,kBAAkB,CACzD,WAAW,CACZ,CAAC;gBACJ,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,SAAS;gBACX,CAAC;gBAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAC/C,CAAC,QAAQ,EAAgC,EAAE,CACzC,QAAQ,YAAY,gBAAgB;oBACpC,QAAQ,CAAC,OAAO,CAAC,KAAK,KAAK,OAAO,CACrC,CAAC;gBACF,MAAM,oBAAoB,GACxB,gBAAgB,EAAE,IAAI,YAAY,oBAAoB;oBACpD,CAAC,CAAC,gBAAgB,CAAC,IAAI;oBACvB,CAAC,CAAC,SAAS,CAAC;gBAEhB,IAAI,CAAC,gBAAgB,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC/C,MAAM,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;oBACxC,QAAQ,CAAC,OAAO;wBACd,mCAAmC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBACtD,MAAM,YAAY,GAAG,oCAAoC,CAAC,SAAS,CAAC,CAAC;oBACrE,4BAA4B,CAAC,YAAY,CAAC,CAAC;oBAC3C,QAAQ,CAAC,IAAI,GAAG,YAAY,CAAC;oBAC7B,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;gBAC3D,CAAC;qBAAM,CAAC;oBACN,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;wBACjC,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC;wBACjC,IACE,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,IAAI,CACzC,CAAC,eAAe,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,MAAM,CACzD,EACD,CAAC;4BACD,MAAM,OAAO,GAAG,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAC/C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CACxB,CAAC;4BACF,oBAAoB,CAAC,gBAAgB,CAAC,IAAI,CACxC,0BAA0B,CAAC,MAAM,EAAE,OAAO,CAAC,CAC5C,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,aAAa,CACjE,sEAAsE,CACvE,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrC,8EAA8E;YAC9E,MAAM,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;YACxC,QAAQ,CAAC,OAAO,GAAG,mCAAmC,CAAC,MAAM,CAC3D,IAAI,CAAC,aAAa,CAAC,MAAM,CAC1B,CAAC;YACF,MAAM,YAAY,GAAG,IAAI,oBAAoB,EAAE,CAAC;YAChD,YAAY,CAAC,gBAAgB,GAAG;gBAC9B,0BAA0B,CAAC,WAAW,EAAE,eAAe,CAAC;aACzD,CAAC;YACF,4BAA4B,CAAC,YAAY,CAAC,CAAC;YAC3C,QAAQ,CAAC,IAAI,GAAG,YAAY,CAAC;YAC7B,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACrC,IAAI,CAAC,EAAE,GAAG,QAAQ,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;QAE3B,MAAM,SAAS,GAAG,IAAI,eAAe,EAAE,CAAC;QACxC,SAAS,CAAC,EAAE,GAAG,UAAU,CAAC;QAC1B,SAAS,CAAC,QAAQ,GAAG,6BAA6B,CAChD,WAAW,EACX,eAAe,CAChB,CAAC;QACF,IAAI,CAAC,UAAU,GAAG,CAAC,SAAS,CAAC,CAAC;QAE9B,iBAAiB,CACf,IAAI,CAAC,KAAK,EACV,IAAI,EACJ,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,eAAe,CACtD,CAAC;QAEF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAE5B,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAED,iFAAiF;AAEjF,MAAM,OAAO,mBAAmB;IACrB,WAAW,CAAc;IACzB,2BAA2B,CAA8B;IAElE,WAAW,GAA2B,EAAE,CAAC;IACzC,kBAAkB,CAAmC;IACrD,oBAAoB,GAAG,KAAK,CAAC;IAC7B,mBAAmB,GAAG,KAAK,CAAC;IAC5B,aAAa,CAAwB;IAErC,YAAY,2BAAwD;QAClE,cAAc,CAAC,IAAI,EAAE;YACnB,WAAW,EAAE,UAAU;YACvB,kBAAkB,EAAE,UAAU;YAC9B,oBAAoB,EAAE,UAAU;YAChC,mBAAmB,EAAE,UAAU;YAC/B,aAAa,EAAE,UAAU;YACzB,qBAAqB,EAAE,MAAM;YAC7B,uBAAuB,EAAE,MAAM;YAC/B,sBAAsB,EAAE,MAAM;YAC9B,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,MAAM;YACnB,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,IAAI;YACjB,WAAW,EAAE,MAAM;SACpB,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,2BAA2B,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,2BAA2B,GAAG,2BAA2B,CAAC;IACjE,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC;IACjD,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,2BAA2B,CAAC,mBAAmB,EAAE,CAAC;IAChE,CAAC;IAED,qBAAqB,CAAC,GAAqC;QACzD,IAAI,CAAC,kBAAkB,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,uBAAuB,CAAC,GAAY;QAClC,IAAI,CAAC,oBAAoB,GAAG,GAAG,CAAC;IAClC,CAAC;IAED,sBAAsB,CAAC,GAAY;QACjC,IAAI,CAAC,mBAAmB,GAAG,GAAG,CAAC;IACjC,CAAC;IAED,gBAAgB,CAAC,GAA0B;QACzC,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;IAC3B,CAAC;IAED,WAAW,CAAC,KAAgB;QAC1B,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IAC5E,CAAC;IAED,IAAI;QACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,oBAAoB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,CACnE,CAAC;QACF,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,CAAC,WAAW,CACV,SAAiB,EACjB,QAAgB,EAChB,WAAmB;QAEnB,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,SAAS,CAAC,EAAE,CAAC;YAC9D,OAAO,uBAAuB,CAAC;QACjC,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;QACpC,KAAK,CAAC,EAAE,GAAG,SAAS,CAAC;QACrB,KAAK,CAAC,KAAK,GAAG,EAAE,CAAC;QAEjB,MAAM,UAAU,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAE3E,MAAM,KAAK,GAAG,CAAC,MAAM,UAAU,CAC7B,UAAU,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,CAAC,CAC7C,CAAuB,CAAC;QAEzB,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,KAAK,CAAC;QACf,CAAC;QAED,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,IAAI,CAAC,kBAAkB,GAAG,UAAU,CAAC;QACrC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,WAAW,CAAC,UAAgC;QAC1C,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC;QACjD,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QAC1C,IAAI,IAAI,CAAC,kBAAkB,KAAK,UAAU,EAAE,CAAC;YAC3C,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;CACF"}

package/lib/stores/editor/editor-state/element-editor-state/testable/LakehouseTestableUtils.d.ts

@@ -0,0 +1,62 @@
+/**
+ * Copyright (c) 2026-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.
+ */
+/**
+ * Shared utilities for Lakehouse test state:
+ * used by both DataProductTestableState and IngestTestableState.
+ */
+import { type Accessor, type AbstractPureGraphManager, type BaseDataResolver, RelationElement, RelationElementsData, type PackageableElement } from '@finos/legend-graph';
+import type { EditorStore } from '../../../EditorStore.js';
+import { RelationElementsDataState } from '../data/EmbeddedDataState.js';
+export declare const createEmptyRelationElement: (itemId: string, columns?: string[]) => RelationElement;
+export declare const buildRelationElementsDataWithColumns: (accessors: Accessor[]) => RelationElementsData;
+export declare const isIngestOrDataProductAccessor: (accessor: Accessor) => accessor is Accessor;
+export interface ElementDataItem {
+    id: string;
+    label: string;
+}
+export declare const getElementDataItems: (element: PackageableElement, graphManager: AbstractPureGraphManager) => ElementDataItem[];
+/**
+ * Wraps a single BaseDataResolver within a test suite's testData array.
+ * Shared between DataProduct and Ingest test suite states — both populate
+ * test data by wiring BaseDataResolver entries to PackageableElements.
+ */
+export declare class LakehouseElementTestDataState {
+    readonly testData: BaseDataResolver;
+    readonly editorStore: EditorStore;
+    readonly relationElementsDataState: RelationElementsDataState | undefined;
+    constructor(testData: BaseDataResolver, editorStore: EditorStore);
+    get element(): PackageableElement;
+    get itemLabel(): string;
+    private initAccessorOptions;
+    private refreshAccessorOptions;
+}
+/**
+ * Structural interface satisfied by both DataProductTestDataState and
+ * IngestTestDataState. Used to type the shared LakehouseTestDataEditor
+ * component without importing the concrete state classes.
+ */
+export interface LakehouseTestDataStateBase {
+    elementTestDataStates: LakehouseElementTestDataState[];
+    selectedElementTestDataState: LakehouseElementTestDataState | undefined;
+    setSelectedElementTestDataState(val: LakehouseElementTestDataState | undefined): void;
+    availableElementsToAdd: PackageableElement[];
+    showAddElementModal: boolean;
+    setShowAddElementModal(val: boolean): void;
+    addElement(path: string): void;
+    deleteElement(state: LakehouseElementTestDataState): void;
+    editorStore: EditorStore;
+}
+//# sourceMappingURL=LakehouseTestableUtils.d.ts.map

package/lib/stores/editor/editor-state/element-editor-state/testable/LakehouseTestableUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LakehouseTestableUtils.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/LakehouseTestableUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH;;;GAGG;AAEH,OAAO,EACL,KAAK,QAAQ,EAEb,KAAK,wBAAwB,EAC7B,KAAK,gBAAgB,EAGrB,eAAe,EACf,oBAAoB,EAKpB,KAAK,kBAAkB,EACxB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,yBAAyB,EAAE,MAAM,8BAA8B,CAAC;AAIzE,eAAO,MAAM,0BAA0B,GACrC,QAAQ,MAAM,EACd,UAAS,MAAM,EAAO,KACrB,eAQF,CAAC;AAEF,eAAO,MAAM,oCAAoC,GAC/C,WAAW,QAAQ,EAAE,KACpB,oBAQF,CAAC;AAEF,eAAO,MAAM,6BAA6B,GACxC,UAAU,QAAQ,KACjB,QAAQ,IAAI,QAEqC,CAAC;AAErD,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;CACf;AAED,eAAO,MAAM,mBAAmB,GAC9B,SAAS,kBAAkB,EAC3B,cAAc,wBAAwB,KACrC,eAAe,EAYjB,CAAC;AAIF;;;;GAIG;AACH,qBAAa,6BAA6B;IACxC,QAAQ,CAAC,QAAQ,EAAE,gBAAgB,CAAC;IACpC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,yBAAyB,EAAE,yBAAyB,GAAG,SAAS,CAAC;gBAE9D,QAAQ,EAAE,gBAAgB,EAAE,WAAW,EAAE,WAAW;IAYhE,IAAI,OAAO,IAAI,kBAAkB,CAEhC;IAED,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,OAAO,CAAC,mBAAmB;YAWb,sBAAsB;CA+DrC;AAID;;;;GAIG;AACH,MAAM,WAAW,0BAA0B;IACzC,qBAAqB,EAAE,6BAA6B,EAAE,CAAC;IACvD,4BAA4B,EAAE,6BAA6B,GAAG,SAAS,CAAC;IACxE,+BAA+B,CAC7B,GAAG,EAAE,6BAA6B,GAAG,SAAS,GAC7C,IAAI,CAAC;IACR,sBAAsB,EAAE,kBAAkB,EAAE,CAAC;IAC7C,mBAAmB,EAAE,OAAO,CAAC;IAC7B,sBAAsB,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI,CAAC;IAC3C,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,aAAa,CAAC,KAAK,EAAE,6BAA6B,GAAG,IAAI,CAAC;IAC1D,WAAW,EAAE,WAAW,CAAC;CAC1B"}

package/lib/stores/editor/editor-state/element-editor-state/testable/LakehouseTestableUtils.js

@@ -0,0 +1,138 @@
+/**
+ * Copyright (c) 2026-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.
+ */
+/**
+ * Shared utilities for Lakehouse test state:
+ * used by both DataProductTestableState and IngestTestableState.
+ */
+import { DataProduct, IngestDefinition, RelationElement, RelationElementsData, RelationRowTestData, getAccessorItemLabelForElement, observe_RelationElement, observe_RelationRowTestData, } from '@finos/legend-graph';
+import { noop } from '@finos/legend-shared';
+import { runInAction } from 'mobx';
+import { RelationElementsDataState } from '../data/EmbeddedDataState.js';
+// ─── Pure helpers ─────────────────────────────────────────────────────────────
+export const createEmptyRelationElement = (itemId, columns = []) => {
+    const row = observe_RelationRowTestData(new RelationRowTestData());
+    row.values = columns.map(() => '');
+    const relationElement = new RelationElement();
+    relationElement.paths = [itemId];
+    relationElement.columns = columns;
+    relationElement.rows = [row];
+    return observe_RelationElement(relationElement);
+};
+export const buildRelationElementsDataWithColumns = (accessors) => {
+    const relData = new RelationElementsData();
+    relData.relationElements = accessors.map((acc) => {
+        const itemId = acc.accessor || 'UNKNOWN';
+        const columns = acc.relationType.columns.map((col) => col.name);
+        return createEmptyRelationElement(itemId, columns);
+    });
+    return relData;
+};
+export const isIngestOrDataProductAccessor = (accessor) => accessor.parentElement instanceof DataProduct ||
+    accessor.parentElement instanceof IngestDefinition;
+export const getElementDataItems = (element, graphManager) => {
+    if (element instanceof DataProduct) {
+        return element.accessPointGroups
+            .flatMap((g) => g.accessPoints)
+            .map((ap) => ({ id: ap.id, label: ap.id }));
+    }
+    if (element instanceof IngestDefinition) {
+        return graphManager
+            .getIngestDefinitionDatasetNames(element)
+            .map((name) => ({ id: name, label: name }));
+    }
+    return [];
+};
+// ─── Shared element test data state ──────────────────────────────────────────
+/**
+ * Wraps a single BaseDataResolver within a test suite's testData array.
+ * Shared between DataProduct and Ingest test suite states — both populate
+ * test data by wiring BaseDataResolver entries to PackageableElements.
+ */
+export class LakehouseElementTestDataState {
+    testData;
+    editorStore;
+    relationElementsDataState;
+    constructor(testData, editorStore) {
+        this.testData = testData;
+        this.editorStore = editorStore;
+        if (testData.data instanceof RelationElementsData) {
+            this.relationElementsDataState = new RelationElementsDataState(editorStore, testData.data);
+            this.initAccessorOptions();
+        }
+    }
+    get element() {
+        return this.testData.element.value;
+    }
+    get itemLabel() {
+        return getAccessorItemLabelForElement(this.element);
+    }
+    initAccessorOptions() {
+        const dataState = this.relationElementsDataState;
+        if (!dataState) {
+            return;
+        }
+        this.refreshAccessorOptions(dataState).catch(noop);
+        dataState.setRefreshAccessorOptions(() => this.refreshAccessorOptions(dataState));
+    }
+    async refreshAccessorOptions(dataState) {
+        const element = this.element;
+        const graphManager = this.editorStore.graphManagerState.graphManager;
+        const graph = this.editorStore.graphManagerState.graph;
+        const items = getElementDataItems(element, graphManager);
+        if (items.length === 0) {
+            dataState.setAccessorOptions(undefined, undefined);
+            return;
+        }
+        const options = await Promise.all(items.map(async (item) => {
+            let columns = [];
+            try {
+                if (element instanceof IngestDefinition) {
+                    const accessor = await graphManager.createAccessorFromPackageableElement(element, graph, { schemaName: undefined, tableName: item.id });
+                    if (accessor) {
+                        columns = accessor.relationType.columns.map((c) => c.name);
+                    }
+                }
+                else if (element instanceof DataProduct) {
+                    const accessor = await graphManager.buildDataProductAccessor(element, graph, { tableName: item.id });
+                    if (accessor) {
+                        columns = accessor.relationType.columns.map((c) => c.name);
+                    }
+                }
+            }
+            catch {
+                // best-effort column resolution
+            }
+            return { label: item.label, value: item.id, columns };
+        }));
+        runInAction(() => {
+            dataState.setAccessorOptions(options, this.itemLabel);
+            const columnsByItem = new Map(options
+                .filter((o) => o.columns.length > 0)
+                .map((o) => [o.value, o.columns]));
+            for (const relState of dataState.relationElementStates) {
+                const rel = relState.relationElement;
+                if (rel.columns.length === 0) {
+                    const key = rel.paths[rel.paths.length - 1];
+                    const cols = key ? columnsByItem.get(key) : undefined;
+                    if (cols) {
+                        rel.columns = cols;
+                    }
+                }
+            }
+        });
+    }
+}
+//# sourceMappingURL=LakehouseTestableUtils.js.map