@rivetkit/cloudflare-workers 2.2.1 → 2.3.0-rc.13

package/dist/mod.js

@@ -1,1093 +1,152 @@
-// src/actor-handler-do.ts
-import { DurableObject, env as env2 } from "cloudflare:workers";
-import invariant2 from "invariant";
-import { createActorRouter, createClientWithDriver as createClientWithDriver2 } from "rivetkit";
-import { getInitialActorKvState } from "rivetkit/driver-helpers";
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class;// src/mod.ts
+var _rivetkitwasm = require('@rivetkit/rivetkit-wasm'); var wasmBindings = _interopRequireWildcard(_rivetkitwasm);
+var _rivetkit_wasm_bgwasm = require('@rivetkit/rivetkit-wasm/rivetkit_wasm_bg.wasm'); var _rivetkit_wasm_bgwasm2 = _interopRequireDefault(_rivetkit_wasm_bgwasm);
 
-// src/actor-driver.ts
-import invariant from "invariant";
-import { lookupInRegistry } from "rivetkit";
-import { promiseWithResolvers } from "rivetkit/utils";
 
-// src/log.ts
-import { getLogger } from "rivetkit/log";
-function logger() {
-  return getLogger("driver-cloudflare-workers");
-}
-
-// src/actor-kv.ts
-var DEFAULT_LIST_LIMIT = 16384;
-function kvGet(sql, key) {
-  const cursor = sql.exec(
-    "SELECT value FROM _rivetkit_kv_storage WHERE key = ?",
-    key
-  );
-  const result = cursor.raw().next();
-  if (!result.done && result.value) {
-    return toUint8Array(result.value[0]);
-  }
-  return null;
-}
-function kvPut(sql, key, value) {
-  sql.exec(
-    "INSERT OR REPLACE INTO _rivetkit_kv_storage (key, value) VALUES (?, ?)",
-    key,
-    value
-  );
-}
-function kvDelete(sql, key) {
-  sql.exec("DELETE FROM _rivetkit_kv_storage WHERE key = ?", key);
-}
-function kvDeleteRange(sql, start, end) {
-  sql.exec(
-    "DELETE FROM _rivetkit_kv_storage WHERE key >= ? AND key < ?",
-    start,
-    end
-  );
-}
-function kvListPrefix(sql, prefix, options) {
-  const upperBound = computePrefixUpperBound(prefix);
-  if (upperBound) {
-    return kvListRange(sql, prefix, upperBound, options);
-  }
-  const direction = (options == null ? void 0 : options.reverse) ? "DESC" : "ASC";
-  const limit = (options == null ? void 0 : options.limit) ?? DEFAULT_LIST_LIMIT;
-  const cursor = sql.exec(
-    `SELECT key, value FROM _rivetkit_kv_storage WHERE key >= ? ORDER BY key ${direction} LIMIT ?`,
-    prefix,
-    limit
-  );
-  return readEntries(cursor);
-}
-function kvListRange(sql, start, end, options) {
-  const direction = (options == null ? void 0 : options.reverse) ? "DESC" : "ASC";
-  const limit = (options == null ? void 0 : options.limit) ?? DEFAULT_LIST_LIMIT;
-  const cursor = sql.exec(
-    `SELECT key, value FROM _rivetkit_kv_storage WHERE key >= ? AND key < ? ORDER BY key ${direction} LIMIT ?`,
-    start,
-    end,
-    limit
-  );
-  return readEntries(cursor);
-}
-function toUint8Array(value) {
-  var _a;
-  if (value instanceof Uint8Array) {
-    return value;
-  }
-  if (value instanceof ArrayBuffer) {
-    return new Uint8Array(value);
-  }
-  throw new Error(
-    `Unexpected SQL value type: ${typeof value} (${(_a = value == null ? void 0 : value.constructor) == null ? void 0 : _a.name})`
-  );
-}
-function readEntries(cursor) {
-  const entries = [];
-  for (const row of cursor.raw()) {
-    entries.push([toUint8Array(row[0]), toUint8Array(row[1])]);
-  }
-  return entries;
-}
-function computePrefixUpperBound(prefix) {
-  const upperBound = prefix.slice();
-  for (let i = upperBound.length - 1; i >= 0; i--) {
-    if (upperBound[i] !== 255) {
-      upperBound[i]++;
-      return upperBound.slice(0, i + 1);
-    }
-  }
-  return null;
-}
-
-// src/global-kv.ts
-var GLOBAL_KV_KEYS = {
-  actorMetadata: (actorId) => {
-    return `actor:${actorId}:metadata`;
-  }
-};
-
-// src/handler.ts
-import { env } from "cloudflare:workers";
-import { createClientWithDriver } from "rivetkit";
-import { buildManagerRouter } from "rivetkit/driver-helpers";
 
-// src/config.ts
-import { z } from "zod/v4";
-var ConfigSchemaBase = z.object({
-  /** Path that the Rivet manager API will be mounted. */
-  managerPath: z.string().optional().default("/api/rivet"),
-  /** Deprecated. Runner key for authentication. */
-  runnerKey: z.string().optional(),
-  /** Disable the welcome message. */
-  noWelcome: z.boolean().optional().default(false),
-  fetch: z.custom().optional()
-});
-var ConfigSchema = ConfigSchemaBase.default(
-  () => ConfigSchemaBase.parse({})
-);
+var _rivetkit = require('rivetkit');
 
-// src/manager-driver.ts
-import {
-  WS_PROTOCOL_ACTOR,
-  WS_PROTOCOL_CONN_PARAMS,
-  WS_PROTOCOL_ENCODING,
-  WS_PROTOCOL_STANDARD,
-  WS_PROTOCOL_TARGET
-} from "rivetkit/driver-helpers";
-import {
-  ActorDuplicateKey,
-  ActorNotFound,
-  InternalError
-} from "rivetkit/errors";
-import { assertUnreachable } from "rivetkit/utils";
-
-// src/actor-id.ts
-function buildActorId(doId, generation) {
-  return `${doId}:${generation}`;
-}
-function parseActorId(actorId) {
-  const parts = actorId.split(":");
-  if (parts.length !== 2) {
-    throw new Error(`Invalid actor ID format: ${actorId}`);
-  }
-  const [doId, generationStr] = parts;
-  const generation = parseInt(generationStr, 10);
-  if (Number.isNaN(generation)) {
-    throw new Error(`Invalid generation number in actor ID: ${actorId}`);
-  }
-  return [doId, generation];
-}
-
-// src/util.ts
-var EMPTY_KEY = "(none)";
-var KEY_SEPARATOR = ",";
-function serializeNameAndKey(name, key) {
-  const escapedName = name.replace(/:/g, "\\:");
-  if (key.length === 0) {
-    return `${escapedName}:${EMPTY_KEY}`;
-  }
-  const serializedKey = serializeKey(key);
-  return `${escapedName}:${serializedKey}`;
-}
-function serializeKey(key) {
-  if (key.length === 0) {
-    return EMPTY_KEY;
-  }
-  const escapedParts = key.map((part) => {
-    if (part === EMPTY_KEY) {
-      return `\\${EMPTY_KEY}`;
-    }
-    let escaped = part.replace(/\\/g, "\\\\");
-    escaped = escaped.replace(/,/g, "\\,");
-    return escaped;
-  });
-  return escapedParts.join(KEY_SEPARATOR);
-}
-
-// src/manager-driver.ts
-var STANDARD_WEBSOCKET_HEADERS = [
-  "connection",
-  "upgrade",
-  "sec-websocket-key",
-  "sec-websocket-version",
-  "sec-websocket-protocol",
-  "sec-websocket-extensions"
-];
-var CloudflareActorsManagerDriver = class {
-  async sendRequest(actorId, actorRequest) {
-    const env3 = getCloudflareAmbientEnv();
-    const [doId] = parseActorId(actorId);
-    logger().debug({
-      msg: "sending request to durable object",
-      actorId,
-      doId,
-      method: actorRequest.method,
-      url: actorRequest.url
-    });
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    return await stub.fetch(actorRequest);
-  }
-  async openWebSocket(path, actorId, encoding, params) {
-    const env3 = getCloudflareAmbientEnv();
-    const [doId] = parseActorId(actorId);
-    logger().debug({
-      msg: "opening websocket to durable object",
-      actorId,
-      doId,
-      path
-    });
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    const protocols = [];
-    protocols.push(WS_PROTOCOL_STANDARD);
-    protocols.push(`${WS_PROTOCOL_TARGET}actor`);
-    protocols.push(`${WS_PROTOCOL_ACTOR}${encodeURIComponent(actorId)}`);
-    protocols.push(`${WS_PROTOCOL_ENCODING}${encoding}`);
-    if (params) {
-      protocols.push(
-        `${WS_PROTOCOL_CONN_PARAMS}${encodeURIComponent(JSON.stringify(params))}`
-      );
-    }
-    const headers = {
-      Upgrade: "websocket",
-      Connection: "Upgrade",
-      "sec-websocket-protocol": protocols.join(", ")
-    };
-    const normalizedPath = path.startsWith("/") ? path : `/${path}`;
-    const url = `http://actor${normalizedPath}`;
-    logger().debug({ msg: "rewriting websocket url", from: path, to: url });
-    const response = await stub.fetch(url, {
-      headers
-    });
-    const webSocket = response.webSocket;
-    if (!webSocket) {
-      throw new InternalError(
-        `missing websocket connection in response from DO
-
-Status: ${response.status}
-Response: ${await response.text()}`
-      );
-    }
-    logger().debug({
-      msg: "durable object websocket connection open",
-      actorId
-    });
-    webSocket.accept();
-    setTimeout(() => {
-      var _a;
-      const event = new Event("open");
-      (_a = webSocket.onopen) == null ? void 0 : _a.call(webSocket, event);
-      webSocket.dispatchEvent(event);
-    }, 0);
-    return webSocket;
-  }
-  async buildGatewayUrl(actorId) {
-    return `http://actor/gateway/${encodeURIComponent(actorId)}`;
-  }
-  async proxyRequest(c, actorRequest, actorId) {
-    const env3 = getCloudflareAmbientEnv();
-    const [doId] = parseActorId(actorId);
-    logger().debug({
-      msg: "forwarding request to durable object",
-      actorId,
-      doId,
-      method: actorRequest.method,
-      url: actorRequest.url
-    });
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    return await stub.fetch(actorRequest);
-  }
-  async proxyWebSocket(c, path, actorId, encoding, params) {
-    logger().debug({
-      msg: "forwarding websocket to durable object",
-      actorId,
-      path
-    });
-    const upgradeHeader = c.req.header("Upgrade");
-    if (!upgradeHeader || upgradeHeader !== "websocket") {
-      return new Response("Expected Upgrade: websocket", {
-        status: 426
-      });
-    }
-    const newUrl = new URL(`http://actor${path}`);
-    const actorRequest = new Request(newUrl, c.req.raw);
-    logger().debug({
-      msg: "rewriting websocket url",
-      from: c.req.url,
-      to: actorRequest.url
-    });
-    const headerKeys = [];
-    actorRequest.headers.forEach((v, k) => {
-      headerKeys.push(k);
-    });
-    for (const k of headerKeys) {
-      if (!STANDARD_WEBSOCKET_HEADERS.includes(k)) {
-        actorRequest.headers.delete(k);
-      }
-    }
-    const protocols = [];
-    protocols.push(WS_PROTOCOL_STANDARD);
-    protocols.push(`${WS_PROTOCOL_TARGET}actor`);
-    protocols.push(`${WS_PROTOCOL_ACTOR}${encodeURIComponent(actorId)}`);
-    protocols.push(`${WS_PROTOCOL_ENCODING}${encoding}`);
-    if (params) {
-      protocols.push(
-        `${WS_PROTOCOL_CONN_PARAMS}${encodeURIComponent(JSON.stringify(params))}`
-      );
-    }
-    actorRequest.headers.set(
-      "sec-websocket-protocol",
-      protocols.join(", ")
-    );
-    const env3 = getCloudflareAmbientEnv();
-    const [doId] = parseActorId(actorId);
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    return await stub.fetch(actorRequest);
-  }
-  async getForId({
-    c,
-    name,
-    actorId
-  }) {
-    const env3 = getCloudflareAmbientEnv();
-    const [doId, expectedGeneration] = parseActorId(actorId);
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    const result = await stub.getMetadata();
-    if (!result) {
-      logger().debug({
-        msg: "getForId: actor not found",
-        actorId
-      });
-      return void 0;
-    }
-    if (result.actorId !== actorId) {
-      logger().debug({
-        msg: "getForId: generation mismatch",
-        requestedActorId: actorId,
-        actualActorId: result.actorId
-      });
-      return void 0;
-    }
-    if (result.destroying) {
-      throw new ActorNotFound(actorId);
-    }
-    return {
-      actorId: result.actorId,
-      name: result.name,
-      key: result.key
-    };
-  }
-  async getWithKey({
-    c,
-    name,
-    key
-  }) {
-    const env3 = getCloudflareAmbientEnv();
-    logger().debug({ msg: "getWithKey: searching for actor", name, key });
-    const nameKeyString = serializeNameAndKey(name, key);
-    const doId = env3.ACTOR_DO.idFromName(nameKeyString).toString();
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    const result = await stub.getMetadata();
-    if (result) {
-      logger().debug({
-        msg: "getWithKey: found actor with matching name and key",
-        actorId: result.actorId,
-        name: result.name,
-        key: result.key
-      });
-      return {
-        actorId: result.actorId,
-        name: result.name,
-        key: result.key
-      };
-    } else {
-      logger().debug({
-        msg: "getWithKey: no actor found with matching name and key",
-        name,
-        key,
-        doId
-      });
-      return void 0;
-    }
-  }
-  async getOrCreateWithKey({
-    c,
-    name,
-    key,
-    input
-  }) {
-    const env3 = getCloudflareAmbientEnv();
-    const nameKeyString = serializeNameAndKey(name, key);
-    const doId = env3.ACTOR_DO.idFromName(nameKeyString);
-    const actor = env3.ACTOR_DO.get(doId);
-    const result = await actor.create({
-      name,
-      key,
-      input,
-      allowExisting: true
-    });
-    if ("success" in result) {
-      const { actorId, created } = result.success;
-      logger().debug({
-        msg: "getOrCreateWithKey result",
-        actorId,
-        name,
-        key,
-        created
-      });
-      return {
-        actorId,
-        name,
-        key
-      };
-    } else if ("error" in result) {
-      throw new Error(`Error: ${JSON.stringify(result.error)}`);
-    } else {
-      assertUnreachable(result);
-    }
-  }
-  async createActor({
-    c,
-    name,
-    key,
-    input
-  }) {
-    const env3 = getCloudflareAmbientEnv();
-    const nameKeyString = serializeNameAndKey(name, key);
-    const doId = env3.ACTOR_DO.idFromName(nameKeyString);
-    const actor = env3.ACTOR_DO.get(doId);
-    const result = await actor.create({
-      name,
-      key,
-      input,
-      allowExisting: false
-    });
-    if ("success" in result) {
-      const { actorId } = result.success;
-      return {
-        actorId,
-        name,
-        key
-      };
-    } else if ("error" in result) {
-      if (result.error.actorAlreadyExists) {
-        throw new ActorDuplicateKey(name, key);
+// src/websocket.ts
+var FetchWebSocket = (_class = class _FetchWebSocket {
+  static __initStatic() {this.CONNECTING = 0}
+  static __initStatic2() {this.OPEN = 1}
+  static __initStatic3() {this.CLOSING = 2}
+  static __initStatic4() {this.CLOSED = 3}
+  __init() {this.binaryType = "arraybuffer"}
+  __init2() {this.onopen = null}
+  __init3() {this.onmessage = null}
+  __init4() {this.onclose = null}
+  __init5() {this.onerror = null}
+  __init6() {this.readyState = _FetchWebSocket.CONNECTING}
+  #socket;
+  #pending = [];
+  constructor(url, protocols) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);_class.prototype.__init4.call(this);_class.prototype.__init5.call(this);_class.prototype.__init6.call(this);
+    void this.#connect(url, protocols);
+  }
+  async #connect(url, protocols) {
+    var _a, _b, _c;
+    try {
+      const protocolList = Array.isArray(protocols) ? protocols : protocols ? [protocols] : [];
+      const headers = new Headers({ Upgrade: "websocket" });
+      if (protocolList.length > 0) {
+        headers.set("Sec-WebSocket-Protocol", protocolList.join(", "));
       }
-      throw new InternalError(
-        `Unknown error creating actor: ${JSON.stringify(result.error)}`
+      const response = await fetch(
+        url.replace(/^ws:/, "http:").replace(/^wss:/, "https:"),
+        { headers }
       );
-    } else {
-      assertUnreachable(result);
-    }
-  }
-  async listActors({ c, name }) {
-    logger().warn({
-      msg: "listActors not fully implemented for Cloudflare Workers",
-      name
-    });
-    return [];
-  }
-  displayInformation() {
-    return {
-      properties: {
-        Driver: "Cloudflare Workers"
+      const socket = response.webSocket;
+      if (!socket) {
+        throw new Error(
+          `websocket upgrade failed with status ${response.status}`
+        );
       }
-    };
-  }
-  setGetUpgradeWebSocket() {
-  }
-  async kvGet(actorId, key) {
-    const env3 = getCloudflareAmbientEnv();
-    const [doId] = parseActorId(actorId);
-    const id = env3.ACTOR_DO.idFromString(doId);
-    const stub = env3.ACTOR_DO.get(id);
-    const value = await stub.managerKvGet(key);
-    return value !== null ? new TextDecoder().decode(value) : null;
-  }
-};
-
-// src/websocket.ts
-import { defineWebSocketHelper, WSContext } from "hono/ws";
-import { WS_PROTOCOL_STANDARD as WS_PROTOCOL_STANDARD2 } from "rivetkit/driver-helpers";
-var upgradeWebSocket = defineWebSocketHelper(async (c, events) => {
-  var _a, _b;
-  const upgradeHeader = c.req.header("Upgrade");
-  if (upgradeHeader !== "websocket") {
-    return;
-  }
-  const webSocketPair = new WebSocketPair();
-  const client = webSocketPair[0];
-  const server = webSocketPair[1];
-  const wsContext = new WSContext({
-    close: (code, reason) => server.close(code, reason),
-    get protocol() {
-      return server.protocol;
-    },
-    raw: server,
-    get readyState() {
-      return server.readyState;
-    },
-    url: server.url ? new URL(server.url) : null,
-    send: (source) => server.send(source)
-  });
-  if (events.onClose) {
-    server.addEventListener(
-      "close",
-      (evt) => {
+      socket.accept();
+      socket.binaryType = this.binaryType;
+      this.#socket = socket;
+      this.readyState = _FetchWebSocket.OPEN;
+      socket.addEventListener("message", (event) => {
         var _a2;
-        return (_a2 = events.onClose) == null ? void 0 : _a2.call(events, evt, wsContext);
-      }
-    );
-  }
-  if (events.onMessage) {
-    server.addEventListener(
-      "message",
-      (evt) => {
+        (_a2 = this.onmessage) == null ? void 0 : _a2.call(this, event);
+      });
+      socket.addEventListener("close", (event) => {
         var _a2;
-        return (_a2 = events.onMessage) == null ? void 0 : _a2.call(events, evt, wsContext);
-      }
-    );
-  }
-  if (events.onError) {
-    server.addEventListener(
-      "error",
-      (evt) => {
+        this.readyState = _FetchWebSocket.CLOSED;
+        (_a2 = this.onclose) == null ? void 0 : _a2.call(this, event);
+      });
+      socket.addEventListener("error", (event) => {
         var _a2;
-        return (_a2 = events.onError) == null ? void 0 : _a2.call(events, evt, wsContext);
-      }
-    );
-  }
-  (_a = server.accept) == null ? void 0 : _a.call(server);
-  (_b = events.onOpen) == null ? void 0 : _b.call(events, new Event("open"), wsContext);
-  const headers = {};
-  const protocols = c.req.header("Sec-WebSocket-Protocol");
-  if (typeof protocols === "string" && protocols.split(",").map((x) => x.trim()).includes(WS_PROTOCOL_STANDARD2)) {
-    headers["Sec-WebSocket-Protocol"] = WS_PROTOCOL_STANDARD2;
-  }
-  return new Response(null, {
-    status: 101,
-    headers,
-    webSocket: client
-  });
-});
-
-// src/handler.ts
-function getCloudflareAmbientEnv() {
-  return env;
-}
-function createInlineClient(registry, inputConfig) {
-  inputConfig = { ...inputConfig, runnerKey: "" };
-  const config = ConfigSchema.parse(inputConfig);
-  const ActorHandler = createActorDurableObject(
-    registry,
-    () => upgradeWebSocket
-  );
-  registry.config.noWelcome = true;
-  registry.config.inspector = {
-    enabled: false,
-    token: () => ""
-  };
-  registry.config.managerBasePath = "/";
-  const parsedConfig = registry.parseConfig();
-  const managerDriver = new CloudflareActorsManagerDriver();
-  const { router } = buildManagerRouter(
-    parsedConfig,
-    managerDriver,
-    () => upgradeWebSocket
-  );
-  const client = createClientWithDriver(managerDriver);
-  return { client, fetch: router.fetch.bind(router), config, ActorHandler };
-}
-function createHandler(registry, inputConfig) {
-  const inline = createInlineClient(registry, inputConfig);
-  const client = inline.client;
-  const fetch = inline.fetch;
-  const config = inline.config;
-  const ActorHandler = inline.ActorHandler;
-  const handler = {
-    fetch: async (request, cfEnv, ctx) => {
-      const url = new URL(request.url);
-      const env3 = Object.assign({ RIVET: client }, cfEnv);
-      if (url.pathname.startsWith(config.managerPath)) {
-        const strippedPath = url.pathname.substring(
-          config.managerPath.length
-        );
-        url.pathname = strippedPath;
-        const modifiedRequest = new Request(url.toString(), request);
-        return fetch(modifiedRequest, env3, ctx);
-      }
-      if (config.fetch) {
-        return config.fetch(request, env3, ctx);
-      } else {
-        return new Response(
-          "This is a RivetKit server.\n\nLearn more at https://rivet.dev\n",
-          { status: 200 }
-        );
+        (_a2 = this.onerror) == null ? void 0 : _a2.call(this, event);
+      });
+      (_a = this.onopen) == null ? void 0 : _a.call(this, new Event("open"));
+      for (const data of this.#pending.splice(0)) {
+        socket.send(data);
       }
+    } catch (error) {
+      console.error("rivetkit cloudflare websocket shim failed", error);
+      this.readyState = _FetchWebSocket.CLOSED;
+      (_b = this.onerror) == null ? void 0 : _b.call(this, error instanceof Event ? error : new Event("error"));
+      (_c = this.onclose) == null ? void 0 : _c.call(this, new CloseEvent("close", { code: 1006 }));
     }
-  };
-  return { handler, ActorHandler };
-}
-
-// src/actor-driver.ts
-var CloudflareDurableObjectGlobalState = class {
-  // Map of actor ID -> DO state
-  #dos = /* @__PURE__ */ new Map();
-  // WeakMap of DO state -> ActorGlobalState for proper GC
-  #actors = /* @__PURE__ */ new WeakMap();
-  getDOState(doId) {
-    const state = this.#dos.get(doId);
-    invariant(
-      state !== void 0,
-      "durable object state not in global state"
-    );
-    return state;
-  }
-  setDOState(doId, state) {
-    this.#dos.set(doId, state);
-  }
-  getActorState(ctx) {
-    return this.#actors.get(ctx);
-  }
-  setActorState(ctx, actorState) {
-    this.#actors.set(ctx, actorState);
   }
-};
-var ActorGlobalState = class {
-  // Initialization state
-  initialized;
-  // Loaded actor state
-  actor;
-  actorInstance;
-  actorPromise;
-  /**
-   * Indicates if `startDestroy` has been called.
-   *
-   * This is stored in memory instead of SQLite since the destroy may be cancelled.
-   *
-   * See the corresponding `destroyed` property in SQLite metadata.
-   */
-  destroying = false;
-  reset() {
-    this.initialized = void 0;
-    this.actor = void 0;
-    this.actorInstance = void 0;
-    this.actorPromise = void 0;
-    this.destroying = false;
-  }
-};
-var CloudflareActorsActorDriver = class {
-  #registryConfig;
-  #managerDriver;
-  #inlineClient;
-  #globalState;
-  constructor(registryConfig, managerDriver, inlineClient, globalState) {
-    this.#registryConfig = registryConfig;
-    this.#managerDriver = managerDriver;
-    this.#inlineClient = inlineClient;
-    this.#globalState = globalState;
-  }
-  #getDOCtx(actorId) {
-    const [doId] = parseActorId(actorId);
-    return this.#globalState.getDOState(doId).ctx;
-  }
-  async loadActor(actorId) {
+  send(data) {
     var _a;
-    const [doId, expectedGeneration] = parseActorId(actorId);
-    const doState = this.#globalState.getDOState(doId);
-    let actorState = this.#globalState.getActorState(doState.ctx);
-    if (actorState == null ? void 0 : actorState.actorInstance) {
-      return actorState.actorInstance;
-    }
-    if (!actorState) {
-      actorState = new ActorGlobalState();
-      actorState.actorPromise = promiseWithResolvers(
-        (reason) => logger().warn({
-          msg: "unhandled actor promise rejection",
-          reason
-        })
-      );
-      this.#globalState.setActorState(doState.ctx, actorState);
-    } else if (actorState.actorPromise) {
-      await actorState.actorPromise.promise;
-      if (!actorState.actorInstance) {
-        throw new Error(
-          `Actor ${actorId} failed to load in concurrent request`
-        );
-      }
-      return actorState.actorInstance;
-    }
-    const sql = doState.ctx.storage.sql;
-    const cursor = sql.exec(
-      "SELECT name, key, destroyed, generation FROM _rivetkit_metadata LIMIT 1"
-    );
-    const result = cursor.raw().next();
-    if (result.done || !result.value) {
-      throw new Error(
-        `Actor ${actorId} is not initialized - missing metadata`
-      );
-    }
-    const name = result.value[0];
-    const key = JSON.parse(result.value[1]);
-    const destroyed = result.value[2];
-    const generation = result.value[3];
-    if (destroyed) {
-      throw new Error(`Actor ${actorId} is destroyed`);
-    }
-    if (generation !== expectedGeneration) {
-      throw new Error(
-        `Actor ${actorId} generation mismatch - expected ${expectedGeneration}, got ${generation}`
-      );
-    }
-    const definition = lookupInRegistry(this.#registryConfig, name);
-    actorState.actorInstance = definition.instantiate();
-    const actorInstance = actorState.actorInstance;
-    await actorInstance.start(
-      this,
-      this.#inlineClient,
-      actorId,
-      name,
-      key,
-      "unknown"
-      // TODO: Support regions in Cloudflare
-    );
-    (_a = actorState.actorPromise) == null ? void 0 : _a.resolve();
-    actorState.actorPromise = void 0;
-    return actorState.actorInstance;
-  }
-  getContext(actorId) {
-    const [doId] = parseActorId(actorId);
-    const state = this.#globalState.getDOState(doId);
-    return { state: state.ctx };
-  }
-  async setAlarm(actor, timestamp) {
-    await this.#getDOCtx(actor.id).storage.setAlarm(timestamp);
-  }
-  async getDatabase(actorId) {
-    return this.#getDOCtx(actorId).storage.sql;
-  }
-  // Batch KV operations
-  async kvBatchPut(actorId, entries) {
-    const sql = this.#getDOCtx(actorId).storage.sql;
-    for (const [key, value] of entries) {
-      kvPut(sql, key, value);
-    }
-  }
-  async kvBatchGet(actorId, keys) {
-    const sql = this.#getDOCtx(actorId).storage.sql;
-    const results = [];
-    for (const key of keys) {
-      results.push(kvGet(sql, key));
-    }
-    return results;
-  }
-  async kvBatchDelete(actorId, keys) {
-    const sql = this.#getDOCtx(actorId).storage.sql;
-    for (const key of keys) {
-      kvDelete(sql, key);
-    }
-  }
-  async kvDeleteRange(actorId, start, end) {
-    const sql = this.#getDOCtx(actorId).storage.sql;
-    kvDeleteRange(sql, start, end);
-  }
-  async kvListPrefix(actorId, prefix, options) {
-    const sql = this.#getDOCtx(actorId).storage.sql;
-    return kvListPrefix(sql, prefix, options);
-  }
-  async kvListRange(actorId, start, end, options) {
-    const sql = this.#getDOCtx(actorId).storage.sql;
-    return kvListRange(sql, start, end, options);
-  }
-  startDestroy(actorId) {
-    const [doId, generation] = parseActorId(actorId);
-    const doState = this.#globalState.getDOState(doId);
-    const actorState = this.#globalState.getActorState(doState.ctx);
-    if (!(actorState == null ? void 0 : actorState.actorInstance)) {
+    if (this.readyState === _FetchWebSocket.CONNECTING) {
+      this.#pending.push(data);
       return;
     }
-    if (actorState.destroying) {
-      return;
-    }
-    actorState.destroying = true;
-    this.#callOnStopAsync(actorId, doId, actorState.actorInstance);
+    (_a = this.#socket) == null ? void 0 : _a.send(data);
   }
-  async #callOnStopAsync(actorId, doId, actor) {
-    await actor.onStop("destroy");
-    const doState = this.#globalState.getDOState(doId);
-    const sql = doState.ctx.storage.sql;
-    sql.exec("UPDATE _rivetkit_metadata SET destroyed = 1 WHERE 1=1");
-    sql.exec("DELETE FROM _rivetkit_kv_storage");
-    await doState.ctx.storage.deleteAlarm();
-    const env3 = getCloudflareAmbientEnv();
-    doState.ctx.waitUntil(
-      env3.ACTOR_KV.delete(GLOBAL_KV_KEYS.actorMetadata(actorId))
-    );
-    const actorHandle = this.#globalState.getActorState(doState.ctx);
-    actorHandle == null ? void 0 : actorHandle.reset();
+  close(code, reason) {
+    var _a;
+    this.readyState = _FetchWebSocket.CLOSING;
+    (_a = this.#socket) == null ? void 0 : _a.close(code, reason);
   }
-};
-function createCloudflareActorsActorDriverBuilder(globalState) {
-  return (config, managerDriver, inlineClient) => {
-    return new CloudflareActorsActorDriver(
-      config,
-      managerDriver,
-      inlineClient,
-      globalState
-    );
-  };
+}, _class.__initStatic(), _class.__initStatic2(), _class.__initStatic3(), _class.__initStatic4(), _class);
+var globalScope = globalThis;
+if (!globalScope.__RIVETKIT_CF_WEBSOCKET_INSTALLED__) {
+  globalScope.WebSocket = FetchWebSocket;
+  globalScope.__RIVETKIT_CF_WEBSOCKET_INSTALLED__ = true;
 }
 
-// src/actor-handler-do.ts
-function createActorDurableObject(registry, getUpgradeWebSocket) {
-  const globalState = new CloudflareDurableObjectGlobalState();
-  const parsedConfig = registry.parseConfig();
-  return class ActorHandler extends DurableObject {
-    /**
-     * This holds a strong reference to ActorGlobalState.
-     * CloudflareDurableObjectGlobalState holds a weak reference so we can
-     * access it elsewhere.
-     **/
-    #state;
-    constructor(...args) {
-      super(...args);
-      this.ctx.storage.sql.exec(`
-				CREATE TABLE IF NOT EXISTS _rivetkit_kv_storage(
-					key BLOB PRIMARY KEY,
-					value BLOB
-				);
-			`);
-      this.ctx.storage.sql.exec(`
-				CREATE TABLE IF NOT EXISTS _rivetkit_metadata(
-					id INTEGER PRIMARY KEY CHECK (id = 1),
-					name TEXT NOT NULL,
-					key TEXT NOT NULL,
-					destroyed INTEGER DEFAULT 0,
-					generation INTEGER DEFAULT 0
-				);
-			`);
-      const state = globalState.getActorState(this.ctx);
-      if (state) {
-        this.#state = state;
-      } else {
-        this.#state = new ActorGlobalState();
-        globalState.setActorState(this.ctx, this.#state);
-      }
-    }
-    async #loadActor() {
-      var _a, _b;
-      invariant2(this.#state, "State should be initialized");
-      if (!this.#state.initialized) {
-        const cursor = this.ctx.storage.sql.exec(
-          "SELECT name, key, destroyed, generation FROM _rivetkit_metadata WHERE id = 1"
-        );
-        const result = cursor.raw().next();
-        if (!result.done && result.value) {
-          const name = result.value[0];
-          const key = JSON.parse(
-            result.value[1]
-          );
-          const destroyed = result.value[2];
-          const generation = result.value[3];
-          if (!destroyed) {
-            logger().debug({
-              msg: "already initialized",
-              name,
-              key,
-              generation
-            });
-            this.#state.initialized = { name, key, generation };
-          } else {
-            logger().debug("actor is destroyed, cannot load");
-            throw new Error("Actor is destroyed");
-          }
-        } else {
-          logger().debug("not initialized");
-          throw new Error("Actor is not initialized");
-        }
-      }
-      if (this.#state.actor) {
-        invariant2(
-          !this.#state.initialized || this.#state.actor.generation === this.#state.initialized.generation,
-          `Stale actor cached: actor generation ${this.#state.actor.generation} != initialized generation ${(_a = this.#state.initialized) == null ? void 0 : _a.generation}. This should not happen.`
+// src/mod.ts
+var DEFAULT_MANAGER_PATH = "/api/rivet";
+function setup(config) {
+  return _rivetkit.setup.call(void 0, {
+    runtime: "wasm",
+    wasm: { bindings: wasmBindings, initInput: _rivetkit_wasm_bgwasm2.default },
+    noWelcome: true,
+    ...config
+  });
+}
+function applyEnv(registry, env, managerPath) {
+  var _a, _b;
+  const config = registry.config;
+  if (!config.endpoint && env.RIVET_ENDPOINT) {
+    config.endpoint = env.RIVET_ENDPOINT;
+  }
+  if (!config.namespace && env.RIVET_NAMESPACE) {
+    config.namespace = env.RIVET_NAMESPACE;
+  }
+  if (!config.token && env.RIVET_TOKEN) {
+    config.token = env.RIVET_TOKEN;
+  }
+  if (env.RIVET_POOL && !((_a = config.envoy) == null ? void 0 : _a.poolName)) {
+    config.envoy = { ...config.envoy, poolName: env.RIVET_POOL };
+  }
+  if (!((_b = config.serverless) == null ? void 0 : _b.basePath)) {
+    config.serverless = { ...config.serverless, basePath: managerPath };
+  }
+}
+function createHandler(registryOrConfig, options = {}) {
+  const managerPath = _nullishCoalesce(options.managerPath, () => ( DEFAULT_MANAGER_PATH));
+  const registry = registryOrConfig instanceof _rivetkit.Registry ? registryOrConfig : setup(registryOrConfig);
+  let envApplied = false;
+  return {
+    async fetch(request, env, ctx) {
+      if (!envApplied) {
+        applyEnv(
+          registry,
+          _nullishCoalesce(env, () => ( {})),
+          managerPath
         );
-        return this.#state.actor;
+        envApplied = true;
       }
-      if (!this.#state.initialized) throw new Error("Not initialized");
-      const actorId = this.ctx.id.toString();
-      globalState.setDOState(actorId, { ctx: this.ctx, env: env2 });
-      const managerDriver = new CloudflareActorsManagerDriver();
-      const inlineClient = createClientWithDriver2(
-        managerDriver
-      );
-      const actorDriverBuilder = createCloudflareActorsActorDriverBuilder(globalState);
-      const actorDriver = actorDriverBuilder(
-        parsedConfig,
-        managerDriver,
-        inlineClient
-      );
-      const actorRouter = createActorRouter(
-        parsedConfig,
-        actorDriver,
-        getUpgradeWebSocket,
-        ((_b = registry.config.test) == null ? void 0 : _b.enabled) ?? false
-      );
-      this.#state.actor = {
-        actorRouter,
-        actorDriver,
-        generation: this.#state.initialized.generation
-      };
-      const actorIdWithGen = buildActorId(
-        actorId,
-        this.#state.initialized.generation
-      );
-      await actorDriver.loadActor(actorIdWithGen);
-      return this.#state.actor;
-    }
-    /** RPC called to get actor metadata without creating it */
-    async getMetadata() {
-      var _a;
-      const cursor = this.ctx.storage.sql.exec(
-        "SELECT name, key, destroyed, generation FROM _rivetkit_metadata WHERE id = 1"
-      );
-      const result = cursor.raw().next();
-      if (!result.done && result.value) {
-        const name = result.value[0];
-        const key = JSON.parse(result.value[1]);
-        const destroyed = result.value[2];
-        const generation = result.value[3];
-        if (destroyed) {
-          logger().debug({
-            msg: "getMetadata: actor is destroyed",
-            name,
-            key,
-            generation
-          });
-          return void 0;
-        }
-        const doId = this.ctx.id.toString();
-        const actorId = buildActorId(doId, generation);
-        const destroying = ((_a = globalState.getActorState(this.ctx)) == null ? void 0 : _a.destroying) ?? false;
-        logger().debug({
-          msg: "getMetadata: found actor metadata",
-          actorId,
-          name,
-          key,
-          generation,
-          destroying
-        });
-        return { actorId, name, key, destroying };
+      const url = new URL(request.url);
+      if (url.pathname === managerPath || url.pathname.startsWith(`${managerPath}/`)) {
+        return registry.handler(request);
       }
-      logger().debug({
-        msg: "getMetadata: no metadata found"
-      });
-      return void 0;
-    }
-    /** RPC called by ManagerDriver.kvGet to read from KV. */
-    async managerKvGet(key) {
-      return kvGet(this.ctx.storage.sql, key);
-    }
-    /** RPC called by the manager to create a DO. Can optionally allow existing actors. */
-    async create(req) {
-      const checkCursor = this.ctx.storage.sql.exec(
-        "SELECT destroyed, generation FROM _rivetkit_metadata WHERE id = 1"
-      );
-      const checkResult = checkCursor.raw().next();
-      let created = false;
-      let generation = 0;
-      if (!checkResult.done && checkResult.value) {
-        const destroyed = checkResult.value[0];
-        generation = checkResult.value[1];
-        if (!destroyed) {
-          if (!req.allowExisting) {
-            logger().debug({
-              msg: "create failed: actor already exists",
-              name: req.name,
-              key: req.key,
-              generation
-            });
-            return { error: { actorAlreadyExists: true } };
-          }
-          logger().debug({
-            msg: "actor already exists",
-            key: req.key,
-            generation
-          });
-          const doId2 = this.ctx.id.toString();
-          const actorId2 = buildActorId(doId2, generation);
-          return { success: { actorId: actorId2, created: false } };
-        }
-        generation = generation + 1;
-        created = true;
-        if (this.#state) {
-          this.#state.actor = void 0;
-        }
-        logger().debug({
-          msg: "resurrecting destroyed actor",
-          key: req.key,
-          oldGeneration: generation - 1,
-          newGeneration: generation
-        });
-      } else {
-        generation = 0;
-        created = true;
-        logger().debug({
-          msg: "creating new actor",
-          key: req.key,
-          generation
-        });
+      if (options.fetch) {
+        return options.fetch(request, env, ctx);
       }
-      this.ctx.storage.sql.exec(
-        `INSERT INTO _rivetkit_metadata (id, name, key, destroyed, generation)
-				VALUES (1, ?, ?, 0, ?)
-				ON CONFLICT(id) DO UPDATE SET
-					name = excluded.name,
-					key = excluded.key,
-					destroyed = 0,
-					generation = excluded.generation`,
-        req.name,
-        JSON.stringify(req.key),
-        generation
+      return new Response(
+        "This is a RivetKit server.\n\nLearn more at https://rivet.dev\n"
       );
-      this.#state.initialized = {
-        name: req.name,
-        key: req.key,
-        generation
-      };
-      const doId = this.ctx.id.toString();
-      const actorId = buildActorId(doId, generation);
-      if (created) {
-        initializeActorKvStorage(this.ctx.storage.sql, req.input);
-        const env3 = getCloudflareAmbientEnv();
-        const actorData = { name: req.name, key: req.key, generation };
-        this.ctx.waitUntil(
-          env3.ACTOR_KV.put(
-            GLOBAL_KV_KEYS.actorMetadata(actorId),
-            JSON.stringify(actorData)
-          )
-        );
-      }
-      await this.#loadActor();
-      logger().debug({
-        msg: created ? "actor created/resurrected" : "returning existing actor",
-        actorId,
-        created,
-        generation
-      });
-      return { success: { actorId, created } };
-    }
-    async fetch(request) {
-      const { actorRouter, generation } = await this.#loadActor();
-      const doId = this.ctx.id.toString();
-      const actorId = buildActorId(doId, generation);
-      return await actorRouter.fetch(request, {
-        actorId
-      });
-    }
-    async alarm() {
-      const { actorDriver, generation } = await this.#loadActor();
-      const doId = this.ctx.id.toString();
-      const actorId = buildActorId(doId, generation);
-      const actor = await actorDriver.loadActor(actorId);
-      await actor.onAlarm();
     }
   };
 }
-function initializeActorKvStorage(sql, input) {
-  const initialKvState = getInitialActorKvState(input);
-  for (const [key, value] of initialKvState) {
-    kvPut(sql, key, value);
-  }
-}
-export {
-  createActorDurableObject,
-  createHandler,
-  createInlineClient
-};
+
+
+
+exports.createHandler = createHandler; exports.setup = setup;
 //# sourceMappingURL=mod.js.map