mastra-pg-pubsub 0.1.0

package/dist/postgres-pubsub.js

@@ -0,0 +1,499 @@
+import { createHash, randomUUID } from 'node:crypto';
+import { PubSub } from '@mastra/core/events';
+import pg from 'pg';
+import { ConsumeLoop } from "./consume-loop.js";
+import { NotifyListener } from "./listener.js";
+import { runMigration } from "./schema.js";
+import { assertValidSchema, notifyChannel, quoteIdentifier } from "./sql.js";
+const DEFAULT_SCHEMA = 'mastra_pubsub';
+function mapKey(topic, subscriptionId) {
+    return JSON.stringify([topic, subscriptionId]);
+}
+function hashPart(value) {
+    return createHash('sha256').update(value).digest('base64url');
+}
+function groupSubscriptionId(topic, group) {
+    return `__group:${hashPart(topic)}:${hashPart(group)}`;
+}
+function resolveConfig(config) {
+    const logger = config.logger ?? {};
+    const schema = config.schema ?? DEFAULT_SCHEMA;
+    assertValidSchema(schema);
+    let maxDeliveryAttempts = config.maxDeliveryAttempts ?? 5;
+    if (maxDeliveryAttempts === 0) {
+        logger.warn?.('maxDeliveryAttempts=0 is treated as Infinity (unbounded redelivery)');
+        maxDeliveryAttempts = Number.POSITIVE_INFINITY;
+    }
+    return {
+        schema,
+        pollIntervalMs: config.pollIntervalMs ?? 1000,
+        ackDeadlineMs: config.ackDeadlineMs ?? 30_000,
+        nackDelayMs: config.nackDelayMs ?? 0,
+        maxDeliveryAttempts,
+        batchSize: config.batchSize ?? 32,
+        maxEventsPerTopic: config.maxEventsPerTopic ?? 10_000,
+        cleanupIntervalMs: config.cleanupIntervalMs ?? 60_000,
+        staleSubscriptionMs: config.staleSubscriptionMs ?? 300_000,
+        listen: config.listen ?? true,
+        deadLetter: config.deadLetter ?? false,
+        logger,
+    };
+}
+/**
+ * A PostgreSQL-backed {@link PubSub} implementation for Mastra.
+ *
+ * Provides at-least-once delivery with ack/nack and visibility timeouts,
+ * competing-consumer groups and groupless fan-out, replay addressed by
+ * per-topic `index`, optional dead-lettering, and low-latency `LISTEN/NOTIFY`
+ * wakeups with polling as the correctness backstop. The only runtime
+ * dependency is `pg`; `@mastra/core` is a peer dependency.
+ */
+export class PostgresPubSub extends PubSub {
+    #config;
+    #pool;
+    #ownsPool;
+    #logger;
+    #instanceId;
+    #channel;
+    #subscriptions = new Map();
+    #cbIndex = new WeakMap();
+    #listener;
+    #migrated;
+    #maintenanceTimer;
+    #closed = false;
+    #pendingPublishes = new Set();
+    /**
+     * @param config - Adapter configuration. Provide exactly one of
+     *   `connectionString` or `pool`.
+     */
+    constructor(config) {
+        super();
+        this.#config = resolveConfig(config);
+        this.#logger = this.#config.logger;
+        if (config.pool && config.connectionString) {
+            throw new Error('Provide either connectionString or pool, not both');
+        }
+        if (config.pool) {
+            this.#pool = config.pool;
+            this.#ownsPool = false;
+        }
+        else if (config.connectionString) {
+            this.#pool = new pg.Pool({ connectionString: config.connectionString });
+            this.#ownsPool = true;
+        }
+        else {
+            throw new Error('Either connectionString or pool is required');
+        }
+        this.#instanceId = randomUUID();
+        this.#channel = notifyChannel(this.#config.schema);
+    }
+    #q(table) {
+        return `${quoteIdentifier(this.#config.schema)}.${quoteIdentifier(table)}`;
+    }
+    /**
+     * Create the schema and tables explicitly. Idempotent and safe to call
+     * concurrently across instances (serialized by an advisory lock). When not
+     * called, migration happens lazily on first use.
+     */
+    async migrate() {
+        if (!this.#migrated) {
+            this.#migrated = runMigration(this.#pool, this.#config.schema, this.#config.deadLetter);
+        }
+        await this.#migrated;
+    }
+    async #ensureReady() {
+        if (this.#closed) {
+            throw new Error('PostgresPubSub is closed');
+        }
+        await this.migrate();
+        this.#startMaintenance();
+    }
+    /**
+     * Publish an event to a topic. Assigns `id`, `createdAt`, and a per-topic
+     * `index` inside a single transaction, fans out one delivery row per active
+     * subscription, and emits a `NOTIFY` wakeup.
+     *
+     * @param topic - The topic to publish to.
+     * @param event - Event without `id`/`createdAt` (assigned here).
+     * @param _options - Accepted for interface compatibility; `localOnly` is
+     *   ignored because delivery is always database-mediated.
+     */
+    async publish(topic, event, _options) {
+        await this.#ensureReady();
+        const task = this.#publish(topic, event);
+        this.#pendingPublishes.add(task);
+        try {
+            await task;
+        }
+        finally {
+            this.#pendingPublishes.delete(task);
+        }
+    }
+    async #publish(topic, event) {
+        const id = randomUUID();
+        const client = await this.#pool.connect();
+        try {
+            await client.query('BEGIN');
+            const indexResult = await client.query(`INSERT INTO ${this.#q('topics')} (topic, next_index)
+         VALUES ($1, 1)
+         ON CONFLICT (topic) DO UPDATE SET next_index = ${this.#q('topics')}.next_index + 1
+         RETURNING next_index`, [topic]);
+            const index = BigInt(indexResult.rows[0]?.next_index ?? '1') - 1n;
+            const eventResult = await client.query(`INSERT INTO ${this.#q('events')} (id, topic, index, type, run_id, data)
+         VALUES ($1, $2, $3, $4, $5, $6)
+         RETURNING seq`, [
+                id,
+                topic,
+                index.toString(),
+                event.type,
+                event.runId,
+                event.data === undefined ? null : JSON.stringify(event.data),
+            ]);
+            const seq = eventResult.rows[0]?.seq;
+            await client.query(`INSERT INTO ${this.#q('deliveries')} (event_seq, subscription_id)
+         SELECT $1, s.id FROM ${this.#q('subscriptions')} s WHERE s.topic = $2`, [seq, topic]);
+            await client.query(`SELECT pg_notify($1, $2)`, [this.#channel, topic]);
+            await client.query('COMMIT');
+        }
+        catch (error) {
+            await client.query('ROLLBACK').catch(() => undefined);
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+        this.#wakeLocal(topic);
+    }
+    #wakeLocal(topic) {
+        for (const sub of this.#subscriptions.values()) {
+            if (sub.topic === topic) {
+                sub.loop.wake();
+            }
+        }
+    }
+    /**
+     * Subscribe to a topic. With `options.group`, members compete and each event
+     * reaches exactly one member (round-robin across local callbacks). Without a
+     * group, the subscription is private to this instance and receives every
+     * event (fan-out).
+     *
+     * @param topic - The topic to subscribe to.
+     * @param cb - Callback invoked per delivered event.
+     * @param options - Subscribe options; `group` selects competing-consumer
+     *   semantics. `batch` is ignored (no native batching).
+     */
+    async subscribe(topic, cb, options) {
+        await this.#subscribeInternal(topic, cb, cb, options?.group);
+    }
+    /**
+     * Shared subscribe path. `userKey` is the callback the caller will later
+     * pass to `unsubscribe`; `registered` is the callback actually invoked on
+     * delivery (they differ only for the replay wrappers).
+     */
+    async #subscribeInternal(topic, userKey, registered, group, startImmediately = true) {
+        await this.#ensureReady();
+        const subscriptionId = group
+            ? groupSubscriptionId(topic, group)
+            : `__private:${this.#instanceId}:${randomUUID()}`;
+        const key = mapKey(topic, subscriptionId);
+        let sub = this.#subscriptions.get(key);
+        if (!sub) {
+            const isGroup = group !== undefined;
+            await this.#upsertSubscription(subscriptionId, topic, isGroup);
+            const registry = { callbacks: [], cursor: 0 };
+            const loop = new ConsumeLoop(this.#pool, this.#config.schema, this.#config, topic, subscriptionId, isGroup, registry);
+            sub = {
+                id: subscriptionId,
+                topic,
+                isGroup,
+                registry,
+                loop,
+                started: false,
+                unregisterWake: undefined,
+            };
+            this.#subscriptions.set(key, sub);
+        }
+        sub.registry.callbacks.push(registered);
+        let registrations = this.#cbIndex.get(userKey);
+        if (!registrations) {
+            registrations = [];
+            this.#cbIndex.set(userKey, registrations);
+        }
+        registrations.push({ sub, registered });
+        if (startImmediately) {
+            await this.#startSubscription(sub);
+            sub.loop.wake();
+        }
+        return sub;
+    }
+    async #startSubscription(sub) {
+        if (sub.started) {
+            return;
+        }
+        sub.loop.start();
+        sub.started = true;
+        if (this.#config.listen) {
+            await this.#registerWake(sub);
+        }
+    }
+    async #registerWake(sub) {
+        if (!this.#listener) {
+            this.#listener = new NotifyListener(this.#pool, this.#config.schema, this.#logger);
+        }
+        sub.unregisterWake = await this.#listener.register(sub.topic, () => sub.loop.wake());
+    }
+    async #upsertSubscription(id, topic, isGroup) {
+        await this.#pool.query(`INSERT INTO ${this.#q('subscriptions')} (id, topic, is_group, last_seen_at)
+       VALUES ($1, $2, $3, now())
+       ON CONFLICT (id) DO UPDATE SET last_seen_at = now()`, [id, topic, isGroup]);
+    }
+    /**
+     * Remove a callback from a topic. Tears down the underlying subscription
+     * (and, for private subscriptions, deletes its database rows) once no local
+     * callbacks remain.
+     *
+     * @param topic - The topic the callback was subscribed to.
+     * @param cb - The callback to remove.
+     */
+    async unsubscribe(topic, cb) {
+        const registrations = this.#cbIndex.get(cb);
+        if (!registrations) {
+            return;
+        }
+        const remaining = [];
+        for (const reg of registrations) {
+            if (reg.sub.topic !== topic) {
+                remaining.push(reg);
+                continue;
+            }
+            const idx = reg.sub.registry.callbacks.indexOf(reg.registered);
+            if (idx !== -1) {
+                reg.sub.registry.callbacks.splice(idx, 1);
+            }
+            if (reg.sub.registry.callbacks.length === 0) {
+                await this.#teardownSubscription(reg.sub);
+            }
+        }
+        if (remaining.length === 0) {
+            this.#cbIndex.delete(cb);
+        }
+        else {
+            this.#cbIndex.set(cb, remaining);
+        }
+    }
+    async #teardownSubscription(sub) {
+        const key = mapKey(sub.topic, sub.id);
+        this.#subscriptions.delete(key);
+        sub.unregisterWake?.();
+        await sub.loop.stop();
+        if (!sub.isGroup) {
+            await this.#pool
+                .query(`DELETE FROM ${this.#q('subscriptions')} WHERE id = $1`, [sub.id])
+                .catch((error) => this.#logger.warn?.('failed to delete private subscription', error));
+        }
+    }
+    /**
+     * Resolve once all in-flight publishes and local deliveries have settled.
+     * Per-event callback errors are logged, never thrown. If locally-owned
+     * deliveries remain unsettled after the bounded drain window, reject so
+     * shutdown callers do not mistake a stuck subscriber for a clean drain.
+     */
+    async flush() {
+        await Promise.allSettled([...this.#pendingPublishes]);
+        let pending = 0;
+        for (let pass = 0; pass < 50; pass++) {
+            await Promise.all([...this.#subscriptions.values()].map((s) => s.loop.drain()));
+            pending = await this.#countPending();
+            if (pending === 0) {
+                return;
+            }
+            await new Promise((resolve) => setTimeout(resolve, 10));
+        }
+        throw new Error(`PostgresPubSub flush timed out with ${pending} unsettled deliveries`);
+    }
+    async #countPending() {
+        const ids = [...this.#subscriptions.values()].map((s) => s.id);
+        if (ids.length === 0) {
+            return 0;
+        }
+        const result = await this.#pool.query(`SELECT count(*)::text AS count FROM ${this.#q('deliveries')}
+       WHERE subscription_id = ANY($1::text[])`, [ids]);
+        return Number(result.rows[0]?.count ?? '0');
+    }
+    /**
+     * Fetch historical events for a topic with `index >= offset`, ordered by
+     * `index`.
+     *
+     * @param topic - The topic to read history for.
+     * @param offset - Inclusive starting `index`. Defaults to `0`.
+     * @returns Events in ascending `index` order.
+     */
+    async getHistory(topic, offset = 0) {
+        await this.#ensureReady();
+        const result = await this.#pool.query(`SELECT id, topic, index, type, run_id, data, created_at
+       FROM ${this.#q('events')}
+       WHERE topic = $1 AND index >= $2
+       ORDER BY index`, [topic, offset]);
+        return result.rows.map(rowToEvent);
+    }
+    /**
+     * Subscribe and replay all history first, then continue live, deduping the
+     * boundary by event `index` so no event is delivered twice.
+     *
+     * @param topic - The topic to subscribe to.
+     * @param cb - Callback for replayed and live events.
+     */
+    async subscribeWithReplay(topic, cb) {
+        await this.subscribeFromOffset(topic, 0, cb);
+    }
+    /**
+     * Subscribe and replay history starting at `offset`, then continue live,
+     * deduping the boundary by event `index`.
+     *
+     * @param topic - The topic to subscribe to.
+     * @param offset - Inclusive starting `index` for replay.
+     * @param cb - Callback for replayed and live events.
+     */
+    async subscribeFromOffset(topic, offset, cb) {
+        const seen = new Set();
+        let live = false;
+        const wrapped = (event, ack, nack) => {
+            if (live && event.index !== undefined) {
+                seen.add(event.index);
+            }
+            return cb(event, ack, nack);
+        };
+        // Create the live subscription row first so events published during replay
+        // get delivery rows, but keep the consume loop paused until historical
+        // callbacks run in order. Replayed deliveries are then settled for this
+        // private subscription to avoid duplicate live delivery at the boundary.
+        const sub = await this.#subscribeInternal(topic, cb, wrapped, undefined, false);
+        try {
+            const replayedIndexes = [];
+            const history = await this.getHistory(topic, offset);
+            for (const event of history) {
+                if (event.index !== undefined && seen.has(event.index)) {
+                    continue;
+                }
+                try {
+                    await cb(event);
+                    if (event.index !== undefined) {
+                        replayedIndexes.push(event.index);
+                    }
+                }
+                catch (error) {
+                    this.#logger.error?.('replay callback threw', error);
+                }
+            }
+            await this.#ackReplayedDeliveries(topic, sub.id, replayedIndexes);
+            live = true;
+            await this.#startSubscription(sub);
+            sub.loop.wake();
+        }
+        catch (error) {
+            await this.unsubscribe(topic, cb).catch((teardownError) => {
+                this.#logger.warn?.('failed to clean up replay subscription after setup failure', teardownError);
+            });
+            throw error;
+        }
+    }
+    async #ackReplayedDeliveries(topic, subscriptionId, indexes) {
+        if (indexes.length === 0) {
+            return;
+        }
+        await this.#pool.query(`DELETE FROM ${this.#q('deliveries')} d
+       USING ${this.#q('events')} e
+       WHERE d.event_seq = e.seq
+         AND d.subscription_id = $1
+         AND e.topic = $2
+         AND e.index = ANY($3::bigint[])`, [subscriptionId, topic, indexes]);
+    }
+    #startMaintenance() {
+        if (this.#maintenanceTimer || this.#config.cleanupIntervalMs <= 0 || this.#closed) {
+            return;
+        }
+        this.#maintenanceTimer = setInterval(() => {
+            this.#runMaintenance().catch((error) => this.#logger.warn?.('maintenance cycle failed', error));
+        }, this.#config.cleanupIntervalMs);
+        this.#maintenanceTimer.unref?.();
+    }
+    async #runMaintenance() {
+        await this.#heartbeat();
+        await this.#pruneStaleSubscriptions();
+        await this.#trimRetention();
+    }
+    async #heartbeat() {
+        const ids = [...this.#subscriptions.values()].filter((s) => !s.isGroup).map((s) => s.id);
+        if (ids.length === 0) {
+            return;
+        }
+        await this.#pool.query(`UPDATE ${this.#q('subscriptions')} SET last_seen_at = now() WHERE id = ANY($1::text[])`, [ids]);
+    }
+    async #pruneStaleSubscriptions() {
+        await this.#pool.query(`DELETE FROM ${this.#q('subscriptions')}
+       WHERE is_group = false
+         AND id LIKE '__private:%'
+         AND last_seen_at < now() - ($1::double precision * interval '1 millisecond')`, [this.#config.staleSubscriptionMs]);
+    }
+    async #trimRetention() {
+        if (this.#config.maxEventsPerTopic <= 0) {
+            return;
+        }
+        await this.#pool.query(`WITH ranked AS (
+         SELECT e.seq,
+                row_number() OVER (PARTITION BY e.topic ORDER BY e.index DESC) AS rn
+         FROM ${this.#q('events')} e
+       )
+       DELETE FROM ${this.#q('events')} e
+       USING ranked
+       WHERE e.seq = ranked.seq
+         AND ranked.rn > $1
+         AND NOT EXISTS (
+           SELECT 1 FROM ${this.#q('deliveries')} d WHERE d.event_seq = e.seq
+         )`, [this.#config.maxEventsPerTopic]);
+    }
+    /**
+     * Stop all loops, release the listener, delete this instance's private
+     * subscriptions, and end the pool when the adapter created it. Idempotent.
+     */
+    async close() {
+        if (this.#closed) {
+            return;
+        }
+        this.#closed = true;
+        if (this.#maintenanceTimer) {
+            clearInterval(this.#maintenanceTimer);
+            this.#maintenanceTimer = undefined;
+        }
+        await Promise.allSettled([...this.#pendingPublishes]);
+        for (const sub of this.#subscriptions.values()) {
+            sub.unregisterWake?.();
+        }
+        await Promise.all([...this.#subscriptions.values()].map((s) => s.loop.stop()));
+        if (this.#listener) {
+            await this.#listener.close();
+            this.#listener = undefined;
+        }
+        const privateIds = [...this.#subscriptions.values()].filter((s) => !s.isGroup).map((s) => s.id);
+        this.#subscriptions.clear();
+        if (privateIds.length > 0) {
+            await this.#pool
+                .query(`DELETE FROM ${this.#q('subscriptions')} WHERE id = ANY($1::text[])`, [privateIds])
+                .catch((error) => this.#logger.warn?.('failed to delete private subscriptions', error));
+        }
+        if (this.#ownsPool) {
+            await this.#pool.end();
+        }
+    }
+}
+function rowToEvent(row) {
+    return {
+        id: row.id,
+        type: row.type,
+        data: row.data,
+        runId: row.run_id,
+        createdAt: row.created_at,
+        index: Number(row.index),
+    };
+}
+//# sourceMappingURL=postgres-pubsub.js.map

package/dist/postgres-pubsub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres-pubsub.js","sourceRoot":"","sources":["../src/postgres-pubsub.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAyB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAG7E,MAAM,cAAc,GAAG,eAAe,CAAC;AA4BvC,SAAS,MAAM,CAAC,KAAa,EAAE,cAAsB;IACnD,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC;AACjD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa;IAC7B,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAa,EAAE,KAAa;IACvD,OAAO,WAAW,QAAQ,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,aAAa,CAAC,MAA4B;IACjD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;IACnC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,cAAc,CAAC;IAC/C,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAE1B,IAAI,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,IAAI,CAAC,CAAC;IAC1D,IAAI,mBAAmB,KAAK,CAAC,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,EAAE,CAAC,qEAAqE,CAAC,CAAC;QACrF,mBAAmB,GAAG,MAAM,CAAC,iBAAiB,CAAC;IACjD,CAAC;IAED,OAAO;QACL,MAAM;QACN,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI;QAC7C,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,MAAM;QAC7C,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC;QACpC,mBAAmB;QACnB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,EAAE;QACjC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,MAAM;QACrD,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,MAAM;QACrD,mBAAmB,EAAE,MAAM,CAAC,mBAAmB,IAAI,OAAO;QAC1D,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI;QAC7B,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,KAAK;QACtC,MAAM;KACP,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,OAAO,cAAe,SAAQ,MAAM;IAC/B,OAAO,CAAiB;IACxB,KAAK,CAAU;IACf,SAAS,CAAU;IACnB,OAAO,CAAe;IACtB,WAAW,CAAS;IACpB,QAAQ,CAAS;IAEjB,cAAc,GAAG,IAAI,GAAG,EAAwB,CAAC;IACjD,QAAQ,GAAG,IAAI,OAAO,EAAiC,CAAC;IAEjE,SAAS,CAA6B;IACtC,SAAS,CAA4B;IACrC,iBAAiB,CAA6B;IAC9C,OAAO,GAAG,KAAK,CAAC;IAChB,iBAAiB,GAAG,IAAI,GAAG,EAAiB,CAAC;IAE7C;;;OAGG;IACH,YAAY,MAA4B;QACtC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QAEnC,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC3C,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC;YACzB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC;aAAM,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;YACnC,IAAI,CAAC,KAAK,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC;YACxE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,EAAE,CAAC;QAChC,IAAI,CAAC,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrD,CAAC;IAED,EAAE,CAAC,KAAa;QACd,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;IAC7E,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC1F,CAAC;QACD,MAAM,IAAI,CAAC,SAAS,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAED;;;;;;;;;OASG;IACM,KAAK,CAAC,OAAO,CACpB,KAAa,EACb,KAAsC,EACtC,QAAkC;QAElC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC;QACb,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,KAAsC;QAClE,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5B,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,KAAK,CACpC,eAAe,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;;0DAEkB,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;8BAC7C,EACtB,CAAC,KAAK,CAAC,CACR,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,GAAG,CAAC,GAAG,EAAE,CAAC;YAElE,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,KAAK,CACpC,eAAe,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;;uBAEjB,EACf;gBACE,EAAE;gBACF,KAAK;gBACL,KAAK,CAAC,QAAQ,EAAE;gBAChB,KAAK,CAAC,IAAI;gBACV,KAAK,CAAC,KAAK;gBACX,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC;aAC7D,CACF,CAAC;YACF,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC;YAErC,MAAM,MAAM,CAAC,KAAK,CAChB,eAAe,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gCACZ,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,uBAAuB,EACvE,CAAC,GAAG,EAAE,KAAK,CAAC,CACb,CAAC;YAEF,MAAM,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,IAAI,GAAG,CAAC,KAAK,KAAK,KAAK,EAAE,CAAC;gBACxB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;;;OAUG;IACM,KAAK,CAAC,SAAS,CACtB,KAAa,EACb,EAAiB,EACjB,OAA0B;QAE1B,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;IAC/D,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,kBAAkB,CACtB,KAAa,EACb,OAAsB,EACtB,UAAyB,EACzB,KAAyB,EACzB,gBAAgB,GAAG,IAAI;QAEvB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,MAAM,cAAc,GAAG,KAAK;YAC1B,CAAC,CAAC,mBAAmB,CAAC,KAAK,EAAE,KAAK,CAAC;YACnC,CAAC,CAAC,aAAa,IAAI,CAAC,WAAW,IAAI,UAAU,EAAE,EAAE,CAAC;QACpD,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAE1C,IAAI,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,OAAO,GAAG,KAAK,KAAK,SAAS,CAAC;YACpC,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YAC/D,MAAM,QAAQ,GAAqB,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,GAAG,IAAI,WAAW,CAC1B,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,CAAC,MAAM,EACnB,IAAI,CAAC,OAAO,EACZ,KAAK,EACL,cAAc,EACd,OAAO,EACP,QAAQ,CACT,CAAC;YACF,GAAG,GAAG;gBACJ,EAAE,EAAE,cAAc;gBAClB,KAAK;gBACL,OAAO;gBACP,QAAQ;gBACR,IAAI;gBACJ,OAAO,EAAE,KAAK;gBACd,cAAc,EAAE,SAAS;aAC1B,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACpC,CAAC;QAED,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxC,IAAI,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,aAAa,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAC5C,CAAC;QACD,aAAa,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,CAAC;QACxC,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,GAAiB;QACxC,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACjB,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC;QACnB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAiB;QACnC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACrF,CAAC;QACD,GAAG,CAAC,cAAc,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACvF,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,EAAU,EAAE,KAAa,EAAE,OAAgB;QACnE,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACpB,eAAe,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC;;2DAEc,EACrD,CAAC,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,CACrB,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACM,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,EAAiB;QACzD,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAmB,EAAE,CAAC;QACrC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,KAAK,KAAK,EAAE,CAAC;gBAC5B,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACpB,SAAS;YACX,CAAC;YACD,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC/D,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;gBACf,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC5C,MAAM,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,GAAiB;QAC3C,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAChC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC;QACvB,MAAM,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACtB,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,KAAK;iBACb,KAAK,CAAC,eAAe,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,gBAAgB,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;iBACxE,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACM,KAAK,CAAC,KAAK;QAClB,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACtD,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;YACrC,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAChF,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YACrC,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;gBAClB,OAAO;YACT,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1D,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,uCAAuC,OAAO,uBAAuB,CAAC,CAAC;IACzF,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC/D,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO,CAAC,CAAC;QACX,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACnC,uCAAuC,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;+CACnB,EACzC,CAAC,GAAG,CAAC,CACN,CAAC;QACF,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,GAAG,CAAC,CAAC;IAC9C,CAAC;IAED;;;;;;;OAOG;IACM,KAAK,CAAC,UAAU,CAAC,KAAa,EAAE,MAAM,GAAG,CAAC;QACjD,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACnC;cACQ,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;;sBAET,EAChB,CAAC,KAAK,EAAE,MAAM,CAAC,CAChB,CAAC;QACF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACrC,CAAC;IAED;;;;;;OAMG;IACM,KAAK,CAAC,mBAAmB,CAAC,KAAa,EAAE,EAAiB;QACjE,MAAM,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;;;OAOG;IACM,KAAK,CAAC,mBAAmB,CAChC,KAAa,EACb,MAAc,EACd,EAAiB;QAEjB,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/B,IAAI,IAAI,GAAG,KAAK,CAAC;QACjB,MAAM,OAAO,GAAkB,CAAC,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAClD,IAAI,IAAI,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBACtC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;QAC9B,CAAC,CAAC;QAEF,2EAA2E;QAC3E,uEAAuE;QACvE,wEAAwE;QACxE,yEAAyE;QACzE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAEhF,IAAI,CAAC;YACH,MAAM,eAAe,GAAa,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACrD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;oBACvD,SAAS;gBACX,CAAC;gBACD,IAAI,CAAC;oBACH,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC;oBAChB,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;wBAC9B,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACpC,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;YACD,MAAM,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,EAAE,eAAe,CAAC,CAAC;YAElE,IAAI,GAAG,IAAI,CAAC;YACZ,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,EAAE;gBACxD,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CACjB,4DAA4D,EAC5D,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,KAAa,EACb,cAAsB,EACtB,OAAiB;QAEjB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACpB,eAAe,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;eAC3B,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;;;;yCAIS,EACnC,CAAC,cAAc,EAAE,KAAK,EAAE,OAAO,CAAC,CACjC,CAAC;IACJ,CAAC;IAED,iBAAiB;QACf,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,OAAO,CAAC,iBAAiB,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CACrC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,0BAA0B,EAAE,KAAK,CAAC,CACvD,CAAC;QACJ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACnC,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC;QACtC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACzF,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACpB,UAAU,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,sDAAsD,EACxF,CAAC,GAAG,CAAC,CACN,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACpB,eAAe,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC;;;sFAGyC,EAChF,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CACnC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,IAAI,CAAC,OAAO,CAAC,iBAAiB,IAAI,CAAC,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CACpB;;;gBAGU,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;;qBAEZ,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC;;;;;2BAKX,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;WACrC,EACL,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,CACjC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACrC,CAAC;QAED,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAEtD,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC;QACzB,CAAC;QACD,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAE/E,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAC7B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAChG,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,CAAC,KAAK;iBACb,KAAK,CAAC,eAAe,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,UAAU,CAAC,CAAC;iBACzF,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC,CAAC;QAC5F,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACzB,CAAC;IACH,CAAC;CACF;AAED,SAAS,UAAU,CAAC,GAAa;IAC/B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,KAAK,EAAE,GAAG,CAAC,MAAM;QACjB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;KACzB,CAAC;AACJ,CAAC"}

package/dist/schema.d.ts

@@ -0,0 +1,21 @@
+import type { Pool } from 'pg';
+/**
+ * Build the ordered list of DDL statements that create the schema and all
+ * tables. Idempotent: every statement uses `IF NOT EXISTS`.
+ *
+ * @param schema - The validated, unquoted schema name.
+ * @param deadLetter - Whether to include the optional `dead_events` table.
+ * @returns Ordered DDL statements to execute within one transaction.
+ */
+export declare function buildDdl(schema: string, deadLetter: boolean): string[];
+/**
+ * Create the schema and tables under a transaction-scoped advisory lock so
+ * concurrent instances serialize their migrations. Idempotent and safe to call
+ * repeatedly.
+ *
+ * @param pool - The connection pool to run the migration on.
+ * @param schema - The validated schema name.
+ * @param deadLetter - Whether to include the optional `dead_events` table.
+ */
+export declare function runMigration(pool: Pool, schema: string, deadLetter: boolean): Promise<void>;
+//# sourceMappingURL=schema.d.ts.map

package/dist/schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../src/schema.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,IAAI,CAAC;AAG3C;;;;;;;GAOG;AACH,wBAAgB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,GAAG,MAAM,EAAE,CA6DtE;AAED;;;;;;;;GAQG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAiBjG"}

package/dist/schema.js

@@ -0,0 +1,97 @@
+import { advisoryLockKey, assertValidSchema, quoteIdentifier } from "./sql.js";
+/**
+ * Build the ordered list of DDL statements that create the schema and all
+ * tables. Idempotent: every statement uses `IF NOT EXISTS`.
+ *
+ * @param schema - The validated, unquoted schema name.
+ * @param deadLetter - Whether to include the optional `dead_events` table.
+ * @returns Ordered DDL statements to execute within one transaction.
+ */
+export function buildDdl(schema, deadLetter) {
+    const s = quoteIdentifier(schema);
+    const statements = [
+        `CREATE SCHEMA IF NOT EXISTS ${s}`,
+        `CREATE TABLE IF NOT EXISTS ${s}.topics (
+       topic TEXT PRIMARY KEY,
+       next_index BIGINT NOT NULL DEFAULT 0
+     )`,
+        `CREATE TABLE IF NOT EXISTS ${s}.events (
+       seq BIGSERIAL PRIMARY KEY,
+       id UUID NOT NULL UNIQUE,
+       topic TEXT NOT NULL,
+       index BIGINT NOT NULL,
+       type TEXT NOT NULL,
+       run_id TEXT NOT NULL,
+       data JSONB,
+       created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+       UNIQUE (topic, index)
+     )`,
+        `CREATE INDEX IF NOT EXISTS events_topic_index_idx
+       ON ${s}.events (topic, index)`,
+        `CREATE TABLE IF NOT EXISTS ${s}.subscriptions (
+       id TEXT PRIMARY KEY,
+       topic TEXT NOT NULL,
+       is_group BOOLEAN NOT NULL,
+       last_seen_at TIMESTAMPTZ NOT NULL DEFAULT now()
+     )`,
+        `CREATE INDEX IF NOT EXISTS subscriptions_topic_idx
+       ON ${s}.subscriptions (topic)`,
+        `CREATE TABLE IF NOT EXISTS ${s}.deliveries (
+       event_seq BIGINT NOT NULL REFERENCES ${s}.events (seq) ON DELETE CASCADE,
+       subscription_id TEXT NOT NULL REFERENCES ${s}.subscriptions (id) ON DELETE CASCADE,
+       delivery_attempt INT NOT NULL DEFAULT 0,
+       visible_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+       PRIMARY KEY (subscription_id, event_seq)
+     )`,
+        `CREATE INDEX IF NOT EXISTS deliveries_claim_idx
+       ON ${s}.deliveries (subscription_id, visible_at, event_seq)`,
+        `CREATE INDEX IF NOT EXISTS deliveries_event_idx
+       ON ${s}.deliveries (event_seq)`,
+    ];
+    if (deadLetter) {
+        statements.push(`CREATE TABLE IF NOT EXISTS ${s}.dead_events (
+         seq BIGSERIAL PRIMARY KEY,
+         event_id UUID NOT NULL,
+         subscription_id TEXT NOT NULL,
+         topic TEXT NOT NULL,
+         index BIGINT NOT NULL,
+         type TEXT NOT NULL,
+         run_id TEXT NOT NULL,
+         data JSONB,
+         created_at TIMESTAMPTZ NOT NULL,
+         delivery_attempt INT NOT NULL,
+         dead_at TIMESTAMPTZ NOT NULL DEFAULT now()
+       )`);
+    }
+    return statements;
+}
+/**
+ * Create the schema and tables under a transaction-scoped advisory lock so
+ * concurrent instances serialize their migrations. Idempotent and safe to call
+ * repeatedly.
+ *
+ * @param pool - The connection pool to run the migration on.
+ * @param schema - The validated schema name.
+ * @param deadLetter - Whether to include the optional `dead_events` table.
+ */
+export async function runMigration(pool, schema, deadLetter) {
+    assertValidSchema(schema);
+    const key = advisoryLockKey(schema);
+    const client = await pool.connect();
+    try {
+        await client.query('BEGIN');
+        await client.query('SELECT pg_advisory_xact_lock($1)', [key]);
+        for (const statement of buildDdl(schema, deadLetter)) {
+            await client.query(statement);
+        }
+        await client.query('COMMIT');
+    }
+    catch (error) {
+        await client.query('ROLLBACK').catch(() => undefined);
+        throw error;
+    }
+    finally {
+        client.release();
+    }
+}
+//# sourceMappingURL=schema.js.map

package/dist/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../src/schema.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE/E;;;;;;;GAOG;AACH,MAAM,UAAU,QAAQ,CAAC,MAAc,EAAE,UAAmB;IAC1D,MAAM,CAAC,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,UAAU,GAAa;QAC3B,+BAA+B,CAAC,EAAE;QAClC,8BAA8B,CAAC;;;OAG5B;QACH,8BAA8B,CAAC;;;;;;;;;;OAU5B;QACH;YACQ,CAAC,wBAAwB;QACjC,8BAA8B,CAAC;;;;;OAK5B;QACH;YACQ,CAAC,wBAAwB;QACjC,8BAA8B,CAAC;8CACW,CAAC;kDACG,CAAC;;;;OAI5C;QACH;YACQ,CAAC,sDAAsD;QAC/D;YACQ,CAAC,yBAAyB;KACnC,CAAC;IAEF,IAAI,UAAU,EAAE,CAAC;QACf,UAAU,CAAC,IAAI,CACb,8BAA8B,CAAC;;;;;;;;;;;;SAY5B,CACJ,CAAC;IACJ,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,IAAU,EAAE,MAAc,EAAE,UAAmB;IAChF,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC1B,MAAM,GAAG,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,MAAM,GAAe,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;IAChD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC5B,MAAM,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9D,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,CAAC;YACrD,MAAM,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;QACD,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,KAAK,CAAC;IACd,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;AACH,CAAC"}