@derivation/rpc 0.1.6 → 0.1.7

package/dist/web-socket-server.js

@@ -0,0 +1,59 @@
+import { parse } from "url";
+import { WebSocketServer } from "ws";
+import { ClientHandler } from "./client-handler";
+import WeakList from "./weak-list";
+import { NodeWebSocketTransport } from "./node-web-socket-transport";
+export function setupWebSocketServer(graph, server, streamEndpoints, mutationEndpoints, options) {
+    const { createContext, presenceHandler, path = "/api/ws" } = options;
+    const clients = new WeakList();
+    graph.afterStep(() => {
+        for (const client of clients) {
+            client.handleStep();
+        }
+    });
+    const wss = new WebSocketServer({ noServer: true, maxPayload: 100 * 1024 });
+    // Handle WebSocket connections for reactive streams
+    wss.on("connection", (ws, req) => {
+        const { pathname } = parse(req.url || "/", true);
+        if (pathname === path) {
+            const messageBuffer = [];
+            let client = null;
+            // Set up temporary message handler to buffer messages
+            const tempMessageHandler = (msg) => {
+                messageBuffer.push(msg);
+            };
+            ws.on("message", tempMessageHandler);
+            // Create context (handle both sync and async)
+            Promise.resolve(createContext(ws, req))
+                .then((context) => {
+                // Remove temporary handler
+                ws.removeListener("message", tempMessageHandler);
+                // Create transport and client handler
+                const transport = new NodeWebSocketTransport(ws);
+                client = new ClientHandler(transport, context, streamEndpoints, mutationEndpoints, presenceHandler);
+                clients.add(client);
+                // Process buffered messages
+                for (const msg of messageBuffer) {
+                    client.handleMessage(msg.toString());
+                }
+                messageBuffer.length = 0;
+            })
+                .catch((err) => {
+                console.error("Error creating context:", err);
+                ws.close();
+            });
+        }
+    });
+    // Handle HTTP upgrade (for /api/ws)
+    server.on("upgrade", (req, socket, head) => {
+        const { pathname } = parse(req.url || "/", true);
+        if (pathname === "/api/ws") {
+            wss.handleUpgrade(req, socket, head, (ws) => {
+                wss.emit("connection", ws, req);
+            });
+        }
+        else {
+            socket.destroy();
+        }
+    });
+}

package/dist/web-socket-transport.d.ts

@@ -0,0 +1,13 @@
+import { Transport } from "./transport";
+/**
+ * Transport implementation for browser WebSocket (client-side).
+ */
+export declare class WebSocketTransport implements Transport {
+    private ws;
+    constructor(ws: globalThis.WebSocket);
+    send(data: string): void;
+    onMessage(handler: (data: string) => void): void;
+    onClose(handler: () => void): void;
+    close(): void;
+    get bufferedAmount(): number;
+}

package/dist/web-socket-transport.js

@@ -0,0 +1,25 @@
+/**
+ * Transport implementation for browser WebSocket (client-side).
+ */
+export class WebSocketTransport {
+    constructor(ws) {
+        this.ws = ws;
+    }
+    send(data) {
+        this.ws.send(data);
+    }
+    onMessage(handler) {
+        this.ws.onmessage = (event) => {
+            handler(event.data);
+        };
+    }
+    onClose(handler) {
+        this.ws.onclose = handler;
+    }
+    close() {
+        this.ws.close();
+    }
+    get bufferedAmount() {
+        return this.ws.bufferedAmount;
+    }
+}

package/dist/websocket-server.d.ts

@@ -1,5 +1,11 @@
-import { Server } from "http";
+import { Server, IncomingMessage } from "http";
+import { WebSocket } from "ws";
 import { StreamEndpoints, MutationEndpoints, RPCDefinition } from "./stream-types";
 import { Graph } from "derivation";
 import { PresenceHandler } from "./presence-manager";
-export declare function setupWebSocketServer<Defs extends RPCDefinition>(graph: Graph, server: Server, streamEndpoints: StreamEndpoints<Defs["streams"]>, mutationEndpoints: MutationEndpoints<Defs["mutations"]>, presenceHandler?: PresenceHandler, path?: string): void;
+export type WebSocketServerOptions<Ctx> = {
+    createContext: (ws: WebSocket, req: IncomingMessage) => Ctx | Promise<Ctx>;
+    presenceHandler?: PresenceHandler;
+    path?: string;
+};
+export declare function setupWebSocketServer<Defs extends RPCDefinition, Ctx = void>(graph: Graph, server: Server, streamEndpoints: StreamEndpoints<Defs["streams"], Ctx>, mutationEndpoints: MutationEndpoints<Defs["mutations"], Ctx>, options: WebSocketServerOptions<Ctx>): void;

package/dist/websocket-server.js

@@ -2,7 +2,9 @@ import { parse } from "url";
 import { WebSocketServer } from "ws";
 import { ClientHandler } from "./client-handler";
 import WeakList from "./weak-list";
-export function setupWebSocketServer(graph, server, streamEndpoints, mutationEndpoints, presenceHandler, path = "/api/ws") {
+import { NodeWebSocketTransport } from "./websocket-transport";
+export function setupWebSocketServer(graph, server, streamEndpoints, mutationEndpoints, options) {
+    const { createContext, presenceHandler, path = "/api/ws" } = options;
     const clients = new WeakList();
     graph.afterStep(() => {
         for (const client of clients) {
@@ -14,10 +16,32 @@ export function setupWebSocketServer(graph, server, streamEndpoints, mutationEnd
     wss.on("connection", (ws, req) => {
         const { pathname } = parse(req.url || "/", true);
         if (pathname === path) {
-            const client = new ClientHandler(ws, streamEndpoints, mutationEndpoints, presenceHandler);
-            clients.add(client);
-            ws.on("message", (msg) => client.handleMessage(msg));
-            ws.on("close", () => client.handleDisconnect());
+            const messageBuffer = [];
+            let client = null;
+            // Set up temporary message handler to buffer messages
+            const tempMessageHandler = (msg) => {
+                messageBuffer.push(msg);
+            };
+            ws.on("message", tempMessageHandler);
+            // Create context (handle both sync and async)
+            Promise.resolve(createContext(ws, req))
+                .then((context) => {
+                // Remove temporary handler
+                ws.removeListener("message", tempMessageHandler);
+                // Create transport and client handler
+                const transport = new NodeWebSocketTransport(ws);
+                client = new ClientHandler(transport, context, streamEndpoints, mutationEndpoints, presenceHandler);
+                clients.add(client);
+                // Process buffered messages
+                for (const msg of messageBuffer) {
+                    client.handleMessage(msg.toString());
+                }
+                messageBuffer.length = 0;
+            })
+                .catch((err) => {
+                console.error("Error creating context:", err);
+                ws.close();
+            });
         }
     });
     // Handle HTTP upgrade (for /api/ws)

package/dist/websocket-transport.d.ts

@@ -0,0 +1,28 @@
+import { WebSocket } from "ws";
+import { Transport } from "./transport";
+/**
+ * Transport implementation for Node.js WebSocket (server-side using 'ws' library).
+ */
+export declare class NodeWebSocketTransport implements Transport {
+    private ws;
+    private messageHandlerSet;
+    private closeHandlerSet;
+    constructor(ws: WebSocket);
+    send(data: string): void;
+    onMessage(handler: (data: string) => void): void;
+    onClose(handler: () => void): void;
+    close(): void;
+    get bufferedAmount(): number;
+}
+/**
+ * Transport implementation for browser WebSocket (client-side).
+ */
+export declare class WebSocketTransport implements Transport {
+    private ws;
+    constructor(ws: globalThis.WebSocket);
+    send(data: string): void;
+    onMessage(handler: (data: string) => void): void;
+    onClose(handler: () => void): void;
+    close(): void;
+    get bufferedAmount(): number;
+}

package/dist/websocket-transport.js

@@ -0,0 +1,58 @@
+/**
+ * Transport implementation for Node.js WebSocket (server-side using 'ws' library).
+ */
+export class NodeWebSocketTransport {
+    constructor(ws) {
+        this.ws = ws;
+        this.messageHandlerSet = false;
+        this.closeHandlerSet = false;
+    }
+    send(data) {
+        this.ws.send(data);
+    }
+    onMessage(handler) {
+        if (!this.messageHandlerSet) {
+            this.ws.on("message", (data) => {
+                handler(data.toString());
+            });
+            this.messageHandlerSet = true;
+        }
+    }
+    onClose(handler) {
+        if (!this.closeHandlerSet) {
+            this.ws.on("close", handler);
+            this.closeHandlerSet = true;
+        }
+    }
+    close() {
+        this.ws.close();
+    }
+    get bufferedAmount() {
+        return this.ws.bufferedAmount;
+    }
+}
+/**
+ * Transport implementation for browser WebSocket (client-side).
+ */
+export class WebSocketTransport {
+    constructor(ws) {
+        this.ws = ws;
+    }
+    send(data) {
+        this.ws.send(data);
+    }
+    onMessage(handler) {
+        this.ws.onmessage = (event) => {
+            handler(event.data);
+        };
+    }
+    onClose(handler) {
+        this.ws.onclose = handler;
+    }
+    close() {
+        this.ws.close();
+    }
+    get bufferedAmount() {
+        return this.ws.bufferedAmount;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@derivation/rpc",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "homepage": "https://github.com/derivationjs/rpc",
   "repository": {
     "type": "git",
@@ -20,6 +20,26 @@
     "./iso": {
       "types": "./dist/iso.d.ts",
       "import": "./dist/iso.js"
+    },
+    "./worker": {
+      "types": "./dist/shared-worker-server.d.ts",
+      "import": "./dist/shared-worker-server.js"
+    },
+    "./client": {
+      "types": "./dist/shared-worker-client.d.ts",
+      "import": "./dist/shared-worker-client.js"
+    },
+    "./transport": {
+      "types": "./dist/transport.d.ts",
+      "import": "./dist/transport.js"
+    },
+    "./browser": {
+      "types": "./dist/web-socket-transport.d.ts",
+      "import": "./dist/web-socket-transport.js"
+    },
+    "./server": {
+      "types": "./dist/web-socket-server.d.ts",
+      "import": "./dist/web-socket-server.js"
     }
   },
   "files": [
@@ -34,7 +54,8 @@
   "dependencies": {
     "@types/node": "^25.0.3",
     "@types/ws": "^8.18.1",
-    "derivation": "^0.1.6",
+    "derivation": "^0.1.7",
+    "@derivation/relational": "^0.1.7",
     "immutable": "^5.1.4",
     "ws": "^8.18.3",
     "zod": "^4.3.3"