@replit/river 0.26.2 → 0.26.4

package/dist/{chunk-IV27BICV.js → chunk-HPOR6FST.js}

@@ -1,11 +1,11 @@
 import {
   ProtocolError,
   Transport
-} from "./chunk-M5X4JTU3.js";
+} from "./chunk-RSCUADHY.js";
 import {
   ClientSessionStateGraph,
   defaultClientTransportOptions
-} from "./chunk-NC54BC47.js";
+} from "./chunk-42SXIIHP.js";
 import {
   ControlMessageHandshakeResponseSchema,
   HandshakeErrorRetriableResponseCodes,
@@ -13,7 +13,7 @@ import {
   getPropagationContext,
   handshakeRequestMessage,
   tracing_default
-} from "./chunk-M75K5TJS.js";
+} from "./chunk-65EMTMCV.js";
 
 // transport/client.ts
 import { SpanStatusCode } from "@opentelemetry/api";
@@ -244,14 +244,14 @@ var ClientTransport = class extends Transport {
         msg.payload.status.code
       ) : false;
       const reason = `handshake failed: ${msg.payload.status.reason}`;
+      const to = session.to;
       this.rejectHandshakeResponse(session, reason, {
         ...session.loggingMetadata,
         transportMessage: msg
       });
       if (retriable) {
-        this.tryReconnecting(session.to);
+        this.tryReconnecting(to);
       } else {
-        this.deleteSession(session);
         this.protocolError({
           type: ProtocolError.HandshakeFailed,
           code: msg.payload.status.code,
@@ -304,6 +304,12 @@ var ClientTransport = class extends Transport {
    * @param to The client ID of the node to connect to.
    */
   connect(to) {
+    if (this.getStatus() !== "open") {
+      this.log?.info(
+        `transport state is no longer open, cancelling attempt to connect to ${to}`
+      );
+      return;
+    }
     let session = this.sessions.get(to);
     session ??= this.createUnconnectedSession(to);
     if (session.state !== "NoConnection" /* NoConnection */) {
@@ -313,13 +319,6 @@ var ClientTransport = class extends Transport {
       );
       return;
     }
-    if (this.getStatus() !== "open") {
-      this.log?.info(
-        `transport state is no longer open, cancelling attempt to connect to ${to}`,
-        session.loggingMetadata
-      );
-      return;
-    }
     if (!this.retryBudget.hasBudget()) {
       const budgetConsumed = this.retryBudget.getBudgetConsumed();
       const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;
@@ -341,22 +340,7 @@ var ClientTransport = class extends Transport {
       backoffMs,
       {
         onBackoffFinished: () => {
-          const reconnectPromise = tracing_default.startActiveSpan(
-            "connect",
-            async (span) => {
-              try {
-                return await this.createNewOutgoingConnection(to);
-              } catch (err) {
-                const errStr = coerceErrorString(err);
-                span.recordException(errStr);
-                span.setStatus({ code: SpanStatusCode.ERROR });
-                throw err;
-              } finally {
-                span.end();
-              }
-            }
-          );
-          this.onBackoffFinished(backingOffSession, reconnectPromise);
+          this.onBackoffFinished(backingOffSession);
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(backingOffSession);
@@ -365,7 +349,19 @@ var ClientTransport = class extends Transport {
     );
     this.updateSession(backingOffSession);
   }
-  onBackoffFinished(session, connPromise) {
+  onBackoffFinished(session) {
+    const connPromise = tracing_default.startActiveSpan("connect", async (span) => {
+      try {
+        return await this.createNewOutgoingConnection(session.to);
+      } catch (err) {
+        const errStr = coerceErrorString(err);
+        span.recordException(errStr);
+        span.setStatus({ code: SpanStatusCode.ERROR });
+        throw err;
+      } finally {
+        span.end();
+      }
+    });
     const connectingSession = ClientSessionStateGraph.transition.BackingOffToConnecting(
       session,
       connPromise,
@@ -373,7 +369,10 @@ var ClientTransport = class extends Transport {
         onConnectionEstablished: (conn) => {
           this.log?.debug(
             `connection to ${connectingSession.to} established`,
-            connectingSession.loggingMetadata
+            {
+              ...conn.loggingMetadata,
+              ...connectingSession.loggingMetadata
+            }
           );
           this.onConnectionEstablished(connectingSession, conn);
         },
@@ -430,4 +429,4 @@ var ClientTransport = class extends Transport {
 export {
   ClientTransport
 };
-//# sourceMappingURL=chunk-IV27BICV.js.map
+//# sourceMappingURL=chunk-HPOR6FST.js.map

package/dist/chunk-HPOR6FST.js.map

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

package/dist/{chunk-MREEJE3X.js → chunk-PXRKBNWN.js}

@@ -8,7 +8,7 @@ import {
   getPropagationContext,
   isStreamClose,
   isStreamOpen
-} from "./chunk-M75K5TJS.js";
+} from "./chunk-65EMTMCV.js";
 
 // router/services.ts
 import { Type } from "@sinclair/typebox";
@@ -1451,4 +1451,4 @@ export {
   createClientHandshakeOptions,
   createServerHandshakeOptions
 };
-//# sourceMappingURL=chunk-MREEJE3X.js.map
+//# sourceMappingURL=chunk-PXRKBNWN.js.map

package/dist/{chunk-M5X4JTU3.js → chunk-RSCUADHY.js}

@@ -5,10 +5,10 @@ import {
 import {
   SessionStateGraph,
   defaultTransportOptions
-} from "./chunk-NC54BC47.js";
+} from "./chunk-42SXIIHP.js";
 import {
   generateId
-} from "./chunk-M75K5TJS.js";
+} from "./chunk-65EMTMCV.js";
 
 // transport/events.ts
 var ProtocolError = {
@@ -160,8 +160,9 @@ var Transport = class {
       status: "disconnect",
       session
     });
+    const to = session.to;
     session.close();
-    this.sessions.delete(session.to);
+    this.sessions.delete(to);
   }
   // common listeners
   onSessionGracePeriodElapsed(session) {
@@ -274,4 +275,4 @@ export {
   Transport,
   Connection
 };
-//# sourceMappingURL=chunk-M5X4JTU3.js.map
+//# sourceMappingURL=chunk-RSCUADHY.js.map

package/dist/{chunk-M5X4JTU3.js.map → chunk-RSCUADHY.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../transport/events.ts","../transport/transport.ts","../transport/connection.ts"],"sourcesContent":["import { type Static } from '@sinclair/typebox';\nimport { Connection } from './connection';\nimport { OpaqueTransportMessage, HandshakeErrorResponseCodes } from './message';\nimport { Session, SessionState } from './sessionStateMachine';\nimport { TransportStatus } from './transport';\n\nexport const ProtocolError = {\n  RetriesExceeded: 'conn_retry_exceeded',\n  HandshakeFailed: 'handshake_failed',\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  sessionStatus: {\n    status: 'connect' | 'disconnect';\n    session: Session<Connection>;\n  };\n  sessionTransition:\n    | { state: SessionState.Connected }\n    | { state: SessionState.Handshaking }\n    | { state: SessionState.Connecting }\n    | { state: SessionState.BackingOff }\n    | { state: SessionState.NoConnection };\n  protocolError:\n    | {\n        type: (typeof ProtocolError)['HandshakeFailed'];\n        code: Static<typeof HandshakeErrorResponseCodes>;\n        message: string;\n      }\n    | {\n        type: Omit<\n          ProtocolErrorType,\n          (typeof ProtocolError)['HandshakeFailed']\n        >;\n        message: string;\n      };\n  transportStatus: {\n    status: TransportStatus;\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  removeAllListeners() {\n    this.eventListeners = {};\n  }\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      // copying ensures that adding more listeners in a handler doesn't\n      // affect the current dispatch.\n      const copy = [...handlers];\n      for (const handler of copy) {\n        handler(event);\n      }\n    }\n  }\n}\n","import {\n  OpaqueTransportMessage,\n  TransportClientId,\n  PartialTransportMessage,\n} from './message';\nimport {\n  BaseLogger,\n  LogFn,\n  Logger,\n  LoggingLevel,\n  createLogProxy,\n} from '../logging/log';\nimport { EventDispatcher, EventHandler, EventMap, EventTypes } from './events';\nimport {\n  ProvidedTransportOptions,\n  TransportOptions,\n  defaultTransportOptions,\n} from './options';\nimport {\n  SessionConnected,\n  SessionConnecting,\n  SessionHandshaking,\n  SessionNoConnection,\n  SessionState,\n} from './sessionStateMachine';\nimport { Connection } from './connection';\nimport { Session, SessionStateGraph } from './sessionStateMachine/transitions';\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 permanently closed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed';\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions\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   * The status of the transport.\n   */\n  private status: TransportStatus;\n\n  /**\n   * The client ID of this transport.\n   */\n  clientId: TransportClientId;\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  log?: Logger;\n\n  sessions: Map<TransportClientId, Session<ConnType>>;\n\n  /**\n   * Creates a new Transport instance.\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.clientId = clientId;\n    this.status = 'open';\n    this.sessions = new Map();\n  }\n\n  bindLogger(fn: LogFn | Logger, level?: LoggingLevel) {\n    // construct logger from fn\n    if (typeof fn === 'function') {\n      this.log = createLogProxy(new BaseLogger(fn, level));\n      return;\n    }\n\n    // object case, just assign\n    this.log = createLogProxy(fn);\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.getStatus() !== 'open') return;\n    this.eventDispatcher.dispatchEvent('message', msg);\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  abstract send(to: TransportClientId, msg: PartialTransportMessage): string;\n\n  protected protocolError(message: EventMap['protocolError']) {\n    this.eventDispatcher.dispatchEvent('protocolError', 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.status = 'closed';\n\n    for (const session of this.sessions.values()) {\n      this.deleteSession(session);\n    }\n\n    this.eventDispatcher.dispatchEvent('transportStatus', {\n      status: this.status,\n    });\n\n    this.eventDispatcher.removeAllListeners();\n\n    this.log?.info(`manually closed transport`, { clientId: this.clientId });\n  }\n\n  getStatus(): TransportStatus {\n    return this.status;\n  }\n\n  protected updateSession<S extends Session<ConnType>>(session: S): S {\n    const activeSession = this.sessions.get(session.to);\n    if (activeSession && activeSession.id !== session.id) {\n      const msg = `attempt to transition active session for ${session.to} but active session (${activeSession.id}) is different from handle (${session.id})`;\n      throw new Error(msg);\n    }\n\n    this.sessions.set(session.to, session);\n\n    if (!activeSession) {\n      this.eventDispatcher.dispatchEvent('sessionStatus', {\n        status: 'connect',\n        session: session,\n      });\n    }\n\n    this.eventDispatcher.dispatchEvent('sessionTransition', {\n      state: session.state,\n      session: session,\n    } as EventMap['sessionTransition']);\n\n    return session;\n  }\n\n  // state transitions\n  protected deleteSession(session: Session<ConnType>) {\n    session.log?.info(`closing session ${session.id}`, session.loggingMetadata);\n\n    this.eventDispatcher.dispatchEvent('sessionStatus', {\n      status: 'disconnect',\n      session: session,\n    });\n\n    session.close();\n    this.sessions.delete(session.to);\n  }\n\n  // common listeners\n  protected onSessionGracePeriodElapsed(session: Session<ConnType>) {\n    this.log?.warn(\n      `session to ${session.to} grace period elapsed, closing`,\n      session.loggingMetadata,\n    );\n\n    this.deleteSession(session);\n  }\n\n  protected onConnectingFailed(\n    session: SessionConnecting<ConnType>,\n  ): SessionNoConnection {\n    // transition to no connection\n    const noConnectionSession =\n      SessionStateGraph.transition.ConnectingToNoConnection(session, {\n        onSessionGracePeriodElapsed: () => {\n          this.onSessionGracePeriodElapsed(noConnectionSession);\n        },\n      });\n\n    return this.updateSession(noConnectionSession);\n  }\n\n  protected onConnClosed(\n    session: SessionHandshaking<ConnType> | SessionConnected<ConnType>,\n  ): SessionNoConnection {\n    // transition to no connection\n    let noConnectionSession: SessionNoConnection;\n    if (session.state === SessionState.Handshaking) {\n      noConnectionSession =\n        SessionStateGraph.transition.HandshakingToNoConnection(session, {\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(noConnectionSession);\n          },\n        });\n    } else {\n      noConnectionSession =\n        SessionStateGraph.transition.ConnectedToNoConnection(session, {\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(noConnectionSession);\n          },\n        });\n    }\n\n    return this.updateSession(noConnectionSession);\n  }\n}\n","import { TelemetryInfo } from '../tracing';\nimport { MessageMetadata } from '../logging';\nimport { generateId } from './id';\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-${generateId()}`; // for debugging, no collision safety needed\n  }\n\n  get loggingMetadata(): MessageMetadata {\n    const metadata: MessageMetadata = { connId: this.id };\n    const spanContext = this.telemetry?.span.spanContext();\n\n    if (this.telemetry?.span.isRecording() && spanContext) {\n      metadata.telemetry = {\n        traceId: spanContext.traceId,\n        spanId: spanContext.spanId,\n      };\n    }\n\n    return metadata;\n  }\n\n  // can't use event emitter because we need this to work in both node + browser\n  private _dataListeners = new Set<(msg: Uint8Array) => void>();\n  private _closeListeners = new Set<() => void>();\n  private _errorListeners = new Set<(err: Error) => void>();\n\n  get dataListeners() {\n    return [...this._dataListeners];\n  }\n\n  get closeListeners() {\n    return [...this._closeListeners];\n  }\n\n  get errorListeners() {\n    return [...this._errorListeners];\n  }\n\n  /**\n   * Handle adding a callback for when a message is received.\n   * @param msg The message that was received.\n   */\n  addDataListener(cb: (msg: Uint8Array) => void) {\n    this._dataListeners.add(cb);\n  }\n\n  removeDataListener(cb: (msg: Uint8Array) => void): void {\n    this._dataListeners.delete(cb);\n  }\n\n  /**\n   * Handle adding a callback for when the connection is closed.\n   * This should also be called if an error happens and after notifying all the error listeners.\n   * @param cb The callback to call when the connection is closed.\n   */\n  addCloseListener(cb: () => void): void {\n    this._closeListeners.add(cb);\n  }\n\n  removeCloseListener(cb: () => void): void {\n    this._closeListeners.delete(cb);\n  }\n\n  /**\n   * Handle adding a callback for when an error is received.\n   * This should only be used for this.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  addErrorListener(cb: (err: Error) => void): void {\n    this._errorListeners.add(cb);\n  }\n\n  removeErrorListener(cb: (err: Error) => void): void {\n    this._errorListeners.delete(cb);\n  }\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"],"mappings":";;;;;;;;;;;;;AAMO,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AAwCO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,qBAAqB;AACnB,SAAK,iBAAiB,CAAC;AAAA,EACzB;AAAA,EAEA,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;AAGZ,YAAM,OAAO,CAAC,GAAG,QAAQ;AACzB,iBAAW,WAAW,MAAM;AAC1B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;AC/BO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA,EAInD;AAAA;AAAA;AAAA;AAAA,EAKR;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA,EACV;AAAA,EAEA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW;AAChB,SAAK,SAAS;AACd,SAAK,WAAW,oBAAI,IAAI;AAAA,EAC1B;AAAA,EAEA,WAAW,IAAoB,OAAsB;AAEnD,QAAI,OAAO,OAAO,YAAY;AAC5B,WAAK,MAAM,eAAe,IAAI,WAAW,IAAI,KAAK,CAAC;AACnD;AAAA,IACF;AAGA,SAAK,MAAM,eAAe,EAAE;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAU,KAA6B;AAC/C,QAAI,KAAK,UAAU,MAAM;AAAQ;AACjC,SAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,EACnD;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,EAUU,cAAc,SAAoC;AAC1D,SAAK,gBAAgB,cAAc,iBAAiB,OAAO;AAAA,EAC7D;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,SAAS;AAEd,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,gBAAgB,cAAc,mBAAmB;AAAA,MACpD,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,SAAK,gBAAgB,mBAAmB;AAExC,SAAK,KAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACzE;AAAA,EAEA,YAA6B;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA,EAEU,cAA2C,SAAe;AAClE,UAAM,gBAAgB,KAAK,SAAS,IAAI,QAAQ,EAAE;AAClD,QAAI,iBAAiB,cAAc,OAAO,QAAQ,IAAI;AACpD,YAAM,MAAM,4CAA4C,QAAQ,EAAE,wBAAwB,cAAc,EAAE,+BAA+B,QAAQ,EAAE;AACnJ,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AAErC,QAAI,CAAC,eAAe;AAClB,WAAK,gBAAgB,cAAc,iBAAiB;AAAA,QAClD,QAAQ;AAAA,QACR;AAAA,MACF,CAAC;AAAA,IACH;AAEA,SAAK,gBAAgB,cAAc,qBAAqB;AAAA,MACtD,OAAO,QAAQ;AAAA,MACf;AAAA,IACF,CAAkC;AAElC,WAAO;AAAA,EACT;AAAA;AAAA,EAGU,cAAc,SAA4B;AAClD,YAAQ,KAAK,KAAK,mBAAmB,QAAQ,EAAE,IAAI,QAAQ,eAAe;AAE1E,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,YAAQ,MAAM;AACd,SAAK,SAAS,OAAO,QAAQ,EAAE;AAAA,EACjC;AAAA;AAAA,EAGU,4BAA4B,SAA4B;AAChE,SAAK,KAAK;AAAA,MACR,cAAc,QAAQ,EAAE;AAAA,MACxB,QAAQ;AAAA,IACV;AAEA,SAAK,cAAc,OAAO;AAAA,EAC5B;AAAA,EAEU,mBACR,SACqB;AAErB,UAAM,sBACJ,kBAAkB,WAAW,yBAAyB,SAAS;AAAA,MAC7D,6BAA6B,MAAM;AACjC,aAAK,4BAA4B,mBAAmB;AAAA,MACtD;AAAA,IACF,CAAC;AAEH,WAAO,KAAK,cAAc,mBAAmB;AAAA,EAC/C;AAAA,EAEU,aACR,SACqB;AAErB,QAAI;AACJ,QAAI,QAAQ,2CAAoC;AAC9C,4BACE,kBAAkB,WAAW,0BAA0B,SAAS;AAAA,QAC9D,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,mBAAmB;AAAA,QACtD;AAAA,MACF,CAAC;AAAA,IACL,OAAO;AACL,4BACE,kBAAkB,WAAW,wBAAwB,SAAS;AAAA,QAC5D,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,mBAAmB;AAAA,QACtD;AAAA,MACF,CAAC;AAAA,IACL;AAEA,WAAO,KAAK,cAAc,mBAAmB;AAAA,EAC/C;AACF;;;AC5PO,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQ,WAAW,CAAC;AAAA,EAChC;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,WAA4B,EAAE,QAAQ,KAAK,GAAG;AACpD,UAAM,cAAc,KAAK,WAAW,KAAK,YAAY;AAErD,QAAI,KAAK,WAAW,KAAK,YAAY,KAAK,aAAa;AACrD,eAAS,YAAY;AAAA,QACnB,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGQ,iBAAiB,oBAAI,IAA+B;AAAA,EACpD,kBAAkB,oBAAI,IAAgB;AAAA,EACtC,kBAAkB,oBAAI,IAA0B;AAAA,EAExD,IAAI,gBAAgB;AAClB,WAAO,CAAC,GAAG,KAAK,cAAc;AAAA,EAChC;AAAA,EAEA,IAAI,iBAAiB;AACnB,WAAO,CAAC,GAAG,KAAK,eAAe;AAAA,EACjC;AAAA,EAEA,IAAI,iBAAiB;AACnB,WAAO,CAAC,GAAG,KAAK,eAAe;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAgB,IAA+B;AAC7C,SAAK,eAAe,IAAI,EAAE;AAAA,EAC5B;AAAA,EAEA,mBAAmB,IAAqC;AACtD,SAAK,eAAe,OAAO,EAAE;AAAA,EAC/B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBAAiB,IAAsB;AACrC,SAAK,gBAAgB,IAAI,EAAE;AAAA,EAC7B;AAAA,EAEA,oBAAoB,IAAsB;AACxC,SAAK,gBAAgB,OAAO,EAAE;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,iBAAiB,IAAgC;AAC/C,SAAK,gBAAgB,IAAI,EAAE;AAAA,EAC7B;AAAA,EAEA,oBAAoB,IAAgC;AAClD,SAAK,gBAAgB,OAAO,EAAE;AAAA,EAChC;AAaF;","names":[]}
+{"version":3,"sources":["../transport/events.ts","../transport/transport.ts","../transport/connection.ts"],"sourcesContent":["import { type Static } from '@sinclair/typebox';\nimport { Connection } from './connection';\nimport { OpaqueTransportMessage, HandshakeErrorResponseCodes } from './message';\nimport { Session, SessionState } from './sessionStateMachine';\nimport { TransportStatus } from './transport';\n\nexport const ProtocolError = {\n  RetriesExceeded: 'conn_retry_exceeded',\n  HandshakeFailed: 'handshake_failed',\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  sessionStatus: {\n    status: 'connect' | 'disconnect';\n    session: Session<Connection>;\n  };\n  sessionTransition:\n    | { state: SessionState.Connected }\n    | { state: SessionState.Handshaking }\n    | { state: SessionState.Connecting }\n    | { state: SessionState.BackingOff }\n    | { state: SessionState.NoConnection };\n  protocolError:\n    | {\n        type: (typeof ProtocolError)['HandshakeFailed'];\n        code: Static<typeof HandshakeErrorResponseCodes>;\n        message: string;\n      }\n    | {\n        type: Omit<\n          ProtocolErrorType,\n          (typeof ProtocolError)['HandshakeFailed']\n        >;\n        message: string;\n      };\n  transportStatus: {\n    status: TransportStatus;\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  removeAllListeners() {\n    this.eventListeners = {};\n  }\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      // copying ensures that adding more listeners in a handler doesn't\n      // affect the current dispatch.\n      const copy = [...handlers];\n      for (const handler of copy) {\n        handler(event);\n      }\n    }\n  }\n}\n","import {\n  OpaqueTransportMessage,\n  TransportClientId,\n  PartialTransportMessage,\n} from './message';\nimport {\n  BaseLogger,\n  LogFn,\n  Logger,\n  LoggingLevel,\n  createLogProxy,\n} from '../logging/log';\nimport { EventDispatcher, EventHandler, EventMap, EventTypes } from './events';\nimport {\n  ProvidedTransportOptions,\n  TransportOptions,\n  defaultTransportOptions,\n} from './options';\nimport {\n  SessionConnected,\n  SessionConnecting,\n  SessionHandshaking,\n  SessionNoConnection,\n  SessionState,\n} from './sessionStateMachine';\nimport { Connection } from './connection';\nimport { Session, SessionStateGraph } from './sessionStateMachine/transitions';\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 permanently closed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed';\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions\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   * The status of the transport.\n   */\n  private status: TransportStatus;\n\n  /**\n   * The client ID of this transport.\n   */\n  clientId: TransportClientId;\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  log?: Logger;\n\n  sessions: Map<TransportClientId, Session<ConnType>>;\n\n  /**\n   * Creates a new Transport instance.\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.clientId = clientId;\n    this.status = 'open';\n    this.sessions = new Map();\n  }\n\n  bindLogger(fn: LogFn | Logger, level?: LoggingLevel) {\n    // construct logger from fn\n    if (typeof fn === 'function') {\n      this.log = createLogProxy(new BaseLogger(fn, level));\n      return;\n    }\n\n    // object case, just assign\n    this.log = createLogProxy(fn);\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.getStatus() !== 'open') return;\n    this.eventDispatcher.dispatchEvent('message', msg);\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  abstract send(to: TransportClientId, msg: PartialTransportMessage): string;\n\n  protected protocolError(message: EventMap['protocolError']) {\n    this.eventDispatcher.dispatchEvent('protocolError', 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.status = 'closed';\n\n    for (const session of this.sessions.values()) {\n      this.deleteSession(session);\n    }\n\n    this.eventDispatcher.dispatchEvent('transportStatus', {\n      status: this.status,\n    });\n\n    this.eventDispatcher.removeAllListeners();\n\n    this.log?.info(`manually closed transport`, { clientId: this.clientId });\n  }\n\n  getStatus(): TransportStatus {\n    return this.status;\n  }\n\n  protected updateSession<S extends Session<ConnType>>(session: S): S {\n    const activeSession = this.sessions.get(session.to);\n    if (activeSession && activeSession.id !== session.id) {\n      const msg = `attempt to transition active session for ${session.to} but active session (${activeSession.id}) is different from handle (${session.id})`;\n      throw new Error(msg);\n    }\n\n    this.sessions.set(session.to, session);\n\n    if (!activeSession) {\n      this.eventDispatcher.dispatchEvent('sessionStatus', {\n        status: 'connect',\n        session: session,\n      });\n    }\n\n    this.eventDispatcher.dispatchEvent('sessionTransition', {\n      state: session.state,\n      session: session,\n    } as EventMap['sessionTransition']);\n\n    return session;\n  }\n\n  // state transitions\n  protected deleteSession(session: Session<ConnType>) {\n    session.log?.info(`closing session ${session.id}`, session.loggingMetadata);\n\n    this.eventDispatcher.dispatchEvent('sessionStatus', {\n      status: 'disconnect',\n      session: session,\n    });\n\n    const to = session.to;\n    session.close();\n    this.sessions.delete(to);\n  }\n\n  // common listeners\n  protected onSessionGracePeriodElapsed(session: Session<ConnType>) {\n    this.log?.warn(\n      `session to ${session.to} grace period elapsed, closing`,\n      session.loggingMetadata,\n    );\n\n    this.deleteSession(session);\n  }\n\n  protected onConnectingFailed(\n    session: SessionConnecting<ConnType>,\n  ): SessionNoConnection {\n    // transition to no connection\n    const noConnectionSession =\n      SessionStateGraph.transition.ConnectingToNoConnection(session, {\n        onSessionGracePeriodElapsed: () => {\n          this.onSessionGracePeriodElapsed(noConnectionSession);\n        },\n      });\n\n    return this.updateSession(noConnectionSession);\n  }\n\n  protected onConnClosed(\n    session: SessionHandshaking<ConnType> | SessionConnected<ConnType>,\n  ): SessionNoConnection {\n    // transition to no connection\n    let noConnectionSession: SessionNoConnection;\n    if (session.state === SessionState.Handshaking) {\n      noConnectionSession =\n        SessionStateGraph.transition.HandshakingToNoConnection(session, {\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(noConnectionSession);\n          },\n        });\n    } else {\n      noConnectionSession =\n        SessionStateGraph.transition.ConnectedToNoConnection(session, {\n          onSessionGracePeriodElapsed: () => {\n            this.onSessionGracePeriodElapsed(noConnectionSession);\n          },\n        });\n    }\n\n    return this.updateSession(noConnectionSession);\n  }\n}\n","import { TelemetryInfo } from '../tracing';\nimport { MessageMetadata } from '../logging';\nimport { generateId } from './id';\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\n  constructor() {\n    this.id = `conn-${generateId()}`; // for debugging, no collision safety needed\n  }\n\n  get loggingMetadata(): MessageMetadata {\n    const metadata: MessageMetadata = { connId: this.id };\n    const spanContext = this.telemetry?.span.spanContext();\n\n    if (this.telemetry?.span.isRecording() && spanContext) {\n      metadata.telemetry = {\n        traceId: spanContext.traceId,\n        spanId: spanContext.spanId,\n      };\n    }\n\n    return metadata;\n  }\n\n  // can't use event emitter because we need this to work in both node + browser\n  private _dataListeners = new Set<(msg: Uint8Array) => void>();\n  private _closeListeners = new Set<() => void>();\n  private _errorListeners = new Set<(err: Error) => void>();\n\n  get dataListeners() {\n    return [...this._dataListeners];\n  }\n\n  get closeListeners() {\n    return [...this._closeListeners];\n  }\n\n  get errorListeners() {\n    return [...this._errorListeners];\n  }\n\n  /**\n   * Handle adding a callback for when a message is received.\n   * @param msg The message that was received.\n   */\n  addDataListener(cb: (msg: Uint8Array) => void) {\n    this._dataListeners.add(cb);\n  }\n\n  removeDataListener(cb: (msg: Uint8Array) => void): void {\n    this._dataListeners.delete(cb);\n  }\n\n  /**\n   * Handle adding a callback for when the connection is closed.\n   * This should also be called if an error happens and after notifying all the error listeners.\n   * @param cb The callback to call when the connection is closed.\n   */\n  addCloseListener(cb: () => void): void {\n    this._closeListeners.add(cb);\n  }\n\n  removeCloseListener(cb: () => void): void {\n    this._closeListeners.delete(cb);\n  }\n\n  /**\n   * Handle adding a callback for when an error is received.\n   * This should only be used for this.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  addErrorListener(cb: (err: Error) => void): void {\n    this._errorListeners.add(cb);\n  }\n\n  removeErrorListener(cb: (err: Error) => void): void {\n    this._errorListeners.delete(cb);\n  }\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"],"mappings":";;;;;;;;;;;;;AAMO,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AAwCO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,qBAAqB;AACnB,SAAK,iBAAiB,CAAC;AAAA,EACzB;AAAA,EAEA,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;AAGZ,YAAM,OAAO,CAAC,GAAG,QAAQ;AACzB,iBAAW,WAAW,MAAM;AAC1B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;AC/BO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA,EAInD;AAAA;AAAA;AAAA;AAAA,EAKR;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA,EACV;AAAA,EAEA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW;AAChB,SAAK,SAAS;AACd,SAAK,WAAW,oBAAI,IAAI;AAAA,EAC1B;AAAA,EAEA,WAAW,IAAoB,OAAsB;AAEnD,QAAI,OAAO,OAAO,YAAY;AAC5B,WAAK,MAAM,eAAe,IAAI,WAAW,IAAI,KAAK,CAAC;AACnD;AAAA,IACF;AAGA,SAAK,MAAM,eAAe,EAAE;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAU,KAA6B;AAC/C,QAAI,KAAK,UAAU,MAAM;AAAQ;AACjC,SAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,EACnD;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,EAUU,cAAc,SAAoC;AAC1D,SAAK,gBAAgB,cAAc,iBAAiB,OAAO;AAAA,EAC7D;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,SAAS;AAEd,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,gBAAgB,cAAc,mBAAmB;AAAA,MACpD,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,SAAK,gBAAgB,mBAAmB;AAExC,SAAK,KAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACzE;AAAA,EAEA,YAA6B;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA,EAEU,cAA2C,SAAe;AAClE,UAAM,gBAAgB,KAAK,SAAS,IAAI,QAAQ,EAAE;AAClD,QAAI,iBAAiB,cAAc,OAAO,QAAQ,IAAI;AACpD,YAAM,MAAM,4CAA4C,QAAQ,EAAE,wBAAwB,cAAc,EAAE,+BAA+B,QAAQ,EAAE;AACnJ,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AAErC,QAAI,CAAC,eAAe;AAClB,WAAK,gBAAgB,cAAc,iBAAiB;AAAA,QAClD,QAAQ;AAAA,QACR;AAAA,MACF,CAAC;AAAA,IACH;AAEA,SAAK,gBAAgB,cAAc,qBAAqB;AAAA,MACtD,OAAO,QAAQ;AAAA,MACf;AAAA,IACF,CAAkC;AAElC,WAAO;AAAA,EACT;AAAA;AAAA,EAGU,cAAc,SAA4B;AAClD,YAAQ,KAAK,KAAK,mBAAmB,QAAQ,EAAE,IAAI,QAAQ,eAAe;AAE1E,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,UAAM,KAAK,QAAQ;AACnB,YAAQ,MAAM;AACd,SAAK,SAAS,OAAO,EAAE;AAAA,EACzB;AAAA;AAAA,EAGU,4BAA4B,SAA4B;AAChE,SAAK,KAAK;AAAA,MACR,cAAc,QAAQ,EAAE;AAAA,MACxB,QAAQ;AAAA,IACV;AAEA,SAAK,cAAc,OAAO;AAAA,EAC5B;AAAA,EAEU,mBACR,SACqB;AAErB,UAAM,sBACJ,kBAAkB,WAAW,yBAAyB,SAAS;AAAA,MAC7D,6BAA6B,MAAM;AACjC,aAAK,4BAA4B,mBAAmB;AAAA,MACtD;AAAA,IACF,CAAC;AAEH,WAAO,KAAK,cAAc,mBAAmB;AAAA,EAC/C;AAAA,EAEU,aACR,SACqB;AAErB,QAAI;AACJ,QAAI,QAAQ,2CAAoC;AAC9C,4BACE,kBAAkB,WAAW,0BAA0B,SAAS;AAAA,QAC9D,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,mBAAmB;AAAA,QACtD;AAAA,MACF,CAAC;AAAA,IACL,OAAO;AACL,4BACE,kBAAkB,WAAW,wBAAwB,SAAS;AAAA,QAC5D,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,mBAAmB;AAAA,QACtD;AAAA,MACF,CAAC;AAAA,IACL;AAEA,WAAO,KAAK,cAAc,mBAAmB;AAAA,EAC/C;AACF;;;AC7PO,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EAEA,cAAc;AACZ,SAAK,KAAK,QAAQ,WAAW,CAAC;AAAA,EAChC;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,WAA4B,EAAE,QAAQ,KAAK,GAAG;AACpD,UAAM,cAAc,KAAK,WAAW,KAAK,YAAY;AAErD,QAAI,KAAK,WAAW,KAAK,YAAY,KAAK,aAAa;AACrD,eAAS,YAAY;AAAA,QACnB,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGQ,iBAAiB,oBAAI,IAA+B;AAAA,EACpD,kBAAkB,oBAAI,IAAgB;AAAA,EACtC,kBAAkB,oBAAI,IAA0B;AAAA,EAExD,IAAI,gBAAgB;AAClB,WAAO,CAAC,GAAG,KAAK,cAAc;AAAA,EAChC;AAAA,EAEA,IAAI,iBAAiB;AACnB,WAAO,CAAC,GAAG,KAAK,eAAe;AAAA,EACjC;AAAA,EAEA,IAAI,iBAAiB;AACnB,WAAO,CAAC,GAAG,KAAK,eAAe;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAgB,IAA+B;AAC7C,SAAK,eAAe,IAAI,EAAE;AAAA,EAC5B;AAAA,EAEA,mBAAmB,IAAqC;AACtD,SAAK,eAAe,OAAO,EAAE;AAAA,EAC/B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBAAiB,IAAsB;AACrC,SAAK,gBAAgB,IAAI,EAAE;AAAA,EAC7B;AAAA,EAEA,oBAAoB,IAAsB;AACxC,SAAK,gBAAgB,OAAO,EAAE;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,iBAAiB,IAAgC;AAC/C,SAAK,gBAAgB,IAAI,EAAE;AAAA,EAC7B;AAAA,EAEA,oBAAoB,IAAgC;AAClD,SAAK,gBAAgB,OAAO,EAAE;AAAA,EAChC;AAaF;","names":[]}

package/dist/{client-654098be.d.ts → client-6c67339a.d.ts}

@@ -1,5 +1,5 @@
-import { C as Connection, T as Transport, n as ClientTransportOptions, L as LeakyBucketRateLimit, p as ClientHandshakeOptions, q as ClientSession, b as ProvidedClientTransportOptions, f as SessionConnecting, e as SessionNoConnection, g as SessionHandshaking, h as SessionConnected, r as SessionBackingOff } from './handshake-1a86f06d.js';
-import { c as TransportClientId, P as PartialTransportMessage, b as OpaqueTransportMessage } from './message-57296605.js';
+import { C as Connection, T as Transport, n as ClientTransportOptions, L as LeakyBucketRateLimit, p as ClientHandshakeOptions, q as ClientSession, b as ProvidedClientTransportOptions, f as SessionConnecting, e as SessionNoConnection, g as SessionHandshaking, h as SessionConnected, r as SessionBackingOff } from './handshake-154a0bb2.js';
+import { c as TransportClientId, P as PartialTransportMessage, b as OpaqueTransportMessage } from './message-ff78a233.js';
 
 declare abstract class ClientTransport<ConnType extends Connection> extends Transport<ConnType> {
     /**
@@ -41,7 +41,7 @@ declare abstract class ClientTransport<ConnType extends Connection> extends Tran
      * @param to The client ID of the node to connect to.
      */
     connect(to: TransportClientId): void;
-    protected onBackoffFinished(session: SessionBackingOff, connPromise: Promise<ConnType>): void;
+    protected onBackoffFinished(session: SessionBackingOff): void;
     private sendHandshake;
     close(): void;
 }

package/dist/{connection-bc2454dc.d.ts → connection-03ffb583.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection } from './handshake-1a86f06d.js';
+import { C as Connection } from './handshake-154a0bb2.js';
 import { W as WsLike } from './wslike-e0b32dd5.js';
 
 declare class WebSocketConnection extends Connection {

package/dist/{handshake-1a86f06d.d.ts → handshake-154a0bb2.d.ts}

@@ -1,5 +1,6 @@
-import { e as TelemetryInfo, M as MessageMetadata, c as TransportClientId, L as Logger, b as OpaqueTransportMessage, P as PartialTransportMessage, a as TransportMessage, H as HandshakeErrorResponseCodes, f as LogFn, g as LoggingLevel, h as HandshakeErrorCustomHandlerFatalResponseCodes } from './message-57296605.js';
+import { f as TelemetryInfo, M as MessageMetadata, c as TransportClientId, L as Logger, b as OpaqueTransportMessage, P as PartialTransportMessage, a as TransportMessage, e as Tags, H as HandshakeErrorResponseCodes, g as LogFn, h as LoggingLevel, j as HandshakeErrorCustomHandlerFatalResponseCodes } from './message-ff78a233.js';
 import { Static, TSchema } from '@sinclair/typebox';
+import * as _sinclair_typebox_errors from '@sinclair/typebox/errors';
 import { C as Codec } from './types-3e5768ec.js';
 
 /**
@@ -210,6 +211,20 @@ declare class SessionHandshaking<ConnType extends Connection> extends Identified
     listeners: SessionHandshakingListeners;
     handshakeTimeout?: ReturnType<typeof setTimeout>;
     constructor(props: SessionHandshakingProps<ConnType>);
+    get loggingMetadata(): {
+        protocolVersion?: string | undefined;
+        clientId?: string | undefined;
+        connectedTo?: string | undefined;
+        sessionId?: string | undefined;
+        connId?: string | undefined;
+        transportMessage?: Partial<OpaqueTransportMessage> | undefined;
+        validationErrors?: _sinclair_typebox_errors.ValueError[] | undefined;
+        tags?: Tags[] | undefined;
+        telemetry?: {
+            traceId: string;
+            spanId: string;
+        } | undefined;
+    };
     onHandshakeData: (msg: Uint8Array) => void;
     sendHandshake(msg: TransportMessage): boolean;
     _handleStateExit(): void;
@@ -236,6 +251,20 @@ declare class SessionConnected<ConnType extends Connection> extends IdentifiedSe
     updateBookkeeping(ack: number, seq: number): void;
     send(msg: PartialTransportMessage): string;
     constructor(props: SessionConnectedProps<ConnType>);
+    get loggingMetadata(): {
+        protocolVersion?: string | undefined;
+        clientId?: string | undefined;
+        connectedTo?: string | undefined;
+        sessionId?: string | undefined;
+        connId?: string | undefined;
+        transportMessage?: Partial<OpaqueTransportMessage> | undefined;
+        validationErrors?: _sinclair_typebox_errors.ValueError[] | undefined;
+        tags?: Tags[] | undefined;
+        telemetry?: {
+            traceId: string;
+            spanId: string;
+        } | undefined;
+    };
     startActiveHeartbeat(): void;
     private sendHeartbeat;
     onMessageData: (msg: Uint8Array) => void;

package/dist/logging/index.d.cts

@@ -1,4 +1,4 @@
-export { f as LogFn, L as Logger, M as MessageMetadata, j as coloredStringLogger, k as jsonLogger, s as stringLogger } from '../message-57296605.js';
+export { g as LogFn, L as Logger, M as MessageMetadata, k as coloredStringLogger, l as jsonLogger, s as stringLogger } from '../message-ff78a233.js';
 import '@sinclair/typebox/value';
 import '@sinclair/typebox';
 import '@opentelemetry/api';

package/dist/logging/index.d.ts

@@ -1,4 +1,4 @@
-export { f as LogFn, L as Logger, M as MessageMetadata, j as coloredStringLogger, k as jsonLogger, s as stringLogger } from '../message-57296605.js';
+export { g as LogFn, L as Logger, M as MessageMetadata, k as coloredStringLogger, l as jsonLogger, s as stringLogger } from '../message-ff78a233.js';
 import '@sinclair/typebox/value';
 import '@sinclair/typebox';
 import '@opentelemetry/api';

package/dist/{message-57296605.d.ts → message-ff78a233.d.ts}

@@ -135,4 +135,4 @@ declare function isStreamOpen(controlFlag: number): boolean;
  */
 declare function isStreamClose(controlFlag: number): boolean;
 
-export { HandshakeErrorResponseCodes as H, Logger as L, MessageMetadata as M, OpaqueTransportMessageSchema as O, PartialTransportMessage as P, TransportMessageSchema as T, TransportMessage as a, OpaqueTransportMessage as b, TransportClientId as c, isStreamClose as d, TelemetryInfo as e, LogFn as f, LoggingLevel as g, HandshakeErrorCustomHandlerFatalResponseCodes as h, isStreamOpen as i, coloredStringLogger as j, jsonLogger as k, stringLogger as s };
+export { HandshakeErrorResponseCodes as H, Logger as L, MessageMetadata as M, OpaqueTransportMessageSchema as O, PartialTransportMessage as P, TransportMessageSchema as T, TransportMessage as a, OpaqueTransportMessage as b, TransportClientId as c, isStreamClose as d, Tags as e, TelemetryInfo as f, LogFn as g, LoggingLevel as h, isStreamOpen as i, HandshakeErrorCustomHandlerFatalResponseCodes as j, coloredStringLogger as k, jsonLogger as l, stringLogger as s };

package/dist/router/index.cjs

@@ -761,7 +761,7 @@ function Err(error) {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.26.2";
+var version = "0.26.4";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {