@aztec/validator-ha-signer 0.0.1-commit.0208eb9

package/dest/slashing_protection_service.js

@@ -0,0 +1,225 @@
+/**
+ * Slashing Protection Service
+ *
+ * Provides distributed locking and slashing protection for validator duties.
+ * Uses an external database to coordinate across multiple validator nodes.
+ */ import { createLogger } from '@aztec/foundation/log';
+import { RunningPromise } from '@aztec/foundation/promise';
+import { sleep } from '@aztec/foundation/sleep';
+import { DutyStatus, getBlockIndexFromDutyIdentifier } from './db/types.js';
+import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
+/**
+ * Slashing Protection Service
+ *
+ * This service ensures that a validator only signs one block/attestation per slot,
+ * even when running multiple redundant nodes (HA setup).
+ *
+ * All nodes in the HA setup try to sign - the first one wins, others get
+ * DutyAlreadySignedError (normal) or SlashingProtectionError (if different data).
+ *
+ * Flow:
+ * 1. checkAndRecord() - Atomically try to acquire lock via tryInsertOrGetExisting
+ * 2. Caller performs the signing operation
+ * 3. recordSuccess() - Update to 'signed' status with signature
+ *    OR deleteDuty() - Delete the record to allow retry
+ */ export class SlashingProtectionService {
+    db;
+    config;
+    log;
+    pollingIntervalMs;
+    signingTimeoutMs;
+    maxStuckDutiesAgeMs;
+    cleanupRunningPromise;
+    lastOldDutiesCleanupAtMs;
+    constructor(db, config){
+        this.db = db;
+        this.config = config;
+        this.log = createLogger('slashing-protection');
+        this.pollingIntervalMs = config.pollingIntervalMs;
+        this.signingTimeoutMs = config.signingTimeoutMs;
+        // Default to 144s (2x 72s Aztec slot duration) if not explicitly configured
+        this.maxStuckDutiesAgeMs = config.maxStuckDutiesAgeMs ?? 144_000;
+        this.cleanupRunningPromise = new RunningPromise(this.cleanup.bind(this), this.log, this.maxStuckDutiesAgeMs);
+    }
+    /**
+   * Check if a duty can be performed and acquire the lock if so.
+   *
+   * This method uses an atomic insert-or-get operation.
+   * It will:
+   * 1. Try to insert a new record with 'signing' status
+   * 2. If insert succeeds, we acquired the lock - return the lockToken
+   * 3. If a record exists, handle based on status:
+   *    - SIGNED: Throw appropriate error (already signed or slashing protection)
+   *    - SIGNING: Wait and poll until status changes, then handle result
+   *
+   * @returns The lockToken that must be used for recordSuccess/deleteDuty
+   * @throws DutyAlreadySignedError if the duty was already completed
+   * @throws SlashingProtectionError if attempting to sign different data for same slot/duty
+   */ async checkAndRecord(params) {
+        const { validatorAddress, slot, dutyType, messageHash, nodeId } = params;
+        const startTime = Date.now();
+        this.log.debug(`Checking duty: ${dutyType} for slot ${slot}`, {
+            validatorAddress: validatorAddress.toString(),
+            nodeId
+        });
+        while(true){
+            // insert if not present, get existing if present
+            const { isNew, record } = await this.db.tryInsertOrGetExisting(params);
+            if (isNew) {
+                // We successfully acquired the lock
+                this.log.info(`Acquired lock for duty ${dutyType} at slot ${slot}`, {
+                    validatorAddress: validatorAddress.toString(),
+                    nodeId
+                });
+                return record.lockToken;
+            }
+            // Record already exists - handle based on status
+            if (record.status === DutyStatus.SIGNED) {
+                // Duty was already signed - check if same or different data
+                if (record.messageHash !== messageHash) {
+                    this.log.verbose(`Slashing protection triggered for duty ${dutyType} at slot ${slot}`, {
+                        validatorAddress: validatorAddress.toString(),
+                        existingMessageHash: record.messageHash,
+                        attemptedMessageHash: messageHash,
+                        existingNodeId: record.nodeId,
+                        attemptingNodeId: nodeId
+                    });
+                    throw new SlashingProtectionError(slot, dutyType, record.blockIndexWithinCheckpoint, record.messageHash, messageHash, record.nodeId);
+                }
+                throw new DutyAlreadySignedError(slot, dutyType, record.blockIndexWithinCheckpoint, record.nodeId);
+            } else if (record.status === DutyStatus.SIGNING) {
+                // Another node is currently signing - check for timeout
+                if (Date.now() - startTime > this.signingTimeoutMs) {
+                    this.log.warn(`Timeout waiting for signing to complete for duty ${dutyType} at slot ${slot}`, {
+                        validatorAddress: validatorAddress.toString(),
+                        timeoutMs: this.signingTimeoutMs,
+                        signingNodeId: record.nodeId
+                    });
+                    throw new DutyAlreadySignedError(slot, dutyType, record.blockIndexWithinCheckpoint, 'unknown (timeout)');
+                }
+                // Wait and poll
+                this.log.debug(`Waiting for signing to complete for duty ${dutyType} at slot ${slot}`, {
+                    validatorAddress: validatorAddress.toString(),
+                    signingNodeId: record.nodeId
+                });
+                await sleep(this.pollingIntervalMs);
+            // Loop continues - next iteration will check status again
+            } else {
+                throw new Error(`Unknown duty status: ${record.status}`);
+            }
+        }
+    }
+    /**
+   * Record a successful signing operation.
+   * Updates the duty status to 'signed' and stores the signature.
+   * Only succeeds if the lockToken matches (caller must be the one who created the duty).
+   *
+   * @returns true if the update succeeded, false if token didn't match
+   */ async recordSuccess(params) {
+        const { rollupAddress, validatorAddress, slot, dutyType, signature, nodeId, lockToken } = params;
+        const blockIndexWithinCheckpoint = getBlockIndexFromDutyIdentifier(params);
+        const success = await this.db.updateDutySigned(rollupAddress, validatorAddress, slot, dutyType, signature.toString(), lockToken, blockIndexWithinCheckpoint);
+        if (success) {
+            this.log.info(`Recorded successful signing for duty ${dutyType} at slot ${slot}`, {
+                validatorAddress: validatorAddress.toString(),
+                nodeId
+            });
+        } else {
+            this.log.warn(`Failed to record successful signing for duty ${dutyType} at slot ${slot}: invalid token`, {
+                validatorAddress: validatorAddress.toString(),
+                nodeId
+            });
+        }
+        return success;
+    }
+    /**
+   * Delete a duty record after a failed signing operation.
+   * Removes the record to allow another node/attempt to retry.
+   * Only succeeds if the lockToken matches (caller must be the one who created the duty).
+   *
+   * @returns true if the delete succeeded, false if token didn't match
+   */ async deleteDuty(params) {
+        const { rollupAddress, validatorAddress, slot, dutyType, lockToken } = params;
+        const blockIndexWithinCheckpoint = getBlockIndexFromDutyIdentifier(params);
+        const success = await this.db.deleteDuty(rollupAddress, validatorAddress, slot, dutyType, lockToken, blockIndexWithinCheckpoint);
+        if (success) {
+            this.log.info(`Deleted duty ${dutyType} at slot ${slot} to allow retry`, {
+                validatorAddress: validatorAddress.toString()
+            });
+        } else {
+            this.log.warn(`Failed to delete duty ${dutyType} at slot ${slot}: invalid token`, {
+                validatorAddress: validatorAddress.toString()
+            });
+        }
+        return success;
+    }
+    /**
+   * Get the node ID for this service
+   */ get nodeId() {
+        return this.config.nodeId;
+    }
+    /**
+   * Start running tasks.
+   * Cleanup runs immediately on start to recover from any previous crashes.
+   */ /**
+   * Start the background cleanup task.
+   * Also performs one-time cleanup of duties with outdated rollup addresses.
+   */ async start() {
+        // One-time cleanup at startup: remove duties from previous rollup versions
+        const numOutdatedRollupDuties = await this.db.cleanupOutdatedRollupDuties(this.config.l1Contracts.rollupAddress);
+        if (numOutdatedRollupDuties > 0) {
+            this.log.info(`Cleaned up ${numOutdatedRollupDuties} duties with outdated rollup address at startup`, {
+                currentRollupAddress: this.config.l1Contracts.rollupAddress.toString()
+            });
+        }
+        this.cleanupRunningPromise.start();
+        this.log.info('Slashing protection service started', {
+            nodeId: this.config.nodeId
+        });
+    }
+    /**
+   * Stop the background cleanup task.
+   */ async stop() {
+        await this.cleanupRunningPromise.stop();
+        this.log.info('Slashing protection service stopped', {
+            nodeId: this.config.nodeId
+        });
+    }
+    /**
+   * Close the database connection.
+   * Should be called after stop() during graceful shutdown.
+   */ async close() {
+        await this.db.close();
+        this.log.info('Slashing protection database connection closed');
+    }
+    /**
+   * Periodic cleanup of stuck duties and optionally old signed duties.
+   * Runs in the background via RunningPromise.
+   */ async cleanup() {
+        // 1. Clean up stuck duties (our own node's duties that got stuck in 'signing' status)
+        const numStuckDuties = await this.db.cleanupOwnStuckDuties(this.config.nodeId, this.maxStuckDutiesAgeMs);
+        if (numStuckDuties > 0) {
+            this.log.verbose(`Cleaned up ${numStuckDuties} stuck duties`, {
+                nodeId: this.config.nodeId,
+                maxStuckDutiesAgeMs: this.maxStuckDutiesAgeMs
+            });
+        }
+        // 2. Clean up old signed duties if configured
+        // we shouldn't run this as often as stuck duty cleanup.
+        if (this.config.cleanupOldDutiesAfterHours !== undefined) {
+            const maxAgeMs = this.config.cleanupOldDutiesAfterHours * 60 * 60 * 1000;
+            const nowMs = Date.now();
+            const shouldRun = this.lastOldDutiesCleanupAtMs === undefined || nowMs - this.lastOldDutiesCleanupAtMs >= maxAgeMs;
+            if (shouldRun) {
+                const numOldDuties = await this.db.cleanupOldDuties(maxAgeMs);
+                this.lastOldDutiesCleanupAtMs = nowMs;
+                if (numOldDuties > 0) {
+                    this.log.verbose(`Cleaned up ${numOldDuties} old signed duties`, {
+                        cleanupOldDutiesAfterHours: this.config.cleanupOldDutiesAfterHours,
+                        maxAgeMs
+                    });
+                }
+            }
+        }
+    }
+}

package/dest/test/pglite_pool.d.ts

@@ -0,0 +1,92 @@
+/**
+ * Vendored pg-compatible Pool/Client wrapper for PGlite.
+ *
+ * Copied from @middle-management/pglite-pg-adapter v0.0.3
+ * https://www.npmjs.com/package/@middle-management/pglite-pg-adapter
+ *
+ * Modifications:
+ * - Converted to ESM and TypeScript
+ * - Uses PGliteInterface instead of PGlite class to avoid TypeScript
+ *   type mismatches from ESM/CJS dual package resolution with private fields
+ * - Simplified rowCount calculation to handle CTEs properly
+ */
+import type { PGliteInterface } from '@electric-sql/pglite';
+import { EventEmitter } from 'events';
+import type { QueryResult, QueryResultRow } from 'pg';
+import { Readable, Writable } from 'stream';
+export interface PoolConfig {
+    pglite: PGliteInterface;
+    max?: number;
+    min?: number;
+}
+interface ClientConfig {
+    pglite: PGliteInterface;
+    host?: string;
+    port?: number;
+    ssl?: boolean;
+}
+export declare class Client extends EventEmitter {
+    protected pglite: PGliteInterface;
+    protected _connected: boolean;
+    readonly host: string;
+    readonly port: number;
+    readonly ssl: boolean;
+    readonly connection: object;
+    readonly copyFrom: () => Writable;
+    readonly copyTo: () => Readable;
+    readonly pauseDrain: () => void;
+    readonly resumeDrain: () => void;
+    readonly escapeLiteral: (str: string) => string;
+    readonly escapeIdentifier: (str: string) => string;
+    readonly setTypeParser: () => void;
+    readonly getTypeParser: () => (value: string) => unknown;
+    constructor(config: ClientConfig);
+    connect(): Promise<void>;
+    end(): Promise<void>;
+    query<R extends QueryResultRow = any>(text: string, values?: any[]): Promise<QueryResult<R>>;
+    protected convertPGliteResult<R extends QueryResultRow>(result: {
+        rows: R[];
+        fields: Array<{
+            name: string;
+            dataTypeID: number;
+        }>;
+        affectedRows?: number;
+    }): QueryResult<R>;
+    get connected(): boolean;
+}
+export declare class Pool extends EventEmitter {
+    private clients;
+    private availableClients;
+    private waitingQueue;
+    private _ended;
+    private pglite;
+    private _config;
+    readonly expiredCount = 0;
+    readonly options: PoolConfig;
+    constructor(config: PoolConfig);
+    get totalCount(): number;
+    get idleCount(): number;
+    get waitingCount(): number;
+    get ending(): boolean;
+    get ended(): boolean;
+    connect(): Promise<PoolClient>;
+    query<R extends QueryResultRow = any>(text: string, values?: any[]): Promise<QueryResult<R>>;
+    releaseClient(client: PoolClient): void;
+    end(): Promise<void>;
+}
+export declare class PoolClient extends Client {
+    private pool;
+    private _released;
+    private _inUse;
+    private _userReleased;
+    constructor(pglite: PGliteInterface, pool: Pool);
+    query<R extends QueryResultRow = any>(text: string, values?: any[]): Promise<QueryResult<R>>;
+    release(): void;
+    end(): Promise<void>;
+    _markInUse(): void;
+    _markAvailable(): void;
+    _markReleased(): void;
+    get connected(): boolean;
+}
+export {};
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicGdsaXRlX3Bvb2wuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy90ZXN0L3BnbGl0ZV9wb29sLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBOzs7Ozs7Ozs7OztHQVdHO0FBQ0gsT0FBTyxLQUFLLEVBQUUsZUFBZSxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDNUQsT0FBTyxFQUFFLFlBQVksRUFBRSxNQUFNLFFBQVEsQ0FBQztBQUN0QyxPQUFPLEtBQUssRUFBRSxXQUFXLEVBQUUsY0FBYyxFQUFFLE1BQU0sSUFBSSxDQUFDO0FBQ3RELE9BQU8sRUFBRSxRQUFRLEVBQUUsUUFBUSxFQUFFLE1BQU0sUUFBUSxDQUFDO0FBRTVDLE1BQU0sV0FBVyxVQUFVO0lBQ3pCLE1BQU0sRUFBRSxlQUFlLENBQUM7SUFDeEIsR0FBRyxDQUFDLEVBQUUsTUFBTSxDQUFDO0lBQ2IsR0FBRyxDQUFDLEVBQUUsTUFBTSxDQUFDO0NBQ2Q7QUFFRCxVQUFVLFlBQVk7SUFDcEIsTUFBTSxFQUFFLGVBQWUsQ0FBQztJQUN4QixJQUFJLENBQUMsRUFBRSxNQUFNLENBQUM7SUFDZCxJQUFJLENBQUMsRUFBRSxNQUFNLENBQUM7SUFDZCxHQUFHLENBQUMsRUFBRSxPQUFPLENBQUM7Q0FDZjtBQUVELHFCQUFhLE1BQU8sU0FBUSxZQUFZO0lBQ3RDLFNBQVMsQ0FBQyxNQUFNLEVBQUUsZUFBZSxDQUFDO0lBQ2xDLFNBQVMsQ0FBQyxVQUFVLFVBQVM7SUFDN0IsUUFBUSxDQUFDLElBQUksRUFBRSxNQUFNLENBQUM7SUFDdEIsUUFBUSxDQUFDLElBQUksRUFBRSxNQUFNLENBQUM7SUFDdEIsUUFBUSxDQUFDLEdBQUcsRUFBRSxPQUFPLENBQUM7SUFDdEIsUUFBUSxDQUFDLFVBQVUsRUFBRSxNQUFNLENBQUM7SUFHNUIsUUFBUSxDQUFDLFFBQVEsaUJBQWtDO0lBQ25ELFFBQVEsQ0FBQyxNQUFNLGlCQUFrQztJQUNqRCxRQUFRLENBQUMsVUFBVSxhQUFrQjtJQUNyQyxRQUFRLENBQUMsV0FBVyxhQUFrQjtJQUN0QyxRQUFRLENBQUMsYUFBYSwwQkFBMkQ7SUFDakYsUUFBUSxDQUFDLGdCQUFnQiwwQkFBMkQ7SUFDcEYsUUFBUSxDQUFDLGFBQWEsYUFBa0I7SUFDeEMsUUFBUSxDQUFDLGFBQWEsbUNBQWdFO0lBRXRGLFlBQVksTUFBTSxFQUFFLFlBQVksRUFPL0I7SUFFRCxPQUFPLElBQUksT0FBTyxDQUFDLElBQUksQ0FBQyxDQU92QjtJQUVELEdBQUcsSUFBSSxPQUFPLENBQUMsSUFBSSxDQUFDLENBT25CO0lBRUssS0FBSyxDQUFDLENBQUMsU0FBUyxjQUFjLEdBQUcsR0FBRyxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsTUFBTSxDQUFDLEVBQUUsR0FBRyxFQUFFLEdBQUcsT0FBTyxDQUFDLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQU1qRztJQUVELFNBQVMsQ0FBQyxtQkFBbUIsQ0FBQyxDQUFDLFNBQVMsY0FBYyxFQUFFLE1BQU0sRUFBRTtRQUM5RCxJQUFJLEVBQUUsQ0FBQyxFQUFFLENBQUM7UUFDVixNQUFNLEVBQUUsS0FBSyxDQUFDO1lBQUUsSUFBSSxFQUFFLE1BQU0sQ0FBQztZQUFDLFVBQVUsRUFBRSxNQUFNLENBQUE7U0FBRSxDQUFDLENBQUM7UUFDcEQsWUFBWSxDQUFDLEVBQUUsTUFBTSxDQUFDO0tBQ3ZCLEdBQUcsV0FBVyxDQUFDLENBQUMsQ0FBQyxDQWdCakI7SUFFRCxJQUFJLFNBQVMsSUFBSSxPQUFPLENBRXZCO0NBQ0Y7QUFFRCxxQkFBYSxJQUFLLFNBQVEsWUFBWTtJQUNwQyxPQUFPLENBQUMsT0FBTyxDQUFvQjtJQUNuQyxPQUFPLENBQUMsZ0JBQWdCLENBQW9CO0lBQzVDLE9BQU8sQ0FBQyxZQUFZLENBQTJDO0lBQy9ELE9BQU8sQ0FBQyxNQUFNLENBQVM7SUFDdkIsT0FBTyxDQUFDLE1BQU0sQ0FBa0I7SUFDaEMsT0FBTyxDQUFDLE9BQU8sQ0FBYTtJQUU1QixRQUFRLENBQUMsWUFBWSxLQUFLO0lBQzFCLFFBQVEsQ0FBQyxPQUFPLEVBQUUsVUFBVSxDQUFDO0lBRTdCLFlBQVksTUFBTSxFQUFFLFVBQVUsRUFLN0I7SUFFRCxJQUFJLFVBQVUsSUFBSSxNQUFNLENBRXZCO0lBRUQsSUFBSSxTQUFTLElBQUksTUFBTSxDQUV0QjtJQUVELElBQUksWUFBWSxJQUFJLE1BQU0sQ0FFekI7SUFFRCxJQUFJLE1BQU0sSUFBSSxPQUFPLENBRXBCO0lBRUQsSUFBSSxLQUFLLElBQUksT0FBTyxDQUVuQjtJQUVELE9BQU8sSUFBSSxPQUFPLENBQUMsVUFBVSxDQUFDLENBb0I3QjtJQUVLLEtBQUssQ0FBQyxDQUFDLFNBQVMsY0FBYyxHQUFHLEdBQUcsRUFBRSxJQUFJLEVBQUUsTUFBTSxFQUFFLE1BQU0sQ0FBQyxFQUFFLEdBQUcsRUFBRSxHQUFHLE9BQU8sQ0FBQyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FPakc7SUFFRCxhQUFhLENBQUMsTUFBTSxFQUFFLFVBQVUsR0FBRyxJQUFJLENBWXRDO0lBRUQsR0FBRyxJQUFJLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FPbkI7Q0FDRjtBQUVELHFCQUFhLFVBQVcsU0FBUSxNQUFNO0lBQ3BDLE9BQU8sQ0FBQyxJQUFJLENBQU87SUFDbkIsT0FBTyxDQUFDLFNBQVMsQ0FBUztJQUMxQixPQUFPLENBQUMsTUFBTSxDQUFRO0lBQ3RCLE9BQU8sQ0FBQyxhQUFhLENBQVM7SUFFOUIsWUFBWSxNQUFNLEVBQUUsZUFBZSxFQUFFLElBQUksRUFBRSxJQUFJLEVBSTlDO0lBRWMsS0FBSyxDQUFDLENBQUMsU0FBUyxjQUFjLEdBQUcsR0FBRyxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsTUFBTSxDQUFDLEVBQUUsR0FBRyxFQUFFLEdBQUcsT0FBTyxDQUFDLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQU0xRztJQUVELE9BQU8sSUFBSSxJQUFJLENBTWQ7SUFFUSxHQUFHLElBQUksT0FBTyxDQUFDLElBQUksQ0FBQyxDQUc1QjtJQUVELFVBQVUsSUFBSSxJQUFJLENBR2pCO0lBRUQsY0FBYyxJQUFJLElBQUksQ0FHckI7SUFFRCxhQUFhLElBQUksSUFBSSxDQUlwQjtJQUVELElBQWEsU0FBUyxJQUFJLE9BQU8sQ0FFaEM7Q0FDRiJ9

package/dest/test/pglite_pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pglite_pool.d.ts","sourceRoot":"","sources":["../../src/test/pglite_pool.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAE5C,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,eAAe,CAAC;IACxB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,UAAU,YAAY;IACpB,MAAM,EAAE,eAAe,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,OAAO,CAAC;CACf;AAED,qBAAa,MAAO,SAAQ,YAAY;IACtC,SAAS,CAAC,MAAM,EAAE,eAAe,CAAC;IAClC,SAAS,CAAC,UAAU,UAAS;IAC7B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC;IACtB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAG5B,QAAQ,CAAC,QAAQ,iBAAkC;IACnD,QAAQ,CAAC,MAAM,iBAAkC;IACjD,QAAQ,CAAC,UAAU,aAAkB;IACrC,QAAQ,CAAC,WAAW,aAAkB;IACtC,QAAQ,CAAC,aAAa,0BAA2D;IACjF,QAAQ,CAAC,gBAAgB,0BAA2D;IACpF,QAAQ,CAAC,aAAa,aAAkB;IACxC,QAAQ,CAAC,aAAa,mCAAgE;IAEtF,YAAY,MAAM,EAAE,YAAY,EAO/B;IAED,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAOvB;IAED,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAOnB;IAEK,KAAK,CAAC,CAAC,SAAS,cAAc,GAAG,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAMjG;IAED,SAAS,CAAC,mBAAmB,CAAC,CAAC,SAAS,cAAc,EAAE,MAAM,EAAE;QAC9D,IAAI,EAAE,CAAC,EAAE,CAAC;QACV,MAAM,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACpD,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,WAAW,CAAC,CAAC,CAAC,CAgBjB;IAED,IAAI,SAAS,IAAI,OAAO,CAEvB;CACF;AAED,qBAAa,IAAK,SAAQ,YAAY;IACpC,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,YAAY,CAA2C;IAC/D,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,OAAO,CAAa;IAE5B,QAAQ,CAAC,YAAY,KAAK;IAC1B,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC;IAE7B,YAAY,MAAM,EAAE,UAAU,EAK7B;IAED,IAAI,UAAU,IAAI,MAAM,CAEvB;IAED,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,IAAI,MAAM,IAAI,OAAO,CAEpB;IAED,IAAI,KAAK,IAAI,OAAO,CAEnB;IAED,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC,CAoB7B;IAEK,KAAK,CAAC,CAAC,SAAS,cAAc,GAAG,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAOjG;IAED,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAYtC;IAED,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAOnB;CACF;AAED,qBAAa,UAAW,SAAQ,MAAM;IACpC,OAAO,CAAC,IAAI,CAAO;IACnB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,MAAM,CAAQ;IACtB,OAAO,CAAC,aAAa,CAAS;IAE9B,YAAY,MAAM,EAAE,eAAe,EAAE,IAAI,EAAE,IAAI,EAI9C;IAEc,KAAK,CAAC,CAAC,SAAS,cAAc,GAAG,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAM1G;IAED,OAAO,IAAI,IAAI,CAMd;IAEQ,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAG5B;IAED,UAAU,IAAI,IAAI,CAGjB;IAED,cAAc,IAAI,IAAI,CAGrB;IAED,aAAa,IAAI,IAAI,CAIpB;IAED,IAAa,SAAS,IAAI,OAAO,CAEhC;CACF"}

package/dest/test/pglite_pool.js

@@ -0,0 +1,210 @@
+/**
+ * Vendored pg-compatible Pool/Client wrapper for PGlite.
+ *
+ * Copied from @middle-management/pglite-pg-adapter v0.0.3
+ * https://www.npmjs.com/package/@middle-management/pglite-pg-adapter
+ *
+ * Modifications:
+ * - Converted to ESM and TypeScript
+ * - Uses PGliteInterface instead of PGlite class to avoid TypeScript
+ *   type mismatches from ESM/CJS dual package resolution with private fields
+ * - Simplified rowCount calculation to handle CTEs properly
+ */ import { EventEmitter } from 'events';
+import { Readable, Writable } from 'stream';
+export class Client extends EventEmitter {
+    pglite;
+    _connected = false;
+    host;
+    port;
+    ssl;
+    connection;
+    // Stub implementations for pg compatibility
+    copyFrom = ()=>new Writable();
+    copyTo = ()=>new Readable();
+    pauseDrain = ()=>{};
+    resumeDrain = ()=>{};
+    escapeLiteral = (str)=>`'${str.replace(/'/g, "''")}'`;
+    escapeIdentifier = (str)=>`"${str.replace(/"/g, '""')}"`;
+    setTypeParser = ()=>{};
+    getTypeParser = ()=>(value)=>value;
+    constructor(config){
+        super();
+        this.pglite = config.pglite;
+        this.host = config.host || 'localhost';
+        this.port = config.port || 5432;
+        this.ssl = typeof config.ssl === 'boolean' ? config.ssl : !!config.ssl;
+        this.connection = {};
+    }
+    connect() {
+        if (this._connected) {
+            return Promise.resolve();
+        }
+        this._connected = true;
+        this.emit('connect');
+        return Promise.resolve();
+    }
+    end() {
+        if (!this._connected) {
+            return Promise.resolve();
+        }
+        this._connected = false;
+        this.emit('end');
+        return Promise.resolve();
+    }
+    async query(text, values) {
+        if (!this._connected) {
+            throw new Error('Client is not connected');
+        }
+        const result = await this.pglite.query(text, values);
+        return this.convertPGliteResult(result);
+    }
+    convertPGliteResult(result) {
+        return {
+            command: '',
+            rowCount: 'affectedRows' in result ? result.affectedRows ?? 0 : result.rows.length,
+            oid: 0,
+            fields: result.fields.map((field)=>({
+                    name: field.name,
+                    tableID: 0,
+                    columnID: 0,
+                    dataTypeID: field.dataTypeID,
+                    dataTypeSize: -1,
+                    dataTypeModifier: -1,
+                    format: 'text'
+                })),
+            rows: result.rows
+        };
+    }
+    get connected() {
+        return this._connected;
+    }
+}
+export class Pool extends EventEmitter {
+    clients = [];
+    availableClients = [];
+    waitingQueue = [];
+    _ended = false;
+    pglite;
+    _config;
+    expiredCount = 0;
+    options;
+    constructor(config){
+        super();
+        this._config = {
+            max: 10,
+            min: 0,
+            ...config
+        };
+        this.pglite = config.pglite;
+        this.options = config;
+    }
+    get totalCount() {
+        return this.clients.length;
+    }
+    get idleCount() {
+        return this.availableClients.length;
+    }
+    get waitingCount() {
+        return this.waitingQueue.length;
+    }
+    get ending() {
+        return this._ended;
+    }
+    get ended() {
+        return this._ended;
+    }
+    connect() {
+        if (this._ended) {
+            return Promise.reject(new Error('Pool is ended'));
+        }
+        if (this.availableClients.length > 0) {
+            const client = this.availableClients.pop();
+            client._markInUse();
+            return Promise.resolve(client);
+        }
+        if (this.clients.length < (this._config.max || 10)) {
+            const client = new PoolClient(this.pglite, this);
+            this.clients.push(client);
+            return Promise.resolve(client);
+        }
+        return new Promise((resolve)=>{
+            this.waitingQueue.push(resolve);
+        });
+    }
+    async query(text, values) {
+        const client = await this.connect();
+        try {
+            return await client.query(text, values);
+        } finally{
+            client.release();
+        }
+    }
+    releaseClient(client) {
+        const index = this.clients.indexOf(client);
+        if (index !== -1) {
+            client._markAvailable();
+            if (this.waitingQueue.length > 0) {
+                const resolve = this.waitingQueue.shift();
+                client._markInUse();
+                resolve(client);
+            } else {
+                this.availableClients.push(client);
+            }
+        }
+    }
+    end() {
+        this._ended = true;
+        this.clients.forEach((client)=>client._markReleased());
+        this.clients = [];
+        this.availableClients = [];
+        this.emit('end');
+        return Promise.resolve();
+    }
+}
+export class PoolClient extends Client {
+    pool;
+    _released = false;
+    _inUse = true;
+    _userReleased = false;
+    constructor(pglite, pool){
+        super({
+            pglite
+        });
+        this.pool = pool;
+        this._connected = true;
+    }
+    async query(text, values) {
+        if (this._userReleased && !this._inUse) {
+            throw new Error('Client has been released back to the pool');
+        }
+        const result = await this.pglite.query(text, values);
+        return this.convertPGliteResult(result);
+    }
+    release() {
+        if (this._released || this._userReleased) {
+            return;
+        }
+        this._userReleased = true;
+        this.pool.releaseClient(this);
+    }
+    end() {
+        this.release();
+        return Promise.resolve();
+    }
+    _markInUse() {
+        this._inUse = true;
+        this._userReleased = false;
+    }
+    _markAvailable() {
+        this._inUse = false;
+        this._userReleased = false;
+    }
+    _markReleased() {
+        this._released = true;
+        this._inUse = false;
+        this._userReleased = true;
+    }
+    get connected() {
+        return this._connected && !this._released;
+    }
+}