rivetkit 2.3.0-rc.6 → 2.3.0-rc.7

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 _chunkSRNOPUC6cjs = require('./chunk-SRNOPUC6.cjs');
 
-var _chunkHTR4YLNTcjs = require('./chunk-HTR4YLNT.cjs');
 
+var _chunk55E7IR6Dcjs = require('./chunk-55E7IR6D.cjs');
 
 
 
@@ -18,7 +18,8 @@ var _chunkHTR4YLNTcjs = require('./chunk-HTR4YLNT.cjs');
 
 
 
-var _chunkENK7C66Gcjs = require('./chunk-ENK7C66G.cjs');
+
+var _chunkZGPX6KAHcjs = require('./chunk-ZGPX6KAH.cjs');
 
 
 
@@ -31,9 +32,11 @@ var _chunkLD5YASJUcjs = require('./chunk-LD5YASJU.cjs');
 
 
 
-var _chunkLIXXFXVRcjs = require('./chunk-LIXXFXVR.cjs');
+
+var _chunkFEOG44WHcjs = require('./chunk-FEOG44WH.cjs');
 require('./chunk-2NDZ7JCR.cjs');
-require('./chunk-JALSAX7Z.cjs');
+require('./chunk-SJLPZEA3.cjs');
+
 
 
 
@@ -57,7 +60,7 @@ require('./chunk-JALSAX7Z.cjs');
 
 
 
-var _chunkK5BA2LEOcjs = require('./chunk-K5BA2LEO.cjs');
+var _chunkX6HIFXNKcjs = require('./chunk-X6HIFXNK.cjs');
 
 
 
@@ -68,11 +71,11 @@ var _chunkK5BA2LEOcjs = require('./chunk-K5BA2LEO.cjs');
 
 
 
-var _chunkQAZLM4WTcjs = require('./chunk-QAZLM4WT.cjs');
+var _chunk63WNTDRCcjs = require('./chunk-63WNTDRC.cjs');
 
 // src/actor/log.ts
 function loggerWithoutContext() {
-  return _chunkK5BA2LEOcjs.getLogger.call(void 0, "actor-runtime");
+  return _chunkX6HIFXNKcjs.getLogger.call(void 0, "actor-runtime");
 }
 
 // src/actor/definition.ts
@@ -105,7 +108,7 @@ function actor(input) {
   warnDeprecatedShutdownTimeoutKeys(
     input == null ? void 0 : input.options
   );
-  const config = _chunkWQ4HNA4Wcjs.ActorConfigSchema.parse(input);
+  const config = _chunkSRNOPUC6cjs.ActorConfigSchema.parse(input);
   return new ActorDefinition(config);
 }
 function isStaticActorDefinition(definition) {
@@ -187,7 +190,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 _chunk63WNTDRCcjs.unsupportedFeature.call(void 0, "async schema validation");
     }
     if (result.issues) {
       return { success: false, issues: [...result.issues] };
@@ -204,7 +207,7 @@ function validateSchemaSync(schemas, key, data) {
 var _ws = require('hono/ws');
 var _virtualwebsocket = require('@rivetkit/virtual-websocket');
 function logger() {
-  return _chunkK5BA2LEOcjs.getLogger.call(void 0, "inline-websocket-adapter");
+  return _chunkX6HIFXNKcjs.getLogger.call(void 0, "inline-websocket-adapter");
 }
 var InlineWebSocketAdapter = class {
   #handler;
@@ -314,24 +317,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 +347,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 +357,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 _chunkK5BA2LEOcjs.getLogger.call(void 0, "registry");
+  return _chunkX6HIFXNKcjs.getLogger.call(void 0, "registry");
 }
 
 // src/serverless/configure.ts
@@ -397,8 +403,8 @@ async function configureServerlessPool(config) {
         throw new Error("configurePool is required for serverless configuration");
       }
       const customConfig = config.configurePool;
-      const clientConfig = _chunkENK7C66Gcjs.convertRegistryConfigToClientConfig.call(void 0, config);
-      const dcsRes = await _chunkENK7C66Gcjs.getDatacenters.call(void 0, clientConfig);
+      const clientConfig = _chunkZGPX6KAHcjs.convertRegistryConfigToClientConfig.call(void 0, config);
+      const dcsRes = await _chunkZGPX6KAHcjs.getDatacenters.call(void 0, clientConfig);
       const poolName = _nullishCoalesce(customConfig.name, () => ( "default"));
       const serverlessToken = _nullishCoalesce(config.token, () => ( config.publicToken));
       const headers = {
@@ -420,7 +426,7 @@ async function configureServerlessPool(config) {
         metadata: _nullishCoalesce(customConfig.metadata, () => ( {})),
         drain_on_version_upgrade: _nullishCoalesce(customConfig.drainOnVersionUpgrade, () => ( true))
       };
-      await _chunkENK7C66Gcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
+      await _chunkZGPX6KAHcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
         datacenters: Object.fromEntries(
           dcsRes.datacenters.map((dc) => [dc.name, serverlessConfig])
         )
@@ -437,7 +443,7 @@ async function configureServerlessPool(config) {
       logger2().warn({
         msg: "serverless pool configuration attempt failed",
         attempts,
-        error: _chunkK5BA2LEOcjs.stringifyError.call(void 0, error)
+        error: _chunkX6HIFXNKcjs.stringifyError.call(void 0, error)
       });
       await sleep(CONFIGURE_RETRY_DELAY_MS);
     }
@@ -445,11 +451,150 @@ async function configureServerlessPool(config) {
   logger2().error({
     msg: "failed to configure serverless pool, validate endpoint is configured correctly then restart this process",
     attempts,
-    error: _chunkK5BA2LEOcjs.stringifyError.call(void 0, lastError)
+    error: _chunkX6HIFXNKcjs.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 = _chunkX6HIFXNKcjs.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: _chunkX6HIFXNKcjs.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: _chunkX6HIFXNKcjs.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: _chunkX6HIFXNKcjs.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: _chunkX6HIFXNKcjs.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 +602,20 @@ var _zod = require('zod');
 var _v4 = require('zod/v4');
 var warnedMissingVersion = false;
 var EnvoyConfigSchema = _v4.z.object({
-  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunkK5BA2LEOcjs.getRivetPool.call(void 0, ), () => ( "default"))),
+  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunkX6HIFXNKcjs.getRivetPool.call(void 0, ), () => ( "default"))),
   version: _v4.z.number().default(() => {
-    const version = _chunkK5BA2LEOcjs.getRivetEnvoyVersion.call(void 0, );
+    const version = _chunkX6HIFXNKcjs.getRivetEnvoyVersion.call(void 0, );
     if (version !== void 0) return version;
-    if (_chunkK5BA2LEOcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
+    if (_chunkX6HIFXNKcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
       warnedMissingVersion = true;
-      _chunkK5BA2LEOcjs.getLogger.call(void 0, "rivetkit").error(
+      _chunkX6HIFXNKcjs.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(_chunkK5BA2LEOcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
+  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunkX6HIFXNKcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
   envoyKey: _v4.z.string().optional()
 });
 
@@ -508,7 +653,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, () => ( _chunkK5BA2LEOcjs.getRivetPublicEndpoint.call(void 0, )))),
+  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkX6HIFXNKcjs.getRivetPublicEndpoint.call(void 0, )))),
   /**
    * Token that clients should use when connecting via the public endpoint.
    *
@@ -516,7 +661,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, () => ( _chunkK5BA2LEOcjs.getRivetPublicToken.call(void 0, ))))
+  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkX6HIFXNKcjs.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 +715,7 @@ var RegistryConfigSchema = _zod.z.object({
    * Runtime binding to use for RivetKit core.
    * */
   runtime: RuntimeKindSchema.optional().transform((val, ctx) => {
-    const rawRuntime = _nullishCoalesce(val, () => ( _chunkK5BA2LEOcjs.getRivetkitRuntime.call(void 0, )));
+    const rawRuntime = _nullishCoalesce(val, () => ( _chunkX6HIFXNKcjs.getRivetkitRuntime.call(void 0, )));
     if (rawRuntime === void 0) {
       return "auto";
     }
@@ -607,7 +752,7 @@ var RegistryConfigSchema = _zod.z.object({
    * */
   logging: _zod.z.object({
     baseLogger: _zod.z.custom().optional(),
-    level: _chunkK5BA2LEOcjs.LogLevelSchema.optional()
+    level: _chunkX6HIFXNKcjs.LogLevelSchema.optional()
   }).optional().default(() => ({})),
   // MARK: Routing
   // // This is a function to allow for lazy configuration of upgradeWebSocket on the
@@ -625,9 +770,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, () => ( _chunkK5BA2LEOcjs.getRivetEngine.call(void 0, ))), () => ( _chunkK5BA2LEOcjs.getRivetEndpoint.call(void 0, )))),
-  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkK5BA2LEOcjs.getRivetToken.call(void 0, )))),
-  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkK5BA2LEOcjs.getRivetNamespace.call(void 0, )))),
+  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunkX6HIFXNKcjs.getRivetEngine.call(void 0, ))), () => ( _chunkX6HIFXNKcjs.getRivetEndpoint.call(void 0, )))),
+  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkX6HIFXNKcjs.getRivetToken.call(void 0, )))),
+  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkX6HIFXNKcjs.getRivetNamespace.call(void 0, )))),
   headers: _zod.z.record(_zod.z.string(), _zod.z.string()).optional().default({}),
   // MARK: Client
   // TODO:
@@ -667,9 +812,9 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Starts the full Rust engine process locally.
    */
-  startEngine: _zod.z.boolean().default(() => _chunkK5BA2LEOcjs.getRivetRunEngine.call(void 0, )),
+  startEngine: _zod.z.boolean().default(() => _chunkX6HIFXNKcjs.getRivetRunEngine.call(void 0, )),
   /** @experimental */
-  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkK5BA2LEOcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkK5BA2LEOcjs.VERSION))),
+  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkX6HIFXNKcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkX6HIFXNKcjs.VERSION))),
   /**
    * @experimental
    *
@@ -717,7 +862,7 @@ var RegistryConfigSchema = _zod.z.object({
   }))
 }).transform((config, ctx) => {
   var _a, _b, _c;
-  const isDevEnv = _chunkK5BA2LEOcjs.isDev.call(void 0, );
+  const isDevEnv = _chunkX6HIFXNKcjs.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 +872,7 @@ var RegistryConfigSchema = _zod.z.object({
     });
   }
   const sqlite = config.runtime === "wasm" && config.sqlite === void 0 ? { backend: "remote" } : config.sqlite;
-  const parsedEndpoint = config.endpoint ? _chunkENK7C66Gcjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedEndpoint = config.endpoint ? _chunkZGPX6KAHcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.endpoint,
     path: ["endpoint"],
     namespace: config.namespace,
@@ -751,7 +896,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 ? _chunkENK7C66Gcjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunkZGPX6KAHcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.serverless.publicEndpoint,
     path: ["serverless", "publicEndpoint"]
   }) : void 0;
@@ -786,7 +931,7 @@ 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 = _chunkSRNOPUC6cjs.getRunMetadata.call(void 0, definition.config.run);
       const metadata = {};
       metadata.icon = _nullishCoalesce(options.icon, () => ( runMeta.icon));
       metadata.name = _nullishCoalesce(options.name, () => ( runMeta.name));
@@ -879,7 +1024,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: _chunkK5BA2LEOcjs.LogLevelSchema.optional().describe(
+    level: _chunkX6HIFXNKcjs.LogLevelSchema.optional().describe(
       "Log level for RivetKit. Default: 'warn'"
     )
   }).optional().describe("Logging configuration."),
@@ -924,7 +1069,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 : _chunk63WNTDRCcjs.decodeBridgeRivetError.call(void 0, bridgeReason);
   if (bridged) {
     throw bridged;
   }
@@ -1154,6 +1299,29 @@ function lastInsertRowIdColumnName(sql) {
   return alias;
 }
 
+// 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/runtime.ts
 function normalizeRuntimeSqlExecuteResult(result) {
   return result;
@@ -1274,32 +1442,15 @@ var NapiCoreRuntime = (_class = class {
   async shutdownRegistry(registry) {
     await asNativeRegistry(registry).shutdown();
   }
-  async registryActorStopThresholdMs(registry) {
-    return await _asyncNullishCoalesce(await asNativeRegistry(registry).actorStopThresholdMs(), async () => ( void 0));
-  }
-  async registryHealth(registry) {
-    const response = await asNativeRegistry(registry).health();
+  async registryDiagnostics(registry) {
+    const diagnostics = await asNativeRegistry(registry).diagnostics();
     return {
-      status: response.status,
-      headers: response.headers,
-      body: response.body
+      mode: diagnostics.mode,
+      envoyActiveActorCount: diagnostics.envoyActiveActorCount
     };
   }
-  async registryMetadata(registry) {
-    const response = asNativeRegistry(registry).metadata();
-    return {
-      status: response.status,
-      headers: response.headers,
-      body: response.body
-    };
-  }
-  async registryMetrics(registry) {
-    const response = asNativeRegistry(registry).metrics();
-    return {
-      status: response.status,
-      headers: response.headers,
-      body: response.body
-    };
+  async registryActorStopThresholdMs(registry) {
+    return await _asyncNullishCoalesce(await asNativeRegistry(registry).actorStopThresholdMs(), async () => ( void 0));
   }
   async handleServerlessRequest(registry, req, onStreamEvent, cancelToken, config) {
     return await asNativeRegistry(registry).handleServerlessRequest(
@@ -1675,7 +1826,7 @@ function validateQueueComplete(schemas, name, response) {
   return result.data;
 }
 function validationError(target, issues) {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunk63WNTDRCcjs.RivetError)(
     "actor",
     "validation_error",
     `Invalid ${target}`,
@@ -1749,16 +1900,16 @@ function normalizeQueueMessage(message) {
 }
 function normalizeWasmBridgeError(error) {
   if (typeof error === "string") {
-    return _nullishCoalesce(_chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
+    return _nullishCoalesce(_chunk63WNTDRCcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
   }
   if (error instanceof Error) {
-    const bridged = _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error.message);
+    const bridged = _chunk63WNTDRCcjs.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 = _chunk63WNTDRCcjs.decodeBridgeRivetError.call(void 0, error.reason);
     if (bridged) {
       return bridged;
     }
@@ -1780,7 +1931,7 @@ function callWasmSync(invoke) {
   }
 }
 function unsupportedWasmMethod(method2) {
-  throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+  throw new (0, _chunk63WNTDRCcjs.RivetError)(
     "runtime",
     "unsupported",
     `Unsupported wasm runtime method: ${method2}`,
@@ -1841,18 +1992,8 @@ var WasmCoreRuntime = (_class2 = class {
   async shutdownRegistry(registry) {
     await callWasm(() => asWasmRegistry(registry).shutdown());
   }
-  async registryHealth() {
-    return {
-      status: 200,
-      headers: { "content-type": "application/json" },
-      body: new TextEncoder().encode(
-        JSON.stringify({
-          status: "ok",
-          runtime: "rivetkit",
-          version: "wasm"
-        })
-      )
-    };
+  async registryDiagnostics() {
+    return { mode: "wasm", envoyActiveActorCount: null };
   }
   async handleServerlessRequest(registry, req, onStreamEvent, cancelToken, config) {
     return await callHandleAsync(
@@ -2281,7 +2422,7 @@ function loadedRuntimeKind(runtime) {
     case "wasm":
       return "wasm";
   }
-  throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+  throw new (0, _chunk63WNTDRCcjs.RivetError)(
     "config",
     "unknown_runtime",
     "RivetKit runtime must be NAPI or wasm.",
@@ -2321,7 +2462,7 @@ function normalizeRuntimeConfigForKind(config, runtimeKind) {
     return config;
   }
   if (sqliteBackendForConfig(config) === "local") {
-    throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+    throw new (0, _chunk63WNTDRCcjs.RivetError)(
       "config",
       "wasm_local_sqlite",
       "WebAssembly runtime cannot use local SQLite. Use remote SQLite instead.",
@@ -2387,22 +2528,30 @@ function getNativeConnPersistState(runtime, ctx, conn) {
   return connState;
 }
 function stateMutationReentrantError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunk63WNTDRCcjs.RivetError)(
     "actor",
     "state_mutation_reentrant",
     "State mutations are not allowed inside onStateChange."
   );
 }
 function databaseNotConfiguredError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunk63WNTDRCcjs.RivetError)(
     "actor",
     "database_not_configured",
     "database is not configured for this actor",
     { public: true }
   );
 }
+function databaseClientNotReadyError() {
+  return new (0, _chunk63WNTDRCcjs.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, _chunk63WNTDRCcjs.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 +2559,7 @@ function stateNotEnabledError() {
   );
 }
 function nativeClientNotConfiguredError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunk63WNTDRCcjs.RivetError)(
     "native",
     "client_not_configured",
     "native actor client is not configured",
@@ -2418,7 +2567,7 @@ function nativeClientNotConfiguredError() {
   );
 }
 function nativeEndpointNotConfiguredError() {
-  return new (0, _chunkQAZLM4WTcjs.RivetError)(
+  return new (0, _chunk63WNTDRCcjs.RivetError)(
     "native",
     "endpoint_not_configured",
     "registry endpoint is required for native envoy startup",
@@ -2453,7 +2602,20 @@ function clearNativeRuntimeState(runtime, ctx) {
   callNativeSync(() => runtime.actorClearRuntimeState(ctx));
 }
 async function cleanupNativeSleepRuntimeState(runtime, ctx) {
-  await runtime.actorWaitForTrackedShutdownWork(ctx);
+  const waitStarted = Date.now();
+  const drained = await runtime.actorWaitForTrackedShutdownWork(ctx);
+  const waitMs = Date.now() - waitStarted;
+  if (drained) {
+    logger2().debug({
+      msg: "sleep cleanup: tracked shutdown work drained",
+      waitMs
+    });
+  } else {
+    logger2().warn({
+      msg: "sleep cleanup: shutdown deadline reached before tracked work drained; closing DB anyway",
+      waitMs
+    });
+  }
   await closeNativeDatabaseClient(runtime, ctx);
   await closeNativeSqlDatabase(runtime, ctx);
   clearNativeRuntimeState(runtime, ctx);
@@ -2558,10 +2720,10 @@ function decodeValue(value) {
   if (!value || value.length === 0) {
     return void 0;
   }
-  return _chunkLIXXFXVRcjs.decodeCborJsonCompat.call(void 0, value);
+  return _chunkFEOG44WHcjs.decodeCborJsonCompat.call(void 0, value);
 }
 function encodeValue(value) {
-  return _chunkLIXXFXVRcjs.encodeCborCompat.call(void 0, value);
+  return _chunkFEOG44WHcjs.encodeCborCompat.call(void 0, value);
 }
 function unwrapTsfnPayload(error, payload) {
   if (error !== null && error !== void 0) {
@@ -2571,16 +2733,16 @@ function unwrapTsfnPayload(error, payload) {
 }
 function normalizeNativeBridgeError(error) {
   if (typeof error === "string") {
-    return _nullishCoalesce(_chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
+    return _nullishCoalesce(_chunk63WNTDRCcjs.decodeBridgeRivetError.call(void 0, error), () => ( error));
   }
   if (error instanceof Error) {
-    const bridged = _chunkQAZLM4WTcjs.decodeBridgeRivetError.call(void 0, error.message);
+    const bridged = _chunk63WNTDRCcjs.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 = _chunk63WNTDRCcjs.decodeBridgeRivetError.call(void 0, error.reason);
     if (bridged) {
       return bridged;
     }
@@ -2588,14 +2750,14 @@ function normalizeNativeBridgeError(error) {
   return error;
 }
 function isStructuredBridgeError(error) {
-  if (error instanceof _chunkQAZLM4WTcjs.RivetError) {
+  if (error instanceof _chunk63WNTDRCcjs.RivetError) {
     return true;
   }
-  return _chunkQAZLM4WTcjs.isRivetErrorLike.call(void 0, error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
+  return _chunk63WNTDRCcjs.isRivetErrorLike.call(void 0, error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
 }
 function encodeNativeCallbackError(error) {
-  const structuredError = isStructuredBridgeError(error) ? error : _chunkK5BA2LEOcjs.deconstructError.call(void 0, error, true);
-  const bridgeError = new Error(_chunkQAZLM4WTcjs.encodeBridgeRivetError.call(void 0, structuredError), {
+  const structuredError = isStructuredBridgeError(error) ? error : _chunkX6HIFXNKcjs.deconstructError.call(void 0, error, true);
+  const bridgeError = new Error(_chunk63WNTDRCcjs.encodeBridgeRivetError.call(void 0, structuredError), {
     cause: error instanceof Error ? error : void 0
   });
   return Object.assign(bridgeError, {
@@ -2625,9 +2787,9 @@ function actorAbortedError() {
   });
 }
 function isClosedTaskRegistrationError(error) {
-  const metadata = error instanceof _chunkQAZLM4WTcjs.RivetError ? error.metadata : void 0;
+  const metadata = error instanceof _chunk63WNTDRCcjs.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 _chunk63WNTDRCcjs.RivetError && error.group === "core" && error.code === _chunk63WNTDRCcjs.INTERNAL_ERROR_CODE && typeof metadataError === "string" && /actor task registration is (closed|not configured)/.test(metadataError);
 }
 async function createCancellationTokenHandle(runtime, signal) {
   if (!signal) {
@@ -2650,7 +2812,7 @@ function decodeWorkflowCbor(data) {
     return null;
   }
   try {
-    return _chunkLIXXFXVRcjs.decodeCborCompat.call(void 0, new Uint8Array(data));
+    return _chunkFEOG44WHcjs.decodeCborCompat.call(void 0, new Uint8Array(data));
   } catch (e3) {
     return null;
   }
@@ -2755,7 +2917,7 @@ function serializeWorkflowHistoryForJson(data) {
   if (data === null) {
     return null;
   }
-  const history = _chunkHTR4YLNTcjs.decodeWorkflowHistoryTransport.call(void 0, data);
+  const history = _chunk55E7IR6Dcjs.decodeWorkflowHistoryTransport.call(void 0, data);
   return {
     nameRegistry: [...history.nameRegistry],
     entries: history.entries.map((entry) => ({
@@ -2833,44 +2995,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 +3030,7 @@ var NativeConnAdapter = class {
     this.#schemas = schemas;
     this.#ctx = ctx;
     this.#queueHibernationRemoval = queueHibernationRemoval;
-    this[_chunkWQ4HNA4Wcjs.CONN_STATE_MANAGER_SYMBOL] = {
+    this[_chunkSRNOPUC6cjs.CONN_STATE_MANAGER_SYMBOL] = {
       stateEnabled: true,
       get state() {
         return thisConn.state;
@@ -2923,13 +3047,19 @@ var NativeConnAdapter = class {
       decodeValue(this.#runtime.connParams(this.#conn))
     );
   }
+  [_chunkSRNOPUC6cjs.RAW_STATE_SYMBOL]() {
+    return this.#readState();
+  }
   get state() {
     const nextState = this.#readState();
     return createWriteThroughProxy(nextState, (nextValue) => {
       this.#writeState(nextValue, { writeNative: true });
+    }, (newValue) => {
+      _chunkFEOG44WHcjs.assertJsonCompatValue.call(void 0, newValue);
     });
   }
   set state(value) {
+    _chunkFEOG44WHcjs.assertJsonCompatValue.call(void 0, value);
     this.#writeState(value, { writeNative: true });
   }
   initializeState(value) {
@@ -3219,7 +3349,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, _chunk63WNTDRCcjs.RivetError)(
         "queue",
         "previous_message_not_completed",
         "Previous completable queue message is not completed. Call `message.complete(...)` before receiving the next message.",
@@ -3388,7 +3518,7 @@ var NativeQueueAdapter = class {
         }
         yield message;
       } catch (error) {
-        if (_chunkQAZLM4WTcjs.isRivetErrorLike.call(void 0, error) && error.group === "actor" && error.code === "aborted") {
+        if (_chunk63WNTDRCcjs.isRivetErrorLike.call(void 0, error) && error.group === "actor" && error.code === "aborted") {
           return;
         }
         throw error;
@@ -3409,7 +3539,7 @@ var NativeQueueAdapter = class {
       ...message,
       complete: async (response) => {
         if (typeof message.complete !== "function") {
-          throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+          throw new (0, _chunk63WNTDRCcjs.RivetError)(
             "queue",
             "complete_not_configured",
             `Queue '${message.name}' does not support completion responses.`,
@@ -3421,7 +3551,7 @@ var NativeQueueAdapter = class {
           );
         }
         if (completed) {
-          throw new (0, _chunkQAZLM4WTcjs.RivetError)(
+          throw new (0, _chunk63WNTDRCcjs.RivetError)(
             "queue",
             "already_completed",
             "Queue message was already completed.",
@@ -3753,6 +3883,70 @@ 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 +3955,9 @@ var ActorContextHandleAdapter = class {
   #abortSignalCleanup;
   #client;
   #clientFactory;
+  #connMap;
   #databaseProvider;
   #db;
-  #dbProxy;
   #dispatchCancelToken;
   #kv;
   #queue;
@@ -3786,7 +3980,7 @@ var ActorContextHandleAdapter = class {
       this.#databaseProvider = databaseProvider;
     }
     this.#request = request;
-    this[_chunkWQ4HNA4Wcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
+    this[_chunkSRNOPUC6cjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
       this
     );
   }
@@ -3806,30 +4000,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();
+  }
+  [_chunkSRNOPUC6cjs.RAW_STATE_SYMBOL]() {
+    if (!this.#stateEnabled) {
+      throw stateNotEnabledError();
+    }
+    return this.#readState();
   }
   get state() {
     if (!this.#stateEnabled) {
@@ -3841,8 +4027,9 @@ var ActorContextHandleAdapter = class {
       (nextValue) => {
         this.#writeState(nextValue, { scheduleSave: true });
       },
-      () => {
+      (newValue) => {
         this.#assertCanMutateState();
+        _chunkFEOG44WHcjs.assertJsonCompatValue.call(void 0, newValue);
       }
     );
   }
@@ -3851,6 +4038,7 @@ var ActorContextHandleAdapter = class {
       throw stateNotEnabledError();
     }
     this.#assertCanMutateState();
+    _chunkFEOG44WHcjs.assertJsonCompatValue.call(void 0, value);
     this.#writeState(value, { scheduleSave: true });
   }
   initializeState(value) {
@@ -3907,25 +4095,10 @@ 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();
@@ -4123,20 +4296,39 @@ var ActorContextHandleAdapter = class {
     });
   }
   keepAwake(promise) {
-    const trackedPromise = Promise.resolve(promise).catch((error) => {
-      logger2().warn({
-        msg: "keepAwake promise rejected",
-        error: _chunkK5BA2LEOcjs.stringifyError.call(void 0, error)
-      });
-    }).then(() => null);
+    const startedAt = Date.now();
+    logger2().debug({
+      msg: "keepAwake registered",
+      at: startedAt
+    });
+    const trackedPromise = Promise.resolve(promise).then(
+      () => {
+        logger2().debug({
+          msg: "keepAwake promise resolved",
+          durationMs: Date.now() - startedAt
+        });
+      },
+      (error) => {
+        logger2().warn({
+          msg: "keepAwake promise rejected",
+          durationMs: Date.now() - startedAt,
+          error: _chunkX6HIFXNKcjs.stringifyError.call(void 0, error)
+        });
+      }
+    ).then(() => null);
     try {
       callNativeSync(
         () => this.#runtime.actorKeepAwake(this.#ctx, trackedPromise)
       );
     } catch (error) {
-      if (!isClosedTaskRegistrationError(error)) {
-        throw error;
+      if (isClosedTaskRegistrationError(error)) {
+        logger2().warn({
+          msg: "keepAwake registration dropped (teardown already started); promise will not delay grace",
+          error: _chunkX6HIFXNKcjs.stringifyError.call(void 0, error)
+        });
+        return promise;
       }
+      throw error;
     }
     return promise;
   }
@@ -4485,9 +4677,9 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   const actionHandlers = Object.fromEntries(
     Object.entries(_nullishCoalesce(config.actions, () => ( {}))).map(([name, handler]) => [name, handler])
   );
-  const createClient = () => _chunkENK7C66Gcjs.createClientWithDriver.call(void 0, 
-    new (0, _chunkENK7C66Gcjs.RemoteEngineControlClient)(
-      _chunkENK7C66Gcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
+  const createClient = () => _chunkZGPX6KAHcjs.createClientWithDriver.call(void 0, 
+    new (0, _chunkZGPX6KAHcjs.RemoteEngineControlClient)(
+      _chunkZGPX6KAHcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
     ),
     { encoding: "bare" }
   );
@@ -4497,7 +4689,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   ), () => ( false));
   const getNativeWorkflowInspector = (ctx) => {
     var _a2;
-    return (_a2 = _chunkWQ4HNA4Wcjs.getRunInspectorConfig.call(void 0, 
+    return (_a2 = _chunkSRNOPUC6cjs.getRunInspectorConfig.call(void 0, 
       config.run,
       callNativeSync(() => runtime.actorId(ctx))
     )) == null ? void 0 : _a2.workflow;
@@ -4550,7 +4742,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       }
     });
     const errorResponse = (error, status) => {
-      const rivetError = _chunkQAZLM4WTcjs.toRivetError.call(void 0, error);
+      const rivetError = _chunk63WNTDRCcjs.toRivetError.call(void 0, error);
       return jsonResponse(
         {
           group: rivetError.group,
@@ -4660,6 +4852,10 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             isWorkflowEnabled: getNativeWorkflowInspector(ctx) !== void 0
           });
         } catch (error) {
+          logger2().error({
+            msg: "error replaying workflow history",
+            error
+          });
           return errorResponse(error);
         }
       }
@@ -4791,7 +4987,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         const action = actionHandlers[actionName];
         if (!action) {
           return errorResponse(
-            new (0, _chunkQAZLM4WTcjs.RivetError)(
+            new (0, _chunk63WNTDRCcjs.RivetError)(
               "action",
               "action_not_found",
               `Action ${actionName} not found`
@@ -4811,6 +5007,10 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           );
           return jsonResponse({ output });
         } catch (error) {
+          logger2().error({
+            msg: "Error handling inspector action request",
+            error
+          });
           return errorResponse(error);
         }
       }
@@ -4824,6 +5024,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();
@@ -5128,7 +5332,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             );
           }
           if (!result) {
-            throw _chunkQAZLM4WTcjs.forbiddenError.call(void 0, );
+            throw _chunk63WNTDRCcjs.forbiddenError.call(void 0, );
           }
         } finally {
           await actorCtx.dispose();
@@ -5174,7 +5378,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               new Response(null, { status: 404 })
             );
           }
-          const rawConnParams = jsRequest.headers.get(_chunkENK7C66Gcjs.HEADER_CONN_PARAMS);
+          const rawConnParams = jsRequest.headers.get(_chunkZGPX6KAHcjs.HEADER_CONN_PARAMS);
           let requestCtx;
           let conn;
           try {
@@ -5239,7 +5443,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       }
     ) : void 0,
     run: (() => {
-      const run = _chunkWQ4HNA4Wcjs.getRunFunction.call(void 0, config.run);
+      const run = _chunkSRNOPUC6cjs.getRunFunction.call(void 0, config.run);
       if (!run) {
         return void 0;
       }
@@ -5258,7 +5462,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         }
       );
     })(),
-    getWorkflowHistory: _chunkWQ4HNA4Wcjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    getWorkflowHistory: _chunkSRNOPUC6cjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
       async (error, payload) => {
         var _a2;
         const { ctx } = unwrapTsfnPayload(error, payload);
@@ -5266,7 +5470,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: _chunkSRNOPUC6cjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
       async (error, payload) => {
         const { ctx, entryId } = unwrapTsfnPayload(
           error,
@@ -5341,7 +5545,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             name
           );
           if (canPublish && !await canPublish(actorCtx)) {
-            throw _chunkQAZLM4WTcjs.forbiddenError.call(void 0, );
+            throw _chunk63WNTDRCcjs.forbiddenError.call(void 0, );
           }
           const decodedBody = decodeValue(body);
           if (wait) {
@@ -5401,7 +5605,7 @@ async function buildServeConfig(config) {
     poolName: config.envoy.poolName,
     handleInspectorHttpInRuntime: true,
     serverlessBasePath: config.serverless.basePath,
-    serverlessPackageVersion: _chunkK5BA2LEOcjs.VERSION,
+    serverlessPackageVersion: _chunkX6HIFXNKcjs.VERSION,
     serverlessClientEndpoint: config.publicEndpoint,
     serverlessClientNamespace: config.publicNamespace,
     serverlessClientToken: config.publicToken,
@@ -5413,13 +5617,13 @@ async function buildServeConfig(config) {
     serveConfig.engineBinaryPath = getEnginePath();
   }
   if ((_a = config.test) == null ? void 0 : _a.enabled) {
-    serveConfig.inspectorTestToken = _nullishCoalesce(_chunkK5BA2LEOcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
+    serveConfig.inspectorTestToken = _nullishCoalesce(_chunkX6HIFXNKcjs.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) && _chunkK5BA2LEOcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
+  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunkX6HIFXNKcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
     trySetProcessEnv("_RIVET_TEST_INSPECTOR_TOKEN", "token");
   }
   const registry = runtime.createRegistry();
@@ -5438,6 +5642,10 @@ async function buildRegistryWithRuntime(config, runtime) {
 }
 async function buildConfiguredRegistry(config) {
   const runtime = await loadConfiguredRuntime(config);
+  if (runtime.kind === "napi") {
+    const { startProcessMetrics } = await Promise.resolve().then(() => _interopRequireWildcard(require("./process-metrics-TYAGKCEJ.cjs")));
+    startProcessMetrics();
+  }
   return buildRegistryWithRuntime(
     normalizeRuntimeConfig(config, runtime),
     runtime
@@ -5461,7 +5669,6 @@ function finishShutdownSignal(signal) {
 }
 var Registry = class {
   #config;
-  
   get config() {
     return this.#config;
   }
@@ -5478,11 +5685,6 @@ var Registry = class {
   #signalHandlers = {};
   constructor(config) {
     this.#config = config;
-    this.routes = {
-      health: () => this.#healthRoute(),
-      metadata: () => this.#metadataRoute(),
-      prometheusMetrics: (request) => this.#prometheusMetricsRoute(request)
-    };
   }
   #ensureServerlessPoolConfigured(config) {
     if (!config.configurePool) return void 0;
@@ -5676,88 +5878,46 @@ var Registry = class {
     };
   }
   /**
-   * Returns a health response suitable for mounting in a user-owned router.
-   */
-  async #healthRoute() {
-    const configured = await this.#activeConfiguredRegistry();
-    if (!configured) {
-      return jsonRouteResponse(503, {
-        status: "not_started",
-        runtime: "rivetkit",
-        version: _chunkK5BA2LEOcjs.VERSION
-      });
-    }
-    const { runtime, registry } = configured;
-    if (!runtime.registryHealth) {
-      return jsonRouteResponse(501, {
-        status: "unsupported",
-        runtime: "rivetkit",
-        version: _chunkK5BA2LEOcjs.VERSION
-      });
-    }
-    const response = await runtime.registryHealth(registry);
-    return new Response(new Uint8Array(response.body), {
-      status: response.status,
-      headers: response.headers
-    });
-  }
-  /**
-   * Returns serverless metadata suitable for mounting in a user-owned router.
-   */
-  async #metadataRoute() {
-    const configured = await this.#activeConfiguredRegistry();
-    if (!configured) {
-      return new Response("registry not started\n", {
-        status: 503,
-        headers: { "content-type": "text/plain; charset=utf-8" }
-      });
-    }
-    const { runtime, registry } = configured;
-    if (!runtime.registryMetadata) {
-      return new Response("metadata is not supported by this runtime\n", {
-        status: 501,
-        headers: { "content-type": "text/plain; charset=utf-8" }
-      });
-    }
-    const response = await runtime.registryMetadata(registry);
-    return new Response(new Uint8Array(response.body), {
-      status: response.status,
-      headers: response.headers
-    });
-  }
-  /**
-   * Returns a Prometheus metrics response suitable for mounting in a user-owned router.
+   * 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 #prometheusMetricsRoute(_request) {
-    const configured = await this.#activeConfiguredRegistry();
-    if (!configured) {
-      return new Response("registry not started\n", {
-        status: 503,
-        headers: { "content-type": "text/plain; charset=utf-8" }
-      });
-    }
-    const { runtime, registry } = configured;
-    if (!runtime.registryMetrics) {
-      return new Response("metrics are not supported by this runtime\n", {
-        status: 501,
-        headers: { "content-type": "text/plain; charset=utf-8" }
-      });
+  async listen(opts = {}) {
+    const port = _nullishCoalesce(opts.port, () => ( 3e3));
+    const config = this.parseConfig();
+    const runtime = _chunkX6HIFXNKcjs.detectRuntime.call(void 0, );
+    const app = new (0, _hono.Hono)();
+    if (opts.publicDir) {
+      const serveStatic = await loadRuntimeServeStatic(runtime);
+      app.use("*", serveStatic({ root: opts.publicDir }));
     }
-    const response = await runtime.registryMetrics(registry);
-    return new Response(new Uint8Array(response.body), {
-      status: response.status,
-      headers: response.headers
-    });
+    app.all("*", (c) => this.handler(c.req.raw));
+    await crossPlatformServe(config, port, app, runtime);
   }
-  async #activeConfiguredRegistry() {
+  async diagnostics() {
+    var _a;
     const candidates = [
       this.#runtimeServerlessPromise,
       this.#runtimeServeConfiguredPromise
     ].filter(
       (candidate) => candidate !== void 0
     );
-    if (candidates.length === 0) return void 0;
-    return await candidates[0];
+    for (const candidate of candidates) {
+      const { runtime, registry } = await candidate;
+      const diagnostics = await ((_a = runtime.registryDiagnostics) == null ? void 0 : _a.call(runtime, registry));
+      if (diagnostics) return diagnostics;
+    }
+    return { mode: "not_started", envoyActiveActorCount: null };
   }
   /**
    * Starts an actor envoy for standalone server deployments.
@@ -5768,8 +5928,8 @@ var Registry = class {
       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);
     }
@@ -5807,8 +5967,8 @@ var Registry = class {
       signal,
       config,
       configuredRegistryPromise
-    ).catch((err) => {
-      logger2().warn({ err }, "shutdown error");
+    ).catch((error) => {
+      logger2().warn({ error }, "shutdown error");
     });
   }
   async #runShutdown(signal, config, configuredRegistryPromise) {
@@ -5820,9 +5980,9 @@ var Registry = class {
           try {
             const { runtime, registry } = await configuredRegistryPromise;
             await runtime.shutdownRegistry(registry);
-          } catch (err) {
+          } catch (error) {
             logger2().warn(
-              { err },
+              { error },
               "runtime registry shutdown errored (mode A)"
             );
           }
@@ -5837,7 +5997,7 @@ var Registry = class {
               await runtime.shutdownRegistry(registry);
             } catch (err) {
               logger2().warn(
-                { err },
+                { error: err },
                 "runtime registry shutdown errored (mode B)"
               );
             }
@@ -5911,7 +6071,7 @@ var Registry = class {
     };
     console.log();
     console.log(
-      `  RivetKit ${_chunkK5BA2LEOcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
+      `  RivetKit ${_chunkX6HIFXNKcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
     );
     if (config.namespace !== "default") {
       logLine("Namespace", config.namespace);
@@ -5939,12 +6099,6 @@ function isServerlessMetadataRequest(request, basePath) {
   const normalizedBase = basePath === "/" ? "" : `/${basePath.replace(/^\/+|\/+$/g, "")}`;
   return parsed.pathname === `${normalizedBase}/metadata`;
 }
-function jsonRouteResponse(status, body) {
-  return new Response(JSON.stringify(body), {
-    status,
-    headers: { "content-type": "application/json" }
-  });
-}
 function setup(input) {
   return new Registry(input);
 }
@@ -5979,5 +6133,5 @@ function setup(input) {
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunkENK7C66Gcjs.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 = _chunkENK7C66Gcjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunkK5BA2LEOcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunkK5BA2LEOcjs.toUint8Array;
+exports.ALLOWED_PUBLIC_HEADERS = _chunkZGPX6KAHcjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunk63WNTDRCcjs.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 = _chunk63WNTDRCcjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunk63WNTDRCcjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunkZGPX6KAHcjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunkX6HIFXNKcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunkX6HIFXNKcjs.toUint8Array;
 //# sourceMappingURL=mod.cjs.map