@replit/river 0.24.0 → 0.24.2

package/dist/util/testHelpers.d.ts

@@ -1,13 +1,13 @@
 import { Static } from '@sinclair/typebox';
-import { P as PayloadType, R as RiverError, a as Procedure, b as Result, c as RiverUncaughtSchema, d as ProcedureResult } from '../services-e1417b33.js';
-import { T as Transport, C as Connection, m as SessionOptions, e as SessionNoConnection, n as ServiceContext, S as Session } from '../handshake-cdead82a.js';
+import { P as PayloadType, R as RiverError, a as Procedure, b as Result, c as RiverUncaughtSchema, d as ProcedureResult } from '../services-adfd0bc3.js';
+import { T as Transport, C as Connection, m as SessionOptions, e as SessionNoConnection, n as ServiceContext, S as Session } from '../handshake-0b88e8fc.js';
 import * as it_pushable from 'it-pushable';
 import { P as PartialTransportMessage, b as OpaqueTransportMessage } from '../message-e6c560fd.js';
 import NodeWs from 'ws';
 import http from 'node:http';
 import net from 'node:net';
 import { W as WsLike } from '../wslike-e0b32dd5.js';
-import '../client-ba0d3315.js';
+import '../client-2ba72e89.js';
 import '../types-3e5768ec.js';
 import '@sinclair/typebox/value';
 import '@opentelemetry/api';

package/dist/util/testHelpers.js

@@ -2,15 +2,15 @@ import {
   Err,
   UNCAUGHT_ERROR,
   pushable
-} from "../chunk-KYYB4DUR.js";
+} from "../chunk-TQ44W7PR.js";
 import {
   SessionStateGraph,
   defaultTransportOptions
-} from "../chunk-JA57I7MG.js";
+} from "../chunk-U4UKZXWA.js";
 import {
   coerceErrorString,
   generateId
-} from "../chunk-PJGGC3LV.js";
+} from "../chunk-STDBD66K.js";
 import "../chunk-4PVU7J25.js";
 
 // util/testHelpers.ts
@@ -94,6 +94,7 @@ function dummyCtx(state, session, extendedContext) {
   return {
     ...extendedContext,
     state,
+    sessionId: session.id,
     to: session.to,
     from: session.from,
     streamId: generateId(),

package/dist/util/testHelpers.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../util/testHelpers.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { pushable } from 'it-pushable';\nimport {\n  Err,\n  PayloadType,\n  Procedure,\n  ServiceContext,\n  ServiceContextWithTransportInfo,\n  UNCAUGHT_ERROR,\n} from '../router';\nimport { RiverError, Result, RiverUncaughtSchema } from '../router/result';\nimport { Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport net from 'node:net';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n} from '../transport/message';\nimport { coerceErrorString } from './stringify';\nimport { Transport } from '../transport/transport';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport { defaultTransportOptions } from '../transport/options';\nimport { generateId } from '../transport/id';\nimport { Connection } from '../transport/connection';\nimport {\n  Session,\n  SessionOptions,\n  SessionState,\n} from '../transport/sessionStateMachine/common';\nimport { SessionStateGraph } from '../transport/sessionStateMachine';\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\nexport function onUdsServeReady(\n  server: net.Server,\n  path: string,\n): Promise<void> {\n  return new Promise<void>((resolve) => {\n    server.listen(path, resolve);\n  });\n}\n\n/**\n * Retrieves the next value from an async iterable iterator.\n * @param iter The async iterable iterator.\n * @returns A promise that resolves to the next value from the iterator.\n */\nexport async function iterNext<T>(iter: AsyncIterableIterator<T>) {\n  return await iter.next().then((res) => res.value as T);\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\nfunction catchProcError(err: unknown) {\n  const errorMsg = coerceErrorString(err);\n  return Err({ code: UNCAUGHT_ERROR, message: errorMsg });\n}\n\nexport const testingSessionOptions: SessionOptions = defaultTransportOptions;\n\nexport function dummySession() {\n  return SessionStateGraph.entrypoints.NoConnection(\n    'client',\n    'server',\n    {\n      onSessionGracePeriodElapsed: () => {\n        /* noop */\n      },\n    },\n    testingSessionOptions,\n  );\n}\n\nfunction dummyCtx<State>(\n  state: State,\n  session: Session<Connection>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n): ServiceContextWithTransportInfo<State> {\n  return {\n    ...extendedContext,\n    state,\n    to: session.to,\n    from: session.from,\n    streamId: generateId(),\n    metadata: {},\n  };\n}\n\nexport function asClientRpc<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n  Init extends PayloadType | null = null,\n>(\n  state: State,\n  proc: Procedure<State, 'rpc', I, O, E, Init>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  return async (\n    msg: Static<I>,\n  ): Promise<\n    Result<Static<O>, Static<E> | Static<typeof RiverUncaughtSchema>>\n  > => {\n    return await proc\n      .handler(dummyCtx(state, session, extendedContext), msg)\n      .catch(catchProcError);\n  };\n}\n\nexport function asClientStream<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n  Init extends PayloadType | null = null,\n>(\n  state: State,\n  proc: Procedure<State, 'stream', I, O, E, Init>,\n  init?: Init extends PayloadType ? Static<Init> : null,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  const input = pushable<Static<I>>({ objectMode: true });\n  const output = pushable<Result<Static<O>, Static<E>>>({\n    objectMode: true,\n  });\n\n  void (async () => {\n    if (init) {\n      const _proc = proc as Procedure<State, 'stream', I, O, E, PayloadType>;\n      await _proc\n        .handler(dummyCtx(state, session, extendedContext), init, input, output)\n        .catch((err: unknown) => output.push(catchProcError(err)));\n    } else {\n      const _proc = proc as Procedure<State, 'stream', I, O, E>;\n      await _proc\n        .handler(dummyCtx(state, session, extendedContext), input, output)\n        .catch((err: unknown) => output.push(catchProcError(err)));\n    }\n  })();\n\n  return [input, output] as const;\n}\n\nexport function asClientSubscription<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n>(\n  state: State,\n  proc: Procedure<State, 'subscription', I, O, E>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  const output = pushable<Result<Static<O>, Static<E>>>({\n    objectMode: true,\n  });\n\n  return (msg: Static<I>) => {\n    void (async () => {\n      return await proc\n        .handler(dummyCtx(state, session, extendedContext), msg, output)\n        .catch((err: unknown) => output.push(catchProcError(err)));\n    })();\n    return output;\n  };\n}\n\nexport function asClientUpload<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n  Init extends PayloadType | null = null,\n>(\n  state: State,\n  proc: Procedure<State, 'upload', I, O, E, Init>,\n  init?: Init extends PayloadType ? Static<Init> : null,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  const input = pushable<Static<I>>({ objectMode: true });\n  if (init) {\n    const _proc = proc as Procedure<State, 'upload', I, O, E, PayloadType>;\n    const result = _proc\n      .handler(dummyCtx(state, session, extendedContext), init, input)\n      .catch(catchProcError);\n    return [input, result] as const;\n  } else {\n    const _proc = proc as Procedure<State, 'upload', I, O, E>;\n    const result = _proc\n      .handler(dummyCtx(state, session, extendedContext), input)\n      .catch(catchProcError);\n    return [input, result] as const;\n  }\n}\n\nexport const getUnixSocketPath = () => {\n  return `/tmp/${nanoid()}.sock`;\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"],"mappings":";;;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;AAaxC,SAAS,cAAc;AAyBhB,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;AAEO,SAAS,gBACd,QACA,MACe;AACf,SAAO,IAAI,QAAc,CAAC,YAAY;AACpC,WAAO,OAAO,MAAM,OAAO;AAAA,EAC7B,CAAC;AACH;AAOA,eAAsB,SAAY,MAAgC;AAChE,SAAO,MAAM,KAAK,KAAK,EAAE,KAAK,CAAC,QAAQ,IAAI,KAAU;AACvD;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;AAEA,SAAS,eAAe,KAAc;AACpC,QAAM,WAAW,kBAAkB,GAAG;AACtC,SAAO,IAAI,EAAE,MAAM,gBAAgB,SAAS,SAAS,CAAC;AACxD;AAEO,IAAM,wBAAwC;AAE9C,SAAS,eAAe;AAC7B,SAAO,kBAAkB,YAAY;AAAA,IACnC;AAAA,IACA;AAAA,IACA;AAAA,MACE,6BAA6B,MAAM;AAAA,MAEnC;AAAA,IACF;AAAA,IACA;AAAA,EACF;AACF;AAEA,SAAS,SACP,OACA,SACA,iBACwC;AACxC,SAAO;AAAA,IACL,GAAG;AAAA,IACH;AAAA,IACA,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,UAAU,WAAW;AAAA,IACrB,UAAU,CAAC;AAAA,EACb;AACF;AAEO,SAAS,YAOd,OACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,SAAO,OACL,QAGG;AACH,WAAO,MAAM,KACV,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,GAAG,EACtD,MAAM,cAAc;AAAA,EACzB;AACF;AAEO,SAAS,eAOd,OACA,MACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,QAAM,QAAQ,SAAoB,EAAE,YAAY,KAAK,CAAC;AACtD,QAAM,SAAS,SAAuC;AAAA,IACpD,YAAY;AAAA,EACd,CAAC;AAED,QAAM,YAAY;AAChB,QAAI,MAAM;AACR,YAAM,QAAQ;AACd,YAAM,MACH,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,MAAM,OAAO,MAAM,EACtE,MAAM,CAAC,QAAiB,OAAO,KAAK,eAAe,GAAG,CAAC,CAAC;AAAA,IAC7D,OAAO;AACL,YAAM,QAAQ;AACd,YAAM,MACH,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,OAAO,MAAM,EAChE,MAAM,CAAC,QAAiB,OAAO,KAAK,eAAe,GAAG,CAAC,CAAC;AAAA,IAC7D;AAAA,EACF,GAAG;AAEH,SAAO,CAAC,OAAO,MAAM;AACvB;AAEO,SAAS,qBAMd,OACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,QAAM,SAAS,SAAuC;AAAA,IACpD,YAAY;AAAA,EACd,CAAC;AAED,SAAO,CAAC,QAAmB;AACzB,UAAM,YAAY;AAChB,aAAO,MAAM,KACV,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,KAAK,MAAM,EAC9D,MAAM,CAAC,QAAiB,OAAO,KAAK,eAAe,GAAG,CAAC,CAAC;AAAA,IAC7D,GAAG;AACH,WAAO;AAAA,EACT;AACF;AAEO,SAAS,eAOd,OACA,MACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,QAAM,QAAQ,SAAoB,EAAE,YAAY,KAAK,CAAC;AACtD,MAAI,MAAM;AACR,UAAM,QAAQ;AACd,UAAM,SAAS,MACZ,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,MAAM,KAAK,EAC9D,MAAM,cAAc;AACvB,WAAO,CAAC,OAAO,MAAM;AAAA,EACvB,OAAO;AACL,UAAM,QAAQ;AACd,UAAM,SAAS,MACZ,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,KAAK,EACxD,MAAM,cAAc;AACvB,WAAO,CAAC,OAAO,MAAM;AAAA,EACvB;AACF;AAEO,IAAM,oBAAoB,MAAM;AACrC,SAAO,QAAQ,OAAO,CAAC;AACzB;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":["../../util/testHelpers.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { pushable } from 'it-pushable';\nimport {\n  Err,\n  PayloadType,\n  Procedure,\n  ServiceContext,\n  ServiceContextWithTransportInfo,\n  UNCAUGHT_ERROR,\n} from '../router';\nimport { RiverError, Result, RiverUncaughtSchema } from '../router/result';\nimport { Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport net from 'node:net';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n} from '../transport/message';\nimport { coerceErrorString } from './stringify';\nimport { Transport } from '../transport/transport';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport { defaultTransportOptions } from '../transport/options';\nimport { generateId } from '../transport/id';\nimport { Connection } from '../transport/connection';\nimport {\n  Session,\n  SessionOptions,\n  SessionState,\n} from '../transport/sessionStateMachine/common';\nimport { SessionStateGraph } from '../transport/sessionStateMachine';\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\nexport function onUdsServeReady(\n  server: net.Server,\n  path: string,\n): Promise<void> {\n  return new Promise<void>((resolve) => {\n    server.listen(path, resolve);\n  });\n}\n\n/**\n * Retrieves the next value from an async iterable iterator.\n * @param iter The async iterable iterator.\n * @returns A promise that resolves to the next value from the iterator.\n */\nexport async function iterNext<T>(iter: AsyncIterableIterator<T>) {\n  return await iter.next().then((res) => res.value as T);\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\nfunction catchProcError(err: unknown) {\n  const errorMsg = coerceErrorString(err);\n  return Err({ code: UNCAUGHT_ERROR, message: errorMsg });\n}\n\nexport const testingSessionOptions: SessionOptions = defaultTransportOptions;\n\nexport function dummySession() {\n  return SessionStateGraph.entrypoints.NoConnection(\n    'client',\n    'server',\n    {\n      onSessionGracePeriodElapsed: () => {\n        /* noop */\n      },\n    },\n    testingSessionOptions,\n  );\n}\n\nfunction dummyCtx<State>(\n  state: State,\n  session: Session<Connection>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n): ServiceContextWithTransportInfo<State> {\n  return {\n    ...extendedContext,\n    state,\n    sessionId: session.id,\n    to: session.to,\n    from: session.from,\n    streamId: generateId(),\n    metadata: {},\n  };\n}\n\nexport function asClientRpc<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n  Init extends PayloadType | null = null,\n>(\n  state: State,\n  proc: Procedure<State, 'rpc', I, O, E, Init>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  return async (\n    msg: Static<I>,\n  ): Promise<\n    Result<Static<O>, Static<E> | Static<typeof RiverUncaughtSchema>>\n  > => {\n    return await proc\n      .handler(dummyCtx(state, session, extendedContext), msg)\n      .catch(catchProcError);\n  };\n}\n\nexport function asClientStream<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n  Init extends PayloadType | null = null,\n>(\n  state: State,\n  proc: Procedure<State, 'stream', I, O, E, Init>,\n  init?: Init extends PayloadType ? Static<Init> : null,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  const input = pushable<Static<I>>({ objectMode: true });\n  const output = pushable<Result<Static<O>, Static<E>>>({\n    objectMode: true,\n  });\n\n  void (async () => {\n    if (init) {\n      const _proc = proc as Procedure<State, 'stream', I, O, E, PayloadType>;\n      await _proc\n        .handler(dummyCtx(state, session, extendedContext), init, input, output)\n        .catch((err: unknown) => output.push(catchProcError(err)));\n    } else {\n      const _proc = proc as Procedure<State, 'stream', I, O, E>;\n      await _proc\n        .handler(dummyCtx(state, session, extendedContext), input, output)\n        .catch((err: unknown) => output.push(catchProcError(err)));\n    }\n  })();\n\n  return [input, output] as const;\n}\n\nexport function asClientSubscription<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n>(\n  state: State,\n  proc: Procedure<State, 'subscription', I, O, E>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  const output = pushable<Result<Static<O>, Static<E>>>({\n    objectMode: true,\n  });\n\n  return (msg: Static<I>) => {\n    void (async () => {\n      return await proc\n        .handler(dummyCtx(state, session, extendedContext), msg, output)\n        .catch((err: unknown) => output.push(catchProcError(err)));\n    })();\n    return output;\n  };\n}\n\nexport function asClientUpload<\n  State extends object,\n  I extends PayloadType,\n  O extends PayloadType,\n  E extends RiverError,\n  Init extends PayloadType | null = null,\n>(\n  state: State,\n  proc: Procedure<State, 'upload', I, O, E, Init>,\n  init?: Init extends PayloadType ? Static<Init> : null,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  const input = pushable<Static<I>>({ objectMode: true });\n  if (init) {\n    const _proc = proc as Procedure<State, 'upload', I, O, E, PayloadType>;\n    const result = _proc\n      .handler(dummyCtx(state, session, extendedContext), init, input)\n      .catch(catchProcError);\n    return [input, result] as const;\n  } else {\n    const _proc = proc as Procedure<State, 'upload', I, O, E>;\n    const result = _proc\n      .handler(dummyCtx(state, session, extendedContext), input)\n      .catch(catchProcError);\n    return [input, result] as const;\n  }\n}\n\nexport const getUnixSocketPath = () => {\n  return `/tmp/${nanoid()}.sock`;\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"],"mappings":";;;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;AAaxC,SAAS,cAAc;AAyBhB,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;AAEO,SAAS,gBACd,QACA,MACe;AACf,SAAO,IAAI,QAAc,CAAC,YAAY;AACpC,WAAO,OAAO,MAAM,OAAO;AAAA,EAC7B,CAAC;AACH;AAOA,eAAsB,SAAY,MAAgC;AAChE,SAAO,MAAM,KAAK,KAAK,EAAE,KAAK,CAAC,QAAQ,IAAI,KAAU;AACvD;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;AAEA,SAAS,eAAe,KAAc;AACpC,QAAM,WAAW,kBAAkB,GAAG;AACtC,SAAO,IAAI,EAAE,MAAM,gBAAgB,SAAS,SAAS,CAAC;AACxD;AAEO,IAAM,wBAAwC;AAE9C,SAAS,eAAe;AAC7B,SAAO,kBAAkB,YAAY;AAAA,IACnC;AAAA,IACA;AAAA,IACA;AAAA,MACE,6BAA6B,MAAM;AAAA,MAEnC;AAAA,IACF;AAAA,IACA;AAAA,EACF;AACF;AAEA,SAAS,SACP,OACA,SACA,iBACwC;AACxC,SAAO;AAAA,IACL,GAAG;AAAA,IACH;AAAA,IACA,WAAW,QAAQ;AAAA,IACnB,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,UAAU,WAAW;AAAA,IACrB,UAAU,CAAC;AAAA,EACb;AACF;AAEO,SAAS,YAOd,OACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,SAAO,OACL,QAGG;AACH,WAAO,MAAM,KACV,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,GAAG,EACtD,MAAM,cAAc;AAAA,EACzB;AACF;AAEO,SAAS,eAOd,OACA,MACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,QAAM,QAAQ,SAAoB,EAAE,YAAY,KAAK,CAAC;AACtD,QAAM,SAAS,SAAuC;AAAA,IACpD,YAAY;AAAA,EACd,CAAC;AAED,QAAM,YAAY;AAChB,QAAI,MAAM;AACR,YAAM,QAAQ;AACd,YAAM,MACH,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,MAAM,OAAO,MAAM,EACtE,MAAM,CAAC,QAAiB,OAAO,KAAK,eAAe,GAAG,CAAC,CAAC;AAAA,IAC7D,OAAO;AACL,YAAM,QAAQ;AACd,YAAM,MACH,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,OAAO,MAAM,EAChE,MAAM,CAAC,QAAiB,OAAO,KAAK,eAAe,GAAG,CAAC,CAAC;AAAA,IAC7D;AAAA,EACF,GAAG;AAEH,SAAO,CAAC,OAAO,MAAM;AACvB;AAEO,SAAS,qBAMd,OACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,QAAM,SAAS,SAAuC;AAAA,IACpD,YAAY;AAAA,EACd,CAAC;AAED,SAAO,CAAC,QAAmB;AACzB,UAAM,YAAY;AAChB,aAAO,MAAM,KACV,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,KAAK,MAAM,EAC9D,MAAM,CAAC,QAAiB,OAAO,KAAK,eAAe,GAAG,CAAC,CAAC;AAAA,IAC7D,GAAG;AACH,WAAO;AAAA,EACT;AACF;AAEO,SAAS,eAOd,OACA,MACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,QAAM,QAAQ,SAAoB,EAAE,YAAY,KAAK,CAAC;AACtD,MAAI,MAAM;AACR,UAAM,QAAQ;AACd,UAAM,SAAS,MACZ,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,MAAM,KAAK,EAC9D,MAAM,cAAc;AACvB,WAAO,CAAC,OAAO,MAAM;AAAA,EACvB,OAAO;AACL,UAAM,QAAQ;AACd,UAAM,SAAS,MACZ,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,KAAK,EACxD,MAAM,cAAc;AACvB,WAAO,CAAC,OAAO,MAAM;AAAA,EACvB;AACF;AAEO,IAAM,oBAAoB,MAAM;AACrC,SAAO,QAAQ,OAAO,CAAC;AACzB;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/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@replit/river",
   "description": "It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!",
-  "version": "0.24.0",
+  "version": "0.24.2",
   "type": "module",
   "exports": {
     ".": {

package/dist/chunk-AASMR3CQ.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../transport/transforms/messageFraming.ts","../transport/impls/uds/connection.ts"],"sourcesContent":["import { Transform, TransformCallback, TransformOptions } from 'node:stream';\n\nexport interface LengthEncodedOptions extends TransformOptions {\n  /** Maximum in-memory buffer size before we throw */\n  maxBufferSizeBytes: number;\n}\n\n/**\n * A transform stream that emits data each time a message with a network/BigEndian uint32 length prefix is received.\n * @extends Transform\n */\nexport class Uint32LengthPrefixFraming extends Transform {\n  receivedBuffer: Buffer;\n  maxBufferSizeBytes: number;\n\n  constructor({ maxBufferSizeBytes, ...options }: LengthEncodedOptions) {\n    super(options);\n    this.maxBufferSizeBytes = maxBufferSizeBytes;\n    this.receivedBuffer = Buffer.alloc(0);\n  }\n\n  _transform(chunk: Buffer, _encoding: BufferEncoding, cb: TransformCallback) {\n    if (\n      this.receivedBuffer.byteLength + chunk.byteLength >\n      this.maxBufferSizeBytes\n    ) {\n      const err = new Error(\n        `buffer overflow: ${this.receivedBuffer.byteLength}B > ${this.maxBufferSizeBytes}B`,\n      );\n\n      this.emit('error', err);\n      cb(err);\n      return;\n    }\n\n    this.receivedBuffer = Buffer.concat([this.receivedBuffer, chunk]);\n\n    // ensure there's enough for a length prefix\n    while (this.receivedBuffer.length > 4) {\n      // read length from buffer (accounting for uint32 prefix)\n      const claimedMessageLength = this.receivedBuffer.readUInt32BE(0) + 4;\n      if (this.receivedBuffer.length >= claimedMessageLength) {\n        // slice the buffer to extract the message\n        const message = this.receivedBuffer.subarray(4, claimedMessageLength);\n        this.push(message);\n        this.receivedBuffer =\n          this.receivedBuffer.subarray(claimedMessageLength);\n      } else {\n        // not enough data for a complete message, wait for more data\n        break;\n      }\n    }\n\n    cb();\n  }\n\n  _flush(cb: TransformCallback) {\n    // if there's any leftover data that doesn't form a complete message\n    if (this.receivedBuffer.length) {\n      this.emit('error', new Error('got incomplete message while flushing'));\n    }\n\n    this.receivedBuffer = Buffer.alloc(0);\n    cb();\n  }\n\n  _destroy(error: Error | null, callback: (error: Error | null) => void): void {\n    this.receivedBuffer = Buffer.alloc(0);\n    super._destroy(error, callback);\n  }\n}\n\nfunction createLengthEncodedStream(options?: Partial<LengthEncodedOptions>) {\n  return new Uint32LengthPrefixFraming({\n    maxBufferSizeBytes: options?.maxBufferSizeBytes ?? 16 * 1024 * 1024, // 16MB\n  });\n}\n\nexport const MessageFramer = {\n  createFramedStream: createLengthEncodedStream,\n  write: (buf: Uint8Array) => {\n    const lengthPrefix = Buffer.alloc(4);\n    lengthPrefix.writeUInt32BE(buf.length, 0);\n    return Buffer.concat([lengthPrefix, buf]);\n  },\n};\n","import { type Socket } from 'node:net';\nimport stream from 'node:stream';\nimport {\n  MessageFramer,\n  Uint32LengthPrefixFraming,\n} from '../../transforms/messageFraming';\nimport { Connection } from '../../connection';\n\nexport class UdsConnection extends Connection {\n  sock: Socket;\n  input: stream.Readable;\n  framer: Uint32LengthPrefixFraming;\n  constructor(sock: Socket) {\n    super();\n    this.framer = MessageFramer.createFramedStream();\n    this.sock = sock;\n    this.input = sock.pipe(this.framer);\n\n    this.sock.on('close', () => {\n      for (const cb of this.closeListeners) {\n        cb();\n      }\n    });\n\n    this.sock.on('error', (err) => {\n      if (err instanceof Error && 'code' in err && err.code === 'EPIPE') {\n        // Ignore EPIPE errors\n        return;\n      }\n\n      for (const cb of this.errorListeners) {\n        cb(err);\n      }\n    });\n\n    this.input.on('data', (msg: Uint8Array) => {\n      for (const cb of this.dataListeners) {\n        cb(msg);\n      }\n    });\n\n    this.sock.on('end', () => {\n      this.sock.destroy();\n    });\n  }\n\n  send(payload: Uint8Array) {\n    if (this.framer.destroyed || !this.sock.writable || this.sock.closed) {\n      return false;\n    }\n    this.sock.write(MessageFramer.write(payload));\n    return true;\n  }\n\n  close() {\n    this.sock.end();\n    this.framer.end();\n  }\n}\n"],"mappings":";;;;;AAAA,SAAS,iBAAsD;AAWxD,IAAM,4BAAN,cAAwC,UAAU;AAAA,EACvD;AAAA,EACA;AAAA,EAEA,YAAY,EAAE,oBAAoB,GAAG,QAAQ,GAAyB;AACpE,UAAM,OAAO;AACb,SAAK,qBAAqB;AAC1B,SAAK,iBAAiB,OAAO,MAAM,CAAC;AAAA,EACtC;AAAA,EAEA,WAAW,OAAe,WAA2B,IAAuB;AAC1E,QACE,KAAK,eAAe,aAAa,MAAM,aACvC,KAAK,oBACL;AACA,YAAM,MAAM,IAAI;AAAA,QACd,oBAAoB,KAAK,eAAe,UAAU,OAAO,KAAK,kBAAkB;AAAA,MAClF;AAEA,WAAK,KAAK,SAAS,GAAG;AACtB,SAAG,GAAG;AACN;AAAA,IACF;AAEA,SAAK,iBAAiB,OAAO,OAAO,CAAC,KAAK,gBAAgB,KAAK,CAAC;AAGhE,WAAO,KAAK,eAAe,SAAS,GAAG;AAErC,YAAM,uBAAuB,KAAK,eAAe,aAAa,CAAC,IAAI;AACnE,UAAI,KAAK,eAAe,UAAU,sBAAsB;AAEtD,cAAM,UAAU,KAAK,eAAe,SAAS,GAAG,oBAAoB;AACpE,aAAK,KAAK,OAAO;AACjB,aAAK,iBACH,KAAK,eAAe,SAAS,oBAAoB;AAAA,MACrD,OAAO;AAEL;AAAA,MACF;AAAA,IACF;AAEA,OAAG;AAAA,EACL;AAAA,EAEA,OAAO,IAAuB;AAE5B,QAAI,KAAK,eAAe,QAAQ;AAC9B,WAAK,KAAK,SAAS,IAAI,MAAM,uCAAuC,CAAC;AAAA,IACvE;AAEA,SAAK,iBAAiB,OAAO,MAAM,CAAC;AACpC,OAAG;AAAA,EACL;AAAA,EAEA,SAAS,OAAqB,UAA+C;AAC3E,SAAK,iBAAiB,OAAO,MAAM,CAAC;AACpC,UAAM,SAAS,OAAO,QAAQ;AAAA,EAChC;AACF;AAEA,SAAS,0BAA0B,SAAyC;AAC1E,SAAO,IAAI,0BAA0B;AAAA,IACnC,oBAAoB,SAAS,sBAAsB,KAAK,OAAO;AAAA;AAAA,EACjE,CAAC;AACH;AAEO,IAAM,gBAAgB;AAAA,EAC3B,oBAAoB;AAAA,EACpB,OAAO,CAAC,QAAoB;AAC1B,UAAM,eAAe,OAAO,MAAM,CAAC;AACnC,iBAAa,cAAc,IAAI,QAAQ,CAAC;AACxC,WAAO,OAAO,OAAO,CAAC,cAAc,GAAG,CAAC;AAAA,EAC1C;AACF;;;AC7EO,IAAM,gBAAN,cAA4B,WAAW;AAAA,EAC5C;AAAA,EACA;AAAA,EACA;AAAA,EACA,YAAY,MAAc;AACxB,UAAM;AACN,SAAK,SAAS,cAAc,mBAAmB;AAC/C,SAAK,OAAO;AACZ,SAAK,QAAQ,KAAK,KAAK,KAAK,MAAM;AAElC,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,UAAI,eAAe,SAAS,UAAU,OAAO,IAAI,SAAS,SAAS;AAEjE;AAAA,MACF;AAEA,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAED,SAAK,MAAM,GAAG,QAAQ,CAAC,QAAoB;AACzC,iBAAW,MAAM,KAAK,eAAe;AACnC,WAAG,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,OAAO,MAAM;AACxB,WAAK,KAAK,QAAQ;AAAA,IACpB,CAAC;AAAA,EACH;AAAA,EAEA,KAAK,SAAqB;AACxB,QAAI,KAAK,OAAO,aAAa,CAAC,KAAK,KAAK,YAAY,KAAK,KAAK,QAAQ;AACpE,aAAO;AAAA,IACT;AACA,SAAK,KAAK,MAAM,cAAc,MAAM,OAAO,CAAC;AAC5C,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ;AACN,SAAK,KAAK,IAAI;AACd,SAAK,OAAO,IAAI;AAAA,EAClB;AACF;","names":[]}