@atcute/xrpc-server 0.1.2 → 0.1.3

package/lib/main/utils/websocket-mock.ts

@@ -0,0 +1,111 @@
+import { AsyncLocalStorage } from 'node:async_hooks';
+
+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';
+
+interface WebSocketHandlerContext {
+	handler: ((ws: WebSocketConnection) => Promisable<void>) | null;
+}
+
+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 class MockWebSocketAdapter implements WebSocketAdapter {
+	#context = new AsyncLocalStorage<WebSocketHandlerContext>();
+
+	upgrade(
+		_request: Request,
+		handler: (ws: WebSocketConnection) => Promisable<void>,
+	): Promisable<Response | undefined> {
+		const ctx = this.#context.getStore();
+		if (!ctx) {
+			return undefined;
+		}
+
+		ctx.handler = handler;
+
+		return new Response(null);
+	}
+
+	attach(router: XRPCRouter): SubscriptionMock {
+		return {
+			subscribe: async (url) => {
+				const ctx: WebSocketHandlerContext = {
+					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<{
+					message: [data: Uint8Array];
+					close: [event: { code: number; reason: string; wasClean: boolean }];
+				}>();
+
+				const controller = new AbortController();
+				const signal = controller.signal;
+
+				const connection: WebSocketConnection = {
+					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: SubscriptionClient = {
+					events,
+					dispose() {
+						if (!signal.aborted) {
+							events.emit('close', { code: 1000, reason: '', wasClean: true });
+							controller.abort();
+						}
+					},
+					[Symbol.dispose]() {
+						this.dispose();
+					},
+				};
+
+				return client;
+			},
+		};
+	}
+}

package/lib/main/xrpc-error.ts

@@ -78,3 +78,23 @@ export class UpstreamTimeoutError extends XRPCError {
 		super({ status, error, description });
 	}
 }
+
+export interface XRPCSubscriptionErrorOptions {
+	closeCode?: number;
+	error: string;
+	description?: string;
+}
+
+export class XRPCSubscriptionError extends Error {
+	readonly closeCode: number;
+	readonly error: string;
+	readonly description?: string;
+
+	constructor({ closeCode = 1008, error, description }: XRPCSubscriptionErrorOptions) {
+		super(`Subscription error: ${error}${description ? ` - ${description}` : ''}`);
+
+		this.closeCode = closeCode;
+		this.error = error;
+		this.description = description;
+	}
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@atcute/xrpc-server",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "a small web framework for handling XRPC operations",
   "license": "0BSD",
   "repository": {
@@ -22,19 +22,21 @@
   "dependencies": {
     "@badrap/valita": "^0.4.6",
     "nanoid": "^5.1.5",
+    "@atcute/cbor": "^2.2.7",
+    "@atcute/identity-resolver": "^1.1.4",
     "@atcute/crypto": "^2.2.5",
     "@atcute/identity": "^1.1.1",
     "@atcute/lexicons": "^1.2.2",
-    "@atcute/identity-resolver": "^1.1.4",
     "@atcute/multibase": "^1.1.6",
     "@atcute/uint8array": "^1.0.5"
   },
   "devDependencies": {
     "@atcute/xrpc-server": "file:",
+    "@types/node": "^24.9.1",
     "@vitest/coverage-v8": "^3.2.4",
     "vitest": "^3.2.4",
-    "@atcute/atproto": "^3.1.6",
-    "@atcute/bluesky": "^3.2.5"
+    "@atcute/bluesky": "^3.2.8",
+    "@atcute/atproto": "^3.1.8"
   },
   "scripts": {
     "build": "tsc --project tsconfig.build.json",