@dotdo/postgres 0.1.0

package/dist/worker/shutdown-manager.d.ts

@@ -0,0 +1,188 @@
+/**
+ * Shutdown Manager for PostgresDO
+ *
+ * Handles graceful shutdown, lifecycle management, and query tracking.
+ * Extracted from PostgresDO to reduce class size and separate concerns.
+ *
+ * @module worker/shutdown-manager
+ */
+/**
+ * Shutdown status for health checks
+ */
+export type ShutdownStatus = 'running' | 'draining' | 'shutdown';
+/**
+ * Shutdown event types for monitoring
+ */
+export type ShutdownEventType = 'shutdown_initiated' | 'drain_started' | 'drain_completed' | 'query_completed_during_drain' | 'forced_shutdown' | 'shutdown_completed' | 'websocket_closed_during_drain';
+/**
+ * Shutdown event for monitoring
+ */
+export interface ShutdownEvent {
+    type: ShutdownEventType;
+    timestamp: Date;
+    details?: Record<string, unknown>;
+}
+/**
+ * Event listener for shutdown events
+ */
+export type ShutdownEventListener = (event: ShutdownEvent) => void;
+/**
+ * Shutdown configuration
+ */
+export interface ShutdownConfig {
+    /** Grace period in ms before forced shutdown (default: 30000) */
+    gracePeriodMs?: number;
+    /** How often to check if queries are complete (default: 100ms) */
+    drainCheckIntervalMs?: number;
+    /** Maximum time to wait for PGlite close (default: 5000ms) */
+    closeTimeoutMs?: number;
+    /** Maximum number of shutdown event listeners to prevent memory leaks (default: 100) */
+    maxListeners?: number;
+}
+/**
+ * Active query tracking
+ */
+export interface ActiveQuery {
+    id: string;
+    sql: string;
+    startTime: number;
+    abortController?: AbortController;
+}
+/**
+ * Interface for closable PGlite instance
+ */
+interface ClosablePGlite {
+    close?(): Promise<void>;
+}
+/**
+ * Interface for DO state with WebSocket support
+ */
+interface WebSocketState {
+    getWebSockets(): WebSocket[];
+}
+/**
+ * ShutdownManager handles graceful shutdown for PostgresDO.
+ *
+ * Features:
+ * - Query tracking with unique IDs
+ * - Graceful draining of in-flight queries
+ * - WebSocket connection cleanup
+ * - Shutdown event notifications
+ * - Memory leak prevention for listeners
+ *
+ * @example
+ * ```typescript
+ * const shutdownManager = new ShutdownManager({
+ *   gracePeriodMs: 30000,
+ *   maxListeners: 100,
+ * })
+ *
+ * // Track queries
+ * const queryId = shutdownManager.trackQuery('SELECT * FROM users')
+ * try {
+ *   // Execute query...
+ * } finally {
+ *   shutdownManager.untrackQuery(queryId)
+ * }
+ *
+ * // Initiate shutdown
+ * await shutdownManager.shutdown(pglite, state)
+ * ```
+ */
+export declare class ShutdownManager {
+    private status;
+    private activeQueries;
+    private queryIdCounter;
+    private config;
+    private eventListeners;
+    private shutdownPromise;
+    constructor(config?: ShutdownConfig);
+    /**
+     * Get current shutdown status
+     */
+    getStatus(): ShutdownStatus;
+    /**
+     * Check if the DO is shutting down (draining or shutdown)
+     */
+    isShuttingDown(): boolean;
+    /**
+     * Get the number of active queries
+     */
+    getActiveQueryCount(): number;
+    /**
+     * Get active query info (for monitoring)
+     */
+    getActiveQueries(): Array<{
+        id: string;
+        sql: string;
+        durationMs: number;
+    }>;
+    /**
+     * Configure shutdown behavior
+     */
+    setConfig(config: Partial<ShutdownConfig>): void;
+    /**
+     * Get shutdown configuration
+     */
+    getConfig(): ShutdownConfig;
+    /**
+     * Get the current number of shutdown event listeners
+     */
+    getListenerCount(): number;
+    /**
+     * Add a shutdown event listener
+     *
+     * To prevent memory leaks, this method:
+     * - Warns when approaching the listener limit (80% of maxListeners)
+     * - Caps the listener array at maxListeners (oldest listeners are dropped)
+     */
+    onShutdownEvent(listener: ShutdownEventListener): () => void;
+    /**
+     * Emit a shutdown event to all listeners
+     */
+    private emitEvent;
+    /**
+     * Generate a unique query ID
+     */
+    private generateQueryId;
+    /**
+     * Track an active query
+     */
+    trackQuery(sql: string): string;
+    /**
+     * Untrack a completed query
+     */
+    untrackQuery(id: string): void;
+    /**
+     * Wait for all active queries to complete
+     */
+    private waitForQueriesToDrain;
+    /**
+     * Close all WebSocket connections during shutdown
+     */
+    private closeAllWebSockets;
+    /**
+     * Initiate graceful shutdown
+     * - Sets status to 'draining'
+     * - Waits for in-flight queries to complete (up to gracePeriodMs)
+     * - Closes PGlite and other resources
+     * - Sets status to 'shutdown'
+     *
+     * Returns a promise that resolves when shutdown is complete
+     */
+    shutdown(pglite: ClosablePGlite | null, state: WebSocketState, config?: Partial<ShutdownConfig>): Promise<void>;
+    /**
+     * Internal shutdown implementation
+     */
+    private performShutdown;
+    /**
+     * Reset the manager (for testing or DO hibernation recovery)
+     */
+    reset(): void;
+}
+/**
+ * Create a shutdown manager instance
+ */
+export declare function createShutdownManager(config?: ShutdownConfig): ShutdownManager;
+export {};
+//# sourceMappingURL=shutdown-manager.d.ts.map

package/dist/worker/shutdown-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdown-manager.d.ts","sourceRoot":"","sources":["../../src/worker/shutdown-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAoDH;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,SAAS,GAAG,UAAU,GAAG,UAAU,CAAA;AAEhE;;GAEG;AACH,MAAM,MAAM,iBAAiB,GACzB,oBAAoB,GACpB,eAAe,GACf,iBAAiB,GACjB,8BAA8B,GAC9B,iBAAiB,GACjB,oBAAoB,GACpB,+BAA+B,CAAA;AAEnC;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,iBAAiB,CAAA;IACvB,SAAS,EAAE,IAAI,CAAA;IACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAClC;AAED;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,CAAC,KAAK,EAAE,aAAa,KAAK,IAAI,CAAA;AAElE;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,iEAAiE;IACjE,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,kEAAkE;IAClE,oBAAoB,CAAC,EAAE,MAAM,CAAA;IAC7B,8DAA8D;IAC9D,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,wFAAwF;IACxF,YAAY,CAAC,EAAE,MAAM,CAAA;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAA;IACV,GAAG,EAAE,MAAM,CAAA;IACX,SAAS,EAAE,MAAM,CAAA;IACjB,eAAe,CAAC,EAAE,eAAe,CAAA;CAClC;AAED;;GAEG;AACH,UAAU,cAAc;IACtB,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;CACxB;AAED;;GAEG;AACH,UAAU,cAAc;IACtB,aAAa,IAAI,SAAS,EAAE,CAAA;CAC7B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,aAAa,CAAiC;IACtD,OAAO,CAAC,cAAc,CAAI;IAC1B,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,cAAc,CAA8B;IACpD,OAAO,CAAC,eAAe,CAA6B;gBAExC,MAAM,GAAE,cAAmB;IASvC;;OAEG;IACH,SAAS,IAAI,cAAc;IAI3B;;OAEG;IACH,cAAc,IAAI,OAAO;IAIzB;;OAEG;IACH,mBAAmB,IAAI,MAAM;IAI7B;;OAEG;IACH,gBAAgB,IAAI,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC;IAS1E;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,IAAI;IAIhD;;OAEG;IACH,SAAS,IAAI,cAAc;IAI3B;;OAEG;IACH,gBAAgB,IAAI,MAAM;IAI1B;;;;;;OAMG;IACH,eAAe,CAAC,QAAQ,EAAE,qBAAqB,GAAG,MAAM,IAAI;IA8B5D;;OAEG;IACH,OAAO,CAAC,SAAS;IAkBjB;;OAEG;IACH,OAAO,CAAC,eAAe;IAIvB;;OAEG;IACH,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM;IAU/B;;OAEG;IACH,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAc9B;;OAEG;YACW,qBAAqB;IAanC;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAkB1B;;;;;;;;OAQG;IACG,QAAQ,CACZ,MAAM,EAAE,cAAc,GAAG,IAAI,EAC7B,KAAK,EAAE,cAAc,EACrB,MAAM,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAC/B,OAAO,CAAC,IAAI,CAAC;IAehB;;OAEG;YACW,eAAe;IA6D7B;;OAEG;IACH,KAAK,IAAI,IAAI;CAMd;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,eAAe,CAE9E"}

package/dist/worker/shutdown-manager.js

@@ -0,0 +1,347 @@
+/**
+ * Shutdown Manager for PostgresDO
+ *
+ * Handles graceful shutdown, lifecycle management, and query tracking.
+ * Extracted from PostgresDO to reduce class size and separate concerns.
+ *
+ * @module worker/shutdown-manager
+ */
+// =============================================================================
+// Constants
+// =============================================================================
+/**
+ * Default grace period in milliseconds before forced shutdown.
+ * Allows in-flight queries to complete before termination.
+ */
+const DEFAULT_GRACE_PERIOD_MS = 30000;
+/**
+ * Default interval in milliseconds for checking if queries have drained.
+ * Frequent checks ensure responsive shutdown when queries complete.
+ */
+const DEFAULT_DRAIN_CHECK_INTERVAL_MS = 100;
+/**
+ * Default timeout in milliseconds for PGlite close operation.
+ * Prevents shutdown from hanging if PGlite fails to close cleanly.
+ */
+const DEFAULT_CLOSE_TIMEOUT_MS = 5000;
+/**
+ * Default maximum number of shutdown event listeners.
+ * Prevents memory leaks from uncleaned listener registrations.
+ */
+const DEFAULT_MAX_LISTENERS = 100;
+/**
+ * WebSocket close code for server shutdown.
+ * Standard code indicating server-initiated closure.
+ */
+const WEBSOCKET_CLOSE_CODE_GOING_AWAY = 1001;
+/**
+ * Threshold percentage for listener warning.
+ * Warns when listener count reaches 80% of max to prevent leaks.
+ */
+const LISTENER_WARNING_THRESHOLD = 0.8;
+/**
+ * Maximum length of SQL query string to include in monitoring output.
+ * Prevents excessive memory usage from very long queries.
+ */
+const QUERY_SQL_TRUNCATE_LENGTH = 100;
+/**
+ * ShutdownManager handles graceful shutdown for PostgresDO.
+ *
+ * Features:
+ * - Query tracking with unique IDs
+ * - Graceful draining of in-flight queries
+ * - WebSocket connection cleanup
+ * - Shutdown event notifications
+ * - Memory leak prevention for listeners
+ *
+ * @example
+ * ```typescript
+ * const shutdownManager = new ShutdownManager({
+ *   gracePeriodMs: 30000,
+ *   maxListeners: 100,
+ * })
+ *
+ * // Track queries
+ * const queryId = shutdownManager.trackQuery('SELECT * FROM users')
+ * try {
+ *   // Execute query...
+ * } finally {
+ *   shutdownManager.untrackQuery(queryId)
+ * }
+ *
+ * // Initiate shutdown
+ * await shutdownManager.shutdown(pglite, state)
+ * ```
+ */
+export class ShutdownManager {
+    status = 'running';
+    activeQueries = new Map();
+    queryIdCounter = 0;
+    config;
+    eventListeners = [];
+    shutdownPromise = null;
+    constructor(config = {}) {
+        this.config = {
+            gracePeriodMs: config.gracePeriodMs ?? DEFAULT_GRACE_PERIOD_MS,
+            drainCheckIntervalMs: config.drainCheckIntervalMs ?? DEFAULT_DRAIN_CHECK_INTERVAL_MS,
+            closeTimeoutMs: config.closeTimeoutMs ?? DEFAULT_CLOSE_TIMEOUT_MS,
+            maxListeners: config.maxListeners ?? DEFAULT_MAX_LISTENERS,
+        };
+    }
+    /**
+     * Get current shutdown status
+     */
+    getStatus() {
+        return this.status;
+    }
+    /**
+     * Check if the DO is shutting down (draining or shutdown)
+     */
+    isShuttingDown() {
+        return this.status !== 'running';
+    }
+    /**
+     * Get the number of active queries
+     */
+    getActiveQueryCount() {
+        return this.activeQueries.size;
+    }
+    /**
+     * Get active query info (for monitoring)
+     */
+    getActiveQueries() {
+        const now = performance.now();
+        return Array.from(this.activeQueries.values()).map((q) => ({
+            id: q.id,
+            sql: q.sql.substring(0, QUERY_SQL_TRUNCATE_LENGTH), // Truncate for safety
+            durationMs: now - q.startTime,
+        }));
+    }
+    /**
+     * Configure shutdown behavior
+     */
+    setConfig(config) {
+        this.config = { ...this.config, ...config };
+    }
+    /**
+     * Get shutdown configuration
+     */
+    getConfig() {
+        return { ...this.config };
+    }
+    /**
+     * Get the current number of shutdown event listeners
+     */
+    getListenerCount() {
+        return this.eventListeners.length;
+    }
+    /**
+     * Add a shutdown event listener
+     *
+     * To prevent memory leaks, this method:
+     * - Warns when approaching the listener limit (80% of maxListeners)
+     * - Caps the listener array at maxListeners (oldest listeners are dropped)
+     */
+    onShutdownEvent(listener) {
+        const maxListeners = this.config.maxListeners ?? DEFAULT_MAX_LISTENERS;
+        const warningThreshold = Math.floor(maxListeners * LISTENER_WARNING_THRESHOLD);
+        // Warn when approaching the limit
+        if (this.eventListeners.length >= warningThreshold && this.eventListeners.length < maxListeners) {
+            console.warn(`Approaching shutdown event listener limit: ${this.eventListeners.length}/${maxListeners}. ` +
+                `Consider calling cleanup functions to prevent memory leaks.`);
+        }
+        // If at capacity, remove the oldest listener to make room
+        if (this.eventListeners.length >= maxListeners) {
+            console.warn(`Shutdown event listener limit (${maxListeners}) reached. Dropping oldest listener. ` +
+                `Call cleanup functions to prevent listener leaks.`);
+            this.eventListeners.shift(); // Remove oldest listener
+        }
+        this.eventListeners.push(listener);
+        return () => {
+            const index = this.eventListeners.indexOf(listener);
+            if (index >= 0) {
+                this.eventListeners.splice(index, 1);
+            }
+        };
+    }
+    /**
+     * Emit a shutdown event to all listeners
+     */
+    emitEvent(type, details) {
+        // Build event without undefined values for exactOptionalPropertyTypes
+        const event = {
+            type,
+            timestamp: new Date(),
+        };
+        if (details !== undefined) {
+            event.details = details;
+        }
+        for (const listener of this.eventListeners) {
+            try {
+                listener(event);
+            }
+            catch (error) {
+                console.error('Shutdown event listener error:', error);
+            }
+        }
+    }
+    /**
+     * Generate a unique query ID
+     */
+    generateQueryId() {
+        return `query-${++this.queryIdCounter}-${Date.now()}`;
+    }
+    /**
+     * Track an active query
+     */
+    trackQuery(sql) {
+        const id = this.generateQueryId();
+        this.activeQueries.set(id, {
+            id,
+            sql,
+            startTime: performance.now(),
+        });
+        return id;
+    }
+    /**
+     * Untrack a completed query
+     */
+    untrackQuery(id) {
+        const query = this.activeQueries.get(id);
+        if (query) {
+            this.activeQueries.delete(id);
+            // Emit event if we're draining
+            if (this.status === 'draining') {
+                this.emitEvent('query_completed_during_drain', {
+                    queryId: id,
+                    remainingQueries: this.activeQueries.size,
+                });
+            }
+        }
+    }
+    /**
+     * Wait for all active queries to complete
+     */
+    async waitForQueriesToDrain(timeoutMs) {
+        const startTime = Date.now();
+        const checkInterval = this.config.drainCheckIntervalMs || DEFAULT_DRAIN_CHECK_INTERVAL_MS;
+        while (this.activeQueries.size > 0) {
+            if (Date.now() - startTime >= timeoutMs) {
+                return false; // Timeout reached
+            }
+            await new Promise((resolve) => setTimeout(resolve, checkInterval));
+        }
+        return true; // All queries completed
+    }
+    /**
+     * Close all WebSocket connections during shutdown
+     */
+    closeAllWebSockets(state) {
+        try {
+            const webSockets = state.getWebSockets();
+            for (const ws of webSockets) {
+                try {
+                    ws.close(WEBSOCKET_CLOSE_CODE_GOING_AWAY, 'Server shutting down');
+                    this.emitEvent('websocket_closed_during_drain', {
+                        reason: 'shutdown',
+                    });
+                }
+                catch {
+                    // WebSocket might already be closed
+                }
+            }
+        }
+        catch {
+            // getWebSockets might not be available
+        }
+    }
+    /**
+     * Initiate graceful shutdown
+     * - Sets status to 'draining'
+     * - Waits for in-flight queries to complete (up to gracePeriodMs)
+     * - Closes PGlite and other resources
+     * - Sets status to 'shutdown'
+     *
+     * Returns a promise that resolves when shutdown is complete
+     */
+    async shutdown(pglite, state, config) {
+        // If already shutting down, return existing promise
+        if (this.shutdownPromise) {
+            return this.shutdownPromise;
+        }
+        // Apply any config overrides
+        if (config) {
+            this.setConfig(config);
+        }
+        this.shutdownPromise = this.performShutdown(pglite, state);
+        return this.shutdownPromise;
+    }
+    /**
+     * Internal shutdown implementation
+     */
+    async performShutdown(pglite, state) {
+        const gracePeriodMs = this.config.gracePeriodMs || DEFAULT_GRACE_PERIOD_MS;
+        const closeTimeoutMs = this.config.closeTimeoutMs || DEFAULT_CLOSE_TIMEOUT_MS;
+        // Emit shutdown initiated event
+        this.emitEvent('shutdown_initiated', {
+            activeQueries: this.activeQueries.size,
+            gracePeriodMs,
+        });
+        // Set status to draining
+        this.status = 'draining';
+        this.emitEvent('drain_started', {
+            activeQueries: this.activeQueries.size,
+        });
+        // Wait for queries to drain (with timeout)
+        const drainedSuccessfully = await this.waitForQueriesToDrain(gracePeriodMs);
+        if (drainedSuccessfully) {
+            this.emitEvent('drain_completed', {
+                duration: Date.now() - (this.config.gracePeriodMs || DEFAULT_GRACE_PERIOD_MS),
+            });
+        }
+        else {
+            // Force shutdown after grace period
+            this.emitEvent('forced_shutdown', {
+                remainingQueries: this.activeQueries.size,
+                activeQueryIds: Array.from(this.activeQueries.keys()),
+            });
+        }
+        // Close all WebSocket connections
+        this.closeAllWebSockets(state);
+        // Close PGlite
+        if (pglite?.close) {
+            try {
+                // Add a timeout for close operation
+                await Promise.race([
+                    pglite.close(),
+                    new Promise((_, reject) => setTimeout(() => reject(new Error('PGlite close timeout')), closeTimeoutMs)),
+                ]);
+            }
+            catch (error) {
+                console.error('Error closing PGlite:', error);
+            }
+        }
+        // Clear state
+        this.status = 'shutdown';
+        this.activeQueries.clear();
+        this.emitEvent('shutdown_completed', {
+            drainedSuccessfully,
+        });
+    }
+    /**
+     * Reset the manager (for testing or DO hibernation recovery)
+     */
+    reset() {
+        this.status = 'running';
+        this.activeQueries.clear();
+        this.shutdownPromise = null;
+        // Keep listeners and config
+    }
+}
+/**
+ * Create a shutdown manager instance
+ */
+export function createShutdownManager(config) {
+    return new ShutdownManager(config);
+}
+//# sourceMappingURL=shutdown-manager.js.map

package/dist/worker/shutdown-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdown-manager.js","sourceRoot":"","sources":["../../src/worker/shutdown-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,gFAAgF;AAChF,YAAY;AACZ,gFAAgF;AAEhF;;;GAGG;AACH,MAAM,uBAAuB,GAAG,KAAK,CAAA;AAErC;;;GAGG;AACH,MAAM,+BAA+B,GAAG,GAAG,CAAA;AAE3C;;;GAGG;AACH,MAAM,wBAAwB,GAAG,IAAI,CAAA;AAErC;;;GAGG;AACH,MAAM,qBAAqB,GAAG,GAAG,CAAA;AAEjC;;;GAGG;AACH,MAAM,+BAA+B,GAAG,IAAI,CAAA;AAE5C;;;GAGG;AACH,MAAM,0BAA0B,GAAG,GAAG,CAAA;AAEtC;;;GAGG;AACH,MAAM,yBAAyB,GAAG,GAAG,CAAA;AA2ErC;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,OAAO,eAAe;IAClB,MAAM,GAAmB,SAAS,CAAA;IAClC,aAAa,GAAG,IAAI,GAAG,EAAuB,CAAA;IAC9C,cAAc,GAAG,CAAC,CAAA;IAClB,MAAM,CAAgB;IACtB,cAAc,GAA4B,EAAE,CAAA;IAC5C,eAAe,GAAyB,IAAI,CAAA;IAEpD,YAAY,SAAyB,EAAE;QACrC,IAAI,CAAC,MAAM,GAAG;YACZ,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,uBAAuB;YAC9D,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,IAAI,+BAA+B;YACpF,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,wBAAwB;YACjE,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,qBAAqB;SAC3D,CAAA;IACH,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAA;IACpB,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,MAAM,KAAK,SAAS,CAAA;IAClC,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAA;IAChC,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;QAC7B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACzD,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,yBAAyB,CAAC,EAAE,sBAAsB;YAC1E,UAAU,EAAE,GAAG,GAAG,CAAC,CAAC,SAAS;SAC9B,CAAC,CAAC,CAAA;IACL,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAA+B;QACvC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAA;IAC7C,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAA;IAC3B,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAA;IACnC,CAAC;IAED;;;;;;OAMG;IACH,eAAe,CAAC,QAA+B;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,qBAAqB,CAAA;QACtE,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,0BAA0B,CAAC,CAAA;QAE9E,kCAAkC;QAClC,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,IAAI,gBAAgB,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,YAAY,EAAE,CAAC;YAChG,OAAO,CAAC,IAAI,CACV,8CAA8C,IAAI,CAAC,cAAc,CAAC,MAAM,IAAI,YAAY,IAAI;gBAC5F,6DAA6D,CAC9D,CAAA;QACH,CAAC;QAED,0DAA0D;QAC1D,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,IAAI,YAAY,EAAE,CAAC;YAC/C,OAAO,CAAC,IAAI,CACV,kCAAkC,YAAY,uCAAuC;gBACnF,mDAAmD,CACtD,CAAA;YACD,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAA,CAAC,yBAAyB;QACvD,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAClC,OAAO,GAAG,EAAE;YACV,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;YACnD,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;gBACf,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;YACtC,CAAC;QACH,CAAC,CAAA;IACH,CAAC;IAED;;OAEG;IACK,SAAS,CAAC,IAAuB,EAAE,OAAiC;QAC1E,sEAAsE;QACtE,MAAM,KAAK,GAAkB;YAC3B,IAAI;YACJ,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAA;QACD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,KAAK,CAAC,OAAO,GAAG,OAAO,CAAA;QACzB,CAAC;QACD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAA;YACjB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;YACxD,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,eAAe;QACrB,OAAO,SAAS,EAAE,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAA;IACvD,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,GAAW;QACpB,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,EAAE,CAAA;QACjC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE;YACzB,EAAE;YACF,GAAG;YACH,SAAS,EAAE,WAAW,CAAC,GAAG,EAAE;SAC7B,CAAC,CAAA;QACF,OAAO,EAAE,CAAA;IACX,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACxC,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YAC7B,+BAA+B;YAC/B,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBAC/B,IAAI,CAAC,SAAS,CAAC,8BAA8B,EAAE;oBAC7C,OAAO,EAAE,EAAE;oBACX,gBAAgB,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;iBAC1C,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB,CAAC,SAAiB;QACnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC5B,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,+BAA+B,CAAA;QAEzF,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACnC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,SAAS,EAAE,CAAC;gBACxC,OAAO,KAAK,CAAA,CAAC,kBAAkB;YACjC,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC,CAAA;QACpE,CAAC;QACD,OAAO,IAAI,CAAA,CAAC,wBAAwB;IACtC,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,KAAqB;QAC9C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,KAAK,CAAC,aAAa,EAAE,CAAA;YACxC,KAAK,MAAM,EAAE,IAAI,UAAU,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACH,EAAE,CAAC,KAAK,CAAC,+BAA+B,EAAE,sBAAsB,CAAC,CAAA;oBACjE,IAAI,CAAC,SAAS,CAAC,+BAA+B,EAAE;wBAC9C,MAAM,EAAE,UAAU;qBACnB,CAAC,CAAA;gBACJ,CAAC;gBAAC,MAAM,CAAC;oBACP,oCAAoC;gBACtC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uCAAuC;QACzC,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,QAAQ,CACZ,MAA6B,EAC7B,KAAqB,EACrB,MAAgC;QAEhC,oDAAoD;QACpD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,eAAe,CAAA;QAC7B,CAAC;QAED,6BAA6B;QAC7B,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QACxB,CAAC;QAED,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;QAC1D,OAAO,IAAI,CAAC,eAAe,CAAA;IAC7B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,eAAe,CAC3B,MAA6B,EAC7B,KAAqB;QAErB,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,uBAAuB,CAAA;QAC1E,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,wBAAwB,CAAA;QAE7E,gCAAgC;QAChC,IAAI,CAAC,SAAS,CAAC,oBAAoB,EAAE;YACnC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;YACtC,aAAa;SACd,CAAC,CAAA;QAEF,yBAAyB;QACzB,IAAI,CAAC,MAAM,GAAG,UAAU,CAAA;QACxB,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE;YAC9B,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;SACvC,CAAC,CAAA;QAEF,2CAA2C;QAC3C,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAA;QAE3E,IAAI,mBAAmB,EAAE,CAAC;YACxB,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE;gBAChC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,uBAAuB,CAAC;aAC9E,CAAC,CAAA;QACJ,CAAC;aAAM,CAAC;YACN,oCAAoC;YACpC,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE;gBAChC,gBAAgB,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;gBACzC,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;aACtD,CAAC,CAAA;QACJ,CAAC;QAED,kCAAkC;QAClC,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAA;QAE9B,eAAe;QACf,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC;gBACH,oCAAoC;gBACpC,MAAM,OAAO,CAAC,IAAI,CAAC;oBACjB,MAAM,CAAC,KAAK,EAAE;oBACd,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAC9B,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,EAAE,cAAc,CAAC,CAC5E;iBACF,CAAC,CAAA;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAA;YAC/C,CAAC;QACH,CAAC;QAED,cAAc;QACd,IAAI,CAAC,MAAM,GAAG,UAAU,CAAA;QACxB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAA;QAE1B,IAAI,CAAC,SAAS,CAAC,oBAAoB,EAAE;YACnC,mBAAmB;SACpB,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,SAAS,CAAA;QACvB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAA;QAC1B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAA;QAC3B,4BAA4B;IAC9B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAAuB;IAC3D,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAA;AACpC,CAAC"}

package/dist/worker/sql-transform.d.ts

@@ -0,0 +1,61 @@
+/**
+ * SQL Transform for PostgreSQL Case Sensitivity
+ * Task: postgres-6u6.2 - SQL Transform (preserveCase, quoteIdentifier, tokenizer, wrapPGliteWithTransform)
+ *
+ * PostgreSQL folds unquoted identifiers to lowercase. This module provides utilities
+ * to preserve case by automatically quoting identifiers.
+ */
+/**
+ * Token types for SQL tokenization
+ */
+export type TokenType = 'keyword' | 'identifier' | 'quoted_identifier' | 'string' | 'number' | 'operator' | 'punctuation' | 'whitespace' | 'comment' | 'parameter';
+/**
+ * SQL Token structure
+ */
+export interface SQLToken {
+    type: TokenType;
+    value: string;
+}
+/**
+ * Quote a SQL identifier with double quotes
+ * Escapes any existing double quotes by doubling them
+ */
+export declare function quoteIdentifier(identifier: string): string;
+/**
+ * Tokenize a SQL string into tokens
+ */
+export declare function tokenizeSQL(sql: string): SQLToken[];
+/**
+ * Transform SQL to preserve case by quoting identifiers
+ */
+export declare function preserveCase(sql: string): string;
+/**
+ * Transform options for PGlite wrapper
+ */
+export interface TransformOptions {
+    preserveCase?: boolean;
+}
+/**
+ * PGlite-like interface for wrapping
+ */
+export interface PGliteLike {
+    query<T = unknown>(sql: string, params?: unknown[]): Promise<{
+        rows: T[];
+        fields: {
+            name: string;
+            dataTypeID: number;
+        }[];
+    }>;
+    exec?(sql: string): Promise<{
+        rows: unknown[];
+        fields: {
+            name: string;
+            dataTypeID: number;
+        }[];
+    }>;
+}
+/**
+ * Wrap a PGlite instance with SQL transformation
+ */
+export declare function wrapPGliteWithTransform<T extends PGliteLike>(pglite: T, options?: TransformOptions): T;
+//# sourceMappingURL=sql-transform.d.ts.map

package/dist/worker/sql-transform.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql-transform.d.ts","sourceRoot":"","sources":["../../src/worker/sql-transform.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH;;GAEG;AACH,MAAM,MAAM,SAAS,GACjB,SAAS,GACT,YAAY,GACZ,mBAAmB,GACnB,QAAQ,GACR,QAAQ,GACR,UAAU,GACV,aAAa,GACb,YAAY,GACZ,SAAS,GACT,WAAW,CAAA;AAEf;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,SAAS,CAAA;IACf,KAAK,EAAE,MAAM,CAAA;CACd;AA6DD;;;GAGG;AACH,wBAAgB,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAE1D;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,QAAQ,EAAE,CAoLnD;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CA+BhD;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,YAAY,CAAC,EAAE,OAAO,CAAA;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAAC,MAAM,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,EAAE,CAAA;KAAE,CAAC,CAAA;IAC3H,IAAI,CAAC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QAAC,MAAM,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,EAAE,CAAA;KAAE,CAAC,CAAA;CACjG;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,CAAC,SAAS,UAAU,EAC1D,MAAM,EAAE,CAAC,EACT,OAAO,GAAE,gBAAqB,GAC7B,CAAC,CA2BH"}