@quereus/quereus 0.4.11 → 0.5.1

package/src/schema/table.ts

@@ -1,398 +1,411 @@
-import type { ColumnSchema } from './column.js';
-import type { AnyVirtualTableModule } from '../vtab/module.js';
-import { MemoryTableModule } from '../vtab/memory/module.js';
-import type { Expression } from '../parser/ast.js';
-import { type ColumnDef, type TableConstraint } from '../parser/ast.js';
-import { getAffinity } from '../common/type-inference.js';
-import { RowOp, SqlDataType, StatusCode, type SqlValue } from '../common/types.js';
-import type * as AST from '../parser/ast.js';
-import { quereusError, QuereusError } from '../common/errors.js';
-import { createLogger } from '../common/logger.js';
-
-const log = createLogger('schema:table');
-const warnLog = log.extend('warn');
-
-/**
- * Represents the schema definition of a table (real or virtual).
- */
-export interface TableSchema {
-	/** Table name */
-	name: string;
-	/** Schema name (e.g., "main", "temp") */
-	schemaName: string;
-	/** Ordered list of column definitions */
-	columns: ReadonlyArray<ColumnSchema>;
-	/** Map from column name (lowercase) to column index */
-	columnIndexMap: ReadonlyMap<string, number>;
-	/** Definition of the primary key, including order and direction */
-	primaryKeyDefinition: ReadonlyArray<PrimaryKeyColumnDefinition>;
-	/** CHECK constraints defined on the table or its columns */
-	checkConstraints: ReadonlyArray<RowConstraintSchema>;
-	/** Reference to the registered module */
-	vtabModule: AnyVirtualTableModule;
-	/** If virtual, aux data passed during module registration */
-	vtabAuxData?: unknown;
-	/** If virtual, the arguments passed in CREATE VIRTUAL TABLE */
-	vtabArgs?: Record<string, SqlValue>;
-	/** If virtual, the name the module was registered with */
-	vtabModuleName: string;
-	/** Whether the table is a temporary table */
-	isTemporary?: boolean;
-	/** Whether the table is a view */
-	isView: boolean;
-	/** Whether the table is a subquery source */
-	subqueryAST?: AST.SelectStmt;
-	/** If virtual, the view definition */
-	viewDefinition?: AST.SelectStmt;
-	/** Table-level constraints */
-	tableConstraints?: readonly TableConstraint[];
-	/** Definitions of secondary indexes (relevant for planning) */
-	indexes?: ReadonlyArray<IndexSchema>;
-	/** Estimated number of rows in the table (for query planning) */
-	readonly estimatedRows?: number;
-	/** Whether the table is read-only */
-	isReadOnly?: boolean;	// default false
-	/** Mutation context variables for this table */
-	mutationContext?: ReadonlyArray<MutationContextDefinition>;
-	/** Foreign key constraints (parsed but not yet enforced by engine) */
-	// foreignKeys?: ReadonlyArray<ForeignKeyConstraintSchema>;
-	/** Unique constraints (beyond primary key) */
-	// uniqueConstraints?: ReadonlyArray<ConstraintSchema>;
-}
-
-/**
- * Builds a map from column names to their indices in the columns array
- *
- * @param columns Array of column schemas
- * @returns Map of lowercase column names to their indices
- */
-export function buildColumnIndexMap(columns: ReadonlyArray<ColumnSchema>): Map<string, number> {
-	const map = new Map<string, number>();
-	columns.forEach((col, index) => {
-		map.set(col.name.toLowerCase(), index);
-	});
-	return map;
-}
-
-/**
- * Extracts just the column indices from a primary key definition
- *
- * @param pkDef Primary key definition array
- * @returns Array of column indices that form the primary key
- */
-export function getPrimaryKeyIndices(pkDef: ReadonlyArray<PrimaryKeyColumnDefinition>): ReadonlyArray<number> {
-	return Object.freeze(pkDef.map(def => def.index));
-}
-
-/**
- * Converts a parsed ColumnDef AST node into a runtime ColumnSchema object
- *
- * @param def Column definition AST node
- * @param defaultNotNull Whether columns should be NOT NULL by default (Third Manifesto approach)
- * @returns A runtime ColumnSchema object
- */
-export function columnDefToSchema(def: ColumnDef, defaultNotNull: boolean = true): ColumnSchema {
-	const schema: Partial<ColumnSchema> & { name: string } = {
-		name: def.name,
-		affinity: getAffinity(def.dataType),
-		notNull: defaultNotNull, // Default based on Third Manifesto principles
-		primaryKey: false,
-		pkOrder: 0,
-		defaultValue: null,
-		collation: 'BINARY',
-		generated: false,
-	};
-
-	for (const constraint of def.constraints ?? []) {
-		switch (constraint.type) {
-			case 'primaryKey':
-				schema.primaryKey = true;
-				schema.pkDirection = constraint.direction;
-				break;
-			case 'notNull':
-				schema.notNull = true;
-				break;
-			case 'null':
-				schema.notNull = false;
-				break;
-			case 'unique':
-				break;
-			case 'default':
-				schema.defaultValue = constraint.expr;
-				break;
-			case 'collate':
-				schema.collation = constraint.collation;
-				break;
-			case 'generated':
-				schema.generated = true;
-				break;
-		}
-	}
-
-	// PK implies NOT NULL (always, regardless of default)
-	if (schema.primaryKey) {
-		schema.notNull = true;
-	}
-
-	// If no explicit nullability constraint and default is nullable,
-	// we need to check if there's an explicit NULL declaration
-	// Note: SQL doesn't have explicit NULL constraints in standard syntax,
-	// so this primarily affects the default behavior
-
-	// Assign a default pkOrder if it's a PK but order isn't specified elsewhere
-	if (schema.primaryKey && schema.pkOrder === 0) {
-		schema.pkOrder = 1;
-	}
-
-	return schema as ColumnSchema;
-}
-
-/**
- * Mutation context variable definition
- */
-export interface MutationContextDefinition {
-	/** Variable name */
-	name: string;
-	/** Type affinity of the variable */
-	affinity: SqlDataType;
-	/** Whether the variable is NOT NULL */
-	notNull: boolean;
-}
-
-/**
- * Converts AST mutation context variable to schema definition
- *
- * @param varDef AST mutation context variable definition
- * @param defaultNotNull Whether variables should be NOT NULL by default
- * @returns Mutation context definition schema object
- */
-export function mutationContextVarToSchema(varDef: AST.MutationContextVar, defaultNotNull: boolean = true): MutationContextDefinition {
-	return {
-		name: varDef.name,
-		affinity: getAffinity(varDef.dataType),
-		notNull: varDef.notNull !== undefined ? varDef.notNull : defaultNotNull,
-	};
-}
-
-/**
- * Defines a column in an index
- */
-export interface IndexColumnSchema {
-	/** Column index in TableSchema.columns */
-	index: number;
-	/** Whether the index should sort in descending order */
-	desc?: boolean;	// default false
-	/** Optional collation sequence for the column */
-	collation?: string;
-}
-
-/**
- * Represents an index definition
- */
-export interface IndexSchema {
-	/** Index name */
-	name: string;
-	/** Columns in the index */
-	columns: ReadonlyArray<IndexColumnSchema>;
-}
-
-/**
- * Creates a basic TableSchema with minimal configuration
- *
- * @param name Table name
- * @param columns Array of column name and type objects
- * @param pkColNames Optional array of primary key column names
- * @param defaultNotNull Whether columns should be NOT NULL by default (defaults to true for Third Manifesto compliance)
- * @returns A frozen TableSchema object
- */
-export function createBasicSchema(name: string, columns: { name: string, type: string }[], pkColNames?: string[], defaultNotNull: boolean = true): Readonly<TableSchema> {
-	const columnSchemas = columns.map(c => columnDefToSchema({
-		name: c.name,
-		dataType: c.type,
-		constraints: []
-	}, defaultNotNull));
-	const columnIndexMap = buildColumnIndexMap(columnSchemas);
-	const pkDef = pkColNames
-		? pkColNames.map(pkName => {
-			const idx = columnIndexMap.get(pkName.toLowerCase());
-			if (idx === undefined) quereusError(`PK column ${pkName} not found`);
-			return { index: idx, desc: false };
-		})
-		: [];
-
-	const defaultMemoryModule = new MemoryTableModule();
-
-	return Object.freeze({
-		name: name,
-		schemaName: 'main',
-		columns: columnSchemas,
-		columnIndexMap: columnIndexMap,
-		primaryKeyDefinition: pkDef,
-		checkConstraints: [] as RowConstraintSchema[],
-		indexes: [],
-		vtabModule: defaultMemoryModule,
-		vtabAuxData: null,
-		vtabArgs: {},
-		vtabModuleName: 'memory',
-		isTemporary: false,
-		isView: false,
-		subqueryAST: undefined,
-		viewDefinition: undefined,
-		tableConstraints: [],
-		primaryKey: pkDef.map(def => columnSchemas[def.index].name),
-	});
-}
-
-/** Bitmask for row operations */
-export const enum RowOpFlag {
-	INSERT = 1,
-	UPDATE = 2,
-	DELETE = 4
-}
-export type RowOpMask = RowOpFlag;
-export const DEFAULT_ROWOP_MASK = RowOpFlag.INSERT | RowOpFlag.UPDATE;
-
-/**
- * Converts an array of row operations to a bitmask
- *
- * @param list Optional array of operation types
- * @returns A bitmask representing the operations
- */
-export function opsToMask(list?: RowOp[]): RowOpMask {
-	if (!list || list.length === 0) {
-		return DEFAULT_ROWOP_MASK;
-	}
-	let mask: RowOpMask = 0 as RowOpMask;
-	list.forEach(op => {
-		switch (op) {
-			case 'insert': mask |= RowOpFlag.INSERT; break;
-			case 'update': mask |= RowOpFlag.UPDATE; break;
-			case 'delete': mask |= RowOpFlag.DELETE; break;
-		}
-	});
-	return mask;
-}
-
-/**
- * Represents a CHECK constraint with operation flags
- */
-export interface RowConstraintSchema {
-	/** Optional constraint name */
-	name?: string;
-	/** Constraint expression */
-	expr: Expression;
-	/** Bitmask of operations the constraint applies to */
-	operations: RowOpMask;
-	/** Whether the constraint is deferrable */
-	deferrable?: boolean;
-	/** Whether the constraint is initially deferred */
-	initiallyDeferred?: boolean;
-}
-
-export interface PrimaryKeyColumnDefinition {
-	index: number;
-	desc?: boolean;	// default false
-	autoIncrement?: boolean;
-	collation?: string;
-}
-
-/**
- * Helper to parse primary key from AST column and table constraints.
- * @param columns Parsed column definitions from AST.
- * @param constraints Parsed table constraints from AST.
- * @returns A ReadonlyArray defining the primary key columns (index and direction), or undefined.
- * @throws QuereusError if multiple primary keys are defined or PK column not found.
- */
-export function findPKDefinition(
-	columns: ReadonlyArray<ColumnSchema>,
-	constraints: ReadonlyArray<AST.TableConstraint> | undefined,
-): ReadonlyArray<PrimaryKeyColumnDefinition> {
-	const columnPK = findColumnPKDefinition(columns);
-	const constraintPK = findConstraintPKDefinition(columns, constraints);
-
-	if (constraintPK && columnPK) {
-		throw new QuereusError("Cannot define both table-level and column-level PRIMARY KEYs", StatusCode.CONSTRAINT);
-	}
-
-	let finalPkDef = constraintPK ?? columnPK;
-
-	if (!finalPkDef) {
-		// Quereus-specific behavior: Include all columns in the primary key when no explicit primary key is defined
-		// This differs from SQLite which would use the first INTEGER column or an implicit rowid
-		// This design choice ensures predictable behavior and avoids potential confusion with SQLite's implicit rules
-		warnLog(`No PRIMARY KEY explicitly defined. Including all columns in primary key.`);
-		finalPkDef = Object.freeze(
-			columns.map((col, index) => ({
-				index,
-				desc: false,
-				collation: col.collation || 'BINARY'
-			}))
-		);
-	}
-
-	// Don't require NOT NULL, we want to be more flexible
-
-	return finalPkDef as ReadonlyArray<PrimaryKeyColumnDefinition>;
-}
-
-function findConstraintPKDefinition(
-	columns: readonly ColumnSchema[],
-	constraints: readonly TableConstraint[] | undefined
-): PrimaryKeyColumnDefinition[] | undefined {
-	const colMap = buildColumnIndexMap(columns);
-	let constraintPKs: PrimaryKeyColumnDefinition[] | undefined;
-
-	if (constraints) {
-		for (const constraint of constraints) {
-			if (constraint.type === 'primaryKey') {
-				if (constraintPKs) {
-					throw new QuereusError("Multiple table-level PRIMARY KEY constraints defined", StatusCode.CONSTRAINT);
-				}
-				if (!constraint.columns || constraint.columns.length === 0) {
-					// An empty column list is fine; means table can have 0-1 rows
-					constraintPKs = [];
-				} else {
-					constraintPKs = constraint.columns.map(colInfo => {
-						const colIndex = colMap.get(colInfo.name.toLowerCase());
-						if (colIndex === undefined) {
-							throw new QuereusError(`PRIMARY KEY column '${colInfo.name}' not found in table definition`, StatusCode.ERROR);
-						}
-						return {
-							index: colIndex,
-							desc: colInfo.direction === 'desc',
-							collation: columns[colIndex].collation || 'BINARY'
-						};
-					});
-				}
-			}
-		}
-	}
-	return constraintPKs;
-}
-
-function findColumnPKDefinition(columns: ReadonlyArray<ColumnSchema>): ReadonlyArray<PrimaryKeyColumnDefinition> | undefined {
-	const pkCols = columns
-		.map((col, index) => ({ ...col, originalIndex: index }))
-		.filter(col => col.primaryKey)
-		.sort((a, b) => a.pkOrder - b.pkOrder);
-
-	if (pkCols.length > 1 && pkCols.some(col => col.pkOrder === 0)) {
-		warnLog("Multiple column-level PRIMARY KEYs defined without explicit pkOrder; consider a table-level PRIMARY KEY for composite keys.");
-	}
-
-	if (pkCols.length > 1) {
-		warnLog('Multiple columns defined as PRIMARY KEY at column level. Forming a composite key.');
-	}
-
-	if (pkCols.length === 0) {
-		return undefined;
-	}
-
-	return Object.freeze(pkCols.map(col => ({
-		index: col.originalIndex,
-		desc: col.affinity === SqlDataType.INTEGER && col.pkDirection === 'desc',
-		autoIncrement: col.affinity === SqlDataType.INTEGER,
-		collation: col.collation || 'BINARY'
-	})));
-}
-
+import type { ColumnSchema } from './column.js';
+import type { AnyVirtualTableModule } from '../vtab/module.js';
+import { MemoryTableModule } from '../vtab/memory/module.js';
+import type { Expression } from '../parser/ast.js';
+import { type ColumnDef, type TableConstraint } from '../parser/ast.js';
+import { getAffinity } from '../common/type-inference.js';
+import { RowOp, SqlDataType, StatusCode, type SqlValue } from '../common/types.js';
+import type * as AST from '../parser/ast.js';
+import { quereusError, QuereusError } from '../common/errors.js';
+import { createLogger } from '../common/logger.js';
+import { inferType } from '../types/registry.js';
+
+const log = createLogger('schema:table');
+const warnLog = log.extend('warn');
+
+/**
+ * Represents the schema definition of a table (real or virtual).
+ */
+export interface TableSchema {
+	/** Table name */
+	name: string;
+	/** Schema name (e.g., "main", "temp") */
+	schemaName: string;
+	/** Ordered list of column definitions */
+	columns: ReadonlyArray<ColumnSchema>;
+	/** Map from column name (lowercase) to column index */
+	columnIndexMap: ReadonlyMap<string, number>;
+	/** Definition of the primary key, including order and direction */
+	primaryKeyDefinition: ReadonlyArray<PrimaryKeyColumnDefinition>;
+	/** CHECK constraints defined on the table or its columns */
+	checkConstraints: ReadonlyArray<RowConstraintSchema>;
+	/** Reference to the registered module */
+	vtabModule: AnyVirtualTableModule;
+	/** If virtual, aux data passed during module registration */
+	vtabAuxData?: unknown;
+	/** If virtual, the arguments passed in CREATE VIRTUAL TABLE */
+	vtabArgs?: Record<string, SqlValue>;
+	/** If virtual, the name the module was registered with */
+	vtabModuleName: string;
+	/** Whether the table is a temporary table */
+	isTemporary?: boolean;
+	/** Whether the table is a view */
+	isView: boolean;
+	/** Whether the table is a subquery source */
+	subqueryAST?: AST.SelectStmt;
+	/** If virtual, the view definition */
+	viewDefinition?: AST.SelectStmt;
+	/** Table-level constraints */
+	tableConstraints?: readonly TableConstraint[];
+	/** Definitions of secondary indexes (relevant for planning) */
+	indexes?: ReadonlyArray<IndexSchema>;
+	/** Estimated number of rows in the table (for query planning) */
+	readonly estimatedRows?: number;
+	/** Whether the table is read-only */
+	isReadOnly?: boolean;	// default false
+	/** Mutation context variables for this table */
+	mutationContext?: ReadonlyArray<MutationContextDefinition>;
+	/** Foreign key constraints (parsed but not yet enforced by engine) */
+	// foreignKeys?: ReadonlyArray<ForeignKeyConstraintSchema>;
+	/** Unique constraints (beyond primary key) */
+	// uniqueConstraints?: ReadonlyArray<ConstraintSchema>;
+}
+
+/**
+ * Builds a map from column names to their indices in the columns array
+ *
+ * @param columns Array of column schemas
+ * @returns Map of lowercase column names to their indices
+ */
+export function buildColumnIndexMap(columns: ReadonlyArray<ColumnSchema>): Map<string, number> {
+	const map = new Map<string, number>();
+	columns.forEach((col, index) => {
+		map.set(col.name.toLowerCase(), index);
+	});
+	return map;
+}
+
+/**
+ * Extracts just the column indices from a primary key definition
+ *
+ * @param pkDef Primary key definition array
+ * @returns Array of column indices that form the primary key
+ */
+export function getPrimaryKeyIndices(pkDef: ReadonlyArray<PrimaryKeyColumnDefinition>): ReadonlyArray<number> {
+	return Object.freeze(pkDef.map(def => def.index));
+}
+
+/**
+ * Converts a parsed ColumnDef AST node into a runtime ColumnSchema object
+ *
+ * @param def Column definition AST node
+ * @param defaultNotNull Whether columns should be NOT NULL by default (Third Manifesto approach)
+ * @returns A runtime ColumnSchema object
+ */
+export function columnDefToSchema(def: ColumnDef, defaultNotNull: boolean = true): ColumnSchema {
+	// Infer logical type from the declared type name
+	const logicalType = inferType(def.dataType);
+
+	const schema: Partial<ColumnSchema> & { name: string } = {
+		name: def.name,
+		logicalType: logicalType,
+		affinity: getAffinity(def.dataType), // Keep for backward compatibility during transition
+		notNull: defaultNotNull, // Default based on Third Manifesto principles
+		primaryKey: false,
+		pkOrder: 0,
+		defaultValue: null,
+		collation: 'BINARY',
+		generated: false,
+	};
+
+	for (const constraint of def.constraints ?? []) {
+		switch (constraint.type) {
+			case 'primaryKey':
+				schema.primaryKey = true;
+				schema.pkDirection = constraint.direction;
+				break;
+			case 'notNull':
+				schema.notNull = true;
+				break;
+			case 'null':
+				schema.notNull = false;
+				break;
+			case 'unique':
+				break;
+			case 'default':
+				schema.defaultValue = constraint.expr;
+				break;
+			case 'collate':
+				schema.collation = constraint.collation;
+				// Validate collation compatibility with type
+				if (constraint.collation && logicalType.supportedCollations &&
+					!logicalType.supportedCollations.includes(constraint.collation)) {
+					throw new QuereusError(
+						`Collation '${constraint.collation}' is not supported for type '${logicalType.name}' on column '${def.name}'`,
+						StatusCode.ERROR
+					);
+				}
+				break;
+			case 'generated':
+				schema.generated = true;
+				break;
+		}
+	}
+
+	// PK implies NOT NULL (always, regardless of default)
+	if (schema.primaryKey) {
+		schema.notNull = true;
+	}
+
+	// If no explicit nullability constraint and default is nullable,
+	// we need to check if there's an explicit NULL declaration
+	// Note: SQL doesn't have explicit NULL constraints in standard syntax,
+	// so this primarily affects the default behavior
+
+	// Assign a default pkOrder if it's a PK but order isn't specified elsewhere
+	if (schema.primaryKey && schema.pkOrder === 0) {
+		schema.pkOrder = 1;
+	}
+
+	return schema as ColumnSchema;
+}
+
+/**
+ * Mutation context variable definition
+ */
+export interface MutationContextDefinition {
+	/** Variable name */
+	name: string;
+	/** Type affinity of the variable */
+	affinity: SqlDataType;
+	/** Whether the variable is NOT NULL */
+	notNull: boolean;
+}
+
+/**
+ * Converts AST mutation context variable to schema definition
+ *
+ * @param varDef AST mutation context variable definition
+ * @param defaultNotNull Whether variables should be NOT NULL by default
+ * @returns Mutation context definition schema object
+ */
+export function mutationContextVarToSchema(varDef: AST.MutationContextVar, defaultNotNull: boolean = true): MutationContextDefinition {
+	return {
+		name: varDef.name,
+		affinity: getAffinity(varDef.dataType),
+		notNull: varDef.notNull !== undefined ? varDef.notNull : defaultNotNull,
+	};
+}
+
+/**
+ * Defines a column in an index
+ */
+export interface IndexColumnSchema {
+	/** Column index in TableSchema.columns */
+	index: number;
+	/** Whether the index should sort in descending order */
+	desc?: boolean;	// default false
+	/** Optional collation sequence for the column */
+	collation?: string;
+}
+
+/**
+ * Represents an index definition
+ */
+export interface IndexSchema {
+	/** Index name */
+	name: string;
+	/** Columns in the index */
+	columns: ReadonlyArray<IndexColumnSchema>;
+}
+
+/**
+ * Creates a basic TableSchema with minimal configuration
+ *
+ * @param name Table name
+ * @param columns Array of column name and type objects
+ * @param pkColNames Optional array of primary key column names
+ * @param defaultNotNull Whether columns should be NOT NULL by default (defaults to true for Third Manifesto compliance)
+ * @returns A frozen TableSchema object
+ */
+export function createBasicSchema(name: string, columns: { name: string, type: string }[], pkColNames?: string[], defaultNotNull: boolean = true): Readonly<TableSchema> {
+	const columnSchemas = columns.map(c => columnDefToSchema({
+		name: c.name,
+		dataType: c.type,
+		constraints: []
+	}, defaultNotNull));
+	const columnIndexMap = buildColumnIndexMap(columnSchemas);
+	const pkDef = pkColNames
+		? pkColNames.map(pkName => {
+			const idx = columnIndexMap.get(pkName.toLowerCase());
+			if (idx === undefined) quereusError(`PK column ${pkName} not found`);
+			return { index: idx, desc: false };
+		})
+		: [];
+
+	const defaultMemoryModule = new MemoryTableModule();
+
+	return Object.freeze({
+		name: name,
+		schemaName: 'main',
+		columns: columnSchemas,
+		columnIndexMap: columnIndexMap,
+		primaryKeyDefinition: pkDef,
+		checkConstraints: [] as RowConstraintSchema[],
+		indexes: [],
+		vtabModule: defaultMemoryModule,
+		vtabAuxData: null,
+		vtabArgs: {},
+		vtabModuleName: 'memory',
+		isTemporary: false,
+		isView: false,
+		subqueryAST: undefined,
+		viewDefinition: undefined,
+		tableConstraints: [],
+		primaryKey: pkDef.map(def => columnSchemas[def.index].name),
+	});
+}
+
+/** Bitmask for row operations */
+export const enum RowOpFlag {
+	INSERT = 1,
+	UPDATE = 2,
+	DELETE = 4
+}
+export type RowOpMask = RowOpFlag;
+export const DEFAULT_ROWOP_MASK = RowOpFlag.INSERT | RowOpFlag.UPDATE;
+
+/**
+ * Converts an array of row operations to a bitmask
+ *
+ * @param list Optional array of operation types
+ * @returns A bitmask representing the operations
+ */
+export function opsToMask(list?: RowOp[]): RowOpMask {
+	if (!list || list.length === 0) {
+		return DEFAULT_ROWOP_MASK;
+	}
+	let mask: RowOpMask = 0 as RowOpMask;
+	list.forEach(op => {
+		switch (op) {
+			case 'insert': mask |= RowOpFlag.INSERT; break;
+			case 'update': mask |= RowOpFlag.UPDATE; break;
+			case 'delete': mask |= RowOpFlag.DELETE; break;
+		}
+	});
+	return mask;
+}
+
+/**
+ * Represents a CHECK constraint with operation flags
+ */
+export interface RowConstraintSchema {
+	/** Optional constraint name */
+	name?: string;
+	/** Constraint expression */
+	expr: Expression;
+	/** Bitmask of operations the constraint applies to */
+	operations: RowOpMask;
+	/** Whether the constraint is deferrable */
+	deferrable?: boolean;
+	/** Whether the constraint is initially deferred */
+	initiallyDeferred?: boolean;
+}
+
+export interface PrimaryKeyColumnDefinition {
+	index: number;
+	desc?: boolean;	// default false
+	autoIncrement?: boolean;
+	collation?: string;
+}
+
+/**
+ * Helper to parse primary key from AST column and table constraints.
+ * @param columns Parsed column definitions from AST.
+ * @param constraints Parsed table constraints from AST.
+ * @returns A ReadonlyArray defining the primary key columns (index and direction), or undefined.
+ * @throws QuereusError if multiple primary keys are defined or PK column not found.
+ */
+export function findPKDefinition(
+	columns: ReadonlyArray<ColumnSchema>,
+	constraints: ReadonlyArray<AST.TableConstraint> | undefined,
+): ReadonlyArray<PrimaryKeyColumnDefinition> {
+	const columnPK = findColumnPKDefinition(columns);
+	const constraintPK = findConstraintPKDefinition(columns, constraints);
+
+	if (constraintPK && columnPK) {
+		throw new QuereusError("Cannot define both table-level and column-level PRIMARY KEYs", StatusCode.CONSTRAINT);
+	}
+
+	let finalPkDef = constraintPK ?? columnPK;
+
+	if (!finalPkDef) {
+		// Quereus-specific behavior: Include all columns in the primary key when no explicit primary key is defined
+		// This differs from SQLite which would use the first INTEGER column or an implicit rowid
+		// This design choice ensures predictable behavior and avoids potential confusion with SQLite's implicit rules
+		warnLog(`No PRIMARY KEY explicitly defined. Including all columns in primary key.`);
+		finalPkDef = Object.freeze(
+			columns.map((col, index) => ({
+				index,
+				desc: false,
+				collation: col.collation || 'BINARY'
+			}))
+		);
+	}
+
+	// Don't require NOT NULL, we want to be more flexible
+
+	return finalPkDef as ReadonlyArray<PrimaryKeyColumnDefinition>;
+}
+
+function findConstraintPKDefinition(
+	columns: readonly ColumnSchema[],
+	constraints: readonly TableConstraint[] | undefined
+): PrimaryKeyColumnDefinition[] | undefined {
+	const colMap = buildColumnIndexMap(columns);
+	let constraintPKs: PrimaryKeyColumnDefinition[] | undefined;
+
+	if (constraints) {
+		for (const constraint of constraints) {
+			if (constraint.type === 'primaryKey') {
+				if (constraintPKs) {
+					throw new QuereusError("Multiple table-level PRIMARY KEY constraints defined", StatusCode.CONSTRAINT);
+				}
+				if (!constraint.columns || constraint.columns.length === 0) {
+					// An empty column list is fine; means table can have 0-1 rows
+					constraintPKs = [];
+				} else {
+					constraintPKs = constraint.columns.map(colInfo => {
+						const colIndex = colMap.get(colInfo.name.toLowerCase());
+						if (colIndex === undefined) {
+							throw new QuereusError(`PRIMARY KEY column '${colInfo.name}' not found in table definition`, StatusCode.ERROR);
+						}
+						return {
+							index: colIndex,
+							desc: colInfo.direction === 'desc',
+							collation: columns[colIndex].collation || 'BINARY'
+						};
+					});
+				}
+			}
+		}
+	}
+	return constraintPKs;
+}
+
+function findColumnPKDefinition(columns: ReadonlyArray<ColumnSchema>): ReadonlyArray<PrimaryKeyColumnDefinition> | undefined {
+	const pkCols = columns
+		.map((col, index) => ({ ...col, originalIndex: index }))
+		.filter(col => col.primaryKey)
+		.sort((a, b) => a.pkOrder - b.pkOrder);
+
+	if (pkCols.length > 1 && pkCols.some(col => col.pkOrder === 0)) {
+		warnLog("Multiple column-level PRIMARY KEYs defined without explicit pkOrder; consider a table-level PRIMARY KEY for composite keys.");
+	}
+
+	if (pkCols.length > 1) {
+		warnLog('Multiple columns defined as PRIMARY KEY at column level. Forming a composite key.');
+	}
+
+	if (pkCols.length === 0) {
+		return undefined;
+	}
+
+	return Object.freeze(pkCols.map(col => ({
+		index: col.originalIndex,
+		desc: col.affinity === SqlDataType.INTEGER && col.pkDirection === 'desc',
+		autoIncrement: col.affinity === SqlDataType.INTEGER,
+		collation: col.collation || 'BINARY'
+	})));
+}
+