@atcute/xrpc-server 0.1.2 → 0.1.3

package/dist/main/utils/websocket-mock.js

@@ -0,0 +1,71 @@
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { EventEmitter } from './event-emitter.js';
+export class MockWebSocketAdapter {
+    #context = new AsyncLocalStorage();
+    upgrade(_request, handler) {
+        const ctx = this.#context.getStore();
+        if (!ctx) {
+            return undefined;
+        }
+        ctx.handler = handler;
+        return new Response(null);
+    }
+    attach(router) {
+        return {
+            subscribe: async (url) => {
+                const ctx = {
+                    handler: null,
+                };
+                await this.#context.run(ctx, async () => {
+                    const urlp = new URL(url, 'http://localhost');
+                    const request = new Request(urlp, {
+                        headers: {
+                            upgrade: 'websocket',
+                            connection: 'upgrade',
+                        },
+                    });
+                    const response = await router.fetch(request);
+                    return response;
+                });
+                if (!ctx.handler) {
+                    throw new Error(`WebSocket upgrade succeeded but no handler was set`);
+                }
+                const events = new EventEmitter();
+                const controller = new AbortController();
+                const signal = controller.signal;
+                const connection = {
+                    signal: signal,
+                    send(data) {
+                        events.emit('message', data);
+                    },
+                    close(code = 1000, reason = '') {
+                        if (!signal.aborted) {
+                            events.emit('close', { code, reason, wasClean: true });
+                            controller.abort();
+                        }
+                    },
+                };
+                {
+                    const handler = ctx.handler;
+                    setTimeout(() => {
+                        handler(connection);
+                    }, 1);
+                }
+                const client = {
+                    events,
+                    dispose() {
+                        if (!signal.aborted) {
+                            events.emit('close', { code: 1000, reason: '', wasClean: true });
+                            controller.abort();
+                        }
+                    },
+                    [Symbol.dispose]() {
+                        this.dispose();
+                    },
+                };
+                return client;
+            },
+        };
+    }
+}
+//# sourceMappingURL=websocket-mock.js.map

package/dist/main/utils/websocket-mock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-mock.js","sourceRoot":"","sources":["../../../lib/main/utils/websocket-mock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAKrD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAkBlD,MAAM,OAAO,oBAAoB;IAChC,QAAQ,GAAG,IAAI,iBAAiB,EAA2B,CAAC;IAE5D,OAAO,CACN,QAAiB,EACjB,OAAsD;QAEtD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;QACrC,IAAI,CAAC,GAAG,EAAE,CAAC;YACV,OAAO,SAAS,CAAC;QAClB,CAAC;QAED,GAAG,CAAC,OAAO,GAAG,OAAO,CAAC;QAEtB,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC;IAED,MAAM,CAAC,MAAkB;QACxB,OAAO;YACN,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;gBACxB,MAAM,GAAG,GAA4B;oBACpC,OAAO,EAAE,IAAI;iBACb,CAAC;gBAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;oBACvC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;oBAC9C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,EAAE;wBACjC,OAAO,EAAE;4BACR,OAAO,EAAE,WAAW;4BACpB,UAAU,EAAE,SAAS;yBACrB;qBACD,CAAC,CAAC;oBAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBAE7C,OAAO,QAAQ,CAAC;gBACjB,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACvE,CAAC;gBAED,MAAM,MAAM,GAAG,IAAI,YAAY,EAG3B,CAAC;gBAEL,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;gBACzC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;gBAEjC,MAAM,UAAU,GAAwB;oBACvC,MAAM,EAAE,MAAM;oBACd,IAAI,CAAC,IAAI;wBACR,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;oBAC9B,CAAC;oBACD,KAAK,CAAC,IAAI,GAAG,IAAI,EAAE,MAAM,GAAG,EAAE;wBAC7B,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;4BACrB,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;4BACvD,UAAU,CAAC,KAAK,EAAE,CAAC;wBACpB,CAAC;oBACF,CAAC;iBACD,CAAC;gBAEF,CAAC;oBACA,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;oBAC5B,UAAU,CAAC,GAAG,EAAE;wBACf,OAAO,CAAC,UAAU,CAAC,CAAC;oBACrB,CAAC,EAAE,CAAC,CAAC,CAAC;gBACP,CAAC;gBAED,MAAM,MAAM,GAAuB;oBAClC,MAAM;oBACN,OAAO;wBACN,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;4BACrB,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;4BACjE,UAAU,CAAC,KAAK,EAAE,CAAC;wBACpB,CAAC;oBACF,CAAC;oBACD,CAAC,MAAM,CAAC,OAAO,CAAC;wBACf,IAAI,CAAC,OAAO,EAAE,CAAC;oBAChB,CAAC;iBACD,CAAC;gBAEF,OAAO,MAAM,CAAC;YACf,CAAC;SACD,CAAC;IACH,CAAC;CACD"}

package/dist/main/xrpc-error.d.ts

@@ -37,4 +37,15 @@ export declare class NotEnoughResourcesError extends XRPCError {
 export declare class UpstreamTimeoutError extends XRPCError {
     constructor({ status, error, description }?: Partial<XRPCErrorOptions>);
 }
+export interface XRPCSubscriptionErrorOptions {
+    closeCode?: number;
+    error: string;
+    description?: string;
+}
+export declare class XRPCSubscriptionError extends Error {
+    readonly closeCode: number;
+    readonly error: string;
+    readonly description?: string;
+    constructor({ closeCode, error, description }: XRPCSubscriptionErrorOptions);
+}
 //# sourceMappingURL=xrpc-error.d.ts.map

package/dist/main/xrpc-error.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"xrpc-error.d.ts","sourceRoot":"","sources":["../../lib/main/xrpc-error.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,gBAAgB;IAChC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,SAAU,SAAQ,KAAK;IACnC,sBAAsB;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAExB,iBAAiB;IACjB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,oBAAoB;IACpB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;gBAElB,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,gBAAgB;IAS5D,UAAU,IAAI,QAAQ;CAGtB;AAED,qBAAa,mBAAoB,SAAQ,SAAS;gBACrC,EAAE,MAAY,EAAE,KAAwB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGnG;AAED,qBAAa,iBAAkB,SAAQ,SAAS;gBACnC,EACX,MAAY,EACZ,KAAgC,EAChC,WAAW,GACX,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGjC;AAED,qBAAa,cAAe,SAAQ,SAAS;gBAChC,EAAE,MAAY,EAAE,KAAmB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAG9F;AAED,qBAAa,sBAAuB,SAAQ,SAAS;gBACxC,EAAE,MAAY,EAAE,KAA2B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGtG;AAED,qBAAa,mBAAoB,SAAQ,SAAS;gBACrC,EAAE,MAAY,EAAE,KAA6B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGxG;AAED,qBAAa,oBAAqB,SAAQ,SAAS;gBACtC,EAAE,MAAY,EAAE,KAAyB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGpG;AAED,qBAAa,uBAAwB,SAAQ,SAAS;gBACzC,EAAE,MAAY,EAAE,KAA4B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGvG;AAED,qBAAa,oBAAqB,SAAQ,SAAS;gBACtC,EAAE,MAAY,EAAE,KAAyB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGpG"}
+{"version":3,"file":"xrpc-error.d.ts","sourceRoot":"","sources":["../../lib/main/xrpc-error.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,gBAAgB;IAChC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,SAAU,SAAQ,KAAK;IACnC,sBAAsB;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAExB,iBAAiB;IACjB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,oBAAoB;IACpB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;gBAElB,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,gBAAgB;IAS5D,UAAU,IAAI,QAAQ;CAGtB;AAED,qBAAa,mBAAoB,SAAQ,SAAS;gBACrC,EAAE,MAAY,EAAE,KAAwB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGnG;AAED,qBAAa,iBAAkB,SAAQ,SAAS;gBACnC,EACX,MAAY,EACZ,KAAgC,EAChC,WAAW,GACX,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGjC;AAED,qBAAa,cAAe,SAAQ,SAAS;gBAChC,EAAE,MAAY,EAAE,KAAmB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAG9F;AAED,qBAAa,sBAAuB,SAAQ,SAAS;gBACxC,EAAE,MAAY,EAAE,KAA2B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGtG;AAED,qBAAa,mBAAoB,SAAQ,SAAS;gBACrC,EAAE,MAAY,EAAE,KAA6B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGxG;AAED,qBAAa,oBAAqB,SAAQ,SAAS;gBACtC,EAAE,MAAY,EAAE,KAAyB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGpG;AAED,qBAAa,uBAAwB,SAAQ,SAAS;gBACzC,EAAE,MAAY,EAAE,KAA4B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGvG;AAED,qBAAa,oBAAqB,SAAQ,SAAS;gBACtC,EAAE,MAAY,EAAE,KAAyB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM;CAGpG;AAED,MAAM,WAAW,4BAA4B;IAC5C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,qBAAsB,SAAQ,KAAK;IAC/C,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;gBAElB,EAAE,SAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,4BAA4B;CAOlF"}

package/dist/main/xrpc-error.js

@@ -55,4 +55,15 @@ export class UpstreamTimeoutError extends XRPCError {
         super({ status, error, description });
     }
 }
+export class XRPCSubscriptionError extends Error {
+    closeCode;
+    error;
+    description;
+    constructor({ closeCode = 1008, error, description }) {
+        super(`Subscription error: ${error}${description ? ` - ${description}` : ''}`);
+        this.closeCode = closeCode;
+        this.error = error;
+        this.description = description;
+    }
+}
 //# sourceMappingURL=xrpc-error.js.map

package/dist/main/xrpc-error.js.map

@@ -1 +1 @@
-{"version":3,"file":"xrpc-error.js","sourceRoot":"","sources":["../../lib/main/xrpc-error.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,SAAU,SAAQ,KAAK;IACnC,sBAAsB;IACb,MAAM,CAAS;IAExB,iBAAiB;IACR,KAAK,CAAS;IACvB,oBAAoB;IACX,WAAW,CAAU;IAE9B,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAoB;QAC3D,KAAK,CAAC,GAAG,KAAK,MAAM,WAAW,IAAI,2BAA2B,EAAE,CAAC,CAAC;QAElE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IAChC,CAAC;IAED,UAAU;QACT,OAAO,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjG,CAAC;CACD;AAED,MAAM,OAAO,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,gBAAgB,EAAE,WAAW,KAAgC,EAAE;QAClG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,iBAAkB,SAAQ,SAAS;IAC/C,YAAY,EACX,MAAM,GAAG,GAAG,EACZ,KAAK,GAAG,wBAAwB,EAChC,WAAW,MACmB,EAAE;QAChC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,cAAe,SAAQ,SAAS;IAC5C,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,WAAW,EAAE,WAAW,KAAgC,EAAE;QAC7F,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,sBAAuB,SAAQ,SAAS;IACpD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,mBAAmB,EAAE,WAAW,KAAgC,EAAE;QACrG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,qBAAqB,EAAE,WAAW,KAAgC,EAAE;QACvG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,iBAAiB,EAAE,WAAW,KAAgC,EAAE;QACnG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,uBAAwB,SAAQ,SAAS;IACrD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,oBAAoB,EAAE,WAAW,KAAgC,EAAE;QACtG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,iBAAiB,EAAE,WAAW,KAAgC,EAAE;QACnG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD"}
+{"version":3,"file":"xrpc-error.js","sourceRoot":"","sources":["../../lib/main/xrpc-error.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,SAAU,SAAQ,KAAK;IACnC,sBAAsB;IACb,MAAM,CAAS;IAExB,iBAAiB;IACR,KAAK,CAAS;IACvB,oBAAoB;IACX,WAAW,CAAU;IAE9B,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAoB;QAC3D,KAAK,CAAC,GAAG,KAAK,MAAM,WAAW,IAAI,2BAA2B,EAAE,CAAC,CAAC;QAElE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IAChC,CAAC;IAED,UAAU;QACT,OAAO,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjG,CAAC;CACD;AAED,MAAM,OAAO,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,gBAAgB,EAAE,WAAW,KAAgC,EAAE;QAClG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,iBAAkB,SAAQ,SAAS;IAC/C,YAAY,EACX,MAAM,GAAG,GAAG,EACZ,KAAK,GAAG,wBAAwB,EAChC,WAAW,MACmB,EAAE;QAChC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,cAAe,SAAQ,SAAS;IAC5C,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,WAAW,EAAE,WAAW,KAAgC,EAAE;QAC7F,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,sBAAuB,SAAQ,SAAS;IACpD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,mBAAmB,EAAE,WAAW,KAAgC,EAAE;QACrG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,qBAAqB,EAAE,WAAW,KAAgC,EAAE;QACvG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,iBAAiB,EAAE,WAAW,KAAgC,EAAE;QACnG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,uBAAwB,SAAQ,SAAS;IACrD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,oBAAoB,EAAE,WAAW,KAAgC,EAAE;QACtG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAED,MAAM,OAAO,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,iBAAiB,EAAE,WAAW,KAAgC,EAAE;QACnG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;CACD;AAQD,MAAM,OAAO,qBAAsB,SAAQ,KAAK;IACtC,SAAS,CAAS;IAClB,KAAK,CAAS;IACd,WAAW,CAAU;IAE9B,YAAY,EAAE,SAAS,GAAG,IAAI,EAAE,KAAK,EAAE,WAAW,EAAgC;QACjF,KAAK,CAAC,uBAAuB,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAE/E,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IAChC,CAAC;CACD"}

package/lib/auth/jwt.ts

@@ -1,5 +1,6 @@
 import * as v from '@badrap/valita';
 
+import type { Did, Nsid } from '@atcute/lexicons';
 import { isDid, isNsid } from '@atcute/lexicons/syntax';
 import { fromBase64Url } from '@atcute/multibase';
 import { decodeUtf8From, encodeUtf8 } from '@atcute/uint8array';
@@ -13,14 +14,26 @@ const nsidString = v.string().assert(isNsid, `must be an nsid`);
 
 const integer = v.number().assert((input) => input >= 0 && Number.isSafeInteger(input), `must be an integer`);
 
-const jwtHeader = v.object({
+export interface JwtHeader {
+	typ?: string;
+	alg: string;
+}
+
+const jwtHeader: v.Type<JwtHeader> = v.object({
 	typ: v.string().optional(),
 	alg: v.string(),
 });
 
-export interface JwtHeader extends v.Infer<typeof jwtHeader> {}
+export interface JwtPayload {
+	iss: Did;
+	aud: Did;
+	exp: number;
+	iat?: number;
+	lxm?: Nsid;
+	jti?: string;
+}
 
-const jwtPayload = v
+const jwtPayload: v.Type<JwtPayload> = v
 	.object({
 		/** issuer */
 		iss: didString,
@@ -40,8 +53,6 @@ const jwtPayload = v
 		path: ['exp'],
 	});
 
-export interface JwtPayload extends v.Infer<typeof jwtPayload> {}
-
 export interface ParsedJwt {
 	header: JwtHeader;
 	payload: JwtPayload;

package/lib/main/index.ts

@@ -1,3 +1,5 @@
 export * from './response.js';
 export * from './router.js';
 export * from './xrpc-error.js';
+
+export * from './types/websocket.js';

package/lib/main/router.ts

@@ -1,14 +1,26 @@
-import { safeParse, type XRPCProcedureMetadata, type XRPCQueryMetadata } from '@atcute/lexicons/validations';
+import {
+	safeParse,
+	type XRPCProcedureMetadata,
+	type XRPCQueryMetadata,
+	type XRPCSubscriptionMetadata,
+} from '@atcute/lexicons/validations';
 
 import type { Literal, Promisable } from '../types/misc.js';
 
-import type { ProcedureConfig, QueryConfig, UnknownOperationContext } from './types/operation.js';
+import type {
+	ProcedureConfig,
+	QueryConfig,
+	SubscriptionConfig,
+	UnknownOperationContext,
+	UnknownSubscriptionContext,
+} from './types/operation.js';
+import type { WebSocketAdapter } from './types/websocket.js';
+import { encodeErrorFrame, encodeMessageFrame, extractMessageType, omitMessageType } from './utils/frames.js';
 import { createAsyncMiddlewareRunner, type Middleware } from './utils/middlewares.js';
 import { constructMimeValidator } from './utils/request-input.js';
 import { constructParamsHandler } from './utils/request-params.js';
 import { invalidRequest, validationError } from './utils/response.js';
-
-import { XRPCError } from './xrpc-error.js';
+import { XRPCError, XRPCSubscriptionError } from './xrpc-error.js';
 
 type InternalRequestContext = {
 	url: URL;
@@ -26,6 +38,7 @@ export type FetchMiddleware = Middleware<[request: Request], Promise<Response>>;
 
 export type NotFoundHandler = (request: Request) => Promisable<Response>;
 export type ExceptionHandler = (error: unknown, request: Request) => Promisable<Response>;
+export type SubscriptionExceptionHandler = (error: unknown, request: Request) => void;
 
 export const defaultExceptionHandler: ExceptionHandler = (error: unknown) => {
 	if (error instanceof XRPCError) {
@@ -46,16 +59,24 @@ export const defaultNotFoundHandler: NotFoundHandler = () => {
 	return new Response('Not Found', { status: 404 });
 };
 
+export const defaultSubscriptionExceptionHandler: SubscriptionExceptionHandler = (error: unknown) => {
+	throw error;
+};
+
 export interface XRPCRouterOptions {
 	middlewares?: FetchMiddleware[];
 	handleNotFound?: NotFoundHandler;
 	handleException?: ExceptionHandler;
+	handleSubscriptionException?: SubscriptionExceptionHandler;
+	websocket?: WebSocketAdapter;
 }
 
 export class XRPCRouter {
 	#handlers: Record<string, InternalRouteData> = {};
 	#handleNotFound: NotFoundHandler;
 	#handleException: ExceptionHandler;
+	#handleSubscriptionException: SubscriptionExceptionHandler;
+	#websocket?: WebSocketAdapter;
 
 	fetch: (request: Request) => Promise<Response>;
 
@@ -63,12 +84,16 @@ export class XRPCRouter {
 		middlewares = [],
 		handleException = defaultExceptionHandler,
 		handleNotFound = defaultNotFoundHandler,
+		handleSubscriptionException = defaultSubscriptionExceptionHandler,
+		websocket,
 	}: XRPCRouterOptions = {}) {
 		const runner = createAsyncMiddlewareRunner([...middlewares, (request) => this.#dispatch(request)]);
 
 		this.fetch = (request) => runner(request);
 		this.#handleException = handleException;
 		this.#handleNotFound = handleNotFound;
+		this.#handleSubscriptionException = handleSubscriptionException;
+		this.#websocket = websocket;
 	}
 
 	async #dispatch(request: Request): Promise<Response> {
@@ -80,8 +105,8 @@ export class XRPCRouter {
 		}
 
 		const nsid = pathname.slice('/xrpc/'.length);
-		const route = this.#handlers[nsid];
 
+		const route = this.#handlers[nsid];
 		if (route === undefined) {
 			return this.#handleNotFound(request);
 		}
@@ -105,6 +130,7 @@ export class XRPCRouter {
 		}
 	}
 
+	/** @deprecated use `addQuery` and `addProcedure` instead */
 	add<TQuery extends XRPCQueryMetadata>(query: TQuery, config: QueryConfig<TQuery>): void;
 	add<TProcedure extends XRPCProcedureMetadata>(
 		procedure: TProcedure,
@@ -113,15 +139,18 @@ export class XRPCRouter {
 	add(operation: XRPCQueryMetadata | XRPCProcedureMetadata, config: any): void {
 		switch (operation.type) {
 			case 'xrpc_query': {
-				return this.#addQuery(operation, config);
+				return this.addQuery(operation, config);
 			}
 			case 'xrpc_procedure': {
-				return this.#addProcedure(operation, config);
+				return this.addProcedure(operation, config);
 			}
 		}
 	}
 
-	#addQuery<TQuery extends XRPCQueryMetadata>(query: TQuery, config: QueryConfig<TQuery>): void {
+	addQuery<TQuery extends XRPCQueryMetadata, TConfig extends QueryConfig<TQuery>>(
+		query: TQuery,
+		config: TConfig,
+	): void {
 		const handleParams = query.params ? constructParamsHandler(query.params) : null;
 
 		const handler = config.handler;
@@ -158,9 +187,9 @@ export class XRPCRouter {
 		};
 	}
 
-	#addProcedure<TProcedure extends XRPCProcedureMetadata>(
+	addProcedure<TProcedure extends XRPCProcedureMetadata, TConfig extends ProcedureConfig<TProcedure>>(
 		procedure: TProcedure,
-		config: ProcedureConfig<TProcedure>,
+		config: TConfig,
 	): void {
 		const handleParams = procedure.params ? constructParamsHandler(procedure.params) : null;
 		const validateInputType = procedure.input ? constructMimeValidator(procedure.input) : null;
@@ -236,4 +265,96 @@ export class XRPCRouter {
 			},
 		};
 	}
+
+	addSubscription<
+		TSubscription extends XRPCSubscriptionMetadata,
+		TConfig extends SubscriptionConfig<TSubscription>,
+	>(subscription: TSubscription, config: TConfig): void {
+		const websocket = this.#websocket;
+		if (websocket === undefined) {
+			throw new Error(`WebSocket adapter not configured`);
+		}
+
+		const nsid = subscription.nsid;
+
+		const handleParams = subscription.params ? constructParamsHandler(subscription.params) : null;
+		const handler = config.handler;
+
+		this.#handlers[nsid] = {
+			method: 'GET',
+			handler: async ({ request, url }) => {
+				{
+					const con = request.headers.get('connection');
+					const req = request.headers.get('upgrade');
+					if (
+						con === null ||
+						req === null ||
+						con.toLowerCase() !== 'upgrade' ||
+						req.toLowerCase() !== 'websocket'
+					) {
+						return invalidRequest(`invalid WebSocket upgrade`);
+					}
+				}
+
+				let params: Record<string, Literal | Literal[]>;
+
+				if (handleParams !== null) {
+					const result = handleParams(url.searchParams);
+					if (!result.ok) {
+						return validationError('params', result);
+					}
+
+					params = result.value;
+				} else {
+					params = {};
+				}
+
+				const upgrade = await websocket.upgrade(request, async (ws) => {
+					const signal = ws.signal;
+
+					const context: UnknownSubscriptionContext = {
+						request: request,
+						params: params,
+						signal: signal,
+					};
+
+					try {
+						for await (const message of handler(context)) {
+							if (signal.aborted) {
+								break;
+							}
+
+							const type = extractMessageType(message, nsid);
+							const body = omitMessageType(message);
+
+							const frame = encodeMessageFrame(body, type);
+							await ws.send(frame);
+						}
+
+						ws.close(1000);
+					} catch (err) {
+						if (err instanceof XRPCSubscriptionError) {
+							const frame = encodeErrorFrame(err.error, err.description);
+
+							try {
+								await ws.send(frame);
+							} catch {}
+
+							ws.close(err.closeCode, err.error);
+							return;
+						}
+
+						ws.close(1011, `internal server error`);
+						this.#handleSubscriptionException(err, request);
+					}
+				});
+
+				if (upgrade !== undefined) {
+					return upgrade;
+				}
+
+				return invalidRequest(`WebSocket upgrade failed`);
+			},
+		};
+	}
 }

package/lib/main/types/operation.ts

@@ -1,10 +1,12 @@
 import type {
 	InferOutput,
 	ObjectSchema,
+	VariantSchema,
 	XRPCBlobBodyParam,
 	XRPCLexBodyParam,
 	XRPCProcedureMetadata,
 	XRPCQueryMetadata,
+	XRPCSubscriptionMetadata,
 } from '@atcute/lexicons/validations';
 
 import type { Literal, Promisable } from '../../types/misc.js';
@@ -85,3 +87,34 @@ export type ProcedureHandler<TProcedure extends XRPCProcedureMetadata> = (
 export type ProcedureConfig<TProcedure extends XRPCProcedureMetadata = XRPCProcedureMetadata> = {
 	handler: ProcedureHandler<TProcedure>;
 };
+
+// #region Subscription
+
+export interface UnknownSubscriptionContext {
+	request: Request;
+	signal: AbortSignal;
+	params: Record<string, Literal | Literal[]>;
+}
+
+export type SubscriptionContext<TSubscription extends XRPCSubscriptionMetadata> = {
+	request: Request;
+	signal: AbortSignal;
+} & (TSubscription['params'] extends ObjectSchema
+	? {
+			params: InferOutput<TSubscription['params']>;
+		}
+	: {
+			// params
+		});
+
+export type SubscriptionHandler<TSubscription extends XRPCSubscriptionMetadata> = (
+	context: SubscriptionContext<TSubscription>,
+) => AsyncIterable<
+	TSubscription['message'] extends ObjectSchema | VariantSchema<any>
+		? InferOutput<TSubscription['message']>
+		: never
+>;
+
+export type SubscriptionConfig<TSubscription extends XRPCSubscriptionMetadata = XRPCSubscriptionMetadata> = {
+	handler: SubscriptionHandler<TSubscription>;
+};

package/lib/main/types/websocket.ts

@@ -0,0 +1,14 @@
+import type { Promisable } from '../../types/misc.js';
+
+export interface WebSocketConnection {
+	signal: AbortSignal;
+	send(data: Uint8Array): void | Promise<void>;
+	close(code?: number, reason?: string): void;
+}
+
+export interface WebSocketAdapter {
+	upgrade(
+		request: Request,
+		handler: (ws: WebSocketConnection) => Promisable<void>,
+	): Promisable<Response | undefined>;
+}

package/lib/main/utils/event-emitter.ts

@@ -0,0 +1,116 @@
+/** Converts a tuple into a listener function */
+export type ListenerFor<T extends any[]> = (...args: T) => void;
+
+/** Generic record of the event name and its argument tuple */
+export type EventMap = {
+	[key: string | symbol]: any[];
+};
+
+type MaybeArray<T> = T | T[];
+
+/** Event emitter */
+export class EventEmitter<Events extends EventMap> {
+	#events?: { [E in keyof Events]?: MaybeArray<ListenerFor<Events[E]>> };
+
+	/**
+	 * Appends a listener for the specified event name
+	 * @param name Name of the event
+	 * @param listener Callback that should be invoked when an event is dispatched
+	 * @returns Cleanup function that can be called to remove it
+	 */
+	on<E extends keyof Events>(name: E, listener: ListenerFor<Events[E]>): () => void {
+		let events = this.#events;
+		let existing: MaybeArray<ListenerFor<Events[E]>> | undefined;
+
+		if (events === undefined) {
+			events = this.#events = Object.create(null);
+		} else {
+			existing = events[name];
+		}
+
+		if (existing === undefined) {
+			events![name] = listener;
+		} else if (typeof existing === 'function') {
+			events![name] = [existing, listener];
+		} else {
+			events![name] = existing.concat(listener);
+		}
+
+		// @ts-expect-error: complains about `listener`
+		return this.off.bind(this, name, listener);
+	}
+
+	/**
+	 * Remove listener from the specified event name
+	 * @param name Name of the event
+	 * @param listener Callback to remove
+	 */
+	off<E extends keyof Events>(name: E, listener: ListenerFor<Events[E]>): void {
+		const events = this.#events;
+
+		if (events === undefined) {
+			return;
+		}
+
+		const list = events[name];
+
+		if (list == undefined) {
+			return;
+		}
+
+		if (list === listener) {
+			delete events[name];
+		} else if (typeof list !== 'function') {
+			const index = list.indexOf(listener);
+
+			if (index !== -1) {
+				if (list.length === 2) {
+					// ^ flips the bit, it's either 0 or 1 here.
+					events[name] = list[index ^ 1];
+				} else {
+					events[name] = list.toSpliced(index, 1);
+				}
+			}
+		}
+	}
+
+	/**
+	 * Emit an event with the specified name and its payload
+	 * @param name Name of the event
+	 * @param args Payload for the event
+	 * @returns Whether a listener has been called
+	 */
+	emit<E extends keyof Events>(name: E, ...args: Events[E]): boolean {
+		const events = this.#events;
+
+		if (events === undefined) {
+			return false;
+		}
+
+		const handler = events[name];
+
+		if (handler === undefined) {
+			return false;
+		}
+
+		if (typeof handler === 'function') {
+			handler.apply(this, args);
+		} else {
+			for (let idx = 0, len = handler.length; idx < len; idx++) {
+				handler[idx].apply(this, args);
+			}
+		}
+
+		return true;
+	}
+
+	/**
+	 * Determines if there is a listener on a specified event name
+	 * @param name Name of the event
+	 * @returns Whether there is a listener registered
+	 */
+	has(name: keyof Events): boolean {
+		const events = this.#events;
+		return events !== undefined && name in events;
+	}
+}

package/lib/main/utils/frames.ts

@@ -0,0 +1,71 @@
+import { encode } from '@atcute/cbor';
+import { concat } from '@atcute/uint8array';
+
+interface MessageFrameHeader {
+	op: 1;
+	t?: string; // Type discriminator for union messages (relative to NSID)
+}
+
+interface ErrorFrameHeader {
+	op: -1;
+}
+
+interface ErrorFrameBody {
+	error: string;
+	message?: string;
+}
+
+export const encodeMessageFrame = (body: unknown, type?: string): Uint8Array => {
+	const header: MessageFrameHeader = {
+		op: 1,
+		t: type,
+	};
+
+	return concat([encode(header), encode(body)]);
+};
+
+export const encodeErrorFrame = (error: string, message?: string): Uint8Array => {
+	const header: ErrorFrameHeader = {
+		op: -1,
+	};
+
+	const body: ErrorFrameBody = {
+		error: error,
+		message: message,
+	};
+
+	return concat([encode(header), encode(body)]);
+};
+
+export const extractMessageType = (message: unknown, nsid: string): string | undefined => {
+	if (typeof message !== 'object' || message === null) {
+		return undefined;
+	}
+
+	const obj = message as Record<string, unknown>;
+	const type = obj.$type;
+
+	if (typeof type !== 'string') {
+		return undefined;
+	}
+
+	// If type starts with the subscription NSID, make it relative
+	// e.g., "com.atproto.sync.subscribeRepos#commit" → "#commit"
+	if (type.startsWith(nsid + '#')) {
+		return type.slice(nsid.length);
+	}
+
+	// Otherwise return the full type
+	return type;
+};
+
+export const omitMessageType = (message: unknown): unknown => {
+	if (typeof message !== 'object' || message === null) {
+		return message;
+	}
+
+	const obj = message as Record<string, unknown>;
+	const { $type, ...rest } = obj;
+
+	return rest;
+};