@replit/river 0.214.0 → 0.215.0

package/dist/{chunk-DRYCLL6N.js → chunk-VXRAFUA3.js}

@@ -20,7 +20,7 @@ import {
   handshakeResponseMessage,
   isAcceptedProtocolVersion,
   isAck
-} from "./chunk-LECVFB2G.js";
+} from "./chunk-RDTQZ7HO.js";
 
 // transport/events.ts
 var ProtocolError = {
@@ -226,6 +226,7 @@ var IdentifiedSession = class extends CommonSession {
   telemetry;
   to;
   protocolVersion;
+  listeners;
   /**
    * Index of the message we will send next (excluding handshake)
    */
@@ -249,7 +250,8 @@ var IdentifiedSession = class extends CommonSession {
       telemetry,
       log,
       protocolVersion,
-      seqSent: messagesSent
+      seqSent: messagesSent,
+      listeners
     } = props;
     super(props);
     this.id = id;
@@ -261,6 +263,7 @@ var IdentifiedSession = class extends CommonSession {
     this.log = log;
     this.protocolVersion = protocolVersion;
     this.seqSent = messagesSent;
+    this.listeners = listeners;
   }
   get loggingMetadata() {
     const metadata = {
@@ -277,7 +280,7 @@ var IdentifiedSession = class extends CommonSession {
     }
     return metadata;
   }
-  constructMsg(partialMsg) {
+  encodeMsg(partialMsg) {
     const msg = {
       ...partialMsg,
       id: generateId(),
@@ -286,18 +289,37 @@ var IdentifiedSession = class extends CommonSession {
       seq: this.seq,
       ack: this.ack
     };
+    const encoded = this.codec.toBuffer(msg);
+    if (!encoded.ok) {
+      this.listeners.onMessageSendFailure(
+        { ...partialMsg, seq: this.seq },
+        encoded.reason
+      );
+      return encoded;
+    }
     this.seq++;
-    return msg;
+    return {
+      ok: true,
+      value: {
+        id: msg.id,
+        seq: msg.seq,
+        msg: partialMsg,
+        data: encoded.value
+      }
+    };
   }
   nextSeq() {
     return this.sendBuffer.length > 0 ? this.sendBuffer[0].seq : this.seq;
   }
   send(msg) {
-    const constructedMsg = this.constructMsg(msg);
-    this.sendBuffer.push(constructedMsg);
+    const encodeResult = this.encodeMsg(msg);
+    if (!encodeResult.ok) {
+      return encodeResult;
+    }
+    this.sendBuffer.push(encodeResult.value);
     return {
       ok: true,
-      value: constructedMsg.id
+      value: encodeResult.value.id
     };
   }
   _handleStateExit() {
@@ -330,23 +352,6 @@ var IdentifiedSessionWithGracePeriod = class extends IdentifiedSession {
     super._handleClose();
   }
 };
-function sendMessage(conn, codec, msg) {
-  const buff = codec.toBuffer(msg);
-  if (!buff.ok) {
-    return buff;
-  }
-  const sent = conn.send(buff.value);
-  if (!sent) {
-    return {
-      ok: false,
-      reason: "failed to send message"
-    };
-  }
-  return {
-    ok: true,
-    value: msg.id
-  };
-}
 
 // transport/sessionStateMachine/SessionConnecting.ts
 var SessionConnecting = class extends IdentifiedSessionWithGracePeriod {
@@ -449,7 +454,21 @@ var SessionWaitingForHandshake = class extends CommonSession {
     this.listeners.onHandshake(parsedMsgRes.value);
   };
   sendHandshake(msg) {
-    return sendMessage(this.conn, this.codec, msg);
+    const buff = this.codec.toBuffer(msg);
+    if (!buff.ok) {
+      return buff;
+    }
+    const sent = this.conn.send(buff.value);
+    if (!sent) {
+      return {
+        ok: false,
+        reason: "failed to send handshake"
+      };
+    }
+    return {
+      ok: true,
+      value: msg.id
+    };
   }
   _handleStateExit() {
     this.conn.removeDataListener();
@@ -498,7 +517,21 @@ var SessionHandshaking = class extends IdentifiedSessionWithGracePeriod {
     this.listeners.onHandshake(parsedMsgRes.value);
   };
   sendHandshake(msg) {
-    return sendMessage(this.conn, this.codec, msg);
+    const buff = this.codec.toBuffer(msg);
+    if (!buff.ok) {
+      return buff;
+    }
+    const sent = this.conn.send(buff.value);
+    if (!sent) {
+      return {
+        ok: false,
+        reason: "failed to send handshake"
+      };
+    }
+    return {
+      ok: true,
+      value: msg.id
+    };
   }
   _handleStateExit() {
     super._handleStateExit();
@@ -533,28 +566,35 @@ var SessionConnected = class extends IdentifiedSession {
     }
     this.startMissingHeartbeatTimeout();
   }
-  assertSendOrdering(constructedMsg) {
-    if (constructedMsg.seq > this.seqSent + 1) {
-      const msg = `invariant violation: would have sent out of order msg (seq: ${constructedMsg.seq}, expected: ${this.seqSent} + 1)`;
+  assertSendOrdering(encodedMsg) {
+    if (encodedMsg.seq > this.seqSent + 1) {
+      const msg = `invariant violation: would have sent out of order msg (seq: ${encodedMsg.seq}, expected: ${this.seqSent} + 1)`;
       this.log?.error(msg, {
         ...this.loggingMetadata,
-        transportMessage: constructedMsg,
         tags: ["invariant-violation"]
       });
       throw new Error(msg);
     }
   }
   send(msg) {
-    const constructedMsg = this.constructMsg(msg);
-    this.assertSendOrdering(constructedMsg);
-    this.sendBuffer.push(constructedMsg);
-    const res = sendMessage(this.conn, this.codec, constructedMsg);
-    if (!res.ok) {
-      this.listeners.onMessageSendFailure(constructedMsg, res.reason);
-      return res;
+    const encodeResult = this.encodeMsg(msg);
+    if (!encodeResult.ok) {
+      return encodeResult;
+    }
+    const encodedMsg = encodeResult.value;
+    this.assertSendOrdering(encodedMsg);
+    this.sendBuffer.push(encodedMsg);
+    const sent = this.conn.send(encodedMsg.data);
+    if (!sent) {
+      const reason = "failed to send message";
+      this.listeners.onMessageSendFailure(
+        { ...encodedMsg.msg, seq: encodedMsg.seq },
+        reason
+      );
+      return { ok: false, reason };
     }
-    this.seqSent = constructedMsg.seq;
-    return res;
+    this.seqSent = encodedMsg.seq;
+    return { ok: true, value: encodedMsg.id };
   }
   constructor(props) {
     super(props);
@@ -572,10 +612,14 @@ var SessionConnected = class extends IdentifiedSession {
       );
       for (const msg of this.sendBuffer) {
         this.assertSendOrdering(msg);
-        const res = sendMessage(this.conn, this.codec, msg);
-        if (!res.ok) {
-          this.listeners.onMessageSendFailure(msg, res.reason);
-          return res;
+        const sent = this.conn.send(msg.data);
+        if (!sent) {
+          const reason = "failed to send buffered message";
+          this.listeners.onMessageSendFailure(
+            { ...msg.msg, seq: msg.seq },
+            reason
+          );
+          return { ok: false, reason };
         }
         this.seqSent = msg.seq;
       }
@@ -1278,6 +1322,17 @@ var Transport = class {
     const noConnectionSession = SessionStateGraph.transition.ConnectingToNoConnection(session, {
       onSessionGracePeriodElapsed: () => {
         this.onSessionGracePeriodElapsed(noConnectionSession);
+      },
+      onMessageSendFailure: (msg, reason) => {
+        this.log?.error(`failed to send message: ${reason}`, {
+          ...noConnectionSession.loggingMetadata,
+          transportMessage: msg
+        });
+        this.protocolError({
+          type: ProtocolError.MessageSendFailure,
+          message: reason
+        });
+        this.deleteSession(noConnectionSession, { unhealthy: true });
       }
     });
     this.updateSession(noConnectionSession);
@@ -1285,18 +1340,32 @@ var Transport = class {
   }
   onConnClosed(session) {
     let noConnectionSession;
+    const listeners = {
+      onSessionGracePeriodElapsed: () => {
+        this.onSessionGracePeriodElapsed(noConnectionSession);
+      },
+      onMessageSendFailure: (msg, reason) => {
+        this.log?.error(`failed to send message: ${reason}`, {
+          ...noConnectionSession.loggingMetadata,
+          transportMessage: msg
+        });
+        this.protocolError({
+          type: ProtocolError.MessageSendFailure,
+          message: reason
+        });
+        this.deleteSession(noConnectionSession, { unhealthy: true });
+      }
+    };
     if (session.state === "Handshaking" /* Handshaking */) {
-      noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(session, {
-        onSessionGracePeriodElapsed: () => {
-          this.onSessionGracePeriodElapsed(noConnectionSession);
-        }
-      });
+      noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(
+        session,
+        listeners
+      );
     } else {
-      noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(session, {
-        onSessionGracePeriodElapsed: () => {
-          this.onSessionGracePeriodElapsed(noConnectionSession);
-        }
-      });
+      noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(
+        session,
+        listeners
+      );
     }
     this.updateSession(noConnectionSession);
     return noConnectionSession;
@@ -1457,6 +1526,17 @@ var ClientTransport = class extends Transport {
       {
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(session);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...session.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(session, { unhealthy: true });
         }
       },
       this.options,
@@ -1521,6 +1601,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(handshakingSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...handshakingSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(handshakingSession, { unhealthy: true });
         }
       }
     );
@@ -1682,6 +1773,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(backingOffSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...backingOffSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(backingOffSession, { unhealthy: true });
         }
       }
     );
@@ -1745,6 +1847,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(connectingSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...connectingSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(connectingSession, { unhealthy: true });
         }
       }
     );
@@ -2075,6 +2188,17 @@ var ServerTransport = class extends Transport {
           {
             onSessionGracePeriodElapsed: () => {
               this.onSessionGracePeriodElapsed(noConnectionSession);
+            },
+            onMessageSendFailure: (msg2, reason) => {
+              this.log?.error(`failed to send message: ${reason}`, {
+                ...noConnectionSession.loggingMetadata,
+                transportMessage: msg2
+              });
+              this.protocolError({
+                type: ProtocolError.MessageSendFailure,
+                message: reason
+              });
+              this.deleteSession(noConnectionSession, { unhealthy: true });
             }
           }
         );
@@ -2345,4 +2469,4 @@ export {
   WebSocketConnection,
   CodecMessageAdapter
 };
-//# sourceMappingURL=chunk-DRYCLL6N.js.map
+//# sourceMappingURL=chunk-VXRAFUA3.js.map