@replit/river 0.200.0-rc.9 → 0.200.2

package/README.md

@@ -65,11 +65,11 @@ Before proceeding, ensure you have TypeScript 5 installed and configured appropr
 
 - Router: a collection of services, namespaced by service name.
 - Service: a collection of procedures with a shared state.
-- Procedure: a single procedure. A procedure declares its type, an input message type, an output message type, optionally an error type, and the associated handler. Valid types are:
-  - `rpc` whose handler has a signature of `Input -> Result<Output, Error>`.
-  - `upload` whose handler has a signature of `AsyncIterableIterator<Input> -> Result<Output, Error>`.
-  - `subscription` whose handler has a signature of `Input -> Pushable<Result<Output, Error>>`.
-  - `stream` whose handler has a signature of `AsyncIterableIterator<Input> -> Pushable<Result<Output, Error>>`.
+- Procedure: a single procedure. A procedure declares its type, a request data type, a response data type, optionally a response error type, and the associated handler. Valid types are:
+  - `rpc`, single request, single response
+  - `upload`, multiple requests, single response
+  - `subscription`, single request, multiple responses
+  - `stream`, multiple requests, multiple response
 - Transport: manages the lifecycle (creation/deletion) of connections and multiplexing read/writes from clients. Both the client and the server must be passed in a subclass of `Transport` to work.
   - Connection: the actual raw underlying transport connection
   - Session: a higher-level abstraction that operates over the span of potentially multiple transport-level connections
@@ -92,11 +92,11 @@ export const ExampleService = ServiceSchema.define(
   // procedures
   {
     add: Procedure.rpc({
-      requestData: Type.Object({ n: Type.Number() }),
+      requestInit: Type.Object({ n: Type.Number() }),
       responseData: Type.Object({ result: Type.Number() }),
-      errors: Type.Never(),
+      requestErrors: Type.Never(),
       // note that a handler is unique per user RPC
-      async handler(ctx, { n }) {
+      async handler({ ctx, reqInit: { n } }) {
         // access and mutate shared state
         ctx.state.count += n;
         return Ok({ result: ctx.state.count });

package/dist/{chunk-42Z2FQIU.js → chunk-6BH2CXVE.js}

@@ -3,10 +3,10 @@ import {
   createSessionTelemetryInfo,
   generateId,
   isAck
-} from "./chunk-EETL2L77.js";
+} from "./chunk-GJUUVID2.js";
 import {
   NaiveJsonCodec
-} from "./chunk-4PVU7J25.js";
+} from "./chunk-AJGIY2UB.js";
 
 // transport/options.ts
 var defaultTransportOptions = {
@@ -414,13 +414,13 @@ var SessionConnected = class extends IdentifiedSession {
     this.conn.addCloseListener(this.listeners.onConnectionClosed);
     this.conn.addErrorListener(this.listeners.onConnectionErrored);
     if (this.sendBuffer.length > 0) {
-      this.log?.debug(
-        `sending ${this.sendBuffer.length} buffered messages`,
+      this.log?.info(
+        `sending ${this.sendBuffer.length} buffered messages, starting at seq ${this.nextSeq()}`,
         this.loggingMetadata
       );
-    }
-    for (const msg of this.sendBuffer) {
-      this.conn.send(this.options.codec.toBuffer(msg));
+      for (const msg of this.sendBuffer) {
+        this.conn.send(this.options.codec.toBuffer(msg));
+      }
     }
     this.isActivelyHeartbeating = false;
     this.heartbeatHandle = setInterval(() => {
@@ -462,6 +462,12 @@ var SessionConnected = class extends IdentifiedSession {
       }
     });
   }
+  closeConnection() {
+    this.conn.removeDataListener(this.onMessageData);
+    this.conn.removeCloseListener(this.listeners.onConnectionClosed);
+    this.conn.removeErrorListener(this.listeners.onConnectionErrored);
+    this.conn.close();
+  }
   onMessageData = (msg) => {
     const parsedMsg = this.parseMsg(msg);
     if (parsedMsg === null) {
@@ -478,8 +484,8 @@ var SessionConnected = class extends IdentifiedSession {
           }
         );
       } else {
-        const reason = `received out-of-order msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;
-        this.log?.error(reason, {
+        const reason = `received out-of-order msg, closing connection (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;
+        this.log?.warn(reason, {
           ...this.loggingMetadata,
           transportMessage: parsedMsg,
           tags: ["invariant-violation"]
@@ -488,7 +494,7 @@ var SessionConnected = class extends IdentifiedSession {
           code: SpanStatusCode.ERROR,
           message: reason
         });
-        this.listeners.onInvalidMessage(reason);
+        this.closeConnection();
       }
       return;
     }
@@ -514,8 +520,10 @@ var SessionConnected = class extends IdentifiedSession {
     this.conn.removeDataListener(this.onMessageData);
     this.conn.removeCloseListener(this.listeners.onConnectionClosed);
     this.conn.removeErrorListener(this.listeners.onConnectionErrored);
-    clearInterval(this.heartbeatHandle);
-    this.heartbeatHandle = void 0;
+    if (this.heartbeatHandle) {
+      clearInterval(this.heartbeatHandle);
+      this.heartbeatHandle = void 0;
+    }
   }
   _handleClose() {
     super._handleClose();
@@ -849,4 +857,4 @@ export {
   ClientSessionStateGraph,
   ServerSessionStateGraph
 };
-//# sourceMappingURL=chunk-42Z2FQIU.js.map
+//# sourceMappingURL=chunk-6BH2CXVE.js.map

package/dist/chunk-6BH2CXVE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/options.ts","../transport/sessionStateMachine/common.ts","../transport/sessionStateMachine/SessionConnecting.ts","../transport/sessionStateMachine/SessionNoConnection.ts","../transport/sessionStateMachine/SessionWaitingForHandshake.ts","../transport/sessionStateMachine/SessionHandshaking.ts","../transport/sessionStateMachine/SessionConnected.ts","../transport/sessionStateMachine/SessionBackingOff.ts","../transport/sessionStateMachine/transitions.ts"],"sourcesContent":["import { NaiveJsonCodec } from '../codec/json';\nimport { ConnectionRetryOptions } from './rateLimit';\nimport { SessionOptions } from './sessionStateMachine/common';\n\nexport type TransportOptions = SessionOptions;\n\nexport type ProvidedTransportOptions = Partial<TransportOptions>;\n\nexport const defaultTransportOptions: TransportOptions = {\n  heartbeatIntervalMs: 1_000,\n  heartbeatsUntilDead: 2,\n  sessionDisconnectGraceMs: 5_000,\n  connectionTimeoutMs: 2_000,\n  handshakeTimeoutMs: 1_000,\n  enableTransparentSessionReconnects: true,\n  codec: NaiveJsonCodec,\n};\n\nexport type ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n  baseIntervalMs: 150,\n  maxJitterMs: 200,\n  maxBackoffMs: 32_000,\n  attemptBudgetCapacity: 5,\n  budgetRestoreIntervalMs: 200,\n};\n\nexport const defaultClientTransportOptions: ClientTransportOptions = {\n  ...defaultTransportOptions,\n  ...defaultConnectionRetryOptions,\n};\n\nexport type ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nexport const defaultServerTransportOptions: ServerTransportOptions = {\n  ...defaultTransportOptions,\n};\n","import { Logger, MessageMetadata } from '../../logging';\nimport { TelemetryInfo } from '../../tracing';\nimport {\n  OpaqueTransportMessage,\n  OpaqueTransportMessageSchema,\n  PartialTransportMessage,\n  ProtocolVersion,\n  TransportClientId,\n  TransportMessage,\n} from '../message';\nimport { Value } from '@sinclair/typebox/value';\nimport { Codec } from '../../codec';\nimport { generateId } from '../id';\n\nexport const enum SessionState {\n  NoConnection = 'NoConnection',\n  BackingOff = 'BackingOff',\n  Connecting = 'Connecting',\n  Handshaking = 'Handshaking',\n  Connected = 'Connected',\n  WaitingForHandshake = 'WaitingForHandshake',\n}\n\nexport const ERR_CONSUMED = `session state has been consumed and is no longer valid`;\n\nabstract class StateMachineState {\n  abstract readonly state: SessionState;\n\n  /*\n   * Whether this state has been consumed\n   * and we've moved on to another state\n   */\n  _isConsumed: boolean;\n\n  // called when we're transitioning to another state\n  // note that this is internal and should not be called directly\n  // by consumers, the proxy will call this when the state is consumed\n  // and we're transitioning to another state\n  abstract _handleStateExit(): void;\n\n  // called when we exit the state machine entirely\n  // note that this is internal and should not be called directly\n  // by consumers, the proxy will call this when .close is closed\n  abstract _handleClose(): void;\n\n  close(): void {\n    this._handleClose();\n  }\n\n  constructor() {\n    this._isConsumed = false;\n\n    // proxy helps us prevent access to properties after the state has been consumed\n    // e.g. if we hold a reference to a state and try to access it after it's been consumed\n    // we intercept the access and throw an error to help catch bugs\n    return new Proxy(this, {\n      get(target, prop) {\n        // always allow access to _isConsumed, id, and state\n        if (prop === '_isConsumed' || prop === 'id' || prop === 'state') {\n          return Reflect.get(target, prop);\n        }\n\n        // modify _handleStateExit\n        if (prop === '_handleStateExit') {\n          return () => {\n            target._isConsumed = true;\n            target._handleStateExit();\n          };\n        }\n\n        // modify _handleClose\n        if (prop === '_handleClose') {\n          return () => {\n            // target is the non-proxied object, we need to set _isConsumed again\n            target._isConsumed = true;\n            target._handleStateExit();\n            target._handleClose();\n          };\n        }\n\n        if (target._isConsumed) {\n          throw new Error(\n            `${ERR_CONSUMED}: getting ${prop.toString()} on consumed state`,\n          );\n        }\n\n        return Reflect.get(target, prop);\n      },\n      set(target, prop, value) {\n        if (target._isConsumed) {\n          throw new Error(\n            `${ERR_CONSUMED}: setting ${prop.toString()} on consumed state`,\n          );\n        }\n\n        return Reflect.set(target, prop, value);\n      },\n    });\n  }\n}\n\nexport interface SessionOptions {\n  /**\n   * Frequency at which to send heartbeat acknowledgements\n   */\n  heartbeatIntervalMs: number;\n  /**\n   * Number of elapsed heartbeats without a response message before we consider\n   * the connection dead.\n   */\n  heartbeatsUntilDead: number;\n  /**\n   * Max duration that a session can be without a connection before we consider\n   * it dead. This deadline is carried between states and is used to determine\n   * when to consider the session a lost cause and delete it entirely.\n   * Generally, this should be strictly greater than the sum of\n   * {@link connectionTimeoutMs} and {@link handshakeTimeoutMs}.\n   */\n  sessionDisconnectGraceMs: number;\n  /**\n   * Connection timeout in milliseconds\n   */\n  connectionTimeoutMs: number;\n  /**\n   * Handshake timeout in milliseconds\n   */\n  handshakeTimeoutMs: number;\n  /**\n   * Whether to enable transparent session reconnects\n   */\n  enableTransparentSessionReconnects: boolean;\n  /**\n   * The codec to use for encoding/decoding messages over the wire\n   */\n  codec: Codec;\n}\n\n// all session states have a from and options\nexport interface CommonSessionProps {\n  from: TransportClientId;\n  options: SessionOptions;\n  log: Logger | undefined;\n}\n\nexport abstract class CommonSession extends StateMachineState {\n  readonly from: TransportClientId;\n  readonly options: SessionOptions;\n\n  log?: Logger;\n  abstract get loggingMetadata(): MessageMetadata;\n\n  constructor({ from, options, log }: CommonSessionProps) {\n    super();\n    this.from = from;\n    this.options = options;\n    this.log = log;\n  }\n\n  parseMsg(msg: Uint8Array): OpaqueTransportMessage | null {\n    const parsedMsg = this.options.codec.fromBuffer(msg);\n\n    if (parsedMsg === null) {\n      const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));\n      this.log?.error(\n        `received malformed msg: ${decodedBuffer}`,\n        this.loggingMetadata,\n      );\n\n      return null;\n    }\n\n    if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n      this.log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n        ...this.loggingMetadata,\n        validationErrors: [\n          ...Value.Errors(OpaqueTransportMessageSchema, parsedMsg),\n        ],\n      });\n\n      return null;\n    }\n\n    return parsedMsg;\n  }\n}\n\nexport type InheritedProperties = Pick<\n  IdentifiedSession,\n  'id' | 'from' | 'to' | 'seq' | 'ack' | 'sendBuffer' | 'telemetry' | 'options'\n>;\n\nexport type SessionId = string;\n\n// all sessions where we know the other side's client id\nexport interface IdentifiedSessionProps extends CommonSessionProps {\n  id: SessionId;\n  to: TransportClientId;\n  seq: number;\n  ack: number;\n  sendBuffer: Array<OpaqueTransportMessage>;\n  telemetry: TelemetryInfo;\n  protocolVersion: ProtocolVersion;\n}\n\nexport abstract class IdentifiedSession extends CommonSession {\n  readonly id: SessionId;\n  readonly telemetry: TelemetryInfo;\n  readonly to: TransportClientId;\n  readonly protocolVersion: ProtocolVersion;\n\n  /**\n   * Index of the message we will send next (excluding handshake)\n   */\n  seq: number;\n\n  /**\n   * Number of unique messages we've received this session (excluding handshake)\n   */\n  ack: number;\n  sendBuffer: Array<OpaqueTransportMessage>;\n\n  constructor(props: IdentifiedSessionProps) {\n    const { id, to, seq, ack, sendBuffer, telemetry, log, protocolVersion } =\n      props;\n    super(props);\n    this.id = id;\n    this.to = to;\n    this.seq = seq;\n    this.ack = ack;\n    this.sendBuffer = sendBuffer;\n    this.telemetry = telemetry;\n    this.log = log;\n    this.protocolVersion = protocolVersion;\n  }\n\n  get loggingMetadata(): MessageMetadata {\n    const spanContext = this.telemetry.span.spanContext();\n\n    const metadata: MessageMetadata = {\n      clientId: this.from,\n      connectedTo: this.to,\n      sessionId: this.id,\n    };\n\n    if (this.telemetry.span.isRecording()) {\n      metadata.telemetry = {\n        traceId: spanContext.traceId,\n        spanId: spanContext.spanId,\n      };\n    }\n\n    return metadata;\n  }\n\n  constructMsg<Payload>(\n    partialMsg: PartialTransportMessage<Payload>,\n  ): TransportMessage<Payload> {\n    const msg = {\n      ...partialMsg,\n      id: generateId(),\n      to: this.to,\n      from: this.from,\n      seq: this.seq,\n      ack: this.ack,\n    };\n\n    this.seq++;\n\n    return msg;\n  }\n\n  nextSeq(): number {\n    return this.sendBuffer.length > 0 ? this.sendBuffer[0].seq : this.seq;\n  }\n\n  send(msg: PartialTransportMessage): string {\n    const constructedMsg = this.constructMsg(msg);\n    this.sendBuffer.push(constructedMsg);\n\n    return constructedMsg.id;\n  }\n\n  _handleStateExit(): void {\n    // noop\n  }\n\n  _handleClose(): void {\n    // zero out the buffer\n    this.sendBuffer.length = 0;\n    this.telemetry.span.end();\n  }\n}\n\nexport interface IdentifiedSessionWithGracePeriodListeners {\n  onSessionGracePeriodElapsed: () => void;\n}\n\nexport interface IdentifiedSessionWithGracePeriodProps\n  extends IdentifiedSessionProps {\n  graceExpiryTime: number;\n  listeners: IdentifiedSessionWithGracePeriodListeners;\n}\n\nexport abstract class IdentifiedSessionWithGracePeriod extends IdentifiedSession {\n  graceExpiryTime: number;\n  protected gracePeriodTimeout?: ReturnType<typeof setTimeout>;\n\n  listeners: IdentifiedSessionWithGracePeriodListeners;\n\n  constructor(props: IdentifiedSessionWithGracePeriodProps) {\n    super(props);\n    this.listeners = props.listeners;\n\n    this.graceExpiryTime = props.graceExpiryTime;\n    this.gracePeriodTimeout = setTimeout(() => {\n      this.listeners.onSessionGracePeriodElapsed();\n    }, this.graceExpiryTime - Date.now());\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n\n    if (this.gracePeriodTimeout) {\n      clearTimeout(this.gracePeriodTimeout);\n      this.gracePeriodTimeout = undefined;\n    }\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n  }\n}\n","import { Connection } from '../connection';\nimport {\n  IdentifiedSessionWithGracePeriod,\n  IdentifiedSessionWithGracePeriodListeners,\n  IdentifiedSessionWithGracePeriodProps,\n  SessionState,\n} from './common';\n\nexport interface SessionConnectingListeners\n  extends IdentifiedSessionWithGracePeriodListeners {\n  onConnectionEstablished: (conn: Connection) => void;\n  onConnectionFailed: (err: unknown) => void;\n\n  // timeout related\n  onConnectionTimeout: () => void;\n}\n\nexport interface SessionConnectingProps<ConnType extends Connection>\n  extends IdentifiedSessionWithGracePeriodProps {\n  connPromise: Promise<ConnType>;\n  listeners: SessionConnectingListeners;\n}\n\n/*\n * A session that is connecting but we don't have access to the raw connection yet.\n * See transitions.ts for valid transitions.\n */\nexport class SessionConnecting<\n  ConnType extends Connection,\n> extends IdentifiedSessionWithGracePeriod {\n  readonly state = SessionState.Connecting as const;\n  connPromise: Promise<ConnType>;\n  listeners: SessionConnectingListeners;\n\n  connectionTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(props: SessionConnectingProps<ConnType>) {\n    super(props);\n    this.connPromise = props.connPromise;\n    this.listeners = props.listeners;\n\n    this.connPromise.then(\n      (conn) => {\n        if (this._isConsumed) return;\n        this.listeners.onConnectionEstablished(conn);\n      },\n      (err) => {\n        if (this._isConsumed) return;\n        this.listeners.onConnectionFailed(err);\n      },\n    );\n\n    this.connectionTimeout = setTimeout(() => {\n      this.listeners.onConnectionTimeout();\n    }, this.options.connectionTimeoutMs);\n  }\n\n  // close a pending connection if it resolves, ignore errors if the promise\n  // ends up rejected anyways\n  bestEffortClose() {\n    // these can technically be stale if the connPromise resolves after the\n    // state has transitioned, but that's fine, this is best effort anyways\n    // we pull these out so even if the state has transitioned, we can still log\n    // without erroring out\n    const logger = this.log;\n    const metadata = this.loggingMetadata;\n\n    this.connPromise\n      .then((conn) => {\n        conn.close();\n        logger?.info(\n          'connection eventually resolved but session has transitioned, closed connection',\n          {\n            ...metadata,\n            ...conn.loggingMetadata,\n          },\n        );\n      })\n      .catch(() => {\n        // ignore errors\n      });\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n\n    if (this.connectionTimeout) {\n      clearTimeout(this.connectionTimeout);\n      this.connectionTimeout = undefined;\n    }\n  }\n\n  _handleClose(): void {\n    // close the pending connection if it resolves\n    this.bestEffortClose();\n    super._handleClose();\n  }\n}\n","import {\n  IdentifiedSessionWithGracePeriod,\n  IdentifiedSessionWithGracePeriodListeners,\n  IdentifiedSessionWithGracePeriodProps,\n  SessionState,\n} from './common';\n\nexport type SessionNoConnectionListeners =\n  IdentifiedSessionWithGracePeriodListeners;\n\nexport type SessionNoConnectionProps = IdentifiedSessionWithGracePeriodProps;\n\n/*\n * A session that is not connected and cannot send or receive messages.\n * See transitions.ts for valid transitions.\n */\nexport class SessionNoConnection extends IdentifiedSessionWithGracePeriod {\n  readonly state = SessionState.NoConnection as const;\n\n  _handleClose(): void {\n    super._handleClose();\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n  }\n}\n","import { Static } from '@sinclair/typebox';\nimport { Connection } from '../connection';\nimport {\n  HandshakeErrorResponseCodes,\n  OpaqueTransportMessage,\n  TransportMessage,\n} from '../message';\nimport { CommonSession, CommonSessionProps, SessionState } from './common';\n\nexport interface SessionWaitingForHandshakeListeners {\n  onConnectionErrored: (err: unknown) => void;\n  onConnectionClosed: () => void;\n  onHandshake: (msg: OpaqueTransportMessage) => void;\n  onInvalidHandshake: (\n    reason: string,\n    code: Static<typeof HandshakeErrorResponseCodes>,\n  ) => void;\n\n  // timeout related\n  onHandshakeTimeout: () => void;\n}\n\nexport interface SessionWaitingForHandshakeProps<ConnType extends Connection>\n  extends CommonSessionProps {\n  conn: ConnType;\n  listeners: SessionWaitingForHandshakeListeners;\n}\n\n/*\n * Server-side session that has a connection but is waiting for the client to identify itself.\n * See transitions.ts for valid transitions.\n */\nexport class SessionWaitingForHandshake<\n  ConnType extends Connection,\n> extends CommonSession {\n  readonly state = SessionState.WaitingForHandshake as const;\n  conn: ConnType;\n  listeners: SessionWaitingForHandshakeListeners;\n\n  handshakeTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(props: SessionWaitingForHandshakeProps<ConnType>) {\n    super(props);\n    this.conn = props.conn;\n    this.listeners = props.listeners;\n\n    this.handshakeTimeout = setTimeout(() => {\n      this.listeners.onHandshakeTimeout();\n    }, this.options.handshakeTimeoutMs);\n\n    this.conn.addDataListener(this.onHandshakeData);\n    this.conn.addErrorListener(this.listeners.onConnectionErrored);\n    this.conn.addCloseListener(this.listeners.onConnectionClosed);\n  }\n\n  get loggingMetadata() {\n    return {\n      clientId: this.from,\n      connId: this.conn.id,\n      ...this.conn.loggingMetadata,\n    };\n  }\n\n  onHandshakeData = (msg: Uint8Array) => {\n    const parsedMsg = this.parseMsg(msg);\n    if (parsedMsg === null) {\n      this.listeners.onInvalidHandshake(\n        'could not parse message',\n        'MALFORMED_HANDSHAKE',\n      );\n\n      return;\n    }\n\n    // after this fires, the listener is responsible for transitioning the session\n    // and thus removing the handshake timeout\n    this.listeners.onHandshake(parsedMsg);\n  };\n\n  sendHandshake(msg: TransportMessage): boolean {\n    return this.conn.send(this.options.codec.toBuffer(msg));\n  }\n\n  _handleStateExit(): void {\n    this.conn.removeDataListener(this.onHandshakeData);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n    clearTimeout(this.handshakeTimeout);\n    this.handshakeTimeout = undefined;\n  }\n\n  _handleClose(): void {\n    this.conn.close();\n  }\n}\n","import { Static } from '@sinclair/typebox';\nimport { Connection } from '../connection';\nimport {\n  OpaqueTransportMessage,\n  TransportMessage,\n  HandshakeErrorResponseCodes,\n} from '../message';\nimport {\n  IdentifiedSessionWithGracePeriod,\n  IdentifiedSessionWithGracePeriodListeners,\n  IdentifiedSessionWithGracePeriodProps,\n  SessionState,\n} from './common';\n\nexport interface SessionHandshakingListeners\n  extends IdentifiedSessionWithGracePeriodListeners {\n  onConnectionErrored: (err: unknown) => void;\n  onConnectionClosed: () => void;\n  onHandshake: (msg: OpaqueTransportMessage) => void;\n  onInvalidHandshake: (\n    reason: string,\n    code: Static<typeof HandshakeErrorResponseCodes>,\n  ) => void;\n\n  // timeout related\n  onHandshakeTimeout: () => void;\n}\n\nexport interface SessionHandshakingProps<ConnType extends Connection>\n  extends IdentifiedSessionWithGracePeriodProps {\n  conn: ConnType;\n  listeners: SessionHandshakingListeners;\n}\n\n/*\n * A session that is handshaking and waiting for the other side to identify itself.\n * See transitions.ts for valid transitions.\n */\nexport class SessionHandshaking<\n  ConnType extends Connection,\n> extends IdentifiedSessionWithGracePeriod {\n  readonly state = SessionState.Handshaking as const;\n  conn: ConnType;\n  listeners: SessionHandshakingListeners;\n\n  handshakeTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(props: SessionHandshakingProps<ConnType>) {\n    super(props);\n    this.conn = props.conn;\n    this.listeners = props.listeners;\n\n    this.handshakeTimeout = setTimeout(() => {\n      this.listeners.onHandshakeTimeout();\n    }, this.options.handshakeTimeoutMs);\n\n    this.conn.addDataListener(this.onHandshakeData);\n    this.conn.addErrorListener(this.listeners.onConnectionErrored);\n    this.conn.addCloseListener(this.listeners.onConnectionClosed);\n  }\n\n  get loggingMetadata() {\n    return {\n      ...super.loggingMetadata,\n      ...this.conn.loggingMetadata,\n    };\n  }\n\n  onHandshakeData = (msg: Uint8Array) => {\n    const parsedMsg = this.parseMsg(msg);\n    if (parsedMsg === null) {\n      this.listeners.onInvalidHandshake(\n        'could not parse message',\n        'MALFORMED_HANDSHAKE',\n      );\n\n      return;\n    }\n\n    this.listeners.onHandshake(parsedMsg);\n  };\n\n  sendHandshake(msg: TransportMessage): boolean {\n    return this.conn.send(this.options.codec.toBuffer(msg));\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n    this.conn.removeDataListener(this.onHandshakeData);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n\n    if (this.handshakeTimeout) {\n      clearTimeout(this.handshakeTimeout);\n      this.handshakeTimeout = undefined;\n    }\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n    this.conn.close();\n  }\n}\n","import { Static } from '@sinclair/typebox';\nimport {\n  ControlFlags,\n  ControlMessageAckSchema,\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  isAck,\n} from '../message';\nimport {\n  IdentifiedSession,\n  IdentifiedSessionProps,\n  SessionState,\n} from './common';\nimport { Connection } from '../connection';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nexport interface SessionConnectedListeners {\n  onConnectionErrored: (err: unknown) => void;\n  onConnectionClosed: () => void;\n  onMessage: (msg: OpaqueTransportMessage) => void;\n  onInvalidMessage: (reason: string) => void;\n}\n\nexport interface SessionConnectedProps<ConnType extends Connection>\n  extends IdentifiedSessionProps {\n  conn: ConnType;\n  listeners: SessionConnectedListeners;\n}\n\n/*\n * A session that is connected and can send and receive messages.\n * See transitions.ts for valid transitions.\n */\nexport class SessionConnected<\n  ConnType extends Connection,\n> extends IdentifiedSession {\n  readonly state = SessionState.Connected as const;\n  conn: ConnType;\n  listeners: SessionConnectedListeners;\n\n  private heartbeatHandle?: ReturnType<typeof setInterval> | undefined;\n  private heartbeatMisses = 0;\n  isActivelyHeartbeating: boolean;\n\n  updateBookkeeping(ack: number, seq: number) {\n    this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n    this.ack = seq + 1;\n    this.heartbeatMisses = 0;\n  }\n\n  send(msg: PartialTransportMessage): string {\n    const constructedMsg = this.constructMsg(msg);\n    this.sendBuffer.push(constructedMsg);\n    this.conn.send(this.options.codec.toBuffer(constructedMsg));\n\n    return constructedMsg.id;\n  }\n\n  constructor(props: SessionConnectedProps<ConnType>) {\n    super(props);\n    this.conn = props.conn;\n    this.listeners = props.listeners;\n\n    this.conn.addDataListener(this.onMessageData);\n    this.conn.addCloseListener(this.listeners.onConnectionClosed);\n    this.conn.addErrorListener(this.listeners.onConnectionErrored);\n\n    // send any buffered messages\n    if (this.sendBuffer.length > 0) {\n      this.log?.info(\n        `sending ${\n          this.sendBuffer.length\n        } buffered messages, starting at seq ${this.nextSeq()}`,\n        this.loggingMetadata,\n      );\n\n      for (const msg of this.sendBuffer) {\n        this.conn.send(this.options.codec.toBuffer(msg));\n      }\n    }\n\n    // dont explicity clear the buffer, we'll just filter out old messages\n    // when we receive an ack\n\n    // setup heartbeat\n    this.isActivelyHeartbeating = false;\n    this.heartbeatHandle = setInterval(() => {\n      const misses = this.heartbeatMisses;\n      const missDuration = misses * this.options.heartbeatIntervalMs;\n      if (misses >= this.options.heartbeatsUntilDead) {\n        this.log?.info(\n          `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,\n          this.loggingMetadata,\n        );\n        this.telemetry.span.addEvent('closing connection due to inactivity');\n\n        // it is OK to close this even on the client when we can't trust the client timer\n        // due to browser throttling or hibernation\n        // at worst, this interval will fire later than what the server expects and the server\n        // will have already closed the connection\n        // this just helps us in cases where we have a proxying setup where the server has closed\n        // the connection but the proxy hasn't synchronized the server-side close to the client so\n        // the client isn't stuck with a pseudo-dead connection forever\n        this.conn.close();\n        clearInterval(this.heartbeatHandle);\n        this.heartbeatHandle = undefined;\n\n        return;\n      }\n\n      if (this.isActivelyHeartbeating) {\n        this.sendHeartbeat();\n      }\n\n      this.heartbeatMisses++;\n    }, this.options.heartbeatIntervalMs);\n  }\n\n  get loggingMetadata() {\n    return {\n      ...super.loggingMetadata,\n      ...this.conn.loggingMetadata,\n    };\n  }\n\n  startActiveHeartbeat() {\n    this.isActivelyHeartbeating = true;\n  }\n\n  private sendHeartbeat() {\n    this.log?.debug('sending heartbeat', this.loggingMetadata);\n    this.send({\n      streamId: 'heartbeat',\n      controlFlags: ControlFlags.AckBit,\n      payload: {\n        type: 'ACK',\n      } satisfies Static<typeof ControlMessageAckSchema>,\n    });\n  }\n\n  closeConnection() {\n    this.conn.removeDataListener(this.onMessageData);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n    this.conn.close();\n  }\n\n  onMessageData = (msg: Uint8Array) => {\n    const parsedMsg = this.parseMsg(msg);\n    if (parsedMsg === null) {\n      this.listeners.onInvalidMessage('could not parse message');\n\n      return;\n    }\n\n    // check message ordering here\n    if (parsedMsg.seq !== this.ack) {\n      if (parsedMsg.seq < this.ack) {\n        this.log?.debug(\n          `received duplicate msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack}), discarding`,\n          {\n            ...this.loggingMetadata,\n            transportMessage: parsedMsg,\n          },\n        );\n      } else {\n        const reason = `received out-of-order msg, closing connection (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;\n        this.log?.warn(reason, {\n          ...this.loggingMetadata,\n          transportMessage: parsedMsg,\n          tags: ['invariant-violation'],\n        });\n\n        this.telemetry.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: reason,\n        });\n\n        // try to recover by closing the connection and re-handshaking\n        // with the session intact\n        this.closeConnection();\n      }\n\n      return;\n    }\n\n    // message is ok to update bookkeeping with\n    this.log?.debug(`received msg`, {\n      ...this.loggingMetadata,\n      transportMessage: parsedMsg,\n    });\n\n    this.updateBookkeeping(parsedMsg.ack, parsedMsg.seq);\n\n    // dispatch directly if its not an explicit ack\n    if (!isAck(parsedMsg.controlFlags)) {\n      this.listeners.onMessage(parsedMsg);\n\n      return;\n    }\n\n    // discard acks (unless we aren't heartbeating in which case just respond)\n    this.log?.debug(`discarding msg (ack bit set)`, {\n      ...this.loggingMetadata,\n      transportMessage: parsedMsg,\n    });\n\n    // if we are not actively heartbeating, we are in passive\n    // heartbeat mode and should send a response to the ack\n    if (!this.isActivelyHeartbeating) {\n      this.sendHeartbeat();\n    }\n  };\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n    this.conn.removeDataListener(this.onMessageData);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n\n    if (this.heartbeatHandle) {\n      clearInterval(this.heartbeatHandle);\n      this.heartbeatHandle = undefined;\n    }\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n    this.conn.close();\n  }\n}\n","import {\n  IdentifiedSessionWithGracePeriod,\n  IdentifiedSessionWithGracePeriodListeners,\n  IdentifiedSessionWithGracePeriodProps,\n  SessionState,\n} from './common';\n\nexport interface SessionBackingOffListeners\n  extends IdentifiedSessionWithGracePeriodListeners {\n  onBackoffFinished: () => void;\n}\n\nexport interface SessionBackingOffProps\n  extends IdentifiedSessionWithGracePeriodProps {\n  backoffMs: number;\n  listeners: SessionBackingOffListeners;\n}\n\n/*\n * A session that is backing off before attempting to connect.\n * See transitions.ts for valid transitions.\n */\nexport class SessionBackingOff extends IdentifiedSessionWithGracePeriod {\n  readonly state = SessionState.BackingOff as const;\n  listeners: SessionBackingOffListeners;\n\n  backoffTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(props: SessionBackingOffProps) {\n    super(props);\n    this.listeners = props.listeners;\n\n    this.backoffTimeout = setTimeout(() => {\n      this.listeners.onBackoffFinished();\n    }, props.backoffMs);\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n\n    if (this.backoffTimeout) {\n      clearTimeout(this.backoffTimeout);\n      this.backoffTimeout = undefined;\n    }\n  }\n}\n","import { OpaqueTransportMessage, TransportClientId } from '..';\nimport {\n  SessionConnecting,\n  SessionConnectingListeners,\n} from './SessionConnecting';\nimport {\n  SessionNoConnection,\n  SessionNoConnectionListeners,\n} from './SessionNoConnection';\nimport {\n  IdentifiedSession,\n  IdentifiedSessionProps,\n  IdentifiedSessionWithGracePeriod,\n  IdentifiedSessionWithGracePeriodProps,\n  SessionOptions,\n} from './common';\nimport { PropagationContext, createSessionTelemetryInfo } from '../../tracing';\nimport {\n  SessionWaitingForHandshake,\n  SessionWaitingForHandshakeListeners,\n} from './SessionWaitingForHandshake';\nimport {\n  SessionHandshaking,\n  SessionHandshakingListeners,\n} from './SessionHandshaking';\nimport {\n  SessionConnected,\n  SessionConnectedListeners,\n} from './SessionConnected';\nimport { generateId } from '../id';\nimport { Connection } from '../connection';\nimport { Logger } from '../../logging';\nimport {\n  SessionBackingOff,\n  SessionBackingOffListeners,\n} from './SessionBackingOff';\nimport { ProtocolVersion } from '../message';\n\nfunction inheritSharedSession(\n  session: IdentifiedSession,\n): IdentifiedSessionProps {\n  return {\n    id: session.id,\n    from: session.from,\n    to: session.to,\n    seq: session.seq,\n    ack: session.ack,\n    sendBuffer: session.sendBuffer,\n    telemetry: session.telemetry,\n    options: session.options,\n    log: session.log,\n    protocolVersion: session.protocolVersion,\n  };\n}\n\nfunction inheritSharedSessionWithGrace(\n  session: IdentifiedSessionWithGracePeriod,\n): Omit<IdentifiedSessionWithGracePeriodProps, 'listeners'> {\n  return {\n    ...inheritSharedSession(session),\n    graceExpiryTime: session.graceExpiryTime,\n  };\n}\n\nexport const SessionStateGraph = {\n  entrypoints: {\n    NoConnection: (\n      to: TransportClientId,\n      from: TransportClientId,\n      listeners: SessionNoConnectionListeners,\n      options: SessionOptions,\n      protocolVersion: ProtocolVersion,\n      log?: Logger,\n    ) => {\n      const id = `session-${generateId()}`;\n      const telemetry = createSessionTelemetryInfo(id, to, from);\n      const sendBuffer: Array<OpaqueTransportMessage> = [];\n\n      const session = new SessionNoConnection({\n        listeners,\n        id,\n        from,\n        to,\n        seq: 0,\n        ack: 0,\n        graceExpiryTime: Date.now() + options.sessionDisconnectGraceMs,\n        sendBuffer,\n        telemetry,\n        options,\n        protocolVersion,\n        log,\n      });\n\n      session.log?.info(`session ${session.id} created in NoConnection state`, {\n        ...session.loggingMetadata,\n        tags: ['state-transition'],\n      });\n\n      return session;\n    },\n    WaitingForHandshake: <ConnType extends Connection>(\n      from: TransportClientId,\n      conn: ConnType,\n      listeners: SessionWaitingForHandshakeListeners,\n      options: SessionOptions,\n      log?: Logger,\n    ): SessionWaitingForHandshake<ConnType> => {\n      const session = new SessionWaitingForHandshake({\n        conn,\n        listeners,\n        from,\n        options,\n        log,\n      });\n\n      session.log?.info(`session created in WaitingForHandshake state`, {\n        ...session.loggingMetadata,\n        tags: ['state-transition'],\n      });\n\n      return session;\n    },\n  },\n  // All of the transitions 'move'/'consume' the old session and return a new one.\n  // After a session is transitioned, any usage of the old session will throw.\n  transition: {\n    // happy path transitions\n    NoConnectionToBackingOff: (\n      oldSession: SessionNoConnection,\n      backoffMs: number,\n      listeners: SessionBackingOffListeners,\n    ): SessionBackingOff => {\n      const carriedState = inheritSharedSessionWithGrace(oldSession);\n      oldSession._handleStateExit();\n\n      const session = new SessionBackingOff({\n        backoffMs,\n        listeners,\n        ...carriedState,\n      });\n\n      session.log?.info(\n        `session ${session.id} transition from NoConnection to BackingOff`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    BackingOffToConnecting: <ConnType extends Connection>(\n      oldSession: SessionBackingOff,\n      connPromise: Promise<ConnType>,\n      listeners: SessionConnectingListeners,\n    ): SessionConnecting<ConnType> => {\n      const carriedState = inheritSharedSessionWithGrace(oldSession);\n      oldSession._handleStateExit();\n\n      const session = new SessionConnecting({\n        connPromise,\n        listeners,\n        ...carriedState,\n      });\n\n      session.log?.info(\n        `session ${session.id} transition from BackingOff to Connecting`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    ConnectingToHandshaking: <ConnType extends Connection>(\n      oldSession: SessionConnecting<ConnType>,\n      conn: ConnType,\n      listeners: SessionHandshakingListeners,\n    ): SessionHandshaking<ConnType> => {\n      const carriedState = inheritSharedSessionWithGrace(oldSession);\n      oldSession._handleStateExit();\n\n      const session = new SessionHandshaking({\n        conn,\n        listeners,\n        ...carriedState,\n      });\n\n      session.log?.info(\n        `session ${session.id} transition from Connecting to Handshaking`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    HandshakingToConnected: <ConnType extends Connection>(\n      oldSession: SessionHandshaking<ConnType>,\n      listeners: SessionConnectedListeners,\n    ): SessionConnected<ConnType> => {\n      const carriedState = inheritSharedSession(oldSession);\n      const conn = oldSession.conn;\n      oldSession._handleStateExit();\n\n      const session = new SessionConnected({\n        conn,\n        listeners,\n        ...carriedState,\n      });\n\n      session.log?.info(\n        `session ${session.id} transition from Handshaking to Connected`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    WaitingForHandshakeToConnected: <ConnType extends Connection>(\n      pendingSession: SessionWaitingForHandshake<ConnType>,\n      oldSession: SessionNoConnection | undefined,\n      sessionId: string,\n      to: TransportClientId,\n      propagationCtx: PropagationContext | undefined,\n      listeners: SessionConnectedListeners,\n      protocolVersion: ProtocolVersion,\n    ): SessionConnected<ConnType> => {\n      const conn = pendingSession.conn;\n      const { from, options } = pendingSession;\n      const carriedState: IdentifiedSessionProps = oldSession\n        ? // old session exists, inherit state\n          inheritSharedSession(oldSession)\n        : // old session does not exist, create new state\n          {\n            id: sessionId,\n            from,\n            to,\n            seq: 0,\n            ack: 0,\n            sendBuffer: [],\n            telemetry: createSessionTelemetryInfo(\n              sessionId,\n              to,\n              from,\n              propagationCtx,\n            ),\n            options,\n            log: pendingSession.log,\n            protocolVersion,\n          };\n\n      pendingSession._handleStateExit();\n      oldSession?._handleStateExit();\n\n      const session = new SessionConnected({\n        conn,\n        listeners,\n        ...carriedState,\n      });\n      session.log?.info(\n        `session ${session.id} transition from WaitingForHandshake to Connected`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    // disconnect paths\n    BackingOffToNoConnection: (\n      oldSession: SessionBackingOff,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection => {\n      const carriedState = inheritSharedSessionWithGrace(oldSession);\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection({\n        listeners,\n        ...carriedState,\n      });\n      session.log?.info(\n        `session ${session.id} transition from BackingOff to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    ConnectingToNoConnection: <ConnType extends Connection>(\n      oldSession: SessionConnecting<ConnType>,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection => {\n      const carriedState = inheritSharedSessionWithGrace(oldSession);\n      oldSession.bestEffortClose();\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection({\n        listeners,\n        ...carriedState,\n      });\n      session.log?.info(\n        `session ${session.id} transition from Connecting to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    HandshakingToNoConnection: <ConnType extends Connection>(\n      oldSession: SessionHandshaking<ConnType>,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection => {\n      const carriedState = inheritSharedSessionWithGrace(oldSession);\n      oldSession.conn.close();\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection({\n        listeners,\n        ...carriedState,\n      });\n      session.log?.info(\n        `session ${session.id} transition from Handshaking to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    ConnectedToNoConnection: <ConnType extends Connection>(\n      oldSession: SessionConnected<ConnType>,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection => {\n      const carriedState = inheritSharedSession(oldSession);\n      const graceExpiryTime =\n        Date.now() + oldSession.options.sessionDisconnectGraceMs;\n      oldSession.conn.close();\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection({\n        listeners,\n        graceExpiryTime,\n        ...carriedState,\n      });\n      session.log?.info(\n        `session ${session.id} transition from Connected to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n  },\n} as const;\n\nconst transitions = SessionStateGraph.transition;\n\nexport const ClientSessionStateGraph = {\n  entrypoint: SessionStateGraph.entrypoints.NoConnection,\n  transition: {\n    // happy paths\n    // NoConnection -> BackingOff: attempt to connect\n    NoConnectionToBackingOff: transitions.NoConnectionToBackingOff,\n    // BackingOff -> Connecting: backoff period elapsed, start connection\n    BackingOffToConnecting: transitions.BackingOffToConnecting,\n    // Connecting -> Handshaking: connection established, start handshake\n    ConnectingToHandshaking: transitions.ConnectingToHandshaking,\n    // Handshaking -> Connected: handshake complete, session ready\n    HandshakingToConnected: transitions.HandshakingToConnected,\n\n    // disconnect paths\n    // BackingOff -> NoConnection: unused\n    BackingOffToNoConnection: transitions.BackingOffToNoConnection,\n    // Connecting -> NoConnection: connection failed or connection timeout\n    ConnectingToNoConnection: transitions.ConnectingToNoConnection,\n    // Handshaking -> NoConnection: connection closed or handshake timeout\n    HandshakingToNoConnection: transitions.HandshakingToNoConnection,\n    // Connected -> NoConnection: connection closed\n    ConnectedToNoConnection: transitions.ConnectedToNoConnection,\n\n    // destroy/close paths\n    // NoConnection -> x: grace period elapsed\n    // BackingOff -> x: grace period elapsed\n    // Connecting -> x: grace period elapsed\n    // Handshaking -> x: grace period elapsed or invalid handshake message or handshake rejection\n    // Connected -> x: grace period elapsed or invalid message\n  },\n};\n\nexport type ClientSession<ConnType extends Connection> =\n  | SessionNoConnection\n  | SessionBackingOff\n  | SessionConnecting<ConnType>\n  | SessionHandshaking<ConnType>\n  | SessionConnected<ConnType>;\n\nexport const ServerSessionStateGraph = {\n  entrypoint: SessionStateGraph.entrypoints.WaitingForHandshake,\n  transition: {\n    // happy paths\n    // WaitingForHandshake -> Connected: handshake complete, session ready\n    WaitingForHandshakeToConnected: transitions.WaitingForHandshakeToConnected,\n\n    // disconnect paths\n    // Connected -> NoConnection: connection closed\n    ConnectedToNoConnection: transitions.ConnectedToNoConnection,\n\n    // destroy/close paths\n    // WaitingForHandshake -> x: handshake timeout elapsed or invalid handshake message or handshake rejection or connection closed\n  },\n};\n\nexport type ServerSession<ConnType extends Connection> =\n  // SessionWaitingForHandshake<ConnType> is stored separately in the server transport\n  SessionConnected<ConnType> | SessionNoConnection;\n\nexport type Session<ConnType extends Connection> =\n  | ClientSession<ConnType>\n  | ServerSession<ConnType>;\n"],"mappings":";;;;;;;;;;;AAQO,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,qBAAqB;AAAA,EACrB,oBAAoB;AAAA,EACpB,oCAAoC;AAAA,EACpC,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AAAA,EACH,GAAG;AACL;AAMO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AACL;;;AC/BA,SAAS,aAAa;AAIf,IAAW,eAAX,kBAAWA,kBAAX;AACL,EAAAA,cAAA,kBAAe;AACf,EAAAA,cAAA,gBAAa;AACb,EAAAA,cAAA,gBAAa;AACb,EAAAA,cAAA,iBAAc;AACd,EAAAA,cAAA,eAAY;AACZ,EAAAA,cAAA,yBAAsB;AANN,SAAAA;AAAA,GAAA;AASX,IAAM,eAAe;AAE5B,IAAe,oBAAf,MAAiC;AAAA;AAAA;AAAA;AAAA;AAAA,EAO/B;AAAA,EAaA,QAAc;AACZ,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,cAAc;AACZ,SAAK,cAAc;AAKnB,WAAO,IAAI,MAAM,MAAM;AAAA,MACrB,IAAI,QAAQ,MAAM;AAEhB,YAAI,SAAS,iBAAiB,SAAS,QAAQ,SAAS,SAAS;AAC/D,iBAAO,QAAQ,IAAI,QAAQ,IAAI;AAAA,QACjC;AAGA,YAAI,SAAS,oBAAoB;AAC/B,iBAAO,MAAM;AACX,mBAAO,cAAc;AACrB,mBAAO,iBAAiB;AAAA,UAC1B;AAAA,QACF;AAGA,YAAI,SAAS,gBAAgB;AAC3B,iBAAO,MAAM;AAEX,mBAAO,cAAc;AACrB,mBAAO,iBAAiB;AACxB,mBAAO,aAAa;AAAA,UACtB;AAAA,QACF;AAEA,YAAI,OAAO,aAAa;AACtB,gBAAM,IAAI;AAAA,YACR,GAAG,YAAY,aAAa,KAAK,SAAS,CAAC;AAAA,UAC7C;AAAA,QACF;AAEA,eAAO,QAAQ,IAAI,QAAQ,IAAI;AAAA,MACjC;AAAA,MACA,IAAI,QAAQ,MAAM,OAAO;AACvB,YAAI,OAAO,aAAa;AACtB,gBAAM,IAAI;AAAA,YACR,GAAG,YAAY,aAAa,KAAK,SAAS,CAAC;AAAA,UAC7C;AAAA,QACF;AAEA,eAAO,QAAQ,IAAI,QAAQ,MAAM,KAAK;AAAA,MACxC;AAAA,IACF,CAAC;AAAA,EACH;AACF;AA6CO,IAAe,gBAAf,cAAqC,kBAAkB;AAAA,EACnD;AAAA,EACA;AAAA,EAET;AAAA,EAGA,YAAY,EAAE,MAAM,SAAS,IAAI,GAAuB;AACtD,UAAM;AACN,SAAK,OAAO;AACZ,SAAK,UAAU;AACf,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,SAAS,KAAgD;AACvD,UAAM,YAAY,KAAK,QAAQ,MAAM,WAAW,GAAG;AAEnD,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,KAAK;AAAA,QACR,2BAA2B,aAAa;AAAA,QACxC,KAAK;AAAA,MACP;AAEA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,KAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QACpE,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,8BAA8B,SAAS;AAAA,QACzD;AAAA,MACF,CAAC;AAED,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AACF;AAoBO,IAAe,oBAAf,cAAyC,cAAc;AAAA,EACnD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EACA;AAAA,EAEA,YAAY,OAA+B;AACzC,UAAM,EAAE,IAAI,IAAI,KAAK,KAAK,YAAY,WAAW,KAAK,gBAAgB,IACpE;AACF,UAAM,KAAK;AACX,SAAK,KAAK;AACV,SAAK,KAAK;AACV,SAAK,MAAM;AACX,SAAK,MAAM;AACX,SAAK,aAAa;AAClB,SAAK,YAAY;AACjB,SAAK,MAAM;AACX,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,cAAc,KAAK,UAAU,KAAK,YAAY;AAEpD,UAAM,WAA4B;AAAA,MAChC,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,IAClB;AAEA,QAAI,KAAK,UAAU,KAAK,YAAY,GAAG;AACrC,eAAS,YAAY;AAAA,QACnB,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,WAAW;AAAA,MACf,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AAEL,WAAO;AAAA,EACT;AAAA,EAEA,UAAkB;AAChB,WAAO,KAAK,WAAW,SAAS,IAAI,KAAK,WAAW,CAAC,EAAE,MAAM,KAAK;AAAA,EACpE;AAAA,EAEA,KAAK,KAAsC;AACzC,UAAM,iBAAiB,KAAK,aAAa,GAAG;AAC5C,SAAK,WAAW,KAAK,cAAc;AAEnC,WAAO,eAAe;AAAA,EACxB;AAAA,EAEA,mBAAyB;AAAA,EAEzB;AAAA,EAEA,eAAqB;AAEnB,SAAK,WAAW,SAAS;AACzB,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AACF;AAYO,IAAe,mCAAf,cAAwD,kBAAkB;AAAA,EAC/E;AAAA,EACU;AAAA,EAEV;AAAA,EAEA,YAAY,OAA8C;AACxD,UAAM,KAAK;AACX,SAAK,YAAY,MAAM;AAEvB,SAAK,kBAAkB,MAAM;AAC7B,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,UAAU,4BAA4B;AAAA,IAC7C,GAAG,KAAK,kBAAkB,KAAK,IAAI,CAAC;AAAA,EACtC;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AAEvB,QAAI,KAAK,oBAAoB;AAC3B,mBAAa,KAAK,kBAAkB;AACpC,WAAK,qBAAqB;AAAA,IAC5B;AAAA,EACF;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AAAA,EACrB;AACF;;;AChTO,IAAM,oBAAN,cAEG,iCAAiC;AAAA,EAChC;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YAAY,OAAyC;AACnD,UAAM,KAAK;AACX,SAAK,cAAc,MAAM;AACzB,SAAK,YAAY,MAAM;AAEvB,SAAK,YAAY;AAAA,MACf,CAAC,SAAS;AACR,YAAI,KAAK;AAAa;AACtB,aAAK,UAAU,wBAAwB,IAAI;AAAA,MAC7C;AAAA,MACA,CAAC,QAAQ;AACP,YAAI,KAAK;AAAa;AACtB,aAAK,UAAU,mBAAmB,GAAG;AAAA,MACvC;AAAA,IACF;AAEA,SAAK,oBAAoB,WAAW,MAAM;AACxC,WAAK,UAAU,oBAAoB;AAAA,IACrC,GAAG,KAAK,QAAQ,mBAAmB;AAAA,EACrC;AAAA;AAAA;AAAA,EAIA,kBAAkB;AAKhB,UAAM,SAAS,KAAK;AACpB,UAAM,WAAW,KAAK;AAEtB,SAAK,YACF,KAAK,CAAC,SAAS;AACd,WAAK,MAAM;AACX,cAAQ;AAAA,QACN;AAAA,QACA;AAAA,UACE,GAAG;AAAA,UACH,GAAG,KAAK;AAAA,QACV;AAAA,MACF;AAAA,IACF,CAAC,EACA,MAAM,MAAM;AAAA,IAEb,CAAC;AAAA,EACL;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AAEvB,QAAI,KAAK,mBAAmB;AAC1B,mBAAa,KAAK,iBAAiB;AACnC,WAAK,oBAAoB;AAAA,IAC3B;AAAA,EACF;AAAA,EAEA,eAAqB;AAEnB,SAAK,gBAAgB;AACrB,UAAM,aAAa;AAAA,EACrB;AACF;;;ACjFO,IAAM,sBAAN,cAAkC,iCAAiC;AAAA,EAC/D;AAAA,EAET,eAAqB;AACnB,UAAM,aAAa;AAAA,EACrB;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AAAA,EACzB;AACF;;;ACMO,IAAM,6BAAN,cAEG,cAAc;AAAA,EACb;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YAAY,OAAkD;AAC5D,UAAM,KAAK;AACX,SAAK,OAAO,MAAM;AAClB,SAAK,YAAY,MAAM;AAEvB,SAAK,mBAAmB,WAAW,MAAM;AACvC,WAAK,UAAU,mBAAmB;AAAA,IACpC,GAAG,KAAK,QAAQ,kBAAkB;AAElC,SAAK,KAAK,gBAAgB,KAAK,eAAe;AAC9C,SAAK,KAAK,iBAAiB,KAAK,UAAU,mBAAmB;AAC7D,SAAK,KAAK,iBAAiB,KAAK,UAAU,kBAAkB;AAAA,EAC9D;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK,KAAK;AAAA,MAClB,GAAG,KAAK,KAAK;AAAA,IACf;AAAA,EACF;AAAA,EAEA,kBAAkB,CAAC,QAAoB;AACrC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU;AAAA,QACb;AAAA,QACA;AAAA,MACF;AAEA;AAAA,IACF;AAIA,SAAK,UAAU,YAAY,SAAS;AAAA,EACtC;AAAA,EAEA,cAAc,KAAgC;AAC5C,WAAO,KAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,EACxD;AAAA,EAEA,mBAAyB;AACvB,SAAK,KAAK,mBAAmB,KAAK,eAAe;AACjD,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,iBAAa,KAAK,gBAAgB;AAClC,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,eAAqB;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;ACxDO,IAAM,qBAAN,cAEG,iCAAiC;AAAA,EAChC;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YAAY,OAA0C;AACpD,UAAM,KAAK;AACX,SAAK,OAAO,MAAM;AAClB,SAAK,YAAY,MAAM;AAEvB,SAAK,mBAAmB,WAAW,MAAM;AACvC,WAAK,UAAU,mBAAmB;AAAA,IACpC,GAAG,KAAK,QAAQ,kBAAkB;AAElC,SAAK,KAAK,gBAAgB,KAAK,eAAe;AAC9C,SAAK,KAAK,iBAAiB,KAAK,UAAU,mBAAmB;AAC7D,SAAK,KAAK,iBAAiB,KAAK,UAAU,kBAAkB;AAAA,EAC9D;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO;AAAA,MACL,GAAG,MAAM;AAAA,MACT,GAAG,KAAK,KAAK;AAAA,IACf;AAAA,EACF;AAAA,EAEA,kBAAkB,CAAC,QAAoB;AACrC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU;AAAA,QACb;AAAA,QACA;AAAA,MACF;AAEA;AAAA,IACF;AAEA,SAAK,UAAU,YAAY,SAAS;AAAA,EACtC;AAAA,EAEA,cAAc,KAAgC;AAC5C,WAAO,KAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,EACxD;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,SAAK,KAAK,mBAAmB,KAAK,eAAe;AACjD,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAE/D,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAClC,WAAK,mBAAmB;AAAA,IAC1B;AAAA,EACF;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;ACxFA,SAAS,sBAAsB;AAmBxB,IAAM,mBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEQ;AAAA,EACA,kBAAkB;AAAA,EAC1B;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AACjB,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,KAAK,KAAsC;AACzC,UAAM,iBAAiB,KAAK,aAAa,GAAG;AAC5C,SAAK,WAAW,KAAK,cAAc;AACnC,SAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,cAAc,CAAC;AAE1D,WAAO,eAAe;AAAA,EACxB;AAAA,EAEA,YAAY,OAAwC;AAClD,UAAM,KAAK;AACX,SAAK,OAAO,MAAM;AAClB,SAAK,YAAY,MAAM;AAEvB,SAAK,KAAK,gBAAgB,KAAK,aAAa;AAC5C,SAAK,KAAK,iBAAiB,KAAK,UAAU,kBAAkB;AAC5D,SAAK,KAAK,iBAAiB,KAAK,UAAU,mBAAmB;AAG7D,QAAI,KAAK,WAAW,SAAS,GAAG;AAC9B,WAAK,KAAK;AAAA,QACR,WACE,KAAK,WAAW,MAClB,uCAAuC,KAAK,QAAQ,CAAC;AAAA,QACrD,KAAK;AAAA,MACP;AAEA,iBAAW,OAAO,KAAK,YAAY;AACjC,aAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,MACjD;AAAA,IACF;AAMA,SAAK,yBAAyB;AAC9B,SAAK,kBAAkB,YAAY,MAAM;AACvC,YAAM,SAAS,KAAK;AACpB,YAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,UAAI,UAAU,KAAK,QAAQ,qBAAqB;AAC9C,aAAK,KAAK;AAAA,UACR,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AASnE,aAAK,KAAK,MAAM;AAChB,sBAAc,KAAK,eAAe;AAClC,aAAK,kBAAkB;AAEvB;AAAA,MACF;AAEA,UAAI,KAAK,wBAAwB;AAC/B,aAAK,cAAc;AAAA,MACrB;AAEA,WAAK;AAAA,IACP,GAAG,KAAK,QAAQ,mBAAmB;AAAA,EACrC;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO;AAAA,MACL,GAAG,MAAM;AAAA,MACT,GAAG,KAAK,KAAK;AAAA,IACf;AAAA,EACF;AAAA,EAEA,uBAAuB;AACrB,SAAK,yBAAyB;AAAA,EAChC;AAAA,EAEQ,gBAAgB;AACtB,SAAK,KAAK,MAAM,qBAAqB,KAAK,eAAe;AACzD,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,kBAAkB;AAChB,SAAK,KAAK,mBAAmB,KAAK,aAAa;AAC/C,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,MAAM;AAAA,EAClB;AAAA,EAEA,gBAAgB,CAAC,QAAoB;AACnC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU,iBAAiB,yBAAyB;AAEzD;AAAA,IACF;AAGA,QAAI,UAAU,QAAQ,KAAK,KAAK;AAC9B,UAAI,UAAU,MAAM,KAAK,KAAK;AAC5B,aAAK,KAAK;AAAA,UACR,oCAAoC,UAAU,GAAG,iBAAiB,KAAK,GAAG;AAAA,UAC1E;AAAA,YACE,GAAG,KAAK;AAAA,YACR,kBAAkB;AAAA,UACpB;AAAA,QACF;AAAA,MACF,OAAO;AACL,cAAM,SAAS,2DAA2D,UAAU,GAAG,iBAAiB,KAAK,GAAG;AAChH,aAAK,KAAK,KAAK,QAAQ;AAAA,UACrB,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,UAClB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AAED,aAAK,UAAU,KAAK,UAAU;AAAA,UAC5B,MAAM,eAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAID,aAAK,gBAAgB;AAAA,MACvB;AAEA;AAAA,IACF;AAGA,SAAK,KAAK,MAAM,gBAAgB;AAAA,MAC9B,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAED,SAAK,kBAAkB,UAAU,KAAK,UAAU,GAAG;AAGnD,QAAI,CAAC,MAAM,UAAU,YAAY,GAAG;AAClC,WAAK,UAAU,UAAU,SAAS;AAElC;AAAA,IACF;AAGA,SAAK,KAAK,MAAM,gCAAgC;AAAA,MAC9C,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAID,QAAI,CAAC,KAAK,wBAAwB;AAChC,WAAK,cAAc;AAAA,IACrB;AAAA,EACF;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,SAAK,KAAK,mBAAmB,KAAK,aAAa;AAC/C,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAEhE,QAAI,KAAK,iBAAiB;AACxB,oBAAc,KAAK,eAAe;AAClC,WAAK,kBAAkB;AAAA,IACzB;AAAA,EACF;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;AChNO,IAAM,oBAAN,cAAgC,iCAAiC;AAAA,EAC7D;AAAA,EACT;AAAA,EAEA;AAAA,EAEA,YAAY,OAA+B;AACzC,UAAM,KAAK;AACX,SAAK,YAAY,MAAM;AAEvB,SAAK,iBAAiB,WAAW,MAAM;AACrC,WAAK,UAAU,kBAAkB;AAAA,IACnC,GAAG,MAAM,SAAS;AAAA,EACpB;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AAAA,EACrB;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AAEvB,QAAI,KAAK,gBAAgB;AACvB,mBAAa,KAAK,cAAc;AAChC,WAAK,iBAAiB;AAAA,IACxB;AAAA,EACF;AACF;;;ACXA,SAAS,qBACP,SACwB;AACxB,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,IAAI,QAAQ;AAAA,IACZ,KAAK,QAAQ;AAAA,IACb,KAAK,QAAQ;AAAA,IACb,YAAY,QAAQ;AAAA,IACpB,WAAW,QAAQ;AAAA,IACnB,SAAS,QAAQ;AAAA,IACjB,KAAK,QAAQ;AAAA,IACb,iBAAiB,QAAQ;AAAA,EAC3B;AACF;AAEA,SAAS,8BACP,SAC0D;AAC1D,SAAO;AAAA,IACL,GAAG,qBAAqB,OAAO;AAAA,IAC/B,iBAAiB,QAAQ;AAAA,EAC3B;AACF;AAEO,IAAM,oBAAoB;AAAA,EAC/B,aAAa;AAAA,IACX,cAAc,CACZ,IACA,MACA,WACA,SACA,iBACA,QACG;AACH,YAAM,KAAK,WAAW,WAAW,CAAC;AAClC,YAAM,YAAY,2BAA2B,IAAI,IAAI,IAAI;AACzD,YAAM,aAA4C,CAAC;AAEnD,YAAM,UAAU,IAAI,oBAAoB;AAAA,QACtC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,iBAAiB,KAAK,IAAI,IAAI,QAAQ;AAAA,QACtC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAED,cAAQ,KAAK,KAAK,WAAW,QAAQ,EAAE,kCAAkC;AAAA,QACvE,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,kBAAkB;AAAA,MAC3B,CAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,qBAAqB,CACnB,MACA,MACA,WACA,SACA,QACyC;AACzC,YAAM,UAAU,IAAI,2BAA2B;AAAA,QAC7C;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAED,cAAQ,KAAK,KAAK,gDAAgD;AAAA,QAChE,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,kBAAkB;AAAA,MAC3B,CAAC;AAED,aAAO;AAAA,IACT;AAAA,EACF;AAAA;AAAA;AAAA,EAGA,YAAY;AAAA;AAAA,IAEV,0BAA0B,CACxB,YACA,WACA,cACsB;AACtB,YAAM,eAAe,8BAA8B,UAAU;AAC7D,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,kBAAkB;AAAA,QACpC;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AAED,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,wBAAwB,CACtB,YACA,aACA,cACgC;AAChC,YAAM,eAAe,8BAA8B,UAAU;AAC7D,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,kBAAkB;AAAA,QACpC;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AAED,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,yBAAyB,CACvB,YACA,MACA,cACiC;AACjC,YAAM,eAAe,8BAA8B,UAAU;AAC7D,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,mBAAmB;AAAA,QACrC;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AAED,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,wBAAwB,CACtB,YACA,cAC+B;AAC/B,YAAM,eAAe,qBAAqB,UAAU;AACpD,YAAM,OAAO,WAAW;AACxB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,iBAAiB;AAAA,QACnC;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AAED,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,gCAAgC,CAC9B,gBACA,YACA,WACA,IACA,gBACA,WACA,oBAC+B;AAC/B,YAAM,OAAO,eAAe;AAC5B,YAAM,EAAE,MAAM,QAAQ,IAAI;AAC1B,YAAM,eAAuC;AAAA;AAAA,QAEzC,qBAAqB,UAAU;AAAA;AAAA;AAAA,QAE/B;AAAA,UACE,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,UACA,KAAK;AAAA,UACL,KAAK;AAAA,UACL,YAAY,CAAC;AAAA,UACb,WAAW;AAAA,YACT;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF;AAAA,UACA;AAAA,UACA,KAAK,eAAe;AAAA,UACpB;AAAA,QACF;AAAA;AAEJ,qBAAe,iBAAiB;AAChC,kBAAY,iBAAiB;AAE7B,YAAM,UAAU,IAAI,iBAAiB;AAAA,QACnC;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,IAEA,0BAA0B,CACxB,YACA,cACwB;AACxB,YAAM,eAAe,8BAA8B,UAAU;AAC7D,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB;AAAA,QACtC;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,0BAA0B,CACxB,YACA,cACwB;AACxB,YAAM,eAAe,8BAA8B,UAAU;AAC7D,iBAAW,gBAAgB;AAC3B,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB;AAAA,QACtC;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,2BAA2B,CACzB,YACA,cACwB;AACxB,YAAM,eAAe,8BAA8B,UAAU;AAC7D,iBAAW,KAAK,MAAM;AACtB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB;AAAA,QACtC;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,yBAAyB,CACvB,YACA,cACwB;AACxB,YAAM,eAAe,qBAAqB,UAAU;AACpD,YAAM,kBACJ,KAAK,IAAI,IAAI,WAAW,QAAQ;AAClC,iBAAW,KAAK,MAAM;AACtB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB;AAAA,QACtC;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,EACF;AACF;AAEA,IAAM,cAAc,kBAAkB;AAE/B,IAAM,0BAA0B;AAAA,EACrC,YAAY,kBAAkB,YAAY;AAAA,EAC1C,YAAY;AAAA;AAAA;AAAA,IAGV,0BAA0B,YAAY;AAAA;AAAA,IAEtC,wBAAwB,YAAY;AAAA;AAAA,IAEpC,yBAAyB,YAAY;AAAA;AAAA,IAErC,wBAAwB,YAAY;AAAA;AAAA;AAAA,IAIpC,0BAA0B,YAAY;AAAA;AAAA,IAEtC,0BAA0B,YAAY;AAAA;AAAA,IAEtC,2BAA2B,YAAY;AAAA;AAAA,IAEvC,yBAAyB,YAAY;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQvC;AACF;AASO,IAAM,0BAA0B;AAAA,EACrC,YAAY,kBAAkB,YAAY;AAAA,EAC1C,YAAY;AAAA;AAAA;AAAA,IAGV,gCAAgC,YAAY;AAAA;AAAA;AAAA,IAI5C,yBAAyB,YAAY;AAAA;AAAA;AAAA,EAIvC;AACF;","names":["SessionState"]}

package/dist/{chunk-4HT6P2ZG.js → chunk-A4JKES5A.js}

@@ -1,11 +1,11 @@
 import {
   ProtocolError,
   Transport
-} from "./chunk-I75XYO5W.js";
+} from "./chunk-QIDEN5PP.js";
 import {
   ClientSessionStateGraph,
   defaultClientTransportOptions
-} from "./chunk-42Z2FQIU.js";
+} from "./chunk-6BH2CXVE.js";
 import {
   ControlMessageHandshakeResponseSchema,
   HandshakeErrorRetriableResponseCodes,
@@ -14,7 +14,7 @@ import {
   getPropagationContext,
   handshakeRequestMessage,
   tracing_default
-} from "./chunk-EETL2L77.js";
+} from "./chunk-GJUUVID2.js";
 
 // transport/client.ts
 import { SpanStatusCode } from "@opentelemetry/api";
@@ -127,22 +127,10 @@ var ClientTransport = class extends Transport {
       this.connect(to);
     }
   }
-  send(to, msg) {
-    if (this.getStatus() === "closed") {
-      const err = "transport is closed, cant send";
-      this.log?.error(err, {
-        clientId: this.clientId,
-        transportMessage: msg,
-        tags: ["invariant-violation"]
-      });
-      throw new Error(err);
-    }
-    let session = this.sessions.get(to);
-    if (!session) {
-      session = this.createUnconnectedSession(to);
-    }
-    return session.send(msg);
-  }
+  /*
+   * Creates a raw unconnected session object.
+   * This is mostly a River internal, you shouldn't need to use this directly.
+   */
   createUnconnectedSession(to) {
     const session = ClientSessionStateGraph.entrypoint(
       to,
@@ -156,7 +144,7 @@ var ClientTransport = class extends Transport {
       currentProtocolVersion,
       this.log
     );
-    this.updateSession(session);
+    this.createSession(session);
     return session;
   }
   // listeners
@@ -197,7 +185,7 @@ var ClientTransport = class extends Transport {
             `invalid handshake: ${reason}`,
             handshakingSession.loggingMetadata
           );
-          this.deleteSession(session);
+          this.deleteSession(session, { unhealthy: true });
           this.protocolError({
             type: ProtocolError.HandshakeFailed,
             code,
@@ -226,7 +214,7 @@ var ClientTransport = class extends Transport {
       message: reason
     });
     this.log?.warn(reason, metadata);
-    this.deleteSession(session);
+    this.deleteSession(session, { unhealthy: true });
   }
   onHandshakeResponse(session, msg) {
     if (!Value.Check(ControlMessageHandshakeResponseSchema, msg.payload)) {
@@ -241,10 +229,10 @@ var ClientTransport = class extends Transport {
       return;
     }
     if (!msg.payload.status.ok) {
-      const retriable = msg.payload.status.code ? Value.Check(
+      const retriable = Value.Check(
         HandshakeErrorRetriableResponseCodes,
         msg.payload.status.code
-      ) : false;
+      );
       const reason = `handshake failed: ${msg.payload.status.reason}`;
       const to = session.to;
       this.rejectHandshakeResponse(session, reason, {
@@ -289,11 +277,13 @@ var ClientTransport = class extends Transport {
         );
         this.onConnClosed(connectedSession);
       },
-      onMessage: (msg2) => this.handleMsg(msg2),
+      onMessage: (msg2) => {
+        this.handleMsg(msg2);
+      },
       onInvalidMessage: (reason) => {
-        this.deleteSession(connectedSession);
+        this.deleteSession(connectedSession, { unhealthy: true });
         this.protocolError({
-          type: ProtocolError.MessageOrderingViolated,
+          type: ProtocolError.InvalidMessage,
           message: reason
         });
       }
@@ -312,8 +302,7 @@ var ClientTransport = class extends Transport {
       );
       return;
     }
-    let session = this.sessions.get(to);
-    session ??= this.createUnconnectedSession(to);
+    const session = this.sessions.get(to) ?? this.createUnconnectedSession(to);
     if (session.state !== "NoConnection" /* NoConnection */) {
       this.log?.debug(
         `session to ${to} has state ${session.state}, skipping connect attempt`,
@@ -405,6 +394,9 @@ var ClientTransport = class extends Transport {
     if (this.handshakeExtensions) {
       metadata = await this.handshakeExtensions.construct();
     }
+    if (session._isConsumed) {
+      return;
+    }
     const requestMsg = handshakeRequestMessage({
       from: this.clientId,
       to: session.to,
@@ -431,4 +423,4 @@ var ClientTransport = class extends Transport {
 export {
   ClientTransport
 };
-//# sourceMappingURL=chunk-4HT6P2ZG.js.map
+//# sourceMappingURL=chunk-A4JKES5A.js.map

package/dist/chunk-A4JKES5A.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/client.ts","../transport/rateLimit.ts"],"sourcesContent":["import { SpanStatusCode } from '@opentelemetry/api';\nimport { ClientHandshakeOptions } from '../router/handshake';\nimport {\n  ControlMessageHandshakeResponseSchema,\n  HandshakeErrorRetriableResponseCodes,\n  OpaqueTransportMessage,\n  TransportClientId,\n  currentProtocolVersion,\n  handshakeRequestMessage,\n} from './message';\nimport {\n  ClientTransportOptions,\n  ProvidedClientTransportOptions,\n  defaultClientTransportOptions,\n} from './options';\nimport { LeakyBucketRateLimit } from './rateLimit';\nimport { Transport } from './transport';\nimport { coerceErrorString } from '../util/stringify';\nimport { ProtocolError } from './events';\nimport { Value } from '@sinclair/typebox/value';\nimport tracer, { getPropagationContext } from '../tracing';\nimport { Connection } from './connection';\nimport { MessageMetadata } from '../logging';\nimport { SessionConnecting } from './sessionStateMachine/SessionConnecting';\nimport { SessionHandshaking } from './sessionStateMachine/SessionHandshaking';\nimport { SessionConnected } from './sessionStateMachine/SessionConnected';\nimport {\n  ClientSession,\n  ClientSessionStateGraph,\n} from './sessionStateMachine/transitions';\nimport { SessionState } from './sessionStateMachine/common';\nimport { SessionNoConnection } from './sessionStateMachine/SessionNoConnection';\nimport { SessionBackingOff } from './sessionStateMachine/SessionBackingOff';\n\nexport abstract class ClientTransport<\n  ConnType extends Connection,\n> extends Transport<ConnType> {\n  /**\n   * The options for this transport.\n   */\n  protected options: ClientTransportOptions;\n\n  retryBudget: LeakyBucketRateLimit;\n\n  /**\n   * A flag indicating whether the transport should automatically reconnect\n   * when a connection is dropped.\n   * Realistically, this should always be true for clients unless you are writing\n   * tests or a special case where you don't want to reconnect.\n   */\n  reconnectOnConnectionDrop = true;\n\n  /**\n   * Optional handshake options for this client.\n   */\n  handshakeExtensions?: ClientHandshakeOptions;\n\n  sessions: Map<TransportClientId, ClientSession<ConnType>>;\n\n  constructor(\n    clientId: TransportClientId,\n    providedOptions?: ProvidedClientTransportOptions,\n  ) {\n    super(clientId, providedOptions);\n    this.sessions = new Map();\n    this.options = {\n      ...defaultClientTransportOptions,\n      ...providedOptions,\n    };\n    this.retryBudget = new LeakyBucketRateLimit(this.options);\n  }\n\n  extendHandshake(options: ClientHandshakeOptions) {\n    this.handshakeExtensions = options;\n  }\n\n  /**\n   * Abstract method that creates a new {@link Connection} object.\n   *\n   * @param to The client ID of the node to connect to.\n   * @returns The new connection object.\n   */\n  protected abstract createNewOutgoingConnection(\n    to: TransportClientId,\n  ): Promise<ConnType>;\n\n  private tryReconnecting(to: TransportClientId) {\n    const oldSession = this.sessions.get(to);\n    if (!this.options.enableTransparentSessionReconnects && oldSession) {\n      this.deleteSession(oldSession);\n    }\n\n    if (this.reconnectOnConnectionDrop && this.getStatus() === 'open') {\n      this.connect(to);\n    }\n  }\n\n  /*\n   * Creates a raw unconnected session object.\n   * This is mostly a River internal, you shouldn't need to use this directly.\n   */\n  createUnconnectedSession(to: string): SessionNoConnection {\n    const session = ClientSessionStateGraph.entrypoint(\n      to,\n      this.clientId,\n      {\n        onSessionGracePeriodElapsed: () => {\n          this.onSessionGracePeriodElapsed(session);\n        },\n      },\n      this.options,\n      currentProtocolVersion,\n      this.log,\n    );\n\n    this.createSession(session);\n\n    return session;\n  }\n\n  // listeners\n  protected onConnectingFailed(session: SessionConnecting<ConnType>) {\n    const noConnectionSession = super.onConnectingFailed(session);\n    this.tryReconnecting(noConnectionSession.to);\n\n    return noConnectionSession;\n  }\n\n  protected onConnClosed(\n    session: SessionHandshaking<ConnType> | SessionConnected<ConnType>,\n  ) {\n    const noConnectionSession = super.onConnClosed(session);\n    this.tryReconnecting(noConnectionSession.to);\n\n    return noConnectionSession;\n  }\n\n  protected onConnectionEstablished(\n    session: SessionConnecting<ConnType>,\n    conn: ConnType,\n  ): SessionHandshaking<ConnType> {\n    // transition to handshaking\n    const handshakingSession =\n      ClientSessionStateGraph.transition.ConnectingToHandshaking(\n        session,\n        conn,\n        {\n          onConnectionErrored: (err) => {\n            // just log, when we error we also emit close\n            const errStr = coerceErrorString(err);\n            this.log?.error(\n              `connection to ${handshakingSession.to} errored during handshake: ${errStr}`,\n              handshakingSession.loggingMetadata,\n            );\n          },\n          onConnectionClosed: () => {\n            this.log?.warn(\n              `connection to ${handshakingSession.to} closed during handshake`,\n              handshakingSession.loggingMetadata,\n            );\n            this.onConnClosed(handshakingSession);\n          },\n          onHandshake: (msg) => {\n            this.onHandshakeResponse(handshakingSession, msg);\n          },\n          onInvalidHandshake: (reason, code) => {\n            this.log?.error(\n              `invalid handshake: ${reason}`,\n              handshakingSession.loggingMetadata,\n            );\n            this.deleteSession(session, { unhealthy: true });\n            this.protocolError({\n              type: ProtocolError.HandshakeFailed,\n              code,\n              message: reason,\n            });\n          },\n          onHandshakeTimeout: () => {\n            this.log?.error(\n              `connection to ${handshakingSession.to} timed out during handshake`,\n              handshakingSession.loggingMetadata,\n            );\n            this.onConnClosed(handshakingSession);\n          },\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(handshakingSession);\n          },\n        },\n      );\n\n    this.updateSession(handshakingSession);\n    void this.sendHandshake(handshakingSession);\n\n    return handshakingSession;\n  }\n\n  private rejectHandshakeResponse(\n    session: SessionHandshaking<ConnType>,\n    reason: string,\n    metadata: MessageMetadata,\n  ) {\n    session.conn.telemetry?.span.setStatus({\n      code: SpanStatusCode.ERROR,\n      message: reason,\n    });\n\n    this.log?.warn(reason, metadata);\n    this.deleteSession(session, { unhealthy: true });\n  }\n\n  protected onHandshakeResponse(\n    session: SessionHandshaking<ConnType>,\n    msg: OpaqueTransportMessage,\n  ) {\n    // invariant: msg is a handshake response\n    if (!Value.Check(ControlMessageHandshakeResponseSchema, msg.payload)) {\n      const reason = `received invalid handshake response`;\n      this.rejectHandshakeResponse(session, reason, {\n        ...session.loggingMetadata,\n        transportMessage: msg,\n        validationErrors: [\n          ...Value.Errors(ControlMessageHandshakeResponseSchema, msg.payload),\n        ],\n      });\n\n      return;\n    }\n\n    // invariant: handshake response should be ok\n    if (!msg.payload.status.ok) {\n      const retriable = Value.Check(\n        HandshakeErrorRetriableResponseCodes,\n        msg.payload.status.code,\n      );\n\n      const reason = `handshake failed: ${msg.payload.status.reason}`;\n      const to = session.to;\n      this.rejectHandshakeResponse(session, reason, {\n        ...session.loggingMetadata,\n        transportMessage: msg,\n      });\n\n      if (retriable) {\n        this.tryReconnecting(to);\n      } else {\n        this.protocolError({\n          type: ProtocolError.HandshakeFailed,\n          code: msg.payload.status.code,\n          message: reason,\n        });\n      }\n\n      return;\n    }\n\n    // invariant: session id should match between client + server\n    if (msg.payload.status.sessionId !== session.id) {\n      const reason = `session id mismatch: expected ${session.id}, got ${msg.payload.status.sessionId}`;\n      this.rejectHandshakeResponse(session, reason, {\n        ...session.loggingMetadata,\n        transportMessage: msg,\n      });\n\n      return;\n    }\n\n    // transition to connected!\n    this.log?.info(`handshake from ${msg.from} ok`, {\n      ...session.loggingMetadata,\n      transportMessage: msg,\n    });\n\n    const connectedSession =\n      ClientSessionStateGraph.transition.HandshakingToConnected(session, {\n        onConnectionErrored: (err) => {\n          // just log, when we error we also emit close\n          const errStr = coerceErrorString(err);\n          this.log?.warn(\n            `connection to ${connectedSession.to} errored: ${errStr}`,\n            connectedSession.loggingMetadata,\n          );\n        },\n        onConnectionClosed: () => {\n          this.log?.info(\n            `connection to ${connectedSession.to} closed`,\n            connectedSession.loggingMetadata,\n          );\n          this.onConnClosed(connectedSession);\n        },\n        onMessage: (msg) => {\n          this.handleMsg(msg);\n        },\n        onInvalidMessage: (reason) => {\n          this.deleteSession(connectedSession, { unhealthy: true });\n          this.protocolError({\n            type: ProtocolError.InvalidMessage,\n            message: reason,\n          });\n        },\n      });\n\n    this.updateSession(connectedSession);\n    this.retryBudget.startRestoringBudget();\n  }\n\n  /**\n   * Manually attempts to connect to a client.\n   * @param to The client ID of the node to connect to.\n   */\n  connect(to: TransportClientId) {\n    if (this.getStatus() !== 'open') {\n      this.log?.info(\n        `transport state is no longer open, cancelling attempt to connect to ${to}`,\n      );\n\n      return;\n    }\n\n    const session = this.sessions.get(to) ?? this.createUnconnectedSession(to);\n    if (session.state !== SessionState.NoConnection) {\n      // already trying to connect\n      this.log?.debug(\n        `session to ${to} has state ${session.state}, skipping connect attempt`,\n        session.loggingMetadata,\n      );\n\n      return;\n    }\n\n    // check budget\n    if (!this.retryBudget.hasBudget()) {\n      const budgetConsumed = this.retryBudget.getBudgetConsumed();\n      const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;\n      this.log?.error(errMsg, session.loggingMetadata);\n      this.protocolError({\n        type: ProtocolError.RetriesExceeded,\n        message: errMsg,\n      });\n\n      return;\n    }\n\n    const backoffMs = this.retryBudget.getBackoffMs();\n\n    this.log?.info(\n      `attempting connection to ${to} (${backoffMs}ms backoff)`,\n      session.loggingMetadata,\n    );\n\n    this.retryBudget.consumeBudget();\n    const backingOffSession =\n      ClientSessionStateGraph.transition.NoConnectionToBackingOff(\n        session,\n        backoffMs,\n        {\n          onBackoffFinished: () => {\n            this.onBackoffFinished(backingOffSession);\n          },\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(backingOffSession);\n          },\n        },\n      );\n\n    this.updateSession(backingOffSession);\n  }\n\n  protected onBackoffFinished(session: SessionBackingOff) {\n    const connPromise = tracer.startActiveSpan('connect', async (span) => {\n      try {\n        return await this.createNewOutgoingConnection(session.to);\n      } catch (err) {\n        // rethrow the error so that the promise is rejected\n        // as it was before we wrapped it in a span\n        const errStr = coerceErrorString(err);\n        span.recordException(errStr);\n        span.setStatus({ code: SpanStatusCode.ERROR });\n        throw err;\n      } finally {\n        span.end();\n      }\n    });\n\n    // transition to connecting\n    const connectingSession =\n      ClientSessionStateGraph.transition.BackingOffToConnecting(\n        session,\n        connPromise,\n        {\n          onConnectionEstablished: (conn) => {\n            this.log?.debug(\n              `connection to ${connectingSession.to} established`,\n              {\n                ...conn.loggingMetadata,\n                ...connectingSession.loggingMetadata,\n              },\n            );\n\n            // cast here because conn can't be narrowed to ConnType\n            // in the callback due to variance rules\n            this.onConnectionEstablished(connectingSession, conn as ConnType);\n          },\n          onConnectionFailed: (error: unknown) => {\n            const errStr = coerceErrorString(error);\n            this.log?.error(\n              `error connecting to ${connectingSession.to}: ${errStr}`,\n              connectingSession.loggingMetadata,\n            );\n            this.onConnectingFailed(connectingSession);\n          },\n          onConnectionTimeout: () => {\n            this.log?.error(\n              `connection to ${connectingSession.to} timed out`,\n              connectingSession.loggingMetadata,\n            );\n            this.onConnectingFailed(connectingSession);\n          },\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(connectingSession);\n          },\n        },\n      );\n\n    this.updateSession(connectingSession);\n  }\n\n  private async sendHandshake(session: SessionHandshaking<ConnType>) {\n    let metadata: unknown = undefined;\n\n    if (this.handshakeExtensions) {\n      metadata = await this.handshakeExtensions.construct();\n    }\n\n    // double-check to make sure we haven't transitioned the session yet\n    if (session._isConsumed) {\n      // bail out, don't need to do anything\n      return;\n    }\n\n    const requestMsg = handshakeRequestMessage({\n      from: this.clientId,\n      to: session.to,\n      sessionId: session.id,\n      expectedSessionState: {\n        nextExpectedSeq: session.ack,\n        nextSentSeq: session.nextSeq(),\n      },\n      metadata,\n      tracing: getPropagationContext(session.telemetry.ctx),\n    });\n\n    this.log?.debug(`sending handshake request to ${session.to}`, {\n      ...session.loggingMetadata,\n      transportMessage: requestMsg,\n    });\n\n    session.sendHandshake(requestMsg);\n  }\n\n  close() {\n    this.retryBudget.close();\n    super.close();\n  }\n}\n","/**\n * Options to control the backoff and retry behavior of the client transport's connection behaviour.\n *\n * River implements exponential backoff with jitter to prevent flooding the server\n * when there's an issue with connection establishment.\n *\n * The backoff is calculated via the following:\n *   backOff = min(jitter + {@link baseIntervalMs} * 2 ^ budget_consumed, {@link maxBackoffMs})\n *\n * We use a leaky bucket rate limit with a budget of {@link attemptBudgetCapacity} reconnection attempts.\n * Budget only starts to restore after a successful handshake at a rate of one budget per {@link budgetRestoreIntervalMs}.\n */\nexport interface ConnectionRetryOptions {\n  /**\n   * The base interval to wait before retrying a connection.\n   */\n  baseIntervalMs: number;\n\n  /**\n   * The maximum random jitter to add to the total backoff time.\n   */\n  maxJitterMs: number;\n\n  /**\n   * The maximum amount of time to wait before retrying a connection.\n   * This does not include the jitter.\n   */\n  maxBackoffMs: number;\n\n  /**\n   * The max number of times to attempt a connection before a successful handshake.\n   * This persists across connections but starts restoring budget after a successful handshake.\n   * The restoration interval depends on {@link budgetRestoreIntervalMs}\n   */\n  attemptBudgetCapacity: number;\n\n  /**\n   * After a successful connection attempt, how long to wait before we restore a single budget.\n   */\n  budgetRestoreIntervalMs: number;\n}\n\nexport class LeakyBucketRateLimit {\n  private budgetConsumed: number;\n  private intervalHandle?: ReturnType<typeof setInterval>;\n  private readonly options: ConnectionRetryOptions;\n\n  constructor(options: ConnectionRetryOptions) {\n    this.options = options;\n    this.budgetConsumed = 0;\n  }\n\n  getBackoffMs() {\n    if (this.getBudgetConsumed() === 0) {\n      return 0;\n    }\n\n    const exponent = Math.max(0, this.getBudgetConsumed() - 1);\n    const jitter = Math.floor(Math.random() * this.options.maxJitterMs);\n    const backoffMs = Math.min(\n      this.options.baseIntervalMs * 2 ** exponent,\n      this.options.maxBackoffMs,\n    );\n\n    return backoffMs + jitter;\n  }\n\n  get totalBudgetRestoreTime() {\n    return (\n      this.options.budgetRestoreIntervalMs * this.options.attemptBudgetCapacity\n    );\n  }\n\n  consumeBudget() {\n    // If we're consuming again, let's ensure that we're not leaking\n    this.stopLeak();\n    this.budgetConsumed = this.getBudgetConsumed() + 1;\n  }\n\n  getBudgetConsumed() {\n    return this.budgetConsumed;\n  }\n\n  hasBudget() {\n    return this.getBudgetConsumed() < this.options.attemptBudgetCapacity;\n  }\n\n  startRestoringBudget() {\n    if (this.intervalHandle) {\n      return;\n    }\n\n    const restoreBudgetForUser = () => {\n      const currentBudget = this.budgetConsumed;\n      if (!currentBudget) {\n        this.stopLeak();\n\n        return;\n      }\n\n      const newBudget = currentBudget - 1;\n      if (newBudget === 0) {\n        return;\n      }\n\n      this.budgetConsumed = newBudget;\n    };\n\n    this.intervalHandle = setInterval(\n      restoreBudgetForUser,\n      this.options.budgetRestoreIntervalMs,\n    );\n  }\n\n  private stopLeak() {\n    if (!this.intervalHandle) {\n      return;\n    }\n\n    clearInterval(this.intervalHandle);\n    this.intervalHandle = undefined;\n  }\n\n  close() {\n    this.stopLeak();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,SAAS,sBAAsB;;;AC0CxB,IAAM,uBAAN,MAA2B;AAAA,EACxB;AAAA,EACA;AAAA,EACS;AAAA,EAEjB,YAAY,SAAiC;AAC3C,SAAK,UAAU;AACf,SAAK,iBAAiB;AAAA,EACxB;AAAA,EAEA,eAAe;AACb,QAAI,KAAK,kBAAkB,MAAM,GAAG;AAClC,aAAO;AAAA,IACT;AAEA,UAAM,WAAW,KAAK,IAAI,GAAG,KAAK,kBAAkB,IAAI,CAAC;AACzD,UAAM,SAAS,KAAK,MAAM,KAAK,OAAO,IAAI,KAAK,QAAQ,WAAW;AAClE,UAAM,YAAY,KAAK;AAAA,MACrB,KAAK,QAAQ,iBAAiB,KAAK;AAAA,MACnC,KAAK,QAAQ;AAAA,IACf;AAEA,WAAO,YAAY;AAAA,EACrB;AAAA,EAEA,IAAI,yBAAyB;AAC3B,WACE,KAAK,QAAQ,0BAA0B,KAAK,QAAQ;AAAA,EAExD;AAAA,EAEA,gBAAgB;AAEd,SAAK,SAAS;AACd,SAAK,iBAAiB,KAAK,kBAAkB,IAAI;AAAA,EACnD;AAAA,EAEA,oBAAoB;AAClB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,YAAY;AACV,WAAO,KAAK,kBAAkB,IAAI,KAAK,QAAQ;AAAA,EACjD;AAAA,EAEA,uBAAuB;AACrB,QAAI,KAAK,gBAAgB;AACvB;AAAA,IACF;AAEA,UAAM,uBAAuB,MAAM;AACjC,YAAM,gBAAgB,KAAK;AAC3B,UAAI,CAAC,eAAe;AAClB,aAAK,SAAS;AAEd;AAAA,MACF;AAEA,YAAM,YAAY,gBAAgB;AAClC,UAAI,cAAc,GAAG;AACnB;AAAA,MACF;AAEA,WAAK,iBAAiB;AAAA,IACxB;AAEA,SAAK,iBAAiB;AAAA,MACpB;AAAA,MACA,KAAK,QAAQ;AAAA,IACf;AAAA,EACF;AAAA,EAEQ,WAAW;AACjB,QAAI,CAAC,KAAK,gBAAgB;AACxB;AAAA,IACF;AAEA,kBAAc,KAAK,cAAc;AACjC,SAAK,iBAAiB;AAAA,EACxB;AAAA,EAEA,QAAQ;AACN,SAAK,SAAS;AAAA,EAChB;AACF;;;AD3GA,SAAS,aAAa;AAef,IAAe,kBAAf,cAEG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA,EAEV;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,4BAA4B;AAAA;AAAA;AAAA;AAAA,EAK5B;AAAA,EAEA;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,WAAW,oBAAI,IAAI;AACxB,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,cAAc,IAAI,qBAAqB,KAAK,OAAO;AAAA,EAC1D;AAAA,EAEA,gBAAgB,SAAiC;AAC/C,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAYQ,gBAAgB,IAAuB;AAC7C,UAAM,aAAa,KAAK,SAAS,IAAI,EAAE;AACvC,QAAI,CAAC,KAAK,QAAQ,sCAAsC,YAAY;AAClE,WAAK,cAAc,UAAU;AAAA,IAC/B;AAEA,QAAI,KAAK,6BAA6B,KAAK,UAAU,MAAM,QAAQ;AACjE,WAAK,QAAQ,EAAE;AAAA,IACjB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,yBAAyB,IAAiC;AACxD,UAAM,UAAU,wBAAwB;AAAA,MACtC;AAAA,MACA,KAAK;AAAA,MACL;AAAA,QACE,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,OAAO;AAAA,QAC1C;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,IACP;AAEA,SAAK,cAAc,OAAO;AAE1B,WAAO;AAAA,EACT;AAAA;AAAA,EAGU,mBAAmB,SAAsC;AACjE,UAAM,sBAAsB,MAAM,mBAAmB,OAAO;AAC5D,SAAK,gBAAgB,oBAAoB,EAAE;AAE3C,WAAO;AAAA,EACT;AAAA,EAEU,aACR,SACA;AACA,UAAM,sBAAsB,MAAM,aAAa,OAAO;AACtD,SAAK,gBAAgB,oBAAoB,EAAE;AAE3C,WAAO;AAAA,EACT;AAAA,EAEU,wBACR,SACA,MAC8B;AAE9B,UAAM,qBACJ,wBAAwB,WAAW;AAAA,MACjC;AAAA,MACA;AAAA,MACA;AAAA,QACE,qBAAqB,CAAC,QAAQ;AAE5B,gBAAM,SAAS,kBAAkB,GAAG;AACpC,eAAK,KAAK;AAAA,YACR,iBAAiB,mBAAmB,EAAE,8BAA8B,MAAM;AAAA,YAC1E,mBAAmB;AAAA,UACrB;AAAA,QACF;AAAA,QACA,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR,iBAAiB,mBAAmB,EAAE;AAAA,YACtC,mBAAmB;AAAA,UACrB;AACA,eAAK,aAAa,kBAAkB;AAAA,QACtC;AAAA,QACA,aAAa,CAAC,QAAQ;AACpB,eAAK,oBAAoB,oBAAoB,GAAG;AAAA,QAClD;AAAA,QACA,oBAAoB,CAAC,QAAQ,SAAS;AACpC,eAAK,KAAK;AAAA,YACR,sBAAsB,MAAM;AAAA,YAC5B,mBAAmB;AAAA,UACrB;AACA,eAAK,cAAc,SAAS,EAAE,WAAW,KAAK,CAAC;AAC/C,eAAK,cAAc;AAAA,YACjB,MAAM,cAAc;AAAA,YACpB;AAAA,YACA,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,QACA,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR,iBAAiB,mBAAmB,EAAE;AAAA,YACtC,mBAAmB;AAAA,UACrB;AACA,eAAK,aAAa,kBAAkB;AAAA,QACtC;AAAA,QACA,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,kBAAkB;AAAA,QACrD;AAAA,MACF;AAAA,IACF;AAEF,SAAK,cAAc,kBAAkB;AACrC,SAAK,KAAK,cAAc,kBAAkB;AAE1C,WAAO;AAAA,EACT;AAAA,EAEQ,wBACN,SACA,QACA,UACA;AACA,YAAQ,KAAK,WAAW,KAAK,UAAU;AAAA,MACrC,MAAM,eAAe;AAAA,MACrB,SAAS;AAAA,IACX,CAAC;AAED,SAAK,KAAK,KAAK,QAAQ,QAAQ;AAC/B,SAAK,cAAc,SAAS,EAAE,WAAW,KAAK,CAAC;AAAA,EACjD;AAAA,EAEU,oBACR,SACA,KACA;AAEA,QAAI,CAAC,MAAM,MAAM,uCAAuC,IAAI,OAAO,GAAG;AACpE,YAAM,SAAS;AACf,WAAK,wBAAwB,SAAS,QAAQ;AAAA,QAC5C,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,QAClB,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,uCAAuC,IAAI,OAAO;AAAA,QACpE;AAAA,MACF,CAAC;AAED;AAAA,IACF;AAGA,QAAI,CAAC,IAAI,QAAQ,OAAO,IAAI;AAC1B,YAAM,YAAY,MAAM;AAAA,QACtB;AAAA,QACA,IAAI,QAAQ,OAAO;AAAA,MACrB;AAEA,YAAM,SAAS,qBAAqB,IAAI,QAAQ,OAAO,MAAM;AAC7D,YAAM,KAAK,QAAQ;AACnB,WAAK,wBAAwB,SAAS,QAAQ;AAAA,QAC5C,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,MACpB,CAAC;AAED,UAAI,WAAW;AACb,aAAK,gBAAgB,EAAE;AAAA,MACzB,OAAO;AACL,aAAK,cAAc;AAAA,UACjB,MAAM,cAAc;AAAA,UACpB,MAAM,IAAI,QAAQ,OAAO;AAAA,UACzB,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA;AAAA,IACF;AAGA,QAAI,IAAI,QAAQ,OAAO,cAAc,QAAQ,IAAI;AAC/C,YAAM,SAAS,iCAAiC,QAAQ,EAAE,SAAS,IAAI,QAAQ,OAAO,SAAS;AAC/F,WAAK,wBAAwB,SAAS,QAAQ;AAAA,QAC5C,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,MACpB,CAAC;AAED;AAAA,IACF;AAGA,SAAK,KAAK,KAAK,kBAAkB,IAAI,IAAI,OAAO;AAAA,MAC9C,GAAG,QAAQ;AAAA,MACX,kBAAkB;AAAA,IACpB,CAAC;AAED,UAAM,mBACJ,wBAAwB,WAAW,uBAAuB,SAAS;AAAA,MACjE,qBAAqB,CAAC,QAAQ;AAE5B,cAAM,SAAS,kBAAkB,GAAG;AACpC,aAAK,KAAK;AAAA,UACR,iBAAiB,iBAAiB,EAAE,aAAa,MAAM;AAAA,UACvD,iBAAiB;AAAA,QACnB;AAAA,MACF;AAAA,MACA,oBAAoB,MAAM;AACxB,aAAK,KAAK;AAAA,UACR,iBAAiB,iBAAiB,EAAE;AAAA,UACpC,iBAAiB;AAAA,QACnB;AACA,aAAK,aAAa,gBAAgB;AAAA,MACpC;AAAA,MACA,WAAW,CAACA,SAAQ;AAClB,aAAK,UAAUA,IAAG;AAAA,MACpB;AAAA,MACA,kBAAkB,CAAC,WAAW;AAC5B,aAAK,cAAc,kBAAkB,EAAE,WAAW,KAAK,CAAC;AACxD,aAAK,cAAc;AAAA,UACjB,MAAM,cAAc;AAAA,UACpB,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAEH,SAAK,cAAc,gBAAgB;AACnC,SAAK,YAAY,qBAAqB;AAAA,EACxC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,QAAQ,IAAuB;AAC7B,QAAI,KAAK,UAAU,MAAM,QAAQ;AAC/B,WAAK,KAAK;AAAA,QACR,uEAAuE,EAAE;AAAA,MAC3E;AAEA;AAAA,IACF;AAEA,UAAM,UAAU,KAAK,SAAS,IAAI,EAAE,KAAK,KAAK,yBAAyB,EAAE;AACzE,QAAI,QAAQ,6CAAqC;AAE/C,WAAK,KAAK;AAAA,QACR,cAAc,EAAE,cAAc,QAAQ,KAAK;AAAA,QAC3C,QAAQ;AAAA,MACV;AAEA;AAAA,IACF;AAGA,QAAI,CAAC,KAAK,YAAY,UAAU,GAAG;AACjC,YAAM,iBAAiB,KAAK,YAAY,kBAAkB;AAC1D,YAAM,SAAS,uBAAuB,EAAE,yCAAyC,cAAc,yBAAyB,KAAK,YAAY,sBAAsB;AAC/J,WAAK,KAAK,MAAM,QAAQ,QAAQ,eAAe;AAC/C,WAAK,cAAc;AAAA,QACjB,MAAM,cAAc;AAAA,QACpB,SAAS;AAAA,MACX,CAAC;AAED;AAAA,IACF;AAEA,UAAM,YAAY,KAAK,YAAY,aAAa;AAEhD,SAAK,KAAK;AAAA,MACR,4BAA4B,EAAE,KAAK,SAAS;AAAA,MAC5C,QAAQ;AAAA,IACV;AAEA,SAAK,YAAY,cAAc;AAC/B,UAAM,oBACJ,wBAAwB,WAAW;AAAA,MACjC;AAAA,MACA;AAAA,MACA;AAAA,QACE,mBAAmB,MAAM;AACvB,eAAK,kBAAkB,iBAAiB;AAAA,QAC1C;AAAA,QACA,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,iBAAiB;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAEF,SAAK,cAAc,iBAAiB;AAAA,EACtC;AAAA,EAEU,kBAAkB,SAA4B;AACtD,UAAM,cAAc,gBAAO,gBAAgB,WAAW,OAAO,SAAS;AACpE,UAAI;AACF,eAAO,MAAM,KAAK,4BAA4B,QAAQ,EAAE;AAAA,MAC1D,SAAS,KAAK;AAGZ,cAAM,SAAS,kBAAkB,GAAG;AACpC,aAAK,gBAAgB,MAAM;AAC3B,aAAK,UAAU,EAAE,MAAM,eAAe,MAAM,CAAC;AAC7C,cAAM;AAAA,MACR,UAAE;AACA,aAAK,IAAI;AAAA,MACX;AAAA,IACF,CAAC;AAGD,UAAM,oBACJ,wBAAwB,WAAW;AAAA,MACjC;AAAA,MACA;AAAA,MACA;AAAA,QACE,yBAAyB,CAAC,SAAS;AACjC,eAAK,KAAK;AAAA,YACR,iBAAiB,kBAAkB,EAAE;AAAA,YACrC;AAAA,cACE,GAAG,KAAK;AAAA,cACR,GAAG,kBAAkB;AAAA,YACvB;AAAA,UACF;AAIA,eAAK,wBAAwB,mBAAmB,IAAgB;AAAA,QAClE;AAAA,QACA,oBAAoB,CAAC,UAAmB;AACtC,gBAAM,SAAS,kBAAkB,KAAK;AACtC,eAAK,KAAK;AAAA,YACR,uBAAuB,kBAAkB,EAAE,KAAK,MAAM;AAAA,YACtD,kBAAkB;AAAA,UACpB;AACA,eAAK,mBAAmB,iBAAiB;AAAA,QAC3C;AAAA,QACA,qBAAqB,MAAM;AACzB,eAAK,KAAK;AAAA,YACR,iBAAiB,kBAAkB,EAAE;AAAA,YACrC,kBAAkB;AAAA,UACpB;AACA,eAAK,mBAAmB,iBAAiB;AAAA,QAC3C;AAAA,QACA,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,iBAAiB;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAEF,SAAK,cAAc,iBAAiB;AAAA,EACtC;AAAA,EAEA,MAAc,cAAc,SAAuC;AACjE,QAAI,WAAoB;AAExB,QAAI,KAAK,qBAAqB;AAC5B,iBAAW,MAAM,KAAK,oBAAoB,UAAU;AAAA,IACtD;AAGA,QAAI,QAAQ,aAAa;AAEvB;AAAA,IACF;AAEA,UAAM,aAAa,wBAAwB;AAAA,MACzC,MAAM,KAAK;AAAA,MACX,IAAI,QAAQ;AAAA,MACZ,WAAW,QAAQ;AAAA,MACnB,sBAAsB;AAAA,QACpB,iBAAiB,QAAQ;AAAA,QACzB,aAAa,QAAQ,QAAQ;AAAA,MAC/B;AAAA,MACA;AAAA,MACA,SAAS,sBAAsB,QAAQ,UAAU,GAAG;AAAA,IACtD,CAAC;AAED,SAAK,KAAK,MAAM,gCAAgC,QAAQ,EAAE,IAAI;AAAA,MAC5D,GAAG,QAAQ;AAAA,MACX,kBAAkB;AAAA,IACpB,CAAC;AAED,YAAQ,cAAc,UAAU;AAAA,EAClC;AAAA,EAEA,QAAQ;AACN,SAAK,YAAY,MAAM;AACvB,UAAM,MAAM;AAAA,EACd;AACF;","names":["msg"]}

package/dist/{chunk-4PVU7J25.js → chunk-AJGIY2UB.js}

@@ -53,4 +53,4 @@ var NaiveJsonCodec = {
 export {
   NaiveJsonCodec
 };
-//# sourceMappingURL=chunk-4PVU7J25.js.map
+//# sourceMappingURL=chunk-AJGIY2UB.js.map

package/dist/chunk-AJGIY2UB.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../codec/json.ts"],"sourcesContent":["import { Codec } from './types';\n\nconst encoder = new TextEncoder();\nconst decoder = new TextDecoder();\n\n// Convert Uint8Array to base64\nfunction uint8ArrayToBase64(uint8Array: Uint8Array) {\n  let binary = '';\n  uint8Array.forEach((byte) => {\n    binary += String.fromCharCode(byte);\n  });\n\n  return btoa(binary);\n}\n\n// Convert base64 to Uint8Array\nfunction base64ToUint8Array(base64: string) {\n  const binaryString = atob(base64);\n  const uint8Array = new Uint8Array(binaryString.length);\n  for (let i = 0; i < binaryString.length; i++) {\n    uint8Array[i] = binaryString.charCodeAt(i);\n  }\n\n  return uint8Array;\n}\n\ninterface Base64EncodedValue {\n  $t: string;\n}\n\n/**\n * Naive JSON codec implementation using JSON.stringify and JSON.parse.\n * @type {Codec}\n */\nexport const NaiveJsonCodec: Codec = {\n  toBuffer: (obj: object) => {\n    return encoder.encode(\n      JSON.stringify(obj, function replacer<\n        T extends object,\n      >(this: T, key: keyof T) {\n        const val = this[key];\n        if (val instanceof Uint8Array) {\n          return { $t: uint8ArrayToBase64(val) } satisfies Base64EncodedValue;\n        } else {\n          return val;\n        }\n      }),\n    );\n  },\n  fromBuffer: (buff: Uint8Array) => {\n    try {\n      const parsed = JSON.parse(\n        decoder.decode(buff),\n        function reviver(_key, val: unknown) {\n          if ((val as Base64EncodedValue | undefined)?.$t) {\n            return base64ToUint8Array((val as Base64EncodedValue).$t);\n          } else {\n            return val;\n          }\n        },\n      ) as unknown;\n\n      if (typeof parsed === 'object') return parsed;\n\n      return null;\n    } catch {\n      return null;\n    }\n  },\n};\n"],"mappings":";AAEA,IAAM,UAAU,IAAI,YAAY;AAChC,IAAM,UAAU,IAAI,YAAY;AAGhC,SAAS,mBAAmB,YAAwB;AAClD,MAAI,SAAS;AACb,aAAW,QAAQ,CAAC,SAAS;AAC3B,cAAU,OAAO,aAAa,IAAI;AAAA,EACpC,CAAC;AAED,SAAO,KAAK,MAAM;AACpB;AAGA,SAAS,mBAAmB,QAAgB;AAC1C,QAAM,eAAe,KAAK,MAAM;AAChC,QAAM,aAAa,IAAI,WAAW,aAAa,MAAM;AACrD,WAAS,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;AAC5C,eAAW,CAAC,IAAI,aAAa,WAAW,CAAC;AAAA,EAC3C;AAEA,SAAO;AACT;AAUO,IAAM,iBAAwB;AAAA,EACnC,UAAU,CAAC,QAAgB;AACzB,WAAO,QAAQ;AAAA,MACb,KAAK,UAAU,KAAK,SAAS,SAElB,KAAc;AACvB,cAAM,MAAM,KAAK,GAAG;AACpB,YAAI,eAAe,YAAY;AAC7B,iBAAO,EAAE,IAAI,mBAAmB,GAAG,EAAE;AAAA,QACvC,OAAO;AACL,iBAAO;AAAA,QACT;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EACA,YAAY,CAAC,SAAqB;AAChC,QAAI;AACF,YAAM,SAAS,KAAK;AAAA,QAClB,QAAQ,OAAO,IAAI;AAAA,QACnB,SAAS,QAAQ,MAAM,KAAc;AACnC,cAAK,KAAwC,IAAI;AAC/C,mBAAO,mBAAoB,IAA2B,EAAE;AAAA,UAC1D,OAAO;AACL,mBAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF;AAEA,UAAI,OAAO,WAAW;AAAU,eAAO;AAEvC,aAAO;AAAA,IACT,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AACF;","names":[]}

package/dist/{chunk-EETL2L77.js → chunk-GJUUVID2.js}

@@ -35,6 +35,9 @@ var ControlMessageCloseSchema = Type.Object({
 });
 var currentProtocolVersion = "v2.0";
 var acceptedProtocolVersions = ["v1.1", currentProtocolVersion];
+function isAcceptedProtocolVersion(version2) {
+  return acceptedProtocolVersions.includes(version2);
+}
 var ControlMessageHandshakeRequestSchema = Type.Object({
   type: Type.Literal("HANDSHAKE_REQ"),
   protocolVersion: Type.String(),
@@ -47,10 +50,7 @@ var ControlMessageHandshakeRequestSchema = Type.Object({
   expectedSessionState: Type.Object({
     // what the client expects the server to send next
     nextExpectedSeq: Type.Integer(),
-    // TODO: remove optional once we know all servers
-    // are nextSentSeq here
-    // what the server expects the client to send next
-    nextSentSeq: Type.Optional(Type.Integer())
+    nextSentSeq: Type.Integer()
   }),
   metadata: Type.Optional(Type.Unknown())
 });
@@ -86,9 +86,7 @@ var ControlMessageHandshakeResponseSchema = Type.Object({
     Type.Object({
       ok: Type.Literal(false),
       reason: Type.String(),
-      // TODO: remove optional once we know all servers
-      // are sending code here
-      code: Type.Optional(HandshakeErrorResponseCodes)
+      code: HandshakeErrorResponseCodes
     })
   ])
 });
@@ -155,19 +153,10 @@ function closeStreamMessage(streamId) {
     }
   };
 }
-function requestCloseStreamMessage(streamId) {
-  return {
-    streamId,
-    controlFlags: 16 /* StreamCloseRequestBit */,
-    payload: {
-      type: "CLOSE"
-    }
-  };
-}
-function abortMessage(streamId, payload) {
+function cancelMessage(streamId, payload) {
   return {
     streamId,
-    controlFlags: 4 /* StreamAbortBit */,
+    controlFlags: 4 /* StreamCancelBit */,
     payload
   };
 }
@@ -186,21 +175,15 @@ function isStreamClose(controlFlag) {
     (controlFlag & 8 /* StreamClosedBit */) === 8 /* StreamClosedBit */
   );
 }
-function isStreamCloseRequest(controlFlag) {
-  return (
-    /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */
-    (controlFlag & 16 /* StreamCloseRequestBit */) === 16 /* StreamCloseRequestBit */
-  );
-}
-function isStreamAbort(controlFlag) {
+function isStreamCancel(controlFlag) {
   return (
     /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */
-    (controlFlag & 4 /* StreamAbortBit */) === 4 /* StreamAbortBit */
+    (controlFlag & 4 /* StreamCancelBit */) === 4 /* StreamCancelBit */
   );
 }
 
 // package.json
-var version = "0.200.0-rc.9";
+var version = "0.200.2";
 
 // util/stringify.ts
 function coerceErrorString(err) {
@@ -301,6 +284,7 @@ export {
   ControlMessageCloseSchema,
   currentProtocolVersion,
   acceptedProtocolVersions,
+  isAcceptedProtocolVersion,
   ControlMessageHandshakeRequestSchema,
   HandshakeErrorRetriableResponseCodes,
   HandshakeErrorCustomHandlerFatalResponseCodes,
@@ -310,13 +294,11 @@ export {
   handshakeRequestMessage,
   handshakeResponseMessage,
   closeStreamMessage,
-  requestCloseStreamMessage,
-  abortMessage,
+  cancelMessage,
   isAck,
   isStreamOpen,
   isStreamClose,
-  isStreamCloseRequest,
-  isStreamAbort,
+  isStreamCancel,
   version,
   getPropagationContext,
   createSessionTelemetryInfo,
@@ -325,4 +307,4 @@ export {
   tracing_default,
   coerceErrorString
 };
-//# sourceMappingURL=chunk-EETL2L77.js.map
+//# sourceMappingURL=chunk-GJUUVID2.js.map

package/dist/chunk-GJUUVID2.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/message.ts","../transport/id.ts","../package.json","../util/stringify.ts","../tracing/index.ts"],"sourcesContent":["import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { PropagationContext } from '../tracing';\nimport { generateId } from './id';\nimport { ErrResult, ReaderErrorSchema } from '../router';\n\n/**\n * Control flags for transport messages.\n */\nexport const enum ControlFlags {\n  /**\n   * Used in heartbeat messages.\n   */\n  AckBit = 0b00001,\n  /**\n   * Used in stream open requests.\n   */\n  StreamOpenBit = 0b00010,\n  /**\n   * Used when a stream is cancelled due errors or to explicit cancellation\n   */\n  StreamCancelBit = 0b00100,\n  /**\n   * Used when writer closes the stream.\n   */\n  StreamClosedBit = 0b01000,\n}\n\n/**\n * Generic Typebox schema for a transport message.\n * @template T The type of the payload.\n * @param {T} t The payload schema.\n * @returns The transport message schema.\n */\nexport const TransportMessageSchema = <T extends TSchema>(t: T) =>\n  Type.Object({\n    id: Type.String(),\n    from: Type.String(),\n    to: Type.String(),\n    seq: Type.Integer(),\n    ack: Type.Integer(),\n    serviceName: Type.Optional(Type.String()),\n    procedureName: Type.Optional(Type.String()),\n    streamId: Type.String(),\n    controlFlags: Type.Integer(),\n    tracing: Type.Optional(\n      Type.Object({\n        traceparent: Type.String(),\n        tracestate: Type.String(),\n      }),\n    ),\n    payload: t,\n  });\n\n/**\n * Defines the schema for a transport acknowledgement message. This is never constructed manually\n * and is only used internally by the library for tracking inflight messages.\n * @returns The transport message schema.\n */\nexport const ControlMessageAckSchema = Type.Object({\n  type: Type.Literal('ACK'),\n});\n\n/**\n * Defines the schema for a transport close message. This is never constructed manually and is only\n * used internally by the library for closing and cleaning up streams.\n */\nexport const ControlMessageCloseSchema = Type.Object({\n  type: Type.Literal('CLOSE'),\n});\n\nexport type ProtocolVersion = 'v1.1' | 'v2.0';\nexport const currentProtocolVersion = 'v2.0' satisfies ProtocolVersion;\nexport const acceptedProtocolVersions = ['v1.1', currentProtocolVersion];\nexport function isAcceptedProtocolVersion(\n  version: string,\n): version is ProtocolVersion {\n  return acceptedProtocolVersions.includes(version);\n}\n\nexport const ControlMessageHandshakeRequestSchema = Type.Object({\n  type: Type.Literal('HANDSHAKE_REQ'),\n  protocolVersion: Type.String(),\n  sessionId: Type.String(),\n  /**\n   * Specifies what the server's expected session state (from the pov of the client). This can be\n   * used by the server to know whether this is a new or a reestablished connection, and whether it\n   * is compatible with what it already has.\n   */\n  expectedSessionState: Type.Object({\n    // what the client expects the server to send next\n    nextExpectedSeq: Type.Integer(),\n    nextSentSeq: Type.Integer(),\n  }),\n\n  metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const HandshakeErrorRetriableResponseCodes = Type.Union([\n  Type.Literal('SESSION_STATE_MISMATCH'),\n]);\n\nexport const HandshakeErrorCustomHandlerFatalResponseCodes = Type.Union([\n  // The custom validation handler rejected the handler because the client is unsupported.\n  Type.Literal('REJECTED_UNSUPPORTED_CLIENT'),\n  // The custom validation handler rejected the handshake.\n  Type.Literal('REJECTED_BY_CUSTOM_HANDLER'),\n]);\n\nexport const HandshakeErrorFatalResponseCodes = Type.Union([\n  HandshakeErrorCustomHandlerFatalResponseCodes,\n  // The ciient sent a handshake that doesn't comply with the extended handshake metadata.\n  Type.Literal('MALFORMED_HANDSHAKE_META'),\n  // The ciient sent a handshake that doesn't comply with ControlMessageHandshakeRequestSchema.\n  Type.Literal('MALFORMED_HANDSHAKE'),\n  // The client's protocol version does not match the server's.\n  Type.Literal('PROTOCOL_VERSION_MISMATCH'),\n]);\n\nexport const HandshakeErrorResponseCodes = Type.Union([\n  HandshakeErrorRetriableResponseCodes,\n  HandshakeErrorFatalResponseCodes,\n]);\n\nexport const ControlMessageHandshakeResponseSchema = Type.Object({\n  type: Type.Literal('HANDSHAKE_RESP'),\n  status: Type.Union([\n    Type.Object({\n      ok: Type.Literal(true),\n      sessionId: Type.String(),\n    }),\n    Type.Object({\n      ok: Type.Literal(false),\n      reason: Type.String(),\n      code: HandshakeErrorResponseCodes,\n    }),\n  ]),\n});\n\nexport const ControlMessagePayloadSchema = Type.Union([\n  ControlMessageCloseSchema,\n  ControlMessageAckSchema,\n  ControlMessageHandshakeRequestSchema,\n  ControlMessageHandshakeResponseSchema,\n]);\n\n/**\n * Defines the schema for an opaque transport message that is agnostic to any\n * procedure/service.\n * @returns The transport message schema.\n */\nexport const OpaqueTransportMessageSchema = TransportMessageSchema(\n  Type.Unknown(),\n);\n\n/**\n * Represents a transport message. This is the same type as {@link TransportMessageSchema} but\n * we can't statically infer generics from generic Typebox schemas so we have to define it again here.\n *\n * TypeScript can't enforce types when a bitmask is involved, so these are the semantics of\n * `controlFlags`:\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `streamId` must be set to a unique value\n *   (suggestion: use `nanoid`).\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `serviceName` and `procedureName` must be set.\n * * If `controlFlags & StreamClosedBit == StreamClosedBit` and the kind is `stream` or `subscription`,\n *   `payload` should be discarded (usually contains a control message).\n * * If `controlFlags & AckBit == AckBit`, the message is an explicit acknowledgement message and doesn't\n *   contain any payload that is relevant to the application so should not be delivered.\n * @template Payload The type of the payload.\n */\nexport interface TransportMessage<Payload = unknown> {\n  id: string;\n  from: TransportClientId;\n  to: TransportClientId;\n  seq: number;\n  ack: number;\n  serviceName?: string;\n  procedureName?: string;\n  streamId: string;\n  controlFlags: number;\n  tracing?: PropagationContext;\n  payload: Payload;\n}\n\nexport type PartialTransportMessage<Payload = unknown> = Omit<\n  TransportMessage<Payload>,\n  'id' | 'from' | 'to' | 'seq' | 'ack'\n>;\n\nexport function handshakeRequestMessage({\n  from,\n  to,\n  sessionId,\n  expectedSessionState,\n  metadata,\n  tracing,\n}: {\n  from: TransportClientId;\n  to: TransportClientId;\n  sessionId: string;\n  expectedSessionState: Static<\n    typeof ControlMessageHandshakeRequestSchema\n  >['expectedSessionState'];\n  metadata?: unknown;\n  tracing?: PropagationContext;\n}): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n  return {\n    id: generateId(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: generateId(),\n    controlFlags: 0,\n    tracing,\n    payload: {\n      type: 'HANDSHAKE_REQ',\n      protocolVersion: currentProtocolVersion,\n      sessionId,\n      expectedSessionState,\n      metadata,\n    } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n  };\n}\n\n/**\n * This is a reason that can be given during the handshake to indicate that the peer has the wrong\n * session state.\n */\nexport const SESSION_STATE_MISMATCH = 'session state mismatch';\n\nexport function handshakeResponseMessage({\n  from,\n  to,\n  status,\n}: {\n  from: TransportClientId;\n  to: TransportClientId;\n  status: Static<typeof ControlMessageHandshakeResponseSchema>['status'];\n}): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n  return {\n    id: generateId(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: generateId(),\n    controlFlags: 0,\n    payload: {\n      type: 'HANDSHAKE_RESP',\n      status,\n    } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n  };\n}\n\nexport function closeStreamMessage(streamId: string): PartialTransportMessage {\n  return {\n    streamId,\n    controlFlags: ControlFlags.StreamClosedBit,\n    payload: {\n      type: 'CLOSE' as const,\n    } satisfies Static<typeof ControlMessagePayloadSchema>,\n  };\n}\n\nexport function cancelMessage(\n  streamId: string,\n  payload: ErrResult<Static<typeof ReaderErrorSchema>>,\n) {\n  return {\n    streamId,\n    controlFlags: ControlFlags.StreamCancelBit,\n    payload,\n  };\n}\n\n/**\n * A type alias for a transport message with an opaque payload.\n * @template T - The type of the opaque payload.\n */\nexport type OpaqueTransportMessage = TransportMessage;\nexport type TransportClientId = string;\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is an ack message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the AckBit, false otherwise.\n */\nexport function isAck(controlFlag: number): boolean {\n  /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n  return (controlFlag & ControlFlags.AckBit) === ControlFlags.AckBit;\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream open message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamOpenBit, false otherwise.\n */\nexport function isStreamOpen(controlFlag: number): boolean {\n  return (\n    /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n    (controlFlag & ControlFlags.StreamOpenBit) === ControlFlags.StreamOpenBit\n  );\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream close message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamCloseBit, false otherwise.\n */\nexport function isStreamClose(controlFlag: number): boolean {\n  return (\n    /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n    (controlFlag & ControlFlags.StreamClosedBit) ===\n    ControlFlags.StreamClosedBit\n  );\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is an cancel message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the CancelBit, false otherwise\n */\nexport function isStreamCancel(controlFlag: number): boolean {\n  return (\n    /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n    (controlFlag & ControlFlags.StreamCancelBit) ===\n    ControlFlags.StreamCancelBit\n  );\n}\n","import { customAlphabet } from 'nanoid';\n\nconst alphabet = customAlphabet(\n  '1234567890abcdefghijklmnopqrstuvxyzABCDEFGHIJKLMNOPQRSTUVXYZ',\n);\nexport const generateId = () => alphabet(12);\n","{\n  \"name\": \"@replit/river\",\n  \"description\": \"It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!\",\n  \"version\": \"0.200.2\",\n  \"type\": \"module\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/router/index.js\",\n      \"require\": \"./dist/router/index.cjs\"\n    },\n    \"./logging\": {\n      \"import\": \"./dist/logging/index.js\",\n      \"require\": \"./dist/logging/index.cjs\"\n    },\n    \"./codec\": {\n      \"import\": \"./dist/codec/index.js\",\n      \"require\": \"./dist/codec/index.cjs\"\n    },\n    \"./transport\": {\n      \"import\": \"./dist/transport/index.js\",\n      \"require\": \"./dist/transport/index.cjs\"\n    },\n    \"./transport/ws/client\": {\n      \"import\": \"./dist/transport/impls/ws/client.js\",\n      \"require\": \"./dist/transport/impls/ws/client.cjs\"\n    },\n    \"./transport/ws/server\": {\n      \"import\": \"./dist/transport/impls/ws/server.js\",\n      \"require\": \"./dist/transport/impls/ws/server.cjs\"\n    },\n    \"./transport/uds/client\": {\n      \"import\": \"./dist/transport/impls/uds/client.js\",\n      \"require\": \"./dist/transport/impls/uds/client.cjs\"\n    },\n    \"./transport/uds/server\": {\n      \"import\": \"./dist/transport/impls/uds/server.js\",\n      \"require\": \"./dist/transport/impls/uds/server.cjs\"\n    },\n    \"./test-util\": {\n      \"import\": \"./dist/util/testHelpers.js\",\n      \"require\": \"./dist/util/testHelpers.cjs\"\n    }\n  },\n  \"sideEffects\": [\n    \"./dist/logging/index.js\"\n  ],\n  \"files\": [\n    \"dist\"\n  ],\n  \"dependencies\": {\n    \"@msgpack/msgpack\": \"^3.0.0-beta2\",\n    \"nanoid\": \"^4.0.2\",\n    \"ws\": \"^8.17.0\"\n  },\n  \"peerDependencies\": {\n    \"@opentelemetry/api\": \"^1.7.0\",\n    \"@sinclair/typebox\": \"~0.32.8\"\n  },\n  \"devDependencies\": {\n    \"@opentelemetry/core\": \"^1.7.0\",\n    \"@opentelemetry/sdk-trace-base\": \"^1.24.1\",\n    \"@opentelemetry/sdk-trace-web\": \"^1.24.1\",\n    \"@stylistic/eslint-plugin\": \"^2.6.4\",\n    \"@types/ws\": \"^8.5.5\",\n    \"@typescript-eslint/eslint-plugin\": \"^7.8.0\",\n    \"@typescript-eslint/parser\": \"^7.8.0\",\n    \"@vitest/ui\": \"^2.0.5\",\n    \"eslint\": \"^8.57.0\",\n    \"eslint-config-prettier\": \"^9.1.0\",\n    \"eslint-plugin-prettier\": \"^5.1.3\",\n    \"prettier\": \"^3.0.0\",\n    \"tsup\": \"^7.2.0\",\n    \"typescript\": \"^5.4.5\",\n    \"vitest\": \"^2.0.5\"\n  },\n  \"scripts\": {\n    \"check\": \"tsc --noEmit && npm run format && npm run lint\",\n    \"format\": \"npx prettier . --check\",\n    \"format:fix\": \"npx prettier . --write\",\n    \"lint\": \"eslint .\",\n    \"lint:fix\": \"eslint . --fix\",\n    \"fix\": \"npm run format:fix && npm run lint:fix\",\n    \"build\": \"rm -rf dist && tsup && du -sh dist\",\n    \"prepack\": \"npm run build\",\n    \"release\": \"npm publish --access public\",\n    \"test:ui\": \"echo \\\"remember to go to /__vitest__ in the webview\\\" && vitest --ui --api.host 0.0.0.0 --api.port 3000\",\n    \"test\": \"vitest\",\n    \"test:single\": \"vitest run --reporter=dot\",\n    \"test:flake\": \"./flake.sh\",\n    \"bench\": \"vitest bench\"\n  },\n  \"engines\": {\n    \"node\": \">=16\"\n  },\n  \"keywords\": [\n    \"rpc\",\n    \"websockets\",\n    \"jsonschema\"\n  ],\n  \"author\": \"Jacky Zhao\",\n  \"license\": \"MIT\"\n}\n","export function coerceErrorString(err: unknown): string {\n  if (err instanceof Error) {\n    return err.message || 'unknown reason';\n  }\n\n  return `[coerced to error] ${String(err)}`;\n}\n","import {\n  Context,\n  Span,\n  SpanKind,\n  context,\n  propagation,\n  trace,\n} from '@opentelemetry/api';\nimport { version as RIVER_VERSION } from '../package.json';\nimport { ValidProcType } from '../router';\nimport { ClientTransport, Connection } from '../transport';\n\nexport interface PropagationContext {\n  traceparent: string;\n  tracestate: string;\n}\n\nexport interface TelemetryInfo {\n  span: Span;\n  ctx: Context;\n}\n\nexport function getPropagationContext(\n  ctx: Context,\n): PropagationContext | undefined {\n  const tracing = {\n    traceparent: '',\n    tracestate: '',\n  };\n  propagation.inject(ctx, tracing);\n\n  return tracing;\n}\n\nexport function createSessionTelemetryInfo(\n  sessionId: string,\n  to: string,\n  from: string,\n  propagationCtx?: PropagationContext,\n): TelemetryInfo {\n  const parentCtx = propagationCtx\n    ? propagation.extract(context.active(), propagationCtx)\n    : context.active();\n\n  const span = tracer.startSpan(\n    `session ${sessionId}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.session.id': sessionId,\n        'river.session.to': to,\n        'river.session.from': from,\n      },\n    },\n    parentCtx,\n  );\n\n  const ctx = trace.setSpan(parentCtx, span);\n\n  return { span, ctx };\n}\n\nexport function createConnectionTelemetryInfo(\n  connection: Connection,\n  info: TelemetryInfo,\n): TelemetryInfo {\n  const span = tracer.startSpan(\n    `connection ${connection.id}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.connection.id': connection.id,\n      },\n      links: [{ context: info.span.spanContext() }],\n    },\n    info.ctx,\n  );\n\n  const ctx = trace.setSpan(info.ctx, span);\n\n  return { span, ctx };\n}\n\nexport function createProcTelemetryInfo(\n  transport: ClientTransport<Connection>,\n  kind: ValidProcType,\n  serviceName: string,\n  procedureName: string,\n  streamId: string,\n): TelemetryInfo {\n  const baseCtx = context.active();\n  const span = tracer.startSpan(\n    `procedure call ${serviceName}.${procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': serviceName,\n        'river.method.name': procedureName,\n        'river.streamId': streamId,\n        'span.kind': 'client',\n      },\n      kind: SpanKind.CLIENT,\n    },\n    baseCtx,\n  );\n\n  const ctx = trace.setSpan(baseCtx, span);\n\n  transport.log?.info(`invoked ${serviceName}.${procedureName}`, {\n    clientId: transport.clientId,\n    transportMessage: {\n      procedureName,\n      serviceName,\n    },\n    telemetry: {\n      traceId: span.spanContext().traceId,\n      spanId: span.spanContext().spanId,\n    },\n  });\n\n  return { span, ctx };\n}\n\nexport function createHandlerSpan<Fn extends (span: Span) => unknown>(\n  kind: ValidProcType,\n  serviceName: string,\n  procedureName: string,\n  streamId: string,\n  tracing: PropagationContext | undefined,\n  fn: Fn,\n): ReturnType<Fn> {\n  const ctx = tracing\n    ? propagation.extract(context.active(), tracing)\n    : context.active();\n\n  return tracer.startActiveSpan<Fn>(\n    `procedure handler ${serviceName}.${procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': serviceName,\n        'river.method.name': procedureName,\n        'river.streamId': streamId,\n        'span.kind': 'server',\n      },\n      kind: SpanKind.SERVER,\n    },\n    ctx,\n    fn,\n  );\n}\n\nconst tracer = trace.getTracer('river', RIVER_VERSION);\nexport default tracer;\n"],"mappings":";AAAA,SAAS,YAA6B;;;ACAtC,SAAS,sBAAsB;AAE/B,IAAM,WAAW;AAAA,EACf;AACF;AACO,IAAM,aAAa,MAAM,SAAS,EAAE;;;AD4BpC,IAAM,yBAAyB,CAAoB,MACxD,KAAK,OAAO;AAAA,EACV,IAAI,KAAK,OAAO;AAAA,EAChB,MAAM,KAAK,OAAO;AAAA,EAClB,IAAI,KAAK,OAAO;AAAA,EAChB,KAAK,KAAK,QAAQ;AAAA,EAClB,KAAK,KAAK,QAAQ;AAAA,EAClB,aAAa,KAAK,SAAS,KAAK,OAAO,CAAC;AAAA,EACxC,eAAe,KAAK,SAAS,KAAK,OAAO,CAAC;AAAA,EAC1C,UAAU,KAAK,OAAO;AAAA,EACtB,cAAc,KAAK,QAAQ;AAAA,EAC3B,SAAS,KAAK;AAAA,IACZ,KAAK,OAAO;AAAA,MACV,aAAa,KAAK,OAAO;AAAA,MACzB,YAAY,KAAK,OAAO;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EACA,SAAS;AACX,CAAC;AAOI,IAAM,0BAA0B,KAAK,OAAO;AAAA,EACjD,MAAM,KAAK,QAAQ,KAAK;AAC1B,CAAC;AAMM,IAAM,4BAA4B,KAAK,OAAO;AAAA,EACnD,MAAM,KAAK,QAAQ,OAAO;AAC5B,CAAC;AAGM,IAAM,yBAAyB;AAC/B,IAAM,2BAA2B,CAAC,QAAQ,sBAAsB;AAChE,SAAS,0BACdA,UAC4B;AAC5B,SAAO,yBAAyB,SAASA,QAAO;AAClD;AAEO,IAAM,uCAAuC,KAAK,OAAO;AAAA,EAC9D,MAAM,KAAK,QAAQ,eAAe;AAAA,EAClC,iBAAiB,KAAK,OAAO;AAAA,EAC7B,WAAW,KAAK,OAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMvB,sBAAsB,KAAK,OAAO;AAAA;AAAA,IAEhC,iBAAiB,KAAK,QAAQ;AAAA,IAC9B,aAAa,KAAK,QAAQ;AAAA,EAC5B,CAAC;AAAA,EAED,UAAU,KAAK,SAAS,KAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,uCAAuC,KAAK,MAAM;AAAA,EAC7D,KAAK,QAAQ,wBAAwB;AACvC,CAAC;AAEM,IAAM,gDAAgD,KAAK,MAAM;AAAA;AAAA,EAEtE,KAAK,QAAQ,6BAA6B;AAAA;AAAA,EAE1C,KAAK,QAAQ,4BAA4B;AAC3C,CAAC;AAEM,IAAM,mCAAmC,KAAK,MAAM;AAAA,EACzD;AAAA;AAAA,EAEA,KAAK,QAAQ,0BAA0B;AAAA;AAAA,EAEvC,KAAK,QAAQ,qBAAqB;AAAA;AAAA,EAElC,KAAK,QAAQ,2BAA2B;AAC1C,CAAC;AAEM,IAAM,8BAA8B,KAAK,MAAM;AAAA,EACpD;AAAA,EACA;AACF,CAAC;AAEM,IAAM,wCAAwC,KAAK,OAAO;AAAA,EAC/D,MAAM,KAAK,QAAQ,gBAAgB;AAAA,EACnC,QAAQ,KAAK,MAAM;AAAA,IACjB,KAAK,OAAO;AAAA,MACV,IAAI,KAAK,QAAQ,IAAI;AAAA,MACrB,WAAW,KAAK,OAAO;AAAA,IACzB,CAAC;AAAA,IACD,KAAK,OAAO;AAAA,MACV,IAAI,KAAK,QAAQ,KAAK;AAAA,MACtB,QAAQ,KAAK,OAAO;AAAA,MACpB,MAAM;AAAA,IACR,CAAC;AAAA,EACH,CAAC;AACH,CAAC;AAEM,IAAM,8BAA8B,KAAK,MAAM;AAAA,EACpD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,+BAA+B;AAAA,EAC1C,KAAK,QAAQ;AACf;AAoCO,SAAS,wBAAwB;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAS0E;AACxE,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,WAAW;AAAA,IACrB,cAAc;AAAA,IACd;AAAA,IACA,SAAS;AAAA,MACP,MAAM;AAAA,MACN,iBAAiB;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQO,SAAS,yBAAyB;AAAA,EACvC;AAAA,EACA;AAAA,EACA;AACF,GAI2E;AACzE,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,WAAW;AAAA,IACrB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,mBAAmB,UAA2C;AAC5E,SAAO;AAAA,IACL;AAAA,IACA,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,IACR;AAAA,EACF;AACF;AAEO,SAAS,cACd,UACA,SACA;AACA,SAAO;AAAA,IACL;AAAA,IACA,cAAc;AAAA,IACd;AAAA,EACF;AACF;AAcO,SAAS,MAAM,aAA8B;AAElD,UAAQ,cAAc,oBAAyB;AACjD;AAOO,SAAS,aAAa,aAA8B;AACzD;AAAA;AAAA,KAEG,cAAc,2BAAgC;AAAA;AAEnD;AAOO,SAAS,cAAc,aAA8B;AAC1D;AAAA;AAAA,KAEG,cAAc,6BACf;AAAA;AAEJ;AAOO,SAAS,eAAe,aAA8B;AAC3D;AAAA;AAAA,KAEG,cAAc,6BACf;AAAA;AAEJ;;;AErUE,cAAW;;;ACHN,SAAS,kBAAkB,KAAsB;AACtD,MAAI,eAAe,OAAO;AACxB,WAAO,IAAI,WAAW;AAAA,EACxB;AAEA,SAAO,sBAAsB,OAAO,GAAG,CAAC;AAC1C;;;ACNA;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAeA,SAAS,sBACd,KACgC;AAChC,QAAM,UAAU;AAAA,IACd,aAAa;AAAA,IACb,YAAY;AAAA,EACd;AACA,cAAY,OAAO,KAAK,OAAO;AAE/B,SAAO;AACT;AAEO,SAAS,2BACd,WACA,IACA,MACA,gBACe;AACf,QAAM,YAAY,iBACd,YAAY,QAAQ,QAAQ,OAAO,GAAG,cAAc,IACpD,QAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,SAAS;AAAA,IACpB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB;AAAA,QACpB,oBAAoB;AAAA,QACpB,sBAAsB;AAAA,MACxB;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,WAAW,IAAI;AAEzC,SAAO,EAAE,MAAM,IAAI;AACrB;AAuBO,SAAS,wBACd,WACA,MACA,aACA,eACA,UACe;AACf,QAAM,UAAU,QAAQ,OAAO;AAC/B,QAAM,OAAO,OAAO;AAAA,IAClB,kBAAkB,WAAW,IAAI,aAAa;AAAA,IAC9C;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB;AAAA,QACxB,qBAAqB;AAAA,QACrB,kBAAkB;AAAA,QAClB,aAAa;AAAA,MACf;AAAA,MACA,MAAM,SAAS;AAAA,IACjB;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,SAAS,IAAI;AAEvC,YAAU,KAAK,KAAK,WAAW,WAAW,IAAI,aAAa,IAAI;AAAA,IAC7D,UAAU,UAAU;AAAA,IACpB,kBAAkB;AAAA,MAChB;AAAA,MACA;AAAA,IACF;AAAA,IACA,WAAW;AAAA,MACT,SAAS,KAAK,YAAY,EAAE;AAAA,MAC5B,QAAQ,KAAK,YAAY,EAAE;AAAA,IAC7B;AAAA,EACF,CAAC;AAED,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,kBACd,MACA,aACA,eACA,UACA,SACA,IACgB;AAChB,QAAM,MAAM,UACR,YAAY,QAAQ,QAAQ,OAAO,GAAG,OAAO,IAC7C,QAAQ,OAAO;AAEnB,SAAO,OAAO;AAAA,IACZ,qBAAqB,WAAW,IAAI,aAAa;AAAA,IACjD;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB;AAAA,QACxB,qBAAqB;AAAA,QACrB,kBAAkB;AAAA,QAClB,aAAa;AAAA,MACf;AAAA,MACA,MAAM,SAAS;AAAA,IACjB;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,IAAM,SAAS,MAAM,UAAU,SAAS,OAAa;AACrD,IAAO,kBAAQ;","names":["version"]}