orez 0.0.1

package/dist/pg-proxy.js

@@ -0,0 +1,385 @@
+/**
+ * tcp proxy that makes pglite speak postgresql wire protocol.
+ *
+ * uses pg-gateway to handle protocol lifecycle for regular connections,
+ * and directly handles the raw socket for replication connections.
+ *
+ * regular connections: forwarded to pglite via execProtocolRaw()
+ * replication connections: intercepted, replication protocol faked
+ */
+import { createServer } from 'node:net';
+import { fromNodeSocket } from 'pg-gateway/node';
+import { handleReplicationQuery, handleStartReplication, } from './replication/handler';
+// database name -> search_path mapping
+const DB_SCHEMA_MAP = {
+    postgres: 'public',
+    zero_cvr: 'zero_cvr, public',
+    zero_cdb: 'zero_cdb, public',
+};
+// query rewrites: make pglite look like real postgres with logical replication
+const QUERY_REWRITES = [
+    // wal_level check
+    {
+        match: /current_setting\s*\(\s*'wal_level'\s*\)/gi,
+        replace: "'logical'::text",
+    },
+    // strip READ ONLY from BEGIN
+    {
+        match: /\bREAD\s+ONLY\b/gi,
+        replace: '',
+    },
+    // redirect pg_replication_slots to our fake table
+    {
+        match: /\bpg_replication_slots\b/g,
+        replace: 'public._zero_replication_slots',
+    },
+];
+// queries to intercept and return no-op success
+const NOOP_QUERY_PATTERNS = [
+    /^\s*SET\s+TRANSACTION\s+SNAPSHOT\s+/i,
+];
+/**
+ * extract query text from a Parse message (0x50).
+ */
+function extractParseQuery(data) {
+    if (data[0] !== 0x50)
+        return null;
+    let offset = 5;
+    while (offset < data.length && data[offset] !== 0)
+        offset++;
+    offset++;
+    const queryStart = offset;
+    while (offset < data.length && data[offset] !== 0)
+        offset++;
+    return new TextDecoder().decode(data.subarray(queryStart, offset));
+}
+/**
+ * rebuild a Parse message with a modified query string.
+ */
+function rebuildParseMessage(data, newQuery) {
+    let offset = 5;
+    while (offset < data.length && data[offset] !== 0)
+        offset++;
+    const nameEnd = offset + 1;
+    const nameBytes = data.subarray(5, nameEnd);
+    offset = nameEnd;
+    while (offset < data.length && data[offset] !== 0)
+        offset++;
+    offset++;
+    const suffix = data.subarray(offset);
+    const encoder = new TextEncoder();
+    const queryBytes = encoder.encode(newQuery);
+    const totalLen = 4 + nameBytes.length + queryBytes.length + 1 + suffix.length;
+    const result = new Uint8Array(1 + totalLen);
+    const dv = new DataView(result.buffer);
+    result[0] = 0x50;
+    dv.setInt32(1, totalLen);
+    let pos = 5;
+    result.set(nameBytes, pos);
+    pos += nameBytes.length;
+    result.set(queryBytes, pos);
+    pos += queryBytes.length;
+    result[pos++] = 0;
+    result.set(suffix, pos);
+    return result;
+}
+/**
+ * rebuild a Simple Query message with a modified query string.
+ */
+function rebuildSimpleQuery(newQuery) {
+    const encoder = new TextEncoder();
+    const queryBytes = encoder.encode(newQuery + '\0');
+    const buf = new Uint8Array(5 + queryBytes.length);
+    buf[0] = 0x51;
+    new DataView(buf.buffer).setInt32(1, 4 + queryBytes.length);
+    buf.set(queryBytes, 5);
+    return buf;
+}
+/**
+ * intercept and rewrite query messages to make pglite look like real postgres.
+ */
+function interceptQuery(data) {
+    const msgType = data[0];
+    if (msgType === 0x51) {
+        const view = new DataView(data.buffer, data.byteOffset, data.byteLength);
+        const len = view.getInt32(1);
+        let query = new TextDecoder()
+            .decode(data.subarray(5, 1 + len - 1))
+            .replace(/\0$/, '');
+        let modified = false;
+        for (const rw of QUERY_REWRITES) {
+            if (rw.match.test(query)) {
+                query = query.replace(rw.match, rw.replace);
+                modified = true;
+                rw.match.lastIndex = 0;
+            }
+            rw.match.lastIndex = 0;
+        }
+        if (modified) {
+            return rebuildSimpleQuery(query);
+        }
+    }
+    else if (msgType === 0x50) {
+        const query = extractParseQuery(data);
+        if (query) {
+            let newQuery = query;
+            let modified = false;
+            for (const rw of QUERY_REWRITES) {
+                if (rw.match.test(newQuery)) {
+                    newQuery = newQuery.replace(rw.match, rw.replace);
+                    modified = true;
+                    rw.match.lastIndex = 0;
+                }
+                rw.match.lastIndex = 0;
+            }
+            if (modified) {
+                return rebuildParseMessage(data, newQuery);
+            }
+        }
+    }
+    return data;
+}
+/**
+ * check if a query should be intercepted as a no-op.
+ */
+function isNoopQuery(data) {
+    let query = null;
+    if (data[0] === 0x51) {
+        const view = new DataView(data.buffer, data.byteOffset, data.byteLength);
+        const len = view.getInt32(1);
+        query = new TextDecoder()
+            .decode(data.subarray(5, 1 + len - 1))
+            .replace(/\0$/, '');
+    }
+    else if (data[0] === 0x50) {
+        query = extractParseQuery(data);
+    }
+    if (!query)
+        return false;
+    return NOOP_QUERY_PATTERNS.some((p) => p.test(query));
+}
+/**
+ * build a synthetic "SET" command complete response.
+ */
+function buildSetCompleteResponse() {
+    const encoder = new TextEncoder();
+    const tag = encoder.encode('SET\0');
+    const cc = new Uint8Array(1 + 4 + tag.length);
+    cc[0] = 0x43;
+    new DataView(cc.buffer).setInt32(1, 4 + tag.length);
+    cc.set(tag, 5);
+    const rfq = new Uint8Array(6);
+    rfq[0] = 0x5a;
+    new DataView(rfq.buffer).setInt32(1, 5);
+    rfq[5] = 0x54; // 'T' = in transaction
+    const result = new Uint8Array(cc.length + rfq.length);
+    result.set(cc, 0);
+    result.set(rfq, cc.length);
+    return result;
+}
+/**
+ * build a synthetic ParseComplete response for extended protocol no-ops.
+ */
+function buildParseCompleteResponse() {
+    const pc = new Uint8Array(5);
+    pc[0] = 0x31; // ParseComplete
+    new DataView(pc.buffer).setInt32(1, 4);
+    return pc;
+}
+/**
+ * strip ReadyForQuery messages from a response buffer.
+ */
+function stripReadyForQuery(data) {
+    if (data.length === 0)
+        return data;
+    const parts = [];
+    let offset = 0;
+    while (offset < data.length) {
+        const msgType = data[offset];
+        if (offset + 5 > data.length)
+            break;
+        const msgLen = new DataView(data.buffer, data.byteOffset + offset + 1).getInt32(0);
+        const totalLen = 1 + msgLen;
+        if (msgType !== 0x5a) {
+            parts.push(data.subarray(offset, offset + totalLen));
+        }
+        offset += totalLen;
+    }
+    if (parts.length === 0)
+        return new Uint8Array(0);
+    if (parts.length === 1)
+        return parts[0];
+    const total = parts.reduce((sum, p) => sum + p.length, 0);
+    const result = new Uint8Array(total);
+    let pos = 0;
+    for (const p of parts) {
+        result.set(p, pos);
+        pos += p.length;
+    }
+    return result;
+}
+// simple mutex for serializing pglite access
+class Mutex {
+    locked = false;
+    queue = [];
+    async acquire() {
+        if (!this.locked) {
+            this.locked = true;
+            return;
+        }
+        return new Promise((resolve) => {
+            this.queue.push(resolve);
+        });
+    }
+    release() {
+        const next = this.queue.shift();
+        if (next) {
+            next();
+        }
+        else {
+            this.locked = false;
+        }
+    }
+}
+const mutex = new Mutex();
+// module-level search_path tracking
+let currentSearchPath = 'public';
+export async function startPgProxy(db, config) {
+    const server = createServer(async (socket) => {
+        let dbName = 'postgres';
+        let isReplicationConnection = false;
+        try {
+            const connection = await fromNodeSocket(socket, {
+                auth: {
+                    method: 'password',
+                    getClearTextPassword() {
+                        return config.pgPassword;
+                    },
+                    validateCredentials(credentials) {
+                        return (credentials.password ===
+                            credentials.clearTextPassword &&
+                            credentials.username === config.pgUser);
+                    },
+                },
+                async onStartup(state) {
+                    const params = state.clientParams;
+                    if (params?.replication === 'database') {
+                        isReplicationConnection = true;
+                    }
+                    dbName = params?.database || 'postgres';
+                    console.info(`[orez] new connection: db=${dbName} user=${params?.user} replication=${params?.replication || 'none'}`);
+                    await db.waitReady;
+                },
+                async onMessage(data, state) {
+                    if (!state.isAuthenticated)
+                        return;
+                    // handle replication connections
+                    if (isReplicationConnection) {
+                        if (data[0] === 0x51) {
+                            const view = new DataView(data.buffer, data.byteOffset, data.byteLength);
+                            const len = view.getInt32(1);
+                            const query = new TextDecoder()
+                                .decode(data.subarray(5, 1 + len - 1))
+                                .replace(/\0$/, '');
+                            console.info(`[orez] repl query: ${query.slice(0, 200)}`);
+                        }
+                        return handleReplicationMessage(data, socket, db, connection);
+                    }
+                    // check for no-op queries
+                    if (isNoopQuery(data)) {
+                        if (data[0] === 0x51) {
+                            return buildSetCompleteResponse();
+                        }
+                        else if (data[0] === 0x50) {
+                            return buildParseCompleteResponse();
+                        }
+                    }
+                    // intercept and rewrite queries
+                    data = interceptQuery(data);
+                    // regular query: set search_path based on database name, then forward
+                    await mutex.acquire();
+                    try {
+                        const searchPath = DB_SCHEMA_MAP[dbName] || 'public';
+                        if (currentSearchPath !== searchPath) {
+                            await db.exec(`SET search_path TO ${searchPath}`);
+                            currentSearchPath = searchPath;
+                        }
+                        let result = await db.execProtocolRaw(data, {
+                            throwOnError: false,
+                        });
+                        // strip ReadyForQuery from non-Sync responses
+                        if (data[0] !== 0x53 && data[0] !== 0x51) {
+                            result = stripReadyForQuery(result);
+                        }
+                        return result;
+                    }
+                    finally {
+                        mutex.release();
+                    }
+                },
+            });
+        }
+        catch (err) {
+            if (!socket.destroyed) {
+                socket.destroy();
+            }
+        }
+    });
+    return new Promise((resolve, reject) => {
+        server.listen(config.pgPort, '127.0.0.1', () => {
+            console.info(`[orez] pg proxy listening on port ${config.pgPort}`);
+            resolve(server);
+        });
+        server.on('error', reject);
+    });
+}
+async function handleReplicationMessage(data, socket, db, connection) {
+    if (data[0] !== 0x51)
+        return undefined;
+    const view = new DataView(data.buffer, data.byteOffset, data.byteLength);
+    const len = view.getInt32(1);
+    const query = new TextDecoder()
+        .decode(data.subarray(5, 1 + len - 1))
+        .replace(/\0$/, '');
+    const upper = query.trim().toUpperCase();
+    // check if this is a START_REPLICATION command
+    if (upper.startsWith('START_REPLICATION')) {
+        await connection.detach();
+        const writer = {
+            write(chunk) {
+                if (!socket.destroyed) {
+                    socket.write(chunk);
+                }
+            },
+        };
+        // drain incoming standby status updates
+        socket.on('data', (_chunk) => { });
+        socket.on('close', () => {
+            socket.destroy();
+        });
+        handleStartReplication(query, writer, db).catch((err) => {
+            console.info(`[orez] replication stream ended: ${err}`);
+        });
+        return undefined;
+    }
+    // handle other replication queries
+    const response = await handleReplicationQuery(query, db);
+    if (response)
+        return response;
+    // fall through to pglite for unrecognized queries
+    await mutex.acquire();
+    try {
+        const searchPath = 'public';
+        if (currentSearchPath !== searchPath) {
+            await db.exec(`SET search_path TO ${searchPath}`);
+            currentSearchPath = searchPath;
+        }
+        return await db.execProtocolRaw(data, {
+            throwOnError: false,
+        });
+    }
+    finally {
+        mutex.release();
+    }
+}
+//# sourceMappingURL=pg-proxy.js.map

package/dist/pg-proxy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-proxy.js","sourceRoot":"","sources":["../src/pg-proxy.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,YAAY,EAA4B,MAAM,UAAU,CAAA;AAEjE,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAKhD,OAAO,EACL,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,uBAAuB,CAAA;AAE9B,uCAAuC;AACvC,MAAM,aAAa,GAA2B;IAC5C,QAAQ,EAAE,QAAQ;IAClB,QAAQ,EAAE,kBAAkB;IAC5B,QAAQ,EAAE,kBAAkB;CAC7B,CAAA;AAED,+EAA+E;AAC/E,MAAM,cAAc,GAA8C;IAChE,kBAAkB;IAClB;QACE,KAAK,EAAE,2CAA2C;QAClD,OAAO,EAAE,iBAAiB;KAC3B;IACD,6BAA6B;IAC7B;QACE,KAAK,EAAE,mBAAmB;QAC1B,OAAO,EAAE,EAAE;KACZ;IACD,kDAAkD;IAClD;QACE,KAAK,EAAE,2BAA2B;QAClC,OAAO,EAAE,gCAAgC;KAC1C;CACF,CAAA;AAED,gDAAgD;AAChD,MAAM,mBAAmB,GAAG;IAC1B,sCAAsC;CACvC,CAAA;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAgB;IACzC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IACjC,IAAI,MAAM,GAAG,CAAC,CAAA;IACd,OAAO,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAAE,MAAM,EAAE,CAAA;IAC3D,MAAM,EAAE,CAAA;IACR,MAAM,UAAU,GAAG,MAAM,CAAA;IACzB,OAAO,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAAE,MAAM,EAAE,CAAA;IAC3D,OAAO,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAA;AACpE,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAC1B,IAAgB,EAChB,QAAgB;IAEhB,IAAI,MAAM,GAAG,CAAC,CAAA;IACd,OAAO,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAAE,MAAM,EAAE,CAAA;IAC3D,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,CAAA;IAC1B,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,OAAO,CAAC,CAAA;IAE3C,MAAM,GAAG,OAAO,CAAA;IAChB,OAAO,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAAE,MAAM,EAAE,CAAA;IAC3D,MAAM,EAAE,CAAA;IAER,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IACpC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;IACjC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IAE3C,MAAM,QAAQ,GACZ,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,CAAA;IAC9D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAA;IAC3C,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IACtC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA;IAChB,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAA;IACxB,IAAI,GAAG,GAAG,CAAC,CAAA;IACX,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAA;IAC1B,GAAG,IAAI,SAAS,CAAC,MAAM,CAAA;IACvB,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAA;IAC3B,GAAG,IAAI,UAAU,CAAC,MAAM,CAAA;IACxB,MAAM,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAA;IACjB,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;IACvB,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,QAAgB;IAC1C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;IACjC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,CAAA;IAClD,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC,CAAA;IACjD,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA;IACb,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC,CAAA;IAC3D,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;IACtB,OAAO,GAAG,CAAA;AACZ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAgB;IACtC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;IAEvB,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACrB,MAAM,IAAI,GAAG,IAAI,QAAQ,CACvB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,UAAU,CAChB,CAAA;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;QAC5B,IAAI,KAAK,GAAG,IAAI,WAAW,EAAE;aAC1B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC;aACrC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;QAErB,IAAI,QAAQ,GAAG,KAAK,CAAA;QACpB,KAAK,MAAM,EAAE,IAAI,cAAc,EAAE,CAAC;YAChC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;gBAC3C,QAAQ,GAAG,IAAI,CAAA;gBACf,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAA;YACxB,CAAC;YACD,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAA;QACxB,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAA;QAClC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAA;QACrC,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,QAAQ,GAAG,KAAK,CAAA;YACpB,IAAI,QAAQ,GAAG,KAAK,CAAA;YACpB,KAAK,MAAM,EAAE,IAAI,cAAc,EAAE,CAAC;gBAChC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC5B,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;oBACjD,QAAQ,GAAG,IAAI,CAAA;oBACf,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAA;gBACxB,CAAC;gBACD,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAA;YACxB,CAAC;YACD,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,mBAAmB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;YAC5C,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAgB;IACnC,IAAI,KAAK,GAAkB,IAAI,CAAA;IAC/B,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACrB,MAAM,IAAI,GAAG,IAAI,QAAQ,CACvB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,UAAU,CAChB,CAAA;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;QAC5B,KAAK,GAAG,IAAI,WAAW,EAAE;aACtB,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC;aACrC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;IACvB,CAAC;SAAM,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC5B,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAA;IACjC,CAAC;IACD,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAA;IACxB,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAM,CAAC,CAAC,CAAA;AACxD,CAAC;AAED;;GAEG;AACH,SAAS,wBAAwB;IAC/B,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;IACjC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;IACnC,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,CAAA;IAC7C,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA;IACZ,IAAI,QAAQ,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,CAAA;IACnD,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA;IAEd,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;IAC7B,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA;IACb,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IACvC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA,CAAC,uBAAuB;IAErC,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAAA;IACrD,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;IACjB,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,CAAA;IAC1B,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,SAAS,0BAA0B;IACjC,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;IAC5B,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,CAAA,CAAC,gBAAgB;IAC7B,IAAI,QAAQ,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IACtC,OAAO,EAAE,CAAA;AACX,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,IAAgB;IAC1C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAA;IAElC,MAAM,KAAK,GAAiB,EAAE,CAAA;IAC9B,IAAI,MAAM,GAAG,CAAC,CAAA;IACd,OAAO,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAA;QAC5B,IAAI,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM;YAAE,MAAK;QACnC,MAAM,MAAM,GAAG,IAAI,QAAQ,CACzB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,GAAG,MAAM,GAAG,CAAC,CAC7B,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;QACb,MAAM,QAAQ,GAAG,CAAC,GAAG,MAAM,CAAA;QAE3B,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACrB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAA;QACtD,CAAC;QAED,MAAM,IAAI,QAAQ,CAAA;IACpB,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;IAChD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAA;IAEvC,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;IACzD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC,CAAA;IACpC,IAAI,GAAG,GAAG,CAAC,CAAA;IACX,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA;QAClB,GAAG,IAAI,CAAC,CAAC,MAAM,CAAA;IACjB,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,6CAA6C;AAC7C,MAAM,KAAK;IACD,MAAM,GAAG,KAAK,CAAA;IACd,KAAK,GAAsB,EAAE,CAAA;IAErC,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAA;YAClB,OAAM;QACR,CAAC;QACD,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACnC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC1B,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,OAAO;QACL,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAA;QAC/B,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,EAAE,CAAA;QACR,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACrB,CAAC;IACH,CAAC;CACF;AAED,MAAM,KAAK,GAAG,IAAI,KAAK,EAAE,CAAA;AAEzB,oCAAoC;AACpC,IAAI,iBAAiB,GAAG,QAAQ,CAAA;AAEhC,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,EAAU,EACV,MAAsB;IAEtB,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,MAAc,EAAE,EAAE;QACnD,IAAI,MAAM,GAAG,UAAU,CAAA;QACvB,IAAI,uBAAuB,GAAG,KAAK,CAAA;QAEnC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE;gBAC9C,IAAI,EAAE;oBACJ,MAAM,EAAE,UAAU;oBAClB,oBAAoB;wBAClB,OAAO,MAAM,CAAC,UAAU,CAAA;oBAC1B,CAAC;oBACD,mBAAmB,CAAC,WAInB;wBACC,OAAO,CACL,WAAW,CAAC,QAAQ;4BAClB,WAAW,CAAC,iBAAiB;4BAC/B,WAAW,CAAC,QAAQ,KAAK,MAAM,CAAC,MAAM,CACvC,CAAA;oBACH,CAAC;iBACF;gBAED,KAAK,CAAC,SAAS,CAAC,KAAK;oBACnB,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAA;oBACjC,IAAI,MAAM,EAAE,WAAW,KAAK,UAAU,EAAE,CAAC;wBACvC,uBAAuB,GAAG,IAAI,CAAA;oBAChC,CAAC;oBACD,MAAM,GAAG,MAAM,EAAE,QAAQ,IAAI,UAAU,CAAA;oBACvC,OAAO,CAAC,IAAI,CACV,6BAA6B,MAAM,SAAS,MAAM,EAAE,IAAI,gBAAgB,MAAM,EAAE,WAAW,IAAI,MAAM,EAAE,CACxG,CAAA;oBACD,MAAM,EAAE,CAAC,SAAS,CAAA;gBACpB,CAAC;gBAED,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK;oBACzB,IAAI,CAAC,KAAK,CAAC,eAAe;wBAAE,OAAM;oBAElC,iCAAiC;oBACjC,IAAI,uBAAuB,EAAE,CAAC;wBAC5B,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;4BACrB,MAAM,IAAI,GAAG,IAAI,QAAQ,CACvB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,UAAU,CAChB,CAAA;4BACD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;4BAC5B,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE;iCAC5B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC;iCACrC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;4BACrB,OAAO,CAAC,IAAI,CACV,sBAAsB,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAC5C,CAAA;wBACH,CAAC;wBACD,OAAO,wBAAwB,CAC7B,IAAI,EACJ,MAAM,EACN,EAAE,EACF,UAAU,CACX,CAAA;oBACH,CAAC;oBAED,0BAA0B;oBAC1B,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;wBACtB,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;4BACrB,OAAO,wBAAwB,EAAE,CAAA;wBACnC,CAAC;6BAAM,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;4BAC5B,OAAO,0BAA0B,EAAE,CAAA;wBACrC,CAAC;oBACH,CAAC;oBAED,gCAAgC;oBAChC,IAAI,GAAG,cAAc,CAAC,IAAI,CAAC,CAAA;oBAE3B,sEAAsE;oBACtE,MAAM,KAAK,CAAC,OAAO,EAAE,CAAA;oBACrB,IAAI,CAAC;wBACH,MAAM,UAAU,GACd,aAAa,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAA;wBACnC,IAAI,iBAAiB,KAAK,UAAU,EAAE,CAAC;4BACrC,MAAM,EAAE,CAAC,IAAI,CACX,sBAAsB,UAAU,EAAE,CACnC,CAAA;4BACD,iBAAiB,GAAG,UAAU,CAAA;wBAChC,CAAC;wBACD,IAAI,MAAM,GAAG,MAAM,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE;4BAC1C,YAAY,EAAE,KAAK;yBACpB,CAAC,CAAA;wBACF,8CAA8C;wBAC9C,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;4BACzC,MAAM,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAA;wBACrC,CAAC;wBACD,OAAO,MAAM,CAAA;oBACf,CAAC;4BAAS,CAAC;wBACT,KAAK,CAAC,OAAO,EAAE,CAAA;oBACjB,CAAC;gBACH,CAAC;aACF,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBACtB,MAAM,CAAC,OAAO,EAAE,CAAA;YAClB,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE;YAC7C,OAAO,CAAC,IAAI,CACV,qCAAqC,MAAM,CAAC,MAAM,EAAE,CACrD,CAAA;YACD,OAAO,CAAC,MAAM,CAAC,CAAA;QACjB,CAAC,CAAC,CAAA;QACF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;AACJ,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,IAAgB,EAChB,MAAc,EACd,EAAU,EACV,UAAsD;IAEtD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;QAAE,OAAO,SAAS,CAAA;IAEtC,MAAM,IAAI,GAAG,IAAI,QAAQ,CACvB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,UAAU,CAChB,CAAA;IACD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;IAC5B,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE;SAC5B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC;SACrC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;IACrB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;IAExC,+CAA+C;IAC/C,IAAI,KAAK,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAC1C,MAAM,UAAU,CAAC,MAAM,EAAE,CAAA;QAEzB,MAAM,MAAM,GAAG;YACb,KAAK,CAAC,KAAiB;gBACrB,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;oBACtB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACrB,CAAC;YACH,CAAC;SACF,CAAA;QAED,wCAAwC;QACxC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,MAAc,EAAE,EAAE,GAAE,CAAC,CAAC,CAAA;QAEzC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,MAAM,CAAC,OAAO,EAAE,CAAA;QAClB,CAAC,CAAC,CAAA;QAEF,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACtD,OAAO,CAAC,IAAI,CACV,oCAAoC,GAAG,EAAE,CAC1C,CAAA;QACH,CAAC,CAAC,CAAA;QACF,OAAO,SAAS,CAAA;IAClB,CAAC;IAED,mCAAmC;IACnC,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;IACxD,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAA;IAE7B,kDAAkD;IAClD,MAAM,KAAK,CAAC,OAAO,EAAE,CAAA;IACrB,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,QAAQ,CAAA;QAC3B,IAAI,iBAAiB,KAAK,UAAU,EAAE,CAAC;YACrC,MAAM,EAAE,CAAC,IAAI,CAAC,sBAAsB,UAAU,EAAE,CAAC,CAAA;YACjD,iBAAiB,GAAG,UAAU,CAAA;QAChC,CAAC;QACD,OAAO,MAAM,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE;YACpC,YAAY,EAAE,KAAK;SACpB,CAAC,CAAA;IACJ,CAAC;YAAS,CAAC;QACT,KAAK,CAAC,OAAO,EAAE,CAAA;IACjB,CAAC;AACH,CAAC"}

package/dist/pglite-manager.d.ts

@@ -0,0 +1,5 @@
+import { PGlite } from '@electric-sql/pglite';
+import type { ZeroLiteConfig } from './config';
+export declare function createPGliteInstance(config: ZeroLiteConfig): Promise<PGlite>;
+export declare function runMigrations(db: PGlite, config: ZeroLiteConfig): Promise<void>;
+//# sourceMappingURL=pglite-manager.d.ts.map

package/dist/pglite-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pglite-manager.d.ts","sourceRoot":"","sources":["../src/pglite-manager.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAA;AAE7C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAA;AAE9C,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,CAAC,CA2BjB;AAED,wBAAsB,aAAa,CACjC,EAAE,EAAE,MAAM,EACV,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,IAAI,CAAC,CA+Df"}

package/dist/pglite-manager.js

@@ -0,0 +1,71 @@
+import { readFileSync, readdirSync, existsSync, mkdirSync } from 'node:fs';
+import { join, resolve } from 'node:path';
+import { PGlite } from '@electric-sql/pglite';
+export async function createPGliteInstance(config) {
+    const dataPath = resolve(config.dataDir, 'pgdata');
+    mkdirSync(dataPath, { recursive: true });
+    console.info(`[orez] creating pglite instance at ${dataPath}`);
+    const db = new PGlite(dataPath);
+    await db.waitReady;
+    console.info('[orez] pglite ready');
+    // create schemas for multi-db simulation
+    await db.exec('CREATE SCHEMA IF NOT EXISTS zero_cvr');
+    await db.exec('CREATE SCHEMA IF NOT EXISTS zero_cdb');
+    // create publication for zero-cache
+    const pubName = process.env.ZERO_APP_PUBLICATIONS || 'zero_pub';
+    const pubs = await db.query(`SELECT count(*) as count FROM pg_publication WHERE pubname = $1`, [pubName]);
+    if (Number(pubs.rows[0].count) === 0) {
+        const quoted = '"' + pubName.replace(/"/g, '""') + '"';
+        await db.exec(`CREATE PUBLICATION ${quoted} FOR ALL TABLES`);
+    }
+    return db;
+}
+export async function runMigrations(db, config) {
+    const migrationsDir = resolve(config.migrationsDir);
+    if (!existsSync(migrationsDir)) {
+        console.info('[orez] no migrations directory found, skipping');
+        return;
+    }
+    // create migrations tracking table
+    await db.exec(`
+    CREATE TABLE IF NOT EXISTS public.migrations (
+      id SERIAL PRIMARY KEY,
+      name TEXT NOT NULL UNIQUE,
+      applied_at TIMESTAMPTZ DEFAULT NOW()
+    )
+  `);
+    // read drizzle journal for correct migration order
+    const journalPath = join(migrationsDir, 'meta', '_journal.json');
+    let files;
+    if (existsSync(journalPath)) {
+        const journal = JSON.parse(readFileSync(journalPath, 'utf-8'));
+        files = journal.entries.map((e) => `${e.tag}.sql`);
+    }
+    else {
+        files = readdirSync(migrationsDir)
+            .filter((f) => f.endsWith('.sql'))
+            .sort();
+    }
+    for (const file of files) {
+        const name = file.replace(/\.sql$/, '');
+        // check if already applied
+        const result = await db.query('SELECT count(*) as count FROM public.migrations WHERE name = $1', [name]);
+        if (Number(result.rows[0].count) > 0) {
+            continue;
+        }
+        console.info(`[orez] applying migration: ${name}`);
+        const sql = readFileSync(join(migrationsDir, file), 'utf-8');
+        // split by drizzle's statement-breakpoint marker
+        const statements = sql
+            .split('--> statement-breakpoint')
+            .map((s) => s.trim())
+            .filter(Boolean);
+        for (const stmt of statements) {
+            await db.exec(stmt);
+        }
+        await db.query('INSERT INTO public.migrations (name) VALUES ($1)', [name]);
+        console.info(`[orez] applied migration: ${name}`);
+    }
+    console.info('[orez] migrations complete');
+}
+//# sourceMappingURL=pglite-manager.js.map

package/dist/pglite-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pglite-manager.js","sourceRoot":"","sources":["../src/pglite-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAA;AAC1E,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAA;AAEzC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAA;AAI7C,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAsB;IAEtB,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAA;IAClD,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IAExC,OAAO,CAAC,IAAI,CAAC,sCAAsC,QAAQ,EAAE,CAAC,CAAA;IAC9D,MAAM,EAAE,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAA;IAE/B,MAAM,EAAE,CAAC,SAAS,CAAA;IAClB,OAAO,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAA;IAEnC,yCAAyC;IACzC,MAAM,EAAE,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAA;IACrD,MAAM,EAAE,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAA;IAErD,oCAAoC;IACpC,MAAM,OAAO,GACX,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,UAAU,CAAA;IACjD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,KAAK,CACzB,iEAAiE,EACjE,CAAC,OAAO,CAAC,CACV,CAAA;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,GAAG,CAAA;QACtD,MAAM,EAAE,CAAC,IAAI,CAAC,sBAAsB,MAAM,iBAAiB,CAAC,CAAA;IAC9D,CAAC;IAED,OAAO,EAAE,CAAA;AACX,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,EAAU,EACV,MAAsB;IAEtB,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;IACnD,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAA;QAC9D,OAAM;IACR,CAAC;IAED,mCAAmC;IACnC,MAAM,EAAE,CAAC,IAAI,CAAC;;;;;;GAMb,CAAC,CAAA;IAEF,mDAAmD;IACnD,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,EAAE,MAAM,EAAE,eAAe,CAAC,CAAA;IAChE,IAAI,KAAe,CAAA;IACnB,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAA;QAC9D,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CACzB,CAAC,CAAkB,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,MAAM,CACvC,CAAA;IACH,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,WAAW,CAAC,aAAa,CAAC;aAC/B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;aACjC,IAAI,EAAE,CAAA;IACX,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAA;QAEvC,2BAA2B;QAC3B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,KAAK,CAC3B,iEAAiE,EACjE,CAAC,IAAI,CAAC,CACP,CAAA;QACD,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACrC,SAAQ;QACV,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,8BAA8B,IAAI,EAAE,CAAC,CAAA;QAClD,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAA;QAE5D,iDAAiD;QACjD,MAAM,UAAU,GAAG,GAAG;aACnB,KAAK,CAAC,0BAA0B,CAAC;aACjC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;aACpB,MAAM,CAAC,OAAO,CAAC,CAAA;QAElB,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACrB,CAAC;QAED,MAAM,EAAE,CAAC,KAAK,CACZ,kDAAkD,EAClD,CAAC,IAAI,CAAC,CACP,CAAA;QACD,OAAO,CAAC,IAAI,CAAC,6BAA6B,IAAI,EAAE,CAAC,CAAA;IACnD,CAAC;IAED,OAAO,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAA;AAC5C,CAAC"}

package/dist/replication/change-tracker.d.ts

@@ -0,0 +1,14 @@
+import type { PGlite } from '@electric-sql/pglite';
+export interface ChangeRecord {
+    id: number;
+    watermark: number;
+    table_name: string;
+    op: 'INSERT' | 'UPDATE' | 'DELETE';
+    row_data: Record<string, unknown> | null;
+    old_data: Record<string, unknown> | null;
+    changed_at: string;
+}
+export declare function installChangeTracking(db: PGlite): Promise<void>;
+export declare function getChangesSince(db: PGlite, watermark: number, limit?: number): Promise<ChangeRecord[]>;
+export declare function getCurrentWatermark(db: PGlite): Promise<number>;
+//# sourceMappingURL=change-tracker.d.ts.map

package/dist/replication/change-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"change-tracker.d.ts","sourceRoot":"","sources":["../../src/replication/change-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAA;AAElD,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAA;IACV,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,MAAM,CAAA;IAClB,EAAE,EAAE,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAA;IAClC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;IACxC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;IACxC,UAAU,EAAE,MAAM,CAAA;CACnB;AAED,wBAAsB,qBAAqB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAsDrE;AA6BD,wBAAsB,eAAe,CACnC,EAAE,EAAE,MAAM,EACV,SAAS,EAAE,MAAM,EACjB,KAAK,SAAO,GACX,OAAO,CAAC,YAAY,EAAE,CAAC,CAMzB;AAED,wBAAsB,mBAAmB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAOrE"}

package/dist/replication/change-tracker.js

@@ -0,0 +1,86 @@
+export async function installChangeTracking(db) {
+    // create changes table and watermark sequence
+    await db.exec(`
+    CREATE SEQUENCE IF NOT EXISTS public._zero_watermark;
+
+    CREATE TABLE IF NOT EXISTS public._zero_changes (
+      id BIGSERIAL PRIMARY KEY,
+      watermark BIGINT NOT NULL DEFAULT nextval('public._zero_watermark'),
+      table_name TEXT NOT NULL,
+      op TEXT NOT NULL,
+      row_data JSONB,
+      old_data JSONB,
+      changed_at TIMESTAMPTZ DEFAULT NOW()
+    );
+
+    CREATE INDEX IF NOT EXISTS _zero_changes_watermark_idx ON public._zero_changes (watermark);
+
+    CREATE TABLE IF NOT EXISTS public._zero_replication_slots (
+      slot_name TEXT PRIMARY KEY,
+      restart_lsn TEXT NOT NULL DEFAULT '0/1000000',
+      confirmed_flush_lsn TEXT NOT NULL DEFAULT '0/1000000',
+      wal_status TEXT NOT NULL DEFAULT 'reserved',
+      plugin TEXT NOT NULL DEFAULT 'pgoutput',
+      slot_type TEXT NOT NULL DEFAULT 'logical',
+      active BOOLEAN NOT NULL DEFAULT false,
+      active_pid INTEGER DEFAULT NULL,
+      created_at TIMESTAMPTZ DEFAULT NOW()
+    );
+  `);
+    // create trigger function
+    await db.exec(`
+    CREATE OR REPLACE FUNCTION public._zero_track_change() RETURNS TRIGGER AS $$
+    BEGIN
+      IF TG_OP = 'DELETE' THEN
+        INSERT INTO public._zero_changes (table_name, op, old_data)
+        VALUES (TG_TABLE_NAME, 'DELETE', row_to_json(OLD)::jsonb);
+        RETURN OLD;
+      ELSIF TG_OP = 'UPDATE' THEN
+        INSERT INTO public._zero_changes (table_name, op, row_data, old_data)
+        VALUES (TG_TABLE_NAME, 'UPDATE', row_to_json(NEW)::jsonb, row_to_json(OLD)::jsonb);
+        RETURN NEW;
+      ELSIF TG_OP = 'INSERT' THEN
+        INSERT INTO public._zero_changes (table_name, op, row_data)
+        VALUES (TG_TABLE_NAME, 'INSERT', row_to_json(NEW)::jsonb);
+        RETURN NEW;
+      END IF;
+      RETURN NULL;
+    END;
+    $$ LANGUAGE plpgsql;
+  `);
+    // install triggers on all public tables
+    await installTriggersOnAllTables(db);
+}
+function quoteIdent(name) {
+    return '"' + name.replace(/"/g, '""') + '"';
+}
+async function installTriggersOnAllTables(db) {
+    const tables = await db.query(`SELECT tablename FROM pg_tables
+     WHERE schemaname = 'public'
+       AND tablename NOT IN ('migrations', '_zero_changes')
+       AND tablename NOT LIKE '_zero_%'`);
+    let count = 0;
+    for (const { tablename } of tables.rows) {
+        const quoted = quoteIdent(tablename);
+        await db.exec(`
+      DROP TRIGGER IF EXISTS _zero_change_trigger ON public.${quoted};
+      CREATE TRIGGER _zero_change_trigger
+        AFTER INSERT OR UPDATE OR DELETE ON public.${quoted}
+        FOR EACH ROW EXECUTE FUNCTION public._zero_track_change();
+    `);
+        count++;
+    }
+    console.info(`[orez] installed change tracking triggers on ${count} tables`);
+}
+export async function getChangesSince(db, watermark, limit = 1000) {
+    const result = await db.query('SELECT * FROM public._zero_changes WHERE watermark > $1 ORDER BY watermark LIMIT $2', [watermark, limit]);
+    return result.rows;
+}
+export async function getCurrentWatermark(db) {
+    const result = await db.query('SELECT last_value, is_called FROM public._zero_watermark');
+    const { last_value, is_called } = result.rows[0];
+    if (!is_called)
+        return 0;
+    return Number(last_value);
+}
+//# sourceMappingURL=change-tracker.js.map

package/dist/replication/change-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"change-tracker.js","sourceRoot":"","sources":["../../src/replication/change-tracker.ts"],"names":[],"mappings":"AAYA,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,EAAU;IACpD,8CAA8C;IAC9C,MAAM,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;GA0Bb,CAAC,CAAA;IAEF,0BAA0B;IAC1B,MAAM,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;GAmBb,CAAC,CAAA;IAEF,wCAAwC;IACxC,MAAM,0BAA0B,CAAC,EAAE,CAAC,CAAA;AACtC,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC9B,OAAO,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,GAAG,CAAA;AAC7C,CAAC;AAED,KAAK,UAAU,0BAA0B,CAAC,EAAU;IAClD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,KAAK,CAC3B;;;wCAGoC,CACrC,CAAA;IAED,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,KAAK,MAAM,EAAE,SAAS,EAAE,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,CAAC,CAAA;QACpC,MAAM,EAAE,CAAC,IAAI,CAAC;8DAC4C,MAAM;;qDAEf,MAAM;;KAEtD,CAAC,CAAA;QACF,KAAK,EAAE,CAAA;IACT,CAAC;IAED,OAAO,CAAC,IAAI,CAAC,gDAAgD,KAAK,SAAS,CAAC,CAAA;AAC9E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,EAAU,EACV,SAAiB,EACjB,KAAK,GAAG,IAAI;IAEZ,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,KAAK,CAC3B,qFAAqF,EACrF,CAAC,SAAS,EAAE,KAAK,CAAC,CACnB,CAAA;IACD,OAAO,MAAM,CAAC,IAAI,CAAA;AACpB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAU;IAClD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,KAAK,CAC3B,0DAA0D,CAC3D,CAAA;IACD,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAChD,IAAI,CAAC,SAAS;QAAE,OAAO,CAAC,CAAA;IACxB,OAAO,MAAM,CAAC,UAAU,CAAC,CAAA;AAC3B,CAAC"}

package/dist/replication/handler.d.ts

@@ -0,0 +1,24 @@
+/**
+ * replication protocol handler.
+ *
+ * intercepts replication-mode queries (IDENTIFY_SYSTEM, CREATE_REPLICATION_SLOT,
+ * START_REPLICATION) and returns fake responses that make zero-cache believe
+ * it's talking to a real postgres with logical replication.
+ */
+import type { PGlite } from '@electric-sql/pglite';
+export interface ReplicationWriter {
+    write(data: Uint8Array): void;
+}
+declare function buildErrorResponse(message: string): Uint8Array;
+/**
+ * handle a replication query. returns response bytes or null if not handled.
+ * async because slot operations need to write to pglite.
+ */
+export declare function handleReplicationQuery(query: string, db: PGlite): Promise<Uint8Array | null>;
+/**
+ * start streaming replication changes to the client.
+ * this runs indefinitely until the connection is closed.
+ */
+export declare function handleStartReplication(query: string, writer: ReplicationWriter, db: PGlite): Promise<void>;
+export { buildErrorResponse };
+//# sourceMappingURL=handler.d.ts.map

package/dist/replication/handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../src/replication/handler.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAA;AAiBlD,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,IAAI,EAAE,UAAU,GAAG,IAAI,CAAA;CAC9B;AA4HD,iBAAS,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,UAAU,CAqBvD;AAED;;;GAGG;AACH,wBAAsB,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,CAoDlG;AAED;;;GAGG;AACH,wBAAsB,sBAAsB,CAC1C,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,iBAAiB,EACzB,EAAE,EAAE,MAAM,GACT,OAAO,CAAC,IAAI,CAAC,CAgFf;AAyDD,OAAO,EAAE,kBAAkB,EAAE,CAAA"}