@replit/river 0.25.1 → 0.25.2

package/dist/chunk-Q3PZBJXR.js.map

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

package/dist/{chunk-PCBPPTXH.js → chunk-QXMZJ7LF.js}

@@ -1,17 +1,17 @@
 import {
   ProtocolError,
   Transport
-} from "./chunk-D5O3ERJU.js";
+} from "./chunk-PF74DCPQ.js";
 import {
-  SessionStateGraph,
+  ServerSessionStateGraph,
   defaultServerTransportOptions
-} from "./chunk-YM5Y4NAT.js";
+} from "./chunk-Q3PZBJXR.js";
 import {
   ControlMessageHandshakeRequestSchema,
   PROTOCOL_VERSION,
   coerceErrorString,
   handshakeResponseMessage
-} from "./chunk-SZ5NBBX7.js";
+} from "./chunk-EHD2JJBW.js";
 
 // transport/server.ts
 import { SpanStatusCode } from "@opentelemetry/api";
@@ -29,9 +29,11 @@ var ServerTransport = class extends Transport {
    * A map of session handshake data for each session.
    */
   sessionHandshakeMetadata = /* @__PURE__ */ new Map();
+  sessions = /* @__PURE__ */ new Map();
   pendingSessions = /* @__PURE__ */ new Set();
   constructor(clientId, providedOptions) {
     super(clientId, providedOptions);
+    this.sessions = /* @__PURE__ */ new Map();
     this.options = {
       ...defaultServerTransportOptions,
       ...providedOptions
@@ -82,7 +84,7 @@ var ServerTransport = class extends Transport {
       clientId: this.clientId
     });
     let receivedHandshake = false;
-    const pendingSession = SessionStateGraph.entrypoints.WaitingForHandshake(
+    const pendingSession = ServerSessionStateGraph.entrypoint(
       this.clientId,
       conn,
       {
@@ -236,26 +238,15 @@ var ServerTransport = class extends Transport {
         );
         return;
       }
-      if (oldSession.state === "Connected" /* Connected */) {
-        const noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(oldSession, {
-          onSessionGracePeriodElapsed: () => {
-            this.onSessionGracePeriodElapsed(noConnectionSession);
-          }
-        });
-        oldSession = noConnectionSession;
-      } else if (oldSession.state === "Handshaking" /* Handshaking */) {
-        const noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(oldSession, {
-          onSessionGracePeriodElapsed: () => {
-            this.onSessionGracePeriodElapsed(noConnectionSession);
-          }
-        });
-        oldSession = noConnectionSession;
-      } else if (oldSession.state === "Connecting" /* Connecting */) {
-        const noConnectionSession = SessionStateGraph.transition.ConnectingToNoConnection(oldSession, {
-          onSessionGracePeriodElapsed: () => {
-            this.onSessionGracePeriodElapsed(noConnectionSession);
+      if (oldSession.state !== "NoConnection" /* NoConnection */) {
+        const noConnectionSession = ServerSessionStateGraph.transition.ConnectedToNoConnection(
+          oldSession,
+          {
+            onSessionGracePeriodElapsed: () => {
+              this.onSessionGracePeriodElapsed(noConnectionSession);
+            }
           }
-        });
+        );
         oldSession = noConnectionSession;
       }
       this.updateSession(oldSession);
@@ -304,7 +295,7 @@ var ServerTransport = class extends Transport {
       }
     });
     session.sendHandshake(responseMsg);
-    const connectedSession = SessionStateGraph.transition.WaitingForHandshakeToConnected(
+    const connectedSession = ServerSessionStateGraph.transition.WaitingForHandshakeToConnected(
       session,
       // by this point oldSession is either no connection or we dont have an old session
       oldSession,
@@ -384,4 +375,4 @@ var ServerTransport = class extends Transport {
 export {
   ServerTransport
 };
-//# sourceMappingURL=chunk-PCBPPTXH.js.map
+//# sourceMappingURL=chunk-QXMZJ7LF.js.map

package/dist/chunk-QXMZJ7LF.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/server.ts"],"sourcesContent":["import { SpanStatusCode } from '@opentelemetry/api';\nimport { ParsedMetadata } from '../router/context';\nimport { ServerHandshakeOptions } from '../router/handshake';\nimport {\n  ControlMessageHandshakeRequestSchema,\n  HandshakeErrorResponseCodes,\n  OpaqueTransportMessage,\n  PROTOCOL_VERSION,\n  PartialTransportMessage,\n  TransportClientId,\n  handshakeResponseMessage,\n} from './message';\nimport {\n  ProvidedServerTransportOptions,\n  ServerTransportOptions,\n  defaultServerTransportOptions,\n} from './options';\nimport { Transport } from './transport';\nimport { coerceErrorString } from '../util/stringify';\nimport { Static } from '@sinclair/typebox';\nimport { Value } from '@sinclair/typebox/value';\nimport { ProtocolError } from './events';\nimport { Connection } from './connection';\nimport { MessageMetadata } from '../logging';\nimport { SessionWaitingForHandshake } from './sessionStateMachine/SessionWaitingForHandshake';\nimport { SessionState } from './sessionStateMachine/common';\nimport {\n  ServerSession,\n  ServerSessionStateGraph,\n} from './sessionStateMachine/transitions';\n\nexport abstract class ServerTransport<\n  ConnType extends Connection,\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;\n\n  /**\n   * A map of session handshake data for each session.\n   */\n  sessionHandshakeMetadata = new Map<TransportClientId, ParsedMetadata>();\n\n  sessions = new Map<TransportClientId, ServerSession<ConnType>>();\n  pendingSessions = new Set<SessionWaitingForHandshake<ConnType>>();\n\n  constructor(\n    clientId: TransportClientId,\n    providedOptions?: ProvidedServerTransportOptions,\n  ) {\n    super(clientId, providedOptions);\n    this.sessions = new Map();\n    this.options = {\n      ...defaultServerTransportOptions,\n      ...providedOptions,\n    };\n    this.log?.info(`initiated server transport`, {\n      clientId: this.clientId,\n      protocolVersion: PROTOCOL_VERSION,\n    });\n  }\n\n  extendHandshake(options: ServerHandshakeOptions) {\n    this.handshakeExtensions = options;\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    const session = this.sessions.get(to);\n    if (!session) {\n      const err = `session to ${to} does not exist`;\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    return session.send(msg);\n  }\n\n  protected deletePendingSession(\n    pendingSession: SessionWaitingForHandshake<ConnType>,\n  ) {\n    pendingSession.close();\n    // we don't dispatch a session disconnect event\n    // for a non-identified session, just delete directly\n\n    this.pendingSessions.delete(pendingSession);\n  }\n\n  protected deleteSession(session: ServerSession<ConnType>): void {\n    this.sessionHandshakeMetadata.delete(session.to);\n    super.deleteSession(session);\n  }\n\n  protected handleConnection(conn: ConnType) {\n    if (this.getStatus() !== 'open') return;\n\n    this.log?.info(`new incoming connection`, {\n      ...conn.loggingMetadata,\n      clientId: this.clientId,\n    });\n\n    let receivedHandshake = false;\n    const pendingSession = ServerSessionStateGraph.entrypoint(\n      this.clientId,\n      conn,\n      {\n        onConnectionClosed: () => {\n          this.log?.warn(\n            `connection from unknown closed before handshake finished`,\n            pendingSession.loggingMetadata,\n          );\n\n          this.deletePendingSession(pendingSession);\n        },\n        onConnectionErrored: (err) => {\n          const errorString = coerceErrorString(err);\n          this.log?.warn(\n            `connection from unknown errored before handshake finished: ${errorString}`,\n            pendingSession.loggingMetadata,\n          );\n\n          this.deletePendingSession(pendingSession);\n        },\n        onHandshakeTimeout: () => {\n          this.log?.warn(\n            `connection from unknown timed out before handshake finished`,\n            pendingSession.loggingMetadata,\n          );\n\n          this.deletePendingSession(pendingSession);\n        },\n        onHandshake: (msg) => {\n          if (receivedHandshake) {\n            this.log?.error(\n              `received multiple handshake messages from pending session`,\n              {\n                ...pendingSession.loggingMetadata,\n                connectedTo: msg.from,\n                transportMessage: msg,\n              },\n            );\n\n            this.deletePendingSession(pendingSession);\n            return;\n          }\n\n          // let this resolve async, we just need to make sure its only\n          // called once so we don't race while transitioning to connected\n          // onHandshakeRequest is async as custom validation may be async\n          receivedHandshake = true;\n          void this.onHandshakeRequest(pendingSession, msg);\n        },\n        onInvalidHandshake: (reason) => {\n          this.log?.error(\n            `invalid handshake: ${reason}`,\n            pendingSession.loggingMetadata,\n          );\n          this.deletePendingSession(pendingSession);\n          this.protocolError(ProtocolError.HandshakeFailed, reason);\n        },\n      },\n      this.options,\n      this.log,\n    );\n\n    this.pendingSessions.add(pendingSession);\n  }\n\n  private rejectHandshakeRequest(\n    session: SessionWaitingForHandshake<ConnType>,\n    to: TransportClientId,\n    reason: string,\n    code: Static<typeof HandshakeErrorResponseCodes>,\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\n    session.sendHandshake(\n      handshakeResponseMessage({\n        from: this.clientId,\n        to,\n        status: {\n          ok: false,\n          code,\n          reason,\n        },\n      }),\n    );\n\n    this.protocolError(ProtocolError.HandshakeFailed, reason);\n    this.deletePendingSession(session);\n  }\n\n  protected async onHandshakeRequest(\n    session: SessionWaitingForHandshake<ConnType>,\n    msg: OpaqueTransportMessage,\n  ) {\n    // invariant: msg is a handshake request\n    if (!Value.Check(ControlMessageHandshakeRequestSchema, msg.payload)) {\n      this.rejectHandshakeRequest(\n        session,\n        msg.from,\n        'received invalid handshake request',\n        'MALFORMED_HANDSHAKE',\n        {\n          ...session.loggingMetadata,\n          transportMessage: msg,\n          connectedTo: msg.from,\n          validationErrors: [\n            ...Value.Errors(ControlMessageHandshakeRequestSchema, msg.payload),\n          ],\n        },\n      );\n\n      return;\n    }\n\n    // invariant: handshake request passes all the validation\n    const gotVersion = msg.payload.protocolVersion;\n    if (gotVersion !== PROTOCOL_VERSION) {\n      this.rejectHandshakeRequest(\n        session,\n        msg.from,\n        `expected protocol version ${PROTOCOL_VERSION}, got ${gotVersion}`,\n        'PROTOCOL_VERSION_MISMATCH',\n        {\n          ...session.loggingMetadata,\n          connectedTo: msg.from,\n          transportMessage: msg,\n        },\n      );\n\n      return;\n    }\n\n    let oldSession = this.sessions.get(msg.from);\n\n    // invariant: must pass custom validation if defined\n    const parsedMetadata = await this.validateHandshakeMetadata(\n      session,\n      oldSession,\n      msg.payload.metadata,\n      msg.from,\n    );\n\n    if (parsedMetadata === false) {\n      return;\n    }\n\n    // 4 connect cases\n    // 1. new session\n    //    we dont have a session and the client is requesting a new one\n    //    we can create the session as normal\n    // 2. client is reconnecting to an existing session but we don't have it\n    //    reject this handshake, there's nothing we can do to salvage it\n    // 3. transparent reconnect (old session exists and is the same as the client wants)\n    //    assign to old session\n    // 4. hard reconnect (oldSession exists but but the client wants a new one)\n    //    we close the old session and create a new one\n    let connectCase:\n      | 'new session'\n      | 'unknown session'\n      | 'transparent reconnection'\n      | 'hard reconnection' = 'new session';\n    const clientNextExpectedSeq =\n      msg.payload.expectedSessionState.nextExpectedSeq;\n    const clientNextSentSeq = msg.payload.expectedSessionState.nextSentSeq ?? 0;\n\n    if (oldSession && oldSession.id === msg.payload.sessionId) {\n      connectCase = 'transparent reconnection';\n\n      // invariant: ordering must be correct\n      const ourNextSeq = oldSession.nextSeq();\n      const ourAck = oldSession.ack;\n\n      // two incorrect cases where we cannot permit a reconnect:\n      // - if the client is about to send a message in the future w.r.t to the server\n      //  - client.seq > server.ack => nextSentSeq > oldSession.ack\n      if (clientNextSentSeq > ourAck) {\n        this.rejectHandshakeRequest(\n          session,\n          msg.from,\n          `client is in the future: server wanted next message to be ${ourAck} but client would have sent ${clientNextSentSeq}`,\n          'SESSION_STATE_MISMATCH',\n          {\n            ...session.loggingMetadata,\n            connectedTo: msg.from,\n            transportMessage: msg,\n          },\n        );\n\n        return;\n      }\n\n      // - if the server is about to send a message in the future w.r.t to the client\n      //  - server.seq > client.ack => oldSession.nextSeq() > nextExpectedSeq\n      if (ourNextSeq > clientNextExpectedSeq) {\n        this.rejectHandshakeRequest(\n          session,\n          msg.from,\n          `server is in the future: client wanted next message to be ${clientNextExpectedSeq} but server would have sent ${ourNextSeq}`,\n          'SESSION_STATE_MISMATCH',\n          {\n            ...session.loggingMetadata,\n            connectedTo: msg.from,\n            transportMessage: msg,\n          },\n        );\n\n        return;\n      }\n\n      // transparent reconnect seems ok, proceed by transitioning old session\n      // to not connected\n      if (oldSession.state !== SessionState.NoConnection) {\n        const noConnectionSession =\n          ServerSessionStateGraph.transition.ConnectedToNoConnection(\n            oldSession,\n            {\n              onSessionGracePeriodElapsed: () => {\n                this.onSessionGracePeriodElapsed(noConnectionSession);\n              },\n            },\n          );\n\n        oldSession = noConnectionSession;\n      }\n\n      this.updateSession(oldSession);\n    } else if (oldSession) {\n      connectCase = 'hard reconnection';\n\n      // just nuke the old session entirely and proceed as if this was new\n      this.log?.info(\n        `client is reconnecting to a new session (${msg.payload.sessionId}) with an old session (${oldSession.id}) already existing, closing old session`,\n        {\n          ...session.loggingMetadata,\n          connectedTo: msg.from,\n          sessionId: msg.payload.sessionId,\n        },\n      );\n      this.deleteSession(oldSession);\n      oldSession = undefined;\n    }\n\n    if (!oldSession && (clientNextSentSeq > 0 || clientNextExpectedSeq > 0)) {\n      // we don't have a session, but the client is trying to reconnect\n      // to an old session. we can't do anything about this, so we reject\n      connectCase = 'unknown session';\n      this.rejectHandshakeRequest(\n        session,\n        msg.from,\n        `client is trying to reconnect to a session the server don't know about: ${msg.payload.sessionId}`,\n        'SESSION_STATE_MISMATCH',\n        {\n          ...session.loggingMetadata,\n          connectedTo: msg.from,\n          transportMessage: msg,\n        },\n      );\n      return;\n    }\n\n    // from this point on, we're committed to connecting\n    const sessionId = msg.payload.sessionId;\n    this.log?.info(\n      `handshake from ${msg.from} ok (${connectCase}), responding with handshake success`,\n      {\n        ...session.loggingMetadata,\n        connectedTo: msg.from,\n      },\n    );\n    const responseMsg = handshakeResponseMessage({\n      from: this.clientId,\n      to: msg.from,\n      status: {\n        ok: true,\n        sessionId,\n      },\n    });\n    session.sendHandshake(responseMsg);\n\n    // transition\n    const connectedSession =\n      ServerSessionStateGraph.transition.WaitingForHandshakeToConnected(\n        session,\n        // by this point oldSession is either no connection or we dont have an old session\n        oldSession,\n        sessionId,\n        msg.from,\n        msg.tracing,\n        {\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.protocolError(ProtocolError.MessageOrderingViolated, reason);\n            this.deleteSession(connectedSession);\n          },\n        },\n      );\n\n    this.sessionHandshakeMetadata.set(connectedSession.to, parsedMetadata);\n    this.updateSession(connectedSession);\n    this.pendingSessions.delete(session);\n    connectedSession.startActiveHeartbeat();\n  }\n\n  private async validateHandshakeMetadata(\n    handshakingSession: SessionWaitingForHandshake<ConnType>,\n    existingSession: ServerSession<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        this.rejectHandshakeRequest(\n          handshakingSession,\n          from,\n          'received malformed handshake metadata',\n          'MALFORMED_HANDSHAKE_META',\n          {\n            ...handshakingSession.loggingMetadata,\n            connectedTo: from,\n            validationErrors: [\n              ...Value.Errors(this.handshakeExtensions.schema, rawMetadata),\n            ],\n          },\n        );\n\n        return false;\n      }\n\n      const previousParsedMetadata = existingSession\n        ? this.sessionHandshakeMetadata.get(existingSession.to)\n        : undefined;\n\n      parsedMetadata = await this.handshakeExtensions.validate(\n        rawMetadata,\n        previousParsedMetadata,\n      );\n\n      // handler rejected the connection\n      if (parsedMetadata === false) {\n        this.rejectHandshakeRequest(\n          handshakingSession,\n          from,\n          'rejected by handshake handler',\n          'REJECTED_BY_CUSTOM_HANDLER',\n          {\n            ...handshakingSession.loggingMetadata,\n            connectedTo: from,\n            clientId: this.clientId,\n          },\n        );\n\n        return false;\n      }\n    }\n\n    return parsedMetadata;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAAA,SAAS,sBAAsB;AAoB/B,SAAS,aAAa;AAWf,IAAe,kBAAf,cAEG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA,EAKA,2BAA2B,oBAAI,IAAuC;AAAA,EAEtE,WAAW,oBAAI,IAAgD;AAAA,EAC/D,kBAAkB,oBAAI,IAA0C;AAAA,EAEhE,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,KAAK,KAAK,8BAA8B;AAAA,MAC3C,UAAU,KAAK;AAAA,MACf,iBAAiB;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,SAAiC;AAC/C,SAAK,sBAAsB;AAAA,EAC7B;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,UAAM,UAAU,KAAK,SAAS,IAAI,EAAE;AACpC,QAAI,CAAC,SAAS;AACZ,YAAM,MAAM,cAAc,EAAE;AAC5B,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,WAAO,QAAQ,KAAK,GAAG;AAAA,EACzB;AAAA,EAEU,qBACR,gBACA;AACA,mBAAe,MAAM;AAIrB,SAAK,gBAAgB,OAAO,cAAc;AAAA,EAC5C;AAAA,EAEU,cAAc,SAAwC;AAC9D,SAAK,yBAAyB,OAAO,QAAQ,EAAE;AAC/C,UAAM,cAAc,OAAO;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB;AACzC,QAAI,KAAK,UAAU,MAAM;AAAQ;AAEjC,SAAK,KAAK,KAAK,2BAA2B;AAAA,MACxC,GAAG,KAAK;AAAA,MACR,UAAU,KAAK;AAAA,IACjB,CAAC;AAED,QAAI,oBAAoB;AACxB,UAAM,iBAAiB,wBAAwB;AAAA,MAC7C,KAAK;AAAA,MACL;AAAA,MACA;AAAA,QACE,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR;AAAA,YACA,eAAe;AAAA,UACjB;AAEA,eAAK,qBAAqB,cAAc;AAAA,QAC1C;AAAA,QACA,qBAAqB,CAAC,QAAQ;AAC5B,gBAAM,cAAc,kBAAkB,GAAG;AACzC,eAAK,KAAK;AAAA,YACR,8DAA8D,WAAW;AAAA,YACzE,eAAe;AAAA,UACjB;AAEA,eAAK,qBAAqB,cAAc;AAAA,QAC1C;AAAA,QACA,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR;AAAA,YACA,eAAe;AAAA,UACjB;AAEA,eAAK,qBAAqB,cAAc;AAAA,QAC1C;AAAA,QACA,aAAa,CAAC,QAAQ;AACpB,cAAI,mBAAmB;AACrB,iBAAK,KAAK;AAAA,cACR;AAAA,cACA;AAAA,gBACE,GAAG,eAAe;AAAA,gBAClB,aAAa,IAAI;AAAA,gBACjB,kBAAkB;AAAA,cACpB;AAAA,YACF;AAEA,iBAAK,qBAAqB,cAAc;AACxC;AAAA,UACF;AAKA,8BAAoB;AACpB,eAAK,KAAK,mBAAmB,gBAAgB,GAAG;AAAA,QAClD;AAAA,QACA,oBAAoB,CAAC,WAAW;AAC9B,eAAK,KAAK;AAAA,YACR,sBAAsB,MAAM;AAAA,YAC5B,eAAe;AAAA,UACjB;AACA,eAAK,qBAAqB,cAAc;AACxC,eAAK,cAAc,cAAc,iBAAiB,MAAM;AAAA,QAC1D;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL,KAAK;AAAA,IACP;AAEA,SAAK,gBAAgB,IAAI,cAAc;AAAA,EACzC;AAAA,EAEQ,uBACN,SACA,IACA,QACA,MACA,UACA;AACA,YAAQ,KAAK,WAAW,KAAK,UAAU;AAAA,MACrC,MAAM,eAAe;AAAA,MACrB,SAAS;AAAA,IACX,CAAC;AAED,SAAK,KAAK,KAAK,QAAQ,QAAQ;AAE/B,YAAQ;AAAA,MACN,yBAAyB;AAAA,QACvB,MAAM,KAAK;AAAA,QACX;AAAA,QACA,QAAQ;AAAA,UACN,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAEA,SAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,SAAK,qBAAqB,OAAO;AAAA,EACnC;AAAA,EAEA,MAAgB,mBACd,SACA,KACA;AAEA,QAAI,CAAC,MAAM,MAAM,sCAAsC,IAAI,OAAO,GAAG;AACnE,WAAK;AAAA,QACH;AAAA,QACA,IAAI;AAAA,QACJ;AAAA,QACA;AAAA,QACA;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,kBAAkB;AAAA,UAClB,aAAa,IAAI;AAAA,UACjB,kBAAkB;AAAA,YAChB,GAAG,MAAM,OAAO,sCAAsC,IAAI,OAAO;AAAA,UACnE;AAAA,QACF;AAAA,MACF;AAEA;AAAA,IACF;AAGA,UAAM,aAAa,IAAI,QAAQ;AAC/B,QAAI,eAAe,kBAAkB;AACnC,WAAK;AAAA,QACH;AAAA,QACA,IAAI;AAAA,QACJ,6BAA6B,gBAAgB,SAAS,UAAU;AAAA,QAChE;AAAA,QACA;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,aAAa,IAAI;AAAA,UACjB,kBAAkB;AAAA,QACpB;AAAA,MACF;AAEA;AAAA,IACF;AAEA,QAAI,aAAa,KAAK,SAAS,IAAI,IAAI,IAAI;AAG3C,UAAM,iBAAiB,MAAM,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA,IAAI,QAAQ;AAAA,MACZ,IAAI;AAAA,IACN;AAEA,QAAI,mBAAmB,OAAO;AAC5B;AAAA,IACF;AAYA,QAAI,cAIsB;AAC1B,UAAM,wBACJ,IAAI,QAAQ,qBAAqB;AACnC,UAAM,oBAAoB,IAAI,QAAQ,qBAAqB,eAAe;AAE1E,QAAI,cAAc,WAAW,OAAO,IAAI,QAAQ,WAAW;AACzD,oBAAc;AAGd,YAAM,aAAa,WAAW,QAAQ;AACtC,YAAM,SAAS,WAAW;AAK1B,UAAI,oBAAoB,QAAQ;AAC9B,aAAK;AAAA,UACH;AAAA,UACA,IAAI;AAAA,UACJ,6DAA6D,MAAM,+BAA+B,iBAAiB;AAAA,UACnH;AAAA,UACA;AAAA,YACE,GAAG,QAAQ;AAAA,YACX,aAAa,IAAI;AAAA,YACjB,kBAAkB;AAAA,UACpB;AAAA,QACF;AAEA;AAAA,MACF;AAIA,UAAI,aAAa,uBAAuB;AACtC,aAAK;AAAA,UACH;AAAA,UACA,IAAI;AAAA,UACJ,6DAA6D,qBAAqB,+BAA+B,UAAU;AAAA,UAC3H;AAAA,UACA;AAAA,YACE,GAAG,QAAQ;AAAA,YACX,aAAa,IAAI;AAAA,YACjB,kBAAkB;AAAA,UACpB;AAAA,QACF;AAEA;AAAA,MACF;AAIA,UAAI,WAAW,6CAAqC;AAClD,cAAM,sBACJ,wBAAwB,WAAW;AAAA,UACjC;AAAA,UACA;AAAA,YACE,6BAA6B,MAAM;AACjC,mBAAK,4BAA4B,mBAAmB;AAAA,YACtD;AAAA,UACF;AAAA,QACF;AAEF,qBAAa;AAAA,MACf;AAEA,WAAK,cAAc,UAAU;AAAA,IAC/B,WAAW,YAAY;AACrB,oBAAc;AAGd,WAAK,KAAK;AAAA,QACR,4CAA4C,IAAI,QAAQ,SAAS,0BAA0B,WAAW,EAAE;AAAA,QACxG;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,aAAa,IAAI;AAAA,UACjB,WAAW,IAAI,QAAQ;AAAA,QACzB;AAAA,MACF;AACA,WAAK,cAAc,UAAU;AAC7B,mBAAa;AAAA,IACf;AAEA,QAAI,CAAC,eAAe,oBAAoB,KAAK,wBAAwB,IAAI;AAGvE,oBAAc;AACd,WAAK;AAAA,QACH;AAAA,QACA,IAAI;AAAA,QACJ,2EAA2E,IAAI,QAAQ,SAAS;AAAA,QAChG;AAAA,QACA;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,aAAa,IAAI;AAAA,UACjB,kBAAkB;AAAA,QACpB;AAAA,MACF;AACA;AAAA,IACF;AAGA,UAAM,YAAY,IAAI,QAAQ;AAC9B,SAAK,KAAK;AAAA,MACR,kBAAkB,IAAI,IAAI,QAAQ,WAAW;AAAA,MAC7C;AAAA,QACE,GAAG,QAAQ;AAAA,QACX,aAAa,IAAI;AAAA,MACnB;AAAA,IACF;AACA,UAAM,cAAc,yBAAyB;AAAA,MAC3C,MAAM,KAAK;AAAA,MACX,IAAI,IAAI;AAAA,MACR,QAAQ;AAAA,QACN,IAAI;AAAA,QACJ;AAAA,MACF;AAAA,IACF,CAAC;AACD,YAAQ,cAAc,WAAW;AAGjC,UAAM,mBACJ,wBAAwB,WAAW;AAAA,MACjC;AAAA;AAAA,MAEA;AAAA,MACA;AAAA,MACA,IAAI;AAAA,MACJ,IAAI;AAAA,MACJ;AAAA,QACE,qBAAqB,CAAC,QAAQ;AAE5B,gBAAM,SAAS,kBAAkB,GAAG;AACpC,eAAK,KAAK;AAAA,YACR,iBAAiB,iBAAiB,EAAE,aAAa,MAAM;AAAA,YACvD,iBAAiB;AAAA,UACnB;AAAA,QACF;AAAA,QACA,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR,iBAAiB,iBAAiB,EAAE;AAAA,YACpC,iBAAiB;AAAA,UACnB;AACA,eAAK,aAAa,gBAAgB;AAAA,QACpC;AAAA,QACA,WAAW,CAACA,SAAQ,KAAK,UAAUA,IAAG;AAAA,QACtC,kBAAkB,CAAC,WAAW;AAC5B,eAAK,cAAc,cAAc,yBAAyB,MAAM;AAChE,eAAK,cAAc,gBAAgB;AAAA,QACrC;AAAA,MACF;AAAA,IACF;AAEF,SAAK,yBAAyB,IAAI,iBAAiB,IAAI,cAAc;AACrE,SAAK,cAAc,gBAAgB;AACnC,SAAK,gBAAgB,OAAO,OAAO;AACnC,qBAAiB,qBAAqB;AAAA,EACxC;AAAA,EAEA,MAAc,0BACZ,oBACA,iBACA,aAGA,MACiC;AACjC,QAAI,iBAAiC,CAAC;AACtC,QAAI,KAAK,qBAAqB;AAE5B,UAAI,CAAC,MAAM,MAAM,KAAK,oBAAoB,QAAQ,WAAW,GAAG;AAC9D,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,YACE,GAAG,mBAAmB;AAAA,YACtB,aAAa;AAAA,YACb,kBAAkB;AAAA,cAChB,GAAG,MAAM,OAAO,KAAK,oBAAoB,QAAQ,WAAW;AAAA,YAC9D;AAAA,UACF;AAAA,QACF;AAEA,eAAO;AAAA,MACT;AAEA,YAAM,yBAAyB,kBAC3B,KAAK,yBAAyB,IAAI,gBAAgB,EAAE,IACpD;AAEJ,uBAAiB,MAAM,KAAK,oBAAoB;AAAA,QAC9C;AAAA,QACA;AAAA,MACF;AAGA,UAAI,mBAAmB,OAAO;AAC5B,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,YACE,GAAG,mBAAmB;AAAA,YACtB,aAAa;AAAA,YACb,UAAU,KAAK;AAAA,UACjB;AAAA,QACF;AAEA,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AACF;","names":["msg"]}

package/dist/{chunk-MBMEJIPU.js → chunk-ZWMWRM23.js}

@@ -1,6 +1,6 @@
 import {
   Connection
-} from "./chunk-D5O3ERJU.js";
+} from "./chunk-PF74DCPQ.js";
 
 // transport/impls/ws/connection.ts
 var WebSocketConnection = class extends Connection {
@@ -47,4 +47,4 @@ var WebSocketConnection = class extends Connection {
 export {
   WebSocketConnection
 };
-//# sourceMappingURL=chunk-MBMEJIPU.js.map
+//# sourceMappingURL=chunk-ZWMWRM23.js.map

package/dist/{client-1321630c.d.ts → client-edc8e2de.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection, T as Transport, o as ClientTransportOptions, L as LeakyBucketRateLimit, p as ClientHandshakeOptions, b as ProvidedClientTransportOptions, f as SessionConnecting, e as SessionNoConnection, g as SessionHandshaking, h as SessionConnected } from './handshake-3772d7ca.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-12d57161.js';
 import { c as TransportClientId, P as PartialTransportMessage, b as OpaqueTransportMessage } from './message-e6c560fd.js';
 
 declare abstract class ClientTransport<ConnType extends Connection> extends Transport<ConnType> {
@@ -18,6 +18,7 @@ declare abstract class ClientTransport<ConnType extends Connection> extends Tran
      * Optional handshake options for this client.
      */
     handshakeExtensions?: ClientHandshakeOptions;
+    sessions: Map<TransportClientId, ClientSession<ConnType>>;
     constructor(clientId: TransportClientId, providedOptions?: ProvidedClientTransportOptions);
     extendHandshake(options: ClientHandshakeOptions): void;
     /**
@@ -42,6 +43,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;
     private sendHandshake;
     close(): void;
 }

package/dist/{connection-bd907ca6.d.ts → connection-ef50ac3a.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection } from './handshake-3772d7ca.js';
+import { C as Connection } from './handshake-12d57161.js';
 import { W as WsLike } from './wslike-e0b32dd5.js';
 
 declare class WebSocketConnection extends Connection {

package/dist/{handshake-3772d7ca.d.ts → handshake-12d57161.d.ts}

@@ -1,4 +1,4 @@
-import { e as TelemetryInfo, M as MessageMetadata, a as TransportMessage, b as OpaqueTransportMessage, P as PartialTransportMessage, c as TransportClientId, L as Logger, f as LogFn, g as LoggingLevel } from './message-e6c560fd.js';
+import { e as TelemetryInfo, M as MessageMetadata, c as TransportClientId, L as Logger, b as OpaqueTransportMessage, P as PartialTransportMessage, a as TransportMessage, f as LogFn, g as LoggingLevel } from './message-e6c560fd.js';
 import { TSchema, Static } from '@sinclair/typebox';
 import { C as Codec } from './types-3e5768ec.js';
 
@@ -57,86 +57,14 @@ declare abstract class Connection {
     abstract close(): void;
 }
 
-interface SessionNoConnectionListeners {
-    onSessionGracePeriodElapsed: () => void;
-}
-declare class SessionNoConnection extends IdentifiedSession {
-    readonly state: SessionState.NoConnection;
-    listeners: SessionNoConnectionListeners;
-    gracePeriodTimeout?: ReturnType<typeof setTimeout>;
-    constructor(listeners: SessionNoConnectionListeners, ...args: ConstructorParameters<typeof IdentifiedSession>);
-    _handleClose(): void;
-    _handleStateExit(): void;
-}
-
-interface SessionConnectingListeners {
-    onConnectionEstablished: (conn: Connection) => void;
-    onConnectionFailed: (err: unknown) => void;
-    onConnectionTimeout: () => void;
-}
-declare class SessionConnecting<ConnType extends Connection> extends IdentifiedSession {
-    readonly state: SessionState.Connecting;
-    connPromise: Promise<ConnType>;
-    listeners: SessionConnectingListeners;
-    connectionTimeout?: ReturnType<typeof setTimeout>;
-    constructor(connPromise: Promise<ConnType>, listeners: SessionConnectingListeners, ...args: ConstructorParameters<typeof IdentifiedSession>);
-    bestEffortClose(): void;
-    _handleStateExit(): void;
-    _handleClose(): void;
-}
-
-interface SessionHandshakingListeners {
-    onConnectionErrored: (err: unknown) => void;
-    onConnectionClosed: () => void;
-    onHandshake: (msg: OpaqueTransportMessage) => void;
-    onInvalidHandshake: (reason: string) => void;
-    onHandshakeTimeout: () => void;
-}
-declare class SessionHandshaking<ConnType extends Connection> extends IdentifiedSession {
-    readonly state: SessionState.Handshaking;
-    conn: ConnType;
-    listeners: SessionHandshakingListeners;
-    handshakeTimeout: ReturnType<typeof setTimeout>;
-    constructor(conn: ConnType, listeners: SessionHandshakingListeners, ...args: ConstructorParameters<typeof IdentifiedSession>);
-    onHandshakeData: (msg: Uint8Array) => void;
-    sendHandshake(msg: TransportMessage): boolean;
-    _handleStateExit(): void;
-    _handleClose(): void;
-}
-
-interface SessionConnectedListeners {
-    onConnectionErrored: (err: unknown) => void;
-    onConnectionClosed: () => void;
-    onMessage: (msg: OpaqueTransportMessage) => void;
-    onInvalidMessage: (reason: string) => void;
-}
-declare class SessionConnected<ConnType extends Connection> extends IdentifiedSession {
-    readonly state: SessionState.Connected;
-    conn: ConnType;
-    listeners: SessionConnectedListeners;
-    private activeHeartbeatHandle?;
-    private activeHeartbeatMisses;
-    private passiveHeartbeatHandle?;
-    get isActivelyHeartbeating(): boolean;
-    updateBookkeeping(ack: number, seq: number): void;
-    send(msg: PartialTransportMessage): string;
-    constructor(conn: ConnType, listeners: SessionConnectedListeners, ...args: ConstructorParameters<typeof IdentifiedSession>);
-    startActiveHeartbeat(): void;
-    waitForNextHeartbeat(): void;
-    private sendHeartbeat;
-    onMessageData: (msg: Uint8Array) => void;
-    _handleStateExit(): void;
-    _handleClose(): void;
-}
-
 declare const enum SessionState {
     NoConnection = "NoConnection",
+    BackingOff = "BackingOff",
     Connecting = "Connecting",
     Handshaking = "Handshaking",
     Connected = "Connected",
     WaitingForHandshake = "WaitingForHandshake"
 }
-type Session<ConnType extends Connection> = SessionNoConnection | SessionConnecting<ConnType> | SessionHandshaking<ConnType> | SessionConnected<ConnType>;
 declare abstract class StateMachineState {
     abstract readonly state: SessionState;
     _isConsumed: boolean;
@@ -172,15 +100,28 @@ interface SessionOptions {
      */
     codec: Codec;
 }
+interface CommonSessionProps {
+    from: TransportClientId;
+    options: SessionOptions;
+    log: Logger | undefined;
+}
 declare abstract class CommonSession extends StateMachineState {
     readonly from: TransportClientId;
     readonly options: SessionOptions;
     log?: Logger;
     abstract get loggingMetadata(): MessageMetadata;
-    constructor(from: TransportClientId, options: SessionOptions, log: Logger | undefined);
+    constructor({ from, options, log }: CommonSessionProps);
     parseMsg(msg: Uint8Array): OpaqueTransportMessage | null;
 }
 type SessionId = string;
+interface IdentifiedSessionProps extends CommonSessionProps {
+    id: SessionId;
+    to: TransportClientId;
+    seq: number;
+    ack: number;
+    sendBuffer: Array<OpaqueTransportMessage>;
+    telemetry: TelemetryInfo;
+}
 declare abstract class IdentifiedSession extends CommonSession {
     readonly id: SessionId;
     readonly telemetry: TelemetryInfo;
@@ -194,7 +135,7 @@ declare abstract class IdentifiedSession extends CommonSession {
      */
     ack: number;
     sendBuffer: Array<OpaqueTransportMessage>;
-    constructor(id: SessionId, from: TransportClientId, to: TransportClientId, seq: number, ack: number, sendBuffer: Array<OpaqueTransportMessage>, telemetry: TelemetryInfo, options: SessionOptions, log: Logger | undefined);
+    constructor(props: IdentifiedSessionProps);
     get loggingMetadata(): MessageMetadata;
     constructMsg<Payload>(partialMsg: PartialTransportMessage<Payload>): TransportMessage<Payload>;
     nextSeq(): number;
@@ -203,6 +144,111 @@ declare abstract class IdentifiedSession extends CommonSession {
     _handleClose(): void;
 }
 
+interface SessionHandshakingListeners {
+    onConnectionErrored: (err: unknown) => void;
+    onConnectionClosed: () => void;
+    onHandshake: (msg: OpaqueTransportMessage) => void;
+    onInvalidHandshake: (reason: string) => void;
+    onHandshakeTimeout: () => void;
+}
+interface SessionHandshakingProps<ConnType extends Connection> extends IdentifiedSessionProps {
+    conn: ConnType;
+    listeners: SessionHandshakingListeners;
+}
+declare class SessionHandshaking<ConnType extends Connection> extends IdentifiedSession {
+    readonly state: SessionState.Handshaking;
+    conn: ConnType;
+    listeners: SessionHandshakingListeners;
+    handshakeTimeout: ReturnType<typeof setTimeout>;
+    constructor(props: SessionHandshakingProps<ConnType>);
+    onHandshakeData: (msg: Uint8Array) => void;
+    sendHandshake(msg: TransportMessage): boolean;
+    _handleStateExit(): void;
+    _handleClose(): void;
+}
+
+interface SessionConnectingListeners {
+    onConnectionEstablished: (conn: Connection) => void;
+    onConnectionFailed: (err: unknown) => void;
+    onConnectionTimeout: () => void;
+}
+interface SessionConnectingProps<ConnType extends Connection> extends IdentifiedSessionProps {
+    connPromise: Promise<ConnType>;
+    listeners: SessionConnectingListeners;
+}
+declare class SessionConnecting<ConnType extends Connection> extends IdentifiedSession {
+    readonly state: SessionState.Connecting;
+    connPromise: Promise<ConnType>;
+    listeners: SessionConnectingListeners;
+    connectionTimeout?: ReturnType<typeof setTimeout>;
+    constructor(props: SessionConnectingProps<ConnType>);
+    bestEffortClose(): void;
+    _handleStateExit(): void;
+    _handleClose(): void;
+}
+
+interface SessionNoConnectionListeners {
+    onSessionGracePeriodElapsed: () => void;
+}
+interface SessionNoConnectionProps extends IdentifiedSessionProps {
+    listeners: SessionNoConnectionListeners;
+}
+declare class SessionNoConnection extends IdentifiedSession {
+    readonly state: SessionState.NoConnection;
+    listeners: SessionNoConnectionListeners;
+    gracePeriodTimeout?: ReturnType<typeof setTimeout>;
+    constructor(props: SessionNoConnectionProps);
+    _handleClose(): void;
+    _handleStateExit(): void;
+}
+
+interface SessionConnectedListeners {
+    onConnectionErrored: (err: unknown) => void;
+    onConnectionClosed: () => void;
+    onMessage: (msg: OpaqueTransportMessage) => void;
+    onInvalidMessage: (reason: string) => void;
+}
+interface SessionConnectedProps<ConnType extends Connection> extends IdentifiedSessionProps {
+    conn: ConnType;
+    listeners: SessionConnectedListeners;
+}
+declare class SessionConnected<ConnType extends Connection> extends IdentifiedSession {
+    readonly state: SessionState.Connected;
+    conn: ConnType;
+    listeners: SessionConnectedListeners;
+    private heartbeatHandle?;
+    private heartbeatMisses;
+    isActivelyHeartbeating: boolean;
+    updateBookkeeping(ack: number, seq: number): void;
+    send(msg: PartialTransportMessage): string;
+    constructor(props: SessionConnectedProps<ConnType>);
+    startActiveHeartbeat(): void;
+    private sendHeartbeat;
+    onMessageData: (msg: Uint8Array) => void;
+    _handleStateExit(): void;
+    _handleClose(): void;
+}
+
+interface SessionBackingOffListeners {
+    onBackoffFinished: () => void;
+}
+interface SessionBackingOffProps extends IdentifiedSessionProps {
+    backoffMs: number;
+    listeners: SessionBackingOffListeners;
+}
+declare class SessionBackingOff extends IdentifiedSession {
+    readonly state: SessionState.BackingOff;
+    listeners: SessionBackingOffListeners;
+    backoffTimeout?: ReturnType<typeof setTimeout>;
+    constructor(props: SessionBackingOffProps);
+    _handleClose(): void;
+    _handleStateExit(): void;
+}
+
+type ClientSession<ConnType extends Connection> = SessionNoConnection | SessionBackingOff | SessionConnecting<ConnType> | SessionHandshaking<ConnType> | SessionConnected<ConnType>;
+type ServerSession<ConnType extends Connection> = SessionConnected<ConnType> | SessionNoConnection;
+type Session<ConnType extends Connection> = ClientSession<ConnType> | ServerSession<ConnType>;
+
 declare const ProtocolError: {
     readonly RetriesExceeded: "conn_retry_exceeded";
     readonly HandshakeFailed: "handshake_failed";
@@ -221,6 +267,8 @@ interface EventMap {
         state: SessionState.Handshaking;
     } | {
         state: SessionState.Connecting;
+    } | {
+        state: SessionState.BackingOff;
     } | {
         state: SessionState.NoConnection;
     };
@@ -282,15 +330,15 @@ interface ConnectionRetryOptions {
 }
 declare class LeakyBucketRateLimit {
     private budgetConsumed;
-    private intervalHandles;
+    private intervalHandle?;
     private readonly options;
     constructor(options: ConnectionRetryOptions);
-    getBackoffMs(user: TransportClientId): number;
+    getBackoffMs(): number;
     get totalBudgetRestoreTime(): number;
-    consumeBudget(user: TransportClientId): void;
-    getBudgetConsumed(user: TransportClientId): number;
-    hasBudget(user: TransportClientId): boolean;
-    startRestoringBudget(user: TransportClientId): void;
+    consumeBudget(): void;
+    getBudgetConsumed(): number;
+    hasBudget(): boolean;
+    startRestoringBudget(): void;
     private stopLeak;
     close(): void;
 }
@@ -485,4 +533,4 @@ interface ServerHandshakeOptions<MetadataSchema extends TSchema = TSchema> {
 declare function createClientHandshakeOptions<MetadataSchema extends TSchema = TSchema>(schema: MetadataSchema, construct: ConstructHandshake<MetadataSchema>): ClientHandshakeOptions;
 declare function createServerHandshakeOptions<MetadataSchema extends TSchema = TSchema>(schema: MetadataSchema, validate: ValidateHandshake<MetadataSchema>): ServerHandshakeOptions;
 
-export { Connection as C, EventMap as E, LeakyBucketRateLimit as L, ProvidedTransportOptions as P, Session as S, Transport as T, TransportStatus as a, ProvidedClientTransportOptions as b, ProvidedServerTransportOptions as c, SessionState as d, SessionNoConnection as e, SessionConnecting as f, SessionHandshaking as g, SessionConnected as h, EventTypes as i, EventHandler as j, ProtocolError as k, ProtocolErrorType as l, SessionOptions as m, ServiceContext as n, ClientTransportOptions as o, ClientHandshakeOptions as p, ServerHandshakeOptions as q, ParsedMetadata as r, ServiceContextWithState as s, ServiceContextWithTransportInfo as t, createClientHandshakeOptions as u, createServerHandshakeOptions as v, CommonSession as w, SessionHandshakingListeners as x, ServerTransportOptions as y };
+export { CommonSessionProps as A, ServerTransportOptions as B, Connection as C, ServerSession as D, EventMap as E, LeakyBucketRateLimit as L, ProvidedTransportOptions as P, Session as S, Transport as T, TransportStatus as a, ProvidedClientTransportOptions as b, ProvidedServerTransportOptions as c, SessionState as d, SessionNoConnection as e, SessionConnecting as f, SessionHandshaking as g, SessionConnected as h, EventTypes as i, EventHandler as j, ProtocolError as k, ProtocolErrorType as l, SessionOptions as m, ClientTransportOptions as n, ServiceContext as o, ClientHandshakeOptions as p, ClientSession as q, SessionBackingOff as r, ServerHandshakeOptions as s, ParsedMetadata as t, ServiceContextWithState as u, ServiceContextWithTransportInfo as v, createClientHandshakeOptions as w, createServerHandshakeOptions as x, CommonSession as y, SessionHandshakingListeners as z };

package/dist/router/index.cjs

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