@replit/river 0.201.0 → 0.203.0

package/dist/{server-48d90a77.d.ts → server-e46399f9.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection, w as CommonSession, d as SessionState, x as CommonSessionProps, T as Transport, y as ServerTransportOptions, r as ServerHandshakeOptions, q as ParsedMetadata, z as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './context-8d263a7f.js';
+import { C as Connection, w as CommonSession, d as SessionState, x as CommonSessionProps, T as Transport, y as ServerTransportOptions, r as ServerHandshakeOptions, q as ParsedMetadata, z as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './context-d6dd8a1a.js';
 import { P as ProtocolVersion, b as OpaqueTransportMessage, e as Tags, a as TransportMessage, H as HandshakeErrorResponseCodes, c as TransportClientId } from './message-3def9ded.js';
 import * as _sinclair_typebox_errors from '@sinclair/typebox/errors';
 import { Static } from '@sinclair/typebox';

package/dist/{services-e71ea921.d.ts → services-56cbea0d.d.ts}

@@ -1,7 +1,7 @@
 import { TObject, TString, TSchema, TUnion, TLiteral, TNever, Static } from '@sinclair/typebox';
-import { C as Connection, n as ClientHandshakeOptions, t as ProcedureHandlerContext, s as ServiceContext } from './context-8d263a7f.js';
+import { C as Connection, n as ClientHandshakeOptions, t as ProcedureHandlerContext, s as ServiceContext } from './context-d6dd8a1a.js';
 import { c as TransportClientId } from './message-3def9ded.js';
-import { C as ClientTransport } from './client-a32692b0.js';
+import { C as ClientTransport } from './client-5d2e41a3.js';
 
 /**
  * {@link UNCAUGHT_ERROR_CODE} is the code that is used when an error is thrown
@@ -504,6 +504,7 @@ declare const Procedure: {
 interface Service<State extends object, Procs extends ProcedureMap<State>> {
     readonly state: State;
     readonly procedures: Procs;
+    [Symbol.asyncDispose]: () => Promise<void>;
 }
 /**
  * Represents any {@link Service} object.
@@ -567,6 +568,10 @@ type ProcType<S extends AnyService, ProcName extends keyof S['procedures']> = S[
  * was created via the {@link Procedure} constructors.
  */
 type BrandedProcedureMap<State> = Record<string, Branded<AnyProcedure<State>>>;
+type MaybeDisposable<State extends object> = State & {
+    [Symbol.asyncDispose]?: () => Promise<void>;
+    [Symbol.dispose]?: () => void;
+};
 /**
  * The configuration for a service.
  */
@@ -574,7 +579,7 @@ interface ServiceConfiguration<State extends object> {
     /**
      * A factory function for creating a fresh state.
      */
-    initializeState: (extendedContext: ServiceContext) => State;
+    initializeState: (extendedContext: ServiceContext) => MaybeDisposable<State>;
 }
 interface SerializedProcedureSchemaProtocolv1 {
     init?: PayloadType;
@@ -636,7 +641,7 @@ declare class ServiceSchema<State extends object, Procedures extends ProcedureMa
     /**
      * Factory function for creating a fresh state.
      */
-    protected readonly initializeState: (extendedContext: ServiceContext) => State;
+    protected readonly initializeState: (extendedContext: ServiceContext) => MaybeDisposable<State>;
     /**
      * The procedures for this service.
      */

package/dist/testUtil/index.cjs

@@ -289,6 +289,14 @@ var StateMachineState = class {
    * and we've moved on to another state
    */
   _isConsumed;
+  /**
+   * Cleanup this state machine state and mark it as consumed.
+   * After calling close, it is an error to access any properties on the state.
+   * You should never need to call this as a consumer.
+   *
+   * If you're looking to close the session from the client,
+   * use `.hardDisconnect` on the client transport.
+   */
   close() {
     this._handleClose();
   }
@@ -343,9 +351,8 @@ var CommonSession = class extends StateMachineState {
   parseMsg(msg) {
     const parsedMsg = this.options.codec.fromBuffer(msg);
     if (parsedMsg === null) {
-      const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));
       this.log?.error(
-        `received malformed msg: ${decodedBuffer}`,
+        `received malformed msg: ${Buffer.from(msg).toString("base64")}`,
         this.loggingMetadata
       );
       return null;
@@ -525,7 +532,7 @@ var SessionNoConnection = class extends IdentifiedSessionWithGracePeriod {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.201.0";
+var version = "0.203.0";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -1757,6 +1764,16 @@ var ClientTransport = class extends Transport {
     );
     this.updateSession(backingOffSession);
   }
+  /**
+   * Manually kills all sessions to the server (including all pending state).
+   * This is useful for when you want to close all connections to a server
+   * and don't want to wait for the grace period to elapse.
+   */
+  hardDisconnect() {
+    for (const session of this.sessions.values()) {
+      this.deleteSession(session);
+    }
+  }
   onBackoffFinished(session) {
     const connPromise = tracing_default.startActiveSpan("connect", async (span) => {
       try {