@finos/legend-cube-builder 0.0.3 → 0.0.5

package/lib/stores/editors/CubeExpressionFilterDraft.js

@@ -0,0 +1,254 @@
+/**
+ * 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 { dependsOnDatabaseType, ExpressionFilter, MESSAGE_CONDITION_NOT_BOOLEAN, QueryEmitter, UNCHECKED, } from '@finos/legend-cube';
+import { action, computed, flow, flowResult, makeObservable, observable, } from 'mobx';
+import { EXPRESSION_FILTER_APPLY_DISABLED_REASON, EXPRESSION_FILTER_PARENTHESES_HINT, EXTEND_TO_ONE_HINT, getExtendPlanProblem, } from '../../__lib__/LegendCubeLabels.js';
+import { CubeEngineError, CubeEngineErrorKind, } from '../../graph-manager/CubeEngine.js';
+import { buildCubeConditionCheckLambdas, readCubeConditionCheck, } from '../CubeExpressionFilterCheck.js';
+import { getCubeExtendUpstream, withoutEngineContext, } from '../CubeExtendTyping.js';
+import { CubeNodeDraft } from './CubeNodeDraft.js';
+/** What a new condition's code starts as (PLAN §11.11) */
+export const NEW_CONDITION_CODE = 'x | ';
+/** An engine message about `and`, `or` or `not` given something other than a Boolean, which Pure's precedence causes */
+const NEEDS_PARENTHESES = /Can't find a match for function '(?:and|or|not)\(/u;
+/** An engine message about a column that can be empty in arithmetic, which `->toOne()` fixes */
+const NEEDS_TO_ONE = /must have a multiplicity \[1\]|Can't find a match for function '(?:plus|minus|times|divide)[^']*\[0\.\.1\]/u;
+const problemOf = (error) => ({
+    message: withoutEngineContext(error.firstLine),
+    location: error.location,
+    hint: NEEDS_PARENTHESES.test(error.detail)
+        ? EXPRESSION_FILTER_PARENTHESES_HINT
+        : NEEDS_TO_ONE.test(error.detail)
+            ? EXTEND_TO_ONE_HINT
+            : undefined,
+});
+const asEngineError = (error) => error instanceof CubeEngineError
+    ? error
+    : new CubeEngineError(CubeEngineErrorKind.NETWORK, error instanceof Error ? error.message : String(error));
+/**
+ * The Filter by Expression editor's draft (PLAN §11.11): the condition's
+ * code, `x | …`. Validate (F10) has the engine parse it, checks Cube's rules,
+ * has the engine check it after the input, located so an error points into
+ * the code, then plans the query up to this node for the cube's database.
+ * Apply waits until the code is the one validated; the node then holds the
+ * engine's lambda and its check, so it needn't be checked again.
+ */
+export class CubeExpressionFilterDraft extends CubeNodeDraft {
+    editorState;
+    code;
+    /** What the engine made of the code the last time it parsed it */
+    checked;
+    /** The problem from the last Validate, while the code is the one validated */
+    problem;
+    /** The check the last Validate gave, for the code as it was then */
+    check;
+    validating = false;
+    /** Anything was changed; until then, `build()` gives the original back */
+    touched = false;
+    /** The database type the last plan was for, named in its problem */
+    planDatabaseType;
+    constructor(original, editorState) {
+        super(original);
+        makeObservable(this, {
+            original: observable.ref,
+            code: observable,
+            checked: observable.ref,
+            problem: observable.ref,
+            check: observable.ref,
+            validating: observable,
+            touched: observable,
+            isValidated: computed,
+            isChecked: computed,
+            applyDisabledReason: computed,
+            setCode: action,
+            validate: flow,
+            follow: action,
+        });
+        this.editorState = editorState;
+        const { code, lambda } = original;
+        this.code = code || NEW_CONDITION_CODE;
+        this.checked = lambda ? { code, lambda, located: lambda } : undefined;
+    }
+    /** The input's schema, when the input is valid */
+    get inputSchema() {
+        const { query } = this.editorState.document;
+        const [inputId] = query.getInputIds(this.original.id);
+        return inputId === undefined
+            ? undefined
+            : this.editorState.analysis.schemas.get(inputId);
+    }
+    /** The code is the one the engine last parsed */
+    get isValidated() {
+        return this.checked?.code === this.code;
+    }
+    /** The code is the one validated, and the engine said it is one Boolean of a row */
+    get isChecked() {
+        if (!this.isValidated) {
+            return false;
+        }
+        const input = this.inputSchema;
+        const check = this.check ?? this.original.check;
+        return (check.kind === 'checked' &&
+            input !== undefined &&
+            this.build().currentCheck(input).kind === 'checked');
+    }
+    /** Why Apply waits: a validation running, or the code not validated yet */
+    get applyDisabledReason() {
+        return this.validating
+            ? EXPRESSION_FILTER_APPLY_DISABLED_REASON.VALIDATING
+            : this.isValidated
+                ? undefined
+                : EXPRESSION_FILTER_APPLY_DISABLED_REASON.NOT_VALIDATED;
+    }
+    /** A code edit drops the problem, until the next Validate */
+    setCode(code) {
+        if (code === this.code) {
+            return;
+        }
+        this.code = code;
+        this.problem = undefined;
+        this.touched = true;
+    }
+    build() {
+        if (!this.touched) {
+            return this.original;
+        }
+        return this.original
+            .withCondition(this.code, this.isValidated ? this.checked?.lambda : undefined)
+            .withCheck(this.check ?? this.original.check);
+    }
+    /**
+     * Goes on from a filter that differs from the original only in its check,
+     * e.g. checked in the background after an Apply, keeping the code
+     */
+    follow(node) {
+        const { original } = this;
+        if (!(node instanceof ExpressionFilter) ||
+            node.id !== original.id ||
+            JSON.stringify([node.code, node.lambda, node.rest]) !==
+                JSON.stringify([original.code, original.lambda, original.rest])) {
+            return false;
+        }
+        this.original = node;
+        return true;
+    }
+    /**
+     * Checks the condition with the engine (F10): parses the code when it
+     * changed since the last Validate, checks Cube's own rules and the
+     * database's, then, when the input is valid, checks the condition after
+     * the input, located in the code, and plans the query up to this node for
+     * the cube's database
+     */
+    *validate() {
+        const { editorState } = this;
+        const { context } = editorState.document;
+        if (this.validating || !context) {
+            return;
+        }
+        this.validating = true;
+        this.touched = true;
+        try {
+            const nodeId = this.original.id;
+            const { code } = this;
+            // 1. parse the code, if it changed
+            let parsed = this.checked?.code === code ? this.checked : undefined;
+            if (!parsed) {
+                const result = (yield editorState.host.engine
+                    .parseExpression(code, `${nodeId}:condition`)
+                    .then((expression) => ({ code, ...expression }), asEngineError));
+                if (result instanceof CubeEngineError) {
+                    this.checked = undefined;
+                    this.problem = problemOf(result);
+                    this.check = UNCHECKED;
+                    return;
+                }
+                parsed = result;
+                this.checked = parsed;
+            }
+            this.problem = undefined;
+            const input = this.inputSchema;
+            if (!input) {
+                return;
+            }
+            // 2. Cube's own checks, which need no engine, then the database's:
+            // a pattern given % or _ on DuckDB, or one that isn't a literal but on
+            // Snowflake, which the model's outline tells
+            const candidate = this.original.withCondition(code, parsed.lambda);
+            if (dependsOnDatabaseType(candidate)) {
+                yield flowResult(editorState.loadModelOutline(context.model));
+            }
+            const cubeProblem = candidate.getConditionProblem() ??
+                editorState.getNodeDatabaseProblems(editorState.document.query, candidate)[0];
+            if (cubeProblem !== undefined) {
+                this.problem = { message: cubeProblem, refused: true };
+                this.check = UNCHECKED;
+                return;
+            }
+            // 3. check the condition after the input, located in its code
+            const query = editorState.document.query.replace(candidate);
+            const answers = (yield editorState.host.engine.typeLambdas(context.model, buildCubeConditionCheckLambdas(query, candidate, parsed.located, input, editorState.registry, true)));
+            const { check, error } = readCubeConditionCheck(answers, candidate, input, getCubeExtendUpstream(new QueryEmitter(query, editorState.registry), candidate.id, context.model));
+            this.check = check;
+            if (check.kind === 'failed') {
+                this.problem =
+                    check.resultType !== undefined
+                        ? { message: MESSAGE_CONDITION_NOT_BOOLEAN(check.resultType) }
+                        : error
+                            ? problemOf(error)
+                            : { message: check.message };
+                return;
+            }
+            // 4. plan the query up to this node, for the cube's database
+            const failure = (yield this.planFrom(query.replace(candidate.withCheck(check)), candidate.id));
+            if (failure) {
+                this.problem = {
+                    message: getExtendPlanProblem(this.planDatabaseType, failure.firstLine),
+                };
+            }
+        }
+        finally {
+            this.validating = false;
+        }
+    }
+    /** Plans the query up to the checked candidate, as a run with it selected would; the database's refusal, if any */
+    async planFrom(query, nodeId) {
+        const { editorState } = this;
+        const { context } = editorState.document;
+        const runtime = context?.runtime;
+        // a cube with no runtime yet can't be planned
+        if (!context || runtime === undefined) {
+            return undefined;
+        }
+        await flowResult(editorState.loadModelOutline(context.model));
+        this.planDatabaseType = editorState.getRunDatabaseType(query, nodeId, context.model, runtime);
+        let lambda;
+        try {
+            lambda = new QueryEmitter(query.selected === nodeId ? query : query.select(nodeId), editorState.registry).emitExecutionLambda({
+                rowLimit: 1,
+                runtime,
+                databaseType: this.planDatabaseType,
+            });
+        }
+        catch (error) {
+            // the query up to here can't run, e.g. a node above is invalid
+            return new CubeEngineError(CubeEngineErrorKind.COMPILE, error instanceof Error ? error.message : String(error), nodeId);
+        }
+        return editorState.host.engine
+            .planLambda(context.model, lambda)
+            .then(() => undefined, asEngineError);
+    }
+}
+//# sourceMappingURL=CubeExpressionFilterDraft.js.map

package/lib/stores/editors/CubeExpressionFilterDraft.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CubeExpressionFilterDraft.js","sourceRoot":"","sources":["../../../src/stores/editors/CubeExpressionFilterDraft.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,qBAAqB,EAErB,gBAAgB,EAGhB,6BAA6B,EAE7B,YAAY,EAGZ,SAAS,GACV,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EACL,MAAM,EACN,QAAQ,EACR,IAAI,EACJ,UAAU,EACV,cAAc,EACd,UAAU,GACX,MAAM,MAAM,CAAC;AACd,OAAO,EACL,uCAAuC,EACvC,kCAAkC,EAClC,kBAAkB,EAClB,oBAAoB,GACrB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EACL,eAAe,EACf,mBAAmB,GAIpB,MAAM,mCAAmC,CAAC;AAE3C,OAAO,EACL,8BAA8B,EAC9B,sBAAsB,GACvB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EACL,qBAAqB,EACrB,oBAAoB,GACrB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAsBnD,0DAA0D;AAC1D,MAAM,CAAC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAEzC,wHAAwH;AACxH,MAAM,iBAAiB,GAAG,oDAAoD,CAAC;AAE/E,gGAAgG;AAChG,MAAM,YAAY,GAChB,6GAA6G,CAAC;AAEhH,MAAM,SAAS,GAAG,CAAC,KAAsB,EAAwB,EAAE,CAAC,CAAC;IACnE,OAAO,EAAE,oBAAoB,CAAC,KAAK,CAAC,SAAS,CAAC;IAC9C,QAAQ,EAAE,KAAK,CAAC,QAAQ;IACxB,IAAI,EAAE,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QACxC,CAAC,CAAC,kCAAkC;QACpC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAC/B,CAAC,CAAC,kBAAkB;YACpB,CAAC,CAAC,SAAS;CAChB,CAAC,CAAC;AAEH,MAAM,aAAa,GAAG,CAAC,KAAc,EAAmB,EAAE,CACxD,KAAK,YAAY,eAAe;IAC9B,CAAC,CAAC,KAAK;IACP,CAAC,CAAC,IAAI,eAAe,CACjB,mBAAmB,CAAC,OAAO,EAC3B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAC;AAER;;;;;;;GAOG;AACH,MAAM,OAAO,yBAA0B,SAAQ,aAA+B;IACnE,WAAW,CAAkB;IACtC,IAAI,CAAS;IACb,kEAAkE;IAClE,OAAO,CAAuC;IAC9C,8EAA8E;IAC9E,OAAO,CAAmC;IAC1C,oEAAoE;IACpE,KAAK,CAA8B;IACnC,UAAU,GAAG,KAAK,CAAC;IACnB,0EAA0E;IAClE,OAAO,GAAG,KAAK,CAAC;IACxB,oEAAoE;IAC5D,gBAAgB,CAAqB;IAE7C,YAAY,QAA0B,EAAE,WAA4B;QAClE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChB,cAAc,CAAuC,IAAI,EAAE;YACzD,QAAQ,EAAE,UAAU,CAAC,GAAG;YACxB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,UAAU,CAAC,GAAG;YACvB,OAAO,EAAE,UAAU,CAAC,GAAG;YACvB,KAAK,EAAE,UAAU,CAAC,GAAG;YACrB,UAAU,EAAE,UAAU;YACtB,OAAO,EAAE,UAAU;YACnB,WAAW,EAAE,QAAQ;YACrB,SAAS,EAAE,QAAQ;YACnB,mBAAmB,EAAE,QAAQ;YAC7B,OAAO,EAAE,MAAM;YACf,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,QAAQ,CAAC;QAClC,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,kBAAkB,CAAC;QACvC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACxE,CAAC;IAED,kDAAkD;IAClD,IAAI,WAAW;QACb,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACtD,OAAO,OAAO,KAAK,SAAS;YAC1B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACrD,CAAC;IAED,iDAAiD;IACjD,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC;IAC1C,CAAC;IAED,oFAAoF;IACpF,IAAI,SAAS;QACX,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAChD,OAAO,CACL,KAAK,CAAC,IAAI,KAAK,SAAS;YACxB,KAAK,KAAK,SAAS;YACnB,IAAI,CAAC,KAAK,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,SAAS,CACpD,CAAC;IACJ,CAAC;IAED,2EAA2E;IAC3E,IAAa,mBAAmB;QAC9B,OAAO,IAAI,CAAC,UAAU;YACpB,CAAC,CAAC,uCAAuC,CAAC,UAAU;YACpD,CAAC,CAAC,IAAI,CAAC,WAAW;gBAChB,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,uCAAuC,CAAC,aAAa,CAAC;IAC9D,CAAC;IAED,6DAA6D;IAC7D,OAAO,CAAC,IAAY;QAClB,IAAI,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACtB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ;aACjB,aAAa,CACZ,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,CACpD;aACA,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC;IAED;;;OAGG;IACM,MAAM,CAAC,IAAe;QAC7B,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;QAC1B,IACE,CAAC,CAAC,IAAI,YAAY,gBAAgB,CAAC;YACnC,IAAI,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE;YACvB,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjD,IAAI,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,EACjE,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACH,CAAC,QAAQ;QACP,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;QAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,WAAW,CAAC,QAAQ,CAAC;QACzC,IAAI,IAAI,CAAC,UAAU,IAAI,CAAC,OAAO,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;YACtB,mCAAmC;YACnC,IAAI,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;YACpE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,MAAM,GAAG,CAAC,MAAM,WAAW,CAAC,IAAI,CAAC,MAAM;qBAC1C,eAAe,CAAC,IAAI,EAAE,GAAG,MAAM,YAAY,CAAC;qBAC5C,IAAI,CACH,CAAC,UAAgC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,UAAU,EAAE,CAAC,EAC/D,aAAa,CACd,CAA+C,CAAC;gBACnD,IAAI,MAAM,YAAY,eAAe,EAAE,CAAC;oBACtC,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;oBACzB,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;oBACjC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;oBACvB,OAAO;gBACT,CAAC;gBACD,MAAM,GAAG,MAAM,CAAC;gBAChB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YACxB,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;YACzB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC;YAC/B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,mEAAmE;YACnE,uEAAuE;YACvE,6CAA6C;YAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACnE,IAAI,qBAAqB,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,MAAM,UAAU,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;YAChE,CAAC;YACD,MAAM,WAAW,GACf,SAAS,CAAC,mBAAmB,EAAE;gBAC/B,WAAW,CAAC,uBAAuB,CACjC,WAAW,CAAC,QAAQ,CAAC,KAAK,EAC1B,SAAS,CACV,CAAC,CAAC,CAAC,CAAC;YACP,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC9B,IAAI,CAAC,OAAO,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBACvD,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;gBACvB,OAAO;YACT,CAAC;YACD,8DAA8D;YAC9D,MAAM,KAAK,GAAG,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CACxD,OAAO,CAAC,KAAK,EACb,8BAA8B,CAC5B,KAAK,EACL,SAAS,EACT,MAAM,CAAC,OAAO,EACd,KAAK,EACL,WAAW,CAAC,QAAQ,EACpB,IAAI,CACL,CACF,CAAkD,CAAC;YACpD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,sBAAsB,CAC7C,OAAO,EACP,SAAS,EACT,KAAK,EACL,qBAAqB,CACnB,IAAI,YAAY,CAAC,KAAK,EAAE,WAAW,CAAC,QAAQ,CAAC,EAC7C,SAAS,CAAC,EAAE,EACZ,OAAO,CAAC,KAAK,CACd,CACF,CAAC;YACF,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;YACnB,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC5B,IAAI,CAAC,OAAO;oBACV,KAAK,CAAC,UAAU,KAAK,SAAS;wBAC5B,CAAC,CAAC,EAAE,OAAO,EAAE,6BAA6B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;wBAC9D,CAAC,CAAC,KAAK;4BACL,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC;4BAClB,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,6DAA6D;YAC7D,MAAM,OAAO,GAAG,CAAC,MAAM,IAAI,CAAC,QAAQ,CAClC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EACzC,SAAS,CAAC,EAAE,CACb,CAAgC,CAAC;YAClC,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC,OAAO,GAAG;oBACb,OAAO,EAAE,oBAAoB,CAC3B,IAAI,CAAC,gBAAgB,EACrB,OAAO,CAAC,SAAS,CAClB;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,mHAAmH;IAC3G,KAAK,CAAC,QAAQ,CACpB,KAAY,EACZ,MAAc;QAEd,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;QAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,WAAW,CAAC,QAAQ,CAAC;QACzC,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC;QACjC,8CAA8C;QAC9C,IAAI,CAAC,OAAO,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,UAAU,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9D,IAAI,CAAC,gBAAgB,GAAG,WAAW,CAAC,kBAAkB,CACpD,KAAK,EACL,MAAM,EACN,OAAO,CAAC,KAAK,EACb,OAAO,CACR,CAAC;QACF,IAAI,MAAU,CAAC;QACf,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,YAAY,CACvB,KAAK,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EACxD,WAAW,CAAC,QAAQ,CACrB,CAAC,mBAAmB,CAAC;gBACpB,QAAQ,EAAE,CAAC;gBACX,OAAO;gBACP,YAAY,EAAE,IAAI,CAAC,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,+DAA+D;YAC/D,OAAO,IAAI,eAAe,CACxB,mBAAmB,CAAC,OAAO,EAC3B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EACtD,MAAM,CACP,CAAC;QACJ,CAAC;QACD,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM;aAC3B,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC;aACjC,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IAC1C,CAAC;CACF"}

package/lib/stores/editors/CubeExtendDraft.d.ts

@@ -0,0 +1,110 @@
+/**
+ * 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 { type CubeType, Extend, type ExtendColumn, type ExtendTyping, type JsonObject, type QueryNode, type Schema } from '@finos/legend-cube';
+import type { GeneratorFn } from '@finos/legend-shared';
+import { type CubeSourceLocation } from '../../graph-manager/CubeEngine.js';
+import type { CubeEditorState } from '../CubeEditorState.js';
+import { CubeNodeDraft } from './CubeNodeDraft.js';
+/** What the engine made of a row's code the last time it was validated */
+export interface CubeExtendCheckedCode {
+    readonly code: string;
+    readonly lambda: JsonObject;
+    readonly located: JsonObject;
+}
+/** A row's problem from the last Validate: the engine's message, where it is in the code, and a hint */
+export interface CubeExtendRowProblem {
+    readonly message: string;
+    readonly location?: CubeSourceLocation | undefined;
+    readonly hint?: string | undefined;
+}
+/** One new column in the editor */
+export interface CubeExtendRow {
+    /** Identifies the row in the editor, and its code to the engine */
+    readonly key: number;
+    readonly name: string;
+    readonly code: string;
+    readonly checked?: CubeExtendCheckedCode | undefined;
+    readonly problem?: CubeExtendRowProblem | undefined;
+    /** The type the engine gave, while the row is as it was typed */
+    readonly type?: CubeType | undefined;
+}
+/** What a new column's code starts as (PLAN §11.7 Q2) */
+export declare const NEW_EXPRESSION_CODE = "x | ";
+/**
+ * The Extend editor's draft (PLAN §11.7): a row per new column, its name and
+ * its code, `x | …`. Validate (F10) has the engine check each changed code,
+ * type the columns as a chain after the input, located so an error points
+ * into the code, and plan the query for the cube's database. Apply waits
+ * until every row's code is the one validated; the node then holds the
+ * engine's lambdas and their typing, so it needn't be typed again.
+ */
+export declare class CubeExtendDraft extends CubeNodeDraft<Extend> {
+    readonly editorState: CubeEditorState;
+    rows: readonly CubeExtendRow[];
+    /** The typing the last Validate gave, for the rows as they were then */
+    typing: ExtendTyping | undefined;
+    validating: boolean;
+    /** Anything was changed; until then, `build()` gives the original back */
+    private touched;
+    /** The database type the last plan was for, named in its problem */
+    private planDatabaseType;
+    constructor(original: Extend, editorState: CubeEditorState);
+    /** The input's schema, when the input is valid */
+    get inputSchema(): Schema | undefined;
+    /** A row named `col_<n>`, the first such name no column has, in any case */
+    private newRow;
+    /** The columns the rows build: a row's lambda only while its code is the one validated */
+    get columns(): ExtendColumn[];
+    /** Every row's code is the one the engine last checked */
+    get isValidated(): boolean;
+    /** Why Apply waits: a validation running, or a code not validated yet */
+    get applyDisabledReason(): string | undefined;
+    private change;
+    addRow(): void;
+    removeRow(key: number): void;
+    /** Moves a row up (-1) or down (+1); a later column may use it, so its typing waits */
+    moveRow(key: number, offset: number): void;
+    setName(key: number, name: string): void;
+    /** A code edit drops the row's problem and type, until the next Validate */
+    setCode(key: number, code: string): void;
+    build(): Extend;
+    /**
+     * Goes on from an Extend that differs from the original only in its
+     * typing, e.g. typed in the background after an Apply, keeping the rows;
+     * rows not validated in this draft take the new types
+     */
+    follow(node: QueryNode): boolean;
+    /**
+     * Checks every row with the engine (F10): parses each code changed since
+     * the last Validate, then, when every code parses and the input is valid,
+     * types the columns after the input, located in their codes, and plans the
+     * query up to this node for the cube's database; a plan that fails is
+     * planned again column by column to find the first that fails. Each
+     * problem goes on its row.
+     */
+    validate(): GeneratorFn<void>;
+    /** Cube's problems, each on its row; whether there was one */
+    private putCubeProblems;
+    /** The problem on its row, or on every row when the engine named none */
+    private putProblem;
+    /**
+     * Plans the query up to the typed candidate, as a run would; when it
+     * fails with several columns, the first k columns for every k, to name
+     * the column the database can't run
+     */
+    private planFrom;
+}
+//# sourceMappingURL=CubeExtendDraft.d.ts.map

package/lib/stores/editors/CubeExtendDraft.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CubeExtendDraft.d.ts","sourceRoot":"","sources":["../../../src/stores/editors/CubeExtendDraft.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,QAAQ,EACb,MAAM,EACN,KAAK,YAAY,EACjB,KAAK,YAAY,EAIjB,KAAK,UAAU,EAGf,KAAK,SAAS,EACd,KAAK,MAAM,EAEZ,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAcxD,OAAO,EAIL,KAAK,kBAAkB,EAExB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAO7D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,0EAA0E;AAC1E,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC;IAC5B,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC;CAC9B;AAED,wGAAwG;AACxG,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,QAAQ,CAAC,EAAE,kBAAkB,GAAG,SAAS,CAAC;IACnD,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC;AAED,mCAAmC;AACnC,MAAM,WAAW,aAAa;IAC5B,mEAAmE;IACnE,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,OAAO,CAAC,EAAE,qBAAqB,GAAG,SAAS,CAAC;IACrD,QAAQ,CAAC,OAAO,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;IACpD,iEAAiE;IACjE,QAAQ,CAAC,IAAI,CAAC,EAAE,QAAQ,GAAG,SAAS,CAAC;CACtC;AAED,yDAAyD;AACzD,eAAO,MAAM,mBAAmB,SAAS,CAAC;AAc1C;;;;;;;GAOG;AACH,qBAAa,eAAgB,SAAQ,aAAa,CAAC,MAAM,CAAC;IACxD,QAAQ,CAAC,WAAW,EAAE,eAAe,CAAC;IACtC,IAAI,EAAE,SAAS,aAAa,EAAE,CAAC;IAC/B,wEAAwE;IACxE,MAAM,EAAE,YAAY,GAAG,SAAS,CAAC;IACjC,UAAU,UAAS;IACnB,0EAA0E;IAC1E,OAAO,CAAC,OAAO,CAAS;IACxB,oEAAoE;IACpE,OAAO,CAAC,gBAAgB,CAAqB;gBAEjC,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe;IAiC1D,kDAAkD;IAClD,IAAI,WAAW,IAAI,MAAM,GAAG,SAAS,CAMpC;IAED,4EAA4E;IAC5E,OAAO,CAAC,MAAM;IAoBd,0FAA0F;IAC1F,IAAI,OAAO,IAAI,YAAY,EAAE,CAM5B;IAED,0DAA0D;IAC1D,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED,yEAAyE;IACzE,IAAa,mBAAmB,IAAI,MAAM,GAAG,SAAS,CAMrD;IAED,OAAO,CAAC,MAAM;IAKd,MAAM,IAAI,IAAI;IAId,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAI5B,uFAAuF;IACvF,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAY1C,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAQxC,4EAA4E;IAC5E,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAUxC,KAAK,IAAI,MAAM;IASf;;;;OAIG;IACM,MAAM,CAAC,IAAI,EAAE,SAAS,GAAG,OAAO;IAuBzC;;;;;;;OAOG;IACF,QAAQ,IAAI,WAAW,CAAC,IAAI,CAAC;IAoH9B,8DAA8D;IAC9D,OAAO,CAAC,eAAe;IAQvB,yEAAyE;IACzE,OAAO,CAAC,UAAU;IAqBlB;;;;OAIG;YACW,QAAQ;CAkFvB"}