rivetkit 0.0.0-skip-ready-wait-flatten-client-options.1cae221 → 0.0.0-sqlite-uds.4e59a38

package/dist/tsup/mod.js

@@ -1,28 +1,26 @@
+import {
+  decodeWorkflowHistoryTransport
+} from "./chunk-IB73AMGJ.js";
 import {
   ACTOR_CONTEXT_INTERNAL_SYMBOL,
   ActorConfigSchema,
   CONN_STATE_MANAGER_SYMBOL,
+  RAW_STATE_SYMBOL,
+  disposeRunInspector,
   getRunFunction,
   getRunInspectorConfig,
   getRunMetadata
-} from "./chunk-7MRGNVSR.js";
-import {
-  decodeWorkflowHistoryTransport
-} from "./chunk-RZEJJJKU.js";
+} from "./chunk-3GLNEWGY.js";
 import {
   ALLOWED_PUBLIC_HEADERS,
-  CURRENT_VERSION,
   HEADER_CONN_PARAMS,
-  HEADER_ENCODING,
-  HTTP_RESPONSE_ERROR_VERSIONED,
-  HttpResponseErrorSchema,
   RemoteEngineControlClient,
   convertRegistryConfigToClientConfig,
   createClientWithDriver,
   getDatacenters,
   tryParseEndpoint,
   updateRunnerConfig
-} from "./chunk-LNMB25HW.js";
+} from "./chunk-XAIMHUPG.js";
 import {
   KEYS,
   makePrefixedKey,
@@ -30,18 +28,18 @@ import {
   queueMetadataKey,
   removePrefixFromKey,
   workflowStoragePrefix
-} from "./chunk-3BFYSYRU.js";
+} from "./chunk-HXUEHHJF.js";
 import {
-  contentTypeForEncoding,
+  assertJsonCompatValue,
   decodeCborCompat,
-  decodeCborJsonCompat,
-  encodeCborCompat,
-  serializeWithEncoding
-} from "./chunk-W54QCENI.js";
-import "./chunk-4ZIWB4CQ.js";
+  encodeCborCompat
+} from "./chunk-KTNBYGRU.js";
+import "./chunk-JZ7TWV65.js";
+import "./chunk-ITWJ6Y7G.js";
 import {
   LogLevelSchema,
   VERSION,
+  configureBaseLogger,
   deconstructError,
   getEnvUniversal,
   getLogger,
@@ -54,15 +52,20 @@ import {
   getRivetPublicEndpoint,
   getRivetPublicToken,
   getRivetRunEngine,
+  getRivetRunEngineHost,
+  getRivetRunEnginePort,
   getRivetRunEngineVersion,
   getRivetToken,
   getRivetTotalSlots,
+  getRivetkitPublicDir,
   getRivetkitRuntime,
+  getRivetkitRuntimeMode,
   isDev,
   noopNext,
+  parsePortEnv,
   stringifyError,
   toUint8Array
-} from "./chunk-D4TSTM6C.js";
+} from "./chunk-XZ62SWSC.js";
 import {
   INTERNAL_ERROR_CODE,
   RivetError,
@@ -70,11 +73,11 @@ import {
   decodeBridgeRivetError,
   encodeBridgeRivetError,
   forbiddenError,
+  isActorAbortedError,
   isRivetErrorLike,
   toRivetError,
   unsupportedFeature
-} from "./chunk-K34B3OVG.js";
-import "./chunk-PCBNKI2J.js";
+} from "./chunk-ZZ3WBRPD.js";
 
 // src/actor/log.ts
 function loggerWithoutContext() {
@@ -146,7 +149,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];
@@ -207,8 +210,8 @@ function validateSchemaSync(schemas, key, data) {
 }
 
 // src/common/inline-websocket-adapter.ts
-import { WSContext } from "hono/ws";
 import { VirtualWebSocket } from "@rivetkit/virtual-websocket";
+import { WSContext } from "hono/ws";
 function logger() {
   return getLogger("inline-websocket-adapter");
 }
@@ -320,24 +323,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) {
@@ -350,8 +353,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);
@@ -361,8 +364,23 @@ var InlineWebSocketAdapter = class {
 };
 
 // src/common/engine.ts
+var ENGINE_HOST = "127.0.0.1";
 var ENGINE_PORT = 6420;
-var ENGINE_ENDPOINT = `http://127.0.0.1:${ENGINE_PORT}`;
+var ENGINE_ENDPOINT = buildEngineEndpoint(ENGINE_HOST, ENGINE_PORT);
+function buildEngineEndpoint(host, port) {
+  const urlHost = host.includes(":") && !host.startsWith("[") ? `[${host}]` : host;
+  return `http://${urlHost}:${port}`;
+}
+function isLocalEngineEndpoint(endpoint) {
+  let url;
+  try {
+    url = new URL(endpoint);
+  } catch {
+    return false;
+  }
+  const hostname = url.hostname.toLowerCase();
+  return hostname === "localhost" || hostname === "0.0.0.0" || hostname === "::" || hostname === "[::]" || hostname === "::1" || hostname === "[::1]" || /^127(?:\.\d{1,3}){0,3}$/.test(hostname);
+}
 
 // src/registry/log.ts
 function logger2() {
@@ -377,10 +395,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;
 }
@@ -394,13 +415,19 @@ 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 = convertRegistryConfigToClientConfig(config);
@@ -415,7 +442,7 @@ async function configureServerlessPool(config) {
         serverless: {
           url: customConfig.url,
           headers,
-          request_lifespan: customConfig.requestLifespan ?? 15 * 60,
+          request_lifespan: customConfig.requestLifespan ?? 60 * 60,
           drain_grace_period: customConfig.drainGracePeriod,
           metadata_poll_interval: customConfig.metadataPollInterval ?? 1e3,
           max_runners: 1e5,
@@ -569,6 +596,8 @@ var RegistryConfigSchema = z3.object({
   maxIncomingMessageSize: z3.number().optional().default(65536),
   /** @experimental */
   maxOutgoingMessageSize: z3.number().optional().default(1048576),
+  /** Enables the experimental per-actor Unix socket API. */
+  experimentalActorUds: z3.boolean().optional().default(false),
   // MARK: Runtime
   /**
    * @experimental
@@ -674,6 +703,18 @@ var RegistryConfigSchema = z3.object({
    * Starts the full Rust engine process locally.
    */
   startEngine: z3.boolean().default(() => getRivetRunEngine()),
+  /**
+   * @experimental
+   *
+   * Host to bind the spawned local engine process to.
+   */
+  engineHost: z3.string().optional().default(() => getRivetRunEngineHost() ?? ENGINE_HOST),
+  /**
+   * @experimental
+   *
+   * Port to bind the spawned local engine process to.
+   */
+  enginePort: z3.number().int().min(1).max(65535).optional().default(() => getRivetRunEnginePort() ?? ENGINE_PORT),
   /** @experimental */
   engineVersion: z3.string().optional().default(() => getRivetRunEngineVersion() ?? VERSION),
   /**
@@ -706,12 +747,12 @@ var RegistryConfigSchema = z3.object({
   shutdown: z3.object({
     /**
      * Wait this many milliseconds for the serve promise to resolve
-     * after calling `CoreRegistry::shutdown()`. Defaults to 30s,
-     * matching Kubernetes `terminationGracePeriodSeconds`.
+     * 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: z3.number().int().min(1e3).optional().default(3e4),
+    gracePeriodMs: z3.number().int().min(1e3).optional(),
     /**
      * If true, rivetkit will not install SIGINT/SIGTERM handlers.
      * Use when the host application owns signal policy and will
@@ -719,7 +760,6 @@ var RegistryConfigSchema = z3.object({
      */
     disableSignalHandlers: z3.boolean().optional().default(false)
   }).optional().default(() => ({
-    gracePeriodMs: 3e4,
     disableSignalHandlers: false
   }))
 }).transform((config, ctx) => {
@@ -752,7 +792,11 @@ var RegistryConfigSchema = z3.object({
       message: "configurePool requires either endpoint or startEngine"
     });
   }
-  const endpoint = config.startEngine ? ENGINE_ENDPOINT : (parsedEndpoint == null ? void 0 : parsedEndpoint.endpoint) ?? (isDevEnv ? ENGINE_ENDPOINT : void 0);
+  const localEngineEndpoint = buildEngineEndpoint(
+    config.engineHost,
+    config.enginePort
+  );
+  const endpoint = config.startEngine ? localEngineEndpoint : (parsedEndpoint == null ? void 0 : parsedEndpoint.endpoint) ?? (isDevEnv ? buildEngineEndpoint(ENGINE_HOST, ENGINE_PORT) : void 0);
   const validateServerlessEndpoint = Boolean(
     config.startEngine || parsedEndpoint
   );
@@ -769,7 +813,7 @@ var RegistryConfigSchema = z3.object({
       path: ["serverless", "publicEndpoint"]
     });
   }
-  const publicEndpoint = (parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.endpoint) ?? (isDevEnv && config.startEngine ? ENGINE_ENDPOINT : void 0);
+  const publicEndpoint = (parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.endpoint) ?? (isDevEnv && config.startEngine ? endpoint : void 0);
   const publicNamespace = parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.namespace;
   const publicToken = (parsedPublicEndpoint == null ? void 0 : parsedPublicEndpoint.token) ?? config.serverless.publicToken;
   return {
@@ -801,7 +845,8 @@ function buildActorNames(config) {
         keys: [
           Array.from(KEYS.PERSIST_DATA),
           Array.from(KEYS.INSPECTOR_TOKEN),
-          Array.from(queueMetadataKey())
+          Array.from(queueMetadataKey()),
+          Array.from(KEYS.LAST_PUSHED_ALARM)
         ],
         prefixes: [
           {
@@ -929,7 +974,8 @@ function shouldAttachNativeKvError(message) {
 }
 function enrichNativeDatabaseError(database, error) {
   var _a;
-  const bridged = typeof error === "string" ? decodeBridgeRivetError(error) : error instanceof Error ? decodeBridgeRivetError(error.message) : void 0;
+  const bridgeReason = typeof error === "string" ? error : error instanceof Error ? error.message : void 0;
+  const bridged = bridgeReason === void 0 ? void 0 : decodeBridgeRivetError(bridgeReason);
   if (bridged) {
     throw bridged;
   }
@@ -1080,7 +1126,7 @@ function wrapJsNativeDatabase(database) {
   const gate = new NativeCloseGate();
   let closePromise;
   let lastInsertRowId = null;
-  const executeNative = async (sql, params) => {
+  const executeNative = async (sql, params, lease) => {
     const lastInsertRowIdColumn = lastInsertRowIdColumnName(sql);
     if (lastInsertRowIdColumn) {
       return {
@@ -1093,7 +1139,12 @@ function wrapJsNativeDatabase(database) {
     const release = gate.enter();
     try {
       const nativeParams = toNativeBindings(sql, params);
-      const result = await database.execute(sql, nativeParams);
+      const result = await database.execute(
+        sql,
+        nativeParams,
+        lease == null ? void 0 : lease.key,
+        lease == null ? void 0 : lease.timeoutMs
+      );
       if (result.lastInsertRowId !== void 0) {
         lastInsertRowId = result.lastInsertRowId;
       }
@@ -1105,11 +1156,11 @@ function wrapJsNativeDatabase(database) {
     }
   };
   return {
-    async exec(sql, callback) {
+    async exec(sql, callback, lease) {
       const release = gate.enter();
       let result;
       try {
-        result = await database.exec(sql);
+        result = await database.exec(sql, lease == null ? void 0 : lease.key, lease == null ? void 0 : lease.timeoutMs);
       } catch (error) {
         enrichNativeDatabaseError(database, error);
       } finally {
@@ -1122,8 +1173,8 @@ function wrapJsNativeDatabase(database) {
         callback(row, result.columns);
       }
     },
-    async execute(sql, params) {
-      return await executeNative(sql, params);
+    async execute(sql, params, lease) {
+      return await executeNative(sql, params, lease);
     },
     async run(sql, params) {
       await executeNative(sql, params);
@@ -1132,9 +1183,6 @@ function wrapJsNativeDatabase(database) {
       const { columns, rows } = await executeNative(sql, params);
       return { columns, rows };
     },
-    async writeMode(callback) {
-      return await callback();
-    },
     nativeMetrics() {
       var _a;
       return normalizeNativeMetrics((_a = database.metrics) == null ? void 0 : _a.call(database));
@@ -1162,6 +1210,82 @@ function lastInsertRowIdColumnName(sql) {
   return alias;
 }
 
+// src/utils/node.ts
+import { createRequire } from "module";
+var nodeCrypto;
+var nodeFsSync;
+var nodeFs;
+var nodePath;
+var nodeOs;
+var nodeChildProcess;
+var nodeStream;
+var nodeUrl;
+var hasImportedDependencies = false;
+function getRequireFn() {
+  return createRequire(import.meta.url);
+}
+function importNodeDependencies() {
+  if (hasImportedDependencies) return;
+  try {
+    const requireFn = getRequireFn();
+    nodeCrypto = requireFn(
+      /* webpackIgnore: true */
+      "node:crypto"
+    );
+    nodeFsSync = requireFn(
+      /* webpackIgnore: true */
+      "node:fs"
+    );
+    nodeFs = requireFn(
+      /* webpackIgnore: true */
+      "node:fs/promises"
+    );
+    nodePath = requireFn(
+      /* webpackIgnore: true */
+      "node:path"
+    );
+    nodeOs = requireFn(
+      /* webpackIgnore: true */
+      "node:os"
+    );
+    nodeChildProcess = requireFn(
+      /* webpackIgnore: true */
+      "node:child_process"
+    );
+    nodeStream = requireFn(
+      /* webpackIgnore: true */
+      "node:stream/promises"
+    );
+    nodeUrl = requireFn(
+      /* webpackIgnore: true */
+      "node:url"
+    );
+    hasImportedDependencies = true;
+  } catch (err) {
+    console.warn(
+      "Node.js modules not available, file system driver will not work",
+      err
+    );
+    throw err;
+  }
+}
+function getNodeFsSync() {
+  if (!nodeFsSync) {
+    throw new Error(
+      "Node fs module not loaded. Ensure importNodeDependencies() has been called."
+    );
+  }
+  return nodeFsSync;
+}
+function getNodePath() {
+  if (!nodePath) {
+    throw new Error(
+      "Node path module not loaded. Ensure importNodeDependencies() has been called."
+    );
+  }
+  return nodePath;
+}
+
 // src/registry/runtime.ts
 function normalizeRuntimeSqlExecuteResult(result) {
   return result;
@@ -1282,11 +1406,31 @@ var NapiCoreRuntime = class {
   async shutdownRegistry(registry) {
     await asNativeRegistry(registry).shutdown();
   }
-  async registryDiagnostics(registry) {
-    const diagnostics = await asNativeRegistry(registry).diagnostics();
+  async registryActorStopThresholdMs(registry) {
+    return await asNativeRegistry(registry).actorStopThresholdMs() ?? void 0;
+  }
+  async registryHealth(registry) {
+    const response = await asNativeRegistry(registry).health();
+    return {
+      status: response.status,
+      headers: response.headers,
+      body: response.body
+    };
+  }
+  async registryMetadata(registry) {
+    const response = asNativeRegistry(registry).metadata();
     return {
-      mode: diagnostics.mode,
-      envoyActiveActorCount: diagnostics.envoyActiveActorCount
+      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 handleServerlessRequest(registry, req, onStreamEvent, cancelToken, config) {
@@ -1297,6 +1441,16 @@ var NapiCoreRuntime = class {
       config
     );
   }
+  async serveListener(registry, listener, config) {
+    await asNativeRegistry(registry).serveListener(
+      {
+        port: listener.port,
+        host: listener.host,
+        publicDir: listener.publicDir
+      },
+      config
+    );
+  }
   createActorFactory(callbacks, config) {
     return asActorFactoryHandle(
       new this.#bindings.NapiActorFactory(callbacks, config)
@@ -1409,6 +1563,12 @@ var NapiCoreRuntime = class {
   actorWaitUntil(ctx, promise) {
     asNativeActorContext(ctx).waitUntil(promise);
   }
+  async actorWaitForTrackedShutdownWork(ctx) {
+    return await asNativeActorContext(ctx).waitForTrackedShutdownWork();
+  }
+  async actorWaitForTrackedShutdownWorkUnbounded(ctx) {
+    await asNativeActorContext(ctx).waitForTrackedShutdownWorkUnbounded();
+  }
   actorKeepAwake(ctx, promise) {
     asNativeActorContext(ctx).keepAwake(promise);
   }
@@ -1463,13 +1623,19 @@ var NapiCoreRuntime = class {
   async actorKvBatchDelete(ctx, keys) {
     await asNativeActorContext(ctx).kv().batchDelete(keys.map(toNapiBuffer));
   }
-  async actorSqlExec(ctx, sql) {
-    return await this.#actorSql(ctx).exec(sql);
+  async actorSqlExec(ctx, sql, lease) {
+    return await this.#actorSql(ctx).exec(
+      sql,
+      lease == null ? void 0 : lease.key,
+      lease == null ? void 0 : lease.timeoutMs
+    );
   }
-  async actorSqlExecute(ctx, sql, params) {
+  async actorSqlExecute(ctx, sql, params, lease) {
     const result = await this.#actorSql(ctx).execute(
       sql,
-      toNapiSqlBindParams(params)
+      toNapiSqlBindParams(params),
+      lease == null ? void 0 : lease.key,
+      lease == null ? void 0 : lease.timeoutMs
     );
     return normalizeRuntimeSqlExecuteResult(result);
   }
@@ -1499,6 +1665,9 @@ var NapiCoreRuntime = class {
     this.#sql.delete(nativeCtx);
     await database.close();
   }
+  async actorUdsProvision(ctx) {
+    return await asNativeActorContext(ctx).provisionActorUds();
+  }
   async actorQueueSend(ctx, name, body) {
     return toNapiQueueMessage(
       await asNativeActorContext(ctx).queue().send(name, toNapiBuffer(body))
@@ -1520,8 +1689,12 @@ var NapiCoreRuntime = 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(
@@ -1540,6 +1713,9 @@ var NapiCoreRuntime = class {
   async actorQueueInspectMessages(ctx) {
     return await asNativeActorContext(ctx).queue().inspectMessages();
   }
+  async actorQueueReset(ctx) {
+    await asNativeActorContext(ctx).queue().reset();
+  }
   actorScheduleAfter(ctx, durationMs, actionName, args) {
     asNativeActorContext(ctx).schedule().after(durationMs, actionName, toNapiBuffer(args));
   }
@@ -1655,20 +1831,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 RivetError(
-    "actor",
-    "validation_error",
-    `Invalid ${target}`,
-    {
-      public: true,
-      metadata: { issues }
-    }
-  );
+  return new RivetError("actor", "validation_error", `Invalid ${target}`, {
+    public: true,
+    metadata: { issues }
+  });
 }
 
 // src/registry/wasm-runtime.ts
@@ -1826,8 +2000,18 @@ var WasmCoreRuntime = class {
   async shutdownRegistry(registry) {
     await callWasm(() => asWasmRegistry(registry).shutdown());
   }
-  async registryDiagnostics() {
-    return { mode: "wasm", envoyActiveActorCount: null };
+  async registryHealth() {
+    return {
+      status: 200,
+      headers: { "content-type": "application/json" },
+      body: new TextEncoder().encode(
+        JSON.stringify({
+          status: "ok",
+          runtime: "rivetkit",
+          version: "wasm"
+        })
+      )
+    };
   }
   async handleServerlessRequest(registry, req, onStreamEvent, cancelToken, config) {
     return await callHandleAsync(
@@ -1839,6 +2023,11 @@ var WasmCoreRuntime = class {
       config
     );
   }
+  async serveListener(_registry, _listener, _config) {
+    throw new Error(
+      "registry.listen() is not supported on the wasm runtime; use registry.serve() and mount the handler in your platform's HTTP server instead"
+    );
+  }
   createActorFactory(callbacks, config) {
     return callWasmSync(
       () => asActorFactoryHandle2(
@@ -1976,6 +2165,18 @@ var WasmCoreRuntime = class {
   actorWaitUntil(ctx, promise) {
     callHandle(asWasmActorContext(ctx), "waitUntil", promise);
   }
+  async actorWaitForTrackedShutdownWork(ctx) {
+    return await callHandle(
+      asWasmActorContext(ctx),
+      "waitForTrackedShutdownWork"
+    );
+  }
+  async actorWaitForTrackedShutdownWorkUnbounded(ctx) {
+    await callHandle(
+      asWasmActorContext(ctx),
+      "waitForTrackedShutdownWorkUnbounded"
+    );
+  }
   actorKeepAwake(ctx, promise) {
     const wasmCtx = asWasmActorContext(ctx);
     const regionId = callHandle(wasmCtx, "beginKeepAwake");
@@ -2063,10 +2264,20 @@ var WasmCoreRuntime = class {
     const kv = childHandle(asWasmActorContext(ctx), "kv");
     await callHandleAsync(kv, "batchDelete", keys);
   }
-  async actorSqlExec(ctx, sql) {
+  async actorSqlExec(ctx, sql, lease) {
+    if (lease) {
+      throw new Error(
+        "sqlite transaction leases are unavailable in the wasm runtime"
+      );
+    }
     return await callWasm(() => this.#actorSql(ctx).exec(sql));
   }
-  async actorSqlExecute(ctx, sql, params) {
+  async actorSqlExecute(ctx, sql, params, lease) {
+    if (lease) {
+      throw new Error(
+        "sqlite transaction leases are unavailable in the wasm runtime"
+      );
+    }
     const result = await callWasm(
       () => this.#actorSql(ctx).execute(sql, params)
     );
@@ -2095,6 +2306,9 @@ var WasmCoreRuntime = class {
     this.#sql.delete(wasmCtx);
     await callWasm(() => database.close());
   }
+  async actorUdsProvision(_ctx) {
+    throw new Error("actor UDS endpoints require the native runtime");
+  }
   async actorQueueSend(ctx, name, body) {
     const queue2 = childHandle(asWasmActorContext(ctx), "queue");
     return normalizeQueueMessage(
@@ -2123,9 +2337,15 @@ var WasmCoreRuntime = 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");
@@ -2156,6 +2376,10 @@ var WasmCoreRuntime = class {
     const queue2 = childHandle(asWasmActorContext(ctx), "queue");
     return await callHandleAsync(queue2, "inspectMessages");
   }
+  async actorQueueReset(ctx) {
+    const queue2 = childHandle(asWasmActorContext(ctx), "queue");
+    await callHandleAsync(queue2, "reset");
+  }
   actorScheduleAfter(ctx, durationMs, actionName, args) {
     const schedule = childHandle(asWasmActorContext(ctx), "schedule");
     callHandle(schedule, "after", wasmNumber(durationMs), actionName, args);
@@ -2217,6 +2441,29 @@ async function loadWasmRuntime(config) {
   };
 }
 
+// src/registry/write-through-proxy.ts
+import onChange from "@rivetkit/on-change";
+function createWriteThroughProxy(value, commit, beforeChange) {
+  if (!value || typeof value !== "object") {
+    return value;
+  }
+  return onChange(
+    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();
@@ -2370,6 +2617,14 @@ function databaseNotConfiguredError() {
     { public: true }
   );
 }
+function databaseClientNotReadyError() {
+  return new 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 RivetError(
     "actor",
@@ -2421,7 +2676,23 @@ function resolveNativeDestroy(runtime, ctx) {
 function clearNativeRuntimeState(runtime, ctx) {
   callNativeSync(() => runtime.actorClearRuntimeState(ctx));
 }
-async function cleanupNativeSleepRuntimeState(runtime, 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: stringifyError(error)
+      });
+    });
+    return;
+  }
+  await (afterTrackedWorkDrained == null ? void 0 : afterTrackedWorkDrained());
   await closeNativeDatabaseClient(runtime, ctx);
   await closeNativeSqlDatabase(runtime, ctx);
   clearNativeRuntimeState(runtime, ctx);
@@ -2453,8 +2724,17 @@ function getOrCreateNativeSqlDatabase(runtime, ctx) {
     return cachedDatabase;
   }
   const database = wrapJsNativeDatabase({
-    exec: (sql) => runtime.actorSqlExec(ctx, sql),
-    execute: (sql, params) => runtime.actorSqlExecute(ctx, sql, params),
+    exec: (sql, leaseKey, leaseTimeoutMs) => runtime.actorSqlExec(
+      ctx,
+      sql,
+      leaseKey === void 0 ? void 0 : { key: leaseKey, timeoutMs: leaseTimeoutMs }
+    ),
+    execute: (sql, params, leaseKey, leaseTimeoutMs) => runtime.actorSqlExecute(
+      ctx,
+      sql,
+      params,
+      leaseKey === void 0 ? void 0 : { key: leaseKey, timeoutMs: leaseTimeoutMs }
+    ),
     query: (sql, params) => runtime.actorSqlQuery(ctx, sql, params),
     run: (sql, params) => runtime.actorSqlRun(ctx, sql, params),
     metrics: () => runtime.actorSqlMetrics(ctx),
@@ -2526,11 +2806,14 @@ function decodeValue(value) {
   if (!value || value.length === 0) {
     return void 0;
   }
-  return decodeCborJsonCompat(value);
+  return decodeCborCompat(value);
 }
 function encodeValue(value) {
   return encodeCborCompat(value);
 }
+function normalizeArgs(value) {
+  return Array.isArray(value) ? value : value === void 0 || value === null ? [] : [value];
+}
 function unwrapTsfnPayload(error, payload) {
   if (error !== null && error !== void 0) {
     throw error;
@@ -2562,27 +2845,7 @@ function isStructuredBridgeError(error) {
   return isRivetErrorLike(error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
 }
 function encodeNativeCallbackError(error) {
-  const structuredError = isStructuredBridgeError(error) ? error : deconstructError(error, logger2(), {
-    bridge: "native_callback"
-  });
-  let stack;
-  if (error instanceof Error) {
-    try {
-      stack = error.stack;
-    } catch {
-      stack = void 0;
-    }
-  }
-  logger2().warn({
-    msg: "native callback error encoded for bridge",
-    group: structuredError.group,
-    code: structuredError.code,
-    message: structuredError.message,
-    metadata: structuredError.metadata,
-    originalError: stringifyError(error),
-    stack,
-    bridge: "native_callback"
-  });
+  const structuredError = isStructuredBridgeError(error) ? error : deconstructError(error, true);
   const bridgeError = new Error(encodeBridgeRivetError(structuredError), {
     cause: error instanceof Error ? error : void 0
   });
@@ -2606,12 +2869,6 @@ 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 RivetError ? error.metadata : void 0;
   const metadataError = metadata && typeof metadata === "object" && "error" in metadata ? metadata.error : void 0;
@@ -2737,6 +2994,14 @@ function serializeWorkflowEntryKind(kind) {
           originalName: kind.val.originalName
         }
       };
+    case "WorkflowVersionCheckEntry":
+      return {
+        tag: kind.tag,
+        val: {
+          resolved: kind.val.resolved,
+          latest: kind.val.latest
+        }
+      };
   }
 }
 function serializeWorkflowHistoryForJson(data) {
@@ -2744,7 +3009,7 @@ function serializeWorkflowHistoryForJson(data) {
     return null;
   }
   const history = decodeWorkflowHistoryTransport(data);
-  return {
+  return jsonSafe({
     nameRegistry: [...history.nameRegistry],
     entries: history.entries.map((entry) => ({
       id: entry.id,
@@ -2768,7 +3033,7 @@ function serializeWorkflowHistoryForJson(data) {
         ]
       )
     )
-  };
+  });
 }
 function toHttpJsonCompatible(value) {
   return JSON.parse(
@@ -2819,45 +3084,7 @@ function wrapNativeCallback(callback) {
 }
 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);
+  return normalizeArgs(decoded);
 }
 function buildRequest(init) {
   const url = init.uri.startsWith("http") ? init.uri : new URL(init.uri, "http://127.0.0.1").toString();
@@ -2911,13 +3138,23 @@ var NativeConnAdapter = class {
       decodeValue(this.#runtime.connParams(this.#conn))
     );
   }
+  [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) => {
+        assertJsonCompatValue(newValue);
+      }
+    );
   }
   set state(value) {
+    assertJsonCompatValue(value);
     this.#writeState(value, { writeNative: true });
   }
   initializeState(value) {
@@ -3269,44 +3506,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) {
@@ -3376,7 +3592,7 @@ var NativeQueueAdapter = class {
         }
         yield message;
       } catch (error) {
-        if (isRivetErrorLike(error) && error.group === "actor" && error.code === "aborted") {
+        if (isActorAbortedError(error)) {
           return;
         }
         throw error;
@@ -3741,6 +3957,68 @@ var TrackedWebSocketHandleAdapter = class {
     return typeof value === "object" && value !== null && "then" in value && typeof value.then === "function";
   }
 };
+var NativeConnectionMap = class {
+  #runtime;
+  #ctx;
+  #schemas;
+  constructor(runtime, ctx, schemas) {
+    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();
+  }
+  [Symbol.toStringTag] = "NativeConnectionMap";
+};
 var ActorContextHandleAdapter = class {
   #runtime;
   #ctx;
@@ -3749,9 +4027,9 @@ var ActorContextHandleAdapter = class {
   #abortSignalCleanup;
   #client;
   #clientFactory;
+  #connMap;
   #databaseProvider;
   #db;
-  #dbProxy;
   #dispatchCancelToken;
   #kv;
   #queue;
@@ -3790,55 +4068,57 @@ var ActorContextHandleAdapter = class {
     }
     return this.#sql;
   }
+  async actorUds() {
+    return await this.#runtime.actorUdsProvision(this.#ctx);
+  }
   get db() {
     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;
+    }
+    const runtimeState = getNativeRuntimeState(this.#runtime, this.#ctx);
+    const cachedClient = runtimeState.databaseClient;
+    if (cachedClient) {
+      this.#db = cachedClient.client;
+      return this.#db;
     }
-    return this.#dbProxy;
+    throw databaseClientNotReadyError();
+  }
+  [RAW_STATE_SYMBOL]() {
+    if (!this.#stateEnabled) {
+      throw stateNotEnabledError();
+    }
+    return this.#readState();
   }
   get state() {
     if (!this.#stateEnabled) {
       throw stateNotEnabledError();
     }
+    const actorState = getNativePersistState(this.#runtime, this.#ctx);
     const nextState = this.#readState();
-    return createWriteThroughProxy(
-      nextState,
-      (nextValue) => {
-        this.#writeState(nextValue, { scheduleSave: true });
-      },
-      () => {
-        this.#assertCanMutateState();
-      }
-    );
+    if (actorState.stateProxy === void 0 || actorState.stateProxyTarget !== nextState) {
+      actorState.stateProxyTarget = nextState;
+      actorState.stateProxy = createWriteThroughProxy(
+        nextState,
+        (nextValue) => {
+          this.#writeState(nextValue, { scheduleSave: true });
+        },
+        (newValue) => {
+          this.#assertCanMutateState();
+          assertJsonCompatValue(newValue);
+        }
+      );
+    }
+    return actorState.stateProxy;
   }
   set state(value) {
     if (!this.#stateEnabled) {
       throw stateNotEnabledError();
     }
     this.#assertCanMutateState();
+    assertJsonCompatValue(value);
     this.#writeState(value, { scheduleSave: true });
   }
   initializeState(value) {
@@ -3895,25 +4175,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();
@@ -4111,35 +4380,17 @@ var ActorContextHandleAdapter = class {
     });
   }
   keepAwake(promise) {
-    const runtimeState = getNativeRuntimeState(this.#runtime, this.#ctx);
-    runtimeState.keepAwakeCount = (runtimeState.keepAwakeCount ?? 0) + 1;
     const trackedPromise = Promise.resolve(promise).catch((error) => {
       logger2().warn({
         msg: "keepAwake promise rejected",
         error: stringifyError(error)
       });
-    }).finally(async () => {
-      runtimeState.keepAwakeCount = Math.max(
-        (runtimeState.keepAwakeCount ?? 1) - 1,
-        0
-      );
-      if (runtimeState.keepAwakeCount === 0 && runtimeState.deferSleepCleanupUntilKeepAwakeIdle) {
-        runtimeState.deferSleepCleanupUntilKeepAwakeIdle = false;
-        await cleanupNativeSleepRuntimeState(
-          this.#runtime,
-          this.#ctx
-        );
-      }
     }).then(() => null);
     try {
       callNativeSync(
         () => this.#runtime.actorKeepAwake(this.#ctx, trackedPromise)
       );
     } catch (error) {
-      runtimeState.keepAwakeCount = Math.max(
-        (runtimeState.keepAwakeCount ?? 1) - 1,
-        0
-      );
       if (!isClosedTaskRegistrationError(error)) {
         throw error;
       }
@@ -4152,7 +4403,10 @@ var ActorContextHandleAdapter = class {
   }
   internalKeepAwake(run) {
     const promise = typeof run === "function" ? run() : run;
-    const trackedPromise = promise.then(() => null);
+    const trackedPromise = promise.then(
+      () => null,
+      () => null
+    );
     try {
       callNativeSync(
         () => this.#runtime.actorRegisterTask(this.#ctx, trackedPromise)
@@ -4208,6 +4462,7 @@ var ActorContextHandleAdapter = class {
     return false;
   }
   sleep() {
+    this.#flushStateChange();
     callNativeSync(() => this.#runtime.actorSleep(this.#ctx));
   }
   destroy() {
@@ -4225,6 +4480,7 @@ var ActorContextHandleAdapter = class {
   }
   async dispose() {
     var _a;
+    this.#flushStateChange();
     (_a = this.#abortSignalCleanup) == null ? void 0 : _a.call(this);
     this.#sql = void 0;
   }
@@ -4252,13 +4508,12 @@ var ActorContextHandleAdapter = class {
     return actorState.state;
   }
   #writeState(value, options) {
-    encodeValue(value);
     const actorState = getNativePersistState(this.#runtime, this.#ctx);
     actorState.state = value;
     if (!options.scheduleSave) {
       return;
     }
-    this.#handleStateChange();
+    this.#scheduleSave();
   }
   #assertCanMutateState() {
     const actorState = getNativePersistState(this.#runtime, this.#ctx);
@@ -4266,9 +4521,31 @@ var ActorContextHandleAdapter = class {
       throw stateMutationReentrantError();
     }
   }
-  #handleStateChange() {
+  // Coalesce the request-save and onStateChange work to once per event loop
+  // tick. A synchronous burst of mutations (for example
+  // `Object.assign(c.state, ...)`) would otherwise cross the NAPI boundary and
+  // run onStateChange once per field, re-serializing the whole state each time
+  // and pinning the event loop on large state.
+  #scheduleSave() {
+    const actorState = getNativePersistState(this.#runtime, this.#ctx);
+    if (actorState.saveScheduled) {
+      return;
+    }
+    actorState.saveScheduled = true;
+    actorState.pendingSaveHandle = setImmediate(() => {
+      this.#flushStateChange();
+    });
+  }
+  #flushStateChange() {
     const actorState = getNativePersistState(this.#runtime, this.#ctx);
-    encodeValue(actorState.state);
+    if (!actorState.saveScheduled) {
+      return;
+    }
+    actorState.saveScheduled = false;
+    if (actorState.pendingSaveHandle !== void 0) {
+      clearImmediate(actorState.pendingSaveHandle);
+      actorState.pendingSaveHandle = void 0;
+    }
     callNativeSync(
       () => this.#runtime.actorRequestSave(this.#ctx, { immediate: false })
     );
@@ -4446,40 +4723,7 @@ function withConnContext(runtime, ctx, conn, clientFactory, schemas = {}, databa
     }
   );
 }
-function buildNativeRequestErrorResponse(encoding, path, error) {
-  const { statusCode, group, code, message, metadata } = deconstructError(
-    error,
-    logger2(),
-    {
-      path,
-      runtime: "native"
-    },
-    false
-  );
-  const body = serializeWithEncoding(
-    encoding,
-    { group, code, message, metadata },
-    HTTP_RESPONSE_ERROR_VERSIONED,
-    CURRENT_VERSION,
-    HttpResponseErrorSchema,
-    (value) => value,
-    (value) => ({
-      group: value.group,
-      code: value.code,
-      message: value.message,
-      metadata: value.metadata === void 0 ? null : runtimeBytesToArrayBuffer(
-        encodeCborCompat(value.metadata)
-      )
-    })
-  );
-  return new Response(body, {
-    status: statusCode,
-    headers: {
-      "Content-Type": contentTypeForEncoding(encoding)
-    }
-  });
-}
-function buildActorConfig(definition, registryConfig) {
+function buildActorConfig(definition, registryConfig, runtimeKind) {
   const config = definition.config;
   const options = config.options ?? {};
   const canHibernate = options.canHibernateWebSocket;
@@ -4488,6 +4732,7 @@ function buildActorConfig(definition, registryConfig) {
     icon: options.icon,
     hasDatabase: config.db !== void 0,
     remoteSqlite: config.db !== void 0 && sqliteBackendForConfig(registryConfig) === "remote",
+    experimentalActorUds: registryConfig.experimentalActorUds,
     hasState: config.state !== void 0 || typeof config.createState === "function",
     canHibernateWebsocket: typeof canHibernate === "boolean" ? canHibernate : void 0,
     stateSaveIntervalMs: options.stateSaveInterval,
@@ -4508,9 +4753,79 @@ function buildActorConfig(definition, registryConfig) {
     maxOutgoingMessageSize: registryConfig.maxOutgoingMessageSize,
     preloadMaxWorkflowBytes: options.preloadMaxWorkflowBytes,
     preloadMaxConnectionsBytes: options.preloadMaxConnectionsBytes,
-    actions: Object.keys(config.actions ?? {}).sort().map((name) => ({ name }))
+    actions: Object.keys(config.actions ?? {}).sort().map((name) => ({ name })),
+    inspectorTabs: buildInspectorTabs(config.inspector, runtimeKind)
   };
 }
+function buildInspectorTabs(inspector, runtimeKind) {
+  if (!inspector || typeof inspector !== "object") return void 0;
+  const tabs = inspector.tabs;
+  if (!Array.isArray(tabs) || tabs.length === 0) return void 0;
+  return tabs.map((raw) => {
+    const entry = raw;
+    if (entry.hidden === true) {
+      return { id: entry.id, hidden: true };
+    }
+    if (runtimeKind === "wasm") {
+      if (entry.source !== void 0) {
+        logger2().warn(
+          {
+            tabId: entry.id,
+            runtimeKind
+          },
+          "inspector.tabs[].source is not supported on wasm runners (current host: wasm). Tab descriptors will still appear in the dashboard strip but the tab body will render a not-available placeholder."
+        );
+      }
+      return {
+        id: entry.id,
+        label: entry.label,
+        icon: entry.icon,
+        source: void 0
+      };
+    }
+    const resolved = entry.source !== void 0 ? getNodePath().resolve(entry.source) : void 0;
+    if (resolved !== void 0) {
+      validateInspectorTabSource(entry.id, resolved);
+    }
+    return {
+      id: entry.id,
+      label: entry.label,
+      icon: entry.icon,
+      source: resolved
+    };
+  });
+}
+function validateInspectorTabSource(tabId, resolved) {
+  if (resolved === getNodePath().parse(resolved).root) {
+    throw new Error(
+      `inspector.tabs[id="${tabId}"].source resolves to the filesystem root (${resolved}). Point it at the tab's own static-asset directory instead.`
+    );
+  }
+  let stat;
+  try {
+    stat = getNodeFsSync().statSync(resolved);
+  } catch (err) {
+    const code = err == null ? void 0 : err.code;
+    if (code === "ENOENT") {
+      throw new Error(
+        `inspector.tabs[id="${tabId}"].source (${resolved}) does not exist.`
+      );
+    }
+    if (code === "EACCES") {
+      throw new Error(
+        `inspector.tabs[id="${tabId}"].source (${resolved}) is not readable (EACCES).`
+      );
+    }
+    throw new Error(
+      `inspector.tabs[id="${tabId}"].source (${resolved}) could not be stat'd: ${(err == null ? void 0 : err.message) ?? err}`
+    );
+  }
+  if (!stat.isDirectory()) {
+    throw new Error(
+      `inspector.tabs[id="${tabId}"].source (${resolved}) must be a directory, got ${stat.isFile() ? "file" : "non-directory"}.`
+    );
+  }
+}
 function buildNativeFactory(runtime, registryConfig, definition) {
   var _a;
   const config = definition.config;
@@ -4602,13 +4917,16 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         }
       );
     };
-    try {
-      await runtime.actorVerifyInspectorAuth(
-        ctx,
-        ((_a2 = jsRequest.headers.get("authorization")) == null ? void 0 : _a2.replace(/^Bearer\s+/i, "")) ?? null
-      );
-    } catch (error) {
-      return errorResponse(error, 401);
+    const isPublicPerActorPath = jsRequest.method === "GET" && url.pathname.startsWith("/inspector/custom-tabs/");
+    if (!isPublicPerActorPath) {
+      try {
+        await runtime.actorVerifyInspectorAuth(
+          ctx,
+          ((_a2 = jsRequest.headers.get("authorization")) == null ? void 0 : _a2.replace(/^Bearer\s+/i, "")) ?? null
+        );
+      } catch (error) {
+        return errorResponse(error, 401);
+      }
     }
     const workflowHistory = () => {
       var _a3;
@@ -4675,6 +4993,47 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           messages
         });
       }
+      if (url.pathname === "/inspector/queue" && jsRequest.method === "DELETE") {
+        await runtime.actorQueueReset(ctx);
+        return jsonResponse({});
+      }
+      if (url.pathname === "/inspector/queue" && jsRequest.method === "POST") {
+        let body;
+        try {
+          body = await jsRequest.json();
+        } catch {
+          return errorResponse(
+            new RivetError(
+              "actor",
+              "invalid_request",
+              "Invalid inspector JSON body",
+              { public: true }
+            ),
+            400
+          );
+        }
+        const name = body.name ?? "";
+        if (name === "") {
+          return errorResponse(
+            new RivetError(
+              "actor",
+              "invalid_request",
+              "Queue message name must not be empty",
+              { public: true }
+            ),
+            400
+          );
+        }
+        const cbor = encodeCborCompat(
+          body.body ?? null
+        );
+        const message = await runtime.actorQueueSend(ctx, name, cbor);
+        return jsonResponse({
+          id: message.id().toString(),
+          name: message.name(),
+          createdAtMs: message.createdAt()
+        });
+      }
       if (url.pathname === "/inspector/traces" && jsRequest.method === "GET") {
         return jsonResponse({ otlp: [], clamped: false });
       }
@@ -4699,6 +5058,10 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             isWorkflowEnabled: getNativeWorkflowInspector(ctx) !== void 0
           });
         } catch (error) {
+          logger2().error({
+            msg: "error replaying workflow history",
+            error
+          });
           return errorResponse(error);
         }
       }
@@ -4839,17 +5202,34 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           );
         }
         const body = await jsRequest.json();
+        if (body.args !== void 0 && body.properties !== void 0) {
+          return jsonResponse(
+            { error: "use either args or properties, not both" },
+            { status: 400 }
+          );
+        }
+        if (body.properties !== void 0 && (body.properties === null || typeof body.properties !== "object" || Array.isArray(body.properties))) {
+          return jsonResponse(
+            { error: "properties must be an object" },
+            { status: 400 }
+          );
+        }
+        const args = body.properties !== void 0 ? [body.properties] : normalizeArgs(body.args);
         try {
           const output = await action(
             actorCtx,
             ...validateActionArgs(
               schemaConfig.actionInputSchemas,
               actionName,
-              body.args ?? []
+              args
             )
           );
           return jsonResponse({ output });
         } catch (error) {
+          logger2().error({
+            msg: "Error handling inspector action request",
+            error
+          });
           return errorResponse(error);
         }
       }
@@ -4863,6 +5243,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();
@@ -4985,29 +5369,33 @@ 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 {
-          const runtimeState = getNativeRuntimeState(runtime, ctx);
-          if ((runtimeState.keepAwakeCount ?? 0) > 0) {
-            runtimeState.deferSleepCleanupUntilKeepAwakeIdle = true;
-          } else {
-            await cleanupNativeSleepRuntimeState(runtime, ctx);
+          try {
+            await cleanupNativeSleepRuntimeState(
+              runtime,
+              ctx,
+              saveActorState
+            );
+          } finally {
+            await actorCtx.dispose();
           }
-          await actorCtx.dispose();
         }
       }
     ),
@@ -5020,6 +5408,9 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             await config.onDestroy(actorCtx);
           }
         } finally {
+          const actorId = callNativeSync(() => runtime.actorId(ctx));
+          nativeRunHandlerActiveByActorId.delete(actorId);
+          disposeRunInspector(config.run, actorId);
           resolveNativeDestroy(runtime, ctx);
           await actorCtx.closeDatabase();
           clearNativeRuntimeState(runtime, ctx);
@@ -5246,17 +5637,6 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               );
             }
             return await toRuntimeHttpResponse(response);
-          } catch (error2) {
-            const encodingHeader = jsRequest.headers.get(HEADER_ENCODING);
-            const encoding = encodingHeader === "cbor" || encodingHeader === "bare" ? encodingHeader : "json";
-            const path = new URL(jsRequest.url).pathname;
-            return await toRuntimeHttpResponse(
-              buildNativeRequestErrorResponse(
-                encoding,
-                path,
-                error2
-              )
-            );
           } finally {
             await (requestCtx == null ? void 0 : requestCtx.dispose());
             if (conn) {
@@ -5304,7 +5684,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           try {
             await run(actorCtx);
           } finally {
-            nativeRunHandlerActiveByActorId.set(actorId, false);
+            nativeRunHandlerActiveByActorId.delete(actorId);
             await actorCtx.dispose();
           }
         }
@@ -5437,7 +5817,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
   };
   return runtime.createActorFactory(
     callbacks,
-    buildActorConfig(definition, registryConfig)
+    buildActorConfig(definition, registryConfig, runtime.kind)
   );
 }
 async function buildServeConfig(config) {
@@ -5460,10 +5840,17 @@ async function buildServeConfig(config) {
     serverlessValidateEndpoint: config.validateServerlessEndpoint,
     serverlessMaxStartPayloadBytes: config.serverless.maxStartPayloadBytes
   };
-  if (config.startEngine) {
+  try {
     const { getEnginePath } = await loadEngineCli();
     serveConfig.engineBinaryPath = getEnginePath();
+  } catch (error) {
+    logger2().warn({
+      msg: "could not resolve a local engine binary; if a local engine must be spawned it will fail with engine.binary_unavailable \u2014 set RIVET_ENGINE_BINARY_PATH or install the @rivetkit/engine-cli platform package",
+      error: stringifyError(error)
+    });
   }
+  serveConfig.engineHost = config.engineHost;
+  serveConfig.enginePort = config.enginePort;
   if ((_a = config.test) == null ? void 0 : _a.enabled) {
     serveConfig.inspectorTestToken = getEnvUniversal("_RIVET_TEST_INSPECTOR_TOKEN") ?? "token";
   }
@@ -5474,6 +5861,9 @@ async function buildRegistryWithRuntime(config, runtime) {
   if (((_a = config.test) == null ? void 0 : _a.enabled) && getEnvUniversal("_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
     trySetProcessEnv("_RIVET_TEST_INSPECTOR_TOKEN", "token");
   }
+  if (runtime.kind === "napi") {
+    importNodeDependencies();
+  }
   const registry = runtime.createRegistry();
   for (const [name, definition] of Object.entries(config.use)) {
     runtime.registerActor(
@@ -5497,8 +5887,24 @@ async function buildConfiguredRegistry(config) {
 }
 
 // src/registry/index.ts
+function signalExitCode(signal) {
+  switch (signal) {
+    case "SIGINT":
+      return 130;
+    case "SIGTERM":
+      return 143;
+  }
+}
+function finishShutdownSignal(signal) {
+  if (process.pid === 1) {
+    process.exit(signalExitCode(signal));
+  }
+  process.kill(process.pid, signal);
+}
 var Registry = class {
   #config;
+  #buildConfiguredRegistry;
+  routes;
   get config() {
     return this.#config;
   }
@@ -5513,18 +5919,34 @@ var Registry = class {
   #shutdownInstalled = false;
   #shutdownInFlight = null;
   #signalHandlers = {};
-  constructor(config) {
+  constructor(config, deps) {
+    var _a;
     this.#config = config;
+    if ((_a = config.logging) == null ? void 0 : _a.baseLogger) {
+      configureBaseLogger(config.logging.baseLogger);
+    }
+    this.#buildConfiguredRegistry = (deps == null ? void 0 : deps.buildConfiguredRegistry) ?? buildConfiguredRegistry;
+    this.routes = {
+      health: () => this.#healthRoute(),
+      metadata: () => this.#metadataRoute(),
+      prometheusMetrics: (request) => this.#prometheusMetricsRoute(request)
+    };
   }
+  /**
+   * Fires `configureServerlessPool` once per process when the registry
+   * config opts into it. Cached on the instance so repeated calls (from
+   * `handler()` and `listen()`) only run the upsert once. The retry loop
+   * inside `configureServerlessPool` tolerates the engine still warming up.
+   */
   #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(() => {
       });
     }
@@ -5545,7 +5967,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(
@@ -5560,7 +5982,7 @@ var Registry = class {
     if (isStartRequest) {
       try {
         await this.#ensureServerlessPoolConfigured(config);
-      } catch (error) {
+      } catch (_error) {
         return new Response(
           JSON.stringify({
             group: "guard",
@@ -5674,7 +6096,7 @@ var Registry = class {
     if (isMetadataRequest && !isEngineMetadataRequest) {
       try {
         await this.#ensureServerlessPoolConfigured(config);
-      } catch (error) {
+      } catch (_error) {
         return new Response(
           JSON.stringify({
             group: "guard",
@@ -5707,40 +6129,147 @@ var Registry = class {
       fetch: (request) => this.handler(request)
     };
   }
-  async diagnostics() {
-    var _a;
+  /**
+   * Bind an HTTP listener provided by the native (Rust) runtime and serve
+   * the registry's serverless endpoints over it. Resolves only after the
+   * registry is shut down (SIGINT/SIGTERM or `nativeRegistry.shutdown()`).
+   *
+   * @param opts.port      Port to listen on. Defaults to `process.env.RIVET_PORT`
+   *                       if set, otherwise 3000.
+   * @param opts.host      Address to bind. Defaults to `0.0.0.0`.
+   * @param opts.publicDir If set, serves static files from this directory
+   *                       as a fallback below the framework routes.
+   *
+   * @example
+   * ```ts
+   * await registry.listen();
+   * await registry.listen({ port: 8080, publicDir: "./public" });
+   * ```
+   */
+  async listen(opts = {}) {
+    const port = opts.port ?? parsePortEnv(process.env.RIVET_PORT) ?? 3e3;
+    const publicDir = opts.publicDir ?? getRivetkitPublicDir();
+    const config = this.parseConfig();
+    const configuredRegistryPromise = buildConfiguredRegistry(config);
+    this.#runtimeServeConfiguredPromise = configuredRegistryPromise;
+    this.#runtimeServerlessPromise = configuredRegistryPromise;
+    this.#installSignalHandlers(config);
+    this.#printWelcome(config, "serverless", {
+      port,
+      host: opts.host,
+      publicDir
+    });
+    this.#ensureServerlessPoolConfigured(config);
+    const { runtime, registry, serveConfig } = await configuredRegistryPromise;
+    await runtime.serveListener(
+      registry,
+      { port, host: opts.host, publicDir },
+      serveConfig
+    );
+  }
+  /**
+   * 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: VERSION
+      });
+    }
+    const { runtime, registry } = configured;
+    if (!runtime.registryHealth) {
+      return jsonRouteResponse(501, {
+        status: "unsupported",
+        runtime: "rivetkit",
+        version: 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.
+   */
+  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" }
+      });
+    }
+    const response = await runtime.registryMetrics(registry);
+    return new Response(new Uint8Array(response.body), {
+      status: response.status,
+      headers: response.headers
+    });
+  }
+  async #activeConfiguredRegistry() {
     const candidates = [
       this.#runtimeServerlessPromise,
       this.#runtimeServeConfiguredPromise
     ].filter(
       (candidate) => candidate !== void 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 };
+    if (candidates.length === 0) return void 0;
+    return await candidates[0];
   }
   /**
    * Starts an actor envoy for standalone server deployments.
    */
   #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;
@@ -5749,58 +6278,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();
-      process.kill(process.pid, signal);
+      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 = ((_a = config.shutdown) == null ? void 0 : _a.gracePeriodMs) ?? 3e4;
+    const modeAPromise = this.#runtimeServeConfiguredPromise;
+    const modeBPromise = this.#runtimeServerlessPromise;
+    const gracePeriodMs = ((_a = config.shutdown) == null ? void 0 : _a.gracePeriodMs) ?? await this.#actorStopThresholdMs(modeAPromise ?? modeBPromise) ?? 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)"
               );
             }
@@ -5822,8 +6379,23 @@ var Registry = class {
         }
       )
     ]);
-    this.#removeSignalHandlers();
-    process.kill(process.pid, 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));
+      if (thresholdMs !== void 0 && Number.isFinite(thresholdMs) && thresholdMs > 0) {
+        return thresholdMs;
+      }
+    } catch (err) {
+      logger2().warn(
+        { err },
+        "failed to read actor stop threshold for shutdown grace"
+      );
+    }
+    return void 0;
   }
   #removeSignalHandlers() {
     for (const [signal, handler] of Object.entries(
@@ -5839,6 +6411,10 @@ var Registry = class {
   /**
    * Starts the actor envoy for standalone server deployments.
    *
+   * Set `RIVETKIT_RUNTIME_MODE=serverless` to instead bind an HTTP listener
+   * via `listen()` (Mode B). Mode A (envoy) and Mode B (listener) are
+   * mutually exclusive per registry instance.
+   *
    * @example
    * ```ts
    * const registry = setup({ use: { counter } });
@@ -5846,10 +6422,20 @@ var Registry = class {
    * ```
    */
   start() {
+    if (getRivetkitRuntimeMode() === "serverless") {
+      const publicDir = getRivetkitPublicDir() ?? "/public";
+      this.listen({ publicDir }).catch((error) => {
+        logger2().error({ error }, "auto-listen failed; exiting");
+        if (typeof process !== "undefined" && typeof process.exit === "function") {
+          process.exit(1);
+        }
+      });
+      return;
+    }
     const config = this.parseConfig();
     this.#startEnvoy(config, true);
   }
-  #printWelcome(config, kind) {
+  #printWelcome(config, kind, listener) {
     if (config.noWelcome || this.#welcomePrinted) return;
     this.#welcomePrinted = true;
     const logLine = (label, value) => {
@@ -5864,13 +6450,20 @@ var Registry = class {
       logLine("Namespace", config.namespace);
     }
     if (config.endpoint) {
-      const endpointType = config.endpoint === ENGINE_ENDPOINT ? "local native" : "remote";
+      const endpointType = config.startEngine || isLocalEngineEndpoint(config.endpoint) ? "local native" : "remote";
       logLine("Endpoint", `${config.endpoint} (${endpointType})`);
     }
     if (kind === "serverless" && config.publicEndpoint) {
       logLine("Client", config.publicEndpoint);
     }
     logLine("Actors", Object.keys(config.use).length.toString());
+    if (listener) {
+      const host = listener.host ?? "0.0.0.0";
+      logLine("Listening", `http://${host}:${listener.port}`);
+      if (listener.publicDir) {
+        logLine("Public Dir", listener.publicDir);
+      }
+    }
     console.log();
   }
 };
@@ -5886,6 +6479,12 @@ 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);
 }