@quereus/quereus 0.1.0

package/dist/src/planner/nodes/add-constraint-node.js

@@ -0,0 +1,47 @@
+import { PlanNode } from './plan-node.js';
+import { PlanNodeType } from './plan-node-type.js';
+/**
+ * Represents adding a constraint to an existing table.
+ * This is a DDL operation that modifies the table schema at runtime.
+ */
+export class AddConstraintNode extends PlanNode {
+    table;
+    constraint;
+    nodeType = PlanNodeType.AddConstraint;
+    constructor(scope, table, constraint) {
+        super(scope);
+        this.table = table;
+        this.constraint = constraint;
+    }
+    getType() {
+        return { typeClass: 'void' };
+    }
+    getRelations() {
+        return [this.table];
+    }
+    getChildren() {
+        return [];
+    }
+    withChildren(newChildren) {
+        if (newChildren.length !== 0) {
+            throw new Error(`AddConstraintNode expects 0 children, got ${newChildren.length}`);
+        }
+        return this; // No children, so no change
+    }
+    toString() {
+        const constraintType = this.constraint.type.toUpperCase();
+        const constraintName = this.constraint.name || 'unnamed';
+        return `ADD CONSTRAINT ${constraintName} ${constraintType}`;
+    }
+    getLogicalAttributes() {
+        return {
+            table: this.table.tableSchema.name,
+            schema: this.table.tableSchema.schemaName,
+            constraintType: this.constraint.type,
+            constraintName: this.constraint.name,
+            hasOperations: !!this.constraint.operations,
+            operations: this.constraint.operations,
+        };
+    }
+}
+//# sourceMappingURL=add-constraint-node.js.map

package/dist/src/planner/nodes/add-constraint-node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"add-constraint-node.js","sourceRoot":"","sources":["../../../../src/planner/nodes/add-constraint-node.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAiB,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAKnD;;;GAGG;AACH,MAAM,OAAO,iBAAkB,SAAQ,QAAQ;IAK3B;IACA;IALA,QAAQ,GAAG,YAAY,CAAC,aAAa,CAAC;IAExD,YACE,KAAY,EACI,KAAyB,EACzB,UAA+B;QAE/C,KAAK,CAAC,KAAK,CAAC,CAAC;QAHG,UAAK,GAAL,KAAK,CAAoB;QACzB,eAAU,GAAV,UAAU,CAAqB;IAGjD,CAAC;IAED,OAAO;QACL,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;IAC/B,CAAC;IAED,YAAY;QACV,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IAED,WAAW;QACT,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,YAAY,CAAC,WAAgC;QAC3C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QACrF,CAAC;QACD,OAAO,IAAI,CAAC,CAAC,4BAA4B;IAC3C,CAAC;IAEQ,QAAQ;QACf,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,SAAS,CAAC;QACzD,OAAO,kBAAkB,cAAc,IAAI,cAAc,EAAE,CAAC;IAC9D,CAAC;IAEQ,oBAAoB;QAC3B,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI;YAClC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU;YACzC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;YACpC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;YACpC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU;YAC3C,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU;SACvC,CAAC;IACJ,CAAC;CACF"}

package/dist/src/planner/nodes/aggregate-function.d.ts

@@ -0,0 +1,34 @@
+import type { ScalarType } from '../../common/datatype.js';
+import { PlanNode, type ScalarPlanNode } from './plan-node.js';
+import { PlanNodeType } from './plan-node-type.js';
+import type { Scope } from '../scopes/scope.js';
+import type { FunctionSchema } from '../../schema/function.js';
+import type * as AST from '../../parser/ast.js';
+/**
+ * Represents an aggregate function call within a SQL query.
+ * This is specifically for aggregate functions (COUNT, SUM, AVG, etc.)
+ */
+export declare class AggregateFunctionCallNode extends PlanNode implements ScalarPlanNode {
+    readonly expression: AST.FunctionExpr;
+    readonly functionName: string;
+    readonly functionSchema: FunctionSchema;
+    readonly args: ReadonlyArray<ScalarPlanNode>;
+    readonly isDistinct: boolean;
+    readonly orderBy?: ReadonlyArray<{
+        expression: ScalarPlanNode;
+        direction: "asc" | "desc";
+    }> | undefined;
+    readonly filter?: ScalarPlanNode | undefined;
+    readonly nodeType = PlanNodeType.ScalarFunctionCall;
+    constructor(scope: Scope, expression: AST.FunctionExpr, functionName: string, functionSchema: FunctionSchema, args: ReadonlyArray<ScalarPlanNode>, isDistinct?: boolean, orderBy?: ReadonlyArray<{
+        expression: ScalarPlanNode;
+        direction: "asc" | "desc";
+    }> | undefined, filter?: ScalarPlanNode | undefined);
+    getType(): ScalarType;
+    getChildren(): readonly ScalarPlanNode[];
+    getRelations(): readonly [];
+    withChildren(newChildren: readonly PlanNode[]): PlanNode;
+    toString(): string;
+    getLogicalAttributes(): Record<string, unknown>;
+}
+//# sourceMappingURL=aggregate-function.d.ts.map

package/dist/src/planner/nodes/aggregate-function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"aggregate-function.d.ts","sourceRoot":"","sources":["../../../../src/planner/nodes/aggregate-function.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,QAAQ,EAAE,KAAK,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE/D,OAAO,KAAK,KAAK,GAAG,MAAM,qBAAqB,CAAC;AAGhD;;;GAGG;AACH,qBAAa,yBAA0B,SAAQ,QAAS,YAAW,cAAc;aAK/D,UAAU,EAAE,GAAG,CAAC,YAAY;aAC5B,YAAY,EAAE,MAAM;aACpB,cAAc,EAAE,cAAc;aAC9B,IAAI,EAAE,aAAa,CAAC,cAAc,CAAC;aACnC,UAAU,EAAE,OAAO;aACnB,OAAO,CAAC,EAAE,aAAa,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,SAAS,EAAE,KAAK,GAAG,MAAM,CAAA;KAAE,CAAC;aAClF,MAAM,CAAC,EAAE,cAAc;IAVxC,QAAQ,CAAC,QAAQ,mCAAmC;gBAGnD,KAAK,EAAE,KAAK,EACI,UAAU,EAAE,GAAG,CAAC,YAAY,EAC5B,YAAY,EAAE,MAAM,EACpB,cAAc,EAAE,cAAc,EAC9B,IAAI,EAAE,aAAa,CAAC,cAAc,CAAC,EACnC,UAAU,GAAE,OAAe,EAC3B,OAAO,CAAC,EAAE,aAAa,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,SAAS,EAAE,KAAK,GAAG,MAAM,CAAA;KAAE,CAAC,YAAA,EAClF,MAAM,CAAC,EAAE,cAAc,YAAA;IAKxC,OAAO,IAAI,UAAU;IAerB,WAAW,IAAI,SAAS,cAAc,EAAE;IAWxC,YAAY,IAAI,SAAS,EAAE;IAI3B,YAAY,CAAC,WAAW,EAAE,SAAS,QAAQ,EAAE,GAAG,QAAQ;IAwD/C,QAAQ,IAAI,MAAM;IAQlB,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CAqBxD"}

package/dist/src/planner/nodes/aggregate-function.js

@@ -0,0 +1,119 @@
+import { PlanNode } from './plan-node.js';
+import { PlanNodeType } from './plan-node-type.js';
+import { isAggregateFunctionSchema } from '../../schema/function.js';
+import { formatExpressionList, formatScalarType } from '../../util/plan-formatter.js';
+/**
+ * Represents an aggregate function call within a SQL query.
+ * This is specifically for aggregate functions (COUNT, SUM, AVG, etc.)
+ */
+export class AggregateFunctionCallNode extends PlanNode {
+    expression;
+    functionName;
+    functionSchema;
+    args;
+    isDistinct;
+    orderBy;
+    filter;
+    nodeType = PlanNodeType.ScalarFunctionCall; // Using same type as scalar functions
+    constructor(scope, expression, functionName, functionSchema, args, isDistinct = false, orderBy, filter) {
+        super(scope);
+        this.expression = expression;
+        this.functionName = functionName;
+        this.functionSchema = functionSchema;
+        this.args = args;
+        this.isDistinct = isDistinct;
+        this.orderBy = orderBy;
+        this.filter = filter;
+    }
+    getType() {
+        // Get the return type from the function schema
+        if (isAggregateFunctionSchema(this.functionSchema)) {
+            return this.functionSchema.returnType;
+        }
+        // Fallback for non-aggregate functions (shouldn't happen)
+        return {
+            typeClass: 'scalar',
+            affinity: 0,
+            nullable: true, // Aggregates can return NULL
+            isReadOnly: true
+        };
+    }
+    getChildren() {
+        const children = [...this.args];
+        if (this.filter) {
+            children.push(this.filter);
+        }
+        if (this.orderBy) {
+            children.push(...this.orderBy.map(item => item.expression));
+        }
+        return children;
+    }
+    getRelations() {
+        return [];
+    }
+    withChildren(newChildren) {
+        const expectedLength = this.args.length + (this.filter ? 1 : 0) + (this.orderBy?.length || 0);
+        if (newChildren.length !== expectedLength) {
+            throw new Error(`AggregateFunctionCallNode expects ${expectedLength} children, got ${newChildren.length}`);
+        }
+        // Type check
+        for (const child of newChildren) {
+            if (!('expression' in child)) {
+                throw new Error('AggregateFunctionCallNode: all children must be ScalarPlanNodes');
+            }
+        }
+        // Split children back into their respective arrays
+        let childIndex = 0;
+        const newArgs = newChildren.slice(childIndex, childIndex + this.args.length);
+        childIndex += this.args.length;
+        let newFilter = undefined;
+        if (this.filter) {
+            newFilter = newChildren[childIndex];
+            childIndex++;
+        }
+        let newOrderBy = undefined;
+        if (this.orderBy?.length) {
+            const newOrderByExpressions = newChildren.slice(childIndex);
+            newOrderBy = this.orderBy.map((item, i) => ({
+                expression: newOrderByExpressions[i],
+                direction: item.direction
+            }));
+        }
+        // Check if anything changed
+        const argsChanged = newArgs.some((arg, i) => arg !== this.args[i]);
+        const filterChanged = newFilter !== this.filter;
+        const orderByChanged = newOrderBy && this.orderBy &&
+            newOrderBy.some((item, i) => item.expression !== this.orderBy[i].expression);
+        if (!argsChanged && !filterChanged && !orderByChanged) {
+            return this;
+        }
+        // Create new instance
+        return new AggregateFunctionCallNode(this.scope, this.expression, this.functionName, this.functionSchema, newArgs, this.isDistinct, newOrderBy, newFilter);
+    }
+    toString() {
+        const distinctStr = this.isDistinct ? 'DISTINCT ' : '';
+        const argsStr = formatExpressionList(this.args);
+        const filterStr = this.filter ? ` FILTER (WHERE ${this.filter.toString()})` : '';
+        const orderByStr = this.orderBy?.length ? ` ORDER BY ${this.orderBy.map(item => `${item.expression.toString()} ${item.direction.toUpperCase()}`).join(', ')}` : '';
+        return `${this.functionName}(${distinctStr}${argsStr})${filterStr}${orderByStr}`;
+    }
+    getLogicalAttributes() {
+        const props = {
+            function: this.functionName,
+            arguments: this.args.map(arg => arg.toString()),
+            resultType: formatScalarType(this.getType()),
+            isDistinct: this.isDistinct
+        };
+        if (this.filter) {
+            props.filter = this.filter.toString();
+        }
+        if (this.orderBy?.length) {
+            props.orderBy = this.orderBy.map(item => ({
+                expression: item.expression.toString(),
+                direction: item.direction
+            }));
+        }
+        return props;
+    }
+}
+//# sourceMappingURL=aggregate-function.js.map

package/dist/src/planner/nodes/aggregate-function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aggregate-function.js","sourceRoot":"","sources":["../../../../src/planner/nodes/aggregate-function.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAuB,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGnD,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAEtF;;;GAGG;AACH,MAAM,OAAO,yBAA0B,SAAQ,QAAQ;IAKrC;IACA;IACA;IACA;IACA;IACA;IACA;IAVR,QAAQ,GAAG,YAAY,CAAC,kBAAkB,CAAC,CAAC,sCAAsC;IAE3F,YACC,KAAY,EACI,UAA4B,EAC5B,YAAoB,EACpB,cAA8B,EAC9B,IAAmC,EACnC,aAAsB,KAAK,EAC3B,OAAkF,EAClF,MAAuB;QAEvC,KAAK,CAAC,KAAK,CAAC,CAAC;QARG,eAAU,GAAV,UAAU,CAAkB;QAC5B,iBAAY,GAAZ,YAAY,CAAQ;QACpB,mBAAc,GAAd,cAAc,CAAgB;QAC9B,SAAI,GAAJ,IAAI,CAA+B;QACnC,eAAU,GAAV,UAAU,CAAiB;QAC3B,YAAO,GAAP,OAAO,CAA2E;QAClF,WAAM,GAAN,MAAM,CAAiB;IAGxC,CAAC;IAED,OAAO;QACN,+CAA+C;QAC/C,IAAI,yBAAyB,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;YACpD,OAAO,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC;QACvC,CAAC;QAED,0DAA0D;QAC1D,OAAO;YACN,SAAS,EAAE,QAAQ;YACnB,QAAQ,EAAE,CAAC;YACX,QAAQ,EAAE,IAAI,EAAE,6BAA6B;YAC7C,UAAU,EAAE,IAAI;SAChB,CAAC;IACH,CAAC;IAED,WAAW;QACV,MAAM,QAAQ,GAAqB,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,YAAY;QACX,OAAO,EAAE,CAAC;IACX,CAAC;IAED,YAAY,CAAC,WAAgC;QAC5C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC;QAC9F,IAAI,WAAW,CAAC,MAAM,KAAK,cAAc,EAAE,CAAC;YAC3C,MAAM,IAAI,KAAK,CAAC,qCAAqC,cAAc,kBAAkB,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5G,CAAC;QAED,aAAa;QACb,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACjC,IAAI,CAAC,CAAC,YAAY,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;YACpF,CAAC;QACF,CAAC;QAED,mDAAmD;QACnD,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAqB,CAAC;QACjG,UAAU,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;QAE/B,IAAI,SAAS,GAA+B,SAAS,CAAC;QACtD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,SAAS,GAAG,WAAW,CAAC,UAAU,CAAmB,CAAC;YACtD,UAAU,EAAE,CAAC;QACd,CAAC;QAED,IAAI,UAAU,GAAyF,SAAS,CAAC;QACjH,IAAI,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC1B,MAAM,qBAAqB,GAAG,WAAW,CAAC,KAAK,CAAC,UAAU,CAAqB,CAAC;YAChF,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC3C,UAAU,EAAE,qBAAqB,CAAC,CAAC,CAAC;gBACpC,SAAS,EAAE,IAAI,CAAC,SAAS;aACzB,CAAC,CAAC,CAAC;QACL,CAAC;QAED,4BAA4B;QAC5B,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,aAAa,GAAG,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC;QAChD,MAAM,cAAc,GAAG,UAAU,IAAI,IAAI,CAAC,OAAO;YAChD,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,KAAK,IAAI,CAAC,OAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAE/E,IAAI,CAAC,WAAW,IAAI,CAAC,aAAa,IAAI,CAAC,cAAc,EAAE,CAAC;YACvD,OAAO,IAAI,CAAC;QACb,CAAC;QAED,sBAAsB;QACtB,OAAO,IAAI,yBAAyB,CACnC,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,EACnB,OAAO,EACP,IAAI,CAAC,UAAU,EACf,UAAU,EACV,SAAS,CACT,CAAC;IACH,CAAC;IAEQ,QAAQ;QAChB,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,MAAM,OAAO,GAAG,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,kBAAkB,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACjF,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnK,OAAO,GAAG,IAAI,CAAC,YAAY,IAAI,WAAW,GAAG,OAAO,IAAI,SAAS,GAAG,UAAU,EAAE,CAAC;IAClF,CAAC;IAEQ,oBAAoB;QAC5B,MAAM,KAAK,GAA4B;YACtC,QAAQ,EAAE,IAAI,CAAC,YAAY;YAC3B,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/C,UAAU,EAAE,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;SAC3B,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvC,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC1B,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACzC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE;gBACtC,SAAS,EAAE,IAAI,CAAC,SAAS;aACzB,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACd,CAAC;CACD"}

package/dist/src/planner/nodes/aggregate-node.d.ts

@@ -0,0 +1,35 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode, type RelationalPlanNode, type ScalarPlanNode, type UnaryRelationalNode, type Attribute } from './plan-node.js';
+import type { RelationType } from '../../common/datatype.js';
+import type { Scope } from '../scopes/scope.js';
+export interface AggregateExpression {
+    expression: ScalarPlanNode;
+    alias: string;
+}
+/**
+ * Represents an aggregation operation.
+ * It takes an input relation and applies aggregate functions,
+ * optionally grouping by specified expressions.
+ */
+export declare class AggregateNode extends PlanNode implements UnaryRelationalNode {
+    readonly source: RelationalPlanNode;
+    readonly groupBy: readonly ScalarPlanNode[];
+    readonly aggregates: readonly AggregateExpression[];
+    readonly preserveAttributeIds?: readonly Attribute[] | undefined;
+    readonly nodeType = PlanNodeType.Aggregate;
+    private outputTypeCache;
+    private attributesCache;
+    constructor(scope: Scope, source: RelationalPlanNode, groupBy: readonly ScalarPlanNode[], aggregates: readonly AggregateExpression[], estimatedCostOverride?: number, preserveAttributeIds?: readonly Attribute[] | undefined);
+    private getGroupByColumnName;
+    private buildOutputType;
+    private buildAttributes;
+    getType(): RelationType;
+    getAttributes(): Attribute[];
+    getChildren(): readonly PlanNode[];
+    getRelations(): readonly [RelationalPlanNode];
+    withChildren(newChildren: readonly PlanNode[]): PlanNode;
+    get estimatedRows(): number | undefined;
+    toString(): string;
+    getLogicalAttributes(): Record<string, unknown>;
+}
+//# sourceMappingURL=aggregate-node.d.ts.map

package/dist/src/planner/nodes/aggregate-node.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"aggregate-node.d.ts","sourceRoot":"","sources":["../../../../src/planner/nodes/aggregate-node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,KAAK,kBAAkB,EAAE,KAAK,cAAc,EAAE,KAAK,mBAAmB,EAAE,KAAK,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAClI,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAMhD,MAAM,WAAW,mBAAmB;IAClC,UAAU,EAAE,cAAc,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;CACf;AAED;;;;GAIG;AACH,qBAAa,aAAc,SAAQ,QAAS,YAAW,mBAAmB;aAQtD,MAAM,EAAE,kBAAkB;aAC1B,OAAO,EAAE,SAAS,cAAc,EAAE;aAClC,UAAU,EAAE,SAAS,mBAAmB,EAAE;aAE1C,oBAAoB,CAAC,EAAE,SAAS,SAAS,EAAE;IAX7D,SAAkB,QAAQ,0BAA0B;IAEpD,OAAO,CAAC,eAAe,CAAuB;IAC9C,OAAO,CAAC,eAAe,CAAsB;gBAG3C,KAAK,EAAE,KAAK,EACI,MAAM,EAAE,kBAAkB,EAC1B,OAAO,EAAE,SAAS,cAAc,EAAE,EAClC,UAAU,EAAE,SAAS,mBAAmB,EAAE,EAC1D,qBAAqB,CAAC,EAAE,MAAM,EACd,oBAAoB,CAAC,EAAE,SAAS,SAAS,EAAE,YAAA;IAS7D,OAAO,CAAC,oBAAoB;IAU5B,OAAO,CAAC,eAAe;IAgCvB,OAAO,CAAC,eAAe;IAgCvB,OAAO,IAAI,YAAY;IAIvB,aAAa,IAAI,SAAS,EAAE;IAI5B,WAAW,IAAI,SAAS,QAAQ,EAAE;IAIlC,YAAY,IAAI,SAAS,CAAC,kBAAkB,CAAC;IAI7C,YAAY,CAAC,WAAW,EAAE,SAAS,QAAQ,EAAE,GAAG,QAAQ;IAyCxD,IAAI,aAAa,IAAI,MAAM,GAAG,SAAS,CAatC;IAEQ,QAAQ,IAAI,MAAM;IAiBlB,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CAgBzD"}

package/dist/src/planner/nodes/aggregate-node.js

@@ -0,0 +1,176 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode } from './plan-node.js';
+import { Cached } from '../../util/cached.js';
+import { formatExpressionList } from '../../util/plan-formatter.js';
+import { StatusCode } from '../../common/types.js';
+import { quereusError } from '../../common/errors.js';
+/**
+ * Represents an aggregation operation.
+ * It takes an input relation and applies aggregate functions,
+ * optionally grouping by specified expressions.
+ */
+export class AggregateNode extends PlanNode {
+    source;
+    groupBy;
+    aggregates;
+    preserveAttributeIds;
+    nodeType = PlanNodeType.Aggregate;
+    outputTypeCache;
+    attributesCache;
+    constructor(scope, source, groupBy, aggregates, estimatedCostOverride, preserveAttributeIds) {
+        super(scope, estimatedCostOverride ?? source.getTotalCost());
+        this.source = source;
+        this.groupBy = groupBy;
+        this.aggregates = aggregates;
+        this.preserveAttributeIds = preserveAttributeIds;
+        this.outputTypeCache = new Cached(() => this.buildOutputType());
+        this.attributesCache = new Cached(() => this.buildAttributes());
+    }
+    // Helper function to extract a meaningful name from a GROUP BY expression
+    getGroupByColumnName(expr, index) {
+        // If it's a column reference, use the column name
+        if (expr.nodeType === PlanNodeType.ColumnReference) {
+            const colRef = expr; // ColumnReferenceNode
+            return colRef.expression.name;
+        }
+        // Otherwise, use a generic name
+        return `group_${index}`;
+    }
+    buildOutputType() {
+        // Build the output relation type based on group by columns and aggregates
+        const columns = [
+            // Group by columns come first
+            ...this.groupBy.map((expr, index) => ({
+                name: this.getGroupByColumnName(expr, index),
+                type: expr.getType(),
+                generated: false
+            })),
+            // Then aggregate columns
+            ...this.aggregates.map(agg => ({
+                name: agg.alias,
+                type: agg.expression.getType(),
+                generated: true
+            }))
+        ];
+        // Determine if result is a set
+        // - Without GROUP BY: always produces exactly 1 row, so it's a set
+        // - With GROUP BY: produces one row per unique group, so it's a set
+        const isSet = true;
+        return {
+            typeClass: 'relation',
+            columns,
+            keys: [], // No keys for aggregate results
+            rowConstraints: [], // No row constraints for aggregate results
+            isReadOnly: true,
+            isSet
+        };
+    }
+    buildAttributes() {
+        // If we have preserved attribute IDs, use them
+        if (this.preserveAttributeIds) {
+            return this.preserveAttributeIds.slice(); // Return a copy
+        }
+        const attributes = [];
+        // Group by columns come first
+        this.groupBy.forEach((expr, index) => {
+            const name = this.getGroupByColumnName(expr, index);
+            attributes.push({
+                id: PlanNode.nextAttrId(),
+                name,
+                type: expr.getType(),
+                sourceRelation: `${this.nodeType}:${this.id}`
+            });
+        });
+        // Then aggregate columns
+        this.aggregates.forEach((agg) => {
+            attributes.push({
+                id: PlanNode.nextAttrId(),
+                name: agg.alias,
+                type: agg.expression.getType(),
+                sourceRelation: `${this.nodeType}:${this.id}`
+            });
+        });
+        return attributes;
+    }
+    getType() {
+        return this.outputTypeCache.value;
+    }
+    getAttributes() {
+        return this.attributesCache.value;
+    }
+    getChildren() {
+        return [this.source, ...this.groupBy, ...this.aggregates.map(agg => agg.expression)];
+    }
+    getRelations() {
+        return [this.source];
+    }
+    withChildren(newChildren) {
+        const expectedLength = 1 + this.groupBy.length + this.aggregates.length;
+        if (newChildren.length !== expectedLength) {
+            quereusError(`AggregateNode expects ${expectedLength} children, got ${newChildren.length}`, StatusCode.INTERNAL);
+        }
+        const [newSource, ...restChildren] = newChildren;
+        const newGroupBy = restChildren.slice(0, this.groupBy.length);
+        const newAggregateExpressions = restChildren.slice(this.groupBy.length);
+        // Type check
+        if (!('getAttributes' in newSource) || typeof newSource.getAttributes !== 'function') {
+            quereusError('AggregateNode: first child must be a RelationalPlanNode', StatusCode.INTERNAL);
+        }
+        // Check if anything changed
+        const sourceChanged = newSource !== this.source;
+        const groupByChanged = newGroupBy.some((expr, i) => expr !== this.groupBy[i]);
+        const aggregatesChanged = newAggregateExpressions.some((expr, i) => expr !== this.aggregates[i].expression);
+        if (!sourceChanged && !groupByChanged && !aggregatesChanged) {
+            return this;
+        }
+        // Build new aggregates array
+        const newAggregates = newAggregateExpressions.map((expr, i) => ({
+            expression: expr,
+            alias: this.aggregates[i].alias
+        }));
+        // Create new instance that preserves original attribute IDs
+        return new AggregateNode(this.scope, newSource, newGroupBy, newAggregates, undefined, // estimatedCostOverride
+        this.getAttributes() // Preserve original attribute IDs
+        );
+    }
+    get estimatedRows() {
+        const sourceRows = this.source.estimatedRows;
+        if (sourceRows === undefined)
+            return undefined;
+        // If we have GROUP BY, the output rows depend on the number of distinct groups
+        // For now, we'll use a conservative estimate
+        if (this.groupBy.length > 0) {
+            // Estimate that we'll have at most sourceRows/2 groups, but at least 1
+            return Math.max(1, Math.floor(sourceRows / 2));
+        }
+        else {
+            // No GROUP BY means we're aggregating the entire table into a single row
+            return 1;
+        }
+    }
+    toString() {
+        const parts = [];
+        if (this.groupBy.length > 0) {
+            parts.push(`GROUP BY ${formatExpressionList(this.groupBy)}`);
+        }
+        if (this.aggregates.length > 0) {
+            const aggregatesStr = this.aggregates.map(agg => `${agg.expression.toString()} AS ${agg.alias}`).join(', ');
+            parts.push(`AGG ${aggregatesStr}`);
+        }
+        return parts.join('  ');
+    }
+    getLogicalAttributes() {
+        const props = {};
+        if (this.groupBy.length > 0) {
+            props.groupBy = this.groupBy.map(expr => expr.toString());
+        }
+        if (this.aggregates.length > 0) {
+            props.aggregates = this.aggregates.map(agg => ({
+                expression: agg.expression.toString(),
+                alias: agg.alias
+            }));
+        }
+        return props;
+    }
+}
+//# sourceMappingURL=aggregate-node.js.map

package/dist/src/planner/nodes/aggregate-node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aggregate-node.js","sourceRoot":"","sources":["../../../../src/planner/nodes/aggregate-node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAA0F,MAAM,gBAAgB,CAAC;AAGlI,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAOtD;;;;GAIG;AACH,MAAM,OAAO,aAAc,SAAQ,QAAQ;IAQvB;IACA;IACA;IAEA;IAXA,QAAQ,GAAG,YAAY,CAAC,SAAS,CAAC;IAE5C,eAAe,CAAuB;IACtC,eAAe,CAAsB;IAE7C,YACE,KAAY,EACI,MAA0B,EAC1B,OAAkC,EAClC,UAA0C,EAC1D,qBAA8B,EACd,oBAA2C;QAE3D,KAAK,CAAC,KAAK,EAAE,qBAAqB,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QAN7C,WAAM,GAAN,MAAM,CAAoB;QAC1B,YAAO,GAAP,OAAO,CAA2B;QAClC,eAAU,GAAV,UAAU,CAAgC;QAE1C,yBAAoB,GAApB,oBAAoB,CAAuB;QAI3D,IAAI,CAAC,eAAe,GAAG,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;QAChE,IAAI,CAAC,eAAe,GAAG,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,0EAA0E;IAClE,oBAAoB,CAAC,IAAoB,EAAE,KAAa;QAC9D,kDAAkD;QAClD,IAAI,IAAI,CAAC,QAAQ,KAAK,YAAY,CAAC,eAAe,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,IAAW,CAAC,CAAC,sBAAsB;YAClD,OAAO,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;QAChC,CAAC;QACD,gCAAgC;QAChC,OAAO,SAAS,KAAK,EAAE,CAAC;IAC1B,CAAC;IAEO,eAAe;QACrB,0EAA0E;QAC1E,MAAM,OAAO,GAAG;YACd,8BAA8B;YAC9B,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC;gBAC5C,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE;gBACpB,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;YACH,yBAAyB;YACzB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7B,IAAI,EAAE,GAAG,CAAC,KAAK;gBACf,IAAI,EAAE,GAAG,CAAC,UAAU,CAAC,OAAO,EAAE;gBAC9B,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;SACJ,CAAC;QAEF,+BAA+B;QAC/B,mEAAmE;QACnE,oEAAoE;QACpE,MAAM,KAAK,GAAG,IAAI,CAAC;QAEnB,OAAO;YACL,SAAS,EAAE,UAAU;YACrB,OAAO;YACP,IAAI,EAAE,EAAE,EAAE,gCAAgC;YAC1C,cAAc,EAAE,EAAE,EAAE,2CAA2C;YAC/D,UAAU,EAAE,IAAI;YAChB,KAAK;SACN,CAAC;IACJ,CAAC;IAEO,eAAe;QACrB,+CAA+C;QAC/C,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC,CAAC,gBAAgB;QAC5D,CAAC;QAED,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,8BAA8B;QAC9B,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACpD,UAAU,CAAC,IAAI,CAAC;gBACd,EAAE,EAAE,QAAQ,CAAC,UAAU,EAAE;gBACzB,IAAI;gBACJ,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE;gBACpB,cAAc,EAAE,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,EAAE,EAAE;aAC9C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,yBAAyB;QACzB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9B,UAAU,CAAC,IAAI,CAAC;gBACd,EAAE,EAAE,QAAQ,CAAC,UAAU,EAAE;gBACzB,IAAI,EAAE,GAAG,CAAC,KAAK;gBACf,IAAI,EAAE,GAAG,CAAC,UAAU,CAAC,OAAO,EAAE;gBAC9B,cAAc,EAAE,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,EAAE,EAAE;aAC9C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;IACpC,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;IACpC,CAAC;IAED,WAAW;QACT,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IACvF,CAAC;IAED,YAAY;QACV,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC;IAED,YAAY,CAAC,WAAgC;QAC3C,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;QACxE,IAAI,WAAW,CAAC,MAAM,KAAK,cAAc,EAAE,CAAC;YAC1C,YAAY,CAAC,yBAAyB,cAAc,kBAAkB,WAAW,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnH,CAAC;QAED,MAAM,CAAC,SAAS,EAAE,GAAG,YAAY,CAAC,GAAG,WAAW,CAAC;QACjD,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9D,MAAM,uBAAuB,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAExE,aAAa;QACb,IAAI,CAAC,CAAC,eAAe,IAAI,SAAS,CAAC,IAAI,OAAQ,SAAiB,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;YAC9F,YAAY,CAAC,yDAAyD,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC/F,CAAC;QAED,4BAA4B;QAC5B,MAAM,aAAa,GAAG,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC;QAChD,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9E,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAE5G,IAAI,CAAC,aAAa,IAAI,CAAC,cAAc,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5D,OAAO,IAAI,CAAC;QACd,CAAC;QAED,6BAA6B;QAC7B,MAAM,aAAa,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9D,UAAU,EAAE,IAAsB;YAClC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK;SAChC,CAAC,CAAC,CAAC;QAEJ,4DAA4D;QAC5D,OAAO,IAAI,aAAa,CACtB,IAAI,CAAC,KAAK,EACV,SAA+B,EAC/B,UAA8B,EAC9B,aAAa,EACb,SAAS,EAAE,wBAAwB;QACnC,IAAI,CAAC,aAAa,EAAE,CAAC,kCAAkC;SACxD,CAAC;IACJ,CAAC;IAED,IAAI,aAAa;QACf,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;QAC7C,IAAI,UAAU,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAE/C,+EAA+E;QAC/E,6CAA6C;QAC7C,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,uEAAuE;YACvE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;QACjD,CAAC;aAAM,CAAC;YACN,yEAAyE;YACzE,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAEQ,QAAQ;QACf,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,YAAY,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAC9C,GAAG,GAAG,CAAC,UAAU,CAAC,QAAQ,EAAE,OAAO,GAAG,CAAC,KAAK,EAAE,CAC/C,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,KAAK,CAAC,IAAI,CAAC,OAAO,aAAa,EAAE,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAEQ,oBAAoB;QAC3B,MAAM,KAAK,GAA4B,EAAE,CAAC;QAE1C,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7C,UAAU,EAAE,GAAG,CAAC,UAAU,CAAC,QAAQ,EAAE;gBACrC,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/src/planner/nodes/array-index-node.d.ts

@@ -0,0 +1,22 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode, type ZeroAryScalarNode } from './plan-node.js';
+import type { ScalarType } from '../../common/datatype.js';
+import type { Scope } from '../scopes/scope.js';
+import type * as AST from '../../parser/ast.js';
+/**
+ * Represents direct access to a value in a row by array index.
+ * This is used when we know the exact position of a value in the output row.
+ */
+export declare class ArrayIndexNode extends PlanNode implements ZeroAryScalarNode {
+    readonly index: number;
+    readonly type: ScalarType;
+    readonly nodeType = PlanNodeType.ArrayIndex;
+    readonly expression: AST.Expression;
+    constructor(scope: Scope, index: number, type: ScalarType);
+    getType(): ScalarType;
+    getChildren(): readonly [];
+    getRelations(): readonly [];
+    withChildren(newChildren: readonly PlanNode[]): PlanNode;
+    toString(): string;
+}
+//# sourceMappingURL=array-index-node.d.ts.map

package/dist/src/planner/nodes/array-index-node.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"array-index-node.d.ts","sourceRoot":"","sources":["../../../../src/planner/nodes/array-index-node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,KAAK,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAClE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,KAAK,KAAK,GAAG,MAAM,qBAAqB,CAAC;AAEhD;;;GAGG;AACH,qBAAa,cAAe,SAAQ,QAAS,YAAW,iBAAiB;aAMvD,KAAK,EAAE,MAAM;aACb,IAAI,EAAE,UAAU;IANjC,SAAkB,QAAQ,2BAA2B;IACrD,SAAgB,UAAU,EAAE,GAAG,CAAC,UAAU,CAAC;gBAG1C,KAAK,EAAE,KAAK,EACI,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,UAAU;IAUjC,OAAO,IAAI,UAAU;IAIrB,WAAW,IAAI,SAAS,EAAE;IAI1B,YAAY,IAAI,SAAS,EAAE;IAI3B,YAAY,CAAC,WAAW,EAAE,SAAS,QAAQ,EAAE,GAAG,QAAQ;IAO/C,QAAQ,IAAI,MAAM;CAG3B"}

package/dist/src/planner/nodes/array-index-node.js

@@ -0,0 +1,41 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode } from './plan-node.js';
+/**
+ * Represents direct access to a value in a row by array index.
+ * This is used when we know the exact position of a value in the output row.
+ */
+export class ArrayIndexNode extends PlanNode {
+    index;
+    type;
+    nodeType = PlanNodeType.ArrayIndex;
+    expression;
+    constructor(scope, index, type) {
+        super(scope);
+        this.index = index;
+        this.type = type;
+        // Create a synthetic expression for this array access
+        this.expression = {
+            type: 'literal',
+            value: `[${index}]`
+        };
+    }
+    getType() {
+        return this.type;
+    }
+    getChildren() {
+        return [];
+    }
+    getRelations() {
+        return [];
+    }
+    withChildren(newChildren) {
+        if (newChildren.length !== 0) {
+            throw new Error(`ArrayIndexNode expects 0 children, got ${newChildren.length}`);
+        }
+        return this; // No children, so no change
+    }
+    toString() {
+        return `[${this.index}]`;
+    }
+}
+//# sourceMappingURL=array-index-node.js.map

package/dist/src/planner/nodes/array-index-node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"array-index-node.js","sourceRoot":"","sources":["../../../../src/planner/nodes/array-index-node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAA0B,MAAM,gBAAgB,CAAC;AAKlE;;;GAGG;AACH,MAAM,OAAO,cAAe,SAAQ,QAAQ;IAM1B;IACA;IANC,QAAQ,GAAG,YAAY,CAAC,UAAU,CAAC;IACrC,UAAU,CAAiB;IAE3C,YACC,KAAY,EACI,KAAa,EACb,IAAgB;QAEhC,KAAK,CAAC,KAAK,CAAC,CAAC;QAHG,UAAK,GAAL,KAAK,CAAQ;QACb,SAAI,GAAJ,IAAI,CAAY;QAGhC,sDAAsD;QACtD,IAAI,CAAC,UAAU,GAAG;YACjB,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,IAAI,KAAK,GAAG;SACA,CAAC;IACtB,CAAC;IAED,OAAO;QACN,OAAO,IAAI,CAAC,IAAI,CAAC;IAClB,CAAC;IAED,WAAW;QACV,OAAO,EAAE,CAAC;IACX,CAAC;IAED,YAAY;QACX,OAAO,EAAE,CAAC;IACX,CAAC;IAED,YAAY,CAAC,WAAgC;QAC5C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,0CAA0C,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QACjF,CAAC;QACD,OAAO,IAAI,CAAC,CAAC,4BAA4B;IAC1C,CAAC;IAEQ,QAAQ;QAChB,OAAO,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC;IAC1B,CAAC;CACD"}

package/dist/src/planner/nodes/block.d.ts

@@ -0,0 +1,26 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode, type RelationalPlanNode } from './plan-node.js';
+import type { BaseType } from '../../common/datatype.js';
+import type { Scope } from '../scopes/scope.js';
+import type { SqlParameters } from '../../common/types.js';
+/**
+ * Represents a block of one or more statements to be executed in sequence.
+ * This is the root of most SQL query plans.
+ */
+export declare class BlockNode extends PlanNode {
+    readonly statements: PlanNode[];
+    /** Snapshot of parameters utilized by the block. */
+    readonly parameters: SqlParameters;
+    readonly nodeType = PlanNodeType.Block;
+    constructor(scope: Scope, statements: PlanNode[], 
+    /** Snapshot of parameters utilized by the block. */
+    parameters: SqlParameters, estimatedCostOverride?: number);
+    getType(): BaseType;
+    getChildren(): readonly PlanNode[];
+    getRelations(): readonly RelationalPlanNode[];
+    toString(): string;
+    getLogicalAttributes(): Record<string, unknown>;
+    get estimatedRows(): number | undefined;
+    withChildren(newChildren: readonly PlanNode[]): PlanNode;
+}
+//# sourceMappingURL=block.d.ts.map

package/dist/src/planner/nodes/block.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"block.d.ts","sourceRoot":"","sources":["../../../../src/planner/nodes/block.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,KAAK,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACnE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAE3D;;;GAGG;AACH,qBAAa,SAAU,SAAQ,QAAQ;aAKnB,UAAU,EAAE,QAAQ,EAAE;IACtC,oDAAoD;aACpC,UAAU,EAAE,aAAa;IAN3C,SAAkB,QAAQ,sBAAsB;gBAG9C,KAAK,EAAE,KAAK,EACI,UAAU,EAAE,QAAQ,EAAE;IACtC,oDAAoD;IACpC,UAAU,EAAE,aAAa,EACzC,qBAAqB,CAAC,EAAE,MAAM;IAOhC,OAAO,IAAI,QAAQ;IAKnB,WAAW,IAAI,SAAS,QAAQ,EAAE;IAIlC,YAAY,IAAI,SAAS,kBAAkB,EAAE;IAIpC,QAAQ,IAAI,MAAM;IAIlB,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAQxD,IAAI,aAAa,IAAI,MAAM,GAAG,SAAS,CAEtC;IAED,YAAY,CAAC,WAAW,EAAE,SAAS,QAAQ,EAAE,GAAG,QAAQ;CAyBzD"}

package/dist/src/planner/nodes/block.js

@@ -0,0 +1,58 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode } from './plan-node.js';
+/**
+ * Represents a block of one or more statements to be executed in sequence.
+ * This is the root of most SQL query plans.
+ */
+export class BlockNode extends PlanNode {
+    statements;
+    parameters;
+    nodeType = PlanNodeType.Block;
+    constructor(scope, statements, 
+    /** Snapshot of parameters utilized by the block. */
+    parameters, estimatedCostOverride) {
+        // Cost: sum of all statement costs
+        const totalCost = statements.reduce((sum, stmt) => sum + stmt.getTotalCost(), 0);
+        super(scope, estimatedCostOverride ?? totalCost);
+        this.statements = statements;
+        this.parameters = parameters;
+    }
+    getType() {
+        // A block doesn't have a well-defined type; it's context-dependent
+        return { typeClass: 'void' };
+    }
+    getChildren() {
+        return this.statements;
+    }
+    getRelations() {
+        return this.statements.filter(s => s.getType().typeClass === "relation");
+    }
+    toString() {
+        return `${this.statements.length} statements`;
+    }
+    getLogicalAttributes() {
+        return {
+            numStatements: this.statements.length,
+            statementTypes: this.statements.map(stmt => stmt.nodeType),
+            parameters: Object.keys(this.parameters)
+        };
+    }
+    get estimatedRows() {
+        return this.getRelations().reduce((acc, s) => acc + (s.estimatedRows ?? 0), 0);
+    }
+    withChildren(newChildren) {
+        // Check if statements changed
+        if (newChildren.length !== this.statements.length) {
+            // Create new instance with new statements
+            return new BlockNode(this.scope, [...newChildren], this.parameters);
+        }
+        // Check if any individual statement changed
+        const statementsChanged = newChildren.some((child, i) => child !== this.statements[i]);
+        if (!statementsChanged) {
+            return this;
+        }
+        // Create new instance with updated statements
+        return new BlockNode(this.scope, [...newChildren], this.parameters);
+    }
+}
+//# sourceMappingURL=block.js.map

package/dist/src/planner/nodes/block.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"block.js","sourceRoot":"","sources":["../../../../src/planner/nodes/block.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAA2B,MAAM,gBAAgB,CAAC;AAKnE;;;GAGG;AACH,MAAM,OAAO,SAAU,SAAQ,QAAQ;IAKnB;IAEA;IANA,QAAQ,GAAG,YAAY,CAAC,KAAK,CAAC;IAEhD,YACE,KAAY,EACI,UAAsB;IACtC,oDAAoD;IACpC,UAAyB,EACzC,qBAA8B;QAE9B,mCAAmC;QACnC,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC;QACjF,KAAK,CAAC,KAAK,EAAE,qBAAqB,IAAI,SAAS,CAAC,CAAC;QAPjC,eAAU,GAAV,UAAU,CAAY;QAEtB,eAAU,GAAV,UAAU,CAAe;IAM3C,CAAC;IAED,OAAO;QACL,mEAAmE;QACnE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;IAC/B,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,SAAS,KAAK,UAAU,CAA6C,CAAC;IACvH,CAAC;IAEQ,QAAQ;QACf,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,aAAa,CAAC;IAChD,CAAC;IAEQ,oBAAoB;QAC3B,OAAO;YACL,aAAa,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM;YACrC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC1D,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;SACzC,CAAC;IACJ,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACjF,CAAC;IAED,YAAY,CAAC,WAAgC;QAC3C,8BAA8B;QAC9B,IAAI,WAAW,CAAC,MAAM,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAClD,0CAA0C;YAC1C,OAAO,IAAI,SAAS,CAClB,IAAI,CAAC,KAAK,EACV,CAAC,GAAG,WAAW,CAAC,EAChB,IAAI,CAAC,UAAU,CAChB,CAAC;QACJ,CAAC;QAED,4CAA4C;QAC5C,MAAM,iBAAiB,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAEvF,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,8CAA8C;QAC9C,OAAO,IAAI,SAAS,CAClB,IAAI,CAAC,KAAK,EACV,CAAC,GAAG,WAAW,CAAC,EAChB,IAAI,CAAC,UAAU,CAChB,CAAC;IACJ,CAAC;CACF"}

package/dist/src/planner/nodes/cache-node.d.ts

@@ -0,0 +1,29 @@
+import { PlanNodeType } from './plan-node-type.js';
+import { PlanNode, type RelationalPlanNode, type UnaryRelationalNode, type Attribute } from './plan-node.js';
+import type { RelationType } from '../../common/datatype.js';
+import type { Scope } from '../scopes/scope.js';
+export type CacheStrategy = 'memory' | 'spill';
+/**
+ * CacheNode provides smart caching for any relational input.
+ *
+ * This node materializes its input on first iteration and serves
+ * subsequent iterations from the cached result. It implements
+ * smart threshold-based policies to avoid excessive memory usage.
+ */
+export declare class CacheNode extends PlanNode implements UnaryRelationalNode {
+    readonly source: RelationalPlanNode;
+    readonly strategy: CacheStrategy;
+    readonly threshold: number;
+    readonly nodeType = PlanNodeType.Cache;
+    constructor(scope: Scope, source: RelationalPlanNode, strategy?: CacheStrategy, threshold?: number, // Rows before switching to pass-through
+    estimatedCostOverride?: number);
+    getAttributes(): Attribute[];
+    getType(): RelationType;
+    getChildren(): readonly [RelationalPlanNode];
+    getRelations(): readonly [RelationalPlanNode];
+    withChildren(newChildren: readonly PlanNode[]): PlanNode;
+    get estimatedRows(): number | undefined;
+    toString(): string;
+    getLogicalAttributes(): Record<string, unknown>;
+}
+//# sourceMappingURL=cache-node.d.ts.map