@replit/river 0.207.3 → 0.208.1

package/dist/testUtil/index.d.cts

@@ -1,4 +1,4 @@
-import { P as ProvidedClientTransportOptions, a as ProvidedServerTransportOptions, a4 as ClientHandshakeOptions, C as ClientTransport, b as Connection, S as ServerHandshakeOptions, M as BaseErrorSchemaType, B as Readable, a5 as ReadableIterator, a6 as ReadableResult, a7 as Transport, a8 as SessionOptions, a9 as ClientTransportOptions, aa as SessionNoConnection, ab as SessionBoundSendFn } from '../services-87887bc5.js';
+import { P as ProvidedClientTransportOptions, a as ProvidedServerTransportOptions, a4 as ClientHandshakeOptions, C as ClientTransport, b as Connection, S as ServerHandshakeOptions, N as BaseErrorSchemaType, D as Readable, a5 as ReadableIterator, a6 as ReadableResult, a7 as Transport, a8 as SessionOptions, a9 as ClientTransportOptions, aa as SessionNoConnection, ab as SessionBoundSendFn } from '../services-44be1b6b.js';
 import { ServerTransport } from '../transport/index.cjs';
 import { T as TransportClientId, a as PartialTransportMessage, O as OpaqueTransportMessage } from '../message-01c3e85a.js';
 import { Static } from '@sinclair/typebox';

package/dist/testUtil/index.d.ts

@@ -1,4 +1,4 @@
-import { P as ProvidedClientTransportOptions, a as ProvidedServerTransportOptions, a4 as ClientHandshakeOptions, C as ClientTransport, b as Connection, S as ServerHandshakeOptions, M as BaseErrorSchemaType, B as Readable, a5 as ReadableIterator, a6 as ReadableResult, a7 as Transport, a8 as SessionOptions, a9 as ClientTransportOptions, aa as SessionNoConnection, ab as SessionBoundSendFn } from '../services-87887bc5.js';
+import { P as ProvidedClientTransportOptions, a as ProvidedServerTransportOptions, a4 as ClientHandshakeOptions, C as ClientTransport, b as Connection, S as ServerHandshakeOptions, N as BaseErrorSchemaType, D as Readable, a5 as ReadableIterator, a6 as ReadableResult, a7 as Transport, a8 as SessionOptions, a9 as ClientTransportOptions, aa as SessionNoConnection, ab as SessionBoundSendFn } from '../services-44be1b6b.js';
 import { ServerTransport } from '../transport/index.js';
 import { T as TransportClientId, a as PartialTransportMessage, O as OpaqueTransportMessage } from '../message-01c3e85a.js';
 import { Static } from '@sinclair/typebox';

package/dist/testUtil/index.js

@@ -5,11 +5,11 @@ import {
   SessionStateGraph,
   defaultClientTransportOptions,
   defaultTransportOptions
-} from "../chunk-BO7MFCO6.js";
+} from "../chunk-2WFRHXDZ.js";
 import {
   currentProtocolVersion,
   getTracer
-} from "../chunk-QGPYCXV4.js";
+} from "../chunk-EVAQ2QMB.js";
 import "../chunk-CC7RN7GI.js";
 
 // testUtil/index.ts
@@ -123,19 +123,13 @@ var InMemoryConnection = class extends Connection {
     this.conn = pipe;
     this.conn.allowHalfOpen = false;
     this.conn.on("data", (data) => {
-      for (const cb of this.dataListeners) {
-        cb(data);
-      }
+      this.dataListener?.(data);
     });
     this.conn.on("close", () => {
-      for (const cb of this.closeListeners) {
-        cb();
-      }
+      this.closeListener?.();
     });
     this.conn.on("error", (err) => {
-      for (const cb of this.errorListeners) {
-        cb(err);
-      }
+      this.errorListener?.(err);
     });
   }
   send(payload) {

package/dist/testUtil/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../testUtil/index.ts","../../testUtil/observable/observable.ts","../../testUtil/duplex/duplexPair.ts","../../testUtil/fixtures/mockTransport.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { Static } from '@sinclair/typebox';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  currentProtocolVersion,\n} from '../transport/message';\nimport { Transport } from '../transport/transport';\nimport { Readable, ReadableResult, ReadableIterator } from '../router/streams';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport {\n  defaultClientTransportOptions,\n  defaultTransportOptions,\n} from '../transport/options';\nimport { Connection } from '../transport/connection';\nimport { SessionState } from '../transport/sessionStateMachine/common';\nimport { SessionStateGraph } from '../transport/sessionStateMachine/transitions';\nimport { BaseErrorSchemaType } from '../router/errors';\nimport { ClientTransport } from '../transport/client';\nimport { ServerTransport } from '../transport/server';\nimport { getTracer } from '../tracing';\n\nexport {\n  createMockTransportNetwork,\n  InMemoryConnection,\n} from './fixtures/mockTransport';\n\n/**\n * Creates a WebSocket client that connects to a local server at the specified port.\n * This should only be used for testing.\n * @param port - The port number to connect to.\n * @returns A Promise that resolves to a WebSocket instance.\n */\nexport function createLocalWebSocketClient(port: number): WsLike {\n  const sock = new NodeWs(`ws://localhost:${port}`);\n  sock.binaryType = 'arraybuffer';\n\n  return sock;\n}\n\n/**\n * Creates a WebSocket server instance using the provided HTTP server.\n * Only used as helper for testing.\n * @param server - The HTTP server instance to use for the WebSocket server.\n * @returns A Promise that resolves to the created WebSocket server instance.\n */\nexport function createWebSocketServer(server: http.Server) {\n  return new WebSocketServer({ server });\n}\n\n/**\n * Starts listening on the given server and returns the automatically allocated port number.\n * This should only be used for testing.\n * @param server - The http server to listen on.\n * @returns A promise that resolves with the allocated port number.\n * @throws An error if a port cannot be allocated.\n */\nexport function onWsServerReady(server: http.Server): Promise<number> {\n  return new Promise((resolve, reject) => {\n    server.listen(() => {\n      const addr = server.address();\n      if (typeof addr === 'object' && addr) {\n        resolve(addr.port);\n      } else {\n        reject(new Error(\"couldn't find a port to allocate\"));\n      }\n    });\n  });\n}\n\nconst readableIterators = new WeakMap<\n  Readable<unknown, Static<BaseErrorSchemaType>>,\n  ReadableIterator<unknown, Static<BaseErrorSchemaType>>\n>();\n\n/**\n * A safe way to access {@link Readble}'s iterator multiple times in test helpers.\n *\n * If there are other iteration attempts outside of the test helpers\n * (this function, {@link readNextResult}, and {@link isReadableDone})\n * it will throw an error.\n */\nexport function getReadableIterator<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): ReadableIterator<T, E> {\n  let iter = readableIterators.get(readable) as\n    | ReadableIterator<T, E>\n    | undefined;\n\n  if (!iter) {\n    iter = readable[Symbol.asyncIterator]();\n    readableIterators.set(readable, iter);\n  }\n\n  return iter;\n}\n\n/**\n * Retrieves the next value from {@link Readable}, or throws an error if the Readable is done.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function readNextResult<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): Promise<ReadableResult<T, E>> {\n  const res = await getReadableIterator(readable).next();\n\n  if (res.done) {\n    throw new Error('readNext from a done Readable');\n  }\n\n  return res.value;\n}\n\n/**\n * Checks if the readable is done iterating, it consumes an iteration in the process.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function isReadableDone<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n) {\n  const res = await getReadableIterator(readable).next();\n\n  return res.done;\n}\n\nexport function payloadToTransportMessage<Payload>(\n  payload: Payload,\n): PartialTransportMessage<Payload> {\n  return {\n    streamId: 'stream',\n    controlFlags: 0,\n    payload,\n  };\n}\n\nexport function createDummyTransportMessage() {\n  return payloadToTransportMessage({\n    msg: 'cool',\n    test: Math.random(),\n  });\n}\n\n/**\n * Waits for a message on the transport.\n * @param {Transport} t - The transport to listen to.\n * @param filter - An optional filter function to apply to the received messages.\n * @returns A promise that resolves with the payload of the first message that passes the filter.\n */\nexport async function waitForMessage(\n  t: Transport<Connection>,\n  filter?: (msg: OpaqueTransportMessage) => boolean,\n  rejectMismatch?: boolean,\n) {\n  return new Promise((resolve, reject) => {\n    function cleanup() {\n      t.removeEventListener('message', onMessage);\n    }\n\n    function onMessage(msg: OpaqueTransportMessage) {\n      if (!filter || filter(msg)) {\n        cleanup();\n        resolve(msg.payload);\n      } else if (rejectMismatch) {\n        cleanup();\n        reject(new Error('message didnt match the filter'));\n      }\n    }\n\n    t.addEventListener('message', onMessage);\n  });\n}\n\nexport const testingSessionOptions = defaultTransportOptions;\nexport const testingClientSessionOptions = defaultClientTransportOptions;\n\nexport function dummySession() {\n  return SessionStateGraph.entrypoints.NoConnection(\n    'client',\n    'server',\n    {\n      onSessionGracePeriodElapsed: () => {\n        /* noop */\n      },\n    },\n    testingSessionOptions,\n    currentProtocolVersion,\n    getTracer(),\n  );\n}\n\nexport function getClientSendFn(\n  clientTransport: ClientTransport<Connection>,\n  serverTransport: ServerTransport<Connection>,\n) {\n  const session =\n    clientTransport.sessions.get(serverTransport.clientId) ??\n    clientTransport.createUnconnectedSession(serverTransport.clientId);\n\n  return clientTransport.getSessionBoundSendFn(\n    serverTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getServerSendFn(\n  serverTransport: ServerTransport<Connection>,\n  clientTransport: ClientTransport<Connection>,\n) {\n  const session = serverTransport.sessions.get(clientTransport.clientId);\n  if (!session) {\n    throw new Error('session not found');\n  }\n\n  return serverTransport.getSessionBoundSendFn(\n    clientTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getTransportConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): Array<ConnType> {\n  const connections = [];\n  for (const session of transport.sessions.values()) {\n    if (session.state === SessionState.Connected) {\n      connections.push(session.conn);\n    }\n  }\n\n  return connections;\n}\n\nexport function numberOfConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): number {\n  return getTransportConnections(transport).length;\n}\n\nexport function closeAllConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n) {\n  for (const conn of getTransportConnections(transport)) {\n    conn.close();\n  }\n}\n","/**\n * Represents an observable value that can be subscribed to for changes.\n * This should only be used in tests\n * @template T - The type of the value being observed.\n */\nexport class Observable<T> {\n  value: T;\n  private listeners: Set<(val: T) => void>;\n\n  constructor(initialValue: T) {\n    this.value = initialValue;\n    this.listeners = new Set();\n  }\n\n  /**\n   * Gets the current value of the observable.\n   */\n  get() {\n    return this.value;\n  }\n\n  /**\n   * Sets the current value of the observable. All listeners will get an update with this value.\n   * @param newValue - The new value to set.\n   */\n  set(tx: (preValue: T) => T) {\n    const newValue = tx(this.value);\n    this.value = newValue;\n    this.listeners.forEach((listener) => listener(newValue));\n  }\n\n  /**\n   * Subscribes to changes in the observable value.\n   * @param listener - A callback function that will be called when the value changes.\n   * @returns A function that can be called to unsubscribe from further notifications.\n   */\n  observe(listener: (val: T) => void) {\n    this.listeners.add(listener);\n    listener(this.get());\n\n    return () => this.listeners.delete(listener);\n  }\n\n  /**\n   * Returns the number of listeners currently observing the observable\n   */\n  get listenerCount(): number {\n    return this.listeners.size;\n  }\n}\n","import { Duplex } from 'node:stream';\nimport assert from 'assert';\n\nconst kCallback = Symbol('Callback');\nconst kInitOtherSide = Symbol('InitOtherSide');\n\n// yoinked from https://github.com/nodejs/node/blob/c3a7b29e56a5ada6327ebb622ba746d022685742/lib/internal/streams/duplexpair.js#L55\n// but with types\nclass DuplexSide extends Duplex {\n  private otherSide: DuplexSide | null;\n  private [kCallback]: (() => void) | null;\n\n  constructor() {\n    super();\n    this[kCallback] = null;\n    this.otherSide = null;\n  }\n\n  [kInitOtherSide](otherSide: DuplexSide) {\n    if (this.otherSide === null) {\n      this.otherSide = otherSide;\n    }\n  }\n\n  _read() {\n    const callback = this[kCallback];\n    if (callback) {\n      this[kCallback] = null;\n      callback();\n    }\n  }\n\n  _write(\n    chunk: Uint8Array,\n    _encoding: BufferEncoding,\n    callback: (error?: Error | null) => void,\n  ) {\n    assert(this.otherSide !== null);\n    assert(this.otherSide[kCallback] === null);\n    if (chunk.length === 0) {\n      process.nextTick(callback);\n    } else {\n      this.otherSide.push(chunk);\n      this.otherSide[kCallback] = callback;\n    }\n  }\n\n  _final(callback: (error?: Error | null) => void) {\n    this.otherSide?.on('end', callback);\n    this.otherSide?.push(null);\n  }\n}\n\nexport function duplexPair(): [DuplexSide, DuplexSide] {\n  const side0 = new DuplexSide();\n  const side1 = new DuplexSide();\n  side0[kInitOtherSide](side1);\n  side1[kInitOtherSide](side0);\n  side0.on('close', () => {\n    setImmediate(() => {\n      side1.destroy();\n    });\n  });\n\n  side1.on('close', () => {\n    setImmediate(() => {\n      side0.destroy();\n    });\n  });\n\n  return [side0, side1];\n}\n","import { TransportClientId } from '../../transport';\nimport { ClientTransport } from '../../transport/client';\nimport { Connection } from '../../transport/connection';\nimport { ServerTransport } from '../../transport/server';\nimport { Observable } from '../observable/observable';\nimport { ProvidedServerTransportOptions } from '../../transport/options';\nimport { TestSetupHelpers, TestTransportOptions } from './transports';\nimport { Duplex } from 'node:stream';\nimport { duplexPair } from '../duplex/duplexPair';\nimport { nanoid } from 'nanoid';\n\nexport class InMemoryConnection extends Connection {\n  conn: Duplex;\n\n  constructor(pipe: Duplex) {\n    super();\n    this.conn = pipe;\n    this.conn.allowHalfOpen = false;\n\n    this.conn.on('data', (data: Uint8Array) => {\n      for (const cb of this.dataListeners) {\n        cb(data);\n      }\n    });\n\n    this.conn.on('close', () => {\n      for (const cb of this.closeListeners) {\n        cb();\n      }\n    });\n\n    this.conn.on('error', (err) => {\n      for (const cb of this.errorListeners) {\n        cb(err);\n      }\n    });\n  }\n\n  send(payload: Uint8Array): boolean {\n    setImmediate(() => {\n      this.conn.write(payload);\n    });\n\n    return true;\n  }\n\n  close(): void {\n    setImmediate(() => {\n      this.conn.end();\n      this.conn.emit('close');\n    });\n  }\n}\n\ninterface BidiConnection {\n  id: string;\n  clientToServer: Duplex;\n  serverToClient: Duplex;\n  clientId: TransportClientId;\n  serverId: TransportClientId;\n  handled: boolean;\n}\n\n// we construct a network of transports connected by node streams here\n// so that we can test the transport layer without needing to actually\n// use real network/websocket connections\n// this is useful for testing the transport layer in isolation\n// and allows us to control network conditions in a way that would be\n// difficult with real network connections (e.g. simulating a phantom\n// disconnect, .pause() vs .removeAllListeners('data'), congestion,\n// latency, differences in ws implementations between node and browsers, etc.)\nexport function createMockTransportNetwork(\n  opts?: TestTransportOptions,\n): TestSetupHelpers {\n  // conn id -> [client->server, server->client]\n  const connections = new Observable<Record<string, BidiConnection>>({});\n\n  const transports: Array<MockClientTransport | MockServerTransport> = [];\n  class MockClientTransport extends ClientTransport<InMemoryConnection> {\n    async createNewOutgoingConnection(\n      to: TransportClientId,\n    ): Promise<InMemoryConnection> {\n      const [clientToServer, serverToClient] = duplexPair();\n      await new Promise((resolve) => setImmediate(resolve));\n\n      const connId = nanoid();\n      connections.set((prev) => ({\n        ...prev,\n        [connId]: {\n          id: connId,\n          clientToServer,\n          serverToClient,\n          clientId: this.clientId,\n          serverId: to,\n          handled: false,\n        },\n      }));\n\n      return new InMemoryConnection(clientToServer);\n    }\n  }\n\n  class MockServerTransport extends ServerTransport<InMemoryConnection> {\n    subscribeCleanup: () => void;\n\n    constructor(\n      clientId: TransportClientId,\n      options?: ProvidedServerTransportOptions,\n    ) {\n      super(clientId, options);\n\n      this.subscribeCleanup = connections.observe((conns) => {\n        // look for any unhandled connections\n        for (const conn of Object.values(conns)) {\n          // if we've already handled this connection, skip it\n          // or if it's not for us, skip it\n          if (conn.handled || conn.serverId !== this.clientId) {\n            continue;\n          }\n\n          conn.handled = true;\n          const connection = new InMemoryConnection(conn.serverToClient);\n          this.handleConnection(connection);\n        }\n      });\n    }\n\n    close() {\n      this.subscribeCleanup();\n      super.close();\n    }\n  }\n\n  return {\n    getClientTransport: (id, handshakeOptions) => {\n      const clientTransport = new MockClientTransport(id, opts?.client);\n      if (handshakeOptions) {\n        clientTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(clientTransport);\n\n      return clientTransport;\n    },\n    getServerTransport: (id = 'SERVER', handshakeOptions) => {\n      const serverTransport = new MockServerTransport(id, opts?.server);\n      if (handshakeOptions) {\n        serverTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(serverTransport);\n\n      return serverTransport;\n    },\n    simulatePhantomDisconnect() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.pause();\n        conn.clientToServer.pause();\n      }\n    },\n    async restartServer() {\n      for (const transport of transports) {\n        if (transport.clientId !== 'SERVER') continue;\n        transport.close();\n      }\n\n      // kill all connections while we're at it\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.destroy();\n        conn.clientToServer.destroy();\n      }\n    },\n    cleanup() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.destroy();\n        conn.clientToServer.destroy();\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;;;ACKjC,IAAM,aAAN,MAAoB;AAAA,EACzB;AAAA,EACQ;AAAA,EAER,YAAY,cAAiB;AAC3B,SAAK,QAAQ;AACb,SAAK,YAAY,oBAAI,IAAI;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM;AACJ,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,IAAwB;AAC1B,UAAM,WAAW,GAAG,KAAK,KAAK;AAC9B,SAAK,QAAQ;AACb,SAAK,UAAU,QAAQ,CAAC,aAAa,SAAS,QAAQ,CAAC;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ,UAA4B;AAClC,SAAK,UAAU,IAAI,QAAQ;AAC3B,aAAS,KAAK,IAAI,CAAC;AAEnB,WAAO,MAAM,KAAK,UAAU,OAAO,QAAQ;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,gBAAwB;AAC1B,WAAO,KAAK,UAAU;AAAA,EACxB;AACF;;;ACjDA,SAAS,cAAc;AACvB,OAAO,YAAY;AAEnB,IAAM,YAAY,OAAO,UAAU;AACnC,IAAM,iBAAiB,OAAO,eAAe;AAI7C,IAAM,aAAN,cAAyB,OAAO;AAAA,EACtB;AAAA,EACR,CAAS,SAAS;AAAA,EAElB,cAAc;AACZ,UAAM;AACN,SAAK,SAAS,IAAI;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,CAAC,cAAc,EAAE,WAAuB;AACtC,QAAI,KAAK,cAAc,MAAM;AAC3B,WAAK,YAAY;AAAA,IACnB;AAAA,EACF;AAAA,EAEA,QAAQ;AACN,UAAM,WAAW,KAAK,SAAS;AAC/B,QAAI,UAAU;AACZ,WAAK,SAAS,IAAI;AAClB,eAAS;AAAA,IACX;AAAA,EACF;AAAA,EAEA,OACE,OACA,WACA,UACA;AACA,WAAO,KAAK,cAAc,IAAI;AAC9B,WAAO,KAAK,UAAU,SAAS,MAAM,IAAI;AACzC,QAAI,MAAM,WAAW,GAAG;AACtB,cAAQ,SAAS,QAAQ;AAAA,IAC3B,OAAO;AACL,WAAK,UAAU,KAAK,KAAK;AACzB,WAAK,UAAU,SAAS,IAAI;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,OAAO,UAA0C;AAC/C,SAAK,WAAW,GAAG,OAAO,QAAQ;AAClC,SAAK,WAAW,KAAK,IAAI;AAAA,EAC3B;AACF;AAEO,SAAS,aAAuC;AACrD,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,cAAc,EAAE,KAAK;AAC3B,QAAM,cAAc,EAAE,KAAK;AAC3B,QAAM,GAAG,SAAS,MAAM;AACtB,iBAAa,MAAM;AACjB,YAAM,QAAQ;AAAA,IAChB,CAAC;AAAA,EACH,CAAC;AAED,QAAM,GAAG,SAAS,MAAM;AACtB,iBAAa,MAAM;AACjB,YAAM,QAAQ;AAAA,IAChB,CAAC;AAAA,EACH,CAAC;AAED,SAAO,CAAC,OAAO,KAAK;AACtB;;;AC9DA,SAAS,cAAc;AAEhB,IAAM,qBAAN,cAAiC,WAAW;AAAA,EACjD;AAAA,EAEA,YAAY,MAAc;AACxB,UAAM;AACN,SAAK,OAAO;AACZ,SAAK,KAAK,gBAAgB;AAE1B,SAAK,KAAK,GAAG,QAAQ,CAAC,SAAqB;AACzC,iBAAW,MAAM,KAAK,eAAe;AACnC,WAAG,IAAI;AAAA,MACT;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,MAAM;AAC1B,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG;AAAA,MACL;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,CAAC,QAAQ;AAC7B,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,KAAK,SAA8B;AACjC,iBAAa,MAAM;AACjB,WAAK,KAAK,MAAM,OAAO;AAAA,IACzB,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEA,QAAc;AACZ,iBAAa,MAAM;AACjB,WAAK,KAAK,IAAI;AACd,WAAK,KAAK,KAAK,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAmBO,SAAS,2BACd,MACkB;AAElB,QAAM,cAAc,IAAI,WAA2C,CAAC,CAAC;AAErE,QAAM,aAA+D,CAAC;AAAA,EACtE,MAAM,4BAA4B,gBAAoC;AAAA,IACpE,MAAM,4BACJ,IAC6B;AAC7B,YAAM,CAAC,gBAAgB,cAAc,IAAI,WAAW;AACpD,YAAM,IAAI,QAAQ,CAAC,YAAY,aAAa,OAAO,CAAC;AAEpD,YAAM,SAAS,OAAO;AACtB,kBAAY,IAAI,CAAC,UAAU;AAAA,QACzB,GAAG;AAAA,QACH,CAAC,MAAM,GAAG;AAAA,UACR,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,UACA,UAAU,KAAK;AAAA,UACf,UAAU;AAAA,UACV,SAAS;AAAA,QACX;AAAA,MACF,EAAE;AAEF,aAAO,IAAI,mBAAmB,cAAc;AAAA,IAC9C;AAAA,EACF;AAAA,EAEA,MAAM,4BAA4B,gBAAoC;AAAA,IACpE;AAAA,IAEA,YACE,UACA,SACA;AACA,YAAM,UAAU,OAAO;AAEvB,WAAK,mBAAmB,YAAY,QAAQ,CAAC,UAAU;AAErD,mBAAW,QAAQ,OAAO,OAAO,KAAK,GAAG;AAGvC,cAAI,KAAK,WAAW,KAAK,aAAa,KAAK,UAAU;AACnD;AAAA,UACF;AAEA,eAAK,UAAU;AACf,gBAAM,aAAa,IAAI,mBAAmB,KAAK,cAAc;AAC7D,eAAK,iBAAiB,UAAU;AAAA,QAClC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IAEA,QAAQ;AACN,WAAK,iBAAiB;AACtB,YAAM,MAAM;AAAA,IACd;AAAA,EACF;AAEA,SAAO;AAAA,IACL,oBAAoB,CAAC,IAAI,qBAAqB;AAC5C,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,oBAAoB,CAAC,KAAK,UAAU,qBAAqB;AACvD,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,4BAA4B;AAC1B,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,MAAM;AAC1B,aAAK,eAAe,MAAM;AAAA,MAC5B;AAAA,IACF;AAAA,IACA,MAAM,gBAAgB;AACpB,iBAAW,aAAa,YAAY;AAClC,YAAI,UAAU,aAAa;AAAU;AACrC,kBAAU,MAAM;AAAA,MAClB;AAGA,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,QAAQ;AAC5B,aAAK,eAAe,QAAQ;AAAA,MAC9B;AAAA,IACF;AAAA,IACA,UAAU;AACR,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,QAAQ;AAC5B,aAAK,eAAe,QAAQ;AAAA,MAC9B;AAAA,IACF;AAAA,EACF;AACF;;;AHjJO,SAAS,2BAA2B,MAAsB;AAC/D,QAAM,OAAO,IAAI,OAAO,kBAAkB,IAAI,EAAE;AAChD,OAAK,aAAa;AAElB,SAAO;AACT;AAQO,SAAS,sBAAsB,QAAqB;AACzD,SAAO,IAAI,gBAAgB,EAAE,OAAO,CAAC;AACvC;AASO,SAAS,gBAAgB,QAAsC;AACpE,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAO,OAAO,MAAM;AAClB,YAAM,OAAO,OAAO,QAAQ;AAC5B,UAAI,OAAO,SAAS,YAAY,MAAM;AACpC,gBAAQ,KAAK,IAAI;AAAA,MACnB,OAAO;AACL,eAAO,IAAI,MAAM,kCAAkC,CAAC;AAAA,MACtD;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AACH;AAEA,IAAM,oBAAoB,oBAAI,QAG5B;AASK,SAAS,oBACd,UACwB;AACxB,MAAI,OAAO,kBAAkB,IAAI,QAAQ;AAIzC,MAAI,CAAC,MAAM;AACT,WAAO,SAAS,OAAO,aAAa,EAAE;AACtC,sBAAkB,IAAI,UAAU,IAAI;AAAA,EACtC;AAEA,SAAO;AACT;AAOA,eAAsB,eACpB,UAC+B;AAC/B,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,MAAI,IAAI,MAAM;AACZ,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO,IAAI;AACb;AAOA,eAAsB,eACpB,UACA;AACA,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,SAAO,IAAI;AACb;AAEO,SAAS,0BACd,SACkC;AAClC,SAAO;AAAA,IACL,UAAU;AAAA,IACV,cAAc;AAAA,IACd;AAAA,EACF;AACF;AAEO,SAAS,8BAA8B;AAC5C,SAAO,0BAA0B;AAAA,IAC/B,KAAK;AAAA,IACL,MAAM,KAAK,OAAO;AAAA,EACpB,CAAC;AACH;AAQA,eAAsB,eACpB,GACA,QACA,gBACA;AACA,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,aAAS,UAAU;AACjB,QAAE,oBAAoB,WAAW,SAAS;AAAA,IAC5C;AAEA,aAAS,UAAU,KAA6B;AAC9C,UAAI,CAAC,UAAU,OAAO,GAAG,GAAG;AAC1B,gBAAQ;AACR,gBAAQ,IAAI,OAAO;AAAA,MACrB,WAAW,gBAAgB;AACzB,gBAAQ;AACR,eAAO,IAAI,MAAM,gCAAgC,CAAC;AAAA,MACpD;AAAA,IACF;AAEA,MAAE,iBAAiB,WAAW,SAAS;AAAA,EACzC,CAAC;AACH;AAEO,IAAM,wBAAwB;AAC9B,IAAM,8BAA8B;AAEpC,SAAS,eAAe;AAC7B,SAAO,kBAAkB,YAAY;AAAA,IACnC;AAAA,IACA;AAAA,IACA;AAAA,MACE,6BAA6B,MAAM;AAAA,MAEnC;AAAA,IACF;AAAA,IACA;AAAA,IACA;AAAA,IACA,UAAU;AAAA,EACZ;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UACJ,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ,KACrD,gBAAgB,yBAAyB,gBAAgB,QAAQ;AAEnE,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UAAU,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ;AACrE,MAAI,CAAC,SAAS;AACZ,UAAM,IAAI,MAAM,mBAAmB;AAAA,EACrC;AAEA,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,wBACd,WACiB;AACjB,QAAM,cAAc,CAAC;AACrB,aAAW,WAAW,UAAU,SAAS,OAAO,GAAG;AACjD,QAAI,QAAQ,uCAAkC;AAC5C,kBAAY,KAAK,QAAQ,IAAI;AAAA,IAC/B;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,oBACd,WACQ;AACR,SAAO,wBAAwB,SAAS,EAAE;AAC5C;AAEO,SAAS,oBACd,WACA;AACA,aAAW,QAAQ,wBAAwB,SAAS,GAAG;AACrD,SAAK,MAAM;AAAA,EACb;AACF;","names":[]}
+{"version":3,"sources":["../../testUtil/index.ts","../../testUtil/observable/observable.ts","../../testUtil/duplex/duplexPair.ts","../../testUtil/fixtures/mockTransport.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { Static } from '@sinclair/typebox';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  currentProtocolVersion,\n} from '../transport/message';\nimport { Transport } from '../transport/transport';\nimport { Readable, ReadableResult, ReadableIterator } from '../router/streams';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport {\n  defaultClientTransportOptions,\n  defaultTransportOptions,\n} from '../transport/options';\nimport { Connection } from '../transport/connection';\nimport { SessionState } from '../transport/sessionStateMachine/common';\nimport { SessionStateGraph } from '../transport/sessionStateMachine/transitions';\nimport { BaseErrorSchemaType } from '../router/errors';\nimport { ClientTransport } from '../transport/client';\nimport { ServerTransport } from '../transport/server';\nimport { getTracer } from '../tracing';\n\nexport {\n  createMockTransportNetwork,\n  InMemoryConnection,\n} from './fixtures/mockTransport';\n\n/**\n * Creates a WebSocket client that connects to a local server at the specified port.\n * This should only be used for testing.\n * @param port - The port number to connect to.\n * @returns A Promise that resolves to a WebSocket instance.\n */\nexport function createLocalWebSocketClient(port: number): WsLike {\n  const sock = new NodeWs(`ws://localhost:${port}`);\n  sock.binaryType = 'arraybuffer';\n\n  return sock;\n}\n\n/**\n * Creates a WebSocket server instance using the provided HTTP server.\n * Only used as helper for testing.\n * @param server - The HTTP server instance to use for the WebSocket server.\n * @returns A Promise that resolves to the created WebSocket server instance.\n */\nexport function createWebSocketServer(server: http.Server) {\n  return new WebSocketServer({ server });\n}\n\n/**\n * Starts listening on the given server and returns the automatically allocated port number.\n * This should only be used for testing.\n * @param server - The http server to listen on.\n * @returns A promise that resolves with the allocated port number.\n * @throws An error if a port cannot be allocated.\n */\nexport function onWsServerReady(server: http.Server): Promise<number> {\n  return new Promise((resolve, reject) => {\n    server.listen(() => {\n      const addr = server.address();\n      if (typeof addr === 'object' && addr) {\n        resolve(addr.port);\n      } else {\n        reject(new Error(\"couldn't find a port to allocate\"));\n      }\n    });\n  });\n}\n\nconst readableIterators = new WeakMap<\n  Readable<unknown, Static<BaseErrorSchemaType>>,\n  ReadableIterator<unknown, Static<BaseErrorSchemaType>>\n>();\n\n/**\n * A safe way to access {@link Readble}'s iterator multiple times in test helpers.\n *\n * If there are other iteration attempts outside of the test helpers\n * (this function, {@link readNextResult}, and {@link isReadableDone})\n * it will throw an error.\n */\nexport function getReadableIterator<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): ReadableIterator<T, E> {\n  let iter = readableIterators.get(readable) as\n    | ReadableIterator<T, E>\n    | undefined;\n\n  if (!iter) {\n    iter = readable[Symbol.asyncIterator]();\n    readableIterators.set(readable, iter);\n  }\n\n  return iter;\n}\n\n/**\n * Retrieves the next value from {@link Readable}, or throws an error if the Readable is done.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function readNextResult<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): Promise<ReadableResult<T, E>> {\n  const res = await getReadableIterator(readable).next();\n\n  if (res.done) {\n    throw new Error('readNext from a done Readable');\n  }\n\n  return res.value;\n}\n\n/**\n * Checks if the readable is done iterating, it consumes an iteration in the process.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function isReadableDone<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n) {\n  const res = await getReadableIterator(readable).next();\n\n  return res.done;\n}\n\nexport function payloadToTransportMessage<Payload>(\n  payload: Payload,\n): PartialTransportMessage<Payload> {\n  return {\n    streamId: 'stream',\n    controlFlags: 0,\n    payload,\n  };\n}\n\nexport function createDummyTransportMessage() {\n  return payloadToTransportMessage({\n    msg: 'cool',\n    test: Math.random(),\n  });\n}\n\n/**\n * Waits for a message on the transport.\n * @param {Transport} t - The transport to listen to.\n * @param filter - An optional filter function to apply to the received messages.\n * @returns A promise that resolves with the payload of the first message that passes the filter.\n */\nexport async function waitForMessage(\n  t: Transport<Connection>,\n  filter?: (msg: OpaqueTransportMessage) => boolean,\n  rejectMismatch?: boolean,\n) {\n  return new Promise((resolve, reject) => {\n    function cleanup() {\n      t.removeEventListener('message', onMessage);\n    }\n\n    function onMessage(msg: OpaqueTransportMessage) {\n      if (!filter || filter(msg)) {\n        cleanup();\n        resolve(msg.payload);\n      } else if (rejectMismatch) {\n        cleanup();\n        reject(new Error('message didnt match the filter'));\n      }\n    }\n\n    t.addEventListener('message', onMessage);\n  });\n}\n\nexport const testingSessionOptions = defaultTransportOptions;\nexport const testingClientSessionOptions = defaultClientTransportOptions;\n\nexport function dummySession() {\n  return SessionStateGraph.entrypoints.NoConnection(\n    'client',\n    'server',\n    {\n      onSessionGracePeriodElapsed: () => {\n        /* noop */\n      },\n    },\n    testingSessionOptions,\n    currentProtocolVersion,\n    getTracer(),\n  );\n}\n\nexport function getClientSendFn(\n  clientTransport: ClientTransport<Connection>,\n  serverTransport: ServerTransport<Connection>,\n) {\n  const session =\n    clientTransport.sessions.get(serverTransport.clientId) ??\n    clientTransport.createUnconnectedSession(serverTransport.clientId);\n\n  return clientTransport.getSessionBoundSendFn(\n    serverTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getServerSendFn(\n  serverTransport: ServerTransport<Connection>,\n  clientTransport: ClientTransport<Connection>,\n) {\n  const session = serverTransport.sessions.get(clientTransport.clientId);\n  if (!session) {\n    throw new Error('session not found');\n  }\n\n  return serverTransport.getSessionBoundSendFn(\n    clientTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getTransportConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): Array<ConnType> {\n  const connections = [];\n  for (const session of transport.sessions.values()) {\n    if (session.state === SessionState.Connected) {\n      connections.push(session.conn);\n    }\n  }\n\n  return connections;\n}\n\nexport function numberOfConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): number {\n  return getTransportConnections(transport).length;\n}\n\nexport function closeAllConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n) {\n  for (const conn of getTransportConnections(transport)) {\n    conn.close();\n  }\n}\n","/**\n * Represents an observable value that can be subscribed to for changes.\n * This should only be used in tests\n * @template T - The type of the value being observed.\n */\nexport class Observable<T> {\n  value: T;\n  private listeners: Set<(val: T) => void>;\n\n  constructor(initialValue: T) {\n    this.value = initialValue;\n    this.listeners = new Set();\n  }\n\n  /**\n   * Gets the current value of the observable.\n   */\n  get() {\n    return this.value;\n  }\n\n  /**\n   * Sets the current value of the observable. All listeners will get an update with this value.\n   * @param newValue - The new value to set.\n   */\n  set(tx: (preValue: T) => T) {\n    const newValue = tx(this.value);\n    this.value = newValue;\n    this.listeners.forEach((listener) => listener(newValue));\n  }\n\n  /**\n   * Subscribes to changes in the observable value.\n   * @param listener - A callback function that will be called when the value changes.\n   * @returns A function that can be called to unsubscribe from further notifications.\n   */\n  observe(listener: (val: T) => void) {\n    this.listeners.add(listener);\n    listener(this.get());\n\n    return () => this.listeners.delete(listener);\n  }\n\n  /**\n   * Returns the number of listeners currently observing the observable\n   */\n  get listenerCount(): number {\n    return this.listeners.size;\n  }\n}\n","import { Duplex } from 'node:stream';\nimport assert from 'assert';\n\nconst kCallback = Symbol('Callback');\nconst kInitOtherSide = Symbol('InitOtherSide');\n\n// yoinked from https://github.com/nodejs/node/blob/c3a7b29e56a5ada6327ebb622ba746d022685742/lib/internal/streams/duplexpair.js#L55\n// but with types\nclass DuplexSide extends Duplex {\n  private otherSide: DuplexSide | null;\n  private [kCallback]: (() => void) | null;\n\n  constructor() {\n    super();\n    this[kCallback] = null;\n    this.otherSide = null;\n  }\n\n  [kInitOtherSide](otherSide: DuplexSide) {\n    if (this.otherSide === null) {\n      this.otherSide = otherSide;\n    }\n  }\n\n  _read() {\n    const callback = this[kCallback];\n    if (callback) {\n      this[kCallback] = null;\n      callback();\n    }\n  }\n\n  _write(\n    chunk: Uint8Array,\n    _encoding: BufferEncoding,\n    callback: (error?: Error | null) => void,\n  ) {\n    assert(this.otherSide !== null);\n    assert(this.otherSide[kCallback] === null);\n    if (chunk.length === 0) {\n      process.nextTick(callback);\n    } else {\n      this.otherSide.push(chunk);\n      this.otherSide[kCallback] = callback;\n    }\n  }\n\n  _final(callback: (error?: Error | null) => void) {\n    this.otherSide?.on('end', callback);\n    this.otherSide?.push(null);\n  }\n}\n\nexport function duplexPair(): [DuplexSide, DuplexSide] {\n  const side0 = new DuplexSide();\n  const side1 = new DuplexSide();\n  side0[kInitOtherSide](side1);\n  side1[kInitOtherSide](side0);\n  side0.on('close', () => {\n    setImmediate(() => {\n      side1.destroy();\n    });\n  });\n\n  side1.on('close', () => {\n    setImmediate(() => {\n      side0.destroy();\n    });\n  });\n\n  return [side0, side1];\n}\n","import { TransportClientId } from '../../transport';\nimport { ClientTransport } from '../../transport/client';\nimport { Connection } from '../../transport/connection';\nimport { ServerTransport } from '../../transport/server';\nimport { Observable } from '../observable/observable';\nimport { ProvidedServerTransportOptions } from '../../transport/options';\nimport { TestSetupHelpers, TestTransportOptions } from './transports';\nimport { Duplex } from 'node:stream';\nimport { duplexPair } from '../duplex/duplexPair';\nimport { nanoid } from 'nanoid';\n\nexport class InMemoryConnection extends Connection {\n  conn: Duplex;\n\n  constructor(pipe: Duplex) {\n    super();\n    this.conn = pipe;\n    this.conn.allowHalfOpen = false;\n\n    this.conn.on('data', (data: Uint8Array) => {\n      this.dataListener?.(data);\n    });\n\n    this.conn.on('close', () => {\n      this.closeListener?.();\n    });\n\n    this.conn.on('error', (err) => {\n      this.errorListener?.(err);\n    });\n  }\n\n  send(payload: Uint8Array): boolean {\n    setImmediate(() => {\n      this.conn.write(payload);\n    });\n\n    return true;\n  }\n\n  close(): void {\n    setImmediate(() => {\n      this.conn.end();\n      this.conn.emit('close');\n    });\n  }\n}\n\ninterface BidiConnection {\n  id: string;\n  clientToServer: Duplex;\n  serverToClient: Duplex;\n  clientId: TransportClientId;\n  serverId: TransportClientId;\n  handled: boolean;\n}\n\n// we construct a network of transports connected by node streams here\n// so that we can test the transport layer without needing to actually\n// use real network/websocket connections\n// this is useful for testing the transport layer in isolation\n// and allows us to control network conditions in a way that would be\n// difficult with real network connections (e.g. simulating a phantom\n// disconnect, .pause() vs .removeAllListeners('data'), congestion,\n// latency, differences in ws implementations between node and browsers, etc.)\nexport function createMockTransportNetwork(\n  opts?: TestTransportOptions,\n): TestSetupHelpers {\n  // conn id -> [client->server, server->client]\n  const connections = new Observable<Record<string, BidiConnection>>({});\n\n  const transports: Array<MockClientTransport | MockServerTransport> = [];\n  class MockClientTransport extends ClientTransport<InMemoryConnection> {\n    async createNewOutgoingConnection(\n      to: TransportClientId,\n    ): Promise<InMemoryConnection> {\n      const [clientToServer, serverToClient] = duplexPair();\n      await new Promise((resolve) => setImmediate(resolve));\n\n      const connId = nanoid();\n      connections.set((prev) => ({\n        ...prev,\n        [connId]: {\n          id: connId,\n          clientToServer,\n          serverToClient,\n          clientId: this.clientId,\n          serverId: to,\n          handled: false,\n        },\n      }));\n\n      return new InMemoryConnection(clientToServer);\n    }\n  }\n\n  class MockServerTransport extends ServerTransport<InMemoryConnection> {\n    subscribeCleanup: () => void;\n\n    constructor(\n      clientId: TransportClientId,\n      options?: ProvidedServerTransportOptions,\n    ) {\n      super(clientId, options);\n\n      this.subscribeCleanup = connections.observe((conns) => {\n        // look for any unhandled connections\n        for (const conn of Object.values(conns)) {\n          // if we've already handled this connection, skip it\n          // or if it's not for us, skip it\n          if (conn.handled || conn.serverId !== this.clientId) {\n            continue;\n          }\n\n          conn.handled = true;\n          const connection = new InMemoryConnection(conn.serverToClient);\n          this.handleConnection(connection);\n        }\n      });\n    }\n\n    close() {\n      this.subscribeCleanup();\n      super.close();\n    }\n  }\n\n  return {\n    getClientTransport: (id, handshakeOptions) => {\n      const clientTransport = new MockClientTransport(id, opts?.client);\n      if (handshakeOptions) {\n        clientTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(clientTransport);\n\n      return clientTransport;\n    },\n    getServerTransport: (id = 'SERVER', handshakeOptions) => {\n      const serverTransport = new MockServerTransport(id, opts?.server);\n      if (handshakeOptions) {\n        serverTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(serverTransport);\n\n      return serverTransport;\n    },\n    simulatePhantomDisconnect() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.pause();\n        conn.clientToServer.pause();\n      }\n    },\n    async restartServer() {\n      for (const transport of transports) {\n        if (transport.clientId !== 'SERVER') continue;\n        transport.close();\n      }\n\n      // kill all connections while we're at it\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.destroy();\n        conn.clientToServer.destroy();\n      }\n    },\n    cleanup() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.destroy();\n        conn.clientToServer.destroy();\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;;;ACKjC,IAAM,aAAN,MAAoB;AAAA,EACzB;AAAA,EACQ;AAAA,EAER,YAAY,cAAiB;AAC3B,SAAK,QAAQ;AACb,SAAK,YAAY,oBAAI,IAAI;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM;AACJ,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,IAAwB;AAC1B,UAAM,WAAW,GAAG,KAAK,KAAK;AAC9B,SAAK,QAAQ;AACb,SAAK,UAAU,QAAQ,CAAC,aAAa,SAAS,QAAQ,CAAC;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ,UAA4B;AAClC,SAAK,UAAU,IAAI,QAAQ;AAC3B,aAAS,KAAK,IAAI,CAAC;AAEnB,WAAO,MAAM,KAAK,UAAU,OAAO,QAAQ;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,gBAAwB;AAC1B,WAAO,KAAK,UAAU;AAAA,EACxB;AACF;;;ACjDA,SAAS,cAAc;AACvB,OAAO,YAAY;AAEnB,IAAM,YAAY,OAAO,UAAU;AACnC,IAAM,iBAAiB,OAAO,eAAe;AAI7C,IAAM,aAAN,cAAyB,OAAO;AAAA,EACtB;AAAA,EACR,CAAS,SAAS;AAAA,EAElB,cAAc;AACZ,UAAM;AACN,SAAK,SAAS,IAAI;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,CAAC,cAAc,EAAE,WAAuB;AACtC,QAAI,KAAK,cAAc,MAAM;AAC3B,WAAK,YAAY;AAAA,IACnB;AAAA,EACF;AAAA,EAEA,QAAQ;AACN,UAAM,WAAW,KAAK,SAAS;AAC/B,QAAI,UAAU;AACZ,WAAK,SAAS,IAAI;AAClB,eAAS;AAAA,IACX;AAAA,EACF;AAAA,EAEA,OACE,OACA,WACA,UACA;AACA,WAAO,KAAK,cAAc,IAAI;AAC9B,WAAO,KAAK,UAAU,SAAS,MAAM,IAAI;AACzC,QAAI,MAAM,WAAW,GAAG;AACtB,cAAQ,SAAS,QAAQ;AAAA,IAC3B,OAAO;AACL,WAAK,UAAU,KAAK,KAAK;AACzB,WAAK,UAAU,SAAS,IAAI;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,OAAO,UAA0C;AAC/C,SAAK,WAAW,GAAG,OAAO,QAAQ;AAClC,SAAK,WAAW,KAAK,IAAI;AAAA,EAC3B;AACF;AAEO,SAAS,aAAuC;AACrD,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,cAAc,EAAE,KAAK;AAC3B,QAAM,cAAc,EAAE,KAAK;AAC3B,QAAM,GAAG,SAAS,MAAM;AACtB,iBAAa,MAAM;AACjB,YAAM,QAAQ;AAAA,IAChB,CAAC;AAAA,EACH,CAAC;AAED,QAAM,GAAG,SAAS,MAAM;AACtB,iBAAa,MAAM;AACjB,YAAM,QAAQ;AAAA,IAChB,CAAC;AAAA,EACH,CAAC;AAED,SAAO,CAAC,OAAO,KAAK;AACtB;;;AC9DA,SAAS,cAAc;AAEhB,IAAM,qBAAN,cAAiC,WAAW;AAAA,EACjD;AAAA,EAEA,YAAY,MAAc;AACxB,UAAM;AACN,SAAK,OAAO;AACZ,SAAK,KAAK,gBAAgB;AAE1B,SAAK,KAAK,GAAG,QAAQ,CAAC,SAAqB;AACzC,WAAK,eAAe,IAAI;AAAA,IAC1B,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,MAAM;AAC1B,WAAK,gBAAgB;AAAA,IACvB,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,CAAC,QAAQ;AAC7B,WAAK,gBAAgB,GAAG;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EAEA,KAAK,SAA8B;AACjC,iBAAa,MAAM;AACjB,WAAK,KAAK,MAAM,OAAO;AAAA,IACzB,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEA,QAAc;AACZ,iBAAa,MAAM;AACjB,WAAK,KAAK,IAAI;AACd,WAAK,KAAK,KAAK,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAmBO,SAAS,2BACd,MACkB;AAElB,QAAM,cAAc,IAAI,WAA2C,CAAC,CAAC;AAErE,QAAM,aAA+D,CAAC;AAAA,EACtE,MAAM,4BAA4B,gBAAoC;AAAA,IACpE,MAAM,4BACJ,IAC6B;AAC7B,YAAM,CAAC,gBAAgB,cAAc,IAAI,WAAW;AACpD,YAAM,IAAI,QAAQ,CAAC,YAAY,aAAa,OAAO,CAAC;AAEpD,YAAM,SAAS,OAAO;AACtB,kBAAY,IAAI,CAAC,UAAU;AAAA,QACzB,GAAG;AAAA,QACH,CAAC,MAAM,GAAG;AAAA,UACR,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,UACA,UAAU,KAAK;AAAA,UACf,UAAU;AAAA,UACV,SAAS;AAAA,QACX;AAAA,MACF,EAAE;AAEF,aAAO,IAAI,mBAAmB,cAAc;AAAA,IAC9C;AAAA,EACF;AAAA,EAEA,MAAM,4BAA4B,gBAAoC;AAAA,IACpE;AAAA,IAEA,YACE,UACA,SACA;AACA,YAAM,UAAU,OAAO;AAEvB,WAAK,mBAAmB,YAAY,QAAQ,CAAC,UAAU;AAErD,mBAAW,QAAQ,OAAO,OAAO,KAAK,GAAG;AAGvC,cAAI,KAAK,WAAW,KAAK,aAAa,KAAK,UAAU;AACnD;AAAA,UACF;AAEA,eAAK,UAAU;AACf,gBAAM,aAAa,IAAI,mBAAmB,KAAK,cAAc;AAC7D,eAAK,iBAAiB,UAAU;AAAA,QAClC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IAEA,QAAQ;AACN,WAAK,iBAAiB;AACtB,YAAM,MAAM;AAAA,IACd;AAAA,EACF;AAEA,SAAO;AAAA,IACL,oBAAoB,CAAC,IAAI,qBAAqB;AAC5C,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,oBAAoB,CAAC,KAAK,UAAU,qBAAqB;AACvD,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,4BAA4B;AAC1B,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,MAAM;AAC1B,aAAK,eAAe,MAAM;AAAA,MAC5B;AAAA,IACF;AAAA,IACA,MAAM,gBAAgB;AACpB,iBAAW,aAAa,YAAY;AAClC,YAAI,UAAU,aAAa;AAAU;AACrC,kBAAU,MAAM;AAAA,MAClB;AAGA,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,QAAQ;AAC5B,aAAK,eAAe,QAAQ;AAAA,MAC9B;AAAA,IACF;AAAA,IACA,UAAU;AACR,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,QAAQ;AAC5B,aAAK,eAAe,QAAQ;AAAA,MAC9B;AAAA,IACF;AAAA,EACF;AACF;;;AH3IO,SAAS,2BAA2B,MAAsB;AAC/D,QAAM,OAAO,IAAI,OAAO,kBAAkB,IAAI,EAAE;AAChD,OAAK,aAAa;AAElB,SAAO;AACT;AAQO,SAAS,sBAAsB,QAAqB;AACzD,SAAO,IAAI,gBAAgB,EAAE,OAAO,CAAC;AACvC;AASO,SAAS,gBAAgB,QAAsC;AACpE,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAO,OAAO,MAAM;AAClB,YAAM,OAAO,OAAO,QAAQ;AAC5B,UAAI,OAAO,SAAS,YAAY,MAAM;AACpC,gBAAQ,KAAK,IAAI;AAAA,MACnB,OAAO;AACL,eAAO,IAAI,MAAM,kCAAkC,CAAC;AAAA,MACtD;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AACH;AAEA,IAAM,oBAAoB,oBAAI,QAG5B;AASK,SAAS,oBACd,UACwB;AACxB,MAAI,OAAO,kBAAkB,IAAI,QAAQ;AAIzC,MAAI,CAAC,MAAM;AACT,WAAO,SAAS,OAAO,aAAa,EAAE;AACtC,sBAAkB,IAAI,UAAU,IAAI;AAAA,EACtC;AAEA,SAAO;AACT;AAOA,eAAsB,eACpB,UAC+B;AAC/B,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,MAAI,IAAI,MAAM;AACZ,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO,IAAI;AACb;AAOA,eAAsB,eACpB,UACA;AACA,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,SAAO,IAAI;AACb;AAEO,SAAS,0BACd,SACkC;AAClC,SAAO;AAAA,IACL,UAAU;AAAA,IACV,cAAc;AAAA,IACd;AAAA,EACF;AACF;AAEO,SAAS,8BAA8B;AAC5C,SAAO,0BAA0B;AAAA,IAC/B,KAAK;AAAA,IACL,MAAM,KAAK,OAAO;AAAA,EACpB,CAAC;AACH;AAQA,eAAsB,eACpB,GACA,QACA,gBACA;AACA,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,aAAS,UAAU;AACjB,QAAE,oBAAoB,WAAW,SAAS;AAAA,IAC5C;AAEA,aAAS,UAAU,KAA6B;AAC9C,UAAI,CAAC,UAAU,OAAO,GAAG,GAAG;AAC1B,gBAAQ;AACR,gBAAQ,IAAI,OAAO;AAAA,MACrB,WAAW,gBAAgB;AACzB,gBAAQ;AACR,eAAO,IAAI,MAAM,gCAAgC,CAAC;AAAA,MACpD;AAAA,IACF;AAEA,MAAE,iBAAiB,WAAW,SAAS;AAAA,EACzC,CAAC;AACH;AAEO,IAAM,wBAAwB;AAC9B,IAAM,8BAA8B;AAEpC,SAAS,eAAe;AAC7B,SAAO,kBAAkB,YAAY;AAAA,IACnC;AAAA,IACA;AAAA,IACA;AAAA,MACE,6BAA6B,MAAM;AAAA,MAEnC;AAAA,IACF;AAAA,IACA;AAAA,IACA;AAAA,IACA,UAAU;AAAA,EACZ;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UACJ,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ,KACrD,gBAAgB,yBAAyB,gBAAgB,QAAQ;AAEnE,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UAAU,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ;AACrE,MAAI,CAAC,SAAS;AACZ,UAAM,IAAI,MAAM,mBAAmB;AAAA,EACrC;AAEA,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,wBACd,WACiB;AACjB,QAAM,cAAc,CAAC;AACrB,aAAW,WAAW,UAAU,SAAS,OAAO,GAAG;AACjD,QAAI,QAAQ,uCAAkC;AAC5C,kBAAY,KAAK,QAAQ,IAAI;AAAA,IAC/B;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,oBACd,WACQ;AACR,SAAO,wBAAwB,SAAS,EAAE;AAC5C;AAEO,SAAS,oBACd,WACA;AACA,aAAW,QAAQ,wBAAwB,SAAS,GAAG;AACrD,SAAK,MAAM;AAAA,EACb;AACF;","names":[]}

package/dist/transport/impls/ws/client.cjs

@@ -189,7 +189,7 @@ var NaiveJsonCodec = {
     const parsed = JSON.parse(
       decoder.decode(buff),
       function reviver(_key, val) {
-        if (val?.$t) {
+        if (val?.$t !== void 0) {
           return base64ToUint8Array(val.$t);
         } else {
           return val;
@@ -675,7 +675,7 @@ function coerceErrorString(err) {
 }
 
 // package.json
-var version = "0.207.3";
+var version = "0.208.1";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -689,7 +689,7 @@ function getPropagationContext(ctx) {
 function createSessionTelemetryInfo(tracer, sessionId, to, from, propagationCtx) {
   const parentCtx = propagationCtx ? import_api2.propagation.extract(import_api2.context.active(), propagationCtx) : import_api2.context.active();
   const span = tracer.startSpan(
-    `river.session.${sessionId}`,
+    `river.session`,
     {
       attributes: {
         component: "river",
@@ -705,7 +705,7 @@ function createSessionTelemetryInfo(tracer, sessionId, to, from, propagationCtx)
 }
 function createConnectionTelemetryInfo(tracer, connection, info) {
   const span = tracer.startSpan(
-    `connection ${connection.id}`,
+    `river.connection`,
     {
       attributes: {
         component: "river",
@@ -735,9 +735,9 @@ var SessionWaitingForHandshake = class extends CommonSession {
     this.handshakeTimeout = setTimeout(() => {
       this.listeners.onHandshakeTimeout();
     }, this.options.handshakeTimeoutMs);
-    this.conn.addDataListener(this.onHandshakeData);
-    this.conn.addErrorListener(this.listeners.onConnectionErrored);
-    this.conn.addCloseListener(this.listeners.onConnectionClosed);
+    this.conn.setDataListener(this.onHandshakeData);
+    this.conn.setErrorListener(this.listeners.onConnectionErrored);
+    this.conn.setCloseListener(this.listeners.onConnectionClosed);
   }
   get loggingMetadata() {
     return {
@@ -761,9 +761,9 @@ var SessionWaitingForHandshake = class extends CommonSession {
     return sendMessage(this.conn, this.codec, msg);
   }
   _handleStateExit() {
-    this.conn.removeDataListener(this.onHandshakeData);
-    this.conn.removeErrorListener(this.listeners.onConnectionErrored);
-    this.conn.removeCloseListener(this.listeners.onConnectionClosed);
+    this.conn.removeDataListener();
+    this.conn.removeErrorListener();
+    this.conn.removeCloseListener();
     clearTimeout(this.handshakeTimeout);
     this.handshakeTimeout = void 0;
   }
@@ -785,9 +785,9 @@ var SessionHandshaking = class extends IdentifiedSessionWithGracePeriod {
     this.handshakeTimeout = setTimeout(() => {
       this.listeners.onHandshakeTimeout();
     }, this.options.handshakeTimeoutMs);
-    this.conn.addDataListener(this.onHandshakeData);
-    this.conn.addErrorListener(this.listeners.onConnectionErrored);
-    this.conn.addCloseListener(this.listeners.onConnectionClosed);
+    this.conn.setDataListener(this.onHandshakeData);
+    this.conn.setErrorListener(this.listeners.onConnectionErrored);
+    this.conn.setCloseListener(this.listeners.onConnectionClosed);
   }
   get loggingMetadata() {
     return {
@@ -811,9 +811,9 @@ var SessionHandshaking = class extends IdentifiedSessionWithGracePeriod {
   }
   _handleStateExit() {
     super._handleStateExit();
-    this.conn.removeDataListener(this.onHandshakeData);
-    this.conn.removeErrorListener(this.listeners.onConnectionErrored);
-    this.conn.removeCloseListener(this.listeners.onConnectionClosed);
+    this.conn.removeDataListener();
+    this.conn.removeErrorListener();
+    this.conn.removeCloseListener();
     if (this.handshakeTimeout) {
       clearTimeout(this.handshakeTimeout);
       this.handshakeTimeout = void 0;
@@ -869,9 +869,9 @@ var SessionConnected = class extends IdentifiedSession {
     super(props);
     this.conn = props.conn;
     this.listeners = props.listeners;
-    this.conn.addDataListener(this.onMessageData);
-    this.conn.addCloseListener(this.listeners.onConnectionClosed);
-    this.conn.addErrorListener(this.listeners.onConnectionErrored);
+    this.conn.setDataListener(this.onMessageData);
+    this.conn.setCloseListener(this.listeners.onConnectionClosed);
+    this.conn.setErrorListener(this.listeners.onConnectionErrored);
   }
   sendBufferedMessages() {
     if (this.sendBuffer.length > 0) {
@@ -980,9 +980,9 @@ var SessionConnected = class extends IdentifiedSession {
   };
   _handleStateExit() {
     super._handleStateExit();
-    this.conn.removeDataListener(this.onMessageData);
-    this.conn.removeCloseListener(this.listeners.onConnectionClosed);
-    this.conn.removeErrorListener(this.listeners.onConnectionErrored);
+    this.conn.removeDataListener();
+    this.conn.removeCloseListener();
+    this.conn.removeErrorListener();
     if (this.heartbeatHandle) {
       clearInterval(this.heartbeatHandle);
       this.heartbeatHandle = void 0;
@@ -1043,60 +1043,44 @@ var Connection = class {
     }
     return metadata;
   }
-  // can't use event emitter because we need this to work in both node + browser
-  _dataListeners = /* @__PURE__ */ new Set();
-  _closeListeners = /* @__PURE__ */ new Set();
-  _errorListeners = /* @__PURE__ */ new Set();
-  get dataListeners() {
-    return [...this._dataListeners];
-  }
-  get closeListeners() {
-    return [...this._closeListeners];
-  }
-  get errorListeners() {
-    return [...this._errorListeners];
-  }
+  dataListener;
+  closeListener;
+  errorListener;
   onData(msg) {
-    for (const cb of this.dataListeners) {
-      cb(msg);
-    }
+    this.dataListener?.(msg);
   }
   onError(err) {
-    for (const cb of this.errorListeners) {
-      cb(err);
-    }
+    this.errorListener?.(err);
   }
   onClose() {
-    for (const cb of this.closeListeners) {
-      cb();
-    }
+    this.closeListener?.();
     this.telemetry?.span.end();
   }
   /**
-   * Handle adding a callback for when a message is received.
-   * @param msg The message that was received.
+   * Set the callback for when a message is received.
+   * @param cb The message handler callback.
    */
-  addDataListener(cb) {
-    this._dataListeners.add(cb);
+  setDataListener(cb) {
+    this.dataListener = cb;
   }
-  removeDataListener(cb) {
-    this._dataListeners.delete(cb);
+  removeDataListener() {
+    this.dataListener = void 0;
   }
   /**
-   * Handle adding a callback for when the connection is closed.
-   * This should also be called if an error happens and after notifying all the error listeners.
+   * Set the callback for when the connection is closed.
+   * This should also be called if an error happens and after notifying the error listener.
    * @param cb The callback to call when the connection is closed.
    */
-  addCloseListener(cb) {
-    this._closeListeners.add(cb);
+  setCloseListener(cb) {
+    this.closeListener = cb;
   }
-  removeCloseListener(cb) {
-    this._closeListeners.delete(cb);
+  removeCloseListener() {
+    this.closeListener = void 0;
   }
   /**
-   * Handle adding a callback for when an error is received.
-   * This should only be used for this.logging errors, all cleanup
-   * should be delegated to addCloseListener.
+   * Set the callback for when an error is received.
+   * This should only be used for logging errors, all cleanup
+   * should be delegated to setCloseListener.
    *
    * The implementer should take care such that the implemented
    * connection will call both the close and error callbacks
@@ -1104,11 +1088,11 @@ var Connection = class {
    *
    * @param cb The callback to call when an error is received.
    */
-  addErrorListener(cb) {
-    this._errorListeners.add(cb);
+  setErrorListener(cb) {
+    this.errorListener = cb;
   }
-  removeErrorListener(cb) {
-    this._errorListeners.delete(cb);
+  removeErrorListener() {
+    this.errorListener = void 0;
   }
 };
 
@@ -1626,7 +1610,7 @@ var Transport = class {
   }
   // common listeners
   onSessionGracePeriodElapsed(session) {
-    this.log?.warn(
+    this.log?.info(
       `session to ${session.to} grace period elapsed, closing`,
       session.loggingMetadata
     );
@@ -1913,15 +1897,6 @@ var ClientTransport = class extends Transport {
     });
     const res = connectedSession.sendBufferedMessages();
     if (!res.ok) {
-      this.log?.error(`failed to send buffered messages: ${res.reason}`, {
-        ...connectedSession.loggingMetadata,
-        transportMessage: msg
-      });
-      this.protocolError({
-        type: ProtocolError.MessageSendFailure,
-        message: res.reason
-      });
-      this.deleteSession(connectedSession, { unhealthy: true });
       return;
     }
     this.updateSession(connectedSession);