rivetkit 2.0.7 → 2.0.9

package/dist/tsup/inspector/mod.cjs

@@ -6,7 +6,7 @@
 
 
 
-var _chunk4EXJ4ITRcjs = require('../chunk-4EXJ4ITR.cjs');
+var _chunkQGRYH6TUcjs = require('../chunk-QGRYH6TU.cjs');
 
 
 
@@ -27,10 +27,10 @@ var _chunk4EXJ4ITRcjs = require('../chunk-4EXJ4ITR.cjs');
 
 
 
-var _chunkOSK2VSJFcjs = require('../chunk-OSK2VSJF.cjs');
-require('../chunk-NDCVQZBS.cjs');
-require('../chunk-ETDWYT2P.cjs');
-require('../chunk-LXAVET4A.cjs');
+var _chunk2MJYYF2Qcjs = require('../chunk-2MJYYF2Q.cjs');
+require('../chunk-SDXTJDDR.cjs');
+require('../chunk-DL7TPF63.cjs');
+require('../chunk-U2IXX6DY.cjs');
 require('../chunk-5QGQK44L.cjs');
 
 
@@ -59,5 +59,5 @@ require('../chunk-5QGQK44L.cjs');
 
 
 
-exports.ActorFeature = _chunkOSK2VSJFcjs.ActorFeature; exports.ActorId = _chunkOSK2VSJFcjs.ActorId; exports.ActorLogEntry = _chunkOSK2VSJFcjs.ActorLogEntry; exports.ActorSchema = _chunkOSK2VSJFcjs.ActorSchema; exports.BuildSchema = _chunkOSK2VSJFcjs.BuildSchema; exports.BuildsSchema = _chunkOSK2VSJFcjs.BuildsSchema; exports.ColumnSchema = _chunkOSK2VSJFcjs.ColumnSchema; exports.ColumnsSchema = _chunkOSK2VSJFcjs.ColumnsSchema; exports.ConnectionSchema = _chunkOSK2VSJFcjs.ConnectionSchema; exports.CreateActorSchema = _chunkOSK2VSJFcjs.CreateActorSchema; exports.DatabaseQuerySchema = _chunkOSK2VSJFcjs.DatabaseQuerySchema; exports.ForeignKeySchema = _chunkOSK2VSJFcjs.ForeignKeySchema; exports.ForeignKeysSchema = _chunkOSK2VSJFcjs.ForeignKeysSchema; exports.OperationSchema = _chunkOSK2VSJFcjs.OperationSchema; exports.PatchSchema = _chunkOSK2VSJFcjs.PatchSchema; exports.RealtimeEventSchema = _chunkOSK2VSJFcjs.RealtimeEventSchema; exports.RecordedRealtimeEventSchema = _chunkOSK2VSJFcjs.RecordedRealtimeEventSchema; exports.TableSchema = _chunkOSK2VSJFcjs.TableSchema; exports.TablesSchema = _chunkOSK2VSJFcjs.TablesSchema; exports.compareSecrets = _chunk4EXJ4ITRcjs.compareSecrets; exports.configureInspectorAccessToken = _chunk4EXJ4ITRcjs.configureInspectorAccessToken; exports.createActorInspectorClient = _chunk4EXJ4ITRcjs.createActorInspectorClient; exports.createManagerInspectorClient = _chunk4EXJ4ITRcjs.createManagerInspectorClient; exports.getInspectorUrl = _chunk4EXJ4ITRcjs.getInspectorUrl; exports.isInspectorEnabled = _chunk4EXJ4ITRcjs.isInspectorEnabled; exports.secureInspector = _chunk4EXJ4ITRcjs.secureInspector;
+exports.ActorFeature = _chunk2MJYYF2Qcjs.ActorFeature; exports.ActorId = _chunk2MJYYF2Qcjs.ActorId; exports.ActorLogEntry = _chunk2MJYYF2Qcjs.ActorLogEntry; exports.ActorSchema = _chunk2MJYYF2Qcjs.ActorSchema; exports.BuildSchema = _chunk2MJYYF2Qcjs.BuildSchema; exports.BuildsSchema = _chunk2MJYYF2Qcjs.BuildsSchema; exports.ColumnSchema = _chunk2MJYYF2Qcjs.ColumnSchema; exports.ColumnsSchema = _chunk2MJYYF2Qcjs.ColumnsSchema; exports.ConnectionSchema = _chunk2MJYYF2Qcjs.ConnectionSchema; exports.CreateActorSchema = _chunk2MJYYF2Qcjs.CreateActorSchema; exports.DatabaseQuerySchema = _chunk2MJYYF2Qcjs.DatabaseQuerySchema; exports.ForeignKeySchema = _chunk2MJYYF2Qcjs.ForeignKeySchema; exports.ForeignKeysSchema = _chunk2MJYYF2Qcjs.ForeignKeysSchema; exports.OperationSchema = _chunk2MJYYF2Qcjs.OperationSchema; exports.PatchSchema = _chunk2MJYYF2Qcjs.PatchSchema; exports.RealtimeEventSchema = _chunk2MJYYF2Qcjs.RealtimeEventSchema; exports.RecordedRealtimeEventSchema = _chunk2MJYYF2Qcjs.RecordedRealtimeEventSchema; exports.TableSchema = _chunk2MJYYF2Qcjs.TableSchema; exports.TablesSchema = _chunk2MJYYF2Qcjs.TablesSchema; exports.compareSecrets = _chunkQGRYH6TUcjs.compareSecrets; exports.configureInspectorAccessToken = _chunkQGRYH6TUcjs.configureInspectorAccessToken; exports.createActorInspectorClient = _chunkQGRYH6TUcjs.createActorInspectorClient; exports.createManagerInspectorClient = _chunkQGRYH6TUcjs.createManagerInspectorClient; exports.getInspectorUrl = _chunkQGRYH6TUcjs.getInspectorUrl; exports.isInspectorEnabled = _chunkQGRYH6TUcjs.isInspectorEnabled; exports.secureInspector = _chunkQGRYH6TUcjs.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-CEh3WKbA.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-CEh3WKbA.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-Bt8rkUzm.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-Bt8rkUzm.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-PD6HCAJE.js";
+} from "../chunk-YR2VY4XS.js";
 import {
   ActorFeature,
   ActorId,
@@ -27,10 +27,10 @@ import {
   RecordedRealtimeEventSchema,
   TableSchema,
   TablesSchema
-} from "../chunk-54MAHBLL.js";
-import "../chunk-NRELKXIX.js";
-import "../chunk-YL4VZMMT.js";
-import "../chunk-RLBM6D4L.js";
+} from "../chunk-WBSPHV5V.js";
+import "../chunk-R7OP5N25.js";
+import "../chunk-DLPIL3VC.js";
+import "../chunk-E63WU5PL.js";
 import "../chunk-YPZFLUO6.js";
 export {
   ActorFeature,

package/dist/tsup/mod.cjs

@@ -10,27 +10,27 @@
 
 
 
-var _chunkF7YL5G7Qcjs = require('./chunk-F7YL5G7Q.cjs');
-require('./chunk-4EXJ4ITR.cjs');
+var _chunk4YV6RDZLcjs = require('./chunk-4YV6RDZL.cjs');
+require('./chunk-QGRYH6TU.cjs');
 
 
 
 
 
-var _chunkNUA6LOOJcjs = require('./chunk-NUA6LOOJ.cjs');
-require('./chunk-OSK2VSJF.cjs');
-require('./chunk-VAF63BEI.cjs');
-require('./chunk-7OOBMCQI.cjs');
+var _chunk4PSLOAXRcjs = require('./chunk-4PSLOAXR.cjs');
+require('./chunk-2MJYYF2Q.cjs');
+require('./chunk-DZZQG7VH.cjs');
+require('./chunk-SOC4HWCG.cjs');
 
 
 
 
-var _chunkNDCVQZBScjs = require('./chunk-NDCVQZBS.cjs');
-require('./chunk-ETDWYT2P.cjs');
+var _chunkSDXTJDDRcjs = require('./chunk-SDXTJDDR.cjs');
+require('./chunk-DL7TPF63.cjs');
 
 
 
-var _chunkLXAVET4Acjs = require('./chunk-LXAVET4A.cjs');
+var _chunkU2IXX6DYcjs = require('./chunk-U2IXX6DY.cjs');
 
 
 var _chunk5QGQK44Lcjs = require('./chunk-5QGQK44L.cjs');
@@ -56,5 +56,5 @@ var _chunk5QGQK44Lcjs = require('./chunk-5QGQK44L.cjs');
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunkNDCVQZBScjs.ALLOWED_PUBLIC_HEADERS; exports.InlineWebSocketAdapter2 = _chunkF7YL5G7Qcjs.InlineWebSocketAdapter2; exports.PATH_CONNECT_WEBSOCKET = _chunkNDCVQZBScjs.PATH_CONNECT_WEBSOCKET; exports.PATH_RAW_WEBSOCKET_PREFIX = _chunkNDCVQZBScjs.PATH_RAW_WEBSOCKET_PREFIX; exports.Registry = _chunkF7YL5G7Qcjs.Registry; exports.RegistryConfigSchema = _chunkF7YL5G7Qcjs.RegistryConfigSchema; exports.UserError = _chunk5QGQK44Lcjs.UserError; exports.actor = _chunkF7YL5G7Qcjs.actor; exports.createActorRouter = _chunkF7YL5G7Qcjs.createActorRouter; exports.createClientWithDriver = _chunkNUA6LOOJcjs.createClientWithDriver; exports.createEngineDriver = _chunkF7YL5G7Qcjs.createEngineDriver; exports.createFileSystemDriver = _chunkF7YL5G7Qcjs.createFileSystemDriver; exports.createMemoryDriver = _chunkF7YL5G7Qcjs.createMemoryDriver; exports.generateConnId = _chunkNUA6LOOJcjs.generateConnId; exports.generateConnToken = _chunkNUA6LOOJcjs.generateConnToken; exports.handleRawWebSocketHandler = _chunkF7YL5G7Qcjs.handleRawWebSocketHandler; exports.handleWebSocketConnect = _chunkF7YL5G7Qcjs.handleWebSocketConnect; exports.lookupInRegistry = _chunkNUA6LOOJcjs.lookupInRegistry; exports.noopNext = _chunkLXAVET4Acjs.noopNext; exports.setup = _chunkF7YL5G7Qcjs.setup; exports.toUint8Array = _chunkLXAVET4Acjs.toUint8Array;
+exports.ALLOWED_PUBLIC_HEADERS = _chunkSDXTJDDRcjs.ALLOWED_PUBLIC_HEADERS; exports.InlineWebSocketAdapter2 = _chunk4YV6RDZLcjs.InlineWebSocketAdapter2; exports.PATH_CONNECT_WEBSOCKET = _chunkSDXTJDDRcjs.PATH_CONNECT_WEBSOCKET; exports.PATH_RAW_WEBSOCKET_PREFIX = _chunkSDXTJDDRcjs.PATH_RAW_WEBSOCKET_PREFIX; exports.Registry = _chunk4YV6RDZLcjs.Registry; exports.RegistryConfigSchema = _chunk4YV6RDZLcjs.RegistryConfigSchema; exports.UserError = _chunk5QGQK44Lcjs.UserError; exports.actor = _chunk4YV6RDZLcjs.actor; exports.createActorRouter = _chunk4YV6RDZLcjs.createActorRouter; exports.createClientWithDriver = _chunk4PSLOAXRcjs.createClientWithDriver; exports.createEngineDriver = _chunk4YV6RDZLcjs.createEngineDriver; exports.createFileSystemDriver = _chunk4YV6RDZLcjs.createFileSystemDriver; exports.createMemoryDriver = _chunk4YV6RDZLcjs.createMemoryDriver; exports.generateConnId = _chunk4PSLOAXRcjs.generateConnId; exports.generateConnToken = _chunk4PSLOAXRcjs.generateConnToken; exports.handleRawWebSocketHandler = _chunk4YV6RDZLcjs.handleRawWebSocketHandler; exports.handleWebSocketConnect = _chunk4YV6RDZLcjs.handleWebSocketConnect; exports.lookupInRegistry = _chunk4PSLOAXRcjs.lookupInRegistry; exports.noopNext = _chunkU2IXX6DYcjs.noopNext; exports.setup = _chunk4YV6RDZLcjs.setup; exports.toUint8Array = _chunkU2IXX6DYcjs.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-CEh3WKbA.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-CEh3WKbA.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-Bt8rkUzm.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-Bt8rkUzm.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-KHRZPP5T.js";
-import "./chunk-PD6HCAJE.js";
+} from "./chunk-FZP2IBIX.js";
+import "./chunk-YR2VY4XS.js";
 import {
   createClientWithDriver,
   generateConnId,
   generateConnToken,
   lookupInRegistry
-} from "./chunk-NII4KKHD.js";
-import "./chunk-54MAHBLL.js";
-import "./chunk-B6N6VM37.js";
-import "./chunk-2NL3KGJ7.js";
+} from "./chunk-KHZ2QSQ4.js";
+import "./chunk-WBSPHV5V.js";
+import "./chunk-APHV6WXU.js";
+import "./chunk-VVCL5DXN.js";
 import {
   ALLOWED_PUBLIC_HEADERS,
   PATH_CONNECT_WEBSOCKET,
   PATH_RAW_WEBSOCKET_PREFIX
-} from "./chunk-NRELKXIX.js";
-import "./chunk-YL4VZMMT.js";
+} from "./chunk-R7OP5N25.js";
+import "./chunk-DLPIL3VC.js";
 import {
   noopNext,
   toUint8Array
-} from "./chunk-RLBM6D4L.js";
+} from "./chunk-E63WU5PL.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 _chunk3ALZ7EGXcjs = require('../chunk-3ALZ7EGX.cjs');
-require('../chunk-F7YL5G7Q.cjs');
-require('../chunk-4EXJ4ITR.cjs');
-require('../chunk-NUA6LOOJ.cjs');
-require('../chunk-OSK2VSJF.cjs');
-require('../chunk-VAF63BEI.cjs');
-require('../chunk-7OOBMCQI.cjs');
-require('../chunk-NDCVQZBS.cjs');
-require('../chunk-ETDWYT2P.cjs');
-require('../chunk-LXAVET4A.cjs');
+var _chunk7OMMIAWPcjs = require('../chunk-7OMMIAWP.cjs');
+require('../chunk-4YV6RDZL.cjs');
+require('../chunk-QGRYH6TU.cjs');
+require('../chunk-4PSLOAXR.cjs');
+require('../chunk-2MJYYF2Q.cjs');
+require('../chunk-DZZQG7VH.cjs');
+require('../chunk-SOC4HWCG.cjs');
+require('../chunk-SDXTJDDR.cjs');
+require('../chunk-DL7TPF63.cjs');
+require('../chunk-U2IXX6DY.cjs');
 require('../chunk-5QGQK44L.cjs');
 
 
 
-exports.getPort = _chunk3ALZ7EGXcjs.getPort; exports.setupTest = _chunk3ALZ7EGXcjs.setupTest;
+exports.getPort = _chunk7OMMIAWPcjs.getPort; exports.setupTest = _chunk7OMMIAWPcjs.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-CEh3WKbA.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-Bt8rkUzm.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-WAT5AE7S.js";
-import "../chunk-KHRZPP5T.js";
-import "../chunk-PD6HCAJE.js";
-import "../chunk-NII4KKHD.js";
-import "../chunk-54MAHBLL.js";
-import "../chunk-B6N6VM37.js";
-import "../chunk-2NL3KGJ7.js";
-import "../chunk-NRELKXIX.js";
-import "../chunk-YL4VZMMT.js";
-import "../chunk-RLBM6D4L.js";
+} from "../chunk-WRSWUDFA.js";
+import "../chunk-FZP2IBIX.js";
+import "../chunk-YR2VY4XS.js";
+import "../chunk-KHZ2QSQ4.js";
+import "../chunk-WBSPHV5V.js";
+import "../chunk-APHV6WXU.js";
+import "../chunk-VVCL5DXN.js";
+import "../chunk-R7OP5N25.js";
+import "../chunk-DLPIL3VC.js";
+import "../chunk-E63WU5PL.js";
 import "../chunk-YPZFLUO6.js";
 export {
   getPort,

package/dist/tsup/utils.cjs

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

package/dist/tsup/utils.js

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rivetkit",
-  "version": "2.0.7",
+  "version": "2.0.9",
   "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@16cf6cc",
     "@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({
@@ -419,12 +428,12 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 			(x) => x.timestamp <= now,
 		);
 		if (runIndex === -1) {
-			// No events are due yet. This will happen if timers fire slightly early.
-			// Ensure we reschedule the alarm for the next upcoming event to avoid losing it.
-			this.#rLog.warn({ msg: "no events are due yet, time may have broken" });
+			// This method is idempotent, so this will happen in scenarios like `start` and
+			// no events are pending.
+			this.#rLog.debug({ msg: "no events are due yet" });
 			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()]) {
@@ -1226,7 +1239,7 @@ export class ActorInstance<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
 				}
 
 				// Connection is dead, remove it
-				this.#rLog.warn({
+				this.#rLog.info({
 					msg: "connection is dead, removing",
 					connId: conn.id,
 					lastSeen,

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-helpers/mod.ts

@@ -1,5 +1,6 @@
 export type { ActorDriver } from "@/actor/driver";
 export type { ActorInstance, AnyActorInstance } from "@/actor/instance";
+export { generateRandomString } from "@/actor/utils";
 export {
 	ALLOWED_PUBLIC_HEADERS,
 	HEADER_ACTOR_ID,

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>;
 	}>,