@atcute/xrpc-server 0.1.2 → 0.1.4

package/dist/main/utils/websocket-mock.d.ts

@@ -0,0 +1,24 @@
+import type { Promisable } from '../../types/misc.js';
+import type { XRPCRouter } from '../router.js';
+import type { WebSocketAdapter, WebSocketConnection } from '../types/websocket.js';
+import { EventEmitter } from './event-emitter.js';
+export interface SubscriptionClient extends Disposable {
+    events: EventEmitter<{
+        message: [data: Uint8Array];
+        close: [event: {
+            code: number;
+            reason: string;
+            wasClean: boolean;
+        }];
+    }>;
+    dispose(): void;
+}
+export interface SubscriptionMock {
+    subscribe(url: string): Promise<SubscriptionClient>;
+}
+export declare class MockWebSocketAdapter implements WebSocketAdapter {
+    #private;
+    upgrade(_request: Request, handler: (ws: WebSocketConnection) => Promisable<void>): Promisable<Response | undefined>;
+    attach(router: XRPCRouter): SubscriptionMock;
+}
+//# sourceMappingURL=websocket-mock.d.ts.map

package/dist/main/utils/websocket-mock.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-mock.d.ts","sourceRoot":"","sources":["../../../lib/main/utils/websocket-mock.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AACnF,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAMlD,MAAM,WAAW,kBAAmB,SAAQ,UAAU;IACrD,MAAM,EAAE,YAAY,CAAC;QACpB,OAAO,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5B,KAAK,EAAE,CAAC,KAAK,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAC;YAAC,QAAQ,EAAE,OAAO,CAAA;SAAE,CAAC,CAAC;KACpE,CAAC,CAAC;IACH,OAAO,IAAI,IAAI,CAAC;CAChB;AAED,MAAM,WAAW,gBAAgB;IAChC,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;CACpD;AAED,qBAAa,oBAAqB,YAAW,gBAAgB;;IAG5D,OAAO,CACN,QAAQ,EAAE,OAAO,EACjB,OAAO,EAAE,CAAC,EAAE,EAAE,mBAAmB,KAAK,UAAU,CAAC,IAAI,CAAC,GACpD,UAAU,CAAC,QAAQ,GAAG,SAAS,CAAC,CASlC;IAED,MAAM,CAAC,MAAM,EAAE,UAAU,GAAG,gBAAgB,CAqE3C;CACD"}

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,EACnB;QACnC,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;IAAA,CAC1B;IAED,MAAM,CAAC,MAAkB,EAAoB;QAC5C,OAAO;YACN,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;gBACzB,MAAM,GAAG,GAA4B;oBACpC,OAAO,EAAE,IAAI;iBACb,CAAC;gBAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC;oBACxC,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;gBAAA,CAChB,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,EAAE;wBACV,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;oBAAA,CAC7B;oBACD,KAAK,CAAC,IAAI,GAAG,IAAI,EAAE,MAAM,GAAG,EAAE,EAAE;wBAC/B,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;oBAAA,CACD;iBACD,CAAC;gBAEF,CAAC;oBACA,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;oBAC5B,UAAU,CAAC,GAAG,EAAE,CAAC;wBAChB,OAAO,CAAC,UAAU,CAAC,CAAC;oBAAA,CACpB,EAAE,CAAC,CAAC,CAAC;gBACP,CAAC;gBAED,MAAM,MAAM,GAAuB;oBAClC,MAAM;oBACN,OAAO,GAAG;wBACT,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;oBAAA,CACD;oBACD,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG;wBAClB,IAAI,CAAC,OAAO,EAAE,CAAC;oBAAA,CACf;iBACD,CAAC;gBAEF,OAAO,MAAM,CAAC;YAAA,CACd;SACD,CAAC;IAAA,CACF;CACD"}

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

@@ -17,7 +17,7 @@ export declare class InvalidRequestError extends XRPCError {
     constructor({ status, error, description }?: Partial<XRPCErrorOptions>);
 }
 export declare class AuthRequiredError extends XRPCError {
-    constructor({ status, error, description, }?: Partial<XRPCErrorOptions>);
+    constructor({ status, error, description }?: Partial<XRPCErrorOptions>);
 }
 export declare class ForbiddenError extends XRPCError {
     constructor({ status, error, description }?: Partial<XRPCErrorOptions>);
@@ -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;IAE9B,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,gBAAgB,EAO3D;IAED,UAAU,IAAI,QAAQ,CAErB;CACD;AAED,qBAAa,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAY,EAAE,KAAwB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAElG;CACD;AAED,qBAAa,iBAAkB,SAAQ,SAAS;IAC/C,YAAY,EACX,MAAY,EACZ,KAAgC,EAChC,WAAW,EACX,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAEhC;CACD;AAED,qBAAa,cAAe,SAAQ,SAAS;IAC5C,YAAY,EAAE,MAAY,EAAE,KAAmB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAE7F;CACD;AAED,qBAAa,sBAAuB,SAAQ,SAAS;IACpD,YAAY,EAAE,MAAY,EAAE,KAA2B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAErG;CACD;AAED,qBAAa,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAY,EAAE,KAA6B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAEvG;CACD;AAED,qBAAa,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAY,EAAE,KAAyB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAEnG;CACD;AAED,qBAAa,uBAAwB,SAAQ,SAAS;IACrD,YAAY,EAAE,MAAY,EAAE,KAA4B,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAEtG;CACD;AAED,qBAAa,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAY,EAAE,KAAyB,EAAE,WAAW,EAAE,GAAE,OAAO,CAAC,gBAAgB,CAAM,EAEnG;CACD;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;IAE9B,YAAY,EAAE,SAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,4BAA4B,EAMjF;CACD"}

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,EAAE;QAC7D,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;IAAA,CAC/B;IAED,UAAU,GAAa;QACtB,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;IAAA,CAChG;CACD;AAED,MAAM,OAAO,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,gBAAgB,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QACpG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,iBAAkB,SAAQ,SAAS;IAC/C,YAAY,EACX,MAAM,GAAG,GAAG,EACZ,KAAK,GAAG,wBAAwB,EAChC,WAAW,GACX,GAA8B,EAAE,EAAE;QAClC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,cAAe,SAAQ,SAAS;IAC5C,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,WAAW,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QAC/F,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,sBAAuB,SAAQ,SAAS;IACpD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,mBAAmB,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QACvG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,mBAAoB,SAAQ,SAAS;IACjD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,qBAAqB,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QACzG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,iBAAiB,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QACrG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,uBAAwB,SAAQ,SAAS;IACrD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,oBAAoB,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QACxG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;CACD;AAED,MAAM,OAAO,oBAAqB,SAAQ,SAAS;IAClD,YAAY,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,GAAG,iBAAiB,EAAE,WAAW,EAAE,GAA8B,EAAE,EAAE;QACrG,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAAA,CACtC;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,EAAE;QACnF,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;IAAA,CAC/B;CACD"}

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

@@ -0,0 +1,23 @@
+import type { XRPCProcedureMetadata, XRPCQueryMetadata } from '@atcute/lexicons/validations';
+import { type XRPCRouterOptions } from './router.js';
+import type { ProcedureConfig, QueryConfig } from './types/operation.js';
+import { type Namespaced } from './utils/namespaced.js';
+type XrpcHandlerRouterOptions = Pick<XRPCRouterOptions, 'middlewares' | 'handleNotFound' | 'handleException'>;
+export type XrpcQueryHandlerOptions<TQuery extends XRPCQueryMetadata> = {
+    lxm: TQuery | Namespaced<TQuery>;
+    routerOptions?: XrpcHandlerRouterOptions;
+} & QueryConfig<TQuery>;
+export type XrpcProcedureHandlerOptions<TProcedure extends XRPCProcedureMetadata> = {
+    lxm: TProcedure | Namespaced<TProcedure>;
+    routerOptions?: XrpcHandlerRouterOptions;
+} & ProcedureConfig<TProcedure>;
+export type XrpcHandlerOptions = XrpcQueryHandlerOptions<XRPCQueryMetadata> | XrpcProcedureHandlerOptions<XRPCProcedureMetadata>;
+/**
+ * create a fetch handler for a single xrpc query or procedure.
+ * requests are expected at `/xrpc/<nsid>`.
+ * subscriptions are not supported.
+ */
+export declare function createXrpcHandler<TQuery extends XRPCQueryMetadata>(options: XrpcQueryHandlerOptions<TQuery>): (request: Request) => Promise<Response>;
+export declare function createXrpcHandler<TProcedure extends XRPCProcedureMetadata>(options: XrpcProcedureHandlerOptions<TProcedure>): (request: Request) => Promise<Response>;
+export {};
+//# sourceMappingURL=xrpc-handler.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"xrpc-handler.d.ts","sourceRoot":"","sources":["../../lib/main/xrpc-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAE7F,OAAO,EAAc,KAAK,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,KAAK,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACzE,OAAO,EAAa,KAAK,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,KAAK,wBAAwB,GAAG,IAAI,CAAC,iBAAiB,EAAE,aAAa,GAAG,gBAAgB,GAAG,iBAAiB,CAAC,CAAC;AAE9G,MAAM,MAAM,uBAAuB,CAAC,MAAM,SAAS,iBAAiB,IAAI;IACvE,GAAG,EAAE,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;IACjC,aAAa,CAAC,EAAE,wBAAwB,CAAC;CACzC,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;AAExB,MAAM,MAAM,2BAA2B,CAAC,UAAU,SAAS,qBAAqB,IAC/E;IACC,GAAG,EAAE,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;IACzC,aAAa,CAAC,EAAE,wBAAwB,CAAC;CACzC,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;AAEjC,MAAM,MAAM,kBAAkB,GAC3B,uBAAuB,CAAC,iBAAiB,CAAC,GAC1C,2BAA2B,CAAC,qBAAqB,CAAC,CAAC;AAEtD;;;;GAIG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,SAAS,iBAAiB,EACjE,OAAO,EAAE,uBAAuB,CAAC,MAAM,CAAC,GACtC,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;AAC3C,wBAAgB,iBAAiB,CAAC,UAAU,SAAS,qBAAqB,EACzE,OAAO,EAAE,2BAA2B,CAAC,UAAU,CAAC,GAC9C,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC"}

package/dist/main/xrpc-handler.js

@@ -0,0 +1,19 @@
+import { XRPCRouter } from './router.js';
+import { unwrapLxm } from './utils/namespaced.js';
+export function createXrpcHandler(options) {
+    const { lxm, handler, routerOptions } = options;
+    const router = new XRPCRouter(routerOptions);
+    const schema = unwrapLxm(lxm);
+    switch (schema.type) {
+        case 'xrpc_query': {
+            router.addQuery(schema, { handler: handler });
+            break;
+        }
+        case 'xrpc_procedure': {
+            router.addProcedure(schema, { handler: handler });
+            break;
+        }
+    }
+    return router.fetch;
+}
+//# sourceMappingURL=xrpc-handler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"xrpc-handler.js","sourceRoot":"","sources":["../../lib/main/xrpc-handler.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAA0B,MAAM,aAAa,CAAC;AAEjE,OAAO,EAAE,SAAS,EAAmB,MAAM,uBAAuB,CAAC;AA8BnE,MAAM,UAAU,iBAAiB,CAAC,OAA2B,EAA2C;IACvG,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;IAEhD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,aAAa,CAAC,CAAC;IAE7C,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAE9B,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,YAAY,EAAE,CAAC;YACnB,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,OAAoD,EAAE,CAAC,CAAC;YAC3F,MAAM;QACP,CAAC;QACD,KAAK,gBAAgB,EAAE,CAAC;YACvB,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,OAA4D,EAAE,CAAC,CAAC;YACvG,MAAM;QACP,CAAC;IACF,CAAC;IAED,OAAO,MAAM,CAAC,KAAK,CAAC;AAAA,CACpB"}

package/dist/middlewares/cors.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"cors.d.ts","sourceRoot":"","sources":["../../lib/middlewares/cors.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,WAAW,WAAW;IAC3B,iDAAiD;IACjD,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,kCAAkC;IAClC,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAsBD,eAAO,MAAM,IAAI,GAAI,UAAS,WAAgB,KAAG,eAsChD,CAAC"}
+{"version":3,"file":"cors.d.ts","sourceRoot":"","sources":["../../lib/middlewares/cors.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,WAAW,WAAW;IAC3B,iDAAiD;IACjD,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,kCAAkC;IAClC,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAsBD,eAAO,MAAM,IAAI,4CAsChB,CAAC"}

package/dist/middlewares/cors.js.map

@@ -1 +1 @@
-{"version":3,"file":"cors.js","sourceRoot":"","sources":["../../lib/middlewares/cors.ts"],"names":[],"mappings":"AASA,MAAM,uBAAuB,GAAG;IAC/B,YAAY;IACZ,kBAAkB;IAElB,iBAAiB;IACjB,kBAAkB;IAClB,qBAAqB;IACrB,iBAAiB;CACjB,CAAC;AAEF,MAAM,uBAAuB,GAAG;IAC/B,cAAc;IAEd,eAAe;IACf,MAAM;IAEN,yBAAyB;IACzB,eAAe;CACf,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,UAAuB,EAAE,EAAmB,EAAE;IAClE,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAChC,IAAI,GAAG,CAAC,CAAC,GAAG,uBAAuB,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CACrG,CAAC,IAAI,EAAE,CAAC;IAET,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAChC,IAAI,GAAG,CAAC,CAAC,GAAG,uBAAuB,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CACrG;SACC,IAAI,EAAE;SACN,IAAI,CAAC,GAAG,CAAC,CAAC;IAEZ,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;QAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC;QAEpD,4BAA4B;QAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;YAEnD,IAAI,cAAc,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE,cAAc,CAAC,CAAC;YAC7D,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;QAErC,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE/E,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;QAC5D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,+BAA+B,EAAE,MAAM,CAAC,CAAC;QAClE,CAAC;QAED,OAAO,QAAQ,CAAC;IACjB,CAAC,CAAC;AACH,CAAC,CAAC"}
+{"version":3,"file":"cors.js","sourceRoot":"","sources":["../../lib/middlewares/cors.ts"],"names":[],"mappings":"AASA,MAAM,uBAAuB,GAAG;IAC/B,YAAY;IACZ,kBAAkB;IAElB,iBAAiB;IACjB,kBAAkB;IAClB,qBAAqB;IACrB,iBAAiB;CACjB,CAAC;AAEF,MAAM,uBAAuB,GAAG;IAC/B,cAAc;IAEd,eAAe;IACf,MAAM;IAEN,yBAAyB;IACzB,eAAe;CACf,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,OAAO,GAAgB,EAAE,EAAmB,EAAE,CAAC;IACnE,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAChC,IAAI,GAAG,CAAC,CAAC,GAAG,uBAAuB,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CACrG,CAAC,IAAI,EAAE,CAAC;IAET,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAChC,IAAI,GAAG,CAAC,CAAC,GAAG,uBAAuB,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CACrG;SACC,IAAI,EAAE;SACN,IAAI,CAAC,GAAG,CAAC,CAAC;IAEZ,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC;QAEpD,4BAA4B;QAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;YAEnD,IAAI,cAAc,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE,cAAc,CAAC,CAAC;YAC7D,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;QAErC,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE/E,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;QAC5D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,+BAA+B,EAAE,MAAM,CAAC,CAAC;QAClE,CAAC;QAED,OAAO,QAAQ,CAAC;IAAA,CAChB,CAAC;AAAA,CACF,CAAC"}

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,6 @@
 export * from './response.js';
+export * from './xrpc-handler.js';
 export * from './router.js';
 export * from './xrpc-error.js';
+
+export * from './types/websocket.js';

package/lib/main/router.ts

@@ -1,14 +1,27 @@
-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 { unwrapLxm, type Namespaced } from './utils/namespaced.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 +39,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 +60,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 +85,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 +106,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,28 +131,41 @@ export class XRPCRouter {
 		}
 	}
 
-	add<TQuery extends XRPCQueryMetadata>(query: TQuery, config: QueryConfig<TQuery>): void;
+	/** @deprecated use `addQuery` and `addProcedure` instead */
+	add<TQuery extends XRPCQueryMetadata>(query: TQuery | Namespaced<TQuery>, config: QueryConfig<TQuery>): void;
 	add<TProcedure extends XRPCProcedureMetadata>(
-		procedure: TProcedure,
+		procedure: TProcedure | Namespaced<TProcedure>,
 		config: ProcedureConfig<TProcedure>,
 	): void;
-	add(operation: XRPCQueryMetadata | XRPCProcedureMetadata, config: any): void {
-		switch (operation.type) {
+	add(
+		operation:
+			| XRPCQueryMetadata
+			| XRPCProcedureMetadata
+			| Namespaced<XRPCQueryMetadata | XRPCProcedureMetadata>,
+		config: any,
+	): void {
+		const schema = unwrapLxm(operation);
+
+		switch (schema.type) {
 			case 'xrpc_query': {
-				return this.#addQuery(operation, config);
+				return this.addQuery(schema, config);
 			}
 			case 'xrpc_procedure': {
-				return this.#addProcedure(operation, config);
+				return this.addProcedure(schema, config);
 			}
 		}
 	}
 
-	#addQuery<TQuery extends XRPCQueryMetadata>(query: TQuery, config: QueryConfig<TQuery>): void {
-		const handleParams = query.params ? constructParamsHandler(query.params) : null;
+	addQuery<TQuery extends XRPCQueryMetadata, TConfig extends QueryConfig<TQuery>>(
+		query: TQuery | Namespaced<TQuery>,
+		config: TConfig,
+	): void {
+		const querySchema = unwrapLxm(query);
+		const handleParams = querySchema.params ? constructParamsHandler(querySchema.params) : null;
 
 		const handler = config.handler;
 
-		this.#handlers[query.nsid] = {
+		this.#handlers[querySchema.nsid] = {
 			method: 'GET',
 			handler: async ({ request, url }) => {
 				let params: Record<string, Literal | Literal[]>;
@@ -158,19 +197,20 @@ export class XRPCRouter {
 		};
 	}
 
-	#addProcedure<TProcedure extends XRPCProcedureMetadata>(
-		procedure: TProcedure,
-		config: ProcedureConfig<TProcedure>,
+	addProcedure<TProcedure extends XRPCProcedureMetadata, TConfig extends ProcedureConfig<TProcedure>>(
+		procedure: TProcedure | Namespaced<TProcedure>,
+		config: TConfig,
 	): void {
-		const handleParams = procedure.params ? constructParamsHandler(procedure.params) : null;
-		const validateInputType = procedure.input ? constructMimeValidator(procedure.input) : null;
+		const procedureSchema = unwrapLxm(procedure);
+		const handleParams = procedureSchema.params ? constructParamsHandler(procedureSchema.params) : null;
+		const validateInputType = procedureSchema.input ? constructMimeValidator(procedureSchema.input) : null;
 
-		const requiresInput = procedure.input !== null;
-		const inputSchema = procedure.input?.type === 'lex' ? procedure.input.schema : null;
+		const requiresInput = procedureSchema.input !== null;
+		const inputSchema = procedureSchema.input?.type === 'lex' ? procedureSchema.input.schema : null;
 
 		const handler = config.handler;
 
-		this.#handlers[procedure.nsid] = {
+		this.#handlers[procedureSchema.nsid] = {
 			method: 'POST',
 			handler: async ({ request, url }) => {
 				let params: Record<string, Literal | Literal[]>;
@@ -203,7 +243,7 @@ export class XRPCRouter {
 						let raw: any;
 						try {
 							raw = await request.json();
-						} catch (err) {
+						} catch {
 							return invalidRequest(`invalid request body (failed to parse json)`);
 						}
 
@@ -236,4 +276,99 @@ export class XRPCRouter {
 			},
 		};
 	}
+
+	addSubscription<
+		TSubscription extends XRPCSubscriptionMetadata,
+		TConfig extends SubscriptionConfig<TSubscription>,
+	>(subscription: TSubscription | Namespaced<TSubscription>, config: TConfig): void {
+		const websocket = this.#websocket;
+		if (websocket === undefined) {
+			throw new Error(`WebSocket adapter not configured`);
+		}
+
+		const subscriptionSchema = unwrapLxm(subscription);
+		const nsid = subscriptionSchema.nsid;
+
+		const handleParams = subscriptionSchema.params
+			? constructParamsHandler(subscriptionSchema.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>;
+}