@livekit/agents 0.0.0-20260120144724

package/dist/utils/ws_transport.cjs

@@ -0,0 +1,51 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ws_transport_exports = {};
+__export(ws_transport_exports, {
+  webSocketStream: () => webSocketStream
+});
+module.exports = __toCommonJS(ws_transport_exports);
+var import_node_stream = require("node:stream");
+var import_ws = __toESM(require("ws"), 1);
+function webSocketStream(wsUrl) {
+  const ws = new import_ws.default(wsUrl);
+  const duplex = (0, import_ws.createWebSocketStream)(ws);
+  duplex.on("error", console.error);
+  duplex.on("end", () => {
+    duplex.end();
+  });
+  const writable = import_node_stream.Writable.toWeb(duplex);
+  const readable = import_node_stream.Readable.toWeb(duplex);
+  return { readable, writable, close: ws.close };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  webSocketStream
+});
+//# sourceMappingURL=ws_transport.cjs.map

package/dist/utils/ws_transport.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/ws_transport.ts"],"sourcesContent":["import { Readable, Writable } from 'node:stream';\nimport WebSocket, { createWebSocketStream } from 'ws';\n\nexport function webSocketStream(wsUrl: string) {\n  const ws = new WebSocket(wsUrl);\n  const duplex = createWebSocketStream(ws);\n  duplex.on('error', console.error);\n\n  // End the write side when the read side ends to properly close the stream.\n  // This is needed because Readable.toWeb() waits for both sides of the duplex\n  // to close before signaling done on the ReadableStream.\n  duplex.on('end', () => {\n    duplex.end();\n  });\n\n  // Convert the writable side\n  const writable = Writable.toWeb(duplex);\n  // Convert the readable side\n  const readable = Readable.toWeb(duplex);\n\n  return { readable, writable, close: ws.close };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,yBAAmC;AACnC,gBAAiD;AAE1C,SAAS,gBAAgB,OAAe;AAC7C,QAAM,KAAK,IAAI,UAAAA,QAAU,KAAK;AAC9B,QAAM,aAAS,iCAAsB,EAAE;AACvC,SAAO,GAAG,SAAS,QAAQ,KAAK;AAKhC,SAAO,GAAG,OAAO,MAAM;AACrB,WAAO,IAAI;AAAA,EACb,CAAC;AAGD,QAAM,WAAW,4BAAS,MAAM,MAAM;AAEtC,QAAM,WAAW,4BAAS,MAAM,MAAM;AAEtC,SAAO,EAAE,UAAU,UAAU,OAAO,GAAG,MAAM;AAC/C;","names":["WebSocket"]}

package/dist/utils/ws_transport.d.cts

@@ -0,0 +1,9 @@
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+export declare function webSocketStream(wsUrl: string): {
+    readable: import("stream/web").ReadableStream<any>;
+    writable: import("stream/web").WritableStream<any>;
+    close: (code?: number | undefined, data?: string | Buffer | undefined) => void;
+};
+//# sourceMappingURL=ws_transport.d.ts.map

package/dist/utils/ws_transport.d.ts

@@ -0,0 +1,9 @@
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+export declare function webSocketStream(wsUrl: string): {
+    readable: import("stream/web").ReadableStream<any>;
+    writable: import("stream/web").WritableStream<any>;
+    close: (code?: number | undefined, data?: string | Buffer | undefined) => void;
+};
+//# sourceMappingURL=ws_transport.d.ts.map

package/dist/utils/ws_transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws_transport.d.ts","sourceRoot":"","sources":["../../src/utils/ws_transport.ts"],"names":[],"mappings":";;;AAGA,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM;;;;EAkB5C"}

package/dist/utils/ws_transport.js

@@ -0,0 +1,17 @@
+import { Readable, Writable } from "node:stream";
+import WebSocket, { createWebSocketStream } from "ws";
+function webSocketStream(wsUrl) {
+  const ws = new WebSocket(wsUrl);
+  const duplex = createWebSocketStream(ws);
+  duplex.on("error", console.error);
+  duplex.on("end", () => {
+    duplex.end();
+  });
+  const writable = Writable.toWeb(duplex);
+  const readable = Readable.toWeb(duplex);
+  return { readable, writable, close: ws.close };
+}
+export {
+  webSocketStream
+};
+//# sourceMappingURL=ws_transport.js.map

package/dist/utils/ws_transport.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/ws_transport.ts"],"sourcesContent":["import { Readable, Writable } from 'node:stream';\nimport WebSocket, { createWebSocketStream } from 'ws';\n\nexport function webSocketStream(wsUrl: string) {\n  const ws = new WebSocket(wsUrl);\n  const duplex = createWebSocketStream(ws);\n  duplex.on('error', console.error);\n\n  // End the write side when the read side ends to properly close the stream.\n  // This is needed because Readable.toWeb() waits for both sides of the duplex\n  // to close before signaling done on the ReadableStream.\n  duplex.on('end', () => {\n    duplex.end();\n  });\n\n  // Convert the writable side\n  const writable = Writable.toWeb(duplex);\n  // Convert the readable side\n  const readable = Readable.toWeb(duplex);\n\n  return { readable, writable, close: ws.close };\n}\n"],"mappings":"AAAA,SAAS,UAAU,gBAAgB;AACnC,OAAO,aAAa,6BAA6B;AAE1C,SAAS,gBAAgB,OAAe;AAC7C,QAAM,KAAK,IAAI,UAAU,KAAK;AAC9B,QAAM,SAAS,sBAAsB,EAAE;AACvC,SAAO,GAAG,SAAS,QAAQ,KAAK;AAKhC,SAAO,GAAG,OAAO,MAAM;AACrB,WAAO,IAAI;AAAA,EACb,CAAC;AAGD,QAAM,WAAW,SAAS,MAAM,MAAM;AAEtC,QAAM,WAAW,SAAS,MAAM,MAAM;AAEtC,SAAO,EAAE,UAAU,UAAU,OAAO,GAAG,MAAM;AAC/C;","names":[]}

package/dist/utils/ws_transport.test.cjs

@@ -0,0 +1,212 @@
+"use strict";
+var import_vitest = require("vitest");
+var import_ws = require("ws");
+var import_ws_transport = require("./ws_transport.cjs");
+(0, import_vitest.describe)("webSocketStream", () => {
+  (0, import_vitest.describe)("readable stream", () => {
+    (0, import_vitest.it)("receives messages from the WebSocket", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.send("hello");
+        serverWs.send("world");
+        serverWs.close();
+      });
+      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const messages = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          messages.push(Buffer.from(value).toString());
+        }
+      } finally {
+        reader.releaseLock();
+      }
+      (0, import_vitest.expect)(messages).toEqual(["hello", "world"]);
+      wss.close();
+    });
+    (0, import_vitest.it)("handles binary messages", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);
+      wss.on("connection", (serverWs) => {
+        serverWs.send(binaryData);
+        serverWs.close();
+      });
+      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const chunks = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          chunks.push(new Uint8Array(value));
+        }
+      } finally {
+        reader.releaseLock();
+      }
+      (0, import_vitest.expect)(chunks).toHaveLength(1);
+      (0, import_vitest.expect)(Array.from(chunks[0])).toEqual([1, 2, 3, 4, 5]);
+      wss.close();
+    });
+    (0, import_vitest.it)("handles empty stream when connection closes immediately", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.close();
+      });
+      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const chunks = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          chunks.push(value);
+        }
+      } finally {
+        reader.releaseLock();
+      }
+      (0, import_vitest.expect)(chunks).toEqual([]);
+      wss.close();
+    });
+  });
+  (0, import_vitest.describe)("writable stream", () => {
+    (0, import_vitest.it)("sends messages through the WebSocket", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const ws = new import_ws.WebSocket(`ws://localhost:${port}`);
+      const connected = new Promise((resolve) => {
+        ws.on("open", resolve);
+      });
+      const messagesReceived = [];
+      const serverClosed = new Promise((resolve) => {
+        wss.on("connection", (serverWs) => {
+          serverWs.on("message", (data) => {
+            messagesReceived.push(data.toString());
+          });
+          serverWs.on("close", resolve);
+        });
+      });
+      await connected;
+      const { writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      await writer.write(new TextEncoder().encode("hello"));
+      await writer.write(new TextEncoder().encode("world"));
+      await writer.close();
+      await serverClosed;
+      (0, import_vitest.expect)(messagesReceived).toEqual(["hello", "world"]);
+      wss.close();
+    });
+    (0, import_vitest.it)("sends binary data through the WebSocket", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const chunksReceived = [];
+      const serverClosed = new Promise((resolve) => {
+        wss.on("connection", (serverWs) => {
+          serverWs.on("message", (data) => {
+            chunksReceived.push(Buffer.from(data));
+          });
+          serverWs.on("close", resolve);
+        });
+      });
+      const { writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);
+      await writer.write(binaryData);
+      await writer.close();
+      await serverClosed;
+      (0, import_vitest.expect)(chunksReceived).toHaveLength(1);
+      (0, import_vitest.expect)(Array.from(chunksReceived[0])).toEqual([10, 20, 30, 40, 50]);
+      wss.close();
+    });
+    (0, import_vitest.it)("buffers writes if readyState is CONNECTING", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const { writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      const messagesReceived = [];
+      const serverClosed = new Promise((resolve) => {
+        wss.on("connection", (serverWs) => {
+          serverWs.on("message", (data) => {
+            messagesReceived.push(data.toString());
+          });
+          serverWs.on("close", resolve);
+        });
+      });
+      await writer.write(new TextEncoder().encode("buffered message"));
+      await writer.close();
+      await serverClosed;
+      (0, import_vitest.expect)(messagesReceived).toEqual(["buffered message"]);
+      wss.close();
+    });
+  });
+  (0, import_vitest.describe)("bidirectional communication", () => {
+    (0, import_vitest.it)("supports echo pattern with readable and writable", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.on("message", (data) => {
+          serverWs.send(data);
+        });
+      });
+      const { readable, writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      const reader = readable.getReader();
+      await writer.write(new TextEncoder().encode("ping1"));
+      await writer.write(new TextEncoder().encode("ping2"));
+      const { value: response1 } = await reader.read();
+      const { value: response2 } = await reader.read();
+      (0, import_vitest.expect)(Buffer.from(response1).toString()).toBe("ping1");
+      (0, import_vitest.expect)(Buffer.from(response2).toString()).toBe("ping2");
+      reader.releaseLock();
+      await writer.close();
+      wss.close();
+    });
+  });
+  (0, import_vitest.describe)("error handling", () => {
+    (0, import_vitest.it)("readable stream ends when WebSocket closes unexpectedly", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.send("before close");
+        serverWs.terminate();
+      });
+      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const chunks = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          chunks.push(Buffer.from(value).toString());
+        }
+      } catch (error) {
+        console.error(error);
+      } finally {
+        reader.releaseLock();
+      }
+      (0, import_vitest.expect)(chunks).toContain("before close");
+      wss.close();
+    });
+  });
+});
+//# sourceMappingURL=ws_transport.test.cjs.map

package/dist/utils/ws_transport.test.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/ws_transport.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { describe, expect, it } from 'vitest';\nimport { WebSocket, WebSocketServer } from 'ws';\nimport { webSocketStream } from './ws_transport.js';\n\ndescribe('webSocketStream', () => {\n  describe('readable stream', () => {\n    it('receives messages from the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('hello');\n        serverWs.send('world');\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const messages: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          messages.push(Buffer.from(value).toString());\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(messages).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('handles binary messages', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send(binaryData);\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(new Uint8Array(value));\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toHaveLength(1);\n      expect(Array.from(chunks[0]!)).toEqual([1, 2, 3, 4, 5]);\n\n      wss.close();\n    });\n\n    it('handles empty stream when connection closes immediately', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.close();\n      });\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(value);\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toEqual([]);\n\n      wss.close();\n    });\n  });\n\n  describe('writable stream', () => {\n    it('sends messages through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n      const ws = new WebSocket(`ws://localhost:${port}`);\n\n      const connected = new Promise<void>((resolve) => {\n        ws.on('open', resolve);\n      });\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      await connected;\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      await writer.write(new TextEncoder().encode('hello'));\n      await writer.write(new TextEncoder().encode('world'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('sends binary data through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const chunksReceived: Buffer[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            chunksReceived.push(Buffer.from(data as Buffer));\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);\n      await writer.write(binaryData);\n      await writer.close();\n\n      await serverClosed;\n\n      expect(chunksReceived).toHaveLength(1);\n      expect(Array.from(chunksReceived[0]!)).toEqual([10, 20, 30, 40, 50]);\n\n      wss.close();\n    });\n\n    it('buffers writes if readyState is CONNECTING', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      // These writes should be buffered\n      await writer.write(new TextEncoder().encode('buffered message'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['buffered message']);\n\n      wss.close();\n    });\n  });\n\n  describe('bidirectional communication', () => {\n    it('supports echo pattern with readable and writable', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      // Server echoes messages back\n      wss.on('connection', (serverWs) => {\n        serverWs.on('message', (data) => {\n          serverWs.send(data);\n        });\n      });\n\n      const { readable, writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n      const reader = readable.getReader();\n\n      // Send messages\n      await writer.write(new TextEncoder().encode('ping1'));\n      await writer.write(new TextEncoder().encode('ping2'));\n\n      // Read echoed responses\n      const { value: response1 } = await reader.read();\n      const { value: response2 } = await reader.read();\n\n      expect(Buffer.from(response1!).toString()).toBe('ping1');\n      expect(Buffer.from(response2!).toString()).toBe('ping2');\n\n      reader.releaseLock();\n      await writer.close();\n\n      wss.close();\n    });\n  });\n\n  describe('error handling', () => {\n    it('readable stream ends when WebSocket closes unexpectedly', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('before close');\n        // Terminate connection abruptly\n        serverWs.terminate();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(Buffer.from(value).toString());\n        }\n      } catch (error) {\n        console.error(error);\n        // Connection terminated, stream may error\n      } finally {\n        reader.releaseLock();\n      }\n\n      // Should have received the message sent before termination\n      expect(chunks).toContain('before close');\n\n      wss.close();\n    });\n  });\n});\n"],"mappings":";AAGA,oBAAqC;AACrC,gBAA2C;AAC3C,0BAAgC;AAAA,IAEhC,wBAAS,mBAAmB,MAAM;AAChC,8BAAS,mBAAmB,MAAM;AAChC,0BAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,OAAO;AACrB,iBAAS,KAAK,OAAO;AACrB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,WAAqB,CAAC;AAC5B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,mBAAS,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC7C;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,gCAAO,QAAQ,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAE3C,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,2BAA2B,YAAY;AACxC,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,aAAa,IAAI,WAAW,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,UAAU;AACxB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,IAAI,WAAW,KAAK,CAAC;AAAA,QACnC;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,gCAAO,MAAM,EAAE,aAAa,CAAC;AAC7B,gCAAO,MAAM,KAAK,OAAO,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEtD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,MAAM;AAAA,MACjB,CAAC;AACD,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,KAAK;AAAA,QACnB;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,gCAAO,MAAM,EAAE,QAAQ,CAAC,CAAC;AAEzB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,8BAAS,mBAAmB,MAAM;AAChC,0BAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AACjD,YAAM,KAAK,IAAI,oBAAU,kBAAkB,IAAI,EAAE;AAEjD,YAAM,YAAY,IAAI,QAAc,CAAC,YAAY;AAC/C,WAAG,GAAG,QAAQ,OAAO;AAAA,MACvB,CAAC;AAED,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM;AACN,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,gCAAO,gBAAgB,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAEnD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,2CAA2C,YAAY;AACxD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,iBAA2B,CAAC;AAClC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,2BAAe,KAAK,OAAO,KAAK,IAAc,CAAC;AAAA,UACjD,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,aAAa,IAAI,WAAW,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AACtD,YAAM,OAAO,MAAM,UAAU;AAC7B,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,gCAAO,cAAc,EAAE,aAAa,CAAC;AACrC,gCAAO,MAAM,KAAK,eAAe,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AAEnE,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,8CAA8C,YAAY;AAC3D,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAGD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,kBAAkB,CAAC;AAC/D,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,gCAAO,gBAAgB,EAAE,QAAQ,CAAC,kBAAkB,CAAC;AAErD,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,8BAAS,+BAA+B,MAAM;AAC5C,0BAAG,oDAAoD,YAAY;AACjE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAGjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,mBAAS,KAAK,IAAI;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,UAAU,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AACvE,YAAM,SAAS,SAAS,UAAU;AAClC,YAAM,SAAS,SAAS,UAAU;AAGlC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AAGpD,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAC/C,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAE/C,gCAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AACvD,gCAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AAEvD,aAAO,YAAY;AACnB,YAAM,OAAO,MAAM;AAEnB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,8BAAS,kBAAkB,MAAM;AAC/B,0BAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,cAAc;AAE5B,iBAAS,UAAU;AAAA,MACrB,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAmB,CAAC;AAC1B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC3C;AAAA,MACF,SAAS,OAAO;AACd,gBAAQ,MAAM,KAAK;AAAA,MAErB,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAGA,gCAAO,MAAM,EAAE,UAAU,cAAc;AAEvC,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AACH,CAAC;","names":[]}

package/dist/utils/ws_transport.test.js

@@ -0,0 +1,211 @@
+import { describe, expect, it } from "vitest";
+import { WebSocket, WebSocketServer } from "ws";
+import { webSocketStream } from "./ws_transport.js";
+describe("webSocketStream", () => {
+  describe("readable stream", () => {
+    it("receives messages from the WebSocket", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.send("hello");
+        serverWs.send("world");
+        serverWs.close();
+      });
+      const { readable } = webSocketStream(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const messages = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          messages.push(Buffer.from(value).toString());
+        }
+      } finally {
+        reader.releaseLock();
+      }
+      expect(messages).toEqual(["hello", "world"]);
+      wss.close();
+    });
+    it("handles binary messages", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);
+      wss.on("connection", (serverWs) => {
+        serverWs.send(binaryData);
+        serverWs.close();
+      });
+      const { readable } = webSocketStream(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const chunks = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          chunks.push(new Uint8Array(value));
+        }
+      } finally {
+        reader.releaseLock();
+      }
+      expect(chunks).toHaveLength(1);
+      expect(Array.from(chunks[0])).toEqual([1, 2, 3, 4, 5]);
+      wss.close();
+    });
+    it("handles empty stream when connection closes immediately", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.close();
+      });
+      const { readable } = webSocketStream(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const chunks = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          chunks.push(value);
+        }
+      } finally {
+        reader.releaseLock();
+      }
+      expect(chunks).toEqual([]);
+      wss.close();
+    });
+  });
+  describe("writable stream", () => {
+    it("sends messages through the WebSocket", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const ws = new WebSocket(`ws://localhost:${port}`);
+      const connected = new Promise((resolve) => {
+        ws.on("open", resolve);
+      });
+      const messagesReceived = [];
+      const serverClosed = new Promise((resolve) => {
+        wss.on("connection", (serverWs) => {
+          serverWs.on("message", (data) => {
+            messagesReceived.push(data.toString());
+          });
+          serverWs.on("close", resolve);
+        });
+      });
+      await connected;
+      const { writable } = webSocketStream(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      await writer.write(new TextEncoder().encode("hello"));
+      await writer.write(new TextEncoder().encode("world"));
+      await writer.close();
+      await serverClosed;
+      expect(messagesReceived).toEqual(["hello", "world"]);
+      wss.close();
+    });
+    it("sends binary data through the WebSocket", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const chunksReceived = [];
+      const serverClosed = new Promise((resolve) => {
+        wss.on("connection", (serverWs) => {
+          serverWs.on("message", (data) => {
+            chunksReceived.push(Buffer.from(data));
+          });
+          serverWs.on("close", resolve);
+        });
+      });
+      const { writable } = webSocketStream(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);
+      await writer.write(binaryData);
+      await writer.close();
+      await serverClosed;
+      expect(chunksReceived).toHaveLength(1);
+      expect(Array.from(chunksReceived[0])).toEqual([10, 20, 30, 40, 50]);
+      wss.close();
+    });
+    it("buffers writes if readyState is CONNECTING", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      const { writable } = webSocketStream(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      const messagesReceived = [];
+      const serverClosed = new Promise((resolve) => {
+        wss.on("connection", (serverWs) => {
+          serverWs.on("message", (data) => {
+            messagesReceived.push(data.toString());
+          });
+          serverWs.on("close", resolve);
+        });
+      });
+      await writer.write(new TextEncoder().encode("buffered message"));
+      await writer.close();
+      await serverClosed;
+      expect(messagesReceived).toEqual(["buffered message"]);
+      wss.close();
+    });
+  });
+  describe("bidirectional communication", () => {
+    it("supports echo pattern with readable and writable", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.on("message", (data) => {
+          serverWs.send(data);
+        });
+      });
+      const { readable, writable } = webSocketStream(`ws://localhost:${port}`);
+      const writer = writable.getWriter();
+      const reader = readable.getReader();
+      await writer.write(new TextEncoder().encode("ping1"));
+      await writer.write(new TextEncoder().encode("ping2"));
+      const { value: response1 } = await reader.read();
+      const { value: response2 } = await reader.read();
+      expect(Buffer.from(response1).toString()).toBe("ping1");
+      expect(Buffer.from(response2).toString()).toBe("ping2");
+      reader.releaseLock();
+      await writer.close();
+      wss.close();
+    });
+  });
+  describe("error handling", () => {
+    it("readable stream ends when WebSocket closes unexpectedly", async () => {
+      const wss = await new Promise((resolve) => {
+        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
+      });
+      const port = wss.address().port;
+      wss.on("connection", (serverWs) => {
+        serverWs.send("before close");
+        serverWs.terminate();
+      });
+      const { readable } = webSocketStream(`ws://localhost:${port}`);
+      const reader = readable.getReader();
+      const chunks = [];
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          chunks.push(Buffer.from(value).toString());
+        }
+      } catch (error) {
+        console.error(error);
+      } finally {
+        reader.releaseLock();
+      }
+      expect(chunks).toContain("before close");
+      wss.close();
+    });
+  });
+});
+//# sourceMappingURL=ws_transport.test.js.map

package/dist/utils/ws_transport.test.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/ws_transport.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { describe, expect, it } from 'vitest';\nimport { WebSocket, WebSocketServer } from 'ws';\nimport { webSocketStream } from './ws_transport.js';\n\ndescribe('webSocketStream', () => {\n  describe('readable stream', () => {\n    it('receives messages from the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('hello');\n        serverWs.send('world');\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const messages: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          messages.push(Buffer.from(value).toString());\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(messages).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('handles binary messages', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send(binaryData);\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(new Uint8Array(value));\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toHaveLength(1);\n      expect(Array.from(chunks[0]!)).toEqual([1, 2, 3, 4, 5]);\n\n      wss.close();\n    });\n\n    it('handles empty stream when connection closes immediately', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.close();\n      });\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(value);\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toEqual([]);\n\n      wss.close();\n    });\n  });\n\n  describe('writable stream', () => {\n    it('sends messages through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n      const ws = new WebSocket(`ws://localhost:${port}`);\n\n      const connected = new Promise<void>((resolve) => {\n        ws.on('open', resolve);\n      });\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      await connected;\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      await writer.write(new TextEncoder().encode('hello'));\n      await writer.write(new TextEncoder().encode('world'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('sends binary data through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const chunksReceived: Buffer[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            chunksReceived.push(Buffer.from(data as Buffer));\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);\n      await writer.write(binaryData);\n      await writer.close();\n\n      await serverClosed;\n\n      expect(chunksReceived).toHaveLength(1);\n      expect(Array.from(chunksReceived[0]!)).toEqual([10, 20, 30, 40, 50]);\n\n      wss.close();\n    });\n\n    it('buffers writes if readyState is CONNECTING', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      // These writes should be buffered\n      await writer.write(new TextEncoder().encode('buffered message'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['buffered message']);\n\n      wss.close();\n    });\n  });\n\n  describe('bidirectional communication', () => {\n    it('supports echo pattern with readable and writable', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      // Server echoes messages back\n      wss.on('connection', (serverWs) => {\n        serverWs.on('message', (data) => {\n          serverWs.send(data);\n        });\n      });\n\n      const { readable, writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n      const reader = readable.getReader();\n\n      // Send messages\n      await writer.write(new TextEncoder().encode('ping1'));\n      await writer.write(new TextEncoder().encode('ping2'));\n\n      // Read echoed responses\n      const { value: response1 } = await reader.read();\n      const { value: response2 } = await reader.read();\n\n      expect(Buffer.from(response1!).toString()).toBe('ping1');\n      expect(Buffer.from(response2!).toString()).toBe('ping2');\n\n      reader.releaseLock();\n      await writer.close();\n\n      wss.close();\n    });\n  });\n\n  describe('error handling', () => {\n    it('readable stream ends when WebSocket closes unexpectedly', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('before close');\n        // Terminate connection abruptly\n        serverWs.terminate();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(Buffer.from(value).toString());\n        }\n      } catch (error) {\n        console.error(error);\n        // Connection terminated, stream may error\n      } finally {\n        reader.releaseLock();\n      }\n\n      // Should have received the message sent before termination\n      expect(chunks).toContain('before close');\n\n      wss.close();\n    });\n  });\n});\n"],"mappings":"AAGA,SAAS,UAAU,QAAQ,UAAU;AACrC,SAAS,WAAW,uBAAuB;AAC3C,SAAS,uBAAuB;AAEhC,SAAS,mBAAmB,MAAM;AAChC,WAAS,mBAAmB,MAAM;AAChC,OAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,OAAO;AACrB,iBAAS,KAAK,OAAO;AACrB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,WAAqB,CAAC;AAC5B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,mBAAS,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC7C;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,aAAO,QAAQ,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAE3C,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,2BAA2B,YAAY;AACxC,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,aAAa,IAAI,WAAW,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,UAAU;AACxB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,IAAI,WAAW,KAAK,CAAC;AAAA,QACnC;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,aAAO,MAAM,EAAE,aAAa,CAAC;AAC7B,aAAO,MAAM,KAAK,OAAO,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEtD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,MAAM;AAAA,MACjB,CAAC;AACD,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,KAAK;AAAA,QACnB;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,aAAO,MAAM,EAAE,QAAQ,CAAC,CAAC;AAEzB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,WAAS,mBAAmB,MAAM;AAChC,OAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AACjD,YAAM,KAAK,IAAI,UAAU,kBAAkB,IAAI,EAAE;AAEjD,YAAM,YAAY,IAAI,QAAc,CAAC,YAAY;AAC/C,WAAG,GAAG,QAAQ,OAAO;AAAA,MACvB,CAAC;AAED,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM;AACN,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,aAAO,gBAAgB,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAEnD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,2CAA2C,YAAY;AACxD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,iBAA2B,CAAC;AAClC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,2BAAe,KAAK,OAAO,KAAK,IAAc,CAAC;AAAA,UACjD,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,aAAa,IAAI,WAAW,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AACtD,YAAM,OAAO,MAAM,UAAU;AAC7B,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,aAAO,cAAc,EAAE,aAAa,CAAC;AACrC,aAAO,MAAM,KAAK,eAAe,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AAEnE,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,8CAA8C,YAAY;AAC3D,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAGD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,kBAAkB,CAAC;AAC/D,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,aAAO,gBAAgB,EAAE,QAAQ,CAAC,kBAAkB,CAAC;AAErD,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,WAAS,+BAA+B,MAAM;AAC5C,OAAG,oDAAoD,YAAY;AACjE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAGjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,mBAAS,KAAK,IAAI;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,UAAU,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AACvE,YAAM,SAAS,SAAS,UAAU;AAClC,YAAM,SAAS,SAAS,UAAU;AAGlC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AAGpD,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAC/C,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAE/C,aAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AACvD,aAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AAEvD,aAAO,YAAY;AACnB,YAAM,OAAO,MAAM;AAEnB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,WAAS,kBAAkB,MAAM;AAC/B,OAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,cAAc;AAE5B,iBAAS,UAAU;AAAA,MACrB,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAmB,CAAC;AAC1B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC3C;AAAA,MACF,SAAS,OAAO;AACd,gBAAQ,MAAM,KAAK;AAAA,MAErB,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAGA,aAAO,MAAM,EAAE,UAAU,cAAc;AAEvC,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AACH,CAAC;","names":[]}