@replit/river 0.12.6 → 0.13.1

package/dist/{chunk-4SDJ5VN4.js → chunk-LDUFHGZU.js}

@@ -1,19 +1,19 @@
-import {
-  NaiveJsonCodec
-} from "./chunk-GZ7HCLLM.js";
 import {
   ControlMessageHandshakeRequestSchema,
   ControlMessageHandshakeResponseSchema,
   OpaqueTransportMessageSchema,
   PROTOCOL_VERSION,
-  bootRequestMessage,
-  bootResponseMessage,
   coerceErrorString,
+  handshakeRequestMessage,
+  handshakeResponseMessage,
   isAck
-} from "./chunk-XFFS4UOD.js";
+} from "./chunk-SCG5S2EC.js";
 import {
   log
 } from "./chunk-H4BYJELI.js";
+import {
+  NaiveJsonCodec
+} from "./chunk-GZ7HCLLM.js";
 
 // transport/events.ts
 var EventDispatcher = class {
@@ -56,8 +56,15 @@ var Connection = class {
 var HEARTBEAT_INTERVAL_MS = 1e3;
 var HEARTBEATS_TILL_DEAD = 2;
 var SESSION_DISCONNECT_GRACE_MS = 5e3;
+var defaultSessionOptions = {
+  heartbeatIntervalMs: HEARTBEAT_INTERVAL_MS,
+  heartbeatsUntilDead: HEARTBEATS_TILL_DEAD,
+  sessionDisconnectGraceMs: SESSION_DISCONNECT_GRACE_MS,
+  codec: NaiveJsonCodec
+};
 var Session = class {
   codec;
+  options;
   /**
    * The buffer of messages that have been sent but not yet acknowledged.
    */
@@ -93,16 +100,17 @@ var Session = class {
    * The interval for sending heartbeats.
    */
   heartbeat;
-  constructor(codec, from, connectedTo, conn) {
+  constructor(from, connectedTo, conn, options) {
+    this.options = options;
     this.debugId = `sess-${unsafeId()}`;
     this.from = from;
     this.to = connectedTo;
     this.connection = conn;
-    this.codec = codec;
+    this.codec = options.codec;
     this.heartbeatMisses = 0;
     this.heartbeat = setInterval(
       () => this.sendHeartbeat(),
-      HEARTBEAT_INTERVAL_MS
+      options.heartbeatIntervalMs
     );
   }
   /**
@@ -138,7 +146,7 @@ var Session = class {
     return fullMsg.id;
   }
   sendHeartbeat() {
-    if (this.heartbeatMisses >= HEARTBEATS_TILL_DEAD) {
+    if (this.heartbeatMisses >= this.options.heartbeatsUntilDead) {
       if (this.connection) {
         log?.info(
           `${this.from} -- closing connection (id: ${this.connection.debugId}) to ${this.to} due to inactivity`
@@ -208,7 +216,7 @@ var Session = class {
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
-    }, SESSION_DISCONNECT_GRACE_MS);
+    }, this.options.sessionDisconnectGraceMs);
   }
   // called on reconnect of the underlying session
   cancelGrace() {
@@ -249,13 +257,13 @@ var Session = class {
 // transport/transport.ts
 import { Value } from "@sinclair/typebox/value";
 import { nanoid as nanoid2 } from "nanoid";
-var DEFAULT_RECONNECT_JITTER_MAX_MS = 500;
-var DEFAULT_RECONNECT_INTERVAL_MS = 250;
+var RECONNECT_JITTER_MAX_MS = 500;
+var RECONNECT_INTERVAL_MS = 250;
 var defaultTransportOptions = {
-  retryIntervalMs: DEFAULT_RECONNECT_INTERVAL_MS,
-  retryJitterMs: DEFAULT_RECONNECT_JITTER_MAX_MS,
+  retryIntervalMs: RECONNECT_INTERVAL_MS,
+  retryJitterMs: RECONNECT_JITTER_MAX_MS,
   retryAttemptsMax: 5,
-  codec: NaiveJsonCodec
+  ...defaultSessionOptions
 };
 var Transport = class {
   /**
@@ -326,7 +334,7 @@ var Transport = class {
     const lastInstanceId = this.connectedInstanceIds.get(connectedTo);
     if (oldSession && lastInstanceId !== void 0 && lastInstanceId !== instanceId) {
       log?.warn(
-        `${this.clientId} -- handshake from ${connectedTo} is a different instance (got: ${instanceId}, last connected to: ${lastInstanceId}), starting a new session`
+        `${this.clientId} -- connection from ${connectedTo} is a different instance (got: ${instanceId}, last connected to: ${lastInstanceId}), starting a new session`
       );
       oldSession.close();
       this.deleteSession(oldSession);
@@ -349,10 +357,10 @@ var Transport = class {
   }
   createSession(connectedTo, conn) {
     const session = new Session(
-      this.codec,
       this.clientId,
       connectedTo,
-      conn
+      conn,
+      this.options
     );
     this.sessions.set(session.to, session);
     this.eventDispatcher.dispatchEvent("sessionStatus", {
@@ -395,7 +403,7 @@ var Transport = class {
   parseMsg(msg) {
     const parsedMsg = this.codec.fromBuffer(msg);
     if (parsedMsg === null) {
-      const decodedBuffer = new TextDecoder().decode(msg);
+      const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));
       log?.error(
         `${this.clientId} -- received malformed msg, killing conn: ${decodedBuffer}`
       );
@@ -409,11 +417,7 @@ var Transport = class {
       );
       return null;
     }
-    return {
-      ...parsedMsg,
-      serviceName: parsedMsg.serviceName === null ? void 0 : parsedMsg.serviceName,
-      procedureName: parsedMsg.procedureName === null ? void 0 : parsedMsg.procedureName
-    };
+    return parsedMsg;
   }
   /**
    * Called when a message is received by this transport.
@@ -432,11 +436,18 @@ var Transport = class {
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {
-      log?.warn(
-        `${this.clientId} -- received out-of-order msg (got: ${msg.seq}, wanted: ${session.nextExpectedSeq}), discarding: ${JSON.stringify(
-          msg
-        )}`
-      );
+      if (msg.seq < session.nextExpectedSeq) {
+        log?.debug(
+          `${this.clientId} -- received duplicate msg (got: ${msg.seq}, wanted: ${session.nextExpectedSeq}), discarding: ${JSON.stringify(
+            msg
+          )}`
+        );
+      } else {
+        log?.error(
+          `${this.clientId} -- received out-of-order msg (got: ${msg.seq}, wanted: ${session.nextExpectedSeq}), marking connection as dead: ${JSON.stringify(msg)}`
+        );
+        session.closeStaleConnection(session.connection);
+      }
       return;
     }
     if (!isAck(msg.controlFlags)) {
@@ -454,7 +465,7 @@ var Transport = class {
   }
   /**
    * Removes a listener from this transport.
-   * @param the type of event to unlisten on
+   * @param the type of event to un-listen on
    * @param handler The message handler to remove.
    */
   removeEventListener(type, handler) {
@@ -531,23 +542,31 @@ var ClientTransport = class extends Transport {
    */
   inflightConnectionPromises;
   tryReconnecting = true;
-  constructor(clientId, providedOptions) {
+  connectedTo;
+  constructor(clientId, connectedTo, providedOptions) {
     super(clientId, providedOptions);
+    this.connectedTo = connectedTo;
     this.inflightConnectionPromises = /* @__PURE__ */ new Map();
   }
   handleConnection(conn, to) {
-    const bootHandler = this.receiveWithBootSequence(conn, () => {
-      conn.removeDataListener(bootHandler);
-      conn.addDataListener((data) => {
-        const parsed = this.parseMsg(data);
+    const handshakeHandler = (data) => {
+      const handshake = this.receiveHandshakeResponseMessage(data);
+      if (!handshake) {
+        conn.close();
+        return;
+      }
+      this.onConnect(conn, handshake.from, handshake.instanceId);
+      conn.removeDataListener(handshakeHandler);
+      conn.addDataListener((data2) => {
+        const parsed = this.parseMsg(data2);
         if (!parsed) {
           conn.close();
           return;
         }
         this.handleMsg(parsed);
       });
-    });
-    conn.addDataListener(bootHandler);
+    };
+    conn.addDataListener(handshakeHandler);
     conn.addCloseListener(() => {
       this.onDisconnect(conn, to);
       void this.connect(to);
@@ -558,6 +577,32 @@ var ClientTransport = class extends Transport {
       );
     });
   }
+  receiveHandshakeResponseMessage(data) {
+    const parsed = this.parseMsg(data);
+    if (!parsed)
+      return false;
+    if (!Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {
+      log?.warn(
+        `${this.clientId} -- received invalid handshake resp: ${JSON.stringify(
+          parsed
+        )}`
+      );
+      return false;
+    }
+    if (!parsed.payload.status.ok) {
+      log?.warn(
+        `${this.clientId} -- received failed handshake resp: ${JSON.stringify(
+          parsed
+        )}`
+      );
+      return false;
+    }
+    const instanceId = parsed.payload.status.instanceId;
+    log?.debug(
+      `${this.clientId} -- handshake from ${parsed.from} ok (instance: ${instanceId})`
+    );
+    return { instanceId, from: parsed.from };
+  }
   /**
    * Manually attempts to connect to a client.
    * @param to The client ID of the node to connect to.
@@ -576,10 +621,7 @@ var ClientTransport = class extends Transport {
     }
     try {
       const conn = await reconnectPromise;
-      this.state = "open";
-      const requestMsg = bootRequestMessage(this.clientId, to, this.instanceId);
-      log?.debug(`${this.clientId} -- sending boot handshake to ${to}`);
-      conn.send(this.codec.toBuffer(requestMsg));
+      this.sendHandshake(to, conn);
     } catch (error) {
       const errStr = coerceErrorString(error);
       this.inflightConnectionPromises.delete(to);
@@ -597,34 +639,14 @@ var ClientTransport = class extends Transport {
       }
     }
   }
-  receiveWithBootSequence(conn, sessionCb) {
-    const bootHandler = (data) => {
-      const parsed = this.parseMsg(data);
-      if (!parsed) {
-        conn.close();
-        return;
-      }
-      if (!Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {
-        log?.warn(
-          `${this.clientId} -- received invalid handshake resp: ${JSON.stringify(parsed)}`
-        );
-        return;
-      }
-      if (!parsed.payload.status.ok) {
-        log?.warn(
-          `${this.clientId} -- received failed handshake resp: ${JSON.stringify(
-            parsed
-          )}`
-        );
-        return;
-      }
-      const serverInstanceId = parsed.payload.status.instanceId;
-      log?.debug(
-        `${this.clientId} -- handshake from ${parsed.from} ok (server instance: ${serverInstanceId})`
-      );
-      sessionCb(this.onConnect(conn, parsed.from, serverInstanceId));
-    };
-    return bootHandler;
+  sendHandshake(to, conn) {
+    const requestMsg = handshakeRequestMessage(
+      this.clientId,
+      to,
+      this.instanceId
+    );
+    log?.debug(`${this.clientId} -- sending handshake request to ${to}`);
+    conn.send(this.codec.toBuffer(requestMsg));
   }
   onDisconnect(conn, connectedTo) {
     this.inflightConnectionPromises.delete(connectedTo);
@@ -641,22 +663,24 @@ var ServerTransport = class extends Transport {
   handleConnection(conn) {
     let session = void 0;
     const client = () => session?.to ?? "unknown";
-    const bootHandler = this.receiveWithBootSequence(
-      conn,
-      (establishedSession) => {
-        session = establishedSession;
-        conn.removeDataListener(bootHandler);
-        conn.addDataListener((data) => {
-          const parsed = this.parseMsg(data);
-          if (!parsed) {
-            conn.close();
-            return;
-          }
-          this.handleMsg(parsed);
-        });
+    const handshakeHandler = (data) => {
+      const handshake = this.receiveHandshakeRequestMessage(data, conn);
+      if (!handshake) {
+        conn.close();
+        return;
       }
-    );
-    conn.addDataListener(bootHandler);
+      session = this.onConnect(conn, handshake.from, handshake.instanceId);
+      conn.removeDataListener(handshakeHandler);
+      conn.addDataListener((data2) => {
+        const parsed = this.parseMsg(data2);
+        if (!parsed) {
+          conn.close();
+          return;
+        }
+        this.handleMsg(parsed);
+      });
+    };
+    conn.addDataListener(handshakeHandler);
     conn.addCloseListener(() => {
       if (!session)
         return;
@@ -673,47 +697,57 @@ var ServerTransport = class extends Transport {
       );
     });
   }
-  receiveWithBootSequence(conn, sessionCb) {
-    const bootHandler = (data) => {
-      const parsed = this.parseMsg(data);
-      if (!parsed) {
-        conn.close();
-        return;
-      }
-      if (!Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {
-        const responseMsg2 = bootResponseMessage(
-          this.clientId,
-          this.instanceId,
-          parsed.from,
-          false
-        );
-        conn.send(this.codec.toBuffer(responseMsg2));
-        log?.warn(
-          `${this.clientId} -- received invalid handshake msg: ${JSON.stringify(
-            parsed
-          )}`
-        );
-        return;
-      }
-      const instanceId = parsed.payload.instanceId;
-      log?.debug(
-        `${this.clientId} -- handshake from ${parsed.from} ok (instance id: ${instanceId}), responding with handshake success`
+  receiveHandshakeRequestMessage(data, conn) {
+    const parsed = this.parseMsg(data);
+    if (!parsed)
+      return false;
+    if (!Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {
+      const responseMsg2 = handshakeResponseMessage(
+        this.clientId,
+        this.instanceId,
+        parsed.from,
+        false
+      );
+      conn.send(this.codec.toBuffer(responseMsg2));
+      log?.warn(
+        `${this.clientId} -- received invalid handshake msg: ${JSON.stringify(
+          parsed
+        )}`
       );
-      const responseMsg = bootResponseMessage(
+      return false;
+    }
+    const gotVersion = parsed.payload.protocolVersion;
+    if (gotVersion !== PROTOCOL_VERSION) {
+      const responseMsg2 = handshakeResponseMessage(
         this.clientId,
         this.instanceId,
         parsed.from,
-        true
+        false
       );
-      conn.send(this.codec.toBuffer(responseMsg));
-      sessionCb(this.onConnect(conn, parsed.from, instanceId));
-    };
-    return bootHandler;
+      conn.send(this.codec.toBuffer(responseMsg2));
+      log?.warn(
+        `${this.clientId} -- received handshake msg with incompatible protocol version (got: ${gotVersion}, expected: ${PROTOCOL_VERSION})`
+      );
+      return false;
+    }
+    const instanceId = parsed.payload.instanceId;
+    log?.debug(
+      `${this.clientId} -- handshake from ${parsed.from} ok (instance id: ${instanceId}), responding with handshake success`
+    );
+    const responseMsg = handshakeResponseMessage(
+      this.clientId,
+      this.instanceId,
+      parsed.from,
+      true
+    );
+    conn.send(this.codec.toBuffer(responseMsg));
+    return { instanceId, from: parsed.from };
   }
 };
 
 export {
   Connection,
+  defaultSessionOptions,
   Session,
   Transport,
   ClientTransport,

package/dist/{chunk-VLBVQX5H.js → chunk-Q7AWJYDQ.js}

@@ -1,6 +1,6 @@
 import {
   Connection
-} from "./chunk-4SDJ5VN4.js";
+} from "./chunk-LDUFHGZU.js";
 
 // transport/impls/ws/connection.ts
 var WebSocketConnection = class extends Connection {

package/dist/{chunk-XFFS4UOD.js → chunk-SCG5S2EC.js}

@@ -7,8 +7,8 @@ var TransportMessageSchema = (t) => Type.Object({
   to: Type.String(),
   seq: Type.Integer(),
   ack: Type.Integer(),
-  serviceName: Type.Optional(Type.Union([Type.String(), Type.Null()])),
-  procedureName: Type.Optional(Type.Union([Type.String(), Type.Null()])),
+  serviceName: Type.Optional(Type.String()),
+  procedureName: Type.Optional(Type.String()),
   streamId: Type.String(),
   controlFlags: Type.Integer(),
   payload: t
@@ -22,7 +22,7 @@ var ControlMessageCloseSchema = Type.Object({
 var PROTOCOL_VERSION = "v1";
 var ControlMessageHandshakeRequestSchema = Type.Object({
   type: Type.Literal("HANDSHAKE_REQ"),
-  protocolVersion: Type.Literal(PROTOCOL_VERSION),
+  protocolVersion: Type.String(),
   instanceId: Type.String()
 });
 var ControlMessageHandshakeResponseSchema = Type.Object({
@@ -34,7 +34,7 @@ var ControlMessageHandshakeResponseSchema = Type.Object({
     }),
     Type.Object({
       ok: Type.Literal(false),
-      reason: Type.Union([Type.Literal("VERSION_MISMATCH")])
+      reason: Type.String()
     })
   ])
 });
@@ -47,7 +47,7 @@ var ControlMessagePayloadSchema = Type.Union([
 var OpaqueTransportMessageSchema = TransportMessageSchema(
   Type.Unknown()
 );
-function bootRequestMessage(from, to, instanceId) {
+function handshakeRequestMessage(from, to, instanceId) {
   return {
     id: nanoid(),
     from,
@@ -63,7 +63,7 @@ function bootRequestMessage(from, to, instanceId) {
     }
   };
 }
-function bootResponseMessage(from, instanceId, to, ok) {
+function handshakeResponseMessage(from, instanceId, to, ok, reason) {
   return {
     id: nanoid(),
     from,
@@ -82,7 +82,7 @@ function bootResponseMessage(from, instanceId, to, ok) {
       type: "HANDSHAKE_RESP",
       status: {
         ok: false,
-        reason: "VERSION_MISMATCH"
+        reason: reason ?? "Unknown reason"
       }
     }
   };
@@ -118,8 +118,8 @@ export {
   ControlMessageHandshakeResponseSchema,
   ControlMessagePayloadSchema,
   OpaqueTransportMessageSchema,
-  bootRequestMessage,
-  bootResponseMessage,
+  handshakeRequestMessage,
+  handshakeResponseMessage,
   isAck,
   isStreamOpen,
   isStreamClose,

package/dist/{messageFraming-b200ef25.d.ts → connection-03e650c8.d.ts}

@@ -1,4 +1,6 @@
-import { Transform, TransformCallback, TransformOptions } from 'node:stream';
+import { C as Connection } from './index-9aa0aabb.js';
+import { Socket } from 'node:net';
+import stream, { Transform, TransformCallback, TransformOptions } from 'node:stream';
 
 interface LengthEncodedOptions extends TransformOptions {
     /** Maximum in-memory buffer size before we throw */
@@ -17,4 +19,17 @@ declare class Uint32LengthPrefixFraming extends Transform {
     _destroy(error: Error | null, callback: (error: Error | null) => void): void;
 }
 
-export { Uint32LengthPrefixFraming as U };
+declare class UdsConnection extends Connection {
+    sock: Socket;
+    input: stream.Readable;
+    framer: Uint32LengthPrefixFraming;
+    constructor(sock: Socket);
+    addDataListener(cb: (msg: Uint8Array) => void): void;
+    removeDataListener(cb: (msg: Uint8Array) => void): void;
+    addCloseListener(cb: () => void): void;
+    addErrorListener(cb: (err: Error) => void): void;
+    send(payload: Uint8Array): boolean;
+    close(): void;
+}
+
+export { UdsConnection as U };

package/dist/{connection-ba37d174.d.ts → connection-d052d027.d.ts}

@@ -1,5 +1,5 @@
 import WebSocket from 'isomorphic-ws';
-import { C as Connection } from './index-54e0f99c.js';
+import { C as Connection } from './index-9aa0aabb.js';
 
 declare class WebSocketConnection extends Connection {
     ws: WebSocket;