@finos/legend-cube-builder 0.0.4 → 0.0.6

package/src/stores/editors/CubeExpressionFilterDraft.ts

@@ -0,0 +1,389 @@
+/**
+ * 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,
+  type ExpressionCheck,
+  ExpressionFilter,
+  type IR,
+  type JsonObject,
+  MESSAGE_CONDITION_NOT_BOOLEAN,
+  type Query,
+  QueryEmitter,
+  type QueryNode,
+  type Schema,
+  UNCHECKED,
+} from '@finos/legend-cube';
+import type { GeneratorFn } from '@finos/legend-shared';
+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,
+  type CubeParsedExpression,
+  type CubeSourceLocation,
+  type NodeId,
+} from '../../graph-manager/CubeEngine.js';
+import type { CubeEditorState } from '../CubeEditorState.js';
+import { CubeJsonFieldsState } from '../variant/CubeJsonFieldsState.js';
+import {
+  buildCubeConditionCheckLambdas,
+  readCubeConditionCheck,
+} from '../CubeExpressionFilterCheck.js';
+import {
+  getCubeExtendUpstream,
+  withoutEngineContext,
+} from '../CubeExtendTyping.js';
+import { CubeNodeDraft } from './CubeNodeDraft.js';
+
+/** What the engine made of the condition's code the last time it was validated */
+export interface CubeConditionCheckedCode {
+  readonly code: string;
+  readonly lambda: JsonObject;
+  readonly located: JsonObject;
+}
+
+/** The condition's problem from the last Validate: a message, where it is in the code, and a hint */
+export interface CubeConditionProblem {
+  readonly message: string;
+  readonly location?: CubeSourceLocation | undefined;
+  readonly hint?: string | undefined;
+  /**
+   * Cube refused the condition before asking the engine (a backslash, `%` or
+   * `_` on DuckDB, a pattern that isn't a literal): the panel's problems show
+   * it, so the editor doesn't say it again (PLAN §11.14)
+   */
+  readonly refused?: true;
+}
+
+/** 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: CubeEngineError): CubeConditionProblem => ({
+  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: unknown): CubeEngineError =>
+  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<ExpressionFilter> {
+  readonly editorState: CubeEditorState;
+  /** A Variant input column's fields, opened from its chip (PLAN §11.20, V.9) */
+  readonly jsonFields: CubeJsonFieldsState;
+  code: string;
+  /** What the engine made of the code the last time it parsed it */
+  checked: CubeConditionCheckedCode | undefined;
+  /** The problem from the last Validate, while the code is the one validated */
+  problem: CubeConditionProblem | undefined;
+  /** The check the last Validate gave, for the code as it was then */
+  check: ExpressionCheck | undefined;
+  validating = false;
+  /** Anything was changed; until then, `build()` gives the original back */
+  private touched = false;
+  /** The database type the last plan was for, named in its problem */
+  private planDatabaseType: string | undefined;
+
+  constructor(original: ExpressionFilter, editorState: CubeEditorState) {
+    super(original);
+    makeObservable<CubeExpressionFilterDraft, 'touched'>(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;
+    this.jsonFields = new CubeJsonFieldsState(editorState, original.id);
+    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(): Schema | undefined {
+    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(): boolean {
+    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(): boolean {
+    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 */
+  override get applyDisabledReason(): string | undefined {
+    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: string): void {
+    if (code === this.code) {
+      return;
+    }
+    this.code = code;
+    this.problem = undefined;
+    this.touched = true;
+  }
+
+  build(): ExpressionFilter {
+    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
+   */
+  override follow(node: QueryNode): boolean {
+    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(): GeneratorFn<void> {
+    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: CubeParsedExpression) => ({ code, ...expression }),
+            asEngineError,
+          )) as CubeConditionCheckedCode | CubeEngineError;
+        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,
+        ),
+      )) as ReadonlyMap<NodeId, Schema | CubeEngineError>;
+      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,
+      )) as CubeEngineError | undefined;
+      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 */
+  private async planFrom(
+    query: Query,
+    nodeId: string,
+  ): Promise<CubeEngineError | undefined> {
+    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: IR;
+    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);
+  }
+}

package/src/stores/editors/CubeExtendDraft.ts

@@ -51,6 +51,8 @@ import {
   type NodeId,
 } from '../../graph-manager/CubeEngine.js';
 import type { CubeEditorState } from '../CubeEditorState.js';
+import { CubeJsonFieldsState } from '../variant/CubeJsonFieldsState.js';
+import { getFreeCubeColumnName } from '../variant/CubeJsonOffers.js';
 import {
   buildCubeExtendTypingLambdas,
   getCubeExtendUpstream,
@@ -110,6 +112,8 @@ const problemOf = (error: CubeEngineError): CubeExtendRowProblem => ({
  */
 export class CubeExtendDraft extends CubeNodeDraft<Extend> {
   readonly editorState: CubeEditorState;
+  /** A Variant input column's fields, opened from its chip (PLAN §11.20, V.9) */
+  readonly jsonFields: CubeJsonFieldsState;
   rows: readonly CubeExtendRow[];
   /** The typing the last Validate gave, for the rows as they were then */
   typing: ExtendTyping | undefined;
@@ -131,6 +135,7 @@ export class CubeExtendDraft extends CubeNodeDraft<Extend> {
       isValidated: computed,
       applyDisabledReason: computed,
       addRow: action,
+      addColumn: action,
       removeRow: action,
       moveRow: action,
       setName: action,
@@ -139,6 +144,7 @@ export class CubeExtendDraft extends CubeNodeDraft<Extend> {
       follow: action,
     });
     this.editorState = editorState;
+    this.jsonFields = new CubeJsonFieldsState(editorState, original.id);
     const types = original.typing.kind === 'typed' ? original.typing.types : [];
     this.rows = original.columns.map(({ name, code, lambda }, index) => ({
       key: nextRowKey++,
@@ -214,6 +220,29 @@ export class CubeExtendDraft extends CubeNodeDraft<Extend> {
     this.change([...this.rows, this.newRow()]);
   }
 
+  /**
+   * Adds a row of this code, named as given or the next free name, compared
+   * as the input's and the other rows' names fold (a JSON field's "+ column",
+   * V.9); it replaces a new row still empty
+   */
+  addColumn(name: string, code: string): void {
+    const blank = this.rows.find(
+      (row) => row.code.trim() === NEW_EXPRESSION_CODE.trim() && !row.checked,
+    );
+    const others = this.rows.filter((row) => row !== blank);
+    this.change([
+      ...others,
+      {
+        key: nextRowKey++,
+        name: getFreeCubeColumnName(name, [
+          ...(this.inputSchema?.names() ?? []),
+          ...others.map((row) => row.name),
+        ]),
+        code,
+      },
+    ]);
+  }
+
   removeRow(key: number): void {
     this.change(this.rows.filter((row) => row.key !== key));
   }

package/src/stores/editors/CubeFunctionDraft.ts

@@ -0,0 +1,92 @@
+/**
+ * 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 EnteredValue,
+  FunctionSource,
+  getLiteralText,
+  type QueryNode,
+} from '@finos/legend-cube';
+import { action, computed, makeObservable, observable } from 'mobx';
+import { readCubeFunctionValue } from '../source-picker/CubeInlineModelTabState.js';
+import { CubeNodeDraft } from './CubeNodeDraft.js';
+
+const textOf = (value: EnteredValue): string =>
+  value === undefined
+    ? ''
+    : value.kind === 'invalid'
+      ? value.text
+      : getLiteralText(value);
+
+const isSameValue = (left: EnteredValue, right: EnteredValue): boolean =>
+  JSON.stringify(left) === JSON.stringify(right);
+
+/**
+ * The draft of a function source (PLAN §6.5): the text of each parameter's
+ * value, as typed. A value is read against its parameter's type, and text
+ * that isn't one is kept as an invalid value, which the source reports.
+ */
+export class CubeFunctionDraft extends CubeNodeDraft<FunctionSource> {
+  valueTexts: string[];
+
+  constructor(original: FunctionSource) {
+    super(original);
+    makeObservable(this, {
+      valueTexts: observable.ref,
+      values: computed,
+      setValueText: action,
+    });
+    this.valueTexts = original.values.map(textOf);
+  }
+
+  /** The value each text gives, against its parameter's type: none for empty text */
+  get values(): readonly EnteredValue[] {
+    return this.original.parameters.map((parameter, index) =>
+      readCubeFunctionValue(this.valueTexts[index] ?? '', parameter),
+    );
+  }
+
+  setValueText(index: number, text: string): void {
+    const texts = [...this.valueTexts];
+    texts[index] = text;
+    this.valueTexts = texts;
+  }
+
+  /** The original while every value is the one it holds */
+  build(): FunctionSource {
+    const { original, values } = this;
+    return values.every((value, index) =>
+      isSameValue(value, original.values[index]),
+    )
+      ? original
+      : original.withValues(values);
+  }
+
+  /** Goes on when the engine types the function again in the background: its values are the same */
+  override follow(node: QueryNode): boolean {
+    if (
+      node instanceof FunctionSource &&
+      node.signature === this.original.signature &&
+      node.values.every((value, index) =>
+        isSameValue(value, this.original.values[index]),
+      )
+    ) {
+      this.original = node;
+      return true;
+    }
+    return false;
+  }
+}

package/src/stores/editors/CubeNodeDraftRegistry.ts

@@ -19,8 +19,12 @@ import {
   Difference,
   Distinct,
   Drop,
+  Explode,
+  ExpressionFilter,
+  Pivot,
   Extend,
   Filter,
+  FunctionSource,
   Group,
   Join,
   Limit,
@@ -35,8 +39,12 @@ import { guaranteeType } from '@finos/legend-shared';
 import type { CubeEditorState } from '../CubeEditorState.js';
 import { CubeConcatDraft } from './CubeConcatDraft.js';
 import { CubeDifferenceDraft } from './CubeDifferenceDraft.js';
+import { CubeExplodeDraft } from './CubeExplodeDraft.js';
+import { CubeExpressionFilterDraft } from './CubeExpressionFilterDraft.js';
+import { CubePivotDraft } from './CubePivotDraft.js';
 import { CubeExtendDraft } from './CubeExtendDraft.js';
 import { CubeFilterDraft } from './CubeFilterDraft.js';
+import { CubeFunctionDraft } from './CubeFunctionDraft.js';
 import { CubeGroupDraft } from './CubeGroupDraft.js';
 import { CubeJoinDraft } from './CubeJoinDraft.js';
 import { type CubeNodeDraft, CubeReadOnlyNodeDraft } from './CubeNodeDraft.js';
@@ -55,13 +63,19 @@ export type CubeNodeDraftFactory = (
 
 /**
  * The draft of each node type the panel edits, by type (PLAN §7.4). A
- * transform with an editor in `CUBE_NODE_EDITORS` has its factory here; a
- * source, which has nothing to edit, has none, and so no Apply or Cancel.
+ * transform with an editor in `CUBE_NODE_EDITORS` has its factory here, as
+ * has a function source, whose parameter values are edited (PLAN §6.5);
+ * another source, which has nothing to edit, has none, and so no Apply or
+ * Cancel.
  */
 export const CUBE_NODE_DRAFT_FACTORIES: ReadonlyMap<
   string,
   CubeNodeDraftFactory
 > = new Map<string, CubeNodeDraftFactory>([
+  [
+    FunctionSource.TYPE,
+    (node) => new CubeFunctionDraft(guaranteeType(node, FunctionSource)),
+  ],
   [Sort.TYPE, (node) => new CubeSortDraft(guaranteeType(node, Sort))],
   [Group.TYPE, (node) => new CubeGroupDraft(guaranteeType(node, Group))],
   [Filter.TYPE, (node) => new CubeFilterDraft(guaranteeType(node, Filter))],
@@ -88,6 +102,24 @@ export const CUBE_NODE_DRAFT_FACTORIES: ReadonlyMap<
     (node, editorState) =>
       new CubeExtendDraft(guaranteeType(node, Extend), editorState),
   ],
+  [
+    ExpressionFilter.TYPE,
+    (node, editorState) =>
+      new CubeExpressionFilterDraft(
+        guaranteeType(node, ExpressionFilter),
+        editorState,
+      ),
+  ],
+  [
+    Pivot.TYPE,
+    (node, editorState) =>
+      new CubePivotDraft(guaranteeType(node, Pivot), editorState),
+  ],
+  [
+    Explode.TYPE,
+    (node, editorState) =>
+      new CubeExplodeDraft(guaranteeType(node, Explode), editorState),
+  ],
 ]);
 
 /**