@replit/river 0.12.1 → 0.12.3

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

@@ -235,7 +235,7 @@ var Session = class {
         log?.info(
           `${this.from} -- closing connection (id: ${this.connection.debugId}) to ${this.to} due to inactivity`
         );
-        this.halfCloseConnection();
+        this.closeStaleConnection(this.connection);
       }
       return;
     }
@@ -296,17 +296,27 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
+  graceCb;
   beginGrace(cb) {
+    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
-      this.resetBufferedMessages();
-      clearInterval(this.heartbeat);
+      this.close();
       cb();
     }, SESSION_DISCONNECT_GRACE_MS);
   }
+  // called on reconnect of the underlying session
   cancelGrace() {
     this.heartbeatMisses = 0;
     clearTimeout(this.disconnectionGrace);
   }
+  // closed when we want to discard the whole session
+  // (i.e. shutdown or session disconnect)
+  close() {
+    this.closeStaleConnection(this.connection);
+    this.cancelGrace();
+    clearInterval(this.heartbeat);
+    this.resetBufferedMessages();
+  }
   get connected() {
     return this.connection !== void 0;
   }
@@ -325,15 +335,6 @@ var Session = class {
     this.seq++;
     return msg;
   }
-  /**
-   * Closes the out-going connection but doesn't remove the listeners
-   * for incoming messages. The connection will eventually call onClose
-   * when it is ready to be cleaned up and only then will {@link connection} be set back
-   * to undefined
-   */
-  halfCloseConnection() {
-    this.connection?.close();
-  }
   inspectSendBuffer() {
     return this.sendBuffer;
   }
@@ -485,19 +486,18 @@ var Transport = class {
       status: "connect",
       conn
     });
-    let session = this.sessions.get(connectedTo);
+    let oldSession = this.sessions.get(connectedTo);
     const lastInstanceId = this.connectedInstanceIds.get(connectedTo);
-    if (session && lastInstanceId !== instanceId && lastInstanceId !== void 0) {
-      log?.debug(
-        `${this.clientId} -- handshake from ${connectedTo} has different server instance (got: ${instanceId}, last connected to: ${lastInstanceId}), starting a new session`
+    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`
       );
-      session.resetBufferedMessages();
-      session.closeStaleConnection();
-      this.deleteSession(session);
-      session = void 0;
+      oldSession.close();
+      this.deleteSession(oldSession);
+      oldSession = void 0;
     }
     this.connectedInstanceIds.set(connectedTo, instanceId);
-    if (session === void 0) {
+    if (oldSession === void 0) {
       const newSession = this.createSession(connectedTo, conn);
       log?.info(
         `${this.clientId} -- new connection (id: ${conn.debugId}) for new session (id: ${newSession.debugId}) to ${connectedTo}`
@@ -505,11 +505,11 @@ var Transport = class {
       return newSession;
     }
     log?.info(
-      `${this.clientId} -- new connection (id: ${conn.debugId}) for existing session (id: ${session.debugId}) to ${connectedTo}`
+      `${this.clientId} -- new connection (id: ${conn.debugId}) for existing session (id: ${oldSession.debugId}) to ${connectedTo}`
     );
-    session.replaceWithNewConnection(conn);
-    session.sendBufferedMessages();
-    return session;
+    oldSession.replaceWithNewConnection(conn);
+    oldSession.sendBufferedMessages();
+    return oldSession;
   }
   createSession(connectedTo, conn) {
     const session = new Session(
@@ -527,24 +527,25 @@ var Transport = class {
   }
   deleteSession(session) {
     this.sessions.delete(session.to);
+    log?.info(
+      `${this.clientId} -- session ${session.debugId} disconnect from ${session.to}`
+    );
     this.eventDispatcher.dispatchEvent("sessionStatus", {
       status: "disconnect",
       session
     });
-    log?.info(
-      `${this.clientId} -- session ${session.debugId} disconnect from ${session.to}`
-    );
   }
   /**
    * The downstream implementation needs to call this when a connection is closed.
    * @param conn The connection object.
+   * @param connectedTo The peer we are connected to.
    */
   onDisconnect(conn, connectedTo) {
     this.eventDispatcher.dispatchEvent("connectionStatus", {
       status: "disconnect",
       conn
     });
-    if (!connectedTo)
+    if (this.state !== "open")
       return;
     const session = this.sessionByClientId(connectedTo);
     log?.info(
@@ -587,6 +588,8 @@ var Transport = class {
    * @param msg The received message.
    */
   handleMsg(msg) {
+    if (this.state !== "open")
+      return;
     const session = this.sessionByClientId(msg.from);
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
@@ -663,10 +666,11 @@ var Transport = class {
    * Closes the transport. Any messages sent while the transport is closed will be silently discarded.
    */
   close() {
+    this.state = "closed";
     for (const session of this.sessions.values()) {
-      session.halfCloseConnection();
+      session.close();
+      this.deleteSession(session);
     }
-    this.state = "closed";
     log?.info(`${this.clientId} -- manually closed transport`);
   }
   /**
@@ -675,10 +679,11 @@ var Transport = class {
    * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.
    */
   destroy() {
+    this.state = "destroyed";
     for (const session of this.sessions.values()) {
-      session.closeStaleConnection(session.connection);
+      session.close();
+      this.deleteSession(session);
     }
-    this.state = "destroyed";
     log?.info(`${this.clientId} -- manually destroyed transport`);
   }
 };

package/dist/transport/impls/stdio/server.d.cts

@@ -1,5 +1,5 @@
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-0c0a69f6.js';
-import { S as StreamConnection } from '../../../connection-bf7811aa.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { S as StreamConnection } from '../../../connection-64b9e0ff.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/stdio/server.d.ts

@@ -1,5 +1,5 @@
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-0c0a69f6.js';
-import { S as StreamConnection } from '../../../connection-bf7811aa.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { S as StreamConnection } from '../../../connection-64b9e0ff.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/stdio/server.js

@@ -1,11 +1,11 @@
 import {
   StreamConnection
-} from "../../../chunk-M6LY25P2.js";
+} from "../../../chunk-H4QR2K3T.js";
 import "../../../chunk-5IZ2UHWV.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ServerTransport
-} from "../../../chunk-QEYN2Z6O.js";
+} from "../../../chunk-2PKONGC3.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import "../../../chunk-H4BYJELI.js";

package/dist/transport/impls/uds/client.cjs

@@ -199,7 +199,7 @@ var Session = class {
         log?.info(
           `${this.from} -- closing connection (id: ${this.connection.debugId}) to ${this.to} due to inactivity`
         );
-        this.halfCloseConnection();
+        this.closeStaleConnection(this.connection);
       }
       return;
     }
@@ -260,17 +260,27 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
+  graceCb;
   beginGrace(cb) {
+    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
-      this.resetBufferedMessages();
-      clearInterval(this.heartbeat);
+      this.close();
       cb();
     }, SESSION_DISCONNECT_GRACE_MS);
   }
+  // called on reconnect of the underlying session
   cancelGrace() {
     this.heartbeatMisses = 0;
     clearTimeout(this.disconnectionGrace);
   }
+  // closed when we want to discard the whole session
+  // (i.e. shutdown or session disconnect)
+  close() {
+    this.closeStaleConnection(this.connection);
+    this.cancelGrace();
+    clearInterval(this.heartbeat);
+    this.resetBufferedMessages();
+  }
   get connected() {
     return this.connection !== void 0;
   }
@@ -289,15 +299,6 @@ var Session = class {
     this.seq++;
     return msg;
   }
-  /**
-   * Closes the out-going connection but doesn't remove the listeners
-   * for incoming messages. The connection will eventually call onClose
-   * when it is ready to be cleaned up and only then will {@link connection} be set back
-   * to undefined
-   */
-  halfCloseConnection() {
-    this.connection?.close();
-  }
   inspectSendBuffer() {
     return this.sendBuffer;
   }
@@ -573,19 +574,18 @@ var Transport = class {
       status: "connect",
       conn
     });
-    let session = this.sessions.get(connectedTo);
+    let oldSession = this.sessions.get(connectedTo);
     const lastInstanceId = this.connectedInstanceIds.get(connectedTo);
-    if (session && lastInstanceId !== instanceId && lastInstanceId !== void 0) {
-      log?.debug(
-        `${this.clientId} -- handshake from ${connectedTo} has different server instance (got: ${instanceId}, last connected to: ${lastInstanceId}), starting a new session`
+    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`
       );
-      session.resetBufferedMessages();
-      session.closeStaleConnection();
-      this.deleteSession(session);
-      session = void 0;
+      oldSession.close();
+      this.deleteSession(oldSession);
+      oldSession = void 0;
     }
     this.connectedInstanceIds.set(connectedTo, instanceId);
-    if (session === void 0) {
+    if (oldSession === void 0) {
       const newSession = this.createSession(connectedTo, conn);
       log?.info(
         `${this.clientId} -- new connection (id: ${conn.debugId}) for new session (id: ${newSession.debugId}) to ${connectedTo}`
@@ -593,11 +593,11 @@ var Transport = class {
       return newSession;
     }
     log?.info(
-      `${this.clientId} -- new connection (id: ${conn.debugId}) for existing session (id: ${session.debugId}) to ${connectedTo}`
+      `${this.clientId} -- new connection (id: ${conn.debugId}) for existing session (id: ${oldSession.debugId}) to ${connectedTo}`
     );
-    session.replaceWithNewConnection(conn);
-    session.sendBufferedMessages();
-    return session;
+    oldSession.replaceWithNewConnection(conn);
+    oldSession.sendBufferedMessages();
+    return oldSession;
   }
   createSession(connectedTo, conn) {
     const session = new Session(
@@ -615,24 +615,25 @@ var Transport = class {
   }
   deleteSession(session) {
     this.sessions.delete(session.to);
+    log?.info(
+      `${this.clientId} -- session ${session.debugId} disconnect from ${session.to}`
+    );
     this.eventDispatcher.dispatchEvent("sessionStatus", {
       status: "disconnect",
       session
     });
-    log?.info(
-      `${this.clientId} -- session ${session.debugId} disconnect from ${session.to}`
-    );
   }
   /**
    * The downstream implementation needs to call this when a connection is closed.
    * @param conn The connection object.
+   * @param connectedTo The peer we are connected to.
    */
   onDisconnect(conn, connectedTo) {
     this.eventDispatcher.dispatchEvent("connectionStatus", {
       status: "disconnect",
       conn
     });
-    if (!connectedTo)
+    if (this.state !== "open")
       return;
     const session = this.sessionByClientId(connectedTo);
     log?.info(
@@ -675,6 +676,8 @@ var Transport = class {
    * @param msg The received message.
    */
   handleMsg(msg) {
+    if (this.state !== "open")
+      return;
     const session = this.sessionByClientId(msg.from);
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
@@ -751,10 +754,11 @@ var Transport = class {
    * Closes the transport. Any messages sent while the transport is closed will be silently discarded.
    */
   close() {
+    this.state = "closed";
     for (const session of this.sessions.values()) {
-      session.halfCloseConnection();
+      session.close();
+      this.deleteSession(session);
     }
-    this.state = "closed";
     log?.info(`${this.clientId} -- manually closed transport`);
   }
   /**
@@ -763,10 +767,11 @@ var Transport = class {
    * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.
    */
   destroy() {
+    this.state = "destroyed";
     for (const session of this.sessions.values()) {
-      session.closeStaleConnection(session.connection);
+      session.close();
+      this.deleteSession(session);
     }
-    this.state = "destroyed";
     log?.info(`${this.clientId} -- manually destroyed transport`);
   }
 };
@@ -872,8 +877,7 @@ var ClientTransport = class extends Transport {
     return bootHandler;
   }
   onDisconnect(conn, connectedTo) {
-    if (connectedTo)
-      this.inflightConnectionPromises.delete(connectedTo);
+    this.inflightConnectionPromises.delete(connectedTo);
     super.onDisconnect(conn, connectedTo);
   }
 };

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

@@ -1,5 +1,5 @@
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-0c0a69f6.js';
-import { U as UdsConnection } from '../../../connection-d880aa4a.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import 'node:net';

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

@@ -1,5 +1,5 @@
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-0c0a69f6.js';
-import { U as UdsConnection } from '../../../connection-d880aa4a.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import 'node:net';

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

@@ -1,10 +1,10 @@
 import {
   UdsConnection
-} from "../../../chunk-TKINU53F.js";
+} from "../../../chunk-7J66WOIC.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ClientTransport
-} from "../../../chunk-QEYN2Z6O.js";
+} from "../../../chunk-2PKONGC3.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {

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

@@ -235,7 +235,7 @@ var Session = class {
         log?.info(
           `${this.from} -- closing connection (id: ${this.connection.debugId}) to ${this.to} due to inactivity`
         );
-        this.halfCloseConnection();
+        this.closeStaleConnection(this.connection);
       }
       return;
     }
@@ -296,17 +296,27 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
+  graceCb;
   beginGrace(cb) {
+    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
-      this.resetBufferedMessages();
-      clearInterval(this.heartbeat);
+      this.close();
       cb();
     }, SESSION_DISCONNECT_GRACE_MS);
   }
+  // called on reconnect of the underlying session
   cancelGrace() {
     this.heartbeatMisses = 0;
     clearTimeout(this.disconnectionGrace);
   }
+  // closed when we want to discard the whole session
+  // (i.e. shutdown or session disconnect)
+  close() {
+    this.closeStaleConnection(this.connection);
+    this.cancelGrace();
+    clearInterval(this.heartbeat);
+    this.resetBufferedMessages();
+  }
   get connected() {
     return this.connection !== void 0;
   }
@@ -325,15 +335,6 @@ var Session = class {
     this.seq++;
     return msg;
   }
-  /**
-   * Closes the out-going connection but doesn't remove the listeners
-   * for incoming messages. The connection will eventually call onClose
-   * when it is ready to be cleaned up and only then will {@link connection} be set back
-   * to undefined
-   */
-  halfCloseConnection() {
-    this.connection?.close();
-  }
   inspectSendBuffer() {
     return this.sendBuffer;
   }
@@ -485,19 +486,18 @@ var Transport = class {
       status: "connect",
       conn
     });
-    let session = this.sessions.get(connectedTo);
+    let oldSession = this.sessions.get(connectedTo);
     const lastInstanceId = this.connectedInstanceIds.get(connectedTo);
-    if (session && lastInstanceId !== instanceId && lastInstanceId !== void 0) {
-      log?.debug(
-        `${this.clientId} -- handshake from ${connectedTo} has different server instance (got: ${instanceId}, last connected to: ${lastInstanceId}), starting a new session`
+    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`
       );
-      session.resetBufferedMessages();
-      session.closeStaleConnection();
-      this.deleteSession(session);
-      session = void 0;
+      oldSession.close();
+      this.deleteSession(oldSession);
+      oldSession = void 0;
     }
     this.connectedInstanceIds.set(connectedTo, instanceId);
-    if (session === void 0) {
+    if (oldSession === void 0) {
       const newSession = this.createSession(connectedTo, conn);
       log?.info(
         `${this.clientId} -- new connection (id: ${conn.debugId}) for new session (id: ${newSession.debugId}) to ${connectedTo}`
@@ -505,11 +505,11 @@ var Transport = class {
       return newSession;
     }
     log?.info(
-      `${this.clientId} -- new connection (id: ${conn.debugId}) for existing session (id: ${session.debugId}) to ${connectedTo}`
+      `${this.clientId} -- new connection (id: ${conn.debugId}) for existing session (id: ${oldSession.debugId}) to ${connectedTo}`
     );
-    session.replaceWithNewConnection(conn);
-    session.sendBufferedMessages();
-    return session;
+    oldSession.replaceWithNewConnection(conn);
+    oldSession.sendBufferedMessages();
+    return oldSession;
   }
   createSession(connectedTo, conn) {
     const session = new Session(
@@ -527,24 +527,25 @@ var Transport = class {
   }
   deleteSession(session) {
     this.sessions.delete(session.to);
+    log?.info(
+      `${this.clientId} -- session ${session.debugId} disconnect from ${session.to}`
+    );
     this.eventDispatcher.dispatchEvent("sessionStatus", {
       status: "disconnect",
       session
     });
-    log?.info(
-      `${this.clientId} -- session ${session.debugId} disconnect from ${session.to}`
-    );
   }
   /**
    * The downstream implementation needs to call this when a connection is closed.
    * @param conn The connection object.
+   * @param connectedTo The peer we are connected to.
    */
   onDisconnect(conn, connectedTo) {
     this.eventDispatcher.dispatchEvent("connectionStatus", {
       status: "disconnect",
       conn
     });
-    if (!connectedTo)
+    if (this.state !== "open")
       return;
     const session = this.sessionByClientId(connectedTo);
     log?.info(
@@ -587,6 +588,8 @@ var Transport = class {
    * @param msg The received message.
    */
   handleMsg(msg) {
+    if (this.state !== "open")
+      return;
     const session = this.sessionByClientId(msg.from);
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
@@ -663,10 +666,11 @@ var Transport = class {
    * Closes the transport. Any messages sent while the transport is closed will be silently discarded.
    */
   close() {
+    this.state = "closed";
     for (const session of this.sessions.values()) {
-      session.halfCloseConnection();
+      session.close();
+      this.deleteSession(session);
     }
-    this.state = "closed";
     log?.info(`${this.clientId} -- manually closed transport`);
   }
   /**
@@ -675,10 +679,11 @@ var Transport = class {
    * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.
    */
   destroy() {
+    this.state = "destroyed";
     for (const session of this.sessions.values()) {
-      session.closeStaleConnection(session.connection);
+      session.close();
+      this.deleteSession(session);
     }
-    this.state = "destroyed";
     log?.info(`${this.clientId} -- manually destroyed transport`);
   }
 };

package/dist/transport/impls/uds/server.d.cts

@@ -1,6 +1,6 @@
 import { Server, Socket } from 'node:net';
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-0c0a69f6.js';
-import { U as UdsConnection } from '../../../connection-d880aa4a.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/uds/server.d.ts

@@ -1,6 +1,6 @@
 import { Server, Socket } from 'node:net';
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-0c0a69f6.js';
-import { U as UdsConnection } from '../../../connection-d880aa4a.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/uds/server.js

@@ -1,10 +1,10 @@
 import {
   UdsConnection
-} from "../../../chunk-TKINU53F.js";
+} from "../../../chunk-7J66WOIC.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ServerTransport
-} from "../../../chunk-QEYN2Z6O.js";
+} from "../../../chunk-2PKONGC3.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {