@replit/river 0.200.0-rc.9 → 0.200.2

package/dist/transport/impls/ws/client.d.cts

@@ -1,7 +1,7 @@
-import { C as ClientTransport } from '../../../client-22a47343.js';
-import { c as TransportClientId } from '../../../message-7d135e38.js';
-import { b as ProvidedClientTransportOptions } from '../../../context-b4aff18f.js';
-import { W as WebSocketConnection } from '../../../connection-260e45a8.js';
+import { C as ClientTransport } from '../../../client-9292552a.js';
+import { c as TransportClientId } from '../../../message-57bb8187.js';
+import { b as ProvidedClientTransportOptions } from '../../../context-69f37ac1.js';
+import { W as WebSocketConnection } from '../../../connection-94dea547.js';
 import { W as WsLike } from '../../../wslike-e0b32dd5.js';
 import '@sinclair/typebox/value';
 import '@sinclair/typebox';

package/dist/transport/impls/ws/client.d.ts

@@ -1,7 +1,7 @@
-import { C as ClientTransport } from '../../../client-22a47343.js';
-import { c as TransportClientId } from '../../../message-7d135e38.js';
-import { b as ProvidedClientTransportOptions } from '../../../context-b4aff18f.js';
-import { W as WebSocketConnection } from '../../../connection-260e45a8.js';
+import { C as ClientTransport } from '../../../client-9292552a.js';
+import { c as TransportClientId } from '../../../message-57bb8187.js';
+import { b as ProvidedClientTransportOptions } from '../../../context-69f37ac1.js';
+import { W as WebSocketConnection } from '../../../connection-94dea547.js';
 import { W as WsLike } from '../../../wslike-e0b32dd5.js';
 import '@sinclair/typebox/value';
 import '@sinclair/typebox';

package/dist/transport/impls/ws/client.js

@@ -1,14 +1,14 @@
 import {
   ClientTransport
-} from "../../../chunk-4HT6P2ZG.js";
+} from "../../../chunk-A4JKES5A.js";
 import {
   WebSocketConnection
-} from "../../../chunk-GR3AQKHL.js";
-import "../../../chunk-I75XYO5W.js";
-import "../../../chunk-VXYHC666.js";
-import "../../../chunk-42Z2FQIU.js";
-import "../../../chunk-EETL2L77.js";
-import "../../../chunk-4PVU7J25.js";
+} from "../../../chunk-HRKM7BIE.js";
+import "../../../chunk-QIDEN5PP.js";
+import "../../../chunk-YTMS7OP6.js";
+import "../../../chunk-6BH2CXVE.js";
+import "../../../chunk-GJUUVID2.js";
+import "../../../chunk-AJGIY2UB.js";
 
 // transport/impls/ws/client.ts
 var WebSocketClientTransport = class extends ClientTransport {

package/dist/transport/impls/ws/client.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../transport/impls/ws/client.ts"],"sourcesContent":["import { ClientTransport } from '../../client';\nimport { TransportClientId } from '../../message';\nimport { ProvidedClientTransportOptions } from '../../options';\nimport { WebSocketConnection } from './connection';\nimport { WsLike } from './wslike';\n\n/**\n * A transport implementation that uses a WebSocket connection with automatic reconnection.\n * @class\n * @extends Transport\n */\nexport class WebSocketClientTransport extends ClientTransport<WebSocketConnection> {\n  /**\n   * A function that returns a Promise that resolves to a websocket URL.\n   */\n  wsGetter: (to: TransportClientId) => Promise<WsLike> | WsLike;\n\n  /**\n   * Creates a new WebSocketClientTransport instance.\n   * @param wsGetter A function that returns a Promise that resolves to a WebSocket instance.\n   * @param clientId The ID of the client using the transport. This should be unique per session.\n   * @param serverId The ID of the server this transport is connecting to.\n   * @param providedOptions An optional object containing configuration options for the transport.\n   */\n  constructor(\n    wsGetter: (to: TransportClientId) => Promise<WsLike> | WsLike,\n    clientId: TransportClientId,\n    providedOptions?: ProvidedClientTransportOptions,\n  ) {\n    super(clientId, providedOptions);\n    this.wsGetter = wsGetter;\n  }\n\n  async createNewOutgoingConnection(to: string) {\n    this.log?.info(`establishing a new websocket to ${to}`, {\n      clientId: this.clientId,\n      connectedTo: to,\n    });\n\n    const ws = await this.wsGetter(to);\n\n    await new Promise<void>((resolve, reject) => {\n      if (ws.readyState === ws.OPEN) {\n        resolve();\n        return;\n      }\n\n      if (ws.readyState === ws.CLOSING || ws.readyState === ws.CLOSED) {\n        reject(new Error('ws is closing or closed'));\n        return;\n      }\n\n      ws.onopen = () => {\n        resolve();\n      };\n\n      ws.onclose = (evt) => {\n        reject(new Error(evt.reason));\n      };\n\n      ws.onerror = (err) => {\n        reject(new Error(err.message));\n      };\n    });\n\n    const conn = new WebSocketConnection(ws);\n    this.log?.info(`raw websocket to ${to} ok`, {\n      clientId: this.clientId,\n      connectedTo: to,\n      ...conn.loggingMetadata,\n    });\n\n    return conn;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AAWO,IAAM,2BAAN,cAAuC,gBAAqC;AAAA;AAAA;AAAA;AAAA,EAIjF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,YACE,UACA,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,MAAM,4BAA4B,IAAY;AAC5C,SAAK,KAAK,KAAK,mCAAmC,EAAE,IAAI;AAAA,MACtD,UAAU,KAAK;AAAA,MACf,aAAa;AAAA,IACf,CAAC;AAED,UAAM,KAAK,MAAM,KAAK,SAAS,EAAE;AAEjC,UAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,UAAI,GAAG,eAAe,GAAG,MAAM;AAC7B,gBAAQ;AACR;AAAA,MACF;AAEA,UAAI,GAAG,eAAe,GAAG,WAAW,GAAG,eAAe,GAAG,QAAQ;AAC/D,eAAO,IAAI,MAAM,yBAAyB,CAAC;AAC3C;AAAA,MACF;AAEA,SAAG,SAAS,MAAM;AAChB,gBAAQ;AAAA,MACV;AAEA,SAAG,UAAU,CAAC,QAAQ;AACpB,eAAO,IAAI,MAAM,IAAI,MAAM,CAAC;AAAA,MAC9B;AAEA,SAAG,UAAU,CAAC,QAAQ;AACpB,eAAO,IAAI,MAAM,IAAI,OAAO,CAAC;AAAA,MAC/B;AAAA,IACF,CAAC;AAED,UAAM,OAAO,IAAI,oBAAoB,EAAE;AACvC,SAAK,KAAK,KAAK,oBAAoB,EAAE,OAAO;AAAA,MAC1C,UAAU,KAAK;AAAA,MACf,aAAa;AAAA,MACb,GAAG,KAAK;AAAA,IACV,CAAC;AAED,WAAO;AAAA,EACT;AACF;","names":[]}
+{"version":3,"sources":["../../../../transport/impls/ws/client.ts"],"sourcesContent":["import { ClientTransport } from '../../client';\nimport { TransportClientId } from '../../message';\nimport { ProvidedClientTransportOptions } from '../../options';\nimport { WebSocketConnection } from './connection';\nimport { WsLike } from './wslike';\n\n/**\n * A transport implementation that uses a WebSocket connection with automatic reconnection.\n * @class\n * @extends Transport\n */\nexport class WebSocketClientTransport extends ClientTransport<WebSocketConnection> {\n  /**\n   * A function that returns a Promise that resolves to a websocket URL.\n   */\n  wsGetter: (to: TransportClientId) => Promise<WsLike> | WsLike;\n\n  /**\n   * Creates a new WebSocketClientTransport instance.\n   * @param wsGetter A function that returns a Promise that resolves to a WebSocket instance.\n   * @param clientId The ID of the client using the transport. This should be unique per session.\n   * @param serverId The ID of the server this transport is connecting to.\n   * @param providedOptions An optional object containing configuration options for the transport.\n   */\n  constructor(\n    wsGetter: (to: TransportClientId) => Promise<WsLike> | WsLike,\n    clientId: TransportClientId,\n    providedOptions?: ProvidedClientTransportOptions,\n  ) {\n    super(clientId, providedOptions);\n    this.wsGetter = wsGetter;\n  }\n\n  async createNewOutgoingConnection(to: string) {\n    this.log?.info(`establishing a new websocket to ${to}`, {\n      clientId: this.clientId,\n      connectedTo: to,\n    });\n\n    const ws = await this.wsGetter(to);\n\n    await new Promise<void>((resolve, reject) => {\n      if (ws.readyState === ws.OPEN) {\n        resolve();\n\n        return;\n      }\n\n      if (ws.readyState === ws.CLOSING || ws.readyState === ws.CLOSED) {\n        reject(new Error('ws is closing or closed'));\n\n        return;\n      }\n\n      ws.onopen = () => {\n        resolve();\n      };\n\n      ws.onclose = (evt) => {\n        reject(new Error(evt.reason));\n      };\n\n      ws.onerror = (err) => {\n        reject(new Error(err.message));\n      };\n    });\n\n    const conn = new WebSocketConnection(ws);\n    this.log?.info(`raw websocket to ${to} ok`, {\n      clientId: this.clientId,\n      connectedTo: to,\n      ...conn.loggingMetadata,\n    });\n\n    return conn;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AAWO,IAAM,2BAAN,cAAuC,gBAAqC;AAAA;AAAA;AAAA;AAAA,EAIjF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,YACE,UACA,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,MAAM,4BAA4B,IAAY;AAC5C,SAAK,KAAK,KAAK,mCAAmC,EAAE,IAAI;AAAA,MACtD,UAAU,KAAK;AAAA,MACf,aAAa;AAAA,IACf,CAAC;AAED,UAAM,KAAK,MAAM,KAAK,SAAS,EAAE;AAEjC,UAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,UAAI,GAAG,eAAe,GAAG,MAAM;AAC7B,gBAAQ;AAER;AAAA,MACF;AAEA,UAAI,GAAG,eAAe,GAAG,WAAW,GAAG,eAAe,GAAG,QAAQ;AAC/D,eAAO,IAAI,MAAM,yBAAyB,CAAC;AAE3C;AAAA,MACF;AAEA,SAAG,SAAS,MAAM;AAChB,gBAAQ;AAAA,MACV;AAEA,SAAG,UAAU,CAAC,QAAQ;AACpB,eAAO,IAAI,MAAM,IAAI,MAAM,CAAC;AAAA,MAC9B;AAEA,SAAG,UAAU,CAAC,QAAQ;AACpB,eAAO,IAAI,MAAM,IAAI,OAAO,CAAC;AAAA,MAC/B;AAAA,IACF,CAAC;AAED,UAAM,OAAO,IAAI,oBAAoB,EAAE;AACvC,SAAK,KAAK,KAAK,oBAAoB,EAAE,OAAO;AAAA,MAC1C,UAAU,KAAK;AAAA,MACf,aAAa;AAAA,MACb,GAAG,KAAK;AAAA,IACV,CAAC;AAED,WAAO;AAAA,EACT;AACF;","names":[]}

package/dist/transport/impls/ws/server.cjs

@@ -103,11 +103,21 @@ var Connection = class {
 };
 
 // transport/impls/ws/connection.ts
+var WS_HEALTHY_CLOSE_CODE = 1e3;
 var WebSocketConnection = class extends Connection {
   ws;
-  constructor(ws) {
+  extras;
+  get loggingMetadata() {
+    const metadata = super.loggingMetadata;
+    if (this.extras) {
+      metadata.extras = this.extras;
+    }
+    return metadata;
+  }
+  constructor(ws, extras) {
     super();
     this.ws = ws;
+    this.extras = extras;
     this.ws.binaryType = "arraybuffer";
     let didError = false;
     this.ws.onerror = () => {
@@ -140,7 +150,7 @@ var WebSocketConnection = class extends Connection {
     return true;
   }
   close() {
-    this.ws.close();
+    this.ws.close(WS_HEALTHY_CLOSE_CODE);
   }
 };
 
@@ -175,6 +185,9 @@ var ControlMessageCloseSchema = import_typebox.Type.Object({
 });
 var currentProtocolVersion = "v2.0";
 var acceptedProtocolVersions = ["v1.1", currentProtocolVersion];
+function isAcceptedProtocolVersion(version2) {
+  return acceptedProtocolVersions.includes(version2);
+}
 var ControlMessageHandshakeRequestSchema = import_typebox.Type.Object({
   type: import_typebox.Type.Literal("HANDSHAKE_REQ"),
   protocolVersion: import_typebox.Type.String(),
@@ -187,10 +200,7 @@ var ControlMessageHandshakeRequestSchema = import_typebox.Type.Object({
   expectedSessionState: import_typebox.Type.Object({
     // what the client expects the server to send next
     nextExpectedSeq: import_typebox.Type.Integer(),
-    // TODO: remove optional once we know all servers
-    // are nextSentSeq here
-    // what the server expects the client to send next
-    nextSentSeq: import_typebox.Type.Optional(import_typebox.Type.Integer())
+    nextSentSeq: import_typebox.Type.Integer()
   }),
   metadata: import_typebox.Type.Optional(import_typebox.Type.Unknown())
 });
@@ -226,9 +236,7 @@ var ControlMessageHandshakeResponseSchema = import_typebox.Type.Object({
     import_typebox.Type.Object({
       ok: import_typebox.Type.Literal(false),
       reason: import_typebox.Type.String(),
-      // TODO: remove optional once we know all servers
-      // are sending code here
-      code: import_typebox.Type.Optional(HandshakeErrorResponseCodes)
+      code: HandshakeErrorResponseCodes
     })
   ])
 });
@@ -398,7 +406,8 @@ var createLogProxy = (log) => ({
 var ProtocolError = {
   RetriesExceeded: "conn_retry_exceeded",
   HandshakeFailed: "handshake_failed",
-  MessageOrderingViolated: "message_ordering_violated"
+  MessageOrderingViolated: "message_ordering_violated",
+  InvalidMessage: "invalid_message"
 };
 var EventDispatcher = class {
   eventListeners = {};
@@ -676,7 +685,7 @@ var SessionNoConnection = class extends IdentifiedSessionWithGracePeriod {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.200.0-rc.9";
+var version = "0.200.2";
 
 // tracing/index.ts
 function createSessionTelemetryInfo(sessionId, to, from, propagationCtx) {
@@ -829,13 +838,13 @@ var SessionConnected = class extends IdentifiedSession {
     this.conn.addCloseListener(this.listeners.onConnectionClosed);
     this.conn.addErrorListener(this.listeners.onConnectionErrored);
     if (this.sendBuffer.length > 0) {
-      this.log?.debug(
-        `sending ${this.sendBuffer.length} buffered messages`,
+      this.log?.info(
+        `sending ${this.sendBuffer.length} buffered messages, starting at seq ${this.nextSeq()}`,
         this.loggingMetadata
       );
-    }
-    for (const msg of this.sendBuffer) {
-      this.conn.send(this.options.codec.toBuffer(msg));
+      for (const msg of this.sendBuffer) {
+        this.conn.send(this.options.codec.toBuffer(msg));
+      }
     }
     this.isActivelyHeartbeating = false;
     this.heartbeatHandle = setInterval(() => {
@@ -877,6 +886,12 @@ var SessionConnected = class extends IdentifiedSession {
       }
     });
   }
+  closeConnection() {
+    this.conn.removeDataListener(this.onMessageData);
+    this.conn.removeCloseListener(this.listeners.onConnectionClosed);
+    this.conn.removeErrorListener(this.listeners.onConnectionErrored);
+    this.conn.close();
+  }
   onMessageData = (msg) => {
     const parsedMsg = this.parseMsg(msg);
     if (parsedMsg === null) {
@@ -893,8 +908,8 @@ var SessionConnected = class extends IdentifiedSession {
           }
         );
       } else {
-        const reason = `received out-of-order msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;
-        this.log?.error(reason, {
+        const reason = `received out-of-order msg, closing connection (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;
+        this.log?.warn(reason, {
           ...this.loggingMetadata,
           transportMessage: parsedMsg,
           tags: ["invariant-violation"]
@@ -903,7 +918,7 @@ var SessionConnected = class extends IdentifiedSession {
           code: import_api2.SpanStatusCode.ERROR,
           message: reason
         });
-        this.listeners.onInvalidMessage(reason);
+        this.closeConnection();
       }
       return;
     }
@@ -929,8 +944,10 @@ var SessionConnected = class extends IdentifiedSession {
     this.conn.removeDataListener(this.onMessageData);
     this.conn.removeCloseListener(this.listeners.onConnectionClosed);
     this.conn.removeErrorListener(this.listeners.onConnectionErrored);
-    clearInterval(this.heartbeatHandle);
-    this.heartbeatHandle = void 0;
+    if (this.heartbeatHandle) {
+      clearInterval(this.heartbeatHandle);
+      this.heartbeatHandle = void 0;
+    }
   }
   _handleClose() {
     super._handleClose();
@@ -1297,12 +1314,12 @@ var Transport = class {
   /**
    * Called when a message is received by this transport.
    * You generally shouldn't need to override this in downstream transport implementations.
-   * @param msg The received message.
+   * @param message The received message.
    */
-  handleMsg(msg) {
+  handleMsg(message) {
     if (this.getStatus() !== "open")
       return;
-    this.eventDispatcher.dispatchEvent("message", msg);
+    this.eventDispatcher.dispatchEvent("message", message);
   }
   /**
    * Adds a listener to this transport.
@@ -1342,28 +1359,59 @@ var Transport = class {
   getStatus() {
     return this.status;
   }
-  updateSession(session) {
+  // state transitions
+  createSession(session) {
     const activeSession = this.sessions.get(session.to);
-    if (activeSession && activeSession.id !== session.id) {
-      const msg = `attempt to transition active session for ${session.to} but active session (${activeSession.id}) is different from handle (${session.id})`;
+    if (activeSession) {
+      const msg = `attempt to create session for ${session.to} but active session (${activeSession.id}) already exists`;
+      this.log?.error(msg, {
+        ...session.loggingMetadata,
+        tags: ["invariant-violation"]
+      });
       throw new Error(msg);
     }
     this.sessions.set(session.to, session);
+    this.eventDispatcher.dispatchEvent("sessionStatus", {
+      status: "connect",
+      session
+    });
+    this.eventDispatcher.dispatchEvent("sessionTransition", {
+      state: session.state,
+      session
+    });
+  }
+  updateSession(session) {
+    const activeSession = this.sessions.get(session.to);
     if (!activeSession) {
-      this.eventDispatcher.dispatchEvent("sessionStatus", {
-        status: "connect",
-        session
+      const msg = `attempt to transition session for ${session.to} but no active session exists`;
+      this.log?.error(msg, {
+        ...session.loggingMetadata,
+        tags: ["invariant-violation"]
+      });
+      throw new Error(msg);
+    }
+    if (activeSession.id !== session.id) {
+      const msg = `attempt to transition active session for ${session.to} but active session (${activeSession.id}) is different from handle (${session.id})`;
+      this.log?.error(msg, {
+        ...session.loggingMetadata,
+        tags: ["invariant-violation"]
       });
+      throw new Error(msg);
     }
+    this.sessions.set(session.to, session);
     this.eventDispatcher.dispatchEvent("sessionTransition", {
       state: session.state,
       session
     });
-    return session;
   }
-  // state transitions
-  deleteSession(session) {
-    session.log?.info(`closing session ${session.id}`, session.loggingMetadata);
+  deleteSession(session, options) {
+    if (session._isConsumed)
+      return;
+    const loggingMetadata = session.loggingMetadata;
+    if (loggingMetadata.tags && options?.unhealthy) {
+      loggingMetadata.tags.push("unhealthy-session");
+    }
+    session.log?.info(`closing session ${session.id}`, loggingMetadata);
     this.eventDispatcher.dispatchEvent("sessionStatus", {
       status: "disconnect",
       session
@@ -1386,7 +1434,8 @@ var Transport = class {
         this.onSessionGracePeriodElapsed(noConnectionSession);
       }
     });
-    return this.updateSession(noConnectionSession);
+    this.updateSession(noConnectionSession);
+    return noConnectionSession;
   }
   onConnClosed(session) {
     let noConnectionSession;
@@ -1403,7 +1452,36 @@ var Transport = class {
         }
       });
     }
-    return this.updateSession(noConnectionSession);
+    this.updateSession(noConnectionSession);
+    return noConnectionSession;
+  }
+  /**
+   * Gets a send closure scoped to a specific session. Sending using the returned
+   * closure after the session has transitioned to a different state will be a noop.
+   *
+   * Session objects themselves can become stale as they transition between
+   * states. As stale sessions cannot be used again (and will throw), holding
+   * onto a session object is not recommended.
+   */
+  getSessionBoundSendFn(to, sessionId) {
+    if (this.getStatus() !== "open") {
+      throw new Error("cannot get a bound send function on a closed transport");
+    }
+    return (msg) => {
+      const session = this.sessions.get(to);
+      if (!session) {
+        throw new Error(
+          `session scope for ${sessionId} has ended (close), can't send`
+        );
+      }
+      const sameSession = session.id === sessionId;
+      if (!sameSession) {
+        throw new Error(
+          `session scope for ${sessionId} has ended (transition), can't send`
+        );
+      }
+      return session.send(msg);
+    };
   }
 };
 
@@ -1447,35 +1525,13 @@ var ServerTransport = class extends Transport {
   extendHandshake(options) {
     this.handshakeExtensions = options;
   }
-  send(to, msg) {
-    if (this.getStatus() === "closed") {
-      const err = "transport is closed, cant send";
-      this.log?.error(err, {
-        clientId: this.clientId,
-        transportMessage: msg,
-        tags: ["invariant-violation"]
-      });
-      throw new Error(err);
-    }
-    const session = this.sessions.get(to);
-    if (!session) {
-      const err = `session to ${to} does not exist`;
-      this.log?.error(err, {
-        clientId: this.clientId,
-        transportMessage: msg,
-        tags: ["invariant-violation"]
-      });
-      throw new Error(err);
-    }
-    return session.send(msg);
-  }
   deletePendingSession(pendingSession) {
     pendingSession.close();
     this.pendingSessions.delete(pendingSession);
   }
-  deleteSession(session) {
+  deleteSession(session, options) {
     this.sessionHandshakeMetadata.delete(session.to);
-    super.deleteSession(session);
+    super.deleteSession(session, options);
   }
   handleConnection(conn) {
     if (this.getStatus() !== "open")
@@ -1588,7 +1644,7 @@ var ServerTransport = class extends Transport {
       return;
     }
     const gotVersion = msg.payload.protocolVersion;
-    if (!acceptedProtocolVersions.includes(gotVersion)) {
+    if (!isAcceptedProtocolVersion(gotVersion)) {
       this.rejectHandshakeRequest(
         session,
         msg.from,
@@ -1602,7 +1658,6 @@ var ServerTransport = class extends Transport {
       );
       return;
     }
-    let oldSession = this.sessions.get(msg.from);
     let parsedMetadata = {};
     if (this.handshakeExtensions) {
       if (!import_value2.Value.Check(this.handshakeExtensions.schema, msg.payload.metadata)) {
@@ -1624,7 +1679,9 @@ var ServerTransport = class extends Transport {
         );
         return;
       }
-      const previousParsedMetadata = oldSession ? this.sessionHandshakeMetadata.get(oldSession.to) : void 0;
+      const previousParsedMetadata = this.sessionHandshakeMetadata.get(
+        msg.from
+      );
       const parsedMetadataOrFailureCode = await this.handshakeExtensions.validate(
         msg.payload.metadata,
         previousParsedMetadata
@@ -1653,7 +1710,8 @@ var ServerTransport = class extends Transport {
     }
     let connectCase = "new session";
     const clientNextExpectedSeq = msg.payload.expectedSessionState.nextExpectedSeq;
-    const clientNextSentSeq = msg.payload.expectedSessionState.nextSentSeq ?? 0;
+    const clientNextSentSeq = msg.payload.expectedSessionState.nextSentSeq;
+    let oldSession = this.sessions.get(msg.from);
     if (this.options.enableTransparentSessionReconnects && oldSession && oldSession.id === msg.payload.sessionId) {
       connectCase = "transparent reconnection";
       const ourNextSeq = oldSession.nextSeq();
@@ -1766,25 +1824,41 @@ var ServerTransport = class extends Transport {
           );
           this.onConnClosed(connectedSession);
         },
-        onMessage: (msg2) => this.handleMsg(msg2),
+        onMessage: (msg2) => {
+          this.handleMsg(msg2);
+        },
         onInvalidMessage: (reason) => {
           this.protocolError({
-            type: ProtocolError.MessageOrderingViolated,
+            type: ProtocolError.InvalidMessage,
             message: reason
           });
-          this.deleteSession(connectedSession);
+          this.deleteSession(connectedSession, { unhealthy: true });
         }
       },
       gotVersion
     );
     this.sessionHandshakeMetadata.set(connectedSession.to, parsedMetadata);
-    this.updateSession(connectedSession);
+    if (oldSession) {
+      this.updateSession(connectedSession);
+    } else {
+      this.createSession(connectedSession);
+    }
     this.pendingSessions.delete(session);
     connectedSession.startActiveHeartbeat();
   }
 };
 
 // transport/impls/ws/server.ts
+function cleanHeaders(headers) {
+  const cleanedHeaders = {};
+  for (const [key, value] of Object.entries(headers)) {
+    if (!key.startsWith("sec-") && value) {
+      const cleanedValue = Array.isArray(value) ? value[0] : value;
+      cleanedHeaders[key] = cleanedValue;
+    }
+  }
+  return cleanedHeaders;
+}
 var WebSocketServerTransport = class extends ServerTransport {
   wss;
   constructor(wss, clientId, providedOptions) {
@@ -1792,8 +1866,10 @@ var WebSocketServerTransport = class extends ServerTransport {
     this.wss = wss;
     this.wss.on("connection", this.connectionHandler);
   }
-  connectionHandler = (ws) => {
-    const conn = new WebSocketConnection(ws);
+  connectionHandler = (ws, req) => {
+    const conn = new WebSocketConnection(ws, {
+      headers: cleanHeaders(req.headersDistinct)
+    });
     this.handleConnection(conn);
   };
   close() {