@replit/river 0.26.5 → 0.26.7

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

@@ -1,7 +1,7 @@
-import { C as ClientTransport } from '../../../client-6c67339a.js';
-import { c as TransportClientId } from '../../../message-ff78a233.js';
-import { b as ProvidedClientTransportOptions } from '../../../handshake-154a0bb2.js';
-import { W as WebSocketConnection } from '../../../connection-03ffb583.js';
+import { C as ClientTransport } from '../../../client-1229b53d.js';
+import { c as TransportClientId } from '../../../message-3edb8fac.js';
+import { b as ProvidedClientTransportOptions } from '../../../handshake-02a61755.js';
+import { W as WebSocketConnection } from '../../../connection-c5ffd44e.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-6c67339a.js';
-import { c as TransportClientId } from '../../../message-ff78a233.js';
-import { b as ProvidedClientTransportOptions } from '../../../handshake-154a0bb2.js';
-import { W as WebSocketConnection } from '../../../connection-03ffb583.js';
+import { C as ClientTransport } from '../../../client-1229b53d.js';
+import { c as TransportClientId } from '../../../message-3edb8fac.js';
+import { b as ProvidedClientTransportOptions } from '../../../handshake-02a61755.js';
+import { W as WebSocketConnection } from '../../../connection-c5ffd44e.js';
 import { W as WsLike } from '../../../wslike-e0b32dd5.js';
 import '@sinclair/typebox/value';
 import '@sinclair/typebox';

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

@@ -1,13 +1,13 @@
 import {
   ClientTransport
-} from "../../../chunk-PZEYFHAB.js";
+} from "../../../chunk-JJX4ALPO.js";
 import {
   WebSocketConnection
-} from "../../../chunk-QEWSUVWV.js";
-import "../../../chunk-IPOUVVKT.js";
-import "../../../chunk-TAH2GVTJ.js";
-import "../../../chunk-XRZ52TYH.js";
-import "../../../chunk-7KB7DHFZ.js";
+} from "../../../chunk-KMH4I245.js";
+import "../../../chunk-I3Z6XQVK.js";
+import "../../../chunk-KTS4HBYE.js";
+import "../../../chunk-F32IOMOX.js";
+import "../../../chunk-NHM3XC6K.js";
 import "../../../chunk-4PVU7J25.js";
 
 // transport/impls/ws/client.ts

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

@@ -105,9 +105,18 @@ var Connection = class {
 // transport/impls/ws/connection.ts
 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 = () => {
@@ -392,7 +401,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 = {};
@@ -668,7 +678,7 @@ var SessionNoConnection = class extends IdentifiedSessionWithGracePeriod {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.26.5";
+var version = "0.26.7";
 
 // tracing/index.ts
 function createSessionTelemetryInfo(sessionId, to, from, propagationCtx) {
@@ -869,6 +879,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) {
@@ -885,8 +901,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"]
@@ -895,7 +911,7 @@ var SessionConnected = class extends IdentifiedSession {
           code: import_api2.SpanStatusCode.ERROR,
           message: reason
         });
-        this.listeners.onInvalidMessage(reason);
+        this.closeConnection();
       }
       return;
     }
@@ -921,8 +937,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();
@@ -1351,8 +1369,12 @@ var Transport = class {
     return session;
   }
   // state transitions
-  deleteSession(session) {
-    session.log?.info(`closing session ${session.id}`, session.loggingMetadata);
+  deleteSession(session, options) {
+    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
@@ -1462,9 +1484,9 @@ var ServerTransport = class extends Transport {
     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")
@@ -1758,10 +1780,10 @@ var ServerTransport = class extends Transport {
         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 });
         }
       }
     );
@@ -1773,6 +1795,16 @@ var ServerTransport = class extends Transport {
 };
 
 // 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) {
@@ -1780,8 +1812,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() {