rivetkit 2.0.38 → 2.0.40

package/dist/tsup/{chunk-IWXMFQDT.cjs → chunk-4KNL47JA.cjs}

@@ -31,7 +31,7 @@ var _chunkKDFWJKMJcjs = require('./chunk-KDFWJKMJ.cjs');
 
 
 
-var _chunkO433HWWGcjs = require('./chunk-O433HWWG.cjs');
+var _chunkHZOBB3KPcjs = require('./chunk-HZOBB3KP.cjs');
 
 
 
@@ -152,12 +152,215 @@ function deserializeActorKey(keyString) {
 // src/actor/router-endpoints.ts
 
 
+// src/actor/instance/keys.ts
+var KEYS = {
+  PERSIST_DATA: Uint8Array.from([1]),
+  CONN_PREFIX: Uint8Array.from([2]),
+  // Prefix for connection keys
+  INSPECTOR_TOKEN: Uint8Array.from([3]),
+  // Inspector token key
+  KV: Uint8Array.from([4])
+  // Prefix for user-facing KV storage
+};
+function makePrefixedKey(key) {
+  const prefixed = new Uint8Array(KEYS.KV.length + key.length);
+  prefixed.set(KEYS.KV, 0);
+  prefixed.set(key, KEYS.KV.length);
+  return prefixed;
+}
+function removePrefixFromKey(prefixedKey) {
+  return prefixedKey.slice(KEYS.KV.length);
+}
+function makeConnKey(connId) {
+  const encoder = new TextEncoder();
+  const connIdBytes = encoder.encode(connId);
+  const key = new Uint8Array(KEYS.CONN_PREFIX.length + connIdBytes.length);
+  key.set(KEYS.CONN_PREFIX, 0);
+  key.set(connIdBytes, KEYS.CONN_PREFIX.length);
+  return key;
+}
+
+// src/actor/instance/kv.ts
+var textEncoder = new TextEncoder();
+var textDecoder = new TextDecoder();
+function encodeKey(key, keyType) {
+  if (key instanceof Uint8Array) {
+    return key;
+  }
+  const resolvedKeyType = _nullishCoalesce(keyType, () => ( "text"));
+  if (resolvedKeyType === "binary") {
+    throw new TypeError("Expected a Uint8Array when keyType is binary");
+  }
+  return textEncoder.encode(key);
+}
+function decodeKey(key, keyType) {
+  const resolvedKeyType = _nullishCoalesce(keyType, () => ( "text"));
+  switch (resolvedKeyType) {
+    case "text":
+      return textDecoder.decode(key);
+    case "binary":
+      return key;
+    default:
+      throw new TypeError("Invalid kv key type");
+  }
+}
+function resolveValueType(value) {
+  if (typeof value === "string") {
+    return "text";
+  }
+  if (value instanceof Uint8Array) {
+    return "binary";
+  }
+  if (value instanceof ArrayBuffer) {
+    return "arrayBuffer";
+  }
+  throw new TypeError("Invalid kv value");
+}
+function encodeValue(value, options) {
+  const type = _nullishCoalesce((options == null ? void 0 : options.type), () => ( resolveValueType(value)));
+  switch (type) {
+    case "text":
+      if (typeof value !== "string") {
+        throw new TypeError("Expected a string when type is text");
+      }
+      return textEncoder.encode(value);
+    case "arrayBuffer":
+      if (!(value instanceof ArrayBuffer)) {
+        throw new TypeError("Expected an ArrayBuffer when type is arrayBuffer");
+      }
+      return new Uint8Array(value);
+    case "binary":
+      if (!(value instanceof Uint8Array)) {
+        throw new TypeError("Expected a Uint8Array when type is binary");
+      }
+      return value;
+    default:
+      throw new TypeError("Invalid kv value type");
+  }
+}
+function decodeValue(value, options) {
+  const type = _nullishCoalesce((options == null ? void 0 : options.type), () => ( "text"));
+  switch (type) {
+    case "text":
+      return textDecoder.decode(value);
+    case "arrayBuffer": {
+      const copy = new Uint8Array(value.byteLength);
+      copy.set(value);
+      return copy.buffer;
+    }
+    case "binary":
+      return value;
+    default:
+      throw new TypeError("Invalid kv value type");
+  }
+}
+var ActorKv = class {
+  #driver;
+  #actorId;
+  constructor(driver, actorId) {
+    this.#driver = driver;
+    this.#actorId = actorId;
+  }
+  /**
+   * Get a single value by key.
+   */
+  async get(key, options) {
+    const results = await this.#driver.kvBatchGet(this.#actorId, [
+      makePrefixedKey(encodeKey(key))
+    ]);
+    const result = results[0];
+    if (!result) {
+      return null;
+    }
+    return decodeValue(result, options);
+  }
+  /**
+   * Get multiple values by keys.
+   */
+  async getBatch(keys, options) {
+    const prefixedKeys = keys.map(
+      (key) => makePrefixedKey(encodeKey(key))
+    );
+    const results = await this.#driver.kvBatchGet(
+      this.#actorId,
+      prefixedKeys
+    );
+    return results.map(
+      (result) => result ? decodeValue(result, options) : null
+    );
+  }
+  /**
+   * Put a single key-value pair.
+   */
+  async put(key, value, options) {
+    await this.#driver.kvBatchPut(this.#actorId, [
+      [makePrefixedKey(encodeKey(key)), encodeValue(value, options)]
+    ]);
+  }
+  /**
+   * Put multiple key-value pairs.
+   */
+  async putBatch(entries, options) {
+    const prefixedEntries = entries.map(
+      ([key, value]) => [
+        makePrefixedKey(encodeKey(key)),
+        encodeValue(value, options)
+      ]
+    );
+    await this.#driver.kvBatchPut(this.#actorId, prefixedEntries);
+  }
+  /**
+   * Delete a single key.
+   */
+  async delete(key) {
+    await this.#driver.kvBatchDelete(this.#actorId, [
+      makePrefixedKey(encodeKey(key))
+    ]);
+  }
+  /**
+   * Delete multiple keys.
+   */
+  async deleteBatch(keys) {
+    const prefixedKeys = keys.map(
+      (key) => makePrefixedKey(encodeKey(key))
+    );
+    await this.#driver.kvBatchDelete(this.#actorId, prefixedKeys);
+  }
+  /**
+   * List all keys with a given prefix.
+   * Returns key-value pairs where keys have the user prefix removed.
+   */
+  async list(prefix, options) {
+    const prefixedPrefix = makePrefixedKey(
+      encodeKey(prefix, options == null ? void 0 : options.keyType)
+    );
+    const results = await this.#driver.kvListPrefix(
+      this.#actorId,
+      prefixedPrefix
+    );
+    return results.map(([key, value]) => [
+      decodeKey(removePrefixFromKey(key), options == null ? void 0 : options.keyType),
+      decodeValue(value, options)
+    ]);
+  }
+};
+
 // src/actor/contexts/base/actor.ts
 var ActorContext = class {
   #actor;
+  #kv;
   constructor(actor) {
     this.#actor = actor;
   }
+  /**
+   * Gets the KV storage interface.
+   */
+  get kv() {
+    if (!this.#kv) {
+      this.#kv = new ActorKv(this.#actor.driver, this.#actor.id);
+    }
+    return this.#kv;
+  }
   /**
    * Get the actor state
    *
@@ -385,7 +588,7 @@ function contentTypeForEncoding(encoding) {
   } else if (encoding === "cbor" || encoding === "bare") {
     return "application/octet-stream";
   } else {
-    _chunkO433HWWGcjs.assertUnreachable.call(void 0, encoding);
+    _chunkHZOBB3KPcjs.assertUnreachable.call(void 0, encoding);
   }
 }
 function serializeWithEncoding(encoding, value, versionedDataHandler, version, zodSchema, toJson, toBare) {
@@ -412,7 +615,7 @@ function serializeWithEncoding(encoding, value, versionedDataHandler, version, z
       version
     );
   } else {
-    _chunkO433HWWGcjs.assertUnreachable.call(void 0, encoding);
+    _chunkHZOBB3KPcjs.assertUnreachable.call(void 0, encoding);
   }
 }
 function deserializeWithEncoding(encoding, buffer, versionedDataHandler, zodSchema, fromJson, fromBare) {
@@ -448,13 +651,13 @@ function deserializeWithEncoding(encoding, buffer, versionedDataHandler, zodSche
     const bareValue = versionedDataHandler.deserializeWithEmbeddedVersion(buffer);
     return fromBare(bareValue);
   } else {
-    _chunkO433HWWGcjs.assertUnreachable.call(void 0, encoding);
+    _chunkHZOBB3KPcjs.assertUnreachable.call(void 0, encoding);
   }
 }
 
 // src/actor/log.ts
 function loggerWithoutContext() {
-  return _chunkO433HWWGcjs.getLogger.call(void 0, "actor-runtime");
+  return _chunkHZOBB3KPcjs.getLogger.call(void 0, "actor-runtime");
 }
 
 // src/actor/utils.ts
@@ -1568,7 +1771,7 @@ async function handleAction(c, config6, actorDriver, actionName, actorId) {
     (value) => ({ output: value }),
     // BARE/CBOR: output needs to be CBOR-encoded to ArrayBuffer
     (value) => ({
-      output: _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, cbor2.encode(value))
+      output: _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, cbor2.encode(value))
     })
   );
   const messageSize = serialized instanceof Uint8Array ? serialized.byteLength : serialized.length;
@@ -1611,7 +1814,7 @@ function getRequestEncoding(req) {
   return result.data;
 }
 function getRequestExposeInternalError(_req) {
-  return _chunkO433HWWGcjs.getEnvUniversal.call(void 0, "RIVET_EXPOSE_ERRORS") === "1" || _chunkO433HWWGcjs.getEnvUniversal.call(void 0, "NODE_ENV") === "development";
+  return _chunkHZOBB3KPcjs.getEnvUniversal.call(void 0, "RIVET_EXPOSE_ERRORS") === "1" || _chunkHZOBB3KPcjs.getEnvUniversal.call(void 0, "NODE_ENV") === "development";
 }
 function getRequestConnParams(req) {
   const paramsParam = req.header(HEADER_CONN_PARAMS);
@@ -1622,7 +1825,7 @@ function getRequestConnParams(req) {
     return JSON.parse(paramsParam);
   } catch (err) {
     throw new (0, _chunkSNAUKDDKcjs.InvalidParams)(
-      `Invalid params JSON: ${_chunkO433HWWGcjs.stringifyError.call(void 0, err)}`
+      `Invalid params JSON: ${_chunkHZOBB3KPcjs.stringifyError.call(void 0, err)}`
     );
   }
 }
@@ -1637,14 +1840,14 @@ var ENGINE_ENDPOINT = `http://127.0.0.1:${ENGINE_PORT}`;
 // src/inspector/config.ts
 
 var defaultTokenFn = () => {
-  const envToken = _chunkO433HWWGcjs.getRivetkitInspectorToken.call(void 0, );
+  const envToken = _chunkHZOBB3KPcjs.getRivetkitInspectorToken.call(void 0, );
   if (envToken) {
     return envToken;
   }
   return "";
 };
 var defaultEnabled = () => {
-  return _chunkO433HWWGcjs.isDev.call(void 0, ) || !_chunkO433HWWGcjs.getRivetkitInspectorDisable.call(void 0, );
+  return _chunkHZOBB3KPcjs.isDev.call(void 0, ) || !_chunkHZOBB3KPcjs.getRivetkitInspectorDisable.call(void 0, );
 };
 var InspectorConfigSchema = _zod.z.object({
   enabled: _zod.z.boolean().default(defaultEnabled),
@@ -1749,10 +1952,10 @@ var DriverConfigSchema = _zod.z.object({
 
 var RunnerConfigSchema = _zod.z.object({
   // MARK: Runner
-  totalSlots: _zod.z.number().default(() => _nullishCoalesce(_chunkO433HWWGcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
-  runnerName: _zod.z.string().default(() => _nullishCoalesce(_chunkO433HWWGcjs.getRivetRunner.call(void 0, ), () => ( "default"))),
-  runnerKey: _zod.z.string().optional().transform((x) => _nullishCoalesce(x, () => ( _chunkO433HWWGcjs.getRivetRunnerKey.call(void 0, )))),
-  version: _zod.z.number().default(() => _nullishCoalesce(_chunkO433HWWGcjs.getRivetRunnerVersion.call(void 0, ), () => ( 1)))
+  totalSlots: _zod.z.number().default(() => _nullishCoalesce(_chunkHZOBB3KPcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
+  runnerName: _zod.z.string().default(() => _nullishCoalesce(_chunkHZOBB3KPcjs.getRivetRunner.call(void 0, ), () => ( "default"))),
+  runnerKey: _zod.z.string().optional().transform((x) => _nullishCoalesce(x, () => ( _chunkHZOBB3KPcjs.getRivetRunnerKey.call(void 0, )))),
+  version: _zod.z.number().default(() => _nullishCoalesce(_chunkHZOBB3KPcjs.getRivetRunnerVersion.call(void 0, ), () => ( 1)))
 });
 
 // src/registry/config/serverless.ts
@@ -1776,9 +1979,9 @@ var ServerlessConfigSchema = _zod.z.object({
    * Downloads and starts the full Rust engine process.
    * Auto-enabled in development mode when no endpoint is provided.
    */
-  spawnEngine: _zod.z.boolean().default(() => _chunkO433HWWGcjs.getRivetRunEngine.call(void 0, )),
+  spawnEngine: _zod.z.boolean().default(() => _chunkHZOBB3KPcjs.getRivetRunEngine.call(void 0, )),
   /** @experimental */
-  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkO433HWWGcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkO433HWWGcjs.VERSION))),
+  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkHZOBB3KPcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkHZOBB3KPcjs.VERSION))),
   /**
    * @experimental
    *
@@ -1807,7 +2010,7 @@ var ServerlessConfigSchema = _zod.z.object({
    *
    * Can also be set via RIVET_PUBLIC_ENDPOINT environment variable.
    */
-  publicEndpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkO433HWWGcjs.getRivetPublicEndpoint.call(void 0, )))),
+  publicEndpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkHZOBB3KPcjs.getRivetPublicEndpoint.call(void 0, )))),
   /**
    * Token that clients should use when connecting via the public endpoint.
    *
@@ -1815,7 +2018,7 @@ var ServerlessConfigSchema = _zod.z.object({
    *
    * Can also be specified in the publicEndpoint URL as `https://namespace:token@host`.
    */
-  publicToken: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkO433HWWGcjs.getRivetPublicToken.call(void 0, ))))
+  publicToken: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkHZOBB3KPcjs.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.
@@ -1857,7 +2060,7 @@ var RegistryConfigSchema = _zod.z.object({
    * */
   logging: _zod.z.object({
     baseLogger: _zod.z.custom().optional(),
-    level: _chunkO433HWWGcjs.LogLevelSchema.optional()
+    level: _chunkHZOBB3KPcjs.LogLevelSchema.optional()
   }).optional().default(() => ({})),
   // MARK: Routing
   // // This is a function to allow for lazy configuration of upgradeWebSocket on the
@@ -1875,9 +2078,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, () => ( _chunkO433HWWGcjs.getRivetEngine.call(void 0, ))), () => ( _chunkO433HWWGcjs.getRivetEndpoint.call(void 0, )))),
-  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkO433HWWGcjs.getRivetToken.call(void 0, )))),
-  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkO433HWWGcjs.getRivetNamespace.call(void 0, )))),
+  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunkHZOBB3KPcjs.getRivetEngine.call(void 0, ))), () => ( _chunkHZOBB3KPcjs.getRivetEndpoint.call(void 0, )))),
+  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkHZOBB3KPcjs.getRivetToken.call(void 0, )))),
+  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkHZOBB3KPcjs.getRivetNamespace.call(void 0, )))),
   headers: _zod.z.record(_zod.z.string(), _zod.z.string()).optional().default({}),
   // MARK: Client
   // TODO:
@@ -1902,6 +2105,12 @@ var RegistryConfigSchema = _zod.z.object({
    * What port to run the manager on.
    */
   managerPort: _zod.z.number().optional().default(6420),
+  /**
+   * @experimental
+   *
+   * What host to bind the manager server to.
+   */
+  managerHost: _zod.z.string().optional(),
   /** @experimental */
   inspector: InspectorConfigSchema,
   // MARK: Runtime-specific
@@ -1912,7 +2121,7 @@ var RegistryConfigSchema = _zod.z.object({
     () => RunnerConfigSchema.parse({})
   )
 }).transform((config6, ctx) => {
-  const isDevEnv = _chunkO433HWWGcjs.isDev.call(void 0, );
+  const isDevEnv = _chunkHZOBB3KPcjs.isDev.call(void 0, );
   const parsedEndpoint = config6.endpoint ? tryParseEndpoint(ctx, {
     endpoint: config6.endpoint,
     path: ["endpoint"],
@@ -2017,7 +2226,7 @@ var DocRegistryConfigSchema = _zod.z.object({
   maxOutgoingMessageSize: _zod.z.number().optional().describe("Maximum size of outgoing WebSocket messages in bytes. Default: 1048576"),
   noWelcome: _zod.z.boolean().optional().describe("Disable the welcome message on startup. Default: false"),
   logging: _zod.z.object({
-    level: _chunkO433HWWGcjs.LogLevelSchema.optional().describe("Log level for RivetKit. Default: 'warn'")
+    level: _chunkHZOBB3KPcjs.LogLevelSchema.optional().describe("Log level for RivetKit. Default: 'warn'")
   }).optional().describe("Logging configuration."),
   endpoint: _zod.z.string().optional().describe("Endpoint URL to connect to Rivet Engine. Supports URL auth syntax: https://namespace:token@api.rivet.dev. Can also be set via RIVET_ENDPOINT environment variable."),
   token: _zod.z.string().optional().describe("Authentication token for Rivet Engine. Can also be set via RIVET_TOKEN environment variable."),
@@ -2033,7 +2242,7 @@ var DocRegistryConfigSchema = _zod.z.object({
 
 // src/common/router.ts
 function logger() {
-  return _chunkO433HWWGcjs.getLogger.call(void 0, "router");
+  return _chunkHZOBB3KPcjs.getLogger.call(void 0, "router");
 }
 function loggerMiddleware(logger4) {
   return async (c, next) => {
@@ -2051,7 +2260,7 @@ function loggerMiddleware(logger4) {
       reqSize: c.req.header("content-length"),
       resSize: c.res.headers.get("content-length"),
       userAgent: c.req.header("user-agent"),
-      ..._chunkO433HWWGcjs.getLogHeaders.call(void 0, ) ? { allHeaders: JSON.stringify(c.req.header()) } : {}
+      ..._chunkHZOBB3KPcjs.getLogHeaders.call(void 0, ) ? { allHeaders: JSON.stringify(c.req.header()) } : {}
     });
   };
 }
@@ -2060,7 +2269,7 @@ function handleRouteNotFound(c) {
 }
 function handleRouteError(error, c) {
   const exposeInternalError = getRequestExposeInternalError(c.req.raw);
-  const { statusCode, group, code, message, metadata } = _chunkO433HWWGcjs.deconstructError.call(void 0, 
+  const { statusCode, group, code, message, metadata } = _chunkHZOBB3KPcjs.deconstructError.call(void 0, 
     error,
     logger(),
     {
@@ -2094,7 +2303,7 @@ function handleRouteError(error, c) {
       group: value.group,
       code: value.code,
       message: value.message,
-      metadata: value.metadata ? _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, cbor3.encode(value.metadata)) : null
+      metadata: value.metadata ? _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, cbor3.encode(value.metadata)) : null
     })
   );
   return c.body(output, { status: statusCode });
@@ -2102,7 +2311,7 @@ function handleRouteError(error, c) {
 function handleMetadataRequest(c, config6, runnerKind, clientEndpoint, clientNamespace, clientToken) {
   const response = {
     runtime: "rivetkit",
-    version: _chunkO433HWWGcjs.VERSION,
+    version: _chunkHZOBB3KPcjs.VERSION,
     runner: {
       kind: runnerKind
     },
@@ -2117,7 +2326,7 @@ function handleHealthRequest(c) {
   return c.json({
     status: "ok",
     runtime: "rivetkit",
-    version: _chunkO433HWWGcjs.VERSION
+    version: _chunkHZOBB3KPcjs.VERSION
   });
 }
 
@@ -2232,7 +2441,7 @@ var cors = () => {
 
 // src/utils/router.ts
 function logger2() {
-  return _chunkO433HWWGcjs.getLogger.call(void 0, "router");
+  return _chunkHZOBB3KPcjs.getLogger.call(void 0, "router");
 }
 function createRouter(basePath, builder) {
   const router = new (0, _zodopenapi.OpenAPIHono)({ strict: false }).basePath(basePath);
@@ -2291,7 +2500,7 @@ function buildOpenApiRequestBody(schema) {
 
 // src/inspector/log.ts
 function inspectorLogger() {
-  return _chunkO433HWWGcjs.getLogger.call(void 0, "inspector");
+  return _chunkHZOBB3KPcjs.getLogger.call(void 0, "inspector");
 }
 
 // src/inspector/handler.ts
@@ -2423,7 +2632,7 @@ async function handleWebSocketInspectorConnect({
             }
           });
         } else {
-          _chunkO433HWWGcjs.assertUnreachable.call(void 0, message.body);
+          _chunkHZOBB3KPcjs.assertUnreachable.call(void 0, message.body);
         }
       } catch (error) {
         inspectorLogger().warn(
@@ -2683,7 +2892,7 @@ var StateManager = class {
   #handleChange(path, value) {
     if (path.startsWith("state")) {
       let invalidPath = "";
-      if (!_chunkO433HWWGcjs.isCborSerializable.call(void 0, 
+      if (!_chunkHZOBB3KPcjs.isCborSerializable.call(void 0, 
         value,
         (invalidPathPart) => {
           invalidPath = invalidPathPart;
@@ -2875,7 +3084,7 @@ var Conn = (_class = class {
             tag: "Event",
             val: {
               name: value.name,
-              args: _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, cbor4.encode(value.args))
+              args: _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, cbor4.encode(value.args))
             }
           }
         })
@@ -2918,7 +3127,7 @@ function getValueLength(value) {
   } else if (value instanceof ArrayBuffer || value instanceof SharedArrayBuffer || value instanceof Uint8Array) {
     return value.byteLength;
   } else {
-    _chunkO433HWWGcjs.assertUnreachable.call(void 0, value);
+    _chunkHZOBB3KPcjs.assertUnreachable.call(void 0, value);
   }
 }
 async function inputDataToBuffer(data) {
@@ -3023,7 +3232,7 @@ async function processMessage(message, actor, conn, handler) {
               tag: "ActionResponse",
               val: {
                 id,
-                output: _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, cbor5.encode(value))
+                output: _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, cbor5.encode(value))
               }
             }
           })
@@ -3051,10 +3260,10 @@ async function processMessage(message, actor, conn, handler) {
         subscribe
       });
     } else {
-      _chunkO433HWWGcjs.assertUnreachable.call(void 0, message.body);
+      _chunkHZOBB3KPcjs.assertUnreachable.call(void 0, message.body);
     }
   } catch (error) {
-    const { group, code, message: message2, metadata } = _chunkO433HWWGcjs.deconstructError.call(void 0, 
+    const { group, code, message: message2, metadata } = _chunkHZOBB3KPcjs.deconstructError.call(void 0, 
       error,
       actor.rLog,
       {
@@ -3062,7 +3271,7 @@ async function processMessage(message, actor, conn, handler) {
         actionId,
         actionName
       },
-      _chunkO433HWWGcjs.getEnvUniversal.call(void 0, "RIVET_EXPOSE_ERRORS") === "1" || _chunkO433HWWGcjs.getEnvUniversal.call(void 0, "NODE_ENV") === "development"
+      _chunkHZOBB3KPcjs.getEnvUniversal.call(void 0, "RIVET_EXPOSE_ERRORS") === "1" || _chunkHZOBB3KPcjs.getEnvUniversal.call(void 0, "NODE_ENV") === "development"
     );
     actor.rLog.debug({
       msg: "sending error response",
@@ -3105,7 +3314,7 @@ async function processMessage(message, actor, conn, handler) {
               group: value.group,
               code: value.code,
               message: value.message,
-              metadata: value.metadata ? _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, 
+              metadata: value.metadata ? _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, 
                 cbor5.encode(value.metadata)
               ) : null,
               actionId: value.actionId !== void 0 ? value.actionId : null
@@ -3131,10 +3340,11 @@ async function routeWebSocket(request, requestPath, requestHeaders, config6, act
       requestPath,
       isHibernatable
     });
-    const closePromiseResolvers = _chunkO433HWWGcjs.promiseWithResolvers.call(void 0, );
+    const closePromiseResolvers = _chunkHZOBB3KPcjs.promiseWithResolvers.call(void 0, );
+    const requestPathWithoutQuery = requestPath.split("?")[0];
     let handler;
     let connDriver;
-    if (requestPath === PATH_CONNECT) {
+    if (requestPathWithoutQuery === PATH_CONNECT) {
       const { driver, setWebSocket } = createWebSocketDriver(
         isHibernatable ? { gatewayId, requestId } : void 0,
         encoding,
@@ -3143,14 +3353,14 @@ async function routeWebSocket(request, requestPath, requestHeaders, config6, act
       );
       handler = handleWebSocketConnect.bind(void 0, setWebSocket);
       connDriver = driver;
-    } else if (requestPath === PATH_WEBSOCKET_BASE || requestPath.startsWith(PATH_WEBSOCKET_PREFIX)) {
+    } else if (requestPathWithoutQuery === PATH_WEBSOCKET_BASE || requestPathWithoutQuery.startsWith(PATH_WEBSOCKET_PREFIX)) {
       const { driver, setWebSocket } = createRawWebSocketDriver(
         isHibernatable ? { gatewayId, requestId } : void 0,
         closePromiseResolvers.promise
       );
       handler = handleRawWebSocket.bind(void 0, setWebSocket);
       connDriver = driver;
-    } else if (requestPath === PATH_INSPECTOR_CONNECT) {
+    } else if (requestPathWithoutQuery === PATH_INSPECTOR_CONNECT) {
       if (!actor.inspectorToken) {
         throw "WebSocket Inspector Unauthorized: actor does not provide inspector access";
       }
@@ -3184,7 +3394,7 @@ async function routeWebSocket(request, requestPath, requestHeaders, config6, act
       exposeInternalError
     });
   } catch (error) {
-    const { group, code } = _chunkO433HWWGcjs.deconstructError.call(void 0, 
+    const { group, code } = _chunkHZOBB3KPcjs.deconstructError.call(void 0, 
       error,
       loggerWithoutContext(),
       {},
@@ -3236,7 +3446,7 @@ async function handleWebSocketConnect(setWebSocket, {
         maxIncomingMessageSize: runConfig.maxIncomingMessageSize
       }).then((message) => {
         actor.processMessage(message, conn).catch((error) => {
-          const { code } = _chunkO433HWWGcjs.deconstructError.call(void 0, 
+          const { code } = _chunkHZOBB3KPcjs.deconstructError.call(void 0, 
             error,
             actor.rLog,
             {
@@ -3247,7 +3457,7 @@ async function handleWebSocketConnect(setWebSocket, {
           ws.close(1011, code);
         });
       }).catch((error) => {
-        const { code } = _chunkO433HWWGcjs.deconstructError.call(void 0, 
+        const { code } = _chunkHZOBB3KPcjs.deconstructError.call(void 0, 
           error,
           actor.rLog,
           {
@@ -3282,7 +3492,7 @@ async function handleWebSocketConnect(setWebSocket, {
       try {
         actor.rLog.warn({ msg: "websocket error" });
       } catch (error) {
-        _chunkO433HWWGcjs.deconstructError.call(void 0, 
+        _chunkHZOBB3KPcjs.deconstructError.call(void 0, 
           error,
           actor.rLog,
           { wsEvent: "error" },
@@ -3338,7 +3548,7 @@ function parseWebSocketProtocols(protocols) {
       }
     }
   }
-  const encoding = EncodingSchema.parse(encodingRaw);
+  const encoding = EncodingSchema.parse(_nullishCoalesce(encodingRaw, () => ( "json")));
   const connParams = connParamsRaw ? JSON.parse(connParamsRaw) : void 0;
   return { encoding, connParams };
 }
@@ -3351,7 +3561,7 @@ function truncateRawWebSocketPathPrefix(path) {
 
 // src/manager/log.ts
 function logger3() {
-  return _chunkO433HWWGcjs.getLogger.call(void 0, "actor-manager");
+  return _chunkHZOBB3KPcjs.getLogger.call(void 0, "actor-manager");
 }
 
 // src/manager/gateway.ts
@@ -3587,7 +3797,7 @@ async function createTestWebSocketProxy(proxyToActorWsPromise) {
     promise: clientToProxyWsPromise,
     resolve: clientToProxyWsResolve,
     reject: clientToProxyWsReject
-  } = _chunkO433HWWGcjs.promiseWithResolvers.call(void 0, );
+  } = _chunkHZOBB3KPcjs.promiseWithResolvers.call(void 0, );
   try {
     logger3().debug({ msg: "awaiting client websocket promise" });
     proxyToActorWs = await proxyToActorWsPromise;
@@ -3981,12 +4191,12 @@ function buildManagerRouter(config6, managerDriver, getUpgradeWebSocket) {
         responses: buildOpenApiResponses(ActorsKvGetResponseSchema)
       });
       router.openapi(route, async (c) => {
-        if (_chunkO433HWWGcjs.getNodeEnv.call(void 0, ) === "development" && !config6.token) {
+        if (_chunkHZOBB3KPcjs.getNodeEnv.call(void 0, ) === "development" && !config6.token) {
           logger3().warn({
             msg: "RIVET_TOKEN is not set, skipping KV store access checks in development mode. This endpoint will be disabled in production, unless you set the token."
           });
         }
-        if (_chunkO433HWWGcjs.getNodeEnv.call(void 0, ) !== "development") {
+        if (_chunkHZOBB3KPcjs.getNodeEnv.call(void 0, ) !== "development") {
           if (!config6.token) {
             throw new (0, _chunkSNAUKDDKcjs.RestrictedFeature)("KV store access");
           }
@@ -4022,7 +4232,7 @@ function buildManagerRouter(config6, managerDriver, getUpgradeWebSocket) {
           const output = await managerDriver[method](...args);
           response = { ok: output };
         } catch (rawErr) {
-          const err = _chunkO433HWWGcjs.deconstructError.call(void 0, rawErr, logger3(), {}, true);
+          const err = _chunkHZOBB3KPcjs.deconstructError.call(void 0, rawErr, logger3(), {}, true);
           response = { err };
         }
         return c.body(cbor6.encode(response));
@@ -4080,7 +4290,7 @@ function buildManagerRouter(config6, managerDriver, getUpgradeWebSocket) {
           return await createTestWebSocketProxy(
             clientToProxyWsPromise
           );
-        })(c, _chunkO433HWWGcjs.noopNext.call(void 0, ));
+        })(c, _chunkHZOBB3KPcjs.noopNext.call(void 0, ));
       });
       router.all(".test/inline-driver/send-request/*", async (c) => {
         const actorId = c.req.header(HEADER_ACTOR_ID);
@@ -4110,9 +4320,9 @@ function buildManagerRouter(config6, managerDriver, getUpgradeWebSocket) {
         } catch (error) {
           logger3().error({
             msg: "error in test inline raw http",
-            error: _chunkO433HWWGcjs.stringifyError.call(void 0, error)
+            error: _chunkHZOBB3KPcjs.stringifyError.call(void 0, error)
           });
-          const err = _chunkO433HWWGcjs.deconstructError.call(void 0, error, logger3(), {}, true);
+          const err = _chunkHZOBB3KPcjs.deconstructError.call(void 0, error, logger3(), {}, true);
           return c.json(
             {
               error: {
@@ -4160,7 +4370,7 @@ function buildManagerRouter(config6, managerDriver, getUpgradeWebSocket) {
         } catch (error) {
           logger3().error({
             msg: "error forcing disconnect",
-            error: _chunkO433HWWGcjs.stringifyError.call(void 0, error)
+            error: _chunkHZOBB3KPcjs.stringifyError.call(void 0, error)
           });
           return c.text(`Error: ${error}`, 500);
         }
@@ -4200,23 +4410,6 @@ function createApiActor(actor) {
 // src/driver-helpers/utils.ts
 
 
-// src/actor/instance/kv.ts
-var KEYS = {
-  PERSIST_DATA: Uint8Array.from([1]),
-  CONN_PREFIX: Uint8Array.from([2]),
-  // Prefix for connection keys
-  INSPECTOR_TOKEN: Uint8Array.from([3])
-  // Inspector token key
-};
-function makeConnKey(connId) {
-  const encoder = new TextEncoder();
-  const connIdBytes = encoder.encode(connId);
-  const key = new Uint8Array(KEYS.CONN_PREFIX.length + connIdBytes.length);
-  key.set(KEYS.CONN_PREFIX, 0);
-  key.set(connIdBytes, KEYS.CONN_PREFIX.length);
-  return key;
-}
-
 // src/schemas/actor-persist/versioned.ts
 
 
@@ -4888,9 +5081,9 @@ var CONN_VERSIONED = _vbare.createVersionedDataHandler.call(void 0, {
 // src/driver-helpers/utils.ts
 function serializeEmptyPersistData(input) {
   const persistData = {
-    input: input !== void 0 ? _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, cbor7.encode(input)) : null,
+    input: input !== void 0 ? _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, cbor7.encode(input)) : null,
     hasInitialized: false,
-    state: _chunkO433HWWGcjs.bufferToArrayBuffer.call(void 0, cbor7.encode(void 0)),
+    state: _chunkHZOBB3KPcjs.bufferToArrayBuffer.call(void 0, cbor7.encode(void 0)),
     scheduledEvents: []
   };
   return ACTOR_VERSIONED.serializeWithEmbeddedVersion(
@@ -4998,5 +5191,6 @@ function getInitialActorKvState(input) {
 
 
 
-exports.createHttpDriver = createHttpDriver; exports.CURRENT_VERSION = CURRENT_VERSION; exports.TO_SERVER_VERSIONED = TO_SERVER_VERSIONED2; exports.TO_CLIENT_VERSIONED = TO_CLIENT_VERSIONED2; exports.HTTP_ACTION_REQUEST_VERSIONED = HTTP_ACTION_REQUEST_VERSIONED; exports.HTTP_ACTION_RESPONSE_VERSIONED = HTTP_ACTION_RESPONSE_VERSIONED; exports.HTTP_RESPONSE_ERROR_VERSIONED = HTTP_RESPONSE_ERROR_VERSIONED; exports.ToClientSchema = ToClientSchema; exports.ToServerSchema = ToServerSchema; exports.HttpActionRequestSchema = HttpActionRequestSchema; exports.HttpActionResponseSchema = HttpActionResponseSchema; exports.HttpResponseErrorSchema = HttpResponseErrorSchema; exports.uint8ArrayToBase64 = uint8ArrayToBase64; exports.contentTypeForEncoding = contentTypeForEncoding; exports.serializeWithEncoding = serializeWithEncoding; exports.deserializeWithEncoding = deserializeWithEncoding; exports.loggerWithoutContext = loggerWithoutContext; exports.assertUnreachable = assertUnreachable2; exports.DeadlineError = DeadlineError; exports.deadline = deadline; exports.generateSecureToken = generateSecureToken; exports.isStatePath = isStatePath; exports.isConnStatePath = isConnStatePath; exports.EncodingSchema = EncodingSchema; exports.CachedSerializer = CachedSerializer; exports.jsonStringifyCompat = jsonStringifyCompat; exports.CONN_CONNECTED_SYMBOL = CONN_CONNECTED_SYMBOL; exports.CONN_SPEAKS_RIVETKIT_SYMBOL = CONN_SPEAKS_RIVETKIT_SYMBOL; exports.CONN_DRIVER_SYMBOL = CONN_DRIVER_SYMBOL; exports.CONN_STATE_MANAGER_SYMBOL = CONN_STATE_MANAGER_SYMBOL; exports.CONN_SEND_MESSAGE_SYMBOL = CONN_SEND_MESSAGE_SYMBOL; exports.Conn = Conn; exports.ActorContext = ActorContext; exports.ActionContext = ActionContext; exports.CURRENT_VERSION2 = CURRENT_VERSION2; exports.ACTOR_VERSIONED = ACTOR_VERSIONED; exports.CONN_VERSIONED = CONN_VERSIONED; exports.BeforeConnectContext = BeforeConnectContext; exports.ConnectContext = ConnectContext; exports.CreateConnStateContext = CreateConnStateContext; exports.RequestContext = RequestContext; exports.WebSocketContext = WebSocketContext; exports.serializeActorKey = serializeActorKey; exports.deserializeActorKey = deserializeActorKey; exports.getValueLength = getValueLength; exports.inputDataToBuffer = inputDataToBuffer; exports.processMessage = processMessage; exports.KEYS = KEYS; exports.makeConnKey = makeConnKey; exports.PATH_CONNECT = PATH_CONNECT; exports.PATH_WEBSOCKET_BASE = PATH_WEBSOCKET_BASE; exports.PATH_WEBSOCKET_PREFIX = PATH_WEBSOCKET_PREFIX; exports.PATH_INSPECTOR_CONNECT = PATH_INSPECTOR_CONNECT; exports.HEADER_ACTOR_QUERY = HEADER_ACTOR_QUERY; exports.HEADER_ENCODING = HEADER_ENCODING; exports.HEADER_CONN_PARAMS = HEADER_CONN_PARAMS; exports.HEADER_ACTOR_ID = HEADER_ACTOR_ID; exports.HEADER_RIVET_TOKEN = HEADER_RIVET_TOKEN; exports.HEADER_RIVET_TARGET = HEADER_RIVET_TARGET; exports.HEADER_RIVET_ACTOR = HEADER_RIVET_ACTOR; exports.WS_PROTOCOL_STANDARD = WS_PROTOCOL_STANDARD; exports.WS_PROTOCOL_TARGET = WS_PROTOCOL_TARGET; exports.WS_PROTOCOL_ACTOR = WS_PROTOCOL_ACTOR; exports.WS_PROTOCOL_ENCODING = WS_PROTOCOL_ENCODING; exports.WS_PROTOCOL_CONN_PARAMS = WS_PROTOCOL_CONN_PARAMS; exports.WS_TEST_PROTOCOL_PATH = WS_TEST_PROTOCOL_PATH; exports.ALLOWED_PUBLIC_HEADERS = ALLOWED_PUBLIC_HEADERS; exports.handleAction = handleAction; exports.handleRawRequest = handleRawRequest; exports.ENGINE_PORT = ENGINE_PORT; exports.ENGINE_ENDPOINT = ENGINE_ENDPOINT; exports.InspectorConfigSchema = InspectorConfigSchema; exports.tryParseEndpoint = tryParseEndpoint; exports.DriverConfigSchema = DriverConfigSchema; exports.ActorsSchema = ActorsSchema; exports.TestConfigSchema = TestConfigSchema; exports.RegistryConfigSchema = RegistryConfigSchema; exports.buildActorNames = buildActorNames; exports.DocInspectorConfigSchema = DocInspectorConfigSchema; exports.DocConfigureRunnerPoolSchema = DocConfigureRunnerPoolSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocRunnerConfigSchema = DocRunnerConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.loggerMiddleware = loggerMiddleware; exports.handleRouteNotFound = handleRouteNotFound; exports.handleRouteError = handleRouteError; exports.handleMetadataRequest = handleMetadataRequest; exports.handleHealthRequest = handleHealthRequest; exports.routeWebSocket = routeWebSocket; exports.parseWebSocketProtocols = parseWebSocketProtocols; exports.truncateRawWebSocketPathPrefix = truncateRawWebSocketPathPrefix; exports.createRouter = createRouter; exports.buildManagerRouter = buildManagerRouter; exports.getInitialActorKvState = getInitialActorKvState;
-//# sourceMappingURL=chunk-IWXMFQDT.cjs.map
+
+exports.createHttpDriver = createHttpDriver; exports.CURRENT_VERSION = CURRENT_VERSION; exports.TO_SERVER_VERSIONED = TO_SERVER_VERSIONED2; exports.TO_CLIENT_VERSIONED = TO_CLIENT_VERSIONED2; exports.HTTP_ACTION_REQUEST_VERSIONED = HTTP_ACTION_REQUEST_VERSIONED; exports.HTTP_ACTION_RESPONSE_VERSIONED = HTTP_ACTION_RESPONSE_VERSIONED; exports.HTTP_RESPONSE_ERROR_VERSIONED = HTTP_RESPONSE_ERROR_VERSIONED; exports.ToClientSchema = ToClientSchema; exports.ToServerSchema = ToServerSchema; exports.HttpActionRequestSchema = HttpActionRequestSchema; exports.HttpActionResponseSchema = HttpActionResponseSchema; exports.HttpResponseErrorSchema = HttpResponseErrorSchema; exports.uint8ArrayToBase64 = uint8ArrayToBase64; exports.contentTypeForEncoding = contentTypeForEncoding; exports.serializeWithEncoding = serializeWithEncoding; exports.deserializeWithEncoding = deserializeWithEncoding; exports.loggerWithoutContext = loggerWithoutContext; exports.assertUnreachable = assertUnreachable2; exports.DeadlineError = DeadlineError; exports.deadline = deadline; exports.generateSecureToken = generateSecureToken; exports.isStatePath = isStatePath; exports.isConnStatePath = isConnStatePath; exports.EncodingSchema = EncodingSchema; exports.CachedSerializer = CachedSerializer; exports.jsonStringifyCompat = jsonStringifyCompat; exports.CONN_CONNECTED_SYMBOL = CONN_CONNECTED_SYMBOL; exports.CONN_SPEAKS_RIVETKIT_SYMBOL = CONN_SPEAKS_RIVETKIT_SYMBOL; exports.CONN_DRIVER_SYMBOL = CONN_DRIVER_SYMBOL; exports.CONN_STATE_MANAGER_SYMBOL = CONN_STATE_MANAGER_SYMBOL; exports.CONN_SEND_MESSAGE_SYMBOL = CONN_SEND_MESSAGE_SYMBOL; exports.Conn = Conn; exports.KEYS = KEYS; exports.makeConnKey = makeConnKey; exports.ActorKv = ActorKv; exports.ActorContext = ActorContext; exports.ActionContext = ActionContext; exports.CURRENT_VERSION2 = CURRENT_VERSION2; exports.ACTOR_VERSIONED = ACTOR_VERSIONED; exports.CONN_VERSIONED = CONN_VERSIONED; exports.BeforeConnectContext = BeforeConnectContext; exports.ConnectContext = ConnectContext; exports.CreateConnStateContext = CreateConnStateContext; exports.RequestContext = RequestContext; exports.WebSocketContext = WebSocketContext; exports.serializeActorKey = serializeActorKey; exports.deserializeActorKey = deserializeActorKey; exports.getValueLength = getValueLength; exports.inputDataToBuffer = inputDataToBuffer; exports.processMessage = processMessage; exports.PATH_CONNECT = PATH_CONNECT; exports.PATH_WEBSOCKET_BASE = PATH_WEBSOCKET_BASE; exports.PATH_WEBSOCKET_PREFIX = PATH_WEBSOCKET_PREFIX; exports.PATH_INSPECTOR_CONNECT = PATH_INSPECTOR_CONNECT; exports.HEADER_ACTOR_QUERY = HEADER_ACTOR_QUERY; exports.HEADER_ENCODING = HEADER_ENCODING; exports.HEADER_CONN_PARAMS = HEADER_CONN_PARAMS; exports.HEADER_ACTOR_ID = HEADER_ACTOR_ID; exports.HEADER_RIVET_TOKEN = HEADER_RIVET_TOKEN; exports.HEADER_RIVET_TARGET = HEADER_RIVET_TARGET; exports.HEADER_RIVET_ACTOR = HEADER_RIVET_ACTOR; exports.WS_PROTOCOL_STANDARD = WS_PROTOCOL_STANDARD; exports.WS_PROTOCOL_TARGET = WS_PROTOCOL_TARGET; exports.WS_PROTOCOL_ACTOR = WS_PROTOCOL_ACTOR; exports.WS_PROTOCOL_ENCODING = WS_PROTOCOL_ENCODING; exports.WS_PROTOCOL_CONN_PARAMS = WS_PROTOCOL_CONN_PARAMS; exports.WS_TEST_PROTOCOL_PATH = WS_TEST_PROTOCOL_PATH; exports.ALLOWED_PUBLIC_HEADERS = ALLOWED_PUBLIC_HEADERS; exports.handleAction = handleAction; exports.handleRawRequest = handleRawRequest; exports.ENGINE_PORT = ENGINE_PORT; exports.ENGINE_ENDPOINT = ENGINE_ENDPOINT; exports.InspectorConfigSchema = InspectorConfigSchema; exports.tryParseEndpoint = tryParseEndpoint; exports.DriverConfigSchema = DriverConfigSchema; exports.ActorsSchema = ActorsSchema; exports.TestConfigSchema = TestConfigSchema; exports.RegistryConfigSchema = RegistryConfigSchema; exports.buildActorNames = buildActorNames; exports.DocInspectorConfigSchema = DocInspectorConfigSchema; exports.DocConfigureRunnerPoolSchema = DocConfigureRunnerPoolSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocRunnerConfigSchema = DocRunnerConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.loggerMiddleware = loggerMiddleware; exports.handleRouteNotFound = handleRouteNotFound; exports.handleRouteError = handleRouteError; exports.handleMetadataRequest = handleMetadataRequest; exports.handleHealthRequest = handleHealthRequest; exports.routeWebSocket = routeWebSocket; exports.parseWebSocketProtocols = parseWebSocketProtocols; exports.truncateRawWebSocketPathPrefix = truncateRawWebSocketPathPrefix; exports.createRouter = createRouter; exports.buildManagerRouter = buildManagerRouter; exports.getInitialActorKvState = getInitialActorKvState;
+//# sourceMappingURL=chunk-4KNL47JA.cjs.map