orpc-fn 0.1.0

package/dist/live/redis-bun.js

@@ -0,0 +1,59 @@
+import { createRedisTransport } from './redis.js';
+/**
+ * `PubSubTransport` for Bun's built-in `RedisClient`. Subscriptions use a
+ * dedicated connection (`client.duplicate()` unless `subscriber` is given)
+ * that re-subscribes its channels after every reconnect; configure retries on
+ * the client (`maxRetries`) since the duplicate copies its options.
+ *
+ * Call `close()` on shutdown to close the subscriber connection it created.
+ */
+export function bunRedisTransport(client, options = {}) {
+    return createRedisTransport({
+        send: (command, args) => client.send(command, args),
+        async connectSubscriber(handlers) {
+            const owned = !options.subscriber;
+            const subscriber = options.subscriber ?? (await client.duplicate());
+            const dispatch = (message, channel) => handlers.onMessage(channel, message);
+            const ours = new Set();
+            // Bun reconnects on its own but drops every subscription. Also fires on
+            // the first connect, when there is nothing to restore yet.
+            subscriber.onconnect = () => handlers.onReconnect();
+            // Fires only when Bun gives up reconnecting (and on close(), which
+            // detaches it first).
+            subscriber.onclose = (error) => handlers.onLost(error ?? new Error('Redis subscriber closed'));
+            return {
+                subscribe: (channel) => {
+                    ours.add(channel);
+                    return subscriber.subscribe(channel, dispatch);
+                },
+                // Bun keeps listeners across a reconnect and does not dedupe them, so
+                // clear the channel before adding its listener again. unsubscribe
+                // throws synchronously outside subscriber mode.
+                resubscribe: async (channel) => {
+                    await Promise.resolve()
+                        .then(() => subscriber.unsubscribe(channel))
+                        .catch(() => { });
+                    return subscriber.subscribe(channel, dispatch);
+                },
+                unsubscribe: (channel) => {
+                    ours.delete(channel);
+                    return subscriber.unsubscribe(channel, dispatch);
+                },
+                close() {
+                    // No-ops, not null: Bun calls `onclose` without checking it.
+                    subscriber.onconnect = () => { };
+                    subscriber.onclose = () => { };
+                    if (owned)
+                        return subscriber.close();
+                    // A given subscriber stays open: remove only this transport's listeners.
+                    for (const channel of ours) {
+                        void Promise.resolve()
+                            .then(() => subscriber.unsubscribe(channel, dispatch))
+                            .catch(() => { });
+                    }
+                }
+            };
+        }
+    });
+}
+//# sourceMappingURL=redis-bun.js.map

package/dist/live/redis-bun.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis-bun.js","sourceRoot":"","sources":["../../src/live/redis-bun.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AAejD;;;;;;;GAOG;AACH,MAAM,UAAU,iBAAiB,CAChC,MAA0B,EAC1B,OAAO,GAAwC,EAAE;IAEjD,OAAO,oBAAoB,CAAC;QAC3B,IAAI,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC;QACnD,KAAK,CAAC,iBAAiB,CAAC,QAAQ;YAC/B,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,UAAU,CAAA;YACjC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC,CAAA;YACnE,MAAM,QAAQ,GAAa,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAC/C,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;YACrC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAA;YAC9B,wEAAwE;YACxE,2DAA2D;YAC3D,UAAU,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAA;YACnD,mEAAmE;YACnE,sBAAsB;YACtB,UAAU,CAAC,OAAO,GAAG,CAAC,KAAK,EAAE,EAAE,CAC9B,QAAQ,CAAC,MAAM,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAA;YAC/D,OAAO;gBACN,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;oBACtB,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;oBACjB,OAAO,UAAU,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAA;gBAC/C,CAAC;gBACD,sEAAsE;gBACtE,kEAAkE;gBAClE,gDAAgD;gBAChD,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;oBAC9B,MAAM,OAAO,CAAC,OAAO,EAAE;yBACrB,IAAI,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;yBAC3C,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;oBACjB,OAAO,UAAU,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAA;gBAC/C,CAAC;gBACD,WAAW,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;oBACpB,OAAO,UAAU,CAAC,WAAW,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAA;gBACjD,CAAC;gBACD,KAAK;oBACJ,6DAA6D;oBAC7D,UAAU,CAAC,SAAS,GAAG,GAAG,EAAE,GAAE,CAAC,CAAA;oBAC/B,UAAU,CAAC,OAAO,GAAG,GAAG,EAAE,GAAE,CAAC,CAAA;oBAC7B,IAAI,KAAK;wBAAE,OAAO,UAAU,CAAC,KAAK,EAAE,CAAA;oBACpC,yEAAyE;oBACzE,KAAK,MAAM,OAAO,IAAI,IAAI,EAAE,CAAC;wBAC5B,KAAK,OAAO,CAAC,OAAO,EAAE;6BACpB,IAAI,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;6BACrD,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;oBAClB,CAAC;gBACF,CAAC;aACD,CAAA;QACF,CAAC;KACD,CAAC,CAAA;AACH,CAAC","sourcesContent":["import { createRedisTransport } from './redis.js'\n\ntype Listener = (message: string, channel: string) => void\n\n/** The part of Bun's `RedisClient` this transport uses. */\nexport interface BunRedisClientLike {\n\tsend(command: string, args: string[]): Promise<unknown>\n\tsubscribe(channel: string, listener: Listener): Promise<unknown>\n\tunsubscribe(channel: string, listener?: Listener): Promise<unknown>\n\tduplicate(): Promise<BunRedisClientLike>\n\tclose(): void\n\tonconnect: ((this: never) => void) | null\n\tonclose: ((this: never, error: Error) => void) | null\n}\n\n/**\n * `PubSubTransport` for Bun's built-in `RedisClient`. Subscriptions use a\n * dedicated connection (`client.duplicate()` unless `subscriber` is given)\n * that re-subscribes its channels after every reconnect; configure retries on\n * the client (`maxRetries`) since the duplicate copies its options.\n *\n * Call `close()` on shutdown to close the subscriber connection it created.\n */\nexport function bunRedisTransport(\n\tclient: BunRedisClientLike,\n\toptions: { subscriber?: BunRedisClientLike } = {}\n) {\n\treturn createRedisTransport({\n\t\tsend: (command, args) => client.send(command, args),\n\t\tasync connectSubscriber(handlers) {\n\t\t\tconst owned = !options.subscriber\n\t\t\tconst subscriber = options.subscriber ?? (await client.duplicate())\n\t\t\tconst dispatch: Listener = (message, channel) =>\n\t\t\t\thandlers.onMessage(channel, message)\n\t\t\tconst ours = new Set<string>()\n\t\t\t// Bun reconnects on its own but drops every subscription. Also fires on\n\t\t\t// the first connect, when there is nothing to restore yet.\n\t\t\tsubscriber.onconnect = () => handlers.onReconnect()\n\t\t\t// Fires only when Bun gives up reconnecting (and on close(), which\n\t\t\t// detaches it first).\n\t\t\tsubscriber.onclose = (error) =>\n\t\t\t\thandlers.onLost(error ?? new Error('Redis subscriber closed'))\n\t\t\treturn {\n\t\t\t\tsubscribe: (channel) => {\n\t\t\t\t\tours.add(channel)\n\t\t\t\t\treturn subscriber.subscribe(channel, dispatch)\n\t\t\t\t},\n\t\t\t\t// Bun keeps listeners across a reconnect and does not dedupe them, so\n\t\t\t\t// clear the channel before adding its listener again. unsubscribe\n\t\t\t\t// throws synchronously outside subscriber mode.\n\t\t\t\tresubscribe: async (channel) => {\n\t\t\t\t\tawait Promise.resolve()\n\t\t\t\t\t\t.then(() => subscriber.unsubscribe(channel))\n\t\t\t\t\t\t.catch(() => {})\n\t\t\t\t\treturn subscriber.subscribe(channel, dispatch)\n\t\t\t\t},\n\t\t\t\tunsubscribe: (channel) => {\n\t\t\t\t\tours.delete(channel)\n\t\t\t\t\treturn subscriber.unsubscribe(channel, dispatch)\n\t\t\t\t},\n\t\t\t\tclose() {\n\t\t\t\t\t// No-ops, not null: Bun calls `onclose` without checking it.\n\t\t\t\t\tsubscriber.onconnect = () => {}\n\t\t\t\t\tsubscriber.onclose = () => {}\n\t\t\t\t\tif (owned) return subscriber.close()\n\t\t\t\t\t// A given subscriber stays open: remove only this transport's listeners.\n\t\t\t\t\tfor (const channel of ours) {\n\t\t\t\t\t\tvoid Promise.resolve()\n\t\t\t\t\t\t\t.then(() => subscriber.unsubscribe(channel, dispatch))\n\t\t\t\t\t\t\t.catch(() => {})\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t})\n}\n"]}

package/dist/live/redis.d.ts

@@ -0,0 +1,45 @@
+import { type PubSubTransport } from './transport.js';
+export type RedisSend = (command: string, args: string[]) => Promise<unknown>;
+/**
+ * Ship a Lua script by SHA instead of resending its source on every call.
+ * Redis forgets its script cache on restart and on `SCRIPT FLUSH`, so a
+ * `NOSCRIPT` reply falls back to an inline `EVAL` (which repopulates the
+ * cache) and drops the stale digest.
+ */
+export declare function createLuaScript(send: RedisSend, source: string): (keys: string[], args: string[]) => Promise<unknown>;
+export type SubscriberConnection = {
+    subscribe(channel: string): Promise<unknown>;
+    unsubscribe(channel: string): Promise<unknown>;
+    /**
+     * Restore one channel after the driver reconnected on its own and lost its
+     * subscriptions. Omit when the client re-subscribes itself (ioredis).
+     */
+    resubscribe?(channel: string): Promise<unknown>;
+    /** Stop delivering and release the connection (if the driver created it). */
+    close(): void;
+};
+/** Callbacks for one connection; they do nothing once it is replaced. */
+export type SubscriberHandlers = {
+    onMessage(channel: string, message: string): void;
+    /** The driver reconnected; its subscriptions must be restored. */
+    onReconnect(): void;
+    /** The connection is gone for good; a fresh one replaces it. */
+    onLost(error: Error): void;
+};
+/**
+ * Shared core of the Redis transports: one SUBSCRIBE per channel no matter
+ * how many listeners, and one dedicated subscriber connection at a time.
+ *
+ * Each connection is a generation. Losing it (or failing to restore its
+ * channels after a reconnect) retires it: the core closes it, ignores
+ * anything it still emits, and tells every listener, whose resubscribe then
+ * opens the next generation. Nothing from an old connection can reach the
+ * channels of a new one.
+ */
+export declare function createRedisTransport(driver: {
+    send: RedisSend;
+    connectSubscriber: (handlers: SubscriberHandlers) => Promise<SubscriberConnection>;
+}): PubSubTransport & {
+    close(): void;
+};
+//# sourceMappingURL=redis.d.ts.map

package/dist/live/redis.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis.d.ts","sourceRoot":"","sources":["../../src/live/redis.ts"],"names":[],"mappings":"AACA,OAAO,EAAc,KAAK,eAAe,EAAE,MAAM,gBAAgB,CAAA;AAEjE,MAAM,MAAM,SAAS,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;AAM7E;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,UAavB,MAAM,EAAE,QAAQ,MAAM,EAAE,sBAW/D;AA8BD,MAAM,MAAM,oBAAoB,GAAG;IAClC,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAA;IAC5C,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAA;IAC9C;;;OAGG;IACH,WAAW,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAA;IAC/C,6EAA6E;IAC7E,KAAK,IAAI,IAAI,CAAA;CACb,CAAA;AAED,yEAAyE;AACzE,MAAM,MAAM,kBAAkB,GAAG;IAChC,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAA;IACjD,kEAAkE;IAClE,WAAW,IAAI,IAAI,CAAA;IACnB,gEAAgE;IAChE,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI,CAAA;CAC1B,CAAA;AAaD;;;;;;;;;GASG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE;IAC5C,IAAI,EAAE,SAAS,CAAA;IACf,iBAAiB,EAAE,CAClB,QAAQ,EAAE,kBAAkB,KACxB,OAAO,CAAC,oBAAoB,CAAC,CAAA;CAClC,GAAG,eAAe,GAAG;IAAE,KAAK,IAAI,IAAI,CAAA;CAAE,CAuItC"}

package/dist/live/redis.js

@@ -0,0 +1,214 @@
+import { errorMessageOf } from '../otel.js';
+import { backlogKey } from './transport.js';
+function isNoScriptError(error) {
+    return errorMessageOf(error).includes('NOSCRIPT');
+}
+/**
+ * Ship a Lua script by SHA instead of resending its source on every call.
+ * Redis forgets its script cache on restart and on `SCRIPT FLUSH`, so a
+ * `NOSCRIPT` reply falls back to an inline `EVAL` (which repopulates the
+ * cache) and drops the stale digest.
+ */
+export function createLuaScript(send, source) {
+    let sha = null;
+    let loading = null;
+    function load() {
+        loading ??= send('SCRIPT', ['LOAD', source])
+            .then(String)
+            .finally(() => {
+            loading = null;
+        });
+        return loading;
+    }
+    return async function evalScript(keys, args) {
+        const tail = [String(keys.length), ...keys, ...args];
+        sha ??= await load();
+        try {
+            return await send('EVALSHA', [sha, ...tail]);
+        }
+        catch (error) {
+            if (!isNoScriptError(error))
+                throw error;
+            sha = null;
+            return await send('EVAL', [source, ...tail]);
+        }
+    };
+}
+/**
+ * Publish payloads and maintain each channel's capped, TTL'd backlog in one
+ * atomic round-trip. Separate PUBLISH/RPUSH/LTRIM/EXPIRE calls can tear (a
+ * message published but missing from the backlog, or a backlog without TTL).
+ *
+ * KEYS = channel1, backlog1, channel2, backlog2, ...
+ * ARGV[1] = max backlog size (0 = no backlog), ARGV[2] = TTL seconds,
+ * ARGV[3..] = pairs of (channel slot, payload)
+ */
+const PUBLISH_LUA = `
+local maxSize = tonumber(ARGV[1])
+local ttl = tonumber(ARGV[2])
+for i = 3, #ARGV, 2 do
+  local slot = tonumber(ARGV[i]) * 2
+  redis.call('PUBLISH', KEYS[slot - 1], ARGV[i + 1])
+  if maxSize > 0 then
+    redis.call('RPUSH', KEYS[slot], ARGV[i + 1])
+  end
+end
+if maxSize > 0 then
+  for slot = 2, #KEYS, 2 do
+    redis.call('LTRIM', KEYS[slot], -maxSize, -1)
+    redis.call('EXPIRE', KEYS[slot], ttl)
+  end
+end
+return (#ARGV - 2) / 2
+`;
+/**
+ * Shared core of the Redis transports: one SUBSCRIBE per channel no matter
+ * how many listeners, and one dedicated subscriber connection at a time.
+ *
+ * Each connection is a generation. Losing it (or failing to restore its
+ * channels after a reconnect) retires it: the core closes it, ignores
+ * anything it still emits, and tells every listener, whose resubscribe then
+ * opens the next generation. Nothing from an old connection can reach the
+ * channels of a new one.
+ */
+export function createRedisTransport(driver) {
+    const publishScript = createLuaScript(driver.send, PUBLISH_LUA);
+    const channels = new Map();
+    // An UNSUBSCRIBE still in flight would cancel a SUBSCRIBE sent after it.
+    const unsubscribing = new Map();
+    let current = null;
+    const connect = () => {
+        let retired = false;
+        const retire = (error) => {
+            if (retired)
+                return;
+            retired = true;
+            if (current?.connection === connection)
+                current = null;
+            void connection.then((open) => open.close()).catch(() => { });
+            if (!error)
+                return;
+            const lost = [...channels.values()].flatMap((state) => [...state.entries]);
+            channels.clear();
+            for (const entry of lost)
+                entry.onLost?.(error);
+        };
+        const handlers = {
+            onMessage(channel, message) {
+                if (retired)
+                    return;
+                for (const entry of [...(channels.get(channel)?.entries ?? [])]) {
+                    try {
+                        entry.listener(message);
+                    }
+                    catch {
+                        // One listener must not starve the others.
+                    }
+                }
+            },
+            onReconnect() {
+                if (retired)
+                    return;
+                // In-flight SUBSCRIBEs are excluded: they still land on their own.
+                const settled = [...channels]
+                    .filter(([, state]) => state.settled)
+                    .map(([channel]) => channel);
+                void connection
+                    .then((open) => Promise.all(settled.map((channel) => open.resubscribe?.(channel))))
+                    .catch((error) => retire(asError(error)));
+            },
+            onLost: (error) => retire(error)
+        };
+        const connection = driver.connectSubscriber(handlers);
+        connection.catch(() => {
+            if (current?.connection === connection)
+                current = null;
+        });
+        current = { connection, retire: () => retire() };
+        return connection;
+    };
+    return {
+        async publish(messages, backlog) {
+            const [first] = messages;
+            if (!first)
+                return;
+            if (!backlog && messages.length === 1) {
+                await driver.send('PUBLISH', [first.channel, first.payload]);
+                return;
+            }
+            const keys = [];
+            const slots = new Map();
+            const args = [
+                String(backlog?.size ?? 0),
+                String(backlog?.ttlSeconds ?? 0)
+            ];
+            for (const { channel, payload } of messages) {
+                let slot = slots.get(channel);
+                if (slot === undefined) {
+                    keys.push(channel, backlogKey(channel));
+                    slot = keys.length / 2;
+                    slots.set(channel, slot);
+                }
+                args.push(String(slot), payload);
+            }
+            await publishScript(keys, args);
+        },
+        async readBacklog(channel) {
+            const items = await driver.send('LRANGE', [
+                backlogKey(channel),
+                '0',
+                '-1'
+            ]);
+            return Array.isArray(items) ? items.map(String) : [];
+        },
+        async subscribe(channel, listener, onLost) {
+            const connection = await (current?.connection ?? connect());
+            const entry = { listener, onLost };
+            await unsubscribing.get(channel);
+            let state = channels.get(channel);
+            if (!state) {
+                const created = {
+                    entries: new Set(),
+                    ready: connection.subscribe(channel),
+                    settled: false
+                };
+                created.ready.then(() => {
+                    created.settled = true;
+                }, () => {
+                    if (channels.get(channel) === created)
+                        channels.delete(channel);
+                });
+                channels.set(channel, created);
+                state = created;
+            }
+            state.entries.add(entry);
+            try {
+                await state.ready;
+            }
+            catch (error) {
+                state.entries.delete(entry);
+                throw error;
+            }
+            const subscribed = state;
+            return async () => {
+                if (!subscribed.entries.delete(entry))
+                    return;
+                if (subscribed.entries.size > 0 || channels.get(channel) !== subscribed)
+                    return;
+                channels.delete(channel);
+                const done = connection.unsubscribe(channel).finally(() => {
+                    if (unsubscribing.get(channel) === done)
+                        unsubscribing.delete(channel);
+                });
+                unsubscribing.set(channel, done);
+                await done;
+            };
+        },
+        close() {
+            current?.retire();
+            channels.clear();
+        }
+    };
+}
+const asError = (error) => error instanceof Error ? error : new Error(errorMessageOf(error));
+//# sourceMappingURL=redis.js.map

package/dist/live/redis.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis.js","sourceRoot":"","sources":["../../src/live/redis.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,UAAU,EAAwB,MAAM,gBAAgB,CAAA;AAIjE,SAAS,eAAe,CAAC,KAAc;IACtC,OAAO,cAAc,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;AAClD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,IAAe,EAAE,MAAc;IAC9D,IAAI,GAAG,GAAkB,IAAI,CAAA;IAC7B,IAAI,OAAO,GAA2B,IAAI,CAAA;IAE1C,SAAS,IAAI;QACZ,OAAO,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;aAC1C,IAAI,CAAC,MAAM,CAAC;aACZ,OAAO,CAAC,GAAG,EAAE;YACb,OAAO,GAAG,IAAI,CAAA;QACf,CAAC,CAAC,CAAA;QACH,OAAO,OAAO,CAAA;IACf,CAAC;IAED,OAAO,KAAK,UAAU,UAAU,CAAC,IAAc,EAAE,IAAc;QAC9D,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC,CAAA;QACpD,GAAG,KAAK,MAAM,IAAI,EAAE,CAAA;QACpB,IAAI,CAAC;YACJ,OAAO,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAA;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;gBAAE,MAAM,KAAK,CAAA;YACxC,GAAG,GAAG,IAAI,CAAA;YACV,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAA;QAC7C,CAAC;IACF,CAAC,CAAA;AACF,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;CAiBnB,CAAA;AAkCD;;;;;;;;;GASG;AACH,MAAM,UAAU,oBAAoB,CAAC,MAKpC;IACA,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAA;IAC/D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAA;IAChD,yEAAyE;IACzE,MAAM,aAAa,GAAG,IAAI,GAAG,EAA4B,CAAA;IACzD,IAAI,OAAO,GAGA,IAAI,CAAA;IAEf,MAAM,OAAO,GAAG,GAAG,EAAE;QACpB,IAAI,OAAO,GAAG,KAAK,CAAA;QACnB,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,EAAE;YAChC,IAAI,OAAO;gBAAE,OAAM;YACnB,OAAO,GAAG,IAAI,CAAA;YACd,IAAI,OAAO,EAAE,UAAU,KAAK,UAAU;gBAAE,OAAO,GAAG,IAAI,CAAA;YACtD,KAAK,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;YAC5D,IAAI,CAAC,KAAK;gBAAE,OAAM;YAClB,MAAM,IAAI,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAA;YAC1E,QAAQ,CAAC,KAAK,EAAE,CAAA;YAChB,KAAK,MAAM,KAAK,IAAI,IAAI;gBAAE,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAA;QAChD,CAAC,CAAA;QACD,MAAM,QAAQ,GAAuB;YACpC,SAAS,CAAC,OAAO,EAAE,OAAO;gBACzB,IAAI,OAAO;oBAAE,OAAM;gBACnB,KAAK,MAAM,KAAK,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;oBACjE,IAAI,CAAC;wBACJ,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAA;oBACxB,CAAC;oBAAC,MAAM,CAAC;wBACR,2CAA2C;oBAC5C,CAAC;gBACF,CAAC;YACF,CAAC;YACD,WAAW;gBACV,IAAI,OAAO;oBAAE,OAAM;gBACnB,mEAAmE;gBACnE,MAAM,OAAO,GAAG,CAAC,GAAG,QAAQ,CAAC;qBAC3B,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;qBACpC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;gBAC7B,KAAK,UAAU;qBACb,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CACd,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAClE;qBACA,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;YAC3C,CAAC;YACD,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;SAChC,CAAA;QACD,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;QACrD,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;YACrB,IAAI,OAAO,EAAE,UAAU,KAAK,UAAU;gBAAE,OAAO,GAAG,IAAI,CAAA;QACvD,CAAC,CAAC,CAAA;QACF,OAAO,GAAG,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,EAAE,CAAA;QAChD,OAAO,UAAU,CAAA;IAClB,CAAC,CAAA;IAED,OAAO;QACN,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO;YAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,QAAQ,CAAA;YACxB,IAAI,CAAC,KAAK;gBAAE,OAAM;YAClB,IAAI,CAAC,OAAO,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvC,MAAM,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAA;gBAC5D,OAAM;YACP,CAAC;YACD,MAAM,IAAI,GAAa,EAAE,CAAA;YACzB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAA;YACvC,MAAM,IAAI,GAAG;gBACZ,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC;gBAC1B,MAAM,CAAC,OAAO,EAAE,UAAU,IAAI,CAAC,CAAC;aAChC,CAAA;YACD,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,QAAQ,EAAE,CAAC;gBAC7C,IAAI,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;gBAC7B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;oBACxB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC,CAAA;oBACvC,IAAI,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA;oBACtB,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAA;gBACzB,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAA;YACjC,CAAC;YACD,MAAM,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAChC,CAAC;QACD,KAAK,CAAC,WAAW,CAAC,OAAO;YACxB,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACzC,UAAU,CAAC,OAAO,CAAC;gBACnB,GAAG;gBACH,IAAI;aACJ,CAAC,CAAA;YACF,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;QACrD,CAAC;QACD,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM;YACxC,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,EAAE,UAAU,IAAI,OAAO,EAAE,CAAC,CAAA;YAC3D,MAAM,KAAK,GAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;YACzC,MAAM,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;YAChC,IAAI,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;YACjC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACZ,MAAM,OAAO,GAAiB;oBAC7B,OAAO,EAAE,IAAI,GAAG,EAAE;oBAClB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC;oBACpC,OAAO,EAAE,KAAK;iBACd,CAAA;gBACD,OAAO,CAAC,KAAK,CAAC,IAAI,CACjB,GAAG,EAAE;oBACJ,OAAO,CAAC,OAAO,GAAG,IAAI,CAAA;gBACvB,CAAC,EACD,GAAG,EAAE;oBACJ,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,OAAO;wBAAE,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;gBAChE,CAAC,CACD,CAAA;gBACD,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;gBAC9B,KAAK,GAAG,OAAO,CAAA;YAChB,CAAC;YACD,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;YACxB,IAAI,CAAC;gBACJ,MAAM,KAAK,CAAC,KAAK,CAAA;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBAC3B,MAAM,KAAK,CAAA;YACZ,CAAC;YACD,MAAM,UAAU,GAAG,KAAK,CAAA;YACxB,OAAO,KAAK,IAAI,EAAE;gBACjB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;oBAAE,OAAM;gBAC7C,IAAI,UAAU,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,UAAU;oBACtE,OAAM;gBACP,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;gBACxB,MAAM,IAAI,GAAG,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;oBACzD,IAAI,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI;wBAAE,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;gBACvE,CAAC,CAAC,CAAA;gBACF,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAA;gBAChC,MAAM,IAAI,CAAA;YACX,CAAC,CAAA;QACF,CAAC;QACD,KAAK;YACJ,OAAO,EAAE,MAAM,EAAE,CAAA;YACjB,QAAQ,CAAC,KAAK,EAAE,CAAA;QACjB,CAAC;KACD,CAAA;AACF,CAAC;AAED,MAAM,OAAO,GAAG,CAAC,KAAc,EAAE,EAAE,CAClC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA","sourcesContent":["import { errorMessageOf } from '../otel.js'\nimport { backlogKey, type PubSubTransport } from './transport.js'\n\nexport type RedisSend = (command: string, args: string[]) => Promise<unknown>\n\nfunction isNoScriptError(error: unknown) {\n\treturn errorMessageOf(error).includes('NOSCRIPT')\n}\n\n/**\n * Ship a Lua script by SHA instead of resending its source on every call.\n * Redis forgets its script cache on restart and on `SCRIPT FLUSH`, so a\n * `NOSCRIPT` reply falls back to an inline `EVAL` (which repopulates the\n * cache) and drops the stale digest.\n */\nexport function createLuaScript(send: RedisSend, source: string) {\n\tlet sha: string | null = null\n\tlet loading: Promise<string> | null = null\n\n\tfunction load() {\n\t\tloading ??= send('SCRIPT', ['LOAD', source])\n\t\t\t.then(String)\n\t\t\t.finally(() => {\n\t\t\t\tloading = null\n\t\t\t})\n\t\treturn loading\n\t}\n\n\treturn async function evalScript(keys: string[], args: string[]) {\n\t\tconst tail = [String(keys.length), ...keys, ...args]\n\t\tsha ??= await load()\n\t\ttry {\n\t\t\treturn await send('EVALSHA', [sha, ...tail])\n\t\t} catch (error) {\n\t\t\tif (!isNoScriptError(error)) throw error\n\t\t\tsha = null\n\t\t\treturn await send('EVAL', [source, ...tail])\n\t\t}\n\t}\n}\n\n/**\n * Publish payloads and maintain each channel's capped, TTL'd backlog in one\n * atomic round-trip. Separate PUBLISH/RPUSH/LTRIM/EXPIRE calls can tear (a\n * message published but missing from the backlog, or a backlog without TTL).\n *\n * KEYS = channel1, backlog1, channel2, backlog2, ...\n * ARGV[1] = max backlog size (0 = no backlog), ARGV[2] = TTL seconds,\n * ARGV[3..] = pairs of (channel slot, payload)\n */\nconst PUBLISH_LUA = `\nlocal maxSize = tonumber(ARGV[1])\nlocal ttl = tonumber(ARGV[2])\nfor i = 3, #ARGV, 2 do\n  local slot = tonumber(ARGV[i]) * 2\n  redis.call('PUBLISH', KEYS[slot - 1], ARGV[i + 1])\n  if maxSize > 0 then\n    redis.call('RPUSH', KEYS[slot], ARGV[i + 1])\n  end\nend\nif maxSize > 0 then\n  for slot = 2, #KEYS, 2 do\n    redis.call('LTRIM', KEYS[slot], -maxSize, -1)\n    redis.call('EXPIRE', KEYS[slot], ttl)\n  end\nend\nreturn (#ARGV - 2) / 2\n`\n\nexport type SubscriberConnection = {\n\tsubscribe(channel: string): Promise<unknown>\n\tunsubscribe(channel: string): Promise<unknown>\n\t/**\n\t * Restore one channel after the driver reconnected on its own and lost its\n\t * subscriptions. Omit when the client re-subscribes itself (ioredis).\n\t */\n\tresubscribe?(channel: string): Promise<unknown>\n\t/** Stop delivering and release the connection (if the driver created it). */\n\tclose(): void\n}\n\n/** Callbacks for one connection; they do nothing once it is replaced. */\nexport type SubscriberHandlers = {\n\tonMessage(channel: string, message: string): void\n\t/** The driver reconnected; its subscriptions must be restored. */\n\tonReconnect(): void\n\t/** The connection is gone for good; a fresh one replaces it. */\n\tonLost(error: Error): void\n}\n\ntype Entry = {\n\tlistener: (payload: string) => void\n\tonLost: ((error: Error) => void) | undefined\n}\n\ntype ChannelState = {\n\tentries: Set<Entry>\n\tready: Promise<unknown>\n\tsettled: boolean\n}\n\n/**\n * Shared core of the Redis transports: one SUBSCRIBE per channel no matter\n * how many listeners, and one dedicated subscriber connection at a time.\n *\n * Each connection is a generation. Losing it (or failing to restore its\n * channels after a reconnect) retires it: the core closes it, ignores\n * anything it still emits, and tells every listener, whose resubscribe then\n * opens the next generation. Nothing from an old connection can reach the\n * channels of a new one.\n */\nexport function createRedisTransport(driver: {\n\tsend: RedisSend\n\tconnectSubscriber: (\n\t\thandlers: SubscriberHandlers\n\t) => Promise<SubscriberConnection>\n}): PubSubTransport & { close(): void } {\n\tconst publishScript = createLuaScript(driver.send, PUBLISH_LUA)\n\tconst channels = new Map<string, ChannelState>()\n\t// An UNSUBSCRIBE still in flight would cancel a SUBSCRIBE sent after it.\n\tconst unsubscribing = new Map<string, Promise<unknown>>()\n\tlet current: {\n\t\tconnection: Promise<SubscriberConnection>\n\t\tretire: () => void\n\t} | null = null\n\n\tconst connect = () => {\n\t\tlet retired = false\n\t\tconst retire = (error?: Error) => {\n\t\t\tif (retired) return\n\t\t\tretired = true\n\t\t\tif (current?.connection === connection) current = null\n\t\t\tvoid connection.then((open) => open.close()).catch(() => {})\n\t\t\tif (!error) return\n\t\t\tconst lost = [...channels.values()].flatMap((state) => [...state.entries])\n\t\t\tchannels.clear()\n\t\t\tfor (const entry of lost) entry.onLost?.(error)\n\t\t}\n\t\tconst handlers: SubscriberHandlers = {\n\t\t\tonMessage(channel, message) {\n\t\t\t\tif (retired) return\n\t\t\t\tfor (const entry of [...(channels.get(channel)?.entries ?? [])]) {\n\t\t\t\t\ttry {\n\t\t\t\t\t\tentry.listener(message)\n\t\t\t\t\t} catch {\n\t\t\t\t\t\t// One listener must not starve the others.\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t\tonReconnect() {\n\t\t\t\tif (retired) return\n\t\t\t\t// In-flight SUBSCRIBEs are excluded: they still land on their own.\n\t\t\t\tconst settled = [...channels]\n\t\t\t\t\t.filter(([, state]) => state.settled)\n\t\t\t\t\t.map(([channel]) => channel)\n\t\t\t\tvoid connection\n\t\t\t\t\t.then((open) =>\n\t\t\t\t\t\tPromise.all(settled.map((channel) => open.resubscribe?.(channel)))\n\t\t\t\t\t)\n\t\t\t\t\t.catch((error) => retire(asError(error)))\n\t\t\t},\n\t\t\tonLost: (error) => retire(error)\n\t\t}\n\t\tconst connection = driver.connectSubscriber(handlers)\n\t\tconnection.catch(() => {\n\t\t\tif (current?.connection === connection) current = null\n\t\t})\n\t\tcurrent = { connection, retire: () => retire() }\n\t\treturn connection\n\t}\n\n\treturn {\n\t\tasync publish(messages, backlog) {\n\t\t\tconst [first] = messages\n\t\t\tif (!first) return\n\t\t\tif (!backlog && messages.length === 1) {\n\t\t\t\tawait driver.send('PUBLISH', [first.channel, first.payload])\n\t\t\t\treturn\n\t\t\t}\n\t\t\tconst keys: string[] = []\n\t\t\tconst slots = new Map<string, number>()\n\t\t\tconst args = [\n\t\t\t\tString(backlog?.size ?? 0),\n\t\t\t\tString(backlog?.ttlSeconds ?? 0)\n\t\t\t]\n\t\t\tfor (const { channel, payload } of messages) {\n\t\t\t\tlet slot = slots.get(channel)\n\t\t\t\tif (slot === undefined) {\n\t\t\t\t\tkeys.push(channel, backlogKey(channel))\n\t\t\t\t\tslot = keys.length / 2\n\t\t\t\t\tslots.set(channel, slot)\n\t\t\t\t}\n\t\t\t\targs.push(String(slot), payload)\n\t\t\t}\n\t\t\tawait publishScript(keys, args)\n\t\t},\n\t\tasync readBacklog(channel) {\n\t\t\tconst items = await driver.send('LRANGE', [\n\t\t\t\tbacklogKey(channel),\n\t\t\t\t'0',\n\t\t\t\t'-1'\n\t\t\t])\n\t\t\treturn Array.isArray(items) ? items.map(String) : []\n\t\t},\n\t\tasync subscribe(channel, listener, onLost) {\n\t\t\tconst connection = await (current?.connection ?? connect())\n\t\t\tconst entry: Entry = { listener, onLost }\n\t\t\tawait unsubscribing.get(channel)\n\t\t\tlet state = channels.get(channel)\n\t\t\tif (!state) {\n\t\t\t\tconst created: ChannelState = {\n\t\t\t\t\tentries: new Set(),\n\t\t\t\t\tready: connection.subscribe(channel),\n\t\t\t\t\tsettled: false\n\t\t\t\t}\n\t\t\t\tcreated.ready.then(\n\t\t\t\t\t() => {\n\t\t\t\t\t\tcreated.settled = true\n\t\t\t\t\t},\n\t\t\t\t\t() => {\n\t\t\t\t\t\tif (channels.get(channel) === created) channels.delete(channel)\n\t\t\t\t\t}\n\t\t\t\t)\n\t\t\t\tchannels.set(channel, created)\n\t\t\t\tstate = created\n\t\t\t}\n\t\t\tstate.entries.add(entry)\n\t\t\ttry {\n\t\t\t\tawait state.ready\n\t\t\t} catch (error) {\n\t\t\t\tstate.entries.delete(entry)\n\t\t\t\tthrow error\n\t\t\t}\n\t\t\tconst subscribed = state\n\t\t\treturn async () => {\n\t\t\t\tif (!subscribed.entries.delete(entry)) return\n\t\t\t\tif (subscribed.entries.size > 0 || channels.get(channel) !== subscribed)\n\t\t\t\t\treturn\n\t\t\t\tchannels.delete(channel)\n\t\t\t\tconst done = connection.unsubscribe(channel).finally(() => {\n\t\t\t\t\tif (unsubscribing.get(channel) === done) unsubscribing.delete(channel)\n\t\t\t\t})\n\t\t\t\tunsubscribing.set(channel, done)\n\t\t\t\tawait done\n\t\t\t}\n\t\t},\n\t\tclose() {\n\t\t\tcurrent?.retire()\n\t\t\tchannels.clear()\n\t\t}\n\t}\n}\n\nconst asError = (error: unknown) =>\n\terror instanceof Error ? error : new Error(errorMessageOf(error))\n"]}

package/dist/live/transport.d.ts

@@ -0,0 +1,35 @@
+/** One payload for one channel. */
+export type PubSubMessage = {
+    channel: string;
+    payload: string;
+};
+export type BacklogOptions = {
+    /** Items kept per channel (oldest dropped). */
+    size: number;
+    /** Seconds the backlog lives after its last write. */
+    ttlSeconds: number;
+};
+/**
+ * The broker behind `createPubSub` / `fnLive`. Implementations:
+ * `orpc-fn/live/redis-bun`, `orpc-fn/live/ioredis` and `orpc-fn/live/memory`.
+ *
+ * The backlog of a channel lives at `${channel}:backlog` (see {@link backlogKey}).
+ */
+export interface PubSubTransport {
+    /**
+     * Publish every message, in order. With `backlog`, also append each payload
+     * to its channel's backlog, trim and refresh its TTL - atomically, so a
+     * message is never published without its backlog entry or vice versa.
+     */
+    publish(messages: readonly PubSubMessage[], backlog?: BacklogOptions): Promise<void>;
+    /** The channel's backlog, oldest first. */
+    readBacklog(channel: string): Promise<string[]>;
+    /**
+     * Deliver every payload published to `channel` until the returned function
+     * is called. `onLost` fires when the subscription can no longer deliver
+     * (connection closed for good); the caller resubscribes.
+     */
+    subscribe(channel: string, listener: (payload: string) => void, onLost?: (error: Error) => void): Promise<() => Promise<void>>;
+}
+export declare const backlogKey: (channel: string) => string;
+//# sourceMappingURL=transport.d.ts.map

package/dist/live/transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/live/transport.ts"],"names":[],"mappings":"AAAA,mCAAmC;AACnC,MAAM,MAAM,aAAa,GAAG;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAA;AAEhE,MAAM,MAAM,cAAc,GAAG;IAC5B,+CAA+C;IAC/C,IAAI,EAAE,MAAM,CAAA;IACZ,sDAAsD;IACtD,UAAU,EAAE,MAAM,CAAA;CAClB,CAAA;AAED;;;;;GAKG;AACH,MAAM,WAAW,eAAe;IAC/B;;;;OAIG;IACH,OAAO,CACN,QAAQ,EAAE,SAAS,aAAa,EAAE,EAClC,OAAO,CAAC,EAAE,cAAc,GACtB,OAAO,CAAC,IAAI,CAAC,CAAA;IAChB,2CAA2C;IAC3C,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;IAC/C;;;;OAIG;IACH,SAAS,CACR,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,EACnC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAC7B,OAAO,CAAC,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,CAAA;CAC/B;AAED,eAAO,MAAM,UAAU,YAAa,MAAM,WAAyB,CAAA"}

package/dist/live/transport.js

@@ -0,0 +1,2 @@
+export const backlogKey = (channel) => `${channel}:backlog`;
+//# sourceMappingURL=transport.js.map

package/dist/live/transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../src/live/transport.ts"],"names":[],"mappings":"AAwCA,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,OAAe,EAAE,EAAE,CAAC,GAAG,OAAO,UAAU,CAAA","sourcesContent":["/** One payload for one channel. */\nexport type PubSubMessage = { channel: string; payload: string }\n\nexport type BacklogOptions = {\n\t/** Items kept per channel (oldest dropped). */\n\tsize: number\n\t/** Seconds the backlog lives after its last write. */\n\tttlSeconds: number\n}\n\n/**\n * The broker behind `createPubSub` / `fnLive`. Implementations:\n * `orpc-fn/live/redis-bun`, `orpc-fn/live/ioredis` and `orpc-fn/live/memory`.\n *\n * The backlog of a channel lives at `${channel}:backlog` (see {@link backlogKey}).\n */\nexport interface PubSubTransport {\n\t/**\n\t * Publish every message, in order. With `backlog`, also append each payload\n\t * to its channel's backlog, trim and refresh its TTL - atomically, so a\n\t * message is never published without its backlog entry or vice versa.\n\t */\n\tpublish(\n\t\tmessages: readonly PubSubMessage[],\n\t\tbacklog?: BacklogOptions\n\t): Promise<void>\n\t/** The channel's backlog, oldest first. */\n\treadBacklog(channel: string): Promise<string[]>\n\t/**\n\t * Deliver every payload published to `channel` until the returned function\n\t * is called. `onLost` fires when the subscription can no longer deliver\n\t * (connection closed for good); the caller resubscribes.\n\t */\n\tsubscribe(\n\t\tchannel: string,\n\t\tlistener: (payload: string) => void,\n\t\tonLost?: (error: Error) => void\n\t): Promise<() => Promise<void>>\n}\n\nexport const backlogKey = (channel: string) => `${channel}:backlog`\n"]}

package/dist/logger.d.ts

@@ -0,0 +1,15 @@
+export type LogAttributes = Record<string, unknown>;
+/** The logger handlers receive. Any app logger with these four methods fits. */
+export interface FnLogger {
+    debug(message: string, attributes?: LogAttributes): void;
+    info(message: string, attributes?: LogAttributes): void;
+    warn(message: string, attributes?: LogAttributes): void;
+    error(message: string, attributes?: LogAttributes): void;
+}
+/**
+ * Default logger: silent for `debug`/`info`, console for `warn`/`error`, so a
+ * library default never floods stdout with one line per call. Inject your own
+ * with `createFn({ logger })`.
+ */
+export declare function createDefaultLogger(scope: string): FnLogger;
+//# sourceMappingURL=logger.d.ts.map

package/dist/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;AAEnD,gFAAgF;AAChF,MAAM,WAAW,QAAQ;IACxB,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,aAAa,GAAG,IAAI,CAAA;IACxD,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,aAAa,GAAG,IAAI,CAAA;IACvD,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,aAAa,GAAG,IAAI,CAAA;IACvD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,aAAa,GAAG,IAAI,CAAA;CACxD;AAKD;;;;GAIG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,QAAQ,CAS3D"}

package/dist/logger.js

@@ -0,0 +1,15 @@
+const format = (scope, message, attributes) => `[${scope}] ${message}${attributes ? ` ${JSON.stringify(attributes)}` : ''}`;
+/**
+ * Default logger: silent for `debug`/`info`, console for `warn`/`error`, so a
+ * library default never floods stdout with one line per call. Inject your own
+ * with `createFn({ logger })`.
+ */
+export function createDefaultLogger(scope) {
+    return {
+        debug() { },
+        info() { },
+        warn: (message, attributes) => console.warn(format(scope, message, attributes)),
+        error: (message, attributes) => console.error(format(scope, message, attributes))
+    };
+}
+//# sourceMappingURL=logger.js.map

package/dist/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAUA,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,OAAe,EAAE,UAA0B,EAAE,EAAE,CAC7E,IAAI,KAAK,KAAK,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAA;AAE7E;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAa;IAChD,OAAO;QACN,KAAK,KAAI,CAAC;QACV,IAAI,KAAI,CAAC;QACT,IAAI,EAAE,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CAC7B,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACjD,KAAK,EAAE,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CAC9B,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;KAClD,CAAA;AACF,CAAC","sourcesContent":["export type LogAttributes = Record<string, unknown>\n\n/** The logger handlers receive. Any app logger with these four methods fits. */\nexport interface FnLogger {\n\tdebug(message: string, attributes?: LogAttributes): void\n\tinfo(message: string, attributes?: LogAttributes): void\n\twarn(message: string, attributes?: LogAttributes): void\n\terror(message: string, attributes?: LogAttributes): void\n}\n\nconst format = (scope: string, message: string, attributes?: LogAttributes) =>\n\t`[${scope}] ${message}${attributes ? ` ${JSON.stringify(attributes)}` : ''}`\n\n/**\n * Default logger: silent for `debug`/`info`, console for `warn`/`error`, so a\n * library default never floods stdout with one line per call. Inject your own\n * with `createFn({ logger })`.\n */\nexport function createDefaultLogger(scope: string): FnLogger {\n\treturn {\n\t\tdebug() {},\n\t\tinfo() {},\n\t\twarn: (message, attributes) =>\n\t\t\tconsole.warn(format(scope, message, attributes)),\n\t\terror: (message, attributes) =>\n\t\t\tconsole.error(format(scope, message, attributes))\n\t}\n}\n"]}

package/dist/mastra.d.ts

@@ -0,0 +1,44 @@
+import { type Tool } from '@mastra/core/tools';
+import { type AnyProcedure } from '@orpc/server';
+import { toToolName } from './meta.js';
+import type { ProcedureInput, ProcedureOutput } from './types.js';
+export { toToolName };
+export type CreateMastraToolOptions = {
+    /** Override the tool id (defaults to the sanitized `fn` name). */
+    id?: string;
+    /** Override the tool description (defaults to the route summary). */
+    description?: string;
+    /** Gate execution behind the AI SDK tool-approval flow. */
+    requireApproval?: boolean;
+    /**
+     * Accept a procedure without an input schema (the tool then takes an empty
+     * object). For no-input queries like `teams.list`.
+     */
+    allowMissingInputSchema?: boolean;
+    onExecuteFinish?: (event: {
+        toolId: string;
+        durationMs: number;
+        outcome: 'success' | 'error';
+    }) => void;
+    /** Key holding the oRPC context in Mastra's `RequestContext`. Default 'orpcContext'. */
+    contextKey?: string;
+};
+/** What the tool takes: the procedure's raw input, or `{}` without a schema. */
+export type MastraToolInput<TProc> = unknown extends ProcedureInput<TProc> ? Record<string, never> : ProcedureInput<TProc>;
+/** A Mastra tool typed by the procedure's raw input and parsed output. */
+export type MastraTool<TProc extends AnyProcedure> = Tool<MastraToolInput<TProc>, ProcedureOutput<TProc>>;
+/** LLM tool parameters must be an object; other inputs are a type error. */
+type ObjectInputCheck<TProc> = MastraToolInput<TProc> extends Record<string, unknown> ? unknown : {
+    'orpc-fn: an agent tool needs an object input': ProcedureInput<TProc>;
+};
+/**
+ * Creates a Mastra tool from a procedure made with `fn()`. Execution calls
+ * the procedure with the oRPC context stored in the Mastra request context
+ * (`requestContext.set('orpcContext', context)`), so auth and guards apply.
+ *
+ * Mastra checks the input against the procedure's schema (with dates coerced
+ * from strings) but passes the raw input on for the procedure to parse;
+ * the output is the procedure's parsed result, passed through unchanged.
+ */
+export declare function createMastraTool<TProc extends AnyProcedure>(procedure: TProc & ObjectInputCheck<TProc>, options?: CreateMastraToolOptions): MastraTool<TProc>;
+//# sourceMappingURL=mastra.d.ts.map

package/dist/mastra.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastra.d.ts","sourceRoot":"","sources":["../src/mastra.ts"],"names":[],"mappings":"AACA,OAAO,EAAc,KAAK,IAAI,EAAE,MAAM,oBAAoB,CAAA;AAC1D,OAAO,EAAE,KAAK,YAAY,EAAQ,MAAM,cAAc,CAAA;AAMtD,OAAO,EAAc,UAAU,EAAE,MAAM,WAAW,CAAA;AAClD,OAAO,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AAEjE,OAAO,EAAE,UAAU,EAAE,CAAA;AAErB,MAAM,MAAM,uBAAuB,GAAG;IACrC,kEAAkE;IAClE,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,qEAAqE;IACrE,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,2DAA2D;IAC3D,eAAe,CAAC,EAAE,OAAO,CAAA;IACzB;;;OAGG;IACH,uBAAuB,CAAC,EAAE,OAAO,CAAA;IACjC,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE;QACzB,MAAM,EAAE,MAAM,CAAA;QACd,UAAU,EAAE,MAAM,CAAA;QAClB,OAAO,EAAE,SAAS,GAAG,OAAO,CAAA;KAC5B,KAAK,IAAI,CAAA;IACV,wFAAwF;IACxF,UAAU,CAAC,EAAE,MAAM,CAAA;CACnB,CAAA;AAED,gFAAgF;AAChF,MAAM,MAAM,eAAe,CAAC,KAAK,IAChC,OAAO,SAAS,cAAc,CAAC,KAAK,CAAC,GAClC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,GACrB,cAAc,CAAC,KAAK,CAAC,CAAA;AAEzB,0EAA0E;AAC1E,MAAM,MAAM,UAAU,CAAC,KAAK,SAAS,YAAY,IAAI,IAAI,CACxD,eAAe,CAAC,KAAK,CAAC,EACtB,eAAe,CAAC,KAAK,CAAC,CACtB,CAAA;AAED,4EAA4E;AAC5E,KAAK,gBAAgB,CAAC,KAAK,IAC1B,eAAe,CAAC,KAAK,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACnD,OAAO,GACP;IAAE,8CAA8C,EAAE,cAAc,CAAC,KAAK,CAAC,CAAA;CAAE,CAAA;AAE7E;;;;;;;;GAQG;AACH,wBAAgB,gBAAgB,CAAC,KAAK,SAAS,YAAY,EAC1D,SAAS,EAAE,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,EAC1C,OAAO,GAAE,uBAA4B,GACnC,UAAU,CAAC,KAAK,CAAC,CA2EnB"}

package/dist/mastra.js

@@ -0,0 +1,78 @@
+import { createTool } from '@mastra/core/tools';
+import { call } from '@orpc/server';
+import { passThroughOutputSchema, rawInputSchema } from './json-schema.js';
+import { readFnMeta, toToolName } from './meta.js';
+export { toToolName };
+/**
+ * Creates a Mastra tool from a procedure made with `fn()`. Execution calls
+ * the procedure with the oRPC context stored in the Mastra request context
+ * (`requestContext.set('orpcContext', context)`), so auth and guards apply.
+ *
+ * Mastra checks the input against the procedure's schema (with dates coerced
+ * from strings) but passes the raw input on for the procedure to parse;
+ * the output is the procedure's parsed result, passed through unchanged.
+ */
+export function createMastraTool(procedure, options = {}) {
+    const meta = readFnMeta(procedure);
+    const name = options.id ?? meta.name;
+    if (!name) {
+        throw new Error('Procedure must be created with fn() to use createMastraTool');
+    }
+    let inputSchema;
+    if (meta.inputSchema) {
+        inputSchema = rawInputSchema(meta.inputSchema);
+        if (inputSchema['~standard'].jsonSchema.input().type !== 'object') {
+            throw new Error(`${name} must take an object input to be an agent tool`);
+        }
+    }
+    else if (!options.allowMissingInputSchema) {
+        throw new Error(`${name} must declare an input schema to be exposed as an agent tool`);
+    }
+    const id = options.id ?? toToolName(name);
+    const contextKey = options.contextKey ?? 'orpcContext';
+    const tool = createTool({
+        id,
+        description: options.description ?? meta.description,
+        requireApproval: options.requireApproval === true,
+        inputSchema: inputSchema ??
+            {
+                '~standard': {
+                    version: 1,
+                    vendor: 'orpc-fn',
+                    validate: () => ({ value: {} }),
+                    jsonSchema: {
+                        input: () => ({ type: 'object', properties: {} }),
+                        output: () => ({ type: 'object', properties: {} })
+                    }
+                }
+            },
+        ...(meta.outputSchema
+            ? { outputSchema: passThroughOutputSchema(meta.outputSchema) }
+            : {}),
+        execute: async (input, ctx) => {
+            const requestContext = ctx?.requestContext;
+            const context = requestContext?.get(contextKey);
+            if (!context) {
+                throw new Error(`createMastraTool execution requires ${contextKey} in the Mastra request context`);
+            }
+            const startedAt = performance.now();
+            const finish = (outcome) => options.onExecuteFinish?.({
+                toolId: id,
+                durationMs: performance.now() - startedAt,
+                outcome
+            });
+            try {
+                const result = await call(procedure, meta.inputSchema ? input : undefined, ctx?.abortSignal ? { context, signal: ctx.abortSignal } : { context });
+                finish('success');
+                return result;
+            }
+            catch (error) {
+                finish('error');
+                throw error;
+            }
+        }
+    });
+    // The runtime schemas are untyped `ToolSchema`s; `MastraTool` is the typed surface.
+    return tool;
+}
+//# sourceMappingURL=mastra.js.map

package/dist/mastra.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastra.js","sourceRoot":"","sources":["../src/mastra.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAa,MAAM,oBAAoB,CAAA;AAC1D,OAAO,EAAqB,IAAI,EAAE,MAAM,cAAc,CAAA;AACtD,OAAO,EACN,uBAAuB,EACvB,cAAc,EAEd,MAAM,kBAAkB,CAAA;AACzB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,WAAW,CAAA;AAGlD,OAAO,EAAE,UAAU,EAAE,CAAA;AAyCrB;;;;;;;;GAQG;AACH,MAAM,UAAU,gBAAgB,CAC/B,SAA0C,EAC1C,OAAO,GAA4B,EAAE;IAErC,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,CAAA;IAClC,MAAM,IAAI,GAAG,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC,IAAI,CAAA;IACpC,IAAI,CAAC,IAAI,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CACd,6DAA6D,CAC7D,CAAA;IACF,CAAC;IACD,IAAI,WAAmC,CAAA;IACvC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACtB,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACnE,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,gDAAgD,CAAC,CAAA;QACzE,CAAC;IACF,CAAC;SAAM,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;QAC7C,MAAM,IAAI,KAAK,CACd,GAAG,IAAI,8DAA8D,CACrE,CAAA;IACF,CAAC;IAED,MAAM,EAAE,GAAG,OAAO,CAAC,EAAE,IAAI,UAAU,CAAC,IAAI,CAAC,CAAA;IACzC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,aAAa,CAAA;IAEtD,MAAM,IAAI,GAAG,UAAU,CAAC;QACvB,EAAE;QACF,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW;QACpD,eAAe,EAAE,OAAO,CAAC,eAAe,KAAK,IAAI;QACjD,WAAW,EACV,WAAW;YACV;gBACA,WAAW,EAAE;oBACZ,OAAO,EAAE,CAAC;oBACV,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;oBAC/B,UAAU,EAAE;wBACX,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;wBACjD,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;qBAClD;iBACD;aACqB;QACxB,GAAG,CAAC,IAAI,CAAC,YAAY;YACpB,CAAC,CAAC,EAAE,YAAY,EAAE,uBAAuB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE;YAC9D,CAAC,CAAC,EAAE,CAAC;QACN,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YAC7B,MAAM,cAAc,GAAG,GAAG,EAAE,cAA4C,CAAA;YACxE,MAAM,OAAO,GAAG,cAAc,EAAE,GAAG,CAAC,UAAU,CAAC,CAAA;YAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CACd,uCAAuC,UAAU,gCAAgC,CACjF,CAAA;YACF,CAAC;YACD,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;YACnC,MAAM,MAAM,GAAG,CAAC,OAA4B,EAAE,EAAE,CAC/C,OAAO,CAAC,eAAe,EAAE,CAAC;gBACzB,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS;gBACzC,OAAO;aACP,CAAC,CAAA;YACH,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CACxB,SAAS,EACT,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,EACpC,GAAG,EAAE,WAAW,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CACrE,CAAA;gBACD,MAAM,CAAC,SAAS,CAAC,CAAA;gBACjB,OAAO,MAAM,CAAA;YACd,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,OAAO,CAAC,CAAA;gBACf,MAAM,KAAK,CAAA;YACZ,CAAC;QACF,CAAC;KACD,CAAC,CAAA;IAEF,oFAAoF;IACpF,OAAO,IAAoC,CAAA;AAC5C,CAAC","sourcesContent":["import type { RequestContext } from '@mastra/core/request-context'\nimport { createTool, type Tool } from '@mastra/core/tools'\nimport { type AnyProcedure, call } from '@orpc/server'\nimport {\n\tpassThroughOutputSchema,\n\trawInputSchema,\n\ttype ToolSchema\n} from './json-schema.js'\nimport { readFnMeta, toToolName } from './meta.js'\nimport type { ProcedureInput, ProcedureOutput } from './types.js'\n\nexport { toToolName }\n\nexport type CreateMastraToolOptions = {\n\t/** Override the tool id (defaults to the sanitized `fn` name). */\n\tid?: string\n\t/** Override the tool description (defaults to the route summary). */\n\tdescription?: string\n\t/** Gate execution behind the AI SDK tool-approval flow. */\n\trequireApproval?: boolean\n\t/**\n\t * Accept a procedure without an input schema (the tool then takes an empty\n\t * object). For no-input queries like `teams.list`.\n\t */\n\tallowMissingInputSchema?: boolean\n\tonExecuteFinish?: (event: {\n\t\ttoolId: string\n\t\tdurationMs: number\n\t\toutcome: 'success' | 'error'\n\t}) => void\n\t/** Key holding the oRPC context in Mastra's `RequestContext`. Default 'orpcContext'. */\n\tcontextKey?: string\n}\n\n/** What the tool takes: the procedure's raw input, or `{}` without a schema. */\nexport type MastraToolInput<TProc> =\n\tunknown extends ProcedureInput<TProc>\n\t\t? Record<string, never>\n\t\t: ProcedureInput<TProc>\n\n/** A Mastra tool typed by the procedure's raw input and parsed output. */\nexport type MastraTool<TProc extends AnyProcedure> = Tool<\n\tMastraToolInput<TProc>,\n\tProcedureOutput<TProc>\n>\n\n/** LLM tool parameters must be an object; other inputs are a type error. */\ntype ObjectInputCheck<TProc> =\n\tMastraToolInput<TProc> extends Record<string, unknown>\n\t\t? unknown\n\t\t: { 'orpc-fn: an agent tool needs an object input': ProcedureInput<TProc> }\n\n/**\n * Creates a Mastra tool from a procedure made with `fn()`. Execution calls\n * the procedure with the oRPC context stored in the Mastra request context\n * (`requestContext.set('orpcContext', context)`), so auth and guards apply.\n *\n * Mastra checks the input against the procedure's schema (with dates coerced\n * from strings) but passes the raw input on for the procedure to parse;\n * the output is the procedure's parsed result, passed through unchanged.\n */\nexport function createMastraTool<TProc extends AnyProcedure>(\n\tprocedure: TProc & ObjectInputCheck<TProc>,\n\toptions: CreateMastraToolOptions = {}\n): MastraTool<TProc> {\n\tconst meta = readFnMeta(procedure)\n\tconst name = options.id ?? meta.name\n\tif (!name) {\n\t\tthrow new Error(\n\t\t\t'Procedure must be created with fn() to use createMastraTool'\n\t\t)\n\t}\n\tlet inputSchema: ToolSchema | undefined\n\tif (meta.inputSchema) {\n\t\tinputSchema = rawInputSchema(meta.inputSchema)\n\t\tif (inputSchema['~standard'].jsonSchema.input().type !== 'object') {\n\t\t\tthrow new Error(`${name} must take an object input to be an agent tool`)\n\t\t}\n\t} else if (!options.allowMissingInputSchema) {\n\t\tthrow new Error(\n\t\t\t`${name} must declare an input schema to be exposed as an agent tool`\n\t\t)\n\t}\n\n\tconst id = options.id ?? toToolName(name)\n\tconst contextKey = options.contextKey ?? 'orpcContext'\n\n\tconst tool = createTool({\n\t\tid,\n\t\tdescription: options.description ?? meta.description,\n\t\trequireApproval: options.requireApproval === true,\n\t\tinputSchema:\n\t\t\tinputSchema ??\n\t\t\t({\n\t\t\t\t'~standard': {\n\t\t\t\t\tversion: 1,\n\t\t\t\t\tvendor: 'orpc-fn',\n\t\t\t\t\tvalidate: () => ({ value: {} }),\n\t\t\t\t\tjsonSchema: {\n\t\t\t\t\t\tinput: () => ({ type: 'object', properties: {} }),\n\t\t\t\t\t\toutput: () => ({ type: 'object', properties: {} })\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} satisfies ToolSchema),\n\t\t...(meta.outputSchema\n\t\t\t? { outputSchema: passThroughOutputSchema(meta.outputSchema) }\n\t\t\t: {}),\n\t\texecute: async (input, ctx) => {\n\t\t\tconst requestContext = ctx?.requestContext as RequestContext | undefined\n\t\t\tconst context = requestContext?.get(contextKey)\n\t\t\tif (!context) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t`createMastraTool execution requires ${contextKey} in the Mastra request context`\n\t\t\t\t)\n\t\t\t}\n\t\t\tconst startedAt = performance.now()\n\t\t\tconst finish = (outcome: 'success' | 'error') =>\n\t\t\t\toptions.onExecuteFinish?.({\n\t\t\t\t\ttoolId: id,\n\t\t\t\t\tdurationMs: performance.now() - startedAt,\n\t\t\t\t\toutcome\n\t\t\t\t})\n\t\t\ttry {\n\t\t\t\tconst result = await call(\n\t\t\t\t\tprocedure,\n\t\t\t\t\tmeta.inputSchema ? input : undefined,\n\t\t\t\t\tctx?.abortSignal ? { context, signal: ctx.abortSignal } : { context }\n\t\t\t\t)\n\t\t\t\tfinish('success')\n\t\t\t\treturn result\n\t\t\t} catch (error) {\n\t\t\t\tfinish('error')\n\t\t\t\tthrow error\n\t\t\t}\n\t\t}\n\t})\n\n\t// The runtime schemas are untyped `ToolSchema`s; `MastraTool` is the typed surface.\n\treturn tool as unknown as MastraTool<TProc>\n}\n"]}