@replit/river 0.22.0 → 0.23.0

package/dist/{chunk-S3YKQT4J.js → chunk-DZOATC6M.js}

@@ -1,6 +1,6 @@
 import {
   Connection
-} from "./chunk-ZPBWKBM5.js";
+} from "./chunk-MJUFKPBT.js";
 
 // transport/transforms/messageFraming.ts
 import { Transform } from "node:stream";
@@ -103,4 +103,4 @@ var UdsConnection = class extends Connection {
 export {
   UdsConnection
 };
-//# sourceMappingURL=chunk-S3YKQT4J.js.map
+//# sourceMappingURL=chunk-DZOATC6M.js.map

package/dist/{chunk-ZPBWKBM5.js → chunk-MJUFKPBT.js}

@@ -11,7 +11,7 @@ import {
   handshakeResponseMessage,
   isAck,
   tracing_default
-} from "./chunk-3MFX6NXA.js";
+} from "./chunk-PCKHBAVP.js";
 import {
   log
 } from "./chunk-OTQNCLFH.js";
@@ -54,12 +54,6 @@ var Session = class {
    * for this session.
    */
   advertisedSessionId;
-  /**
-   * The metadata for this session, as parsed from the handshake.
-   *
-   * Will only ever be populated on the server side.
-   */
-  metadata;
   /**
    * Number of messages we've sent along this session (excluding handshake and acks)
    */
@@ -706,6 +700,10 @@ var ClientTransport = class extends Transport {
    * tests or a special case where you don't want to reconnect.
    */
   reconnectOnConnectionDrop = true;
+  /**
+   * Optional handshake options for this client.
+   */
+  handshakeExtensions;
   constructor(clientId, providedOptions) {
     super(clientId, providedOptions);
     this.options = {
@@ -715,6 +713,9 @@ var ClientTransport = class extends Transport {
     this.inflightConnectionPromises = /* @__PURE__ */ new Map();
     this.retryBudget = new LeakyBucketRateLimit(this.options);
   }
+  extendHandshake(options) {
+    this.handshakeExtensions = options;
+  }
   handleConnection(conn, to) {
     if (this.state !== "open")
       return;
@@ -938,10 +939,10 @@ var ClientTransport = class extends Transport {
   }
   async sendHandshake(to, conn) {
     let metadata;
-    if (this.options.handshake) {
-      metadata = await this.options.handshake.get();
-      if (!Value.Check(this.options.handshake.schema, metadata)) {
-        log?.error(`handshake metadata did not match schema`, {
+    if (this.handshakeExtensions) {
+      metadata = await this.handshakeExtensions.construct();
+      if (!Value.Check(this.handshakeExtensions.schema, metadata)) {
+        log?.error(`constructed handshake metadata did not match schema`, {
           clientId: this.clientId,
           connectedTo: to,
           tags: ["invariant-violation"]
@@ -982,17 +983,29 @@ var ServerTransport = class extends Transport {
    * The options for this transport.
    */
   options;
+  /**
+   * Optional handshake options for the server.
+   */
+  handshakeExtensions;
+  /**
+   * A map of session handshake data for each session.
+   */
+  sessionHandshakeMetadata;
   constructor(clientId, providedOptions) {
     super(clientId, providedOptions);
     this.options = {
       ...defaultServerTransportOptions,
       ...providedOptions
     };
+    this.sessionHandshakeMetadata = /* @__PURE__ */ new WeakMap();
     log?.info(`initiated server transport`, {
       clientId: this.clientId,
       protocolVersion: PROTOCOL_VERSION
     });
   }
+  extendHandshake(options) {
+    this.handshakeExtensions = options;
+  }
   handleConnection(conn) {
     if (this.state !== "open")
       return;
@@ -1075,6 +1088,52 @@ var ServerTransport = class extends Transport {
       );
     });
   }
+  async validateHandshakeMetadata(conn, session, rawMetadata, from) {
+    let parsedMetadata = {};
+    if (this.handshakeExtensions) {
+      if (!Value.Check(this.handshakeExtensions.schema, rawMetadata)) {
+        conn.telemetry?.span.setStatus({
+          code: SpanStatusCode2.ERROR,
+          message: "malformed handshake meta"
+        });
+        const reason = "received malformed handshake metadata";
+        const responseMsg = handshakeResponseMessage(this.clientId, from, {
+          ok: false,
+          reason
+        });
+        conn.send(this.codec.toBuffer(responseMsg));
+        log?.warn(`received malformed handshake metadata from ${from}`, {
+          clientId: this.clientId,
+          connId: conn.id
+        });
+        this.protocolError(ProtocolError.HandshakeFailed, reason);
+        return false;
+      }
+      parsedMetadata = await this.handshakeExtensions.validate(
+        rawMetadata,
+        session
+      );
+      if (parsedMetadata === false) {
+        const reason = "rejected by handshake handler";
+        conn.telemetry?.span.setStatus({
+          code: SpanStatusCode2.ERROR,
+          message: reason
+        });
+        const responseMsg = handshakeResponseMessage(this.clientId, from, {
+          ok: false,
+          reason
+        });
+        conn.send(this.codec.toBuffer(responseMsg));
+        log?.warn(`rejected handshake from ${from}`, {
+          clientId: this.clientId,
+          connId: conn.id
+        });
+        this.protocolError(ProtocolError.HandshakeFailed, reason);
+        return false;
+      }
+    }
+    return parsedMetadata;
+  }
   async receiveHandshakeRequestMessage(data, conn) {
     const parsed = this.parseMsg(data);
     if (!parsed) {
@@ -1130,86 +1189,23 @@ var ServerTransport = class extends Transport {
       this.protocolError(ProtocolError.HandshakeFailed, reason);
       return false;
     }
+    const oldSession = this.sessions.get(parsed.from);
+    const parsedMetadata = await this.validateHandshakeMetadata(
+      conn,
+      oldSession,
+      parsed.payload.metadata,
+      parsed.from
+    );
+    if (parsedMetadata === false) {
+      return false;
+    }
     const { session, isReconnect } = this.getOrCreateSession(
       parsed.from,
       conn,
       parsed.payload.sessionId,
       parsed.tracing
     );
-    let handshakeMetadata;
-    if (this.options.handshake) {
-      if (!Value.Check(
-        this.options.handshake.requestSchema,
-        parsed.payload.metadata
-      )) {
-        conn.telemetry?.span.setStatus({
-          code: SpanStatusCode2.ERROR,
-          message: "malformed handshake meta"
-        });
-        const reason = "received malformed handshake metadata";
-        const responseMsg2 = handshakeResponseMessage(
-          this.clientId,
-          parsed.from,
-          { ok: false, reason }
-        );
-        conn.send(this.codec.toBuffer(responseMsg2));
-        log?.warn(`received malformed handshake metadata from ${parsed.from}`, {
-          clientId: this.clientId,
-          connId: conn.id
-        });
-        this.protocolError(ProtocolError.HandshakeFailed, reason);
-        this.deleteSession(session);
-        return false;
-      }
-      const parsedMetadata = await this.options.handshake.parse(
-        parsed.payload.metadata,
-        session,
-        isReconnect
-      );
-      if (parsedMetadata === false) {
-        conn.telemetry?.span.setStatus({
-          code: SpanStatusCode2.ERROR,
-          message: "rejected by handshake handler"
-        });
-        const reason = "rejected by handshake handler";
-        const responseMsg2 = handshakeResponseMessage(
-          this.clientId,
-          parsed.from,
-          { ok: false, reason }
-        );
-        conn.send(this.codec.toBuffer(responseMsg2));
-        log?.warn(`rejected handshake from ${parsed.from}`, {
-          clientId: this.clientId,
-          connId: conn.id
-        });
-        this.protocolError(ProtocolError.HandshakeFailed, reason);
-        this.deleteSession(session);
-        return false;
-      }
-      if (!Value.Check(this.options.handshake.parsedSchema, parsedMetadata)) {
-        conn.telemetry?.span.setStatus({
-          code: SpanStatusCode2.ERROR,
-          message: "malformed handshake meta"
-        });
-        const reason = "failed to parse handshake metadata";
-        const responseMsg2 = handshakeResponseMessage(
-          this.clientId,
-          parsed.from,
-          { ok: false, reason }
-        );
-        conn.send(this.codec.toBuffer(responseMsg2));
-        log?.error(`failed to parse handshake metadata`, {
-          clientId: this.clientId,
-          connId: conn.id
-        });
-        this.protocolError(ProtocolError.HandshakeFailed, reason);
-        this.deleteSession(session);
-        return false;
-      }
-      handshakeMetadata = parsedMetadata;
-    }
-    handshakeMetadata ??= {};
-    session.metadata = handshakeMetadata;
+    this.sessionHandshakeMetadata.set(session, parsedMetadata);
     log?.debug(
       `handshake from ${parsed.from} ok, responding with handshake success`,
       { clientId: this.clientId, connId: conn.id }
@@ -1233,4 +1229,4 @@ export {
   ClientTransport,
   ServerTransport
 };
-//# sourceMappingURL=chunk-ZPBWKBM5.js.map
+//# sourceMappingURL=chunk-MJUFKPBT.js.map

package/dist/chunk-MJUFKPBT.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/session.ts","../transport/events.ts","../transport/transport.ts","../transport/rateLimit.ts"],"sourcesContent":["import { customAlphabet } from 'nanoid';\nimport {\n  ControlFlags,\n  ControlMessageAckSchema,\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  TransportClientId,\n  TransportMessage,\n} from './message';\nimport { Codec } from '../codec';\nimport { MessageMetadata, log } from '../logging/log';\nimport { Static } from '@sinclair/typebox';\nimport {\n  PropagationContext,\n  TelemetryInfo,\n  createSessionTelemetryInfo,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nconst nanoid = customAlphabet('1234567890abcdefghijklmnopqrstuvxyz', 6);\nexport const unsafeId = () => nanoid();\n\ntype SequenceNumber = number;\n\n/**\n * A connection is the actual raw underlying transport connection.\n * It’s responsible for dispatching to/from the actual connection itself\n * This should be instantiated as soon as the client/server has a connection\n * It’s tied to the lifecycle of the underlying transport connection (i.e. if the WS drops, this connection should be deleted)\n */\nexport abstract class Connection {\n  id: string;\n  telemetry?: TelemetryInfo;\n  constructor() {\n    this.id = `conn-${nanoid(12)}`; // for debugging, no collision safety needed\n  }\n\n  /**\n   * Handle adding a callback for when a message is received.\n   * @param msg The message that was received.\n   */\n  abstract addDataListener(cb: (msg: Uint8Array) => void): void;\n  abstract removeDataListener(cb: (msg: Uint8Array) => void): void;\n\n  /**\n   * Handle adding a callback for when the connection is closed.\n   * This should also be called if an error happens.\n   * @param cb The callback to call when the connection is closed.\n   */\n  abstract addCloseListener(cb: () => void): void;\n\n  /**\n   * Handle adding a callback for when an error is received.\n   * This should only be used for logging errors, all cleanup\n   * should be delegated to addCloseListener.\n   *\n   * The implementer should take care such that the implemented\n   * connection will call both the close and error callbacks\n   * on an error.\n   *\n   * @param cb The callback to call when an error is received.\n   */\n  abstract addErrorListener(cb: (err: Error) => void): void;\n\n  /**\n   * Sends a message over the connection.\n   * @param msg The message to send.\n   * @returns true if the message was sent, false otherwise.\n   */\n  abstract send(msg: Uint8Array): boolean;\n\n  /**\n   * Closes the connection.\n   */\n  abstract close(): void;\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   * Duration to wait between connection disconnect and actual session disconnect\n   */\n  sessionDisconnectGraceMs: number;\n  /**\n   * The codec to use for encoding/decoding messages over the wire\n   */\n  codec: Codec;\n}\n\n/**\n * A session is a higher-level abstraction that operates over the span of potentially multiple transport-level connections\n * - It’s responsible for tracking any metadata for a particular client that might need to be persisted across connections (i.e. the sendBuffer, ack, seq)\n * - This will only be considered disconnected if\n *    - the server tells the client that we’ve reconnected but it doesn’t recognize us anymore (server definitely died) or\n *    - we hit a grace period after a connection disconnect\n */\nexport class Session<ConnType extends Connection> {\n  private codec: Codec;\n  private options: SessionOptions;\n  telemetry: TelemetryInfo;\n\n  /**\n   * The buffer of messages that have been sent but not yet acknowledged.\n   */\n  private sendBuffer: Array<OpaqueTransportMessage> = [];\n\n  /**\n   * The active connection associated with this session\n   */\n  connection?: ConnType;\n  readonly from: TransportClientId;\n  readonly to: TransportClientId;\n\n  /**\n   * The unique ID of this session.\n   */\n  id: string;\n\n  /**\n   * What the other side advertised as their session ID\n   * for this session.\n   */\n  advertisedSessionId?: string;\n\n  /**\n   * Number of messages we've sent along this session (excluding handshake and acks)\n   */\n  private seq: SequenceNumber = 0;\n\n  /**\n   * Number of unique messages we've received this session (excluding handshake and acks)\n   */\n  private ack: SequenceNumber = 0;\n\n  /**\n   * The grace period between when the inner connection is disconnected\n   * and when we should consider the entire session disconnected.\n   */\n  private disconnectionGrace?: ReturnType<typeof setTimeout>;\n\n  /**\n   * Number of heartbeats we've sent without a response.\n   */\n  private heartbeatMisses: number;\n\n  /**\n   * The interval for sending heartbeats.\n   */\n  private heartbeat: ReturnType<typeof setInterval>;\n\n  constructor(\n    conn: ConnType | undefined,\n    from: TransportClientId,\n    to: TransportClientId,\n    options: SessionOptions,\n    propagationCtx?: PropagationContext,\n  ) {\n    this.id = `session-${nanoid(12)}`;\n    this.options = options;\n    this.from = from;\n    this.to = to;\n    this.connection = conn;\n    this.codec = options.codec;\n\n    // setup heartbeat\n    this.heartbeatMisses = 0;\n    this.heartbeat = setInterval(\n      () => this.sendHeartbeat(),\n      options.heartbeatIntervalMs,\n    );\n    this.telemetry = createSessionTelemetryInfo(this, propagationCtx);\n  }\n\n  get loggingMetadata(): Omit<MessageMetadata, 'parsedMsg'> {\n    return {\n      clientId: this.from,\n      connectedTo: this.to,\n      sessionId: this.id,\n      connId: this.connection?.id,\n    };\n  }\n\n  /**\n   * Sends a message over the session's connection.\n   * If the connection is not ready or the message fails to send, the message can be buffered for retry unless skipped.\n   *\n   * @param msg The partial message to be sent, which will be constructed into a full message.\n   * @param addToSendBuff Whether to add the message to the send buffer for retry.\n   * @returns The full transport ID of the message that was attempted to be sent.\n   */\n  send(msg: PartialTransportMessage): string {\n    const fullMsg: TransportMessage = this.constructMsg(msg);\n    log?.debug(`sending msg`, {\n      ...this.loggingMetadata,\n      fullTransportMessage: fullMsg,\n    });\n\n    if (this.connection) {\n      const ok = this.connection.send(this.codec.toBuffer(fullMsg));\n      if (ok) return fullMsg.id;\n      log?.info(\n        `failed to send msg to ${fullMsg.to}, connection is probably dead`,\n        {\n          ...this.loggingMetadata,\n          fullTransportMessage: fullMsg,\n        },\n      );\n    } else {\n      log?.info(\n        `failed to send msg to ${fullMsg.to}, connection not ready yet`,\n        { ...this.loggingMetadata, fullTransportMessage: fullMsg },\n      );\n    }\n\n    return fullMsg.id;\n  }\n\n  sendHeartbeat() {\n    const misses = this.heartbeatMisses;\n    const missDuration = misses * this.options.heartbeatIntervalMs;\n    if (misses > this.options.heartbeatsUntilDead) {\n      if (this.connection) {\n        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        this.closeStaleConnection();\n      }\n      return;\n    }\n\n    this.send({\n      streamId: 'heartbeat',\n      controlFlags: ControlFlags.AckBit,\n      payload: {\n        type: 'ACK',\n      } satisfies Static<typeof ControlMessageAckSchema>,\n    });\n    this.heartbeatMisses++;\n  }\n\n  resetBufferedMessages() {\n    this.sendBuffer = [];\n    this.seq = 0;\n    this.ack = 0;\n  }\n\n  sendBufferedMessages(conn: ConnType) {\n    log?.info(`resending ${this.sendBuffer.length} buffered messages`, {\n      ...this.loggingMetadata,\n      connId: conn.id,\n    });\n    for (const msg of this.sendBuffer) {\n      log?.debug(`resending msg`, {\n        ...this.loggingMetadata,\n        fullTransportMessage: msg,\n        connId: conn.id,\n      });\n      const ok = conn.send(this.codec.toBuffer(msg));\n      if (!ok) {\n        // this should never happen unless the transport has an\n        // incorrect implementation of `createNewOutgoingConnection`\n        const errMsg = `failed to send buffered message to ${this.to} (sus, this is a fresh connection)`;\n        conn.telemetry?.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: errMsg,\n        });\n\n        log?.error(errMsg, {\n          ...this.loggingMetadata,\n          fullTransportMessage: msg,\n          connId: conn.id,\n          tags: ['invariant-violation'],\n        });\n        conn.close();\n        return;\n      }\n    }\n  }\n\n  updateBookkeeping(ack: number, seq: number) {\n    if (seq + 1 < this.ack) {\n      log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {\n        ...this.loggingMetadata,\n        tags: ['invariant-violation'],\n      });\n      return;\n    }\n\n    this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n    this.ack = seq + 1;\n  }\n\n  closeStaleConnection(conn?: ConnType) {\n    if (this.connection === undefined || this.connection === conn) return;\n    log?.info(\n      `closing old inner connection from session to ${this.to}`,\n      this.loggingMetadata,\n    );\n    this.connection.close();\n    this.connection = undefined;\n  }\n\n  replaceWithNewConnection(newConn: ConnType) {\n    this.closeStaleConnection(newConn);\n    this.cancelGrace();\n    this.connection = newConn;\n    this.sendBufferedMessages(newConn);\n  }\n\n  beginGrace(cb: () => void) {\n    log?.info(\n      `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,\n      this.loggingMetadata,\n    );\n    this.disconnectionGrace = setTimeout(() => {\n      this.close();\n      cb();\n    }, this.options.sessionDisconnectGraceMs);\n  }\n\n  // called on reconnect of the underlying session\n  cancelGrace() {\n    this.heartbeatMisses = 0;\n    clearTimeout(this.disconnectionGrace);\n    this.disconnectionGrace = undefined;\n  }\n\n  // closed when we want to discard the whole session\n  // (i.e. shutdown or session disconnect)\n  close() {\n    this.closeStaleConnection();\n    this.cancelGrace();\n    this.resetBufferedMessages();\n    clearInterval(this.heartbeat);\n  }\n\n  get connected() {\n    return this.connection !== undefined;\n  }\n\n  get nextExpectedSeq() {\n    return this.ack;\n  }\n\n  constructMsg<Payload>(\n    partialMsg: PartialTransportMessage<Payload>,\n  ): TransportMessage<Payload> {\n    const msg = {\n      ...partialMsg,\n      id: unsafeId(),\n      to: this.to,\n      from: this.from,\n      seq: this.seq,\n      ack: this.ack,\n    };\n\n    this.seq++;\n    this.sendBuffer.push(msg);\n    return msg;\n  }\n\n  inspectSendBuffer(): ReadonlyArray<OpaqueTransportMessage> {\n    return this.sendBuffer;\n  }\n}\n","import { OpaqueTransportMessage } from './message';\nimport { Connection, Session } from './session';\n\ntype ConnectionStatus = 'connect' | 'disconnect';\nexport const ProtocolError = {\n  RetriesExceeded: 'conn_retry_exceeded',\n  HandshakeFailed: 'handshake_failed',\n  UseAfterDestroy: 'use_after_destroy',\n  MessageOrderingViolated: 'message_ordering_violated',\n} as const;\n\nexport type ProtocolErrorType =\n  (typeof ProtocolError)[keyof typeof ProtocolError];\n\nexport interface EventMap {\n  message: OpaqueTransportMessage;\n  connectionStatus: {\n    status: ConnectionStatus;\n    conn: Connection;\n  };\n  sessionStatus: {\n    status: ConnectionStatus;\n    session: Session<Connection>;\n  };\n  protocolError: {\n    type: ProtocolErrorType;\n    message: string;\n  };\n}\n\nexport type EventTypes = keyof EventMap;\nexport type EventHandler<K extends EventTypes> = (\n  event: EventMap[K],\n) => unknown;\n\nexport class EventDispatcher<T extends EventTypes> {\n  private eventListeners: { [K in T]?: Set<EventHandler<K>> } = {};\n\n  numberOfListeners<K extends T>(eventType: K) {\n    return this.eventListeners[eventType]?.size ?? 0;\n  }\n\n  addEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n    if (!this.eventListeners[eventType]) {\n      this.eventListeners[eventType] = new Set();\n    }\n\n    this.eventListeners[eventType]?.add(handler);\n  }\n\n  removeEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n    const handlers = this.eventListeners[eventType];\n    if (handlers) {\n      this.eventListeners[eventType]?.delete(handler);\n    }\n  }\n\n  dispatchEvent<K extends T>(eventType: K, event: EventMap[K]) {\n    const handlers = this.eventListeners[eventType];\n    if (handlers) {\n      for (const handler of handlers) {\n        handler(event);\n      }\n    }\n  }\n}\n","import { Codec } from '../codec/types';\nimport { Value } from '@sinclair/typebox/value';\nimport {\n  OpaqueTransportMessage,\n  OpaqueTransportMessageSchema,\n  TransportClientId,\n  ControlMessageHandshakeRequestSchema,\n  ControlMessageHandshakeResponseSchema,\n  handshakeRequestMessage,\n  handshakeResponseMessage,\n  PartialTransportMessage,\n  ControlFlags,\n  ControlMessagePayloadSchema,\n  isAck,\n  PROTOCOL_VERSION,\n  ClientHandshakeOptions,\n  ServerHandshakeOptions,\n} from './message';\nimport { log } from '../logging/log';\nimport {\n  EventDispatcher,\n  EventHandler,\n  EventTypes,\n  ProtocolError,\n  ProtocolErrorType,\n} from './events';\nimport { Connection, Session, SessionOptions } from './session';\nimport { Static, TSchema } from '@sinclair/typebox';\nimport { coerceErrorString } from '../util/stringify';\nimport { ConnectionRetryOptions, LeakyBucketRateLimit } from './rateLimit';\nimport { NaiveJsonCodec } from '../codec';\nimport tracer, {\n  PropagationContext,\n  createConnectionTelemetryInfo,\n  getPropagationContext,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\nimport { ParsedMetadata } from '../router/context';\n\n/**\n * Represents the possible states of a transport.\n * @property {'open'} open - The transport is open and operational (note that this doesn't mean it is actively connected)\n * @property {'closed'} closed - The transport is closed and not operational, but can be reopened.\n * @property {'destroyed'} destroyed - The transport is permanently destroyed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed' | 'destroyed';\n\ntype 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  codec: NaiveJsonCodec,\n};\n\ntype ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n  baseIntervalMs: 250,\n  maxJitterMs: 200,\n  maxBackoffMs: 32_000,\n  attemptBudgetCapacity: 5,\n  budgetRestoreIntervalMs: 200,\n};\n\nconst defaultClientTransportOptions: ClientTransportOptions = {\n  ...defaultTransportOptions,\n  ...defaultConnectionRetryOptions,\n};\n\ntype ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nconst defaultServerTransportOptions: ServerTransportOptions = {\n  ...defaultTransportOptions,\n};\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions and connections. Its responsibilities include:\n *\n *  1) Constructing a new {@link Session} and {@link Connection} on {@link TransportMessage}s from new clients.\n *     After constructing the {@link Connection}, {@link onConnect} is called which adds it to the connection map.\n *  2) Delegating message listening of the connection to the newly created {@link Connection}.\n *     From this point on, the {@link Connection} is responsible for *reading* and *writing*\n *     messages from the connection.\n *  3) When a connection is closed, the {@link Transport} calls {@link onDisconnect} which closes the\n *     connection via {@link Connection.close} and removes it from the {@link connections} map.\n\n *\n * ```plaintext\n *            ▲\n *  incoming  │\n *  messages  │\n *            ▼\n *      ┌─────────────┐   1:N   ┌───────────┐   1:1*  ┌────────────┐\n *      │  Transport  │ ◄─────► │  Session  │ ◄─────► │ Connection │\n *      └─────────────┘         └───────────┘         └────────────┘\n *            ▲                               * (may or may not be initialized yet)\n *            │\n *            ▼\n *      ┌───────────┐\n *      │ Message   │\n *      │ Listeners │\n *      └───────────┘\n * ```\n * @abstract\n */\nexport abstract class Transport<ConnType extends Connection> {\n  /**\n   * A flag indicating whether the transport has been destroyed.\n   * A destroyed transport will not attempt to reconnect and cannot be used again.\n   */\n  state: TransportStatus;\n\n  /**\n   * The {@link Codec} used to encode and decode messages.\n   */\n  codec: Codec;\n\n  /**\n   * The client ID of this transport.\n   */\n  clientId: TransportClientId;\n\n  /**\n   * The map of {@link Session}s managed by this transport.\n   */\n  sessions: Map<TransportClientId, Session<ConnType>>;\n\n  /**\n   * The map of {@link Connection}s managed by this transport.\n   */\n  get connections() {\n    return new Map(\n      [...this.sessions]\n        .map(([client, session]) => [client, session.connection])\n        .filter((entry): entry is [string, ConnType] => entry[1] !== undefined),\n    );\n  }\n\n  /**\n   * The event dispatcher for handling events of type EventTypes.\n   */\n  eventDispatcher: EventDispatcher<EventTypes>;\n\n  /**\n   * The options for this transport.\n   */\n  protected options: TransportOptions;\n\n  /**\n   * Creates a new Transport instance.\n   * This should also set up {@link onConnect}, and {@link onDisconnect} listeners.\n   * @param codec The codec used to encode and decode messages.\n   * @param clientId The client ID of this transport.\n   */\n  constructor(\n    clientId: TransportClientId,\n    providedOptions?: ProvidedTransportOptions,\n  ) {\n    this.options = { ...defaultTransportOptions, ...providedOptions };\n    this.eventDispatcher = new EventDispatcher();\n    this.sessions = new Map();\n    this.codec = this.options.codec;\n    this.clientId = clientId;\n    this.state = 'open';\n  }\n\n  /**\n   * This is called immediately after a new connection is established and we\n   * may or may not know the identity of the connected client.\n   * It should attach all the necessary listeners to the connection for lifecycle\n   * events (i.e. data, close, error)\n   *\n   * This method is implemented by {@link ClientTransport} and {@link ServerTransport}.\n   */\n  protected abstract handleConnection(\n    conn: ConnType,\n    to: TransportClientId,\n  ): void;\n\n  /**\n   * Called when a new connection is established\n   * and we know the identity of the connected client.\n   * @param conn The connection object.\n   */\n  protected onConnect(\n    conn: ConnType,\n    connectedTo: TransportClientId,\n    session: Session<ConnType>,\n    isReconnect: boolean,\n  ) {\n    this.eventDispatcher.dispatchEvent('connectionStatus', {\n      status: 'connect',\n      conn,\n    });\n\n    conn.telemetry = createConnectionTelemetryInfo(\n      conn,\n      session.telemetry.span,\n    );\n\n    if (isReconnect) {\n      session.replaceWithNewConnection(conn);\n      log?.info(`reconnected to ${connectedTo}`, session.loggingMetadata);\n    }\n  }\n\n  protected createSession(\n    to: TransportClientId,\n    conn?: ConnType,\n    propagationCtx?: PropagationContext,\n  ) {\n    const session = new Session<ConnType>(\n      conn,\n      this.clientId,\n      to,\n      this.options,\n      propagationCtx,\n    );\n    this.sessions.set(session.to, session);\n    this.eventDispatcher.dispatchEvent('sessionStatus', {\n      status: 'connect',\n      session,\n    });\n    return session;\n  }\n\n  protected getOrCreateSession(\n    to: TransportClientId,\n    conn?: ConnType,\n    sessionId?: string,\n    propagationCtx?: PropagationContext,\n  ) {\n    let session = this.sessions.get(to);\n    let isReconnect = session !== undefined;\n\n    if (\n      session?.advertisedSessionId !== undefined &&\n      sessionId !== undefined &&\n      session.advertisedSessionId !== sessionId\n    ) {\n      log?.info(\n        `session for ${to} already exists but has a different session id (expected: ${session.advertisedSessionId}, got: ${sessionId}), creating a new one`,\n        session.loggingMetadata,\n      );\n      this.deleteSession(session);\n      isReconnect = false;\n      session = undefined;\n    }\n\n    if (!session) {\n      session = this.createSession(to, conn, propagationCtx);\n      log?.info(\n        `no session for ${to}, created a new one`,\n        session.loggingMetadata,\n      );\n    }\n\n    if (sessionId !== undefined) {\n      session.advertisedSessionId = sessionId;\n    }\n\n    return { session, isReconnect };\n  }\n\n  protected deleteSession(session: Session<ConnType>) {\n    session.close();\n    session.telemetry.span.end();\n    this.sessions.delete(session.to);\n    log?.info(\n      `session ${session.id} disconnect from ${session.to}`,\n      session.loggingMetadata,\n    );\n    this.eventDispatcher.dispatchEvent('sessionStatus', {\n      status: 'disconnect',\n      session,\n    });\n  }\n\n  /**\n   * The downstream implementation needs to call this when a connection is closed.\n   * @param conn The connection object.\n   * @param connectedTo The peer we are connected to.\n   */\n  protected onDisconnect(conn: ConnType, session: Session<ConnType>) {\n    conn.telemetry?.span.end();\n    this.eventDispatcher.dispatchEvent('connectionStatus', {\n      status: 'disconnect',\n      conn,\n    });\n\n    session.connection = undefined;\n    session.beginGrace(() => {\n      session.telemetry.span.addEvent('session grace period expired');\n      this.deleteSession(session);\n    });\n  }\n\n  /**\n   * Parses a message from a Uint8Array into a {@link OpaqueTransportMessage}.\n   * @param msg The message to parse.\n   * @returns The parsed message, or null if the message is malformed or invalid.\n   */\n  protected parseMsg(msg: Uint8Array): OpaqueTransportMessage | null {\n    const parsedMsg = this.codec.fromBuffer(msg);\n\n    if (parsedMsg === null) {\n      const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));\n      log?.error(`received malformed msg, killing conn: ${decodedBuffer}`, {\n        clientId: this.clientId,\n      });\n      return null;\n    }\n\n    if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n      log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n        clientId: this.clientId,\n      });\n      return null;\n    }\n\n    return parsedMsg;\n  }\n\n  /**\n   * Called when a message is received by this transport.\n   * You generally shouldn't need to override this in downstream transport implementations.\n   * @param msg The received message.\n   */\n  protected handleMsg(msg: OpaqueTransportMessage) {\n    if (this.state !== 'open') return;\n    const session = this.sessions.get(msg.from);\n    if (!session) {\n      log?.error(`no existing session for ${msg.from}`, {\n        clientId: this.clientId,\n        fullTransportMessage: msg,\n        tags: ['invariant-violation'],\n      });\n      return;\n    }\n\n    // got a msg so we know the other end is alive, reset the grace period\n    session.cancelGrace();\n\n    log?.debug(`received msg`, {\n      clientId: this.clientId,\n      fullTransportMessage: msg,\n    });\n    if (msg.seq !== session.nextExpectedSeq) {\n      if (msg.seq < session.nextExpectedSeq) {\n        log?.debug(\n          `received duplicate msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq}), discarding`,\n          { clientId: this.clientId, fullTransportMessage: msg },\n        );\n      } else {\n        const errMsg = `received out-of-order msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq})`;\n        log?.error(`${errMsg}, marking connection as dead`, {\n          clientId: this.clientId,\n          fullTransportMessage: msg,\n          tags: ['invariant-violation'],\n        });\n        this.protocolError(ProtocolError.MessageOrderingViolated, errMsg);\n        session.telemetry.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: 'message order violated',\n        });\n        session.close();\n      }\n\n      return;\n    }\n\n    session.updateBookkeeping(msg.ack, msg.seq);\n\n    // don't dispatch explicit acks\n    if (!isAck(msg.controlFlags)) {\n      this.eventDispatcher.dispatchEvent('message', msg);\n    } else {\n      log?.debug(`discarding msg (ack bit set)`, {\n        clientId: this.clientId,\n        fullTransportMessage: msg,\n      });\n    }\n  }\n\n  /**\n   * Adds a listener to this transport.\n   * @param the type of event to listen for\n   * @param handler The message handler to add.\n   */\n  addEventListener<K extends EventTypes, T extends EventHandler<K>>(\n    type: K,\n    handler: T,\n  ): void {\n    this.eventDispatcher.addEventListener(type, handler);\n  }\n\n  /**\n   * Removes a listener from this transport.\n   * @param the type of event to un-listen on\n   * @param handler The message handler to remove.\n   */\n  removeEventListener<K extends EventTypes, T extends EventHandler<K>>(\n    type: K,\n    handler: T,\n  ): void {\n    this.eventDispatcher.removeEventListener(type, handler);\n  }\n\n  /**\n   * Sends a message over this transport, delegating to the appropriate connection to actually\n   * send the message.\n   * @param msg The message to send.\n   * @returns The ID of the sent message or undefined if it wasn't sent\n   */\n  send(\n    to: TransportClientId,\n    msg: PartialTransportMessage,\n  ): string | undefined {\n    if (this.state === 'destroyed') {\n      const err = 'transport is destroyed, cant send';\n      log?.error(err, {\n        clientId: this.clientId,\n        partialTransportMessage: msg,\n        tags: ['invariant-violation'],\n      });\n      this.protocolError(ProtocolError.UseAfterDestroy, err);\n      return undefined;\n    } else if (this.state === 'closed') {\n      log?.info(`transport closed when sending, discarding`, {\n        clientId: this.clientId,\n        partialTransportMessage: msg,\n      });\n      return undefined;\n    }\n\n    return this.getOrCreateSession(to).session.send(msg);\n  }\n\n  // control helpers\n  sendCloseStream(to: TransportClientId, streamId: string) {\n    return this.send(to, {\n      streamId: streamId,\n      controlFlags: ControlFlags.StreamClosedBit,\n      payload: {\n        type: 'CLOSE' as const,\n      } satisfies Static<typeof ControlMessagePayloadSchema>,\n    });\n  }\n\n  protected protocolError(type: ProtocolErrorType, message: string) {\n    this.eventDispatcher.dispatchEvent('protocolError', { type, message });\n  }\n\n  /**\n   * Default close implementation for transports. You should override this in the downstream\n   * implementation if you need to do any additional cleanup and call super.close() at the end.\n   * Closes the transport. Any messages sent while the transport is closed will be silently discarded.\n   */\n  close() {\n    this.state = 'closed';\n    for (const session of this.sessions.values()) {\n      this.deleteSession(session);\n    }\n\n    log?.info(`manually closed transport`, { clientId: this.clientId });\n  }\n\n  /**\n   * Default destroy implementation for transports. You should override this in the downstream\n   * implementation if you need to do any additional cleanup and call super.destroy() at the end.\n   * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.\n   */\n  destroy() {\n    this.state = 'destroyed';\n    for (const session of this.sessions.values()) {\n      this.deleteSession(session);\n    }\n\n    log?.info(`manually destroyed transport`, { clientId: this.clientId });\n  }\n}\n\nexport abstract class ClientTransport<\n  ConnType extends Connection,\n  HandshakeMetadata extends TSchema = TSchema,\n> extends Transport<ConnType> {\n  /**\n   * The options for this transport.\n   */\n  protected options: ClientTransportOptions;\n\n  /**\n   * The map of reconnect promises for each client ID.\n   */\n  inflightConnectionPromises: Map<TransportClientId, Promise<ConnType>>;\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<HandshakeMetadata>;\n\n  constructor(\n    clientId: TransportClientId,\n    providedOptions?: ProvidedClientTransportOptions,\n  ) {\n    super(clientId, providedOptions);\n    this.options = {\n      ...defaultClientTransportOptions,\n      ...providedOptions,\n    };\n    this.inflightConnectionPromises = new Map();\n    this.retryBudget = new LeakyBucketRateLimit(this.options);\n  }\n\n  extendHandshake(options: ClientHandshakeOptions<HandshakeMetadata>) {\n    this.handshakeExtensions = options;\n  }\n\n  protected handleConnection(conn: ConnType, to: TransportClientId): void {\n    if (this.state !== 'open') return;\n    let session: Session<ConnType> | undefined = undefined;\n\n    // kill the conn after the grace period if we haven't received a handshake\n    const handshakeTimeout = setTimeout(() => {\n      if (!session) {\n        log?.warn(\n          `connection to ${to} timed out waiting for handshake, closing`,\n          { clientId: this.clientId, connectedTo: to, connId: conn.id },\n        );\n        conn.close();\n      }\n    }, this.options.sessionDisconnectGraceMs);\n\n    const handshakeHandler = (data: Uint8Array) => {\n      const maybeSession = this.receiveHandshakeResponseMessage(data, conn);\n      if (!maybeSession) {\n        conn.close();\n        return;\n      } else {\n        session = maybeSession;\n        clearTimeout(handshakeTimeout);\n      }\n\n      // when we are done handshake sequence,\n      // remove handshake listener and use the normal message listener\n      conn.removeDataListener(handshakeHandler);\n      conn.addDataListener((data) => {\n        const parsed = this.parseMsg(data);\n        if (!parsed) {\n          conn.telemetry?.span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: 'message parse failure',\n          });\n          conn.close();\n          return;\n        }\n\n        this.handleMsg(parsed);\n      });\n    };\n\n    conn.addDataListener(handshakeHandler);\n    conn.addCloseListener(() => {\n      if (session) {\n        this.onDisconnect(conn, session);\n      }\n      log?.info(`connection to ${to} disconnected`, {\n        ...session?.loggingMetadata,\n        clientId: this.clientId,\n        connectedTo: to,\n      });\n      this.inflightConnectionPromises.delete(to);\n      if (this.reconnectOnConnectionDrop) {\n        void this.connect(to);\n      }\n    });\n    conn.addErrorListener((err) => {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'connection error',\n      });\n      log?.warn(`error in connection to ${to}: ${coerceErrorString(err)}`, {\n        ...session?.loggingMetadata,\n        clientId: this.clientId,\n        connectedTo: to,\n      });\n    });\n  }\n\n  receiveHandshakeResponseMessage(\n    data: Uint8Array,\n    conn: ConnType,\n  ): Session<ConnType> | false {\n    const parsed = this.parseMsg(data);\n    if (!parsed) {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'non-transport message',\n      });\n      this.protocolError(\n        ProtocolError.HandshakeFailed,\n        'received non-transport message',\n      );\n      return false;\n    }\n\n    if (!Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'invalid handshake response',\n      });\n      log?.warn(`received invalid handshake resp`, {\n        clientId: this.clientId,\n        connectedTo: parsed.from,\n        fullTransportMessage: parsed,\n      });\n      this.protocolError(\n        ProtocolError.HandshakeFailed,\n        'invalid handshake resp',\n      );\n      return false;\n    }\n\n    if (!parsed.payload.status.ok) {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'handshake rejected',\n      });\n      log?.warn(`received handshake rejection`, {\n        clientId: this.clientId,\n        connectedTo: parsed.from,\n        fullTransportMessage: parsed,\n      });\n      this.protocolError(\n        ProtocolError.HandshakeFailed,\n        parsed.payload.status.reason,\n      );\n      return false;\n    }\n\n    log?.debug(`handshake from ${parsed.from} ok`, {\n      clientId: this.clientId,\n      connectedTo: parsed.from,\n      fullTransportMessage: parsed,\n    });\n\n    const { session, isReconnect } = this.getOrCreateSession(\n      parsed.from,\n      conn,\n      parsed.payload.status.sessionId,\n    );\n\n    this.onConnect(conn, parsed.from, session, isReconnect);\n\n    // After a successful connection, we start restoring the budget\n    // so that the next time we try to connect, we don't hit the client\n    // with backoff forever.\n    this.retryBudget.startRestoringBudget(parsed.from);\n    return session;\n  }\n\n  /**\n   * Abstract method that creates a new {@link Connection} object.\n   * This should call {@link handleConnection} when the connection is created.\n   * The downstream client implementation needs to implement this.\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  /**\n   * Manually attempts to connect to a client.\n   * @param to The client ID of the node to connect to.\n   */\n  async connect(to: TransportClientId): Promise<void> {\n    const canProceedWithConnection = () => this.state === 'open';\n    if (!canProceedWithConnection()) {\n      log?.info(\n        `transport state is no longer open, cancelling attempt to connect to ${to}`,\n        { clientId: this.clientId, connectedTo: to },\n      );\n      return;\n    }\n\n    let reconnectPromise = this.inflightConnectionPromises.get(to);\n    if (!reconnectPromise) {\n      // check budget\n      const budgetConsumed = this.retryBudget.getBudgetConsumed(to);\n      if (!this.retryBudget.hasBudget(to)) {\n        const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;\n        log?.warn(errMsg, { clientId: this.clientId, connectedTo: to });\n        this.protocolError(ProtocolError.RetriesExceeded, errMsg);\n        return;\n      }\n\n      let sleep = Promise.resolve();\n      const backoffMs = this.retryBudget.getBackoffMs(to);\n      if (backoffMs > 0) {\n        sleep = new Promise((resolve) => setTimeout(resolve, backoffMs));\n      }\n\n      log?.info(`attempting connection to ${to} (${backoffMs}ms backoff)`, {\n        clientId: this.clientId,\n        connectedTo: to,\n      });\n      this.retryBudget.consumeBudget(to);\n      reconnectPromise = tracer.startActiveSpan('connect', async (span) => {\n        try {\n          span.addEvent('backoff', { backoffMs });\n          await sleep;\n          if (!canProceedWithConnection()) {\n            throw new Error('transport state is no longer open');\n          }\n\n          span.addEvent('connecting');\n          const conn = await this.createNewOutgoingConnection(to);\n          if (!canProceedWithConnection()) {\n            log?.info(\n              `transport state is no longer open, closing pre-handshake connection to ${to}`,\n              {\n                clientId: this.clientId,\n                connectedTo: to,\n                connId: conn.id,\n              },\n            );\n            conn.close();\n            throw new Error('transport state is no longer open');\n          }\n\n          span.addEvent('sending handshake');\n          const ok = await this.sendHandshake(to, conn);\n          if (!ok) {\n            conn.close();\n            throw new Error('failed to send handshake');\n          }\n\n          return conn;\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      this.inflightConnectionPromises.set(to, reconnectPromise);\n    } else {\n      log?.info(`attempting connection to ${to} (reusing previous attempt)`, {\n        clientId: this.clientId,\n        connectedTo: to,\n      });\n    }\n\n    try {\n      await reconnectPromise;\n    } catch (error: unknown) {\n      this.inflightConnectionPromises.delete(to);\n      const errStr = coerceErrorString(error);\n\n      if (!this.reconnectOnConnectionDrop || !canProceedWithConnection()) {\n        log?.warn(`connection to ${to} failed (${errStr})`, {\n          clientId: this.clientId,\n          connectedTo: to,\n        });\n      } else {\n        log?.warn(`connection to ${to} failed (${errStr}), retrying`, {\n          clientId: this.clientId,\n          connectedTo: to,\n        });\n        return this.connect(to);\n      }\n    }\n  }\n\n  protected deleteSession(session: Session<ConnType>) {\n    this.inflightConnectionPromises.delete(session.to);\n    super.deleteSession(session);\n  }\n\n  protected async sendHandshake(to: TransportClientId, conn: ConnType) {\n    let metadata: Static<HandshakeMetadata> | undefined;\n\n    if (this.handshakeExtensions) {\n      metadata = await this.handshakeExtensions.construct();\n      if (!Value.Check(this.handshakeExtensions.schema, metadata)) {\n        log?.error(`constructed handshake metadata did not match schema`, {\n          clientId: this.clientId,\n          connectedTo: to,\n          tags: ['invariant-violation'],\n        });\n        this.protocolError(\n          ProtocolError.HandshakeFailed,\n          'handshake metadata did not match schema',\n        );\n        conn.telemetry?.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: 'handshake meta mismatch',\n        });\n        return false;\n      }\n    }\n\n    const { session } = this.getOrCreateSession(to, conn);\n    const requestMsg = handshakeRequestMessage(\n      this.clientId,\n      to,\n      session.id,\n      metadata,\n      getPropagationContext(session.telemetry.ctx),\n    );\n    log?.debug(`sending handshake request to ${to}`, {\n      clientId: this.clientId,\n      connectedTo: to,\n    });\n    conn.send(this.codec.toBuffer(requestMsg));\n    return true;\n  }\n\n  close() {\n    this.retryBudget.close();\n    super.close();\n  }\n}\n\nexport abstract class ServerTransport<\n  ConnType extends Connection,\n  CustomMetadataSchema extends TSchema = TSchema,\n> extends Transport<ConnType> {\n  /**\n   * The options for this transport.\n   */\n  protected options: ServerTransportOptions;\n\n  /**\n   * Optional handshake options for the server.\n   */\n  handshakeExtensions?: ServerHandshakeOptions<CustomMetadataSchema>;\n\n  /**\n   * A map of session handshake data for each session.\n   */\n  sessionHandshakeMetadata: WeakMap<Session<ConnType>, ParsedMetadata>;\n\n  constructor(\n    clientId: TransportClientId,\n    providedOptions?: ProvidedServerTransportOptions,\n  ) {\n    super(clientId, providedOptions);\n    this.options = {\n      ...defaultServerTransportOptions,\n      ...providedOptions,\n    };\n    this.sessionHandshakeMetadata = new WeakMap();\n    log?.info(`initiated server transport`, {\n      clientId: this.clientId,\n      protocolVersion: PROTOCOL_VERSION,\n    });\n  }\n\n  extendHandshake(options: ServerHandshakeOptions<CustomMetadataSchema>) {\n    this.handshakeExtensions = options;\n  }\n\n  protected handleConnection(conn: ConnType) {\n    if (this.state !== 'open') return;\n\n    log?.info(`new incoming connection`, {\n      clientId: this.clientId,\n      connId: conn.id,\n    });\n\n    let session: Session<ConnType> | undefined = undefined;\n    const client = () => session?.to ?? 'unknown';\n\n    // kill the conn after the grace period if we haven't received a handshake\n    const handshakeTimeout = setTimeout(() => {\n      if (!session) {\n        log?.warn(\n          `connection to ${client()} timed out waiting for handshake, closing`,\n          {\n            clientId: this.clientId,\n            connectedTo: client(),\n            connId: conn.id,\n          },\n        );\n        conn.telemetry?.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: 'handshake timeout',\n        });\n        conn.close();\n      }\n    }, this.options.sessionDisconnectGraceMs);\n\n    const buffer: Array<Uint8Array> = [];\n    let receivedHandshakeMessage = false;\n\n    const handshakeHandler = (data: Uint8Array) => {\n      // if we've already received, just buffer the data\n      if (receivedHandshakeMessage) {\n        buffer.push(data);\n        return;\n      }\n\n      receivedHandshakeMessage = true;\n      clearTimeout(handshakeTimeout);\n\n      void this.receiveHandshakeRequestMessage(data, conn).then(\n        (maybeSession) => {\n          if (!maybeSession) {\n            conn.close();\n            return;\n          }\n\n          session = maybeSession;\n\n          // when we are done handshake sequence,\n          // remove handshake listener and use the normal message listener\n          const dataHandler = (data: Uint8Array) => {\n            const parsed = this.parseMsg(data);\n            if (!parsed) {\n              conn.close();\n              return;\n            }\n\n            this.handleMsg(parsed);\n          };\n\n          // process any data we missed\n          for (const data of buffer) {\n            dataHandler(data);\n          }\n\n          conn.removeDataListener(handshakeHandler);\n          conn.addDataListener(dataHandler);\n          buffer.length = 0;\n        },\n      );\n    };\n\n    conn.addDataListener(handshakeHandler);\n    conn.addCloseListener(() => {\n      if (!session) return;\n      log?.info(`connection to ${client()} disconnected`, {\n        clientId: this.clientId,\n        connId: conn.id,\n      });\n      this.onDisconnect(conn, session);\n    });\n\n    conn.addErrorListener((err) => {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'connection error',\n      });\n      if (!session) return;\n      log?.warn(\n        `connection to ${client()} got an error: ${coerceErrorString(err)}`,\n        { clientId: this.clientId, connId: conn.id },\n      );\n    });\n  }\n\n  private async validateHandshakeMetadata(\n    conn: ConnType,\n    session: Session<ConnType> | undefined,\n    rawMetadata: Static<\n      typeof ControlMessageHandshakeRequestSchema\n    >['metadata'],\n    from: TransportClientId,\n  ): Promise<ParsedMetadata | false> {\n    let parsedMetadata: ParsedMetadata = {};\n    if (this.handshakeExtensions) {\n      // check that the metadata that was sent is the correct shape\n      if (!Value.Check(this.handshakeExtensions.schema, rawMetadata)) {\n        conn.telemetry?.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: 'malformed handshake meta',\n        });\n        const reason = 'received malformed handshake metadata';\n        const responseMsg = handshakeResponseMessage(this.clientId, from, {\n          ok: false,\n          reason,\n        });\n        conn.send(this.codec.toBuffer(responseMsg));\n        // note: do _not_ log the metadata, it may contain secrets\n        log?.warn(`received malformed handshake metadata from ${from}`, {\n          clientId: this.clientId,\n          connId: conn.id,\n        });\n        this.protocolError(ProtocolError.HandshakeFailed, reason);\n        return false;\n      }\n\n      parsedMetadata = await this.handshakeExtensions.validate(\n        rawMetadata,\n        session,\n      );\n\n      // handler rejected the connection\n      if (parsedMetadata === false) {\n        const reason = 'rejected by handshake handler';\n        conn.telemetry?.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: reason,\n        });\n        const responseMsg = handshakeResponseMessage(this.clientId, from, {\n          ok: false,\n          reason,\n        });\n        conn.send(this.codec.toBuffer(responseMsg));\n        log?.warn(`rejected handshake from ${from}`, {\n          clientId: this.clientId,\n          connId: conn.id,\n        });\n        this.protocolError(ProtocolError.HandshakeFailed, reason);\n        return false;\n      }\n    }\n\n    return parsedMetadata;\n  }\n\n  async receiveHandshakeRequestMessage(\n    data: Uint8Array,\n    conn: ConnType,\n  ): Promise<Session<ConnType> | false> {\n    const parsed = this.parseMsg(data);\n    if (!parsed) {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'non-transport message',\n      });\n      this.protocolError(\n        ProtocolError.HandshakeFailed,\n        'received non-transport message',\n      );\n      return false;\n    }\n\n    if (!Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'invalid handshake request',\n      });\n      const reason = 'received invalid handshake msg';\n      const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n        ok: false,\n        reason,\n      });\n      conn.send(this.codec.toBuffer(responseMsg));\n      // note: do _not_ log the payload, it may contain secrets\n      const logData = { ...(parsed.payload ?? {}), metadata: 'redacted' };\n      log?.warn(reason, {\n        clientId: this.clientId,\n        connId: conn.id,\n        partialTransportMessage: { ...parsed, payload: logData },\n      });\n      this.protocolError(\n        ProtocolError.HandshakeFailed,\n        'invalid handshake request',\n      );\n      return false;\n    }\n\n    // double check protocol version here\n    const gotVersion = parsed.payload.protocolVersion;\n    if (gotVersion !== PROTOCOL_VERSION) {\n      conn.telemetry?.span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: 'incorrect protocol version',\n      });\n\n      const reason = `incorrect version (got: ${gotVersion} wanted ${PROTOCOL_VERSION})`;\n      const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n        ok: false,\n        reason,\n      });\n      conn.send(this.codec.toBuffer(responseMsg));\n      log?.warn(\n        `received handshake msg with incompatible protocol version (got: ${gotVersion}, expected: ${PROTOCOL_VERSION})`,\n        { clientId: this.clientId, connId: conn.id },\n      );\n      this.protocolError(ProtocolError.HandshakeFailed, reason);\n      return false;\n    }\n\n    const oldSession = this.sessions.get(parsed.from);\n    const parsedMetadata = await this.validateHandshakeMetadata(\n      conn,\n      oldSession,\n      parsed.payload.metadata,\n      parsed.from,\n    );\n\n    if (parsedMetadata === false) {\n      return false;\n    }\n\n    const { session, isReconnect } = this.getOrCreateSession(\n      parsed.from,\n      conn,\n      parsed.payload.sessionId,\n      parsed.tracing,\n    );\n\n    this.sessionHandshakeMetadata.set(session, parsedMetadata);\n\n    log?.debug(\n      `handshake from ${parsed.from} ok, responding with handshake success`,\n      { clientId: this.clientId, connId: conn.id },\n    );\n    const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n      ok: true,\n      sessionId: session.id,\n    });\n    conn.send(this.codec.toBuffer(responseMsg));\n    this.onConnect(conn, parsed.from, session, isReconnect);\n\n    return session;\n  }\n}\n","import { TransportClientId } from './message';\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: Map<TransportClientId, number>;\n  private intervalHandles: Map<\n    TransportClientId,\n    ReturnType<typeof setInterval>\n  >;\n  private readonly options: ConnectionRetryOptions;\n\n  constructor(options: ConnectionRetryOptions) {\n    this.options = options;\n    this.budgetConsumed = new Map();\n    this.intervalHandles = new Map();\n  }\n\n  getBackoffMs(user: TransportClientId) {\n    if (!this.budgetConsumed.has(user)) return 0;\n\n    const exponent = Math.max(0, this.getBudgetConsumed(user) - 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(user: TransportClientId) {\n    // If we're consuming again, let's ensure that we're not leaking\n    this.stopLeak(user);\n    this.budgetConsumed.set(user, this.getBudgetConsumed(user) + 1);\n  }\n\n  getBudgetConsumed(user: TransportClientId) {\n    return this.budgetConsumed.get(user) ?? 0;\n  }\n\n  hasBudget(user: TransportClientId) {\n    return this.getBudgetConsumed(user) < this.options.attemptBudgetCapacity;\n  }\n\n  startRestoringBudget(user: TransportClientId) {\n    if (this.intervalHandles.has(user)) {\n      return;\n    }\n\n    const restoreBudgetForUser = () => {\n      const currentBudget = this.budgetConsumed.get(user);\n      if (!currentBudget) {\n        this.stopLeak(user);\n        return;\n      }\n\n      const newBudget = currentBudget - 1;\n      if (newBudget === 0) {\n        this.budgetConsumed.delete(user);\n        return;\n      }\n\n      this.budgetConsumed.set(user, newBudget);\n    };\n\n    const intervalHandle = setInterval(\n      restoreBudgetForUser,\n      this.options.budgetRestoreIntervalMs,\n    );\n\n    this.intervalHandles.set(user, intervalHandle);\n  }\n\n  private stopLeak(user: TransportClientId) {\n    if (!this.intervalHandles.has(user)) {\n      return;\n    }\n\n    clearInterval(this.intervalHandles.get(user));\n    this.intervalHandles.delete(user);\n  }\n\n  close() {\n    for (const user of this.intervalHandles.keys()) {\n      this.stopLeak(user);\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,sBAAsB;AAiB/B,SAAS,sBAAsB;AAE/B,IAAM,SAAS,eAAe,uCAAuC,CAAC;AAC/D,IAAM,WAAW,MAAM,OAAO;AAU9B,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQ,OAAO,EAAE,CAAC;AAAA,EAC9B;AAwCF;AA6BO,IAAM,UAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKQ,aAA4C,CAAC;AAAA;AAAA;AAAA;AAAA,EAKrD;AAAA,EACS;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA;AAAA;AAAA;AAAA;AAAA,EAKQ,MAAsB;AAAA;AAAA;AAAA;AAAA,EAKtB,MAAsB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMtB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAER,YACE,MACA,MACA,IACA,SACA,gBACA;AACA,SAAK,KAAK,WAAW,OAAO,EAAE,CAAC;AAC/B,SAAK,UAAU;AACf,SAAK,OAAO;AACZ,SAAK,KAAK;AACV,SAAK,aAAa;AAClB,SAAK,QAAQ,QAAQ;AAGrB,SAAK,kBAAkB;AACvB,SAAK,YAAY;AAAA,MACf,MAAM,KAAK,cAAc;AAAA,MACzB,QAAQ;AAAA,IACV;AACA,SAAK,YAAY,2BAA2B,MAAM,cAAc;AAAA,EAClE;AAAA,EAEA,IAAI,kBAAsD;AACxD,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,QAAQ,KAAK,YAAY;AAAA,IAC3B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,KAAK,KAAsC;AACzC,UAAM,UAA4B,KAAK,aAAa,GAAG;AACvD,SAAK,MAAM,eAAe;AAAA,MACxB,GAAG,KAAK;AAAA,MACR,sBAAsB;AAAA,IACxB,CAAC;AAED,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,KAAK,WAAW,KAAK,KAAK,MAAM,SAAS,OAAO,CAAC;AAC5D,UAAI;AAAI,eAAO,QAAQ;AACvB,WAAK;AAAA,QACH,yBAAyB,QAAQ,EAAE;AAAA,QACnC;AAAA,UACE,GAAG,KAAK;AAAA,UACR,sBAAsB;AAAA,QACxB;AAAA,MACF;AAAA,IACF,OAAO;AACL,WAAK;AAAA,QACH,yBAAyB,QAAQ,EAAE;AAAA,QACnC,EAAE,GAAG,KAAK,iBAAiB,sBAAsB,QAAQ;AAAA,MAC3D;AAAA,IACF;AAEA,WAAO,QAAQ;AAAA,EACjB;AAAA,EAEA,gBAAgB;AACd,UAAM,SAAS,KAAK;AACpB,UAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,QAAI,SAAS,KAAK,QAAQ,qBAAqB;AAC7C,UAAI,KAAK,YAAY;AACnB,aAAK;AAAA,UACH,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,qBAAqB;AAAA,MAC5B;AACA;AAAA,IACF;AAEA,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AACD,SAAK;AAAA,EACP;AAAA,EAEA,wBAAwB;AACtB,SAAK,aAAa,CAAC;AACnB,SAAK,MAAM;AACX,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,qBAAqB,MAAgB;AACnC,SAAK,KAAK,aAAa,KAAK,WAAW,MAAM,sBAAsB;AAAA,MACjE,GAAG,KAAK;AAAA,MACR,QAAQ,KAAK;AAAA,IACf,CAAC;AACD,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,MAAM,iBAAiB;AAAA,QAC1B,GAAG,KAAK;AAAA,QACR,sBAAsB;AAAA,QACtB,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,YAAM,KAAK,KAAK,KAAK,KAAK,MAAM,SAAS,GAAG,CAAC;AAC7C,UAAI,CAAC,IAAI;AAGP,cAAM,SAAS,sCAAsC,KAAK,EAAE;AAC5D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,eAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,MAAM,QAAQ;AAAA,UACjB,GAAG,KAAK;AAAA,UACR,sBAAsB;AAAA,UACtB,QAAQ,KAAK;AAAA,UACb,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM;AACX;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,QAAI,MAAM,IAAI,KAAK,KAAK;AACtB,WAAK,MAAM,sBAAsB,GAAG,UAAU,KAAK,GAAG,IAAI;AAAA,QACxD,GAAG,KAAK;AAAA,QACR,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAEA,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AAAA,EACnB;AAAA,EAEA,qBAAqB,MAAiB;AACpC,QAAI,KAAK,eAAe,UAAa,KAAK,eAAe;AAAM;AAC/D,SAAK;AAAA,MACH,gDAAgD,KAAK,EAAE;AAAA,MACvD,KAAK;AAAA,IACP;AACA,SAAK,WAAW,MAAM;AACtB,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,yBAAyB,SAAmB;AAC1C,SAAK,qBAAqB,OAAO;AACjC,SAAK,YAAY;AACjB,SAAK,aAAa;AAClB,SAAK,qBAAqB,OAAO;AAAA,EACnC;AAAA,EAEA,WAAW,IAAgB;AACzB,SAAK;AAAA,MACH,YAAY,KAAK,QAAQ,wBAAwB,oCAAoC,KAAK,EAAE;AAAA,MAC5F,KAAK;AAAA,IACP;AACA,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,MAAM;AACX,SAAG;AAAA,IACL,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA;AAAA,EAGA,cAAc;AACZ,SAAK,kBAAkB;AACvB,iBAAa,KAAK,kBAAkB;AACpC,SAAK,qBAAqB;AAAA,EAC5B;AAAA;AAAA;AAAA,EAIA,QAAQ;AACN,SAAK,qBAAqB;AAC1B,SAAK,YAAY;AACjB,SAAK,sBAAsB;AAC3B,kBAAc,KAAK,SAAS;AAAA,EAC9B;AAAA,EAEA,IAAI,YAAY;AACd,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,SAAS;AAAA,MACb,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,SAAK,WAAW,KAAK,GAAG;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,oBAA2D;AACzD,WAAO,KAAK;AAAA,EACd;AACF;;;AClXO,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AA0BO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,kBAA+B,WAAc;AAC3C,WAAO,KAAK,eAAe,SAAS,GAAG,QAAQ;AAAA,EACjD;AAAA,EAEA,iBAA8B,WAAc,SAA0B;AACpE,QAAI,CAAC,KAAK,eAAe,SAAS,GAAG;AACnC,WAAK,eAAe,SAAS,IAAI,oBAAI,IAAI;AAAA,IAC3C;AAEA,SAAK,eAAe,SAAS,GAAG,IAAI,OAAO;AAAA,EAC7C;AAAA,EAEA,oBAAiC,WAAc,SAA0B;AACvE,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,WAAK,eAAe,SAAS,GAAG,OAAO,OAAO;AAAA,IAChD;AAAA,EACF;AAAA,EAEA,cAA2B,WAAc,OAAoB;AAC3D,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,iBAAW,WAAW,UAAU;AAC9B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;AChEA,SAAS,aAAa;;;AC2Cf,IAAM,uBAAN,MAA2B;AAAA,EACxB;AAAA,EACA;AAAA,EAIS;AAAA,EAEjB,YAAY,SAAiC;AAC3C,SAAK,UAAU;AACf,SAAK,iBAAiB,oBAAI,IAAI;AAC9B,SAAK,kBAAkB,oBAAI,IAAI;AAAA,EACjC;AAAA,EAEA,aAAa,MAAyB;AACpC,QAAI,CAAC,KAAK,eAAe,IAAI,IAAI;AAAG,aAAO;AAE3C,UAAM,WAAW,KAAK,IAAI,GAAG,KAAK,kBAAkB,IAAI,IAAI,CAAC;AAC7D,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,cAAc,MAAyB;AAErC,SAAK,SAAS,IAAI;AAClB,SAAK,eAAe,IAAI,MAAM,KAAK,kBAAkB,IAAI,IAAI,CAAC;AAAA,EAChE;AAAA,EAEA,kBAAkB,MAAyB;AACzC,WAAO,KAAK,eAAe,IAAI,IAAI,KAAK;AAAA,EAC1C;AAAA,EAEA,UAAU,MAAyB;AACjC,WAAO,KAAK,kBAAkB,IAAI,IAAI,KAAK,QAAQ;AAAA,EACrD;AAAA,EAEA,qBAAqB,MAAyB;AAC5C,QAAI,KAAK,gBAAgB,IAAI,IAAI,GAAG;AAClC;AAAA,IACF;AAEA,UAAM,uBAAuB,MAAM;AACjC,YAAM,gBAAgB,KAAK,eAAe,IAAI,IAAI;AAClD,UAAI,CAAC,eAAe;AAClB,aAAK,SAAS,IAAI;AAClB;AAAA,MACF;AAEA,YAAM,YAAY,gBAAgB;AAClC,UAAI,cAAc,GAAG;AACnB,aAAK,eAAe,OAAO,IAAI;AAC/B;AAAA,MACF;AAEA,WAAK,eAAe,IAAI,MAAM,SAAS;AAAA,IACzC;AAEA,UAAM,iBAAiB;AAAA,MACrB;AAAA,MACA,KAAK,QAAQ;AAAA,IACf;AAEA,SAAK,gBAAgB,IAAI,MAAM,cAAc;AAAA,EAC/C;AAAA,EAEQ,SAAS,MAAyB;AACxC,QAAI,CAAC,KAAK,gBAAgB,IAAI,IAAI,GAAG;AACnC;AAAA,IACF;AAEA,kBAAc,KAAK,gBAAgB,IAAI,IAAI,CAAC;AAC5C,SAAK,gBAAgB,OAAO,IAAI;AAAA,EAClC;AAAA,EAEA,QAAQ;AACN,eAAW,QAAQ,KAAK,gBAAgB,KAAK,GAAG;AAC9C,WAAK,SAAS,IAAI;AAAA,IACpB;AAAA,EACF;AACF;;;ADlGA,SAAS,kBAAAA,uBAAsB;AAexB,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AAAA,EACH,GAAG;AACL;AAMA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AACL;AAgCO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA;AAAA,EAK3D;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,cAAc;AAChB,WAAO,IAAI;AAAA,MACT,CAAC,GAAG,KAAK,QAAQ,EACd,IAAI,CAAC,CAAC,QAAQ,OAAO,MAAM,CAAC,QAAQ,QAAQ,UAAU,CAAC,EACvD,OAAO,CAAC,UAAuC,MAAM,CAAC,MAAM,MAAS;AAAA,IAC1E;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQV,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW,oBAAI,IAAI;AACxB,SAAK,QAAQ,KAAK,QAAQ;AAC1B,SAAK,WAAW;AAChB,SAAK,QAAQ;AAAA,EACf;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAoBU,UACR,MACA,aACA,SACA,aACA;AACA,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,SAAK,YAAY;AAAA,MACf;AAAA,MACA,QAAQ,UAAU;AAAA,IACpB;AAEA,QAAI,aAAa;AACf,cAAQ,yBAAyB,IAAI;AACrC,WAAK,KAAK,kBAAkB,WAAW,IAAI,QAAQ,eAAe;AAAA,IACpE;AAAA,EACF;AAAA,EAEU,cACR,IACA,MACA,gBACA;AACA,UAAM,UAAU,IAAI;AAAA,MAClB;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,MACL;AAAA,IACF;AACA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AACrC,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEU,mBACR,IACA,MACA,WACA,gBACA;AACA,QAAI,UAAU,KAAK,SAAS,IAAI,EAAE;AAClC,QAAI,cAAc,YAAY;AAE9B,QACE,SAAS,wBAAwB,UACjC,cAAc,UACd,QAAQ,wBAAwB,WAChC;AACA,WAAK;AAAA,QACH,eAAe,EAAE,6DAA6D,QAAQ,mBAAmB,UAAU,SAAS;AAAA,QAC5H,QAAQ;AAAA,MACV;AACA,WAAK,cAAc,OAAO;AAC1B,oBAAc;AACd,gBAAU;AAAA,IACZ;AAEA,QAAI,CAAC,SAAS;AACZ,gBAAU,KAAK,cAAc,IAAI,MAAM,cAAc;AACrD,WAAK;AAAA,QACH,kBAAkB,EAAE;AAAA,QACpB,QAAQ;AAAA,MACV;AAAA,IACF;AAEA,QAAI,cAAc,QAAW;AAC3B,cAAQ,sBAAsB;AAAA,IAChC;AAEA,WAAO,EAAE,SAAS,YAAY;AAAA,EAChC;AAAA,EAEU,cAAc,SAA4B;AAClD,YAAQ,MAAM;AACd,YAAQ,UAAU,KAAK,IAAI;AAC3B,SAAK,SAAS,OAAO,QAAQ,EAAE;AAC/B,SAAK;AAAA,MACH,WAAW,QAAQ,EAAE,oBAAoB,QAAQ,EAAE;AAAA,MACnD,QAAQ;AAAA,IACV;AACA,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,aAAa,MAAgB,SAA4B;AACjE,SAAK,WAAW,KAAK,IAAI;AACzB,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,YAAQ,aAAa;AACrB,YAAQ,WAAW,MAAM;AACvB,cAAQ,UAAU,KAAK,SAAS,8BAA8B;AAC9D,WAAK,cAAc,OAAO;AAAA,IAC5B,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,SAAS,KAAgD;AACjE,UAAM,YAAY,KAAK,MAAM,WAAW,GAAG;AAE3C,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,MAAM,yCAAyC,aAAa,IAAI;AAAA,QACnE,UAAU,KAAK;AAAA,MACjB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QAC/D,UAAU,KAAK;AAAA,MACjB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAU,KAA6B;AAC/C,QAAI,KAAK,UAAU;AAAQ;AAC3B,UAAM,UAAU,KAAK,SAAS,IAAI,IAAI,IAAI;AAC1C,QAAI,CAAC,SAAS;AACZ,WAAK,MAAM,2BAA2B,IAAI,IAAI,IAAI;AAAA,QAChD,UAAU,KAAK;AAAA,QACf,sBAAsB;AAAA,QACtB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAGA,YAAQ,YAAY;AAEpB,SAAK,MAAM,gBAAgB;AAAA,MACzB,UAAU,KAAK;AAAA,MACf,sBAAsB;AAAA,IACxB,CAAC;AACD,QAAI,IAAI,QAAQ,QAAQ,iBAAiB;AACvC,UAAI,IAAI,MAAM,QAAQ,iBAAiB;AACrC,aAAK;AAAA,UACH,oCAAoC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AAAA,UACnF,EAAE,UAAU,KAAK,UAAU,sBAAsB,IAAI;AAAA,QACvD;AAAA,MACF,OAAO;AACL,cAAM,SAAS,uCAAuC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AACrG,aAAK,MAAM,GAAG,MAAM,gCAAgC;AAAA,UAClD,UAAU,KAAK;AAAA,UACf,sBAAsB;AAAA,UACtB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,cAAc,cAAc,yBAAyB,MAAM;AAChE,gBAAQ,UAAU,KAAK,UAAU;AAAA,UAC/B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,gBAAQ,MAAM;AAAA,MAChB;AAEA;AAAA,IACF;AAEA,YAAQ,kBAAkB,IAAI,KAAK,IAAI,GAAG;AAG1C,QAAI,CAAC,MAAM,IAAI,YAAY,GAAG;AAC5B,WAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,IACnD,OAAO;AACL,WAAK,MAAM,gCAAgC;AAAA,QACzC,UAAU,KAAK;AAAA,QACf,sBAAsB;AAAA,MACxB,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBACE,MACA,SACM;AACN,SAAK,gBAAgB,iBAAiB,MAAM,OAAO;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,oBACE,MACA,SACM;AACN,SAAK,gBAAgB,oBAAoB,MAAM,OAAO;AAAA,EACxD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,KACE,IACA,KACoB;AACpB,QAAI,KAAK,UAAU,aAAa;AAC9B,YAAM,MAAM;AACZ,WAAK,MAAM,KAAK;AAAA,QACd,UAAU,KAAK;AAAA,QACf,yBAAyB;AAAA,QACzB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD,WAAK,cAAc,cAAc,iBAAiB,GAAG;AACrD,aAAO;AAAA,IACT,WAAW,KAAK,UAAU,UAAU;AAClC,WAAK,KAAK,6CAA6C;AAAA,QACrD,UAAU,KAAK;AAAA,QACf,yBAAyB;AAAA,MAC3B,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO,KAAK,mBAAmB,EAAE,EAAE,QAAQ,KAAK,GAAG;AAAA,EACrD;AAAA;AAAA,EAGA,gBAAgB,IAAuB,UAAkB;AACvD,WAAO,KAAK,KAAK,IAAI;AAAA,MACnB;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEU,cAAc,MAAyB,SAAiB;AAChE,SAAK,gBAAgB,cAAc,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAAA,EACvE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAU;AACR,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,gCAAgC,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACvE;AACF;AAEO,IAAe,kBAAf,cAGG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,4BAA4B;AAAA;AAAA;AAAA;AAAA,EAK5B;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,6BAA6B,oBAAI,IAAI;AAC1C,SAAK,cAAc,IAAI,qBAAqB,KAAK,OAAO;AAAA,EAC1D;AAAA,EAEA,gBAAgB,SAAoD;AAClE,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB,IAA6B;AACtE,QAAI,KAAK,UAAU;AAAQ;AAC3B,QAAI,UAAyC;AAG7C,UAAM,mBAAmB,WAAW,MAAM;AACxC,UAAI,CAAC,SAAS;AACZ,aAAK;AAAA,UACH,iBAAiB,EAAE;AAAA,UACnB,EAAE,UAAU,KAAK,UAAU,aAAa,IAAI,QAAQ,KAAK,GAAG;AAAA,QAC9D;AACA,aAAK,MAAM;AAAA,MACb;AAAA,IACF,GAAG,KAAK,QAAQ,wBAAwB;AAExC,UAAM,mBAAmB,CAAC,SAAqB;AAC7C,YAAM,eAAe,KAAK,gCAAgC,MAAM,IAAI;AACpE,UAAI,CAAC,cAAc;AACjB,aAAK,MAAM;AACX;AAAA,MACF,OAAO;AACL,kBAAU;AACV,qBAAa,gBAAgB;AAAA,MAC/B;AAIA,WAAK,mBAAmB,gBAAgB;AACxC,WAAK,gBAAgB,CAACC,UAAS;AAC7B,cAAM,SAAS,KAAK,SAASA,KAAI;AACjC,YAAI,CAAC,QAAQ;AACX,eAAK,WAAW,KAAK,UAAU;AAAA,YAC7B,MAAMD,gBAAe;AAAA,YACrB,SAAS;AAAA,UACX,CAAC;AACD,eAAK,MAAM;AACX;AAAA,QACF;AAEA,aAAK,UAAU,MAAM;AAAA,MACvB,CAAC;AAAA,IACH;AAEA,SAAK,gBAAgB,gBAAgB;AACrC,SAAK,iBAAiB,MAAM;AAC1B,UAAI,SAAS;AACX,aAAK,aAAa,MAAM,OAAO;AAAA,MACjC;AACA,WAAK,KAAK,iBAAiB,EAAE,iBAAiB;AAAA,QAC5C,GAAG,SAAS;AAAA,QACZ,UAAU,KAAK;AAAA,QACf,aAAa;AAAA,MACf,CAAC;AACD,WAAK,2BAA2B,OAAO,EAAE;AACzC,UAAI,KAAK,2BAA2B;AAClC,aAAK,KAAK,QAAQ,EAAE;AAAA,MACtB;AAAA,IACF,CAAC;AACD,SAAK,iBAAiB,CAAC,QAAQ;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK,KAAK,0BAA0B,EAAE,KAAK,kBAAkB,GAAG,CAAC,IAAI;AAAA,QACnE,GAAG,SAAS;AAAA,QACZ,UAAU,KAAK;AAAA,QACf,aAAa;AAAA,MACf,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAAA,EAEA,gCACE,MACA,MAC2B;AAC3B,UAAM,SAAS,KAAK,SAAS,IAAI;AACjC,QAAI,CAAC,QAAQ;AACX,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,uCAAuC,OAAO,OAAO,GAAG;AACvE,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK,KAAK,mCAAmC;AAAA,QAC3C,UAAU,KAAK;AAAA,QACf,aAAa,OAAO;AAAA,QACpB,sBAAsB;AAAA,MACxB,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,OAAO,QAAQ,OAAO,IAAI;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK,KAAK,gCAAgC;AAAA,QACxC,UAAU,KAAK;AAAA,QACf,aAAa,OAAO;AAAA,QACpB,sBAAsB;AAAA,MACxB,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd,OAAO,QAAQ,OAAO;AAAA,MACxB;AACA,aAAO;AAAA,IACT;AAEA,SAAK,MAAM,kBAAkB,OAAO,IAAI,OAAO;AAAA,MAC7C,UAAU,KAAK;AAAA,MACf,aAAa,OAAO;AAAA,MACpB,sBAAsB;AAAA,IACxB,CAAC;AAED,UAAM,EAAE,SAAS,YAAY,IAAI,KAAK;AAAA,MACpC,OAAO;AAAA,MACP;AAAA,MACA,OAAO,QAAQ,OAAO;AAAA,IACxB;AAEA,SAAK,UAAU,MAAM,OAAO,MAAM,SAAS,WAAW;AAKtD,SAAK,YAAY,qBAAqB,OAAO,IAAI;AACjD,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAM,QAAQ,IAAsC;AAClD,UAAM,2BAA2B,MAAM,KAAK,UAAU;AACtD,QAAI,CAAC,yBAAyB,GAAG;AAC/B,WAAK;AAAA,QACH,uEAAuE,EAAE;AAAA,QACzE,EAAE,UAAU,KAAK,UAAU,aAAa,GAAG;AAAA,MAC7C;AACA;AAAA,IACF;AAEA,QAAI,mBAAmB,KAAK,2BAA2B,IAAI,EAAE;AAC7D,QAAI,CAAC,kBAAkB;AAErB,YAAM,iBAAiB,KAAK,YAAY,kBAAkB,EAAE;AAC5D,UAAI,CAAC,KAAK,YAAY,UAAU,EAAE,GAAG;AACnC,cAAM,SAAS,uBAAuB,EAAE,yCAAyC,cAAc,yBAAyB,KAAK,YAAY,sBAAsB;AAC/J,aAAK,KAAK,QAAQ,EAAE,UAAU,KAAK,UAAU,aAAa,GAAG,CAAC;AAC9D,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD;AAAA,MACF;AAEA,UAAI,QAAQ,QAAQ,QAAQ;AAC5B,YAAM,YAAY,KAAK,YAAY,aAAa,EAAE;AAClD,UAAI,YAAY,GAAG;AACjB,gBAAQ,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,SAAS,CAAC;AAAA,MACjE;AAEA,WAAK,KAAK,4BAA4B,EAAE,KAAK,SAAS,eAAe;AAAA,QACnE,UAAU,KAAK;AAAA,QACf,aAAa;AAAA,MACf,CAAC;AACD,WAAK,YAAY,cAAc,EAAE;AACjC,yBAAmB,gBAAO,gBAAgB,WAAW,OAAO,SAAS;AACnE,YAAI;AACF,eAAK,SAAS,WAAW,EAAE,UAAU,CAAC;AACtC,gBAAM;AACN,cAAI,CAAC,yBAAyB,GAAG;AAC/B,kBAAM,IAAI,MAAM,mCAAmC;AAAA,UACrD;AAEA,eAAK,SAAS,YAAY;AAC1B,gBAAM,OAAO,MAAM,KAAK,4BAA4B,EAAE;AACtD,cAAI,CAAC,yBAAyB,GAAG;AAC/B,iBAAK;AAAA,cACH,0EAA0E,EAAE;AAAA,cAC5E;AAAA,gBACE,UAAU,KAAK;AAAA,gBACf,aAAa;AAAA,gBACb,QAAQ,KAAK;AAAA,cACf;AAAA,YACF;AACA,iBAAK,MAAM;AACX,kBAAM,IAAI,MAAM,mCAAmC;AAAA,UACrD;AAEA,eAAK,SAAS,mBAAmB;AACjC,gBAAM,KAAK,MAAM,KAAK,cAAc,IAAI,IAAI;AAC5C,cAAI,CAAC,IAAI;AACP,iBAAK,MAAM;AACX,kBAAM,IAAI,MAAM,0BAA0B;AAAA,UAC5C;AAEA,iBAAO;AAAA,QACT,SAAS,KAAK;AAGZ,gBAAM,SAAS,kBAAkB,GAAG;AACpC,eAAK,gBAAgB,MAAM;AAC3B,eAAK,UAAU,EAAE,MAAMA,gBAAe,MAAM,CAAC;AAC7C,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF,CAAC;AAED,WAAK,2BAA2B,IAAI,IAAI,gBAAgB;AAAA,IAC1D,OAAO;AACL,WAAK,KAAK,4BAA4B,EAAE,+BAA+B;AAAA,QACrE,UAAU,KAAK;AAAA,QACf,aAAa;AAAA,MACf,CAAC;AAAA,IACH;AAEA,QAAI;AACF,YAAM;AAAA,IACR,SAAS,OAAgB;AACvB,WAAK,2BAA2B,OAAO,EAAE;AACzC,YAAM,SAAS,kBAAkB,KAAK;AAEtC,UAAI,CAAC,KAAK,6BAA6B,CAAC,yBAAyB,GAAG;AAClE,aAAK,KAAK,iBAAiB,EAAE,YAAY,MAAM,KAAK;AAAA,UAClD,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf,CAAC;AAAA,MACH,OAAO;AACL,aAAK,KAAK,iBAAiB,EAAE,YAAY,MAAM,eAAe;AAAA,UAC5D,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf,CAAC;AACD,eAAO,KAAK,QAAQ,EAAE;AAAA,MACxB;AAAA,IACF;AAAA,EACF;AAAA,EAEU,cAAc,SAA4B;AAClD,SAAK,2BAA2B,OAAO,QAAQ,EAAE;AACjD,UAAM,cAAc,OAAO;AAAA,EAC7B;AAAA,EAEA,MAAgB,cAAc,IAAuB,MAAgB;AACnE,QAAI;AAEJ,QAAI,KAAK,qBAAqB;AAC5B,iBAAW,MAAM,KAAK,oBAAoB,UAAU;AACpD,UAAI,CAAC,MAAM,MAAM,KAAK,oBAAoB,QAAQ,QAAQ,GAAG;AAC3D,aAAK,MAAM,uDAAuD;AAAA,UAChE,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,UACb,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK;AAAA,UACH,cAAc;AAAA,UACd;AAAA,QACF;AACA,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,eAAO;AAAA,MACT;AAAA,IACF;AAEA,UAAM,EAAE,QAAQ,IAAI,KAAK,mBAAmB,IAAI,IAAI;AACpD,UAAM,aAAa;AAAA,MACjB,KAAK;AAAA,MACL;AAAA,MACA,QAAQ;AAAA,MACR;AAAA,MACA,sBAAsB,QAAQ,UAAU,GAAG;AAAA,IAC7C;AACA,SAAK,MAAM,gCAAgC,EAAE,IAAI;AAAA,MAC/C,UAAU,KAAK;AAAA,MACf,aAAa;AAAA,IACf,CAAC;AACD,SAAK,KAAK,KAAK,MAAM,SAAS,UAAU,CAAC;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ;AACN,SAAK,YAAY,MAAM;AACvB,UAAM,MAAM;AAAA,EACd;AACF;AAEO,IAAe,kBAAf,cAGG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,2BAA2B,oBAAI,QAAQ;AAC5C,SAAK,KAAK,8BAA8B;AAAA,MACtC,UAAU,KAAK;AAAA,MACf,iBAAiB;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,SAAuD;AACrE,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB;AACzC,QAAI,KAAK,UAAU;AAAQ;AAE3B,SAAK,KAAK,2BAA2B;AAAA,MACnC,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,QAAI,UAAyC;AAC7C,UAAM,SAAS,MAAM,SAAS,MAAM;AAGpC,UAAM,mBAAmB,WAAW,MAAM;AACxC,UAAI,CAAC,SAAS;AACZ,aAAK;AAAA,UACH,iBAAiB,OAAO,CAAC;AAAA,UACzB;AAAA,YACE,UAAU,KAAK;AAAA,YACf,aAAa,OAAO;AAAA,YACpB,QAAQ,KAAK;AAAA,UACf;AAAA,QACF;AACA,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,aAAK,MAAM;AAAA,MACb;AAAA,IACF,GAAG,KAAK,QAAQ,wBAAwB;AAExC,UAAM,SAA4B,CAAC;AACnC,QAAI,2BAA2B;AAE/B,UAAM,mBAAmB,CAAC,SAAqB;AAE7C,UAAI,0BAA0B;AAC5B,eAAO,KAAK,IAAI;AAChB;AAAA,MACF;AAEA,iCAA2B;AAC3B,mBAAa,gBAAgB;AAE7B,WAAK,KAAK,+BAA+B,MAAM,IAAI,EAAE;AAAA,QACnD,CAAC,iBAAiB;AAChB,cAAI,CAAC,cAAc;AACjB,iBAAK,MAAM;AACX;AAAA,UACF;AAEA,oBAAU;AAIV,gBAAM,cAAc,CAACC,UAAqB;AACxC,kBAAM,SAAS,KAAK,SAASA,KAAI;AACjC,gBAAI,CAAC,QAAQ;AACX,mBAAK,MAAM;AACX;AAAA,YACF;AAEA,iBAAK,UAAU,MAAM;AAAA,UACvB;AAGA,qBAAWA,SAAQ,QAAQ;AACzB,wBAAYA,KAAI;AAAA,UAClB;AAEA,eAAK,mBAAmB,gBAAgB;AACxC,eAAK,gBAAgB,WAAW;AAChC,iBAAO,SAAS;AAAA,QAClB;AAAA,MACF;AAAA,IACF;AAEA,SAAK,gBAAgB,gBAAgB;AACrC,SAAK,iBAAiB,MAAM;AAC1B,UAAI,CAAC;AAAS;AACd,WAAK,KAAK,iBAAiB,OAAO,CAAC,iBAAiB;AAAA,QAClD,UAAU,KAAK;AAAA,QACf,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,WAAK,aAAa,MAAM,OAAO;AAAA,IACjC,CAAC;AAED,SAAK,iBAAiB,CAAC,QAAQ;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMD,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,UAAI,CAAC;AAAS;AACd,WAAK;AAAA,QACH,iBAAiB,OAAO,CAAC,kBAAkB,kBAAkB,GAAG,CAAC;AAAA,QACjE,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,MAC7C;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,0BACZ,MACA,SACA,aAGA,MACiC;AACjC,QAAI,iBAAiC,CAAC;AACtC,QAAI,KAAK,qBAAqB;AAE5B,UAAI,CAAC,MAAM,MAAM,KAAK,oBAAoB,QAAQ,WAAW,GAAG;AAC9D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,SAAS;AACf,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAE1C,aAAK,KAAK,8CAA8C,IAAI,IAAI;AAAA,UAC9D,UAAU,KAAK;AAAA,UACf,QAAQ,KAAK;AAAA,QACf,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAEA,uBAAiB,MAAM,KAAK,oBAAoB;AAAA,QAC9C;AAAA,QACA;AAAA,MACF;AAGA,UAAI,mBAAmB,OAAO;AAC5B,cAAM,SAAS;AACf,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,aAAK,KAAK,2BAA2B,IAAI,IAAI;AAAA,UAC3C,UAAU,KAAK;AAAA,UACf,QAAQ,KAAK;AAAA,QACf,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,+BACJ,MACA,MACoC;AACpC,UAAM,SAAS,KAAK,SAAS,IAAI;AACjC,QAAI,CAAC,QAAQ;AACX,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,sCAAsC,OAAO,OAAO,GAAG;AACtE,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,YAAM,SAAS;AACf,YAAME,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAE1C,YAAM,UAAU,EAAE,GAAI,OAAO,WAAW,CAAC,GAAI,UAAU,WAAW;AAClE,WAAK,KAAK,QAAQ;AAAA,QAChB,UAAU,KAAK;AAAA,QACf,QAAQ,KAAK;AAAA,QACb,yBAAyB,EAAE,GAAG,QAAQ,SAAS,QAAQ;AAAA,MACzD,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAGA,UAAM,aAAa,OAAO,QAAQ;AAClC,QAAI,eAAe,kBAAkB;AACnC,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMF,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AAED,YAAM,SAAS,2BAA2B,UAAU,WAAW,gBAAgB;AAC/E,YAAME,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAC1C,WAAK;AAAA,QACH,mEAAmE,UAAU,eAAe,gBAAgB;AAAA,QAC5G,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,MAC7C;AACA,WAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,aAAO;AAAA,IACT;AAEA,UAAM,aAAa,KAAK,SAAS,IAAI,OAAO,IAAI;AAChD,UAAM,iBAAiB,MAAM,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,QAAI,mBAAmB,OAAO;AAC5B,aAAO;AAAA,IACT;AAEA,UAAM,EAAE,SAAS,YAAY,IAAI,KAAK;AAAA,MACpC,OAAO;AAAA,MACP;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,SAAK,yBAAyB,IAAI,SAAS,cAAc;AAEzD,SAAK;AAAA,MACH,kBAAkB,OAAO,IAAI;AAAA,MAC7B,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,IAC7C;AACA,UAAM,cAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,MACvE,IAAI;AAAA,MACJ,WAAW,QAAQ;AAAA,IACrB,CAAC;AACD,SAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,SAAK,UAAU,MAAM,OAAO,MAAM,SAAS,WAAW;AAEtD,WAAO;AAAA,EACT;AACF;","names":["SpanStatusCode","data","responseMsg"]}

package/dist/{chunk-3MFX6NXA.js → chunk-PCKHBAVP.js}

@@ -104,7 +104,7 @@ function isStreamClose(controlFlag) {
 }
 
 // package.json
-var version = "0.22.0";
+var version = "0.23.0";
 
 // util/stringify.ts
 function coerceErrorString(err) {
@@ -222,4 +222,4 @@ export {
   tracing_default,
   coerceErrorString
 };
-//# sourceMappingURL=chunk-3MFX6NXA.js.map
+//# sourceMappingURL=chunk-PCKHBAVP.js.map

package/dist/chunk-PCKHBAVP.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/message.ts","../package.json","../util/stringify.ts","../tracing/index.ts"],"sourcesContent":["import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport { Connection, Session } from './session';\nimport { PropagationContext } from '../tracing';\nimport { ParsedMetadata } from '../router/context';\n\n/**\n * Control flags for transport messages.\n * An RPC message is coded with StreamOpenBit | StreamClosedBit.\n * Streams are expected to start with StreamOpenBit sent and the client SHOULD send an empty\n * message with StreamClosedBit to close the stream handler on the server, indicating that\n * it will not be using the stream anymore.\n */\nexport const enum ControlFlags {\n  AckBit = 0b0001,\n  StreamOpenBit = 0b0010,\n  StreamClosedBit = 0b0100,\n}\n\nexport interface ClientHandshakeOptions<MetadataSchema extends TSchema> {\n  /**\n   * Schema for the metadata that the client sends to the server\n   * during the handshake.\n   */\n  schema: MetadataSchema;\n\n  /**\n   * Gets the {@link HandshakeRequestMetadata} to send to the server.\n   */\n  construct: () => Static<MetadataSchema> | Promise<Static<MetadataSchema>>;\n}\n\nexport interface ServerHandshakeOptions<MetadataSchema extends TSchema> {\n  /**\n   * Schema for the metadata that the server receives from the client\n   * during the handshake.\n   */\n  schema: MetadataSchema;\n\n  /**\n   * Parses the {@link HandshakeRequestMetadata} sent by the client, transforming\n   * it into {@link ParsedHandshakeMetadata}.\n   *\n   * May return `false` if the client should be rejected.\n   *\n   * @param metadata - The metadata sent by the client.\n   * @param session - The session that the client would be associated with.\n   * @param isReconnect - Whether the client is reconnecting to the session,\n   *                      or if this is a new session.\n   */\n  validate: (\n    metadata: Static<MetadataSchema>,\n    session?: Session<Connection>,\n  ) => false | ParsedMetadata | Promise<false | ParsedMetadata>;\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 const PROTOCOL_VERSION = 'v1.1';\nexport const ControlMessageHandshakeRequestSchema = Type.Object({\n  type: Type.Literal('HANDSHAKE_REQ'),\n  protocolVersion: Type.String(),\n  sessionId: Type.String(),\n  metadata: Type.Optional(Type.Unknown()),\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    }),\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: string;\n  to: string;\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: TransportClientId,\n  to: TransportClientId,\n  sessionId: string,\n  metadata?: unknown,\n  tracing?: PropagationContext,\n): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n  return {\n    id: nanoid(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: nanoid(),\n    controlFlags: 0,\n    tracing,\n    payload: {\n      type: 'HANDSHAKE_REQ',\n      protocolVersion: PROTOCOL_VERSION,\n      sessionId,\n      metadata,\n    } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n  };\n}\n\nexport function handshakeResponseMessage(\n  from: TransportClientId,\n  to: TransportClientId,\n  status: Static<typeof ControlMessageHandshakeResponseSchema>['status'],\n): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n  return {\n    id: nanoid(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: nanoid(),\n    controlFlags: 0,\n    payload: {\n      type: 'HANDSHAKE_RESP',\n      status,\n    } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\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  \"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.23.0\",\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    \"it-pushable\": \"^3.2.3\",\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    \"@types/ws\": \"^8.5.5\",\n    \"@typescript-eslint/eslint-plugin\": \"^7.1.0\",\n    \"@typescript-eslint/parser\": \"^7.1.0\",\n    \"@vitest/ui\": \"^1.3.1\",\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.2.2\",\n    \"vitest\": \"^1.3.1\",\n    \"vscode-langservers-extracted\": \"4.8.0\"\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 --test-timeout=500\",\n    \"test:single\": \"vitest run --test-timeout=500 --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 { Connection, OpaqueTransportMessage, Session } 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  return tracing;\n}\n\nexport function createSessionTelemetryInfo(\n  session: Session<Connection>,\n  propagationCtx?: PropagationContext,\n): TelemetryInfo {\n  const ctx = propagationCtx\n    ? propagation.extract(context.active(), propagationCtx)\n    : context.active();\n\n  const span = tracer.startSpan(\n    `session ${session.id}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.session.id': session.id,\n        'river.session.to': session.to,\n        'river.session.from': session.from,\n      },\n    },\n    ctx,\n  );\n\n  return { span, ctx };\n}\n\nexport function createConnectionTelemetryInfo(\n  connection: Connection,\n  sessionSpan: Span,\n): TelemetryInfo {\n  const ctx = trace.setSpan(context.active(), sessionSpan);\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: sessionSpan.spanContext() }],\n    },\n    ctx,\n  );\n\n  return { span, ctx };\n}\n\nexport function createProcTelemetryInfo(\n  kind: ValidProcType,\n  serviceName: string,\n  procedureName: string,\n  streamId: string,\n): TelemetryInfo {\n  const ctx = 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    ctx,\n  );\n\n  return { span, ctx };\n}\n\nexport function createHandlerSpan(\n  kind: ValidProcType,\n  message: OpaqueTransportMessage,\n  fn: (span: Span) => Promise<unknown>,\n) {\n  const ctx = message.tracing\n    ? propagation.extract(context.active(), message.tracing)\n    : context.active();\n\n  return tracer.startActiveSpan(\n    `procedure handler ${message.serviceName}.${message.procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': message.serviceName,\n        'river.method.name': message.procedureName,\n        'river.streamId': message.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;AACtC,SAAS,cAAc;AA6DhB,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;AAEM,IAAM,mBAAmB;AACzB,IAAM,uCAAuC,KAAK,OAAO;AAAA,EAC9D,MAAM,KAAK,QAAQ,eAAe;AAAA,EAClC,iBAAiB,KAAK,OAAO;AAAA,EAC7B,WAAW,KAAK,OAAO;AAAA,EACvB,UAAU,KAAK,SAAS,KAAK,QAAQ,CAAC;AACxC,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,IACtB,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,wBACd,MACA,IACA,WACA,UACA,SACuE;AACvE,SAAO;AAAA,IACL,IAAI,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,cAAc;AAAA,IACd;AAAA,IACA,SAAS;AAAA,MACP,MAAM;AAAA,MACN,iBAAiB;AAAA,MACjB;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,yBACd,MACA,IACA,QACwE;AACxE,SAAO;AAAA,IACL,IAAI,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;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;;;AC7PE,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;AAC/B,SAAO;AACT;AAEO,SAAS,2BACd,SACA,gBACe;AACf,QAAM,MAAM,iBACR,YAAY,QAAQ,QAAQ,OAAO,GAAG,cAAc,IACpD,QAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,QAAQ,EAAE;AAAA,IACrB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB,QAAQ;AAAA,QAC5B,oBAAoB,QAAQ;AAAA,QAC5B,sBAAsB,QAAQ;AAAA,MAChC;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,8BACd,YACA,aACe;AACf,QAAM,MAAM,MAAM,QAAQ,QAAQ,OAAO,GAAG,WAAW;AACvD,QAAM,OAAO,OAAO;AAAA,IAClB,cAAc,WAAW,EAAE;AAAA,IAC3B;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,uBAAuB,WAAW;AAAA,MACpC;AAAA,MACA,OAAO,CAAC,EAAE,SAAS,YAAY,YAAY,EAAE,CAAC;AAAA,IAChD;AAAA,IACA;AAAA,EACF;AAEA,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,wBACd,MACA,aACA,eACA,UACe;AACf,QAAM,MAAM,QAAQ,OAAO;AAC3B,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,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,kBACd,MACA,SACA,IACA;AACA,QAAM,MAAM,QAAQ,UAChB,YAAY,QAAQ,QAAQ,OAAO,GAAG,QAAQ,OAAO,IACrD,QAAQ,OAAO;AAEnB,SAAO,OAAO;AAAA,IACZ,qBAAqB,QAAQ,WAAW,IAAI,QAAQ,aAAa;AAAA,IACjE;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB,QAAQ;AAAA,QAChC,qBAAqB,QAAQ;AAAA,QAC7B,kBAAkB,QAAQ;AAAA,QAC1B,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":[]}

package/dist/{chunk-GCLEWC26.js → chunk-VOJVLWVX.js}

@@ -6,21 +6,25 @@ import {
   getPropagationContext,
   isStreamClose,
   isStreamOpen
-} from "./chunk-3MFX6NXA.js";
+} from "./chunk-PCKHBAVP.js";
 import {
   log
 } from "./chunk-OTQNCLFH.js";
 
 // router/services.ts
 import { Type } from "@sinclair/typebox";
-function serializeSchema(services) {
-  return Object.entries(services).reduce(
-    (acc, [name, value]) => {
-      acc[name] = value.serialize();
-      return acc;
-    },
-    {}
-  );
+function serializeSchema(services, handshakeSchema) {
+  const serializedServiceObject = Object.entries(services).reduce((acc, [name, value]) => {
+    acc[name] = value.serialize();
+    return acc;
+  }, {});
+  const schema = {
+    services: serializedServiceObject
+  };
+  if (handshakeSchema) {
+    schema.handshakeSchema = Type.Strict(handshakeSchema);
+  }
+  return schema;
 }
 var ServiceSchema = class _ServiceSchema {
   /**
@@ -636,7 +640,10 @@ var defaultClientOptions = {
   connectOnInvoke: true,
   eagerlyConnect: true
 };
-var createClient = (transport, serverId, providedClientOptions = {}) => {
+function createClient(transport, serverId, providedClientOptions = {}) {
+  if (providedClientOptions.handshakeOptions) {
+    transport.extendHandshake(providedClientOptions.handshakeOptions);
+  }
   const options = { ...defaultClientOptions, ...providedClientOptions };
   if (options.eagerlyConnect) {
     void transport.connect(serverId);
@@ -672,7 +679,7 @@ var createClient = (transport, serverId, providedClientOptions = {}) => {
       throw new Error(`invalid river call, unknown procedure type ${procType}`);
     }
   }, []);
-};
+}
 function createSessionDisconnectHandler(from, cb) {
   return (evt) => {
     if (evt.status === "disconnect" && evt.session.to === from) {
@@ -941,7 +948,7 @@ var RiverServer = class {
   // map of client to their open streams by streamId
   clientStreams;
   disconnectedSessions;
-  constructor(transport, services, extendedContext) {
+  constructor(transport, services, handshakeOptions, extendedContext) {
     const instances = {};
     this.services = instances;
     this.contextMap = /* @__PURE__ */ new Map();
@@ -953,6 +960,9 @@ var RiverServer = class {
         state: instance.state
       });
     }
+    if (handshakeOptions) {
+      transport.extendHandshake(handshakeOptions);
+    }
     this.transport = transport;
     this.disconnectedSessions = /* @__PURE__ */ new Set();
     this.streamMap = /* @__PURE__ */ new Map();
@@ -1109,7 +1119,8 @@ var RiverServer = class {
         })
       );
     };
-    if (session.metadata === void 0) {
+    const sessionMeta = this.transport.sessionHandshakeMetadata.get(session);
+    if (!sessionMeta) {
       log?.error(`session doesn't have handshake metadata`, {
         ...session.loggingMetadata,
         tags: ["invariant-violation"]
@@ -1123,8 +1134,8 @@ var RiverServer = class {
       to: message.to,
       from: message.from,
       streamId: message.streamId,
-      // we've already validated that the session has handshake metadata
-      session
+      session,
+      metadata: sessionMeta
     };
     switch (procedure.type) {
       case "rpc":
@@ -1344,8 +1355,13 @@ var RiverServer = class {
     this.streamMap.delete(id);
   };
 };
-function createServer(transport, services, extendedContext) {
-  return new RiverServer(transport, services, extendedContext);
+function createServer(transport, services, providedServerOptions) {
+  return new RiverServer(
+    transport,
+    services,
+    providedServerOptions?.handshakeOptions,
+    providedServerOptions?.extendedContext
+  );
 }
 
 export {
@@ -1360,4 +1376,4 @@ export {
   createClient,
   createServer
 };
-//# sourceMappingURL=chunk-GCLEWC26.js.map
+//# sourceMappingURL=chunk-VOJVLWVX.js.map