rivetkit 2.3.0-rc.10 → 2.3.0-rc.12

package/dist/tsup/mod.cjs

@@ -1,15 +1,15 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2; var _class3;
 
 
 
 
 
 
-var _chunkWQ4HNA4Wcjs = require('./chunk-WQ4HNA4W.cjs');
 
+var _chunkWHYBAEWGcjs = require('./chunk-WHYBAEWG.cjs');
 
-var _chunkG5RULGYQcjs = require('./chunk-G5RULGYQ.cjs');
 
+var _chunkC7AAIILHcjs = require('./chunk-C7AAIILH.cjs');
 
 
 
@@ -18,23 +18,23 @@ var _chunkG5RULGYQcjs = require('./chunk-G5RULGYQ.cjs');
 
 
 
-var _chunkTE4VCDNYcjs = require('./chunk-TE4VCDNY.cjs');
 
+var _chunk2U6RLFKXcjs = require('./chunk-2U6RLFKX.cjs');
 
 
 
 
 
 
-var _chunkLD5YASJUcjs = require('./chunk-LD5YASJU.cjs');
 
+var _chunkVE2X4KMGcjs = require('./chunk-VE2X4KMG.cjs');
 
 
 
-var _chunkPWFGP2UScjs = require('./chunk-PWFGP2US.cjs');
-require('./chunk-2NDZ7JCR.cjs');
-require('./chunk-Z4C3W2CQ.cjs');
 
+var _chunkSS56HFM2cjs = require('./chunk-SS56HFM2.cjs');
+require('./chunk-ZA7FLHKH.cjs');
+require('./chunk-UETC5RF7.cjs');
 
 
 
@@ -57,9 +57,9 @@ require('./chunk-Z4C3W2CQ.cjs');
 
 
 
-var _chunkJ5P6S2LCcjs = require('./chunk-J5P6S2LC.cjs');
 
 
+var _chunkVNMIAPPFcjs = require('./chunk-VNMIAPPF.cjs');
 
 
 
@@ -68,11 +68,13 @@ var _chunkJ5P6S2LCcjs = require('./chunk-J5P6S2LC.cjs');
 
 
 
-var _chunkQAZLM4WTcjs = require('./chunk-QAZLM4WT.cjs');
+
+
+var _chunkKORQB2IRcjs = require('./chunk-KORQB2IR.cjs');
 
 // src/actor/log.ts
 function loggerWithoutContext() {
-  return _chunkJ5P6S2LCcjs.getLogger.call(void 0, "actor-runtime");
+  return _chunkVNMIAPPFcjs.getLogger.call(void 0, "actor-runtime");
 }
 
 // src/actor/definition.ts
@@ -105,7 +107,7 @@ function actor(input) {
   warnDeprecatedShutdownTimeoutKeys(
     input == null ? void 0 : input.options
   );
-  const config = _chunkWQ4HNA4Wcjs.ActorConfigSchema.parse(input);
+  const config = _chunkWHYBAEWGcjs.ActorConfigSchema.parse(input);
   return new ActorDefinition(config);
 }
 function isStaticActorDefinition(definition) {
@@ -140,7 +142,7 @@ function hasSchemaConfigKey(schemas, key) {
   if (!schemas) {
     return false;
   }
-  return Object.prototype.hasOwnProperty.call(schemas, key);
+  return Object.hasOwn(schemas, key);
 }
 function getEventCanSubscribe(schemas, key) {
   const schema = schemas == null ? void 0 : schemas[key];
@@ -187,7 +189,7 @@ function validateSchemaSync(schemas, key, data) {
   if (isStandardSchema(schema)) {
     const result = schema["~standard"].validate(data);
     if (isPromiseLike(result)) {
-      throw _chunkQAZLM4WTcjs.unsupportedFeature.call(void 0, "async schema validation");
+      throw _chunkKORQB2IRcjs.unsupportedFeature.call(void 0, "async schema validation");
     }
     if (result.issues) {
       return { success: false, issues: [...result.issues] };
@@ -201,10 +203,10 @@ function validateSchemaSync(schemas, key, data) {
 }
 
 // src/common/inline-websocket-adapter.ts
-var _ws = require('hono/ws');
 var _virtualwebsocket = require('@rivetkit/virtual-websocket');
+var _ws = require('hono/ws');
 function logger() {
-  return _chunkJ5P6S2LCcjs.getLogger.call(void 0, "inline-websocket-adapter");
+  return _chunkVNMIAPPFcjs.getLogger.call(void 0, "inline-websocket-adapter");
 }
 var InlineWebSocketAdapter = class {
   #handler;
@@ -314,24 +316,24 @@ var InlineWebSocketAdapter = class {
       this.#close(1011, "Internal error during initialization");
     }
   }
-  #handleError(err) {
-    console.error("INLINE_WEBSOCKET_ADAPTER_ERROR", err);
+  #handleError(error) {
+    console.error("INLINE_WEBSOCKET_ADAPTER_ERROR", error);
     logger().error({
       msg: "error in websocket",
-      error: err,
-      errorMessage: err instanceof Error ? err.message : String(err),
-      stack: err instanceof Error ? err.stack : void 0
+      error,
+      errorMessage: error instanceof Error ? error.message : String(error),
+      stack: error instanceof Error ? error.stack : void 0
     });
     try {
-      this.#handler.onError(err, this.#wsContext);
-    } catch (handlerErr) {
+      this.#handler.onError(error, this.#wsContext);
+    } catch (error2) {
       logger().error({
         msg: "error in onError handler",
-        error: handlerErr
+        error: error2
       });
     }
-    this.#clientWs.triggerError(err);
-    this.#actorWs.triggerError(err);
+    this.#clientWs.triggerError(error);
+    this.#actorWs.triggerError(error);
   }
   #close(code, reason) {
     if (this.#readyState === 3 || this.#readyState === 2) {
@@ -344,8 +346,8 @@ var InlineWebSocketAdapter = class {
         { code, reason, wasClean: true },
         this.#wsContext
       );
-    } catch (err) {
-      logger().error({ msg: "error closing websocket", error: err });
+    } catch (error) {
+      logger().error({ msg: "error closing websocket", error });
     } finally {
       this.#readyState = 3;
       this.#clientWs.triggerClose(code, reason);
@@ -354,13 +356,16 @@ var InlineWebSocketAdapter = class {
   }
 };
 
+// src/registry/index.ts
+var _hono = require('hono');
+
 // src/common/engine.ts
 var ENGINE_PORT = 6420;
 var ENGINE_ENDPOINT = `http://127.0.0.1:${ENGINE_PORT}`;
 
 // src/registry/log.ts
 function logger2() {
-  return _chunkJ5P6S2LCcjs.getLogger.call(void 0, "registry");
+  return _chunkVNMIAPPFcjs.getLogger.call(void 0, "registry");
 }
 
 // src/serverless/configure.ts
@@ -371,10 +376,13 @@ function sleep(ms) {
 }
 function configureTimeoutMs() {
   const value = process.env.RIVET_SERVERLESS_CONFIGURE_TIMEOUT_MS;
-  if (value === void 0 || value === "") return DEFAULT_CONFIGURE_TIMEOUT_MS;
+  if (value === void 0 || value === "")
+    return DEFAULT_CONFIGURE_TIMEOUT_MS;
   const parsed = Number(value);
   if (!Number.isFinite(parsed) || parsed < 0) {
-    throw new Error("RIVET_SERVERLESS_CONFIGURE_TIMEOUT_MS must be a finite non-negative number");
+    throw new Error(
+      "RIVET_SERVERLESS_CONFIGURE_TIMEOUT_MS must be a finite non-negative number"
+    );
   }
   return parsed;
 }
@@ -388,17 +396,23 @@ async function configureServerlessPool(config) {
     attempts += 1;
     try {
       if (!config.namespace) {
-        throw new Error("namespace is required for serverless configuration");
+        throw new Error(
+          "namespace is required for serverless configuration"
+        );
       }
       if (!config.endpoint) {
-        throw new Error("endpoint is required for serverless configuration");
+        throw new Error(
+          "endpoint is required for serverless configuration"
+        );
       }
       if (!config.configurePool) {
-        throw new Error("configurePool is required for serverless configuration");
+        throw new Error(
+          "configurePool is required for serverless configuration"
+        );
       }
       const customConfig = config.configurePool;
-      const clientConfig = _chunkTE4VCDNYcjs.convertRegistryConfigToClientConfig.call(void 0, config);
-      const dcsRes = await _chunkTE4VCDNYcjs.getDatacenters.call(void 0, clientConfig);
+      const clientConfig = _chunk2U6RLFKXcjs.convertRegistryConfigToClientConfig.call(void 0, config);
+      const dcsRes = await _chunk2U6RLFKXcjs.getDatacenters.call(void 0, clientConfig);
       const poolName = _nullishCoalesce(customConfig.name, () => ( "default"));
       const serverlessToken = _nullishCoalesce(config.token, () => ( config.publicToken));
       const headers = {
@@ -420,7 +434,7 @@ async function configureServerlessPool(config) {
         metadata: _nullishCoalesce(customConfig.metadata, () => ( {})),
         drain_on_version_upgrade: _nullishCoalesce(customConfig.drainOnVersionUpgrade, () => ( true))
       };
-      await _chunkTE4VCDNYcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
+      await _chunk2U6RLFKXcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
         datacenters: Object.fromEntries(
           dcsRes.datacenters.map((dc) => [dc.name, serverlessConfig])
         )
@@ -437,7 +451,7 @@ async function configureServerlessPool(config) {
       logger2().warn({
         msg: "serverless pool configuration attempt failed",
         attempts,
-        error: _chunkJ5P6S2LCcjs.stringifyError.call(void 0, error)
+        error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, error)
       });
       await sleep(CONFIGURE_RETRY_DELAY_MS);
     }
@@ -445,11 +459,150 @@ async function configureServerlessPool(config) {
   logger2().error({
     msg: "failed to configure serverless pool, validate endpoint is configured correctly then restart this process",
     attempts,
-    error: _chunkJ5P6S2LCcjs.stringifyError.call(void 0, lastError)
+    error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, lastError)
   });
   throw lastError;
 }
 
+// src/utils/serve.ts
+var _getport = require('get-port'); var _getport2 = _interopRequireDefault(_getport);
+var serveStaticLoaderPromises = {};
+async function crossPlatformServe(config, httpPort, app, runtime = _chunkVNMIAPPFcjs.detectRuntime.call(void 0, )) {
+  logger2().debug({ msg: "detected runtime for serve", runtime });
+  switch (runtime) {
+    case "deno":
+      return serveDeno(config, httpPort, app);
+    case "bun":
+      return serveBun(config, httpPort, app);
+    case "node":
+      return serveNode(config, httpPort, app);
+    default:
+      return serveNode(config, httpPort, app);
+  }
+}
+async function loadRuntimeServeStatic(runtime) {
+  if (!serveStaticLoaderPromises[runtime]) {
+    if (runtime === "node") {
+      const nodeServeStaticModule = "@hono/node-server/serve-static";
+      serveStaticLoaderPromises[runtime] = Promise.resolve().then(() => _interopRequireWildcard(require(
+        /* webpackIgnore: true */
+        nodeServeStaticModule
+      ))).then((x) => x.serveStatic);
+    } else if (runtime === "bun") {
+      const bunModule = "hono/bun";
+      serveStaticLoaderPromises[runtime] = Promise.resolve().then(() => _interopRequireWildcard(require(
+        /* webpackIgnore: true */
+        bunModule
+      ))).then((x) => x.serveStatic);
+    } else if (runtime === "deno") {
+      const denoModule = "hono/deno";
+      serveStaticLoaderPromises[runtime] = Promise.resolve().then(() => _interopRequireWildcard(require(
+        /* webpackIgnore: true */
+        denoModule
+      ))).then((x) => x.serveStatic);
+    } else {
+      throw new Error(`unsupported runtime: ${runtime}`);
+    }
+  }
+  return await serveStaticLoaderPromises[runtime];
+}
+async function serveNode(config, httpPort, app) {
+  const nodeServerModule = "@hono/node-server";
+  let serve;
+  try {
+    const dep = await Promise.resolve().then(() => _interopRequireWildcard(require(
+      /* webpackIgnore: true */
+      nodeServerModule
+    )));
+    serve = dep.serve;
+  } catch (err) {
+    logger2().error({
+      msg: "failed to import @hono/node-server. please run 'npm install @hono/node-server @hono/node-ws'",
+      error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, err)
+    });
+    process.exit(1);
+  }
+  const nodeWsModule = "@hono/node-ws";
+  let createNodeWebSocket;
+  try {
+    const dep = await Promise.resolve().then(() => _interopRequireWildcard(require(
+      /* webpackIgnore: true */
+      nodeWsModule
+    )));
+    createNodeWebSocket = dep.createNodeWebSocket;
+  } catch (err) {
+    logger2().error({
+      msg: "failed to import @hono/node-ws. please run 'npm install @hono/node-server @hono/node-ws'",
+      error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, err)
+    });
+    process.exit(1);
+  }
+  const { injectWebSocket, upgradeWebSocket } = createNodeWebSocket({
+    app
+  });
+  const port = httpPort;
+  const hostname = config.httpHost;
+  const server = serve(
+    { fetch: app.fetch, port, hostname },
+    () => logger2().info({ msg: "server listening", port, hostname })
+  );
+  injectWebSocket(server);
+  const closeServer = () => {
+    server.close();
+  };
+  return { upgradeWebSocket, closeServer };
+}
+async function serveDeno(config, httpPort, app) {
+  const honoDenoModule = "hono/deno";
+  let upgradeWebSocket;
+  try {
+    const dep = await Promise.resolve().then(() => _interopRequireWildcard(require(
+      /* webpackIgnore: true */
+      honoDenoModule
+    )));
+    upgradeWebSocket = dep.upgradeWebSocket;
+  } catch (err) {
+    logger2().error({
+      msg: "failed to import hono/deno",
+      error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, err)
+    });
+    process.exit(1);
+  }
+  const port = httpPort;
+  const hostname = config.httpHost;
+  Deno.serve({ port, hostname }, app.fetch);
+  logger2().info({ msg: "server listening", port, hostname });
+  return { upgradeWebSocket };
+}
+async function serveBun(config, httpPort, app) {
+  const honoBunModule = "hono/bun";
+  let createBunWebSocket;
+  try {
+    const dep = await Promise.resolve().then(() => _interopRequireWildcard(require(
+      /* webpackIgnore: true */
+      honoBunModule
+    )));
+    createBunWebSocket = dep.createBunWebSocket;
+  } catch (err) {
+    logger2().error({
+      msg: "failed to import hono/bun",
+      error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, err)
+    });
+    process.exit(1);
+  }
+  const { websocket, upgradeWebSocket } = createBunWebSocket();
+  const port = httpPort;
+  const hostname = config.httpHost;
+  Bun.serve({
+    fetch: app.fetch,
+    port,
+    hostname,
+    websocket
+  });
+  logger2().info({ msg: "server listening", port, hostname });
+  return { upgradeWebSocket };
+}
+
 // src/registry/config/index.ts
 var _zod = require('zod');
 
@@ -457,20 +610,20 @@ var _zod = require('zod');
 var _v4 = require('zod/v4');
 var warnedMissingVersion = false;
 var EnvoyConfigSchema = _v4.z.object({
-  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunkJ5P6S2LCcjs.getRivetPool.call(void 0, ), () => ( "default"))),
+  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunkVNMIAPPFcjs.getRivetPool.call(void 0, ), () => ( "default"))),
   version: _v4.z.number().default(() => {
-    const version = _chunkJ5P6S2LCcjs.getRivetEnvoyVersion.call(void 0, );
+    const version = _chunkVNMIAPPFcjs.getRivetEnvoyVersion.call(void 0, );
     if (version !== void 0) return version;
-    if (_chunkJ5P6S2LCcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
+    if (_chunkVNMIAPPFcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
       warnedMissingVersion = true;
-      _chunkJ5P6S2LCcjs.getLogger.call(void 0, "rivetkit").error(
+      _chunkVNMIAPPFcjs.getLogger.call(void 0, "rivetkit").error(
         "RIVET_ENVOY_VERSION is not set. Actors will not be versioned, which means they won't be drained on deploy. This is only needed when self-hosting or using a custom envoy (not needed for Rivet Compute). Set this as a build arg in your Dockerfile. See https://rivet.dev/docs/actors/versions"
       );
     }
     return 1;
   }),
   // Deprecated.
-  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunkJ5P6S2LCcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
+  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunkVNMIAPPFcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
   envoyKey: _v4.z.string().optional()
 });
 
@@ -508,7 +661,7 @@ var ServerlessConfigSchema = _v4.z.object({
    *
    * Can also be set via RIVET_PUBLIC_ENDPOINT environment variable.
    */
-  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkJ5P6S2LCcjs.getRivetPublicEndpoint.call(void 0, )))),
+  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkVNMIAPPFcjs.getRivetPublicEndpoint.call(void 0, )))),
   /**
    * Token that clients should use when connecting via the public endpoint.
    *
@@ -516,7 +669,7 @@ var ServerlessConfigSchema = _v4.z.object({
    *
    * Can also be specified in the publicEndpoint URL as `https://namespace:token@host`.
    */
-  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkJ5P6S2LCcjs.getRivetPublicToken.call(void 0, ))))
+  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkVNMIAPPFcjs.getRivetPublicToken.call(void 0, ))))
   // There is no publicNamespace config option because the frontend and backend
   // cannot use different namespaces. The namespace is extracted from the
   // publicEndpoint URL auth syntax if provided.
@@ -570,7 +723,7 @@ var RegistryConfigSchema = _zod.z.object({
    * Runtime binding to use for RivetKit core.
    * */
   runtime: RuntimeKindSchema.optional().transform((val, ctx) => {
-    const rawRuntime = _nullishCoalesce(val, () => ( _chunkJ5P6S2LCcjs.getRivetkitRuntime.call(void 0, )));
+    const rawRuntime = _nullishCoalesce(val, () => ( _chunkVNMIAPPFcjs.getRivetkitRuntime.call(void 0, )));
     if (rawRuntime === void 0) {
       return "auto";
     }
@@ -607,7 +760,7 @@ var RegistryConfigSchema = _zod.z.object({
    * */
   logging: _zod.z.object({
     baseLogger: _zod.z.custom().optional(),
-    level: _chunkJ5P6S2LCcjs.LogLevelSchema.optional()
+    level: _chunkVNMIAPPFcjs.LogLevelSchema.optional()
   }).optional().default(() => ({})),
   // MARK: Routing
   // // This is a function to allow for lazy configuration of upgradeWebSocket on the
@@ -625,9 +778,9 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Can also be set via RIVET_ENDPOINT environment variables.
    */
-  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunkJ5P6S2LCcjs.getRivetEngine.call(void 0, ))), () => ( _chunkJ5P6S2LCcjs.getRivetEndpoint.call(void 0, )))),
-  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkJ5P6S2LCcjs.getRivetToken.call(void 0, )))),
-  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkJ5P6S2LCcjs.getRivetNamespace.call(void 0, )))),
+  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunkVNMIAPPFcjs.getRivetEngine.call(void 0, ))), () => ( _chunkVNMIAPPFcjs.getRivetEndpoint.call(void 0, )))),
+  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkVNMIAPPFcjs.getRivetToken.call(void 0, )))),
+  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkVNMIAPPFcjs.getRivetNamespace.call(void 0, )))),
   headers: _zod.z.record(_zod.z.string(), _zod.z.string()).optional().default({}),
   // MARK: Client
   // TODO:
@@ -667,9 +820,9 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Starts the full Rust engine process locally.
    */
-  startEngine: _zod.z.boolean().default(() => _chunkJ5P6S2LCcjs.getRivetRunEngine.call(void 0, )),
+  startEngine: _zod.z.boolean().default(() => _chunkVNMIAPPFcjs.getRivetRunEngine.call(void 0, )),
   /** @experimental */
-  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkJ5P6S2LCcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkJ5P6S2LCcjs.VERSION))),
+  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkVNMIAPPFcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkVNMIAPPFcjs.VERSION))),
   /**
    * @experimental
    *
@@ -703,7 +856,7 @@ var RegistryConfigSchema = _zod.z.object({
      * after calling `CoreRegistry::shutdown()`. Defaults to the
      * engine-provided actor stop threshold once the envoy connects.
      *
-     * Must be >= rivetkit-core's drain timeout (20s) + margin.
+     * Must be long enough for rivetkit-core to drain the envoy.
      */
     gracePeriodMs: _zod.z.number().int().min(1e3).optional(),
     /**
@@ -717,7 +870,7 @@ var RegistryConfigSchema = _zod.z.object({
   }))
 }).transform((config, ctx) => {
   var _a, _b, _c;
-  const isDevEnv = _chunkJ5P6S2LCcjs.isDev.call(void 0, );
+  const isDevEnv = _chunkVNMIAPPFcjs.isDev.call(void 0, );
   const sqliteBackend = _nullishCoalesce(((_a = config.sqlite) == null ? void 0 : _a.backend), () => ( ((_b = config.test) == null ? void 0 : _b.sqliteBackend)));
   if (config.runtime === "wasm" && sqliteBackend === "local") {
     ctx.addIssue({
@@ -727,7 +880,7 @@ var RegistryConfigSchema = _zod.z.object({
     });
   }
   const sqlite = config.runtime === "wasm" && config.sqlite === void 0 ? { backend: "remote" } : config.sqlite;
-  const parsedEndpoint = config.endpoint ? _chunkTE4VCDNYcjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedEndpoint = config.endpoint ? _chunk2U6RLFKXcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.endpoint,
     path: ["endpoint"],
     namespace: config.namespace,
@@ -751,7 +904,7 @@ var RegistryConfigSchema = _zod.z.object({
   );
   const namespace = _nullishCoalesce(_nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.namespace), () => ( config.namespace)), () => ( "default"));
   const token = _nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.token), () => ( config.token));
-  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunkTE4VCDNYcjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunk2U6RLFKXcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.serverless.publicEndpoint,
     path: ["serverless", "publicEndpoint"]
   }) : void 0;
@@ -786,30 +939,30 @@ function buildActorNames(config) {
     Object.keys(config.use).map((actorName) => {
       const definition = config.use[actorName];
       const options = _nullishCoalesce(definition.config.options, () => ( {}));
-      const runMeta = _chunkWQ4HNA4Wcjs.getRunMetadata.call(void 0, definition.config.run);
+      const runMeta = _chunkWHYBAEWGcjs.getRunMetadata.call(void 0, definition.config.run);
       const metadata = {};
       metadata.icon = _nullishCoalesce(options.icon, () => ( runMeta.icon));
       metadata.name = _nullishCoalesce(options.name, () => ( runMeta.name));
       metadata.preload = {
         keys: [
-          Array.from(_chunkLD5YASJUcjs.KEYS.PERSIST_DATA),
-          Array.from(_chunkLD5YASJUcjs.KEYS.INSPECTOR_TOKEN),
-          Array.from(_chunkLD5YASJUcjs.queueMetadataKey.call(void 0, )),
-          Array.from(_chunkLD5YASJUcjs.KEYS.LAST_PUSHED_ALARM)
+          Array.from(_chunkVE2X4KMGcjs.KEYS.PERSIST_DATA),
+          Array.from(_chunkVE2X4KMGcjs.KEYS.INSPECTOR_TOKEN),
+          Array.from(_chunkVE2X4KMGcjs.queueMetadataKey.call(void 0, )),
+          Array.from(_chunkVE2X4KMGcjs.KEYS.LAST_PUSHED_ALARM)
         ],
         prefixes: [
           {
-            prefix: Array.from(_chunkLD5YASJUcjs.workflowStoragePrefix.call(void 0, )),
+            prefix: Array.from(_chunkVE2X4KMGcjs.workflowStoragePrefix.call(void 0, )),
             maxBytes: _nullishCoalesce(options.preloadMaxWorkflowBytes, () => ( 131072)),
             partial: false
           },
           {
-            prefix: Array.from(_chunkLD5YASJUcjs.KEYS.CONN_PREFIX),
+            prefix: Array.from(_chunkVE2X4KMGcjs.KEYS.CONN_PREFIX),
             maxBytes: _nullishCoalesce(options.preloadMaxConnectionsBytes, () => ( 65536)),
             partial: false
           },
           {
-            prefix: Array.from(_chunkLD5YASJUcjs.queueMessagesPrefix.call(void 0, )),
+            prefix: Array.from(_chunkVE2X4KMGcjs.queueMessagesPrefix.call(void 0, )),
             maxBytes: 65536,
             partial: false
           }
@@ -879,7 +1032,7 @@ var DocRegistryConfigSchema = _zod.z.object({
   noWelcome: _zod.z.boolean().optional().describe("Disable the welcome message on startup. Default: false"),
   sqlite: DocSqliteConfigSchema,
   logging: _zod.z.object({
-    level: _chunkJ5P6S2LCcjs.LogLevelSchema.optional().describe(
+    level: _chunkVNMIAPPFcjs.LogLevelSchema.optional().describe(
       "Log level for RivetKit. Default: 'warn'"
     )
   }).optional().describe("Logging configuration."),
@@ -924,7 +1077,7 @@ function shouldAttachNativeKvError(message) {
 function enrichNativeDatabaseError(database, error) {
   var _a;
   const bridgeReason = typeof error === "string" ? error : error instanceof Error ? error.message : void 0;
-  const bridged = bridgeReason === void 0 ? void 0 : _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, bridgeReason);
+  const bridged = bridgeReason === void 0 ? void 0 : _chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, bridgeReason);
   if (bridged) {
     throw bridged;
   }
@@ -1424,6 +1577,9 @@ var NapiCoreRuntime = (_class = class {
   async actorWaitForTrackedShutdownWork(ctx) {
     return await asNativeActorContext(ctx).waitForTrackedShutdownWork();
   }
+  async actorWaitForTrackedShutdownWorkUnbounded(ctx) {
+    await asNativeActorContext(ctx).waitForTrackedShutdownWorkUnbounded();
+  }
   actorKeepAwake(ctx, promise) {
     asNativeActorContext(ctx).keepAwake(promise);
   }
@@ -1535,8 +1691,12 @@ var NapiCoreRuntime = (_class = class {
       )
     );
   }
-  async actorQueueWaitForNamesAvailable(ctx, names, options) {
-    await asNativeActorContext(ctx).queue().waitForNamesAvailable(names, options);
+  async actorQueueWaitForNamesAvailable(ctx, names, options, signal) {
+    await asNativeActorContext(ctx).queue().waitForNamesAvailable(
+      names,
+      options,
+      signal ? asNativeCancellationToken(signal) : signal
+    );
   }
   async actorQueueEnqueueAndWait(ctx, name, body, options, signal) {
     return await asNativeActorContext(ctx).queue().enqueueAndWait(
@@ -1670,20 +1830,18 @@ function validateQueueComplete(schemas, name, response) {
     response
   );
   if (!result.success) {
-    throw validationError(`queue \`${name}\` completion response`, result.issues);
+    throw validationError(
+      `queue \`${name}\` completion response`,
+      result.issues
+    );
   }
   return result.data;
 }
 function validationError(target, issues) {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
-    "actor",
-    "validation_error",
-    `Invalid ${target}`,
-    {
-      public: true,
-      metadata: { issues }
-    }
-  );
+  return new (0, _chunkKORQB2IRcjs.RivetError)("actor", "validation_error", `Invalid ${target}`, {
+    public: true,
+    metadata: { issues }
+  });
 }
 
 // src/registry/wasm-runtime.ts
@@ -1749,16 +1907,16 @@ function normalizeQueueMessage(message) {
 }
 function normalizeWasmBridgeError(error) {
   if (typeof error === "string") {
-    return _nullishCoalesce(_chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
+    return _nullishCoalesce(_chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
   }
   if (error instanceof Error) {
-    const bridged = _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error.message);
+    const bridged = _chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, error.message);
     if (bridged) {
       return bridged;
     }
   }
   if (typeof error === "object" && error !== null && "reason" in error && typeof error.reason === "string") {
-    const bridged = _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error.reason);
+    const bridged = _chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, error.reason);
     if (bridged) {
       return bridged;
     }
@@ -1780,7 +1938,7 @@ function callWasmSync(invoke) {
   }
 }
 function unsupportedWasmMethod(method2) {
-  throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+  throw new (0, _chunkKORQB2IRcjs.RivetError)(
     "runtime",
     "unsupported",
     `Unsupported wasm runtime method: ${method2}`,
@@ -2007,6 +2165,12 @@ var WasmCoreRuntime = (_class2 = class {
       "waitForTrackedShutdownWork"
     );
   }
+  async actorWaitForTrackedShutdownWorkUnbounded(ctx) {
+    await callHandle(
+      asWasmActorContext(ctx),
+      "waitForTrackedShutdownWorkUnbounded"
+    );
+  }
   actorKeepAwake(ctx, promise) {
     const wasmCtx = asWasmActorContext(ctx);
     const regionId = callHandle(wasmCtx, "beginKeepAwake");
@@ -2154,9 +2318,15 @@ var WasmCoreRuntime = (_class2 = class {
       )
     );
   }
-  async actorQueueWaitForNamesAvailable(ctx, names, options) {
+  async actorQueueWaitForNamesAvailable(ctx, names, options, signal) {
     const queue2 = childHandle(asWasmActorContext(ctx), "queue");
-    await callHandleAsync(queue2, "waitForNamesAvailable", names, options);
+    await callHandleAsync(
+      queue2,
+      "waitForNamesAvailable",
+      names,
+      options,
+      signal
+    );
   }
   async actorQueueEnqueueAndWait(ctx, name, body, options, signal) {
     const queue2 = childHandle(asWasmActorContext(ctx), "queue");
@@ -2248,6 +2418,29 @@ async function loadWasmRuntime(config) {
   };
 }
 
+// src/registry/write-through-proxy.ts
+var _onchange = require('@rivetkit/on-change'); var _onchange2 = _interopRequireDefault(_onchange);
+function createWriteThroughProxy(value, commit, beforeChange) {
+  if (!value || typeof value !== "object") {
+    return value;
+  }
+  return _onchange2.default.call(void 0, 
+    value,
+    () => {
+      commit(value);
+    },
+    {
+      // Rejection is throw-based: beforeChange throws to prevent the
+      // mutation. We always return true so on-change applies the change
+      // if beforeChange did not throw.
+      onValidate(_path, newValue) {
+        beforeChange == null ? void 0 : beforeChange(newValue);
+        return true;
+      }
+    }
+  );
+}
+
 // src/registry/native.ts
 var textEncoder = new TextEncoder();
 var textDecoder = new TextDecoder();
@@ -2281,7 +2474,7 @@ function loadedRuntimeKind(runtime) {
     case "wasm":
       return "wasm";
   }
-  throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+  throw new (0, _chunkKORQB2IRcjs.RivetError)(
     "config",
     "unknown_runtime",
     "RivetKit runtime must be NAPI or wasm.",
@@ -2321,7 +2514,7 @@ function normalizeRuntimeConfigForKind(config, runtimeKind) {
     return config;
   }
   if (sqliteBackendForConfig(config) === "local") {
-    throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+    throw new (0, _chunkKORQB2IRcjs.RivetError)(
       "config",
       "wasm_local_sqlite",
       "WebAssembly runtime cannot use local SQLite. Use remote SQLite instead.",
@@ -2387,22 +2580,30 @@ function getNativeConnPersistState(runtime, ctx, conn) {
   return connState;
 }
 function stateMutationReentrantError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
     "actor",
     "state_mutation_reentrant",
     "State mutations are not allowed inside onStateChange."
   );
 }
 function databaseNotConfiguredError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
     "actor",
     "database_not_configured",
     "database is not configured for this actor",
     { public: true }
   );
 }
+function databaseClientNotReadyError() {
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
+    "actor",
+    "database_client_not_ready",
+    "actor database client was not initialized before user code ran. this is an internal lifecycle error; the migration callback should have pre-warmed the client. file an issue if you can reproduce.",
+    { public: true }
+  );
+}
 function stateNotEnabledError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
     "actor",
     "state_not_enabled",
     "State not enabled. Must implement `createState` or `state` to use state. (https://www.rivet.dev/docs/actors/state/#initializing-state)",
@@ -2410,7 +2611,7 @@ function stateNotEnabledError() {
   );
 }
 function nativeClientNotConfiguredError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
     "native",
     "client_not_configured",
     "native actor client is not configured",
@@ -2418,7 +2619,7 @@ function nativeClientNotConfiguredError() {
   );
 }
 function nativeEndpointNotConfiguredError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
     "native",
     "endpoint_not_configured",
     "registry endpoint is required for native envoy startup",
@@ -2452,8 +2653,23 @@ function resolveNativeDestroy(runtime, ctx) {
 function clearNativeRuntimeState(runtime, ctx) {
   callNativeSync(() => runtime.actorClearRuntimeState(ctx));
 }
-async function cleanupNativeSleepRuntimeState(runtime, ctx) {
-  await runtime.actorWaitForTrackedShutdownWork(ctx);
+async function cleanupNativeSleepRuntimeState(runtime, ctx, afterTrackedWorkDrained) {
+  const drained = await runtime.actorWaitForTrackedShutdownWork(ctx);
+  if (!drained) {
+    await closeNativeDatabaseClient(runtime, ctx);
+    await closeNativeSqlDatabase(runtime, ctx);
+    void runtime.actorWaitForTrackedShutdownWorkUnbounded(ctx).then(async () => {
+      await (afterTrackedWorkDrained == null ? void 0 : afterTrackedWorkDrained());
+      clearNativeRuntimeState(runtime, ctx);
+    }).catch((error) => {
+      logger2().warn({
+        msg: "deferred native sleep cleanup failed",
+        error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, error)
+      });
+    });
+    return;
+  }
+  await (afterTrackedWorkDrained == null ? void 0 : afterTrackedWorkDrained());
   await closeNativeDatabaseClient(runtime, ctx);
   await closeNativeSqlDatabase(runtime, ctx);
   clearNativeRuntimeState(runtime, ctx);
@@ -2558,10 +2774,10 @@ function decodeValue(value) {
   if (!value || value.length === 0) {
     return void 0;
   }
-  return _chunkPWFGP2UScjs.decodeCborJsonCompat.call(void 0, value);
+  return _chunkSS56HFM2cjs.decodeCborCompat.call(void 0, value);
 }
 function encodeValue(value) {
-  return _chunkPWFGP2UScjs.encodeCborCompat.call(void 0, value);
+  return _chunkSS56HFM2cjs.encodeCborCompat.call(void 0, value);
 }
 function unwrapTsfnPayload(error, payload) {
   if (error !== null && error !== void 0) {
@@ -2571,16 +2787,16 @@ function unwrapTsfnPayload(error, payload) {
 }
 function normalizeNativeBridgeError(error) {
   if (typeof error === "string") {
-    return _nullishCoalesce(_chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
+    return _nullishCoalesce(_chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
   }
   if (error instanceof Error) {
-    const bridged = _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error.message);
+    const bridged = _chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, error.message);
     if (bridged) {
       return bridged;
     }
   }
   if (typeof error === "object" && error !== null && "reason" in error && typeof error.reason === "string") {
-    const bridged = _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error.reason);
+    const bridged = _chunkKORQB2IRcjs.decodeBridgeRivetError.call(void 0, error.reason);
     if (bridged) {
       return bridged;
     }
@@ -2588,14 +2804,14 @@ function normalizeNativeBridgeError(error) {
   return error;
 }
 function isStructuredBridgeError(error) {
-  if (error instanceof _chunkQAZLM4WTcjs.RivetError) {
+  if (error instanceof _chunkKORQB2IRcjs.RivetError) {
     return true;
   }
-  return _chunkQAZLM4WTcjs.isRivetErrorLike.call(void 0, error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
+  return _chunkKORQB2IRcjs.isRivetErrorLike.call(void 0, error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
 }
 function encodeNativeCallbackError(error) {
-  const structuredError = isStructuredBridgeError(error) ? error : _chunkJ5P6S2LCcjs.deconstructError.call(void 0, error, true);
-  const bridgeError = new Error(_chunkQAZLM4WTcjs.encodeBridgeRivetError.call(void 0, structuredError), {
+  const structuredError = isStructuredBridgeError(error) ? error : _chunkVNMIAPPFcjs.deconstructError.call(void 0, error, true);
+  const bridgeError = new Error(_chunkKORQB2IRcjs.encodeBridgeRivetError.call(void 0, structuredError), {
     cause: error instanceof Error ? error : void 0
   });
   return Object.assign(bridgeError, {
@@ -2618,16 +2834,10 @@ function callNativeSync(invoke) {
     throw normalizeNativeBridgeError(error);
   }
 }
-function actorAbortedError() {
-  return Object.assign(new Error("Actor aborted"), {
-    group: "actor",
-    code: "aborted"
-  });
-}
 function isClosedTaskRegistrationError(error) {
-  const metadata = error instanceof _chunkQAZLM4WTcjs.RivetError ? error.metadata : void 0;
+  const metadata = error instanceof _chunkKORQB2IRcjs.RivetError ? error.metadata : void 0;
   const metadataError = metadata && typeof metadata === "object" && "error" in metadata ? metadata.error : void 0;
-  return error instanceof _chunkQAZLM4WTcjs.RivetError && error.group === "core" && error.code === _chunkQAZLM4WTcjs.INTERNAL_ERROR_CODE && typeof metadataError === "string" && /actor task registration is (closed|not configured)/.test(metadataError);
+  return error instanceof _chunkKORQB2IRcjs.RivetError && error.group === "core" && error.code === _chunkKORQB2IRcjs.INTERNAL_ERROR_CODE && typeof metadataError === "string" && /actor task registration is (closed|not configured)/.test(metadataError);
 }
 async function createCancellationTokenHandle(runtime, signal) {
   if (!signal) {
@@ -2650,7 +2860,7 @@ function decodeWorkflowCbor(data) {
     return null;
   }
   try {
-    return _chunkPWFGP2UScjs.decodeCborCompat.call(void 0, new Uint8Array(data));
+    return _chunkSS56HFM2cjs.decodeCborCompat.call(void 0, new Uint8Array(data));
   } catch (e3) {
     return null;
   }
@@ -2755,8 +2965,8 @@ function serializeWorkflowHistoryForJson(data) {
   if (data === null) {
     return null;
   }
-  const history = _chunkG5RULGYQcjs.decodeWorkflowHistoryTransport.call(void 0, data);
-  return {
+  const history = _chunkC7AAIILHcjs.decodeWorkflowHistoryTransport.call(void 0, data);
+  return jsonSafe({
     nameRegistry: [...history.nameRegistry],
     entries: history.entries.map((entry) => ({
       id: entry.id,
@@ -2780,7 +2990,7 @@ function serializeWorkflowHistoryForJson(data) {
         ]
       )
     )
-  };
+  });
 }
 function toHttpJsonCompatible(value) {
   return JSON.parse(
@@ -2833,44 +3043,6 @@ function decodeArgs(value) {
   const decoded = decodeValue(value);
   return Array.isArray(decoded) ? decoded : decoded === void 0 ? [] : [decoded];
 }
-function createWriteThroughProxy(value, commit, beforeChange) {
-  if (!value || typeof value !== "object") {
-    return value;
-  }
-  const proxies = /* @__PURE__ */ new WeakMap();
-  const wrap = (target) => {
-    const cached = proxies.get(target);
-    if (cached) {
-      return cached;
-    }
-    const proxy = new Proxy(target, {
-      get(innerTarget, property, receiver) {
-        const result = Reflect.get(innerTarget, property, receiver);
-        return result && typeof result === "object" ? wrap(result) : result;
-      },
-      set(innerTarget, property, nextValue, receiver) {
-        beforeChange == null ? void 0 : beforeChange();
-        const updated = Reflect.set(
-          innerTarget,
-          property,
-          nextValue,
-          receiver
-        );
-        commit(value);
-        return updated;
-      },
-      deleteProperty(innerTarget, property) {
-        beforeChange == null ? void 0 : beforeChange();
-        const updated = Reflect.deleteProperty(innerTarget, property);
-        commit(value);
-        return updated;
-      }
-    });
-    proxies.set(target, proxy);
-    return proxy;
-  };
-  return wrap(value);
-}
 function buildRequest(init) {
   const url = init.uri.startsWith("http") ? init.uri : new URL(init.uri, "http://127.0.0.1").toString();
   const body = init.body && init.body.length > 0 ? runtimeBytesToArrayBuffer(init.body) : void 0;
@@ -2906,7 +3078,7 @@ var NativeConnAdapter = class {
     this.#schemas = schemas;
     this.#ctx = ctx;
     this.#queueHibernationRemoval = queueHibernationRemoval;
-    this[_chunkWQ4HNA4Wcjs.CONN_STATE_MANAGER_SYMBOL] = {
+    this[_chunkWHYBAEWGcjs.CONN_STATE_MANAGER_SYMBOL] = {
       stateEnabled: true,
       get state() {
         return thisConn.state;
@@ -2923,13 +3095,23 @@ var NativeConnAdapter = class {
       decodeValue(this.#runtime.connParams(this.#conn))
     );
   }
+  [_chunkWHYBAEWGcjs.RAW_STATE_SYMBOL]() {
+    return this.#readState();
+  }
   get state() {
     const nextState = this.#readState();
-    return createWriteThroughProxy(nextState, (nextValue) => {
-      this.#writeState(nextValue, { writeNative: true });
-    });
+    return createWriteThroughProxy(
+      nextState,
+      (nextValue) => {
+        this.#writeState(nextValue, { writeNative: true });
+      },
+      (newValue) => {
+        _chunkSS56HFM2cjs.assertJsonCompatValue.call(void 0, newValue);
+      }
+    );
   }
   set state(value) {
+    _chunkSS56HFM2cjs.assertJsonCompatValue.call(void 0, value);
     this.#writeState(value, { writeNative: true });
   }
   initializeState(value) {
@@ -3034,7 +3216,7 @@ var NativeKvAdapter = class {
     const value = await callNative(
       () => this.#runtime.actorKvGet(
         this.#ctx,
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(key))
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(key))
       )
     );
     return value ? decodeNativeKvValue(new Uint8Array(value), options) : null;
@@ -3043,7 +3225,7 @@ var NativeKvAdapter = class {
     await callNative(
       () => this.#runtime.actorKvPut(
         this.#ctx,
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(key)),
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(key)),
         toRuntimeBytes(value)
       )
     );
@@ -3052,7 +3234,7 @@ var NativeKvAdapter = class {
     await callNative(
       () => this.#runtime.actorKvDelete(
         this.#ctx,
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(key))
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(key))
       )
     );
   }
@@ -3060,8 +3242,8 @@ var NativeKvAdapter = class {
     await callNative(
       () => this.#runtime.actorKvDeleteRange(
         this.#ctx,
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(start)),
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(end))
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(start)),
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, encodeNativeKvUserKey(end))
       )
     );
   }
@@ -3074,7 +3256,7 @@ var NativeKvAdapter = class {
     const entries = await callNative(
       () => this.#runtime.actorKvListPrefix(
         this.#ctx,
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, 
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, 
           encodeNativeKvUserKey(
             prefix,
             options == null ? void 0 : options.keyType
@@ -3088,7 +3270,7 @@ var NativeKvAdapter = class {
     );
     return entries.map((entry) => [
       decodeNativeKvKey(
-        _chunkLD5YASJUcjs.removePrefixFromKey.call(void 0, new Uint8Array(entry.key)),
+        _chunkVE2X4KMGcjs.removePrefixFromKey.call(void 0, new Uint8Array(entry.key)),
         options == null ? void 0 : options.keyType
       ),
       decodeNativeKvValue(new Uint8Array(entry.value), options)
@@ -3107,13 +3289,13 @@ var NativeKvAdapter = class {
     const entries = await callNative(
       () => this.#runtime.actorKvListRange(
         this.#ctx,
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, 
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, 
           encodeNativeKvUserKey(
             start,
             options == null ? void 0 : options.keyType
           )
         ),
-        _chunkLD5YASJUcjs.makePrefixedKey.call(void 0, 
+        _chunkVE2X4KMGcjs.makePrefixedKey.call(void 0, 
           encodeNativeKvUserKey(
             end,
             options == null ? void 0 : options.keyType
@@ -3127,7 +3309,7 @@ var NativeKvAdapter = class {
     );
     return entries.map((entry) => [
       decodeNativeKvKey(
-        _chunkLD5YASJUcjs.removePrefixFromKey.call(void 0, new Uint8Array(entry.key)),
+        _chunkVE2X4KMGcjs.removePrefixFromKey.call(void 0, new Uint8Array(entry.key)),
         options == null ? void 0 : options.keyType
       ),
       decodeNativeKvValue(new Uint8Array(entry.value), options)
@@ -3219,7 +3401,7 @@ var NativeQueueAdapter = class {
   async nextBatch(options) {
     const completable = (options == null ? void 0 : options.completable) === true;
     if (this.#pendingCompletableMessageIds.size > 0) {
-      throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+      throw new (0, _chunkKORQB2IRcjs.RivetError)(
         "queue",
         "previous_message_not_completed",
         "Previous completable queue message is not completed. Call `message.complete(...)` before receiving the next message.",
@@ -3281,44 +3463,23 @@ var NativeQueueAdapter = class {
     }
   }
   async waitForNamesAvailable(names, options) {
-    if (!(options == null ? void 0 : options.signal)) {
+    const { token, cleanup } = await createCancellationTokenHandle(
+      this.#runtime,
+      options == null ? void 0 : options.signal
+    );
+    try {
       await callNative(
         () => this.#runtime.actorQueueWaitForNamesAvailable(
           this.#ctx,
           [...names],
           {
             timeoutMs: options == null ? void 0 : options.timeout
-          }
+          },
+          token
         )
       );
-      return;
-    }
-    const deadline = options.timeout === void 0 ? void 0 : Date.now() + options.timeout;
-    for (; ; ) {
-      if (options.signal.aborted) {
-        throw actorAbortedError();
-      }
-      const remainingTimeout = deadline === void 0 ? void 0 : Math.max(0, deadline - Date.now());
-      const sliceTimeout = remainingTimeout === void 0 ? 100 : Math.min(remainingTimeout, 100);
-      try {
-        await callNative(
-          () => this.#runtime.actorQueueWaitForNamesAvailable(
-            this.#ctx,
-            [...names],
-            {
-              timeoutMs: sliceTimeout
-            }
-          )
-        );
-        return;
-      } catch (error) {
-        if (error.group === "queue" && error.code === "timed_out") {
-          if (remainingTimeout === void 0 || remainingTimeout > 100) {
-            continue;
-          }
-        }
-        throw error;
-      }
+    } finally {
+      cleanup == null ? void 0 : cleanup();
     }
   }
   async enqueueAndWait(name, body, options) {
@@ -3388,7 +3549,7 @@ var NativeQueueAdapter = class {
         }
         yield message;
       } catch (error) {
-        if (_chunkQAZLM4WTcjs.isRivetErrorLike.call(void 0, error) && error.group === "actor" && error.code === "aborted") {
+        if (_chunkKORQB2IRcjs.isRivetErrorLike.call(void 0, error) && error.group === "actor" && error.code === "aborted") {
           return;
         }
         throw error;
@@ -3409,7 +3570,7 @@ var NativeQueueAdapter = class {
       ...message,
       complete: async (response) => {
         if (typeof message.complete !== "function") {
-          throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+          throw new (0, _chunkKORQB2IRcjs.RivetError)(
             "queue",
             "complete_not_configured",
             `Queue '${message.name}' does not support completion responses.`,
@@ -3421,7 +3582,7 @@ var NativeQueueAdapter = class {
           );
         }
         if (completed) {
-          throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+          throw new (0, _chunkKORQB2IRcjs.RivetError)(
             "queue",
             "already_completed",
             "Queue message was already completed.",
@@ -3753,6 +3914,68 @@ var TrackedWebSocketHandleAdapter = class {
     return typeof value === "object" && value !== null && "then" in value && typeof value.then === "function";
   }
 };
+var NativeConnectionMap = (_class3 = class {
+  #runtime;
+  #ctx;
+  #schemas;
+  constructor(runtime, ctx, schemas) {;_class3.prototype.__init3.call(this);
+    this.#runtime = runtime;
+    this.#ctx = ctx;
+    this.#schemas = schemas;
+  }
+  #connToAdapter(conn) {
+    return new NativeConnAdapter(
+      this.#runtime,
+      conn,
+      this.#schemas,
+      this.#ctx,
+      (connId) => callNativeSync(
+        () => this.#runtime.actorQueueHibernationRemoval(
+          this.#ctx,
+          connId
+        )
+      )
+    );
+  }
+  get size() {
+    return callNativeSync(() => this.#runtime.actorConns(this.#ctx)).length;
+  }
+  get(key) {
+    const conns = callNativeSync(() => this.#runtime.actorConns(this.#ctx));
+    const conn = conns.find((c) => this.#runtime.connId(c) === key);
+    if (!conn) return void 0;
+    return this.#connToAdapter(conn);
+  }
+  has(key) {
+    const conns = callNativeSync(() => this.#runtime.actorConns(this.#ctx));
+    return conns.some((c) => this.#runtime.connId(c) === key);
+  }
+  keys() {
+    const conns = callNativeSync(() => this.#runtime.actorConns(this.#ctx));
+    return conns.map((c) => this.#runtime.connId(c))[Symbol.iterator]();
+  }
+  values() {
+    const conns = callNativeSync(() => this.#runtime.actorConns(this.#ctx));
+    return conns.map((c) => this.#connToAdapter(c))[Symbol.iterator]();
+  }
+  entries() {
+    const conns = callNativeSync(() => this.#runtime.actorConns(this.#ctx));
+    return conns.map(
+      (c) => [this.#runtime.connId(c), this.#connToAdapter(c)]
+    )[Symbol.iterator]();
+  }
+  forEach(callback, thisArg) {
+    const conns = callNativeSync(() => this.#runtime.actorConns(this.#ctx));
+    for (const conn of conns) {
+      const id = this.#runtime.connId(conn);
+      callback.call(thisArg, this.#connToAdapter(conn), id, this);
+    }
+  }
+  [Symbol.iterator]() {
+    return this.entries();
+  }
+  __init3() {this[Symbol.toStringTag] = "NativeConnectionMap"}
+}, _class3);
 var ActorContextHandleAdapter = class {
   #runtime;
   #ctx;
@@ -3761,9 +3984,9 @@ var ActorContextHandleAdapter = class {
   #abortSignalCleanup;
   #client;
   #clientFactory;
+  #connMap;
   #databaseProvider;
   #db;
-  #dbProxy;
   #dispatchCancelToken;
   #kv;
   #queue;
@@ -3786,7 +4009,7 @@ var ActorContextHandleAdapter = class {
       this.#databaseProvider = databaseProvider;
     }
     this.#request = request;
-    this[_chunkWQ4HNA4Wcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
+    this[_chunkWHYBAEWGcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
       this
     );
   }
@@ -3806,30 +4029,22 @@ var ActorContextHandleAdapter = class {
     if (!this.#databaseProvider) {
       throw databaseNotConfiguredError();
     }
-    if (!this.#dbProxy) {
-      this.#dbProxy = new Proxy(
-        {},
-        {
-          get: (_target, property) => {
-            if (property === "then") {
-              return void 0;
-            }
-            return async (...args) => {
-              const client = await this.ensureDatabaseClient();
-              const value = Reflect.get(
-                client,
-                property
-              );
-              if (typeof value !== "function") {
-                return value;
-              }
-              return await value.apply(client, args);
-            };
-          }
-        }
-      );
+    if (this.#db) {
+      return this.#db;
     }
-    return this.#dbProxy;
+    const runtimeState = getNativeRuntimeState(this.#runtime, this.#ctx);
+    const cachedClient = runtimeState.databaseClient;
+    if (cachedClient) {
+      this.#db = cachedClient.client;
+      return this.#db;
+    }
+    throw databaseClientNotReadyError();
+  }
+  [_chunkWHYBAEWGcjs.RAW_STATE_SYMBOL]() {
+    if (!this.#stateEnabled) {
+      throw stateNotEnabledError();
+    }
+    return this.#readState();
   }
   get state() {
     if (!this.#stateEnabled) {
@@ -3844,8 +4059,9 @@ var ActorContextHandleAdapter = class {
         (nextValue) => {
           this.#writeState(nextValue, { scheduleSave: true });
         },
-        () => {
+        (newValue) => {
           this.#assertCanMutateState();
+          _chunkSS56HFM2cjs.assertJsonCompatValue.call(void 0, newValue);
         }
       );
     }
@@ -3856,6 +4072,7 @@ var ActorContextHandleAdapter = class {
       throw stateNotEnabledError();
     }
     this.#assertCanMutateState();
+    _chunkSS56HFM2cjs.assertJsonCompatValue.call(void 0, value);
     this.#writeState(value, { scheduleSave: true });
   }
   initializeState(value) {
@@ -3912,25 +4129,14 @@ var ActorContextHandleAdapter = class {
     return callNativeSync(() => this.#runtime.actorRegion(this.#ctx));
   }
   get conns() {
-    return new Map(
-      callNativeSync(() => this.#runtime.actorConns(this.#ctx)).map(
-        (conn) => [
-          this.#runtime.connId(conn),
-          new NativeConnAdapter(
-            this.#runtime,
-            conn,
-            this.#schemas,
-            this.#ctx,
-            (connId) => callNativeSync(
-              () => this.#runtime.actorQueueHibernationRemoval(
-                this.#ctx,
-                connId
-              )
-            )
-          )
-        ]
-      )
-    );
+    if (!this.#connMap) {
+      this.#connMap = new NativeConnectionMap(
+        this.#runtime,
+        this.#ctx,
+        this.#schemas
+      );
+    }
+    return this.#connMap;
   }
   get log() {
     return logger2();
@@ -4131,7 +4337,7 @@ var ActorContextHandleAdapter = class {
     const trackedPromise = Promise.resolve(promise).catch((error) => {
       logger2().warn({
         msg: "keepAwake promise rejected",
-        error: _chunkJ5P6S2LCcjs.stringifyError.call(void 0, error)
+        error: _chunkVNMIAPPFcjs.stringifyError.call(void 0, error)
       });
     }).then(() => null);
     try {
@@ -4207,6 +4413,7 @@ var ActorContextHandleAdapter = class {
     return false;
   }
   sleep() {
+    this.#flushStateChange();
     callNativeSync(() => this.#runtime.actorSleep(this.#ctx));
   }
   destroy() {
@@ -4512,9 +4719,9 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   const actionHandlers = Object.fromEntries(
     Object.entries(_nullishCoalesce(config.actions, () => ( {}))).map(([name, handler]) => [name, handler])
   );
-  const createClient = () => _chunkTE4VCDNYcjs.createClientWithDriver.call(void 0, 
-    new (0, _chunkTE4VCDNYcjs.RemoteEngineControlClient)(
-      _chunkTE4VCDNYcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
+  const createClient = () => _chunk2U6RLFKXcjs.createClientWithDriver.call(void 0, 
+    new (0, _chunk2U6RLFKXcjs.RemoteEngineControlClient)(
+      _chunk2U6RLFKXcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
     ),
     { encoding: "bare" }
   );
@@ -4524,7 +4731,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   ), () => ( false));
   const getNativeWorkflowInspector = (ctx) => {
     var _a2;
-    return (_a2 = _chunkWQ4HNA4Wcjs.getRunInspectorConfig.call(void 0, 
+    return (_a2 = _chunkWHYBAEWGcjs.getRunInspectorConfig.call(void 0, 
       config.run,
       callNativeSync(() => runtime.actorId(ctx))
     )) == null ? void 0 : _a2.workflow;
@@ -4577,7 +4784,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       }
     });
     const errorResponse = (error, status) => {
-      const rivetError = _chunkQAZLM4WTcjs.toRivetError.call(void 0, error);
+      const rivetError = _chunkKORQB2IRcjs.toRivetError.call(void 0, error);
       return jsonResponse(
         {
           group: rivetError.group,
@@ -4687,6 +4894,10 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             isWorkflowEnabled: getNativeWorkflowInspector(ctx) !== void 0
           });
         } catch (error) {
+          logger2().error({
+            msg: "error replaying workflow history",
+            error
+          });
           return errorResponse(error);
         }
       }
@@ -4818,7 +5029,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         const action = actionHandlers[actionName];
         if (!action) {
           return errorResponse(
-            new (0, _chunkQAZLM4WTcjs.RivetError)(
+            new (0, _chunkKORQB2IRcjs.RivetError)(
               "action",
               "action_not_found",
               `Action ${actionName} not found`
@@ -4838,6 +5049,10 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           );
           return jsonResponse({ output });
         } catch (error) {
+          logger2().error({
+            msg: "Error handling inspector action request",
+            error
+          });
           return errorResponse(error);
         }
       }
@@ -4851,6 +5066,10 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         { status: 404 }
       );
     } catch (error) {
+      logger2().error({
+        msg: "Error handling inspector request",
+        error
+      });
       return errorResponse(error);
     } finally {
       await actorCtx.dispose();
@@ -4973,24 +5192,30 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       async (error, payload) => {
         const { ctx } = unwrapTsfnPayload(error, payload);
         const actorCtx = makeActorCtx(ctx);
+        const saveActorState = async () => {
+          if (runtime.kind === "wasm") {
+            await runtime.actorSaveState(
+              ctx,
+              actorCtx.serializeForTick("save")
+            );
+          } else {
+            await actorCtx.saveState({
+              immediate: true
+            });
+          }
+        };
         try {
           if (onSleep) {
-            try {
-              await onSleep(actorCtx);
-            } finally {
-              if (runtime.kind === "wasm") {
-                await runtime.actorSaveState(
-                  ctx,
-                  actorCtx.serializeForTick("save")
-                );
-              } else {
-                await actorCtx.saveState({ immediate: true });
-              }
-            }
+            await onSleep(actorCtx);
           }
+          await saveActorState();
         } finally {
           try {
-            await cleanupNativeSleepRuntimeState(runtime, ctx);
+            await cleanupNativeSleepRuntimeState(
+              runtime,
+              ctx,
+              saveActorState
+            );
           } finally {
             await actorCtx.dispose();
           }
@@ -5155,7 +5380,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             );
           }
           if (!result) {
-            throw _chunkQAZLM4WTcjs.forbiddenError.call(void 0, );
+            throw _chunkKORQB2IRcjs.forbiddenError.call(void 0, );
           }
         } finally {
           await actorCtx.dispose();
@@ -5201,7 +5426,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               new Response(null, { status: 404 })
             );
           }
-          const rawConnParams = jsRequest.headers.get(_chunkTE4VCDNYcjs.HEADER_CONN_PARAMS);
+          const rawConnParams = jsRequest.headers.get(_chunk2U6RLFKXcjs.HEADER_CONN_PARAMS);
           let requestCtx;
           let conn;
           try {
@@ -5266,7 +5491,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       }
     ) : void 0,
     run: (() => {
-      const run = _chunkWQ4HNA4Wcjs.getRunFunction.call(void 0, config.run);
+      const run = _chunkWHYBAEWGcjs.getRunFunction.call(void 0, config.run);
       if (!run) {
         return void 0;
       }
@@ -5285,7 +5510,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         }
       );
     })(),
-    getWorkflowHistory: _chunkWQ4HNA4Wcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    getWorkflowHistory: _chunkWHYBAEWGcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
       async (error, payload) => {
         var _a2;
         const { ctx } = unwrapTsfnPayload(error, payload);
@@ -5293,7 +5518,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         return history == null ? void 0 : encodeValue(history);
       }
     ) : void 0,
-    replayWorkflow: _chunkWQ4HNA4Wcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    replayWorkflow: _chunkWHYBAEWGcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
       async (error, payload) => {
         const { ctx, entryId } = unwrapTsfnPayload(
           error,
@@ -5368,7 +5593,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             name
           );
           if (canPublish && !await canPublish(actorCtx)) {
-            throw _chunkQAZLM4WTcjs.forbiddenError.call(void 0, );
+            throw _chunkKORQB2IRcjs.forbiddenError.call(void 0, );
           }
           const decodedBody = decodeValue(body);
           if (wait) {
@@ -5428,7 +5653,7 @@ async function buildServeConfig(config) {
     poolName: config.envoy.poolName,
     handleInspectorHttpInRuntime: true,
     serverlessBasePath: config.serverless.basePath,
-    serverlessPackageVersion: _chunkJ5P6S2LCcjs.VERSION,
+    serverlessPackageVersion: _chunkVNMIAPPFcjs.VERSION,
     serverlessClientEndpoint: config.publicEndpoint,
     serverlessClientNamespace: config.publicNamespace,
     serverlessClientToken: config.publicToken,
@@ -5440,13 +5665,13 @@ async function buildServeConfig(config) {
     serveConfig.engineBinaryPath = getEnginePath();
   }
   if ((_a = config.test) == null ? void 0 : _a.enabled) {
-    serveConfig.inspectorTestToken = _nullishCoalesce(_chunkJ5P6S2LCcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
+    serveConfig.inspectorTestToken = _nullishCoalesce(_chunkVNMIAPPFcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
   }
   return serveConfig;
 }
 async function buildRegistryWithRuntime(config, runtime) {
   var _a;
-  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunkJ5P6S2LCcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
+  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunkVNMIAPPFcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
     trySetProcessEnv("_RIVET_TEST_INSPECTOR_TOKEN", "token");
   }
   const registry = runtime.createRegistry();
@@ -5488,6 +5713,7 @@ function finishShutdownSignal(signal) {
 }
 var Registry = class {
   #config;
+  #buildConfiguredRegistry;
   
   get config() {
     return this.#config;
@@ -5503,8 +5729,9 @@ var Registry = class {
   #shutdownInstalled = false;
   #shutdownInFlight = null;
   #signalHandlers = {};
-  constructor(config) {
+  constructor(config, deps) {
     this.#config = config;
+    this.#buildConfiguredRegistry = _nullishCoalesce((deps == null ? void 0 : deps.buildConfiguredRegistry), () => ( buildConfiguredRegistry));
     this.routes = {
       health: () => this.#healthRoute(),
       metadata: () => this.#metadataRoute(),
@@ -5514,12 +5741,12 @@ var Registry = class {
   #ensureServerlessPoolConfigured(config) {
     if (!config.configurePool) return void 0;
     if (!this.#configureServerlessPoolPromise) {
-      this.#configureServerlessPoolPromise = configureServerlessPool(config).catch(
-        (error) => {
-          this.#configureServerlessPoolPromise = void 0;
-          throw error;
-        }
-      );
+      this.#configureServerlessPoolPromise = configureServerlessPool(
+        config
+      ).catch((error) => {
+        this.#configureServerlessPoolPromise = void 0;
+        throw error;
+      });
       this.#configureServerlessPoolPromise.catch(() => {
       });
     }
@@ -5540,7 +5767,7 @@ var Registry = class {
     const config = this.parseConfig();
     this.#printWelcome(config, "serverless");
     if (!this.#runtimeServerlessPromise) {
-      this.#runtimeServerlessPromise = buildConfiguredRegistry(config);
+      this.#runtimeServerlessPromise = this.#buildConfiguredRegistry(config);
     }
     const { runtime, registry, serveConfig } = await this.#runtimeServerlessPromise;
     const isStartRequest = isServerlessStartRequest(
@@ -5555,7 +5782,7 @@ var Registry = class {
     if (isStartRequest) {
       try {
         await this.#ensureServerlessPoolConfigured(config);
-      } catch (error) {
+      } catch (_error) {
         return new Response(
           JSON.stringify({
             group: "guard",
@@ -5669,7 +5896,7 @@ var Registry = class {
     if (isMetadataRequest && !isEngineMetadataRequest) {
       try {
         await this.#ensureServerlessPoolConfigured(config);
-      } catch (error) {
+      } catch (_error) {
         return new Response(
           JSON.stringify({
             group: "guard",
@@ -5702,6 +5929,33 @@ var Registry = class {
       fetch: (request) => this.handler(request)
     };
   }
+  /**
+   * Starts an HTTP server that dispatches every request through the
+   * serverless handler. Uses `crossPlatformServe` to pick the right
+   * runtime (Node, Bun, Deno).
+   *
+   * @param opts.port      Port to listen on. Defaults to 3000.
+   * @param opts.publicDir If set, serves static files from this directory
+   *                       before falling through to the registry handler.
+   *
+   * @example
+   * ```ts
+   * await registry.listen();
+   * await registry.listen({ port: 8080, publicDir: "./public" });
+   * ```
+   */
+  async listen(opts = {}) {
+    const port = _nullishCoalesce(opts.port, () => ( 3e3));
+    const config = this.parseConfig();
+    const runtime = _chunkVNMIAPPFcjs.detectRuntime.call(void 0, );
+    const app = new (0, _hono.Hono)();
+    if (opts.publicDir) {
+      const serveStatic = await loadRuntimeServeStatic(runtime);
+      app.use("*", serveStatic({ root: opts.publicDir }));
+    }
+    app.all("*", (c) => this.handler(c.req.raw));
+    await crossPlatformServe(config, port, app, runtime);
+  }
   /**
    * Returns a health response suitable for mounting in a user-owned router.
    */
@@ -5711,7 +5965,7 @@ var Registry = class {
       return jsonRouteResponse(503, {
         status: "not_started",
         runtime: "rivetkit",
-        version: _chunkJ5P6S2LCcjs.VERSION
+        version: _chunkVNMIAPPFcjs.VERSION
       });
     }
     const { runtime, registry } = configured;
@@ -5719,7 +5973,7 @@ var Registry = class {
       return jsonRouteResponse(501, {
         status: "unsupported",
         runtime: "rivetkit",
-        version: _chunkJ5P6S2LCcjs.VERSION
+        version: _chunkVNMIAPPFcjs.VERSION
       });
     }
     const response = await runtime.registryHealth(registry);
@@ -5791,20 +6045,20 @@ var Registry = class {
    */
   #startEnvoy(config, printWelcome) {
     if (!this.#runtimeServePromise) {
-      const configuredRegistryPromise = buildConfiguredRegistry(config);
+      const configuredRegistryPromise = this.#buildConfiguredRegistry(config);
       this.#runtimeServeConfiguredPromise = configuredRegistryPromise;
       this.#runtimeServePromise = configuredRegistryPromise.then(async ({ runtime, registry, serveConfig }) => {
         await runtime.serveRegistry(registry, serveConfig);
-      }).catch((err) => {
-        logger2().warn({ err }, "runtime registry serve errored");
+      }).catch((error) => {
+        logger2().warn({ error }, "runtime registry serve errored");
       });
-      this.#installSignalHandlers(config, configuredRegistryPromise);
+      this.#installSignalHandlers(config);
     }
     if (printWelcome) {
       this.#printWelcome(config, "serverful");
     }
   }
-  #installSignalHandlers(config, configuredRegistryPromise) {
+  #installSignalHandlers(config) {
     var _a;
     if (this.#shutdownInstalled) return;
     if ((_a = config.shutdown) == null ? void 0 : _a.disableSignalHandlers) return;
@@ -5813,58 +6067,86 @@ var Registry = class {
     }
     this.#shutdownInstalled = true;
     const install = (signal) => {
-      const handler = () => this.#onShutdownSignal(
-        signal,
-        config,
-        configuredRegistryPromise
-      );
+      const handler = () => this.#onShutdownSignal(signal, config);
       this.#signalHandlers[signal] = handler;
       process.on(signal, handler);
     };
     install("SIGINT");
     install("SIGTERM");
   }
-  #onShutdownSignal(signal, config, configuredRegistryPromise) {
+  #onShutdownSignal(signal, config) {
     if (this.#shutdownInFlight !== null) {
       this.#removeSignalHandlers();
       finishShutdownSignal(signal);
       return;
     }
-    this.#shutdownInFlight = this.#runShutdown(
-      signal,
-      config,
-      configuredRegistryPromise
-    ).catch((err) => {
+    this.#shutdownInFlight = this.#drain(config).catch((err) => {
+      logger2().warn({ err }, "shutdown error");
+    }).then(() => {
+      this.#removeSignalHandlers();
+      finishShutdownSignal(signal);
+    });
+  }
+  /**
+   * Gracefully drains all live registries.
+   *
+   * Programmatic counterpart to the SIGINT/SIGTERM handlers: tears down
+   * every live `CoreRegistry` (both `start()` and `handler()` modes) and
+   * waits for the serve promise to resolve, all bounded by the shutdown
+   * grace period. Unlike a signal-driven shutdown, this does not re-raise a
+   * signal or exit the process. The caller owns process lifetime.
+   *
+   * Idempotent: concurrent or repeated calls share a single drain. Safe to
+   * call even if nothing has been started.
+   *
+   * @example
+   * ```ts
+   * const registry = setup({ use: { counter } });
+   * registry.start();
+   * // ...later, on your own shutdown trigger:
+   * await registry.shutdown();
+   * ```
+   */
+  async shutdown() {
+    if (this.#shutdownInFlight !== null) return this.#shutdownInFlight;
+    const config = this.parseConfig();
+    this.#removeSignalHandlers();
+    this.#shutdownInFlight = this.#drain(config).catch((err) => {
       logger2().warn({ err }, "shutdown error");
     });
+    return this.#shutdownInFlight;
   }
-  async #runShutdown(signal, config, configuredRegistryPromise) {
+  async #drain(config) {
     var _a;
-    const gracePeriodMs = await _asyncNullishCoalesce(await _asyncNullishCoalesce(((_a = config.shutdown) == null ? void 0 : _a.gracePeriodMs), async () => ( await this.#actorStopThresholdMs(configuredRegistryPromise))), async () => ( 30 * 60 * 1e3));
+    const modeAPromise = this.#runtimeServeConfiguredPromise;
+    const modeBPromise = this.#runtimeServerlessPromise;
+    const gracePeriodMs = await _asyncNullishCoalesce(await _asyncNullishCoalesce(((_a = config.shutdown) == null ? void 0 : _a.gracePeriodMs), async () => ( await this.#actorStopThresholdMs(_nullishCoalesce(modeAPromise, () => ( modeBPromise))))), async () => ( 30 * 60 * 1e3));
     const drain = async () => {
-      const registries = [
-        (async () => {
-          try {
-            const { runtime, registry } = await configuredRegistryPromise;
-            await runtime.shutdownRegistry(registry);
-          } catch (err) {
-            logger2().warn(
-              { err },
-              "runtime registry shutdown errored (mode A)"
-            );
-          }
-        })()
-      ];
-      const runtimeServerlessPromise = this.#runtimeServerlessPromise;
-      if (runtimeServerlessPromise !== void 0) {
+      const registries = [];
+      if (modeAPromise !== void 0) {
         registries.push(
           (async () => {
             try {
-              const { runtime, registry } = await runtimeServerlessPromise;
+              const { runtime, registry } = await modeAPromise;
               await runtime.shutdownRegistry(registry);
             } catch (err) {
               logger2().warn(
                 { err },
+                "runtime registry shutdown errored (mode A)"
+              );
+            }
+          })()
+        );
+      }
+      if (modeBPromise !== void 0) {
+        registries.push(
+          (async () => {
+            try {
+              const { runtime, registry } = await modeBPromise;
+              await runtime.shutdownRegistry(registry);
+            } catch (err) {
+              logger2().warn(
+                { error: err },
                 "runtime registry shutdown errored (mode B)"
               );
             }
@@ -5886,11 +6168,10 @@ var Registry = class {
         }
       )
     ]);
-    this.#removeSignalHandlers();
-    finishShutdownSignal(signal);
   }
   async #actorStopThresholdMs(configuredRegistryPromise) {
     var _a;
+    if (configuredRegistryPromise === void 0) return void 0;
     try {
       const { runtime, registry } = await configuredRegistryPromise;
       const thresholdMs = await ((_a = runtime.registryActorStopThresholdMs) == null ? void 0 : _a.call(runtime, registry));
@@ -5938,7 +6219,7 @@ var Registry = class {
     };
     console.log();
     console.log(
-      `  RivetKit ${_chunkJ5P6S2LCcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
+      `  RivetKit ${_chunkVNMIAPPFcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
     );
     if (config.namespace !== "default") {
       logLine("Namespace", config.namespace);
@@ -6006,5 +6287,5 @@ function setup(input) {
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunkTE4VCDNYcjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunkQAZLM4WTcjs.RivetError; exports.ActorsSchema = ActorsSchema; exports.DocConfigurePoolSchema = DocConfigurePoolSchema; exports.DocEnvoyConfigSchema = DocEnvoyConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocSqliteConfigSchema = DocSqliteConfigSchema; exports.InlineWebSocketAdapter = InlineWebSocketAdapter; exports.Registry = Registry; exports.RegistryConfigSchema = RegistryConfigSchema; exports.RivetError = _chunkQAZLM4WTcjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunkQAZLM4WTcjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunkTE4VCDNYcjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunkJ5P6S2LCcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunkJ5P6S2LCcjs.toUint8Array;
+exports.ALLOWED_PUBLIC_HEADERS = _chunk2U6RLFKXcjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunkKORQB2IRcjs.RivetError; exports.ActorsSchema = ActorsSchema; exports.DocConfigurePoolSchema = DocConfigurePoolSchema; exports.DocEnvoyConfigSchema = DocEnvoyConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocSqliteConfigSchema = DocSqliteConfigSchema; exports.InlineWebSocketAdapter = InlineWebSocketAdapter; exports.Registry = Registry; exports.RegistryConfigSchema = RegistryConfigSchema; exports.RivetError = _chunkKORQB2IRcjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunkKORQB2IRcjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunk2U6RLFKXcjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunkVNMIAPPFcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunkVNMIAPPFcjs.toUint8Array;
 //# sourceMappingURL=mod.cjs.map