@rivetkit/cloudflare-workers 0.9.1 → 0.9.2

package/dist/mod.d.cts

@@ -2,29 +2,23 @@ import * as hono_types from 'hono/types';
 import * as hono from 'hono';
 import { Hono } from 'hono';
 import * as _rivetkit_core_utils from '@rivetkit/core/utils';
+import * as _rivetkit_core_driver_helpers from '@rivetkit/core/driver-helpers';
 import * as _rivetkit_core from '@rivetkit/core';
 import { ActorKey, Registry } from '@rivetkit/core';
-import * as _rivetkit_core_driver_helpers from '@rivetkit/core/driver-helpers';
 import { z } from 'zod';
 import { Client } from '@rivetkit/core/client';
 import { DurableObject } from 'cloudflare:workers';
 
 declare const ConfigSchema: z.ZodDefault<z.ZodObject<Omit<{
     driver: z.ZodDefault<z.ZodOptional<z.ZodObject<{
-        topology: z.ZodEnum<["standalone", "partition", "coordinate"]>;
-        manager: z.ZodType<_rivetkit_core_driver_helpers.ManagerDriver, z.ZodTypeDef, _rivetkit_core_driver_helpers.ManagerDriver>;
-        actor: z.ZodType<_rivetkit_core_driver_helpers.ActorDriver, z.ZodTypeDef, _rivetkit_core_driver_helpers.ActorDriver>;
-        coordinate: z.ZodOptional<z.ZodType<_rivetkit_core.CoordinateDriver, z.ZodTypeDef, _rivetkit_core.CoordinateDriver>>;
+        manager: z.ZodType<(registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver, z.ZodTypeDef, (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver>;
+        actor: z.ZodType<(registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver, z.ZodTypeDef, (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver>;
     }, "strip", z.ZodTypeAny, {
-        actor: _rivetkit_core_driver_helpers.ActorDriver;
-        topology: "standalone" | "partition" | "coordinate";
-        manager: _rivetkit_core_driver_helpers.ManagerDriver;
-        coordinate?: _rivetkit_core.CoordinateDriver | undefined;
+        actor: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver;
+        manager: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver;
     }, {
-        actor: _rivetkit_core_driver_helpers.ActorDriver;
-        topology: "standalone" | "partition" | "coordinate";
-        manager: _rivetkit_core_driver_helpers.ManagerDriver;
-        coordinate?: _rivetkit_core.CoordinateDriver | undefined;
+        actor: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver;
+        manager: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver;
     }>>>;
     getUpgradeWebSocket: z.ZodOptional<z.ZodType<() => _rivetkit_core_utils.UpgradeWebSocket, z.ZodTypeDef, () => _rivetkit_core_utils.UpgradeWebSocket>>;
     cors: z.ZodOptional<z.ZodType<{
@@ -43,36 +37,10 @@ declare const ConfigSchema: z.ZodDefault<z.ZodObject<Omit<{
         exposeHeaders?: string[];
     }>>;
     maxIncomingMessageSize: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-    actorPeer: z.ZodDefault<z.ZodOptional<z.ZodObject<{
-        leaseDuration: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        renewLeaseGrace: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        checkLeaseInterval: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        checkLeaseJitter: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        messageAckTimeout: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-    }, "strip", z.ZodTypeAny, {
-        leaseDuration: number;
-        renewLeaseGrace: number;
-        checkLeaseInterval: number;
-        checkLeaseJitter: number;
-        messageAckTimeout: number;
-    }, {
-        leaseDuration?: number | undefined;
-        renewLeaseGrace?: number | undefined;
-        checkLeaseInterval?: number | undefined;
-        checkLeaseJitter?: number | undefined;
-        messageAckTimeout?: number | undefined;
-    }>>>;
 }, "driver" | "getUpgradeWebSocket"> & {
     app: z.ZodOptional<z.ZodType<Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/">, z.ZodTypeDef, Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/">>>;
 }, "strip", z.ZodTypeAny, {
     maxIncomingMessageSize: number;
-    actorPeer: {
-        leaseDuration: number;
-        renewLeaseGrace: number;
-        checkLeaseInterval: number;
-        checkLeaseJitter: number;
-        messageAckTimeout: number;
-    };
     cors?: {
         origin: string | string[] | ((origin: string, c: hono.Context) => string | undefined | null);
         allowMethods?: string[] | ((origin: string, c: hono.Context) => string[]);
@@ -92,13 +60,6 @@ declare const ConfigSchema: z.ZodDefault<z.ZodObject<Omit<{
         exposeHeaders?: string[];
     } | undefined;
     maxIncomingMessageSize?: number | undefined;
-    actorPeer?: {
-        leaseDuration?: number | undefined;
-        renewLeaseGrace?: number | undefined;
-        checkLeaseInterval?: number | undefined;
-        checkLeaseJitter?: number | undefined;
-        messageAckTimeout?: number | undefined;
-    } | undefined;
     app?: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/"> | undefined;
 }>>;
 type InputConfig = z.input<typeof ConfigSchema>;

package/dist/mod.d.ts

@@ -2,29 +2,23 @@ import * as hono_types from 'hono/types';
 import * as hono from 'hono';
 import { Hono } from 'hono';
 import * as _rivetkit_core_utils from '@rivetkit/core/utils';
+import * as _rivetkit_core_driver_helpers from '@rivetkit/core/driver-helpers';
 import * as _rivetkit_core from '@rivetkit/core';
 import { ActorKey, Registry } from '@rivetkit/core';
-import * as _rivetkit_core_driver_helpers from '@rivetkit/core/driver-helpers';
 import { z } from 'zod';
 import { Client } from '@rivetkit/core/client';
 import { DurableObject } from 'cloudflare:workers';
 
 declare const ConfigSchema: z.ZodDefault<z.ZodObject<Omit<{
     driver: z.ZodDefault<z.ZodOptional<z.ZodObject<{
-        topology: z.ZodEnum<["standalone", "partition", "coordinate"]>;
-        manager: z.ZodType<_rivetkit_core_driver_helpers.ManagerDriver, z.ZodTypeDef, _rivetkit_core_driver_helpers.ManagerDriver>;
-        actor: z.ZodType<_rivetkit_core_driver_helpers.ActorDriver, z.ZodTypeDef, _rivetkit_core_driver_helpers.ActorDriver>;
-        coordinate: z.ZodOptional<z.ZodType<_rivetkit_core.CoordinateDriver, z.ZodTypeDef, _rivetkit_core.CoordinateDriver>>;
+        manager: z.ZodType<(registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver, z.ZodTypeDef, (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver>;
+        actor: z.ZodType<(registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver, z.ZodTypeDef, (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver>;
     }, "strip", z.ZodTypeAny, {
-        actor: _rivetkit_core_driver_helpers.ActorDriver;
-        topology: "standalone" | "partition" | "coordinate";
-        manager: _rivetkit_core_driver_helpers.ManagerDriver;
-        coordinate?: _rivetkit_core.CoordinateDriver | undefined;
+        actor: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver;
+        manager: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver;
     }, {
-        actor: _rivetkit_core_driver_helpers.ActorDriver;
-        topology: "standalone" | "partition" | "coordinate";
-        manager: _rivetkit_core_driver_helpers.ManagerDriver;
-        coordinate?: _rivetkit_core.CoordinateDriver | undefined;
+        actor: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig, managerDriver: _rivetkit_core_driver_helpers.ManagerDriver, inlineClient: _rivetkit_core.AnyClient) => _rivetkit_core_driver_helpers.ActorDriver;
+        manager: (registryConfig: _rivetkit_core.RegistryConfig, runConfig: _rivetkit_core.RunConfig) => _rivetkit_core_driver_helpers.ManagerDriver;
     }>>>;
     getUpgradeWebSocket: z.ZodOptional<z.ZodType<() => _rivetkit_core_utils.UpgradeWebSocket, z.ZodTypeDef, () => _rivetkit_core_utils.UpgradeWebSocket>>;
     cors: z.ZodOptional<z.ZodType<{
@@ -43,36 +37,10 @@ declare const ConfigSchema: z.ZodDefault<z.ZodObject<Omit<{
         exposeHeaders?: string[];
     }>>;
     maxIncomingMessageSize: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-    actorPeer: z.ZodDefault<z.ZodOptional<z.ZodObject<{
-        leaseDuration: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        renewLeaseGrace: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        checkLeaseInterval: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        checkLeaseJitter: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        messageAckTimeout: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-    }, "strip", z.ZodTypeAny, {
-        leaseDuration: number;
-        renewLeaseGrace: number;
-        checkLeaseInterval: number;
-        checkLeaseJitter: number;
-        messageAckTimeout: number;
-    }, {
-        leaseDuration?: number | undefined;
-        renewLeaseGrace?: number | undefined;
-        checkLeaseInterval?: number | undefined;
-        checkLeaseJitter?: number | undefined;
-        messageAckTimeout?: number | undefined;
-    }>>>;
 }, "driver" | "getUpgradeWebSocket"> & {
     app: z.ZodOptional<z.ZodType<Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/">, z.ZodTypeDef, Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/">>>;
 }, "strip", z.ZodTypeAny, {
     maxIncomingMessageSize: number;
-    actorPeer: {
-        leaseDuration: number;
-        renewLeaseGrace: number;
-        checkLeaseInterval: number;
-        checkLeaseJitter: number;
-        messageAckTimeout: number;
-    };
     cors?: {
         origin: string | string[] | ((origin: string, c: hono.Context) => string | undefined | null);
         allowMethods?: string[] | ((origin: string, c: hono.Context) => string[]);
@@ -92,13 +60,6 @@ declare const ConfigSchema: z.ZodDefault<z.ZodObject<Omit<{
         exposeHeaders?: string[];
     } | undefined;
     maxIncomingMessageSize?: number | undefined;
-    actorPeer?: {
-        leaseDuration?: number | undefined;
-        renewLeaseGrace?: number | undefined;
-        checkLeaseInterval?: number | undefined;
-        checkLeaseJitter?: number | undefined;
-        messageAckTimeout?: number | undefined;
-    } | undefined;
     app?: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/"> | undefined;
 }>>;
 type InputConfig = z.input<typeof ConfigSchema>;

package/dist/mod.js

@@ -1,15 +1,23 @@
 // src/handler.ts
 import { AsyncLocalStorage } from "async_hooks";
-import { PartitionTopologyManager } from "@rivetkit/core/topologies/partition";
 import { Hono } from "hono";
 import invariant2 from "invariant";
 
 // src/actor-handler-do.ts
 import { DurableObject } from "cloudflare:workers";
+import {
+  createActorRouter,
+  createClientWithDriver,
+  createInlineClientDriver
+} from "@rivetkit/core";
 import { serializeEmptyPersistData } from "@rivetkit/core/driver-helpers";
-import { PartitionTopologyActor } from "@rivetkit/core/topologies/partition";
 
 // src/actor-driver.ts
+import {
+  createGenericConnDrivers,
+  GenericConnGlobalState,
+  lookupInRegistry
+} from "@rivetkit/core";
 import invariant from "invariant";
 var CloudflareDurableObjectGlobalState = class {
   // Single map for all actor state
@@ -23,14 +31,76 @@ var CloudflareDurableObjectGlobalState = class {
     this.#dos.set(actorId, state);
   }
 };
+var ActorHandler = class {
+  actor;
+  actorPromise = Promise.withResolvers();
+  genericConnGlobalState = new GenericConnGlobalState();
+};
 var CloudflareActorsActorDriver = class {
+  #registryConfig;
+  #runConfig;
+  #managerDriver;
+  #inlineClient;
   #globalState;
-  constructor(globalState) {
+  #actors = /* @__PURE__ */ new Map();
+  constructor(registryConfig, runConfig, managerDriver, inlineClient, globalState) {
+    this.#registryConfig = registryConfig;
+    this.#runConfig = runConfig;
+    this.#managerDriver = managerDriver;
+    this.#inlineClient = inlineClient;
     this.#globalState = globalState;
   }
   #getDOCtx(actorId) {
     return this.#globalState.getDOState(actorId).ctx;
   }
+  async loadActor(actorId) {
+    var _a;
+    let handler = this.#actors.get(actorId);
+    if (handler) {
+      if (handler.actorPromise) await handler.actorPromise.promise;
+      if (!handler.actor) throw new Error("Actor should be loaded");
+      return handler.actor;
+    }
+    handler = new ActorHandler();
+    this.#actors.set(actorId, handler);
+    const doState = this.#globalState.getDOState(actorId);
+    const storage = doState.ctx.storage;
+    const [name, key] = await Promise.all([
+      storage.get(KEYS.NAME),
+      storage.get(KEYS.KEY)
+    ]);
+    if (!name) {
+      throw new Error(`Actor ${actorId} is not initialized - missing name`);
+    }
+    if (!key) {
+      throw new Error(`Actor ${actorId} is not initialized - missing key`);
+    }
+    const definition = lookupInRegistry(this.#registryConfig, name);
+    handler.actor = definition.instantiate();
+    const connDrivers = createGenericConnDrivers(
+      handler.genericConnGlobalState
+    );
+    await handler.actor.start(
+      connDrivers,
+      this,
+      this.#inlineClient,
+      actorId,
+      name,
+      key,
+      "unknown"
+      // TODO: Support regions in Cloudflare
+    );
+    (_a = handler.actorPromise) == null ? void 0 : _a.resolve();
+    handler.actorPromise = void 0;
+    return handler.actor;
+  }
+  getGenericConnGlobalState(actorId) {
+    const handler = this.#actors.get(actorId);
+    if (!handler) {
+      throw new Error(`Actor ${actorId} not loaded`);
+    }
+    return handler.genericConnGlobalState;
+  }
   getContext(actorId) {
     const state = this.#globalState.getDOState(actorId);
     return { ctx: state.ctx, env: state.env };
@@ -48,6 +118,17 @@ var CloudflareActorsActorDriver = class {
     return this.#getDOCtx(actorId).storage.sql;
   }
 };
+function createCloudflareActorsActorDriverBuilder(globalState) {
+  return (registryConfig, runConfig, managerDriver, inlineClient) => {
+    return new CloudflareActorsActorDriver(
+      registryConfig,
+      runConfig,
+      managerDriver,
+      inlineClient,
+      globalState
+    );
+  };
+}
 
 // src/log.ts
 import { getLogger } from "@rivetkit/core/log";
@@ -96,23 +177,27 @@ function createActorDurableObject(registry, runConfig) {
         return this.#actor;
       }
       if (!this.#initialized) throw new Error("Not initialized");
-      runConfig.driver.actor = new CloudflareActorsActorDriver(globalState);
-      const actorTopology = new PartitionTopologyActor(
+      const actorId = this.ctx.id.toString();
+      globalState.setDOState(actorId, { ctx: this.ctx, env: this.env });
+      runConfig.driver.actor = createCloudflareActorsActorDriverBuilder(globalState);
+      const managerDriver = runConfig.driver.manager(
         registry.config,
         runConfig
       );
-      const actorId = this.ctx.id.toString();
-      globalState.setDOState(actorId, { ctx: this.ctx, env: this.env });
+      const inlineClient = createClientWithDriver(
+        createInlineClientDriver(managerDriver)
+      );
+      const actorDriver = runConfig.driver.actor(
+        registry.config,
+        runConfig,
+        managerDriver,
+        inlineClient
+      );
+      const actorRouter = createActorRouter(runConfig, actorDriver);
       this.#actor = {
-        actorTopology
+        actorRouter
       };
-      await actorTopology.start(
-        actorId,
-        this.#initialized.name,
-        this.#initialized.key,
-        // TODO:
-        "unknown"
-      );
+      await actorDriver.loadActor(actorId);
       return this.#actor;
     }
     /** RPC called by the service that creates the DO to initialize it. */
@@ -133,27 +218,32 @@ function createActorDurableObject(registry, runConfig) {
     }
     async fetch(request) {
       return await CF_AMBIENT_ENV.run(this.env, async () => {
-        const { actorTopology } = await this.#loadActor();
-        const ctx = this.ctx;
-        return await actorTopology.router.fetch(
-          request,
-          this.env,
-          // Implement execution context so we can wait on requests
-          {
-            waitUntil(promise) {
-              ctx.waitUntil(promise);
-            },
-            passThroughOnException() {
-            },
-            props: {}
-          }
-        );
+        const { actorRouter } = await this.#loadActor();
+        const actorId = this.ctx.id.toString();
+        return await actorRouter.fetch(request, {
+          actorId
+        });
       });
     }
     async alarm() {
       return await CF_AMBIENT_ENV.run(this.env, async () => {
-        const { actorTopology } = await this.#loadActor();
-        await actorTopology.actor.onAlarm();
+        await this.#loadActor();
+        const actorId = this.ctx.id.toString();
+        const managerDriver = runConfig.driver.manager(
+          registry.config,
+          runConfig
+        );
+        const inlineClient = createClientWithDriver(
+          createInlineClientDriver(managerDriver)
+        );
+        const actorDriver = runConfig.driver.actor(
+          registry.config,
+          runConfig,
+          managerDriver,
+          inlineClient
+        );
+        const actor = await actorDriver.loadActor(actorId);
+        await actor.onAlarm();
       });
     }
   };
@@ -221,105 +311,103 @@ var STANDARD_WEBSOCKET_HEADERS = [
   "sec-websocket-extensions"
 ];
 var CloudflareActorsManagerDriver = class {
-  connRoutingHandler;
-  constructor() {
-    this.connRoutingHandler = {
-      custom: {
-        sendRequest: async (actorId, actorRequest) => {
-          const env = getCloudflareAmbientEnv();
-          logger().debug("sending request to durable object", {
-            actorId,
-            method: actorRequest.method,
-            url: actorRequest.url
-          });
-          const id = env.ACTOR_DO.idFromString(actorId);
-          const stub = env.ACTOR_DO.get(id);
-          return await stub.fetch(actorRequest);
-        },
-        openWebSocket: async (actorId, encodingKind, params) => {
-          const env = getCloudflareAmbientEnv();
-          logger().debug("opening websocket to durable object", { actorId });
-          const id = env.ACTOR_DO.idFromString(actorId);
-          const stub = env.ACTOR_DO.get(id);
-          const headers = {
-            Upgrade: "websocket",
-            Connection: "Upgrade",
-            [HEADER_EXPOSE_INTERNAL_ERROR]: "true",
-            [HEADER_ENCODING]: encodingKind
-          };
-          if (params) {
-            headers[HEADER_CONN_PARAMS] = JSON.stringify(params);
-          }
-          headers["sec-websocket-protocol"] = "rivetkit";
-          const response = await stub.fetch("http://actor/connect/websocket", {
-            headers
-          });
-          const webSocket = response.webSocket;
-          if (!webSocket) {
-            throw new InternalError(
-              "missing websocket connection in response from DO"
-            );
-          }
-          logger().debug("durable object websocket connection open", {
-            actorId
-          });
-          webSocket.accept();
-          setTimeout(() => {
-            var _a;
-            (_a = webSocket.onopen) == null ? void 0 : _a.call(webSocket, new Event("open"));
-          }, 0);
-          return webSocket;
-        },
-        proxyRequest: async (c, actorRequest, actorId) => {
-          logger().debug("forwarding request to durable object", {
-            actorId,
-            method: actorRequest.method,
-            url: actorRequest.url
-          });
-          const id = c.env.ACTOR_DO.idFromString(actorId);
-          const stub = c.env.ACTOR_DO.get(id);
-          return await stub.fetch(actorRequest);
-        },
-        proxyWebSocket: async (c, path, actorId, encoding, params, authData) => {
-          logger().debug("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);
-          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);
-            }
-          }
-          actorRequest.headers.set(HEADER_EXPOSE_INTERNAL_ERROR, "true");
-          actorRequest.headers.set(HEADER_ENCODING, encoding);
-          if (params) {
-            actorRequest.headers.set(
-              HEADER_CONN_PARAMS,
-              JSON.stringify(params)
-            );
-          }
-          if (authData) {
-            actorRequest.headers.set(
-              HEADER_AUTH_DATA,
-              JSON.stringify(authData)
-            );
-          }
-          const id = c.env.ACTOR_DO.idFromString(actorId);
-          const stub = c.env.ACTOR_DO.get(id);
-          return await stub.fetch(actorRequest);
-        }
-      }
+  async sendRequest(actorId, actorRequest) {
+    const env = getCloudflareAmbientEnv();
+    logger().debug("sending request to durable object", {
+      actorId,
+      method: actorRequest.method,
+      url: actorRequest.url
+    });
+    const id = env.ACTOR_DO.idFromString(actorId);
+    const stub = env.ACTOR_DO.get(id);
+    return await stub.fetch(actorRequest);
+  }
+  async openWebSocket(path, actorId, encoding, params) {
+    const env = getCloudflareAmbientEnv();
+    logger().debug("opening websocket to durable object", { actorId, path });
+    const id = env.ACTOR_DO.idFromString(actorId);
+    const stub = env.ACTOR_DO.get(id);
+    const headers = {
+      Upgrade: "websocket",
+      Connection: "Upgrade",
+      [HEADER_EXPOSE_INTERNAL_ERROR]: "true",
+      [HEADER_ENCODING]: encoding
     };
+    if (params) {
+      headers[HEADER_CONN_PARAMS] = JSON.stringify(params);
+    }
+    headers["sec-websocket-protocol"] = "rivetkit";
+    const url = `http://actor${path}`;
+    logger().debug("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"
+      );
+    }
+    logger().debug("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 proxyRequest(c, actorRequest, actorId) {
+    logger().debug("forwarding request to durable object", {
+      actorId,
+      method: actorRequest.method,
+      url: actorRequest.url
+    });
+    const id = c.env.ACTOR_DO.idFromString(actorId);
+    const stub = c.env.ACTOR_DO.get(id);
+    return await stub.fetch(actorRequest);
+  }
+  async proxyWebSocket(c, path, actorId, encoding, params, authData) {
+    logger().debug("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("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);
+      }
+    }
+    actorRequest.headers.set(HEADER_EXPOSE_INTERNAL_ERROR, "true");
+    actorRequest.headers.set(HEADER_ENCODING, encoding);
+    if (params) {
+      actorRequest.headers.set(HEADER_CONN_PARAMS, JSON.stringify(params));
+    }
+    if (authData) {
+      actorRequest.headers.set(HEADER_AUTH_DATA, JSON.stringify(authData));
+    }
+    const id = c.env.ACTOR_DO.idFromString(actorId);
+    const stub = c.env.ACTOR_DO.get(id);
+    return await stub.fetch(actorRequest);
   }
   async getForId({
     c,
@@ -501,30 +589,28 @@ function createServer(registry, inputConfig) {
   const config = ConfigSchema.parse(inputConfig);
   const runConfig = {
     driver: {
-      topology: "partition",
-      manager: new CloudflareActorsManagerDriver(),
-      // HACK: We can't build the actor driver until we're inside the Druable Object
+      manager: () => new CloudflareActorsManagerDriver(),
+      // HACK: We can't build the actor driver until we're inside the Durable Object
       actor: void 0
     },
     getUpgradeWebSocket: () => upgradeWebSocket,
     ...config
   };
-  const ActorHandler = createActorDurableObject(registry, runConfig);
-  const managerTopology = new PartitionTopologyManager(
-    registry.config,
-    runConfig
-  );
+  const ActorHandler2 = createActorDurableObject(registry, runConfig);
+  const serverOutput = registry.createServer(runConfig);
   return {
-    client: managerTopology.inlineClient,
+    client: serverOutput.client,
     createHandler: (hono) => {
       const app = hono ?? new Hono();
-      app.route("/registry", managerTopology.router);
+      if (!hono) {
+        app.route("/registry", serverOutput.hono);
+      }
       const handler = {
         fetch: (request, env, ctx) => {
           return CF_AMBIENT_ENV.run(env, () => app.fetch(request, env, ctx));
         }
       };
-      return { handler, ActorHandler };
+      return { handler, ActorHandler: ActorHandler2 };
     }
   };
 }