rivetkit 2.0.7-rc.1 → 2.0.8

package/dist/tsup/inspector/mod.cjs

@@ -6,7 +6,7 @@
 
 
 
-var _chunkG4ABMAQYcjs = require('../chunk-G4ABMAQY.cjs');
+var _chunkQGUQB3NCcjs = require('../chunk-QGUQB3NC.cjs');
 
 
 
@@ -27,10 +27,10 @@ var _chunkG4ABMAQYcjs = require('../chunk-G4ABMAQY.cjs');
 
 
 
-var _chunkB3TLRM4Qcjs = require('../chunk-B3TLRM4Q.cjs');
-require('../chunk-SAZCNSVY.cjs');
-require('../chunk-3MBP4WNC.cjs');
-require('../chunk-6EUWRXLT.cjs');
+var _chunkL5MHM6JJcjs = require('../chunk-L5MHM6JJ.cjs');
+require('../chunk-FGOZELKN.cjs');
+require('../chunk-TPJNKVFB.cjs');
+require('../chunk-DQVVH5ZK.cjs');
 require('../chunk-5QGQK44L.cjs');
 
 
@@ -59,5 +59,5 @@ require('../chunk-5QGQK44L.cjs');
 
 
 
-exports.ActorFeature = _chunkB3TLRM4Qcjs.ActorFeature; exports.ActorId = _chunkB3TLRM4Qcjs.ActorId; exports.ActorLogEntry = _chunkB3TLRM4Qcjs.ActorLogEntry; exports.ActorSchema = _chunkB3TLRM4Qcjs.ActorSchema; exports.BuildSchema = _chunkB3TLRM4Qcjs.BuildSchema; exports.BuildsSchema = _chunkB3TLRM4Qcjs.BuildsSchema; exports.ColumnSchema = _chunkB3TLRM4Qcjs.ColumnSchema; exports.ColumnsSchema = _chunkB3TLRM4Qcjs.ColumnsSchema; exports.ConnectionSchema = _chunkB3TLRM4Qcjs.ConnectionSchema; exports.CreateActorSchema = _chunkB3TLRM4Qcjs.CreateActorSchema; exports.DatabaseQuerySchema = _chunkB3TLRM4Qcjs.DatabaseQuerySchema; exports.ForeignKeySchema = _chunkB3TLRM4Qcjs.ForeignKeySchema; exports.ForeignKeysSchema = _chunkB3TLRM4Qcjs.ForeignKeysSchema; exports.OperationSchema = _chunkB3TLRM4Qcjs.OperationSchema; exports.PatchSchema = _chunkB3TLRM4Qcjs.PatchSchema; exports.RealtimeEventSchema = _chunkB3TLRM4Qcjs.RealtimeEventSchema; exports.RecordedRealtimeEventSchema = _chunkB3TLRM4Qcjs.RecordedRealtimeEventSchema; exports.TableSchema = _chunkB3TLRM4Qcjs.TableSchema; exports.TablesSchema = _chunkB3TLRM4Qcjs.TablesSchema; exports.compareSecrets = _chunkG4ABMAQYcjs.compareSecrets; exports.configureInspectorAccessToken = _chunkG4ABMAQYcjs.configureInspectorAccessToken; exports.createActorInspectorClient = _chunkG4ABMAQYcjs.createActorInspectorClient; exports.createManagerInspectorClient = _chunkG4ABMAQYcjs.createManagerInspectorClient; exports.getInspectorUrl = _chunkG4ABMAQYcjs.getInspectorUrl; exports.isInspectorEnabled = _chunkG4ABMAQYcjs.isInspectorEnabled; exports.secureInspector = _chunkG4ABMAQYcjs.secureInspector;
+exports.ActorFeature = _chunkL5MHM6JJcjs.ActorFeature; exports.ActorId = _chunkL5MHM6JJcjs.ActorId; exports.ActorLogEntry = _chunkL5MHM6JJcjs.ActorLogEntry; exports.ActorSchema = _chunkL5MHM6JJcjs.ActorSchema; exports.BuildSchema = _chunkL5MHM6JJcjs.BuildSchema; exports.BuildsSchema = _chunkL5MHM6JJcjs.BuildsSchema; exports.ColumnSchema = _chunkL5MHM6JJcjs.ColumnSchema; exports.ColumnsSchema = _chunkL5MHM6JJcjs.ColumnsSchema; exports.ConnectionSchema = _chunkL5MHM6JJcjs.ConnectionSchema; exports.CreateActorSchema = _chunkL5MHM6JJcjs.CreateActorSchema; exports.DatabaseQuerySchema = _chunkL5MHM6JJcjs.DatabaseQuerySchema; exports.ForeignKeySchema = _chunkL5MHM6JJcjs.ForeignKeySchema; exports.ForeignKeysSchema = _chunkL5MHM6JJcjs.ForeignKeysSchema; exports.OperationSchema = _chunkL5MHM6JJcjs.OperationSchema; exports.PatchSchema = _chunkL5MHM6JJcjs.PatchSchema; exports.RealtimeEventSchema = _chunkL5MHM6JJcjs.RealtimeEventSchema; exports.RecordedRealtimeEventSchema = _chunkL5MHM6JJcjs.RecordedRealtimeEventSchema; exports.TableSchema = _chunkL5MHM6JJcjs.TableSchema; exports.TablesSchema = _chunkL5MHM6JJcjs.TablesSchema; exports.compareSecrets = _chunkQGUQB3NCcjs.compareSecrets; exports.configureInspectorAccessToken = _chunkQGUQB3NCcjs.configureInspectorAccessToken; exports.createActorInspectorClient = _chunkQGUQB3NCcjs.createActorInspectorClient; exports.createManagerInspectorClient = _chunkQGUQB3NCcjs.createManagerInspectorClient; exports.getInspectorUrl = _chunkQGUQB3NCcjs.getInspectorUrl; exports.isInspectorEnabled = _chunkQGUQB3NCcjs.isInspectorEnabled; exports.secureInspector = _chunkQGUQB3NCcjs.secureInspector;
 //# sourceMappingURL=mod.cjs.map

package/dist/tsup/inspector/mod.d.cts

@@ -1,5 +1,5 @@
-import { ak as ActorFeature, c as RunConfig, H as RunConfigInput, ah as ManagerDriver } from '../conn-DdzHTm2E.cjs';
-export { ao as Actor, al as ActorId, am as ActorLogEntry, an as ActorSchema, aO as Build, aN as BuildSchema, aQ as Builds, aP as BuildsSchema, aG as Column, aF as ColumnSchema, aI as Columns, aH as ColumnsSchema, au as Connection, at as ConnectionSchema, aS as CreateActor, aR as CreateActorSchema, aA as DatabaseQuery, az as DatabaseQuerySchema, aK as ForeignKey, aJ as ForeignKeySchema, aM as ForeignKeys, aL as ForeignKeysSchema, aq as Operation, ap as OperationSchema, as as Patch, ar as PatchSchema, aw as RealtimeEvent, av as RealtimeEventSchema, ay as RecordedRealtimeEvent, ax as RecordedRealtimeEventSchema, aC as Table, aB as TableSchema, aE as Tables, aD as TablesSchema } from '../conn-DdzHTm2E.cjs';
+import { ak as ActorFeature, c as RunConfig, H as RunConfigInput, ah as ManagerDriver } from '../conn-ChAuuTr0.cjs';
+export { ao as Actor, al as ActorId, am as ActorLogEntry, an as ActorSchema, aO as Build, aN as BuildSchema, aQ as Builds, aP as BuildsSchema, aG as Column, aF as ColumnSchema, aI as Columns, aH as ColumnsSchema, au as Connection, at as ConnectionSchema, aS as CreateActor, aR as CreateActorSchema, aA as DatabaseQuery, az as DatabaseQuerySchema, aK as ForeignKey, aJ as ForeignKeySchema, aM as ForeignKeys, aL as ForeignKeysSchema, aq as Operation, ap as OperationSchema, as as Patch, ar as PatchSchema, aw as RealtimeEvent, av as RealtimeEventSchema, ay as RecordedRealtimeEvent, ax as RecordedRealtimeEventSchema, aC as Table, aB as TableSchema, aE as Tables, aD as TablesSchema } from '../conn-ChAuuTr0.cjs';
 import * as hono_utils_http_status from 'hono/utils/http-status';
 import * as hono_utils_types from 'hono/utils/types';
 import * as hono_client from 'hono/client';

package/dist/tsup/inspector/mod.d.ts

@@ -1,5 +1,5 @@
-import { ak as ActorFeature, c as RunConfig, H as RunConfigInput, ah as ManagerDriver } from '../conn-DCSQgIlw.js';
-export { ao as Actor, al as ActorId, am as ActorLogEntry, an as ActorSchema, aO as Build, aN as BuildSchema, aQ as Builds, aP as BuildsSchema, aG as Column, aF as ColumnSchema, aI as Columns, aH as ColumnsSchema, au as Connection, at as ConnectionSchema, aS as CreateActor, aR as CreateActorSchema, aA as DatabaseQuery, az as DatabaseQuerySchema, aK as ForeignKey, aJ as ForeignKeySchema, aM as ForeignKeys, aL as ForeignKeysSchema, aq as Operation, ap as OperationSchema, as as Patch, ar as PatchSchema, aw as RealtimeEvent, av as RealtimeEventSchema, ay as RecordedRealtimeEvent, ax as RecordedRealtimeEventSchema, aC as Table, aB as TableSchema, aE as Tables, aD as TablesSchema } from '../conn-DCSQgIlw.js';
+import { ak as ActorFeature, c as RunConfig, H as RunConfigInput, ah as ManagerDriver } from '../conn-CjUkMEcm.js';
+export { ao as Actor, al as ActorId, am as ActorLogEntry, an as ActorSchema, aO as Build, aN as BuildSchema, aQ as Builds, aP as BuildsSchema, aG as Column, aF as ColumnSchema, aI as Columns, aH as ColumnsSchema, au as Connection, at as ConnectionSchema, aS as CreateActor, aR as CreateActorSchema, aA as DatabaseQuery, az as DatabaseQuerySchema, aK as ForeignKey, aJ as ForeignKeySchema, aM as ForeignKeys, aL as ForeignKeysSchema, aq as Operation, ap as OperationSchema, as as Patch, ar as PatchSchema, aw as RealtimeEvent, av as RealtimeEventSchema, ay as RecordedRealtimeEvent, ax as RecordedRealtimeEventSchema, aC as Table, aB as TableSchema, aE as Tables, aD as TablesSchema } from '../conn-CjUkMEcm.js';
 import * as hono_utils_http_status from 'hono/utils/http-status';
 import * as hono_utils_types from 'hono/utils/types';
 import * as hono_client from 'hono/client';

package/dist/tsup/inspector/mod.js

@@ -6,7 +6,7 @@ import {
   getInspectorUrl,
   isInspectorEnabled,
   secureInspector
-} from "../chunk-4GP7BZSR.js";
+} from "../chunk-WP7YG7S5.js";
 import {
   ActorFeature,
   ActorId,
@@ -27,10 +27,10 @@ import {
   RecordedRealtimeEventSchema,
   TableSchema,
   TablesSchema
-} from "../chunk-GZVBFXBI.js";
-import "../chunk-YQ4XQYPM.js";
-import "../chunk-PUSQNDJG.js";
-import "../chunk-RVP5RUSC.js";
+} from "../chunk-DOZBWJRI.js";
+import "../chunk-S6EAEZQA.js";
+import "../chunk-SFRRXLRM.js";
+import "../chunk-A6TV3QU6.js";
 import "../chunk-YPZFLUO6.js";
 export {
   ActorFeature,

package/dist/tsup/mod.cjs

@@ -10,27 +10,27 @@
 
 
 
-var _chunk3Y45CIF4cjs = require('./chunk-3Y45CIF4.cjs');
-require('./chunk-G4ABMAQY.cjs');
+var _chunkCKSA7NOScjs = require('./chunk-CKSA7NOS.cjs');
+require('./chunk-QGUQB3NC.cjs');
 
 
 
 
 
-var _chunk5ZOHIKWGcjs = require('./chunk-5ZOHIKWG.cjs');
-require('./chunk-B3TLRM4Q.cjs');
-require('./chunk-K4ENQCC4.cjs');
-require('./chunk-DFS77KAA.cjs');
+var _chunkKDNB2BQXcjs = require('./chunk-KDNB2BQX.cjs');
+require('./chunk-L5MHM6JJ.cjs');
+require('./chunk-ESD2JX3L.cjs');
+require('./chunk-3WRAGTDC.cjs');
 
 
 
 
-var _chunkSAZCNSVYcjs = require('./chunk-SAZCNSVY.cjs');
-require('./chunk-3MBP4WNC.cjs');
+var _chunkFGOZELKNcjs = require('./chunk-FGOZELKN.cjs');
+require('./chunk-TPJNKVFB.cjs');
 
 
 
-var _chunk6EUWRXLTcjs = require('./chunk-6EUWRXLT.cjs');
+var _chunkDQVVH5ZKcjs = require('./chunk-DQVVH5ZK.cjs');
 
 
 var _chunk5QGQK44Lcjs = require('./chunk-5QGQK44L.cjs');
@@ -56,5 +56,5 @@ var _chunk5QGQK44Lcjs = require('./chunk-5QGQK44L.cjs');
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunkSAZCNSVYcjs.ALLOWED_PUBLIC_HEADERS; exports.InlineWebSocketAdapter2 = _chunk3Y45CIF4cjs.InlineWebSocketAdapter2; exports.PATH_CONNECT_WEBSOCKET = _chunkSAZCNSVYcjs.PATH_CONNECT_WEBSOCKET; exports.PATH_RAW_WEBSOCKET_PREFIX = _chunkSAZCNSVYcjs.PATH_RAW_WEBSOCKET_PREFIX; exports.Registry = _chunk3Y45CIF4cjs.Registry; exports.RegistryConfigSchema = _chunk3Y45CIF4cjs.RegistryConfigSchema; exports.UserError = _chunk5QGQK44Lcjs.UserError; exports.actor = _chunk3Y45CIF4cjs.actor; exports.createActorRouter = _chunk3Y45CIF4cjs.createActorRouter; exports.createClientWithDriver = _chunk5ZOHIKWGcjs.createClientWithDriver; exports.createEngineDriver = _chunk3Y45CIF4cjs.createEngineDriver; exports.createFileSystemDriver = _chunk3Y45CIF4cjs.createFileSystemDriver; exports.createMemoryDriver = _chunk3Y45CIF4cjs.createMemoryDriver; exports.generateConnId = _chunk5ZOHIKWGcjs.generateConnId; exports.generateConnToken = _chunk5ZOHIKWGcjs.generateConnToken; exports.handleRawWebSocketHandler = _chunk3Y45CIF4cjs.handleRawWebSocketHandler; exports.handleWebSocketConnect = _chunk3Y45CIF4cjs.handleWebSocketConnect; exports.lookupInRegistry = _chunk5ZOHIKWGcjs.lookupInRegistry; exports.noopNext = _chunk6EUWRXLTcjs.noopNext; exports.setup = _chunk3Y45CIF4cjs.setup; exports.toUint8Array = _chunk6EUWRXLTcjs.toUint8Array;
+exports.ALLOWED_PUBLIC_HEADERS = _chunkFGOZELKNcjs.ALLOWED_PUBLIC_HEADERS; exports.InlineWebSocketAdapter2 = _chunkCKSA7NOScjs.InlineWebSocketAdapter2; exports.PATH_CONNECT_WEBSOCKET = _chunkFGOZELKNcjs.PATH_CONNECT_WEBSOCKET; exports.PATH_RAW_WEBSOCKET_PREFIX = _chunkFGOZELKNcjs.PATH_RAW_WEBSOCKET_PREFIX; exports.Registry = _chunkCKSA7NOScjs.Registry; exports.RegistryConfigSchema = _chunkCKSA7NOScjs.RegistryConfigSchema; exports.UserError = _chunk5QGQK44Lcjs.UserError; exports.actor = _chunkCKSA7NOScjs.actor; exports.createActorRouter = _chunkCKSA7NOScjs.createActorRouter; exports.createClientWithDriver = _chunkKDNB2BQXcjs.createClientWithDriver; exports.createEngineDriver = _chunkCKSA7NOScjs.createEngineDriver; exports.createFileSystemDriver = _chunkCKSA7NOScjs.createFileSystemDriver; exports.createMemoryDriver = _chunkCKSA7NOScjs.createMemoryDriver; exports.generateConnId = _chunkKDNB2BQXcjs.generateConnId; exports.generateConnToken = _chunkKDNB2BQXcjs.generateConnToken; exports.handleRawWebSocketHandler = _chunkCKSA7NOScjs.handleRawWebSocketHandler; exports.handleWebSocketConnect = _chunkCKSA7NOScjs.handleWebSocketConnect; exports.lookupInRegistry = _chunkKDNB2BQXcjs.lookupInRegistry; exports.noopNext = _chunkDQVVH5ZKcjs.noopNext; exports.setup = _chunkCKSA7NOScjs.setup; exports.toUint8Array = _chunkDQVVH5ZKcjs.toUint8Array;
 //# sourceMappingURL=mod.cjs.map

package/dist/tsup/mod.d.cts

@@ -1,5 +1,5 @@
-import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-DdzHTm2E.cjs';
-export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-DdzHTm2E.cjs';
+import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-ChAuuTr0.cjs';
+export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-ChAuuTr0.cjs';
 import { WSContext } from 'hono/ws';
 export { n as noopNext } from './utils-fwx3o3K9.cjs';
 import * as hono_types from 'hono/types';
@@ -108,7 +108,7 @@ type ActorRouter = Hono<{
 /**
  * Creates a router that runs on the partitioned instance.
  */
-declare function createActorRouter(runConfig: RunConfig, actorDriver: ActorDriver): ActorRouter;
+declare function createActorRouter(runConfig: RunConfig, actorDriver: ActorDriver, isTest: boolean): ActorRouter;
 
 declare function actor<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TActions extends Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase>>(input: ActorConfigInput<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TActions>): ActorDefinition<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TActions>;
 
@@ -122,23 +122,23 @@ declare const ConfigSchema: z.ZodDefault<z.ZodObject<{
     runnerKey: z.ZodDefault<z.ZodString>;
     totalSlots: z.ZodDefault<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
+    totalSlots: number;
     endpoint: string;
     namespace: string;
     runnerName: string;
     runnerKey: string;
-    totalSlots: number;
     token?: string | undefined;
     app?: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/"> | undefined;
     pegboardEndpoint?: string | undefined;
 }, {
     token?: string | undefined;
+    totalSlots?: number | undefined;
     endpoint?: string | undefined;
     namespace?: string | undefined;
     runnerName?: string | undefined;
     app?: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/"> | undefined;
     pegboardEndpoint?: string | undefined;
     runnerKey?: string | undefined;
-    totalSlots?: number | undefined;
 }>>;
 type InputConfig = z.input<typeof ConfigSchema>;
 

package/dist/tsup/mod.d.ts

@@ -1,5 +1,5 @@
-import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-DCSQgIlw.js';
-export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-DCSQgIlw.js';
+import { U as UniversalWebSocket, R as RivetEvent, a as RivetCloseEvent, b as RivetMessageEvent, A as ActorDriver, c as RunConfig, E as Encoding, d as AnyDatabaseProvider, e as Actions, f as ActorConfigInput, g as ActorDefinition, D as DriverConfig } from './conn-CjUkMEcm.js';
+export { n as ActionContext, r as ActionContextOf, z as ActorConfig, x as ActorConfigSchema, q as ActorContext, s as ActorContextOf, m as ActorKey, l as ActorQuery, w as ActorTypes, t as AnyActorDefinition, v as AnyActorInstance, j as AnyClient, y as AuthIntent, C as Client, o as Conn, p as ConnectionStatus, I as InitContext, O as OnConnectOptions, K as Registry, G as RegistryActors, F as RegistryConfig, J as RegistryConfigSchema, H as RunConfigInput, k as createClientWithDriver, h as generateConnId, i as generateConnToken, u as lookupInRegistry, L as setup, B as test } from './conn-CjUkMEcm.js';
 import { WSContext } from 'hono/ws';
 export { n as noopNext } from './utils-fwx3o3K9.js';
 import * as hono_types from 'hono/types';
@@ -108,7 +108,7 @@ type ActorRouter = Hono<{
 /**
  * Creates a router that runs on the partitioned instance.
  */
-declare function createActorRouter(runConfig: RunConfig, actorDriver: ActorDriver): ActorRouter;
+declare function createActorRouter(runConfig: RunConfig, actorDriver: ActorDriver, isTest: boolean): ActorRouter;
 
 declare function actor<TState, TConnParams, TConnState, TVars, TInput, TDatabase extends AnyDatabaseProvider, TActions extends Actions<TState, TConnParams, TConnState, TVars, TInput, TDatabase>>(input: ActorConfigInput<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TActions>): ActorDefinition<TState, TConnParams, TConnState, TVars, TInput, TDatabase, TActions>;
 
@@ -122,23 +122,23 @@ declare const ConfigSchema: z.ZodDefault<z.ZodObject<{
     runnerKey: z.ZodDefault<z.ZodString>;
     totalSlots: z.ZodDefault<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
+    totalSlots: number;
     endpoint: string;
     namespace: string;
     runnerName: string;
     runnerKey: string;
-    totalSlots: number;
     token?: string | undefined;
     app?: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/"> | undefined;
     pegboardEndpoint?: string | undefined;
 }, {
     token?: string | undefined;
+    totalSlots?: number | undefined;
     endpoint?: string | undefined;
     namespace?: string | undefined;
     runnerName?: string | undefined;
     app?: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/"> | undefined;
     pegboardEndpoint?: string | undefined;
     runnerKey?: string | undefined;
-    totalSlots?: number | undefined;
 }>>;
 type InputConfig = z.input<typeof ConfigSchema>;
 

package/dist/tsup/mod.js

@@ -10,27 +10,27 @@ import {
   handleRawWebSocketHandler,
   handleWebSocketConnect,
   setup
-} from "./chunk-HPT3I7UU.js";
-import "./chunk-4GP7BZSR.js";
+} from "./chunk-E77RVI3P.js";
+import "./chunk-WP7YG7S5.js";
 import {
   createClientWithDriver,
   generateConnId,
   generateConnToken,
   lookupInRegistry
-} from "./chunk-BW5DPM6Z.js";
-import "./chunk-GZVBFXBI.js";
-import "./chunk-E4UVJKSV.js";
-import "./chunk-JD54PXWP.js";
+} from "./chunk-N7OVEOMU.js";
+import "./chunk-DOZBWJRI.js";
+import "./chunk-2FAWAPRT.js";
+import "./chunk-PVKV2O2E.js";
 import {
   ALLOWED_PUBLIC_HEADERS,
   PATH_CONNECT_WEBSOCKET,
   PATH_RAW_WEBSOCKET_PREFIX
-} from "./chunk-YQ4XQYPM.js";
-import "./chunk-PUSQNDJG.js";
+} from "./chunk-S6EAEZQA.js";
+import "./chunk-SFRRXLRM.js";
 import {
   noopNext,
   toUint8Array
-} from "./chunk-RVP5RUSC.js";
+} from "./chunk-A6TV3QU6.js";
 import {
   UserError
 } from "./chunk-YPZFLUO6.js";

package/dist/tsup/test/mod.cjs

@@ -1,19 +1,19 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
-var _chunkTZGUSEIJcjs = require('../chunk-TZGUSEIJ.cjs');
-require('../chunk-3Y45CIF4.cjs');
-require('../chunk-G4ABMAQY.cjs');
-require('../chunk-5ZOHIKWG.cjs');
-require('../chunk-B3TLRM4Q.cjs');
-require('../chunk-K4ENQCC4.cjs');
-require('../chunk-DFS77KAA.cjs');
-require('../chunk-SAZCNSVY.cjs');
-require('../chunk-3MBP4WNC.cjs');
-require('../chunk-6EUWRXLT.cjs');
+var _chunk6INXQCH7cjs = require('../chunk-6INXQCH7.cjs');
+require('../chunk-CKSA7NOS.cjs');
+require('../chunk-QGUQB3NC.cjs');
+require('../chunk-KDNB2BQX.cjs');
+require('../chunk-L5MHM6JJ.cjs');
+require('../chunk-ESD2JX3L.cjs');
+require('../chunk-3WRAGTDC.cjs');
+require('../chunk-FGOZELKN.cjs');
+require('../chunk-TPJNKVFB.cjs');
+require('../chunk-DQVVH5ZK.cjs');
 require('../chunk-5QGQK44L.cjs');
 
 
 
-exports.getPort = _chunkTZGUSEIJcjs.getPort; exports.setupTest = _chunkTZGUSEIJcjs.setupTest;
+exports.getPort = _chunk6INXQCH7cjs.getPort; exports.setupTest = _chunk6INXQCH7cjs.setupTest;
 //# sourceMappingURL=mod.cjs.map

package/dist/tsup/test/mod.d.cts

@@ -1,5 +1,5 @@
 import { TestContext } from 'vitest';
-import { K as Registry, C as Client } from '../conn-DdzHTm2E.cjs';
+import { K as Registry, C as Client } from '../conn-ChAuuTr0.cjs';
 import 'hono/streaming';
 import 'hono/ws';
 import '../utils.cjs';

package/dist/tsup/test/mod.d.ts

@@ -1,5 +1,5 @@
 import { TestContext } from 'vitest';
-import { K as Registry, C as Client } from '../conn-DCSQgIlw.js';
+import { K as Registry, C as Client } from '../conn-CjUkMEcm.js';
 import 'hono/streaming';
 import 'hono/ws';
 import '../utils.js';

package/dist/tsup/test/mod.js

@@ -1,16 +1,16 @@
 import {
   getPort,
   setupTest
-} from "../chunk-SBKRVQS2.js";
-import "../chunk-HPT3I7UU.js";
-import "../chunk-4GP7BZSR.js";
-import "../chunk-BW5DPM6Z.js";
-import "../chunk-GZVBFXBI.js";
-import "../chunk-E4UVJKSV.js";
-import "../chunk-JD54PXWP.js";
-import "../chunk-YQ4XQYPM.js";
-import "../chunk-PUSQNDJG.js";
-import "../chunk-RVP5RUSC.js";
+} from "../chunk-RM2V2IRK.js";
+import "../chunk-E77RVI3P.js";
+import "../chunk-WP7YG7S5.js";
+import "../chunk-N7OVEOMU.js";
+import "../chunk-DOZBWJRI.js";
+import "../chunk-2FAWAPRT.js";
+import "../chunk-PVKV2O2E.js";
+import "../chunk-S6EAEZQA.js";
+import "../chunk-SFRRXLRM.js";
+import "../chunk-A6TV3QU6.js";
 import "../chunk-YPZFLUO6.js";
 export {
   getPort,

package/dist/tsup/utils.cjs

@@ -11,7 +11,7 @@
 
 
 
-var _chunk6EUWRXLTcjs = require('./chunk-6EUWRXLT.cjs');
+var _chunkDQVVH5ZKcjs = require('./chunk-DQVVH5ZK.cjs');
 require('./chunk-5QGQK44L.cjs');
 
 
@@ -26,5 +26,5 @@ require('./chunk-5QGQK44L.cjs');
 
 
 
-exports.SinglePromiseQueue = _chunk6EUWRXLTcjs.SinglePromiseQueue; exports.VERSION = _chunk6EUWRXLTcjs.VERSION; exports.assertUnreachable = _chunk6EUWRXLTcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunk6EUWRXLTcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunk6EUWRXLTcjs.combineUrlPath; exports.dbg = _chunk6EUWRXLTcjs.dbg; exports.getEnvUniversal = _chunk6EUWRXLTcjs.getEnvUniversal; exports.httpUserAgent = _chunk6EUWRXLTcjs.httpUserAgent; exports.promiseWithResolvers = _chunk6EUWRXLTcjs.promiseWithResolvers; exports.setLongTimeout = _chunk6EUWRXLTcjs.setLongTimeout; exports.stringifyError = _chunk6EUWRXLTcjs.stringifyError; exports.toUint8Array = _chunk6EUWRXLTcjs.toUint8Array;
+exports.SinglePromiseQueue = _chunkDQVVH5ZKcjs.SinglePromiseQueue; exports.VERSION = _chunkDQVVH5ZKcjs.VERSION; exports.assertUnreachable = _chunkDQVVH5ZKcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkDQVVH5ZKcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkDQVVH5ZKcjs.combineUrlPath; exports.dbg = _chunkDQVVH5ZKcjs.dbg; exports.getEnvUniversal = _chunkDQVVH5ZKcjs.getEnvUniversal; exports.httpUserAgent = _chunkDQVVH5ZKcjs.httpUserAgent; exports.promiseWithResolvers = _chunkDQVVH5ZKcjs.promiseWithResolvers; exports.setLongTimeout = _chunkDQVVH5ZKcjs.setLongTimeout; exports.stringifyError = _chunkDQVVH5ZKcjs.stringifyError; exports.toUint8Array = _chunkDQVVH5ZKcjs.toUint8Array;
 //# sourceMappingURL=utils.cjs.map

package/dist/tsup/utils.js

@@ -11,7 +11,7 @@ import {
   setLongTimeout,
   stringifyError,
   toUint8Array
-} from "./chunk-RVP5RUSC.js";
+} from "./chunk-A6TV3QU6.js";
 import "./chunk-YPZFLUO6.js";
 export {
   SinglePromiseQueue,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rivetkit",
-  "version": "2.0.7-rc.1",
+  "version": "2.0.8",
   "description": "Lightweight libraries for building stateful actors on edge platforms",
   "license": "Apache-2.0",
   "keywords": [
@@ -155,7 +155,7 @@
     "@bare-ts/lib": "~0.3.0",
     "@hono/standard-validator": "^0.1.3",
     "@hono/zod-openapi": "^0.19.10",
-    "@rivetkit/engine-runner": "https://pkg.pr.new/rivet-dev/engine/@rivetkit/engine-runner@3002",
+    "@rivetkit/engine-runner": "https://pkg.pr.new/rivet-dev/engine/@rivetkit/engine-runner@03e3532",
     "@rivetkit/fast-json-patch": "^3.1.2",
     "cbor-x": "^1.6.0",
     "hono": "^4.7.0",

package/src/actor/instance.ts

@@ -373,6 +373,9 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 			this.#config.options.connectionLivenessInterval,
 		);
 		this.#checkConnectionsLiveness();
+
+		// Trigger any pending alarms
+		await this._onAlarm();
 	}
 
 	async #scheduleEventInner(newEvent: PersistedScheduleEvent) {
@@ -401,6 +404,12 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 		}
 	}
 
+	/**
+	 * Triggers any pending alarms.
+	 *
+	 * This method is idempotent. It's called automatically when the actor wakes
+	 * in order to trigger any pending alarms.
+	 */
 	async _onAlarm() {
 		const now = Date.now();
 		this.actorContext.log.debug({
@@ -424,7 +433,7 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 			this.#rLog.warn({ msg: "no events are due yet, time may have broken" });
 			if (this.#persist.scheduledEvents.length > 0) {
 				const nextTs = this.#persist.scheduledEvents[0].timestamp;
-				this.actorContext.log.warn({
+				this.actorContext.log.debug({
 					msg: "alarm fired early, rescheduling for next event",
 					now,
 					nextTs,
@@ -786,7 +795,7 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 	}
 
 	/**
-	 * Connection disconnected.
+	 * Call when conn is disconnected. Used by transports.
 	 *
 	 * If a clean diconnect, will be removed immediately.
 	 *
@@ -800,7 +809,7 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 		// If socket ID is provided, check if it matches the current socket ID
 		// If it doesn't match, this is a stale disconnect event from an old socket
 		if (socketId && conn.__socket && socketId !== conn.__socket.socketId) {
-			this.rLog.debug({
+			this.#rLog.debug({
 				msg: "ignoring stale disconnect event",
 				connId: conn.id,
 				eventSocketId: socketId,
@@ -825,6 +834,9 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 
 			// Remove socket
 			conn.__socket = undefined;
+
+			// Update sleep
+			this.#resetSleepTimer();
 		}
 	}
 
@@ -848,6 +860,7 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 
 		// Remove from state
 		this.#connections.delete(conn.id);
+		this.#rLog.debug({ msg: "removed conn", connId: conn.id });
 
 		// Remove subscriptions
 		for (const eventName of [...conn.subscriptions.values()]) {

package/src/actor/router-endpoints.ts

@@ -292,7 +292,8 @@ export async function handleWebSocketConnect(
 			// Handle cleanup asynchronously
 			handlersPromise
 				.then(({ conn, actor }) => {
-					actor.__connDisconnected(conn, event.wasClean, socketId);
+					const wasClean = event.wasClean || event.code === 1000;
+					actor.__connDisconnected(conn, wasClean, socketId);
 				})
 				.catch((error) => {
 					deconstructError(

package/src/actor/router.ts

@@ -69,6 +69,7 @@ export type ActorRouter = Hono<{ Bindings: ActorRouterBindings }>;
 export function createActorRouter(
 	runConfig: RunConfig,
 	actorDriver: ActorDriver,
+	isTest: boolean,
 ): ActorRouter {
 	const router = new Hono<{ Bindings: ActorRouterBindings }>({ strict: false });
 
@@ -84,37 +85,39 @@ export function createActorRouter(
 		return c.text("ok");
 	});
 
-	// Test endpoint to force disconnect a connection non-cleanly
-	router.post("/.test/force-disconnect", async (c) => {
-		const connId = c.req.query("conn");
+	if (isTest) {
+		// Test endpoint to force disconnect a connection non-cleanly
+		router.post("/.test/force-disconnect", async (c) => {
+			const connId = c.req.query("conn");
 
-		if (!connId) {
-			return c.text("Missing conn query parameter", 400);
-		}
+			if (!connId) {
+				return c.text("Missing conn query parameter", 400);
+			}
 
-		const actor = await actorDriver.loadActor(c.env.actorId);
-		const conn = actor.__getConnForId(connId);
+			const actor = await actorDriver.loadActor(c.env.actorId);
+			const conn = actor.__getConnForId(connId);
 
-		if (!conn) {
-			return c.text(`Connection not found: ${connId}`, 404);
-		}
+			if (!conn) {
+				return c.text(`Connection not found: ${connId}`, 404);
+			}
 
-		// Force close the websocket/SSE connection without clean shutdown
-		const driverState = conn.__driverState;
-		if (driverState && ConnDriverKind.WEBSOCKET in driverState) {
-			const ws = driverState[ConnDriverKind.WEBSOCKET].websocket;
+			// Force close the websocket/SSE connection without clean shutdown
+			const driverState = conn.__driverState;
+			if (driverState && ConnDriverKind.WEBSOCKET in driverState) {
+				const ws = driverState[ConnDriverKind.WEBSOCKET].websocket;
 
-			// Force close without sending close frame
-			(ws.raw as any).terminate();
-		} else if (driverState && ConnDriverKind.SSE in driverState) {
-			const stream = driverState[ConnDriverKind.SSE].stream;
+				// Force close without sending close frame
+				(ws.raw as any).terminate();
+			} else if (driverState && ConnDriverKind.SSE in driverState) {
+				const stream = driverState[ConnDriverKind.SSE].stream;
 
-			// Force close the SSE stream
-			stream.abort();
-		}
+				// Force close the SSE stream
+				stream.abort();
+			}
 
-		return c.json({ success: true });
-	});
+			return c.json({ success: true });
+		});
+	}
 
 	router.get(PATH_CONNECT_WEBSOCKET, async (c) => {
 		const upgradeWebSocket = runConfig.getUpgradeWebSocket?.();

package/src/client/actor-conn.ts

@@ -914,6 +914,8 @@ enc
 		if (!this.#transport) {
 			// Nothing to do
 		} else if ("websocket" in this.#transport) {
+			logger().debug("closing ws");
+
 			const ws = this.#transport.websocket;
 			// Check if WebSocket is already closed or closing
 			if (
@@ -927,10 +929,12 @@ enc
 					logger().debug({ msg: "ws closed" });
 					resolve(undefined);
 				});
-				ws.close();
+				ws.close(1000, "Normal closure");
 				await promise;
 			}
 		} else if ("sse" in this.#transport) {
+			logger().debug("closing sse");
+
 			// Send close request to server for SSE connections
 			if (this.#connectionId && this.#connectionToken) {
 				try {

package/src/client/config.ts

@@ -19,6 +19,8 @@ export const ClientConfigSchema = z.object({
 		.optional()
 		.transform((x) => x ?? getEnvUniversal("RIVET_TOKEN")),
 
+	totalSlots: z.number().optional(),
+
 	headers: z.record(z.string()).optional().default({}),
 
 	/** Endpoint to connect to the Rivet engine. Can be configured via RIVET_ENGINE env var. */

package/src/driver-test-suite/mod.ts

@@ -35,6 +35,7 @@ export interface SkipTests {
 	schedule?: boolean;
 	sleep?: boolean;
 	sse?: boolean;
+	inline?: boolean;
 }
 
 export interface DriverTestConfig {
@@ -79,7 +80,10 @@ export interface DriverDeployOutput {
 export function runDriverTests(
 	driverTestConfigPartial: Omit<DriverTestConfig, "clientType" | "transport">,
 ) {
-	for (const clientType of ["http", "inline"] as ClientType[]) {
+	const clientTypes: ClientType[] = driverTestConfigPartial.skip?.inline
+		? ["http"]
+		: ["http", "inline"];
+	for (const clientType of clientTypes) {
 		const driverTestConfig: DriverTestConfig = {
 			...driverTestConfigPartial,
 			clientType,
@@ -148,7 +152,12 @@ export function runDriverTests(
 export async function createTestRuntime(
 	registryPath: string,
 	driverFactory: (registry: Registry<any>) => Promise<{
-		rivetEngine?: { endpoint: string; namespace: string; runnerName: string };
+		rivetEngine?: {
+			endpoint: string;
+			namespace: string;
+			runnerName: string;
+			token: string;
+		};
 		driver: DriverConfig;
 		cleanup?: () => Promise<void>;
 	}>,