@rivetkit/cloudflare-workers 0.9.1 → 0.9.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/mod.cjs +228 -142
- package/dist/mod.cjs.map +1 -1
- package/dist/mod.d.cts +7 -46
- package/dist/mod.d.ts +7 -46
- package/dist/mod.js +227 -141
- package/dist/mod.js.map +1 -1
- package/package.json +4 -3
- package/src/actor-driver.ts +117 -1
- package/src/actor-handler-do.ts +70 -40
- package/src/handler.ts +9 -11
- package/src/manager-driver.ts +150 -143
package/dist/mod.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/handler.ts","../src/actor-handler-do.ts","../src/actor-driver.ts","../src/log.ts","../src/config.ts","../src/manager-driver.ts","../src/util.ts","../src/websocket.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport type { Registry, RunConfig } from \"@rivetkit/core\";\nimport type { Client } from \"@rivetkit/core/client\";\nimport { PartitionTopologyManager } from \"@rivetkit/core/topologies/partition\";\nimport { Hono } from \"hono\";\nimport invariant from \"invariant\";\nimport {\n\ttype ActorHandlerInterface,\n\tcreateActorDurableObject,\n\ttype DurableObjectConstructor,\n} from \"./actor-handler-do\";\nimport { ConfigSchema, type InputConfig } from \"./config\";\nimport { CloudflareActorsManagerDriver } from \"./manager-driver\";\nimport { upgradeWebSocket } from \"./websocket\";\n\n/** Cloudflare Workers env */\nexport interface Bindings {\n\tACTOR_KV: KVNamespace;\n\tACTOR_DO: DurableObjectNamespace<ActorHandlerInterface>;\n}\n\n/**\n * Stores the env for the current request. Required since some contexts like the inline client driver does not have access to the Hono context.\n *\n * Use getCloudflareAmbientEnv unless using CF_AMBIENT_ENV.run.\n */\nexport const CF_AMBIENT_ENV = new AsyncLocalStorage<Bindings>();\n\nexport function getCloudflareAmbientEnv(): Bindings {\n\tconst env = CF_AMBIENT_ENV.getStore();\n\tinvariant(env, \"missing CF_AMBIENT_ENV\");\n\treturn env;\n}\n\ninterface Handler {\n\thandler: ExportedHandler<Bindings>;\n\tActorHandler: DurableObjectConstructor;\n}\n\ninterface SetupOutput<A extends Registry<any>> {\n\tclient: Client<A>;\n\tcreateHandler: (hono?: Hono) => Handler;\n}\n\nexport function createServerHandler<R extends Registry<any>>(\n\tregistry: R,\n\tinputConfig?: InputConfig,\n): Handler {\n\tconst { createHandler } = createServer(registry, inputConfig);\n\treturn createHandler();\n}\n\nexport function createServer<R extends Registry<any>>(\n\tregistry: R,\n\tinputConfig?: InputConfig,\n): SetupOutput<R> {\n\tconst config = ConfigSchema.parse(inputConfig);\n\n\t// Create config\n\tconst runConfig = {\n\t\tdriver: {\n\t\t\ttopology: \"partition\",\n\t\t\tmanager: new CloudflareActorsManagerDriver(),\n\t\t\t// HACK: We can't build the actor driver until we're inside the Druable Object\n\t\t\tactor: undefined as any,\n\t\t},\n\t\tgetUpgradeWebSocket: () => upgradeWebSocket,\n\t\t...config,\n\t} satisfies RunConfig;\n\n\t// Create Durable Object\n\tconst ActorHandler = createActorDurableObject(registry, runConfig);\n\n\tconst managerTopology = new PartitionTopologyManager(\n\t\tregistry.config,\n\t\trunConfig,\n\t);\n\n\treturn {\n\t\tclient: managerTopology.inlineClient as Client<R>,\n\t\tcreateHandler: (hono) => {\n\t\t\t// Build base router\n\t\t\tconst app = hono ?? new Hono();\n\n\t\t\t// Mount registry\n\t\t\tapp.route(\"/registry\", managerTopology.router);\n\n\t\t\t// Create Cloudflare handler\n\t\t\tconst handler = {\n\t\t\t\tfetch: (request, env, ctx) => {\n\t\t\t\t\treturn CF_AMBIENT_ENV.run(env, () => app.fetch(request, env, ctx));\n\t\t\t\t},\n\t\t\t} satisfies ExportedHandler<Bindings>;\n\n\t\t\treturn { handler, ActorHandler };\n\t\t},\n\t};\n}\n","import { DurableObject } from \"cloudflare:workers\";\nimport type { ActorKey, Registry, RunConfig } from \"@rivetkit/core\";\nimport { serializeEmptyPersistData } from \"@rivetkit/core/driver-helpers\";\nimport { PartitionTopologyActor } from \"@rivetkit/core/topologies/partition\";\nimport type { ExecutionContext } from \"hono\";\nimport {\n\tCloudflareActorsActorDriver,\n\tCloudflareDurableObjectGlobalState,\n} from \"./actor-driver\";\nimport { type Bindings, CF_AMBIENT_ENV } from \"./handler\";\nimport { logger } from \"./log\";\n\nexport const KEYS = {\n\tNAME: \"rivetkit:name\",\n\tKEY: \"rivetkit:key\",\n\tPERSIST_DATA: \"rivetkit:data\",\n};\n\nexport interface ActorHandlerInterface extends DurableObject {\n\tinitialize(req: ActorInitRequest): Promise<void>;\n}\n\nexport interface ActorInitRequest {\n\tname: string;\n\tkey: ActorKey;\n\tinput?: unknown;\n}\n\ninterface InitializedData {\n\tname: string;\n\tkey: ActorKey;\n}\n\nexport type DurableObjectConstructor = new (\n\t...args: ConstructorParameters<typeof DurableObject<Bindings>>\n) => DurableObject<Bindings>;\n\ninterface LoadedActor {\n\tactorTopology: PartitionTopologyActor;\n}\n\nexport function createActorDurableObject(\n\tregistry: Registry<any>,\n\trunConfig: RunConfig,\n): DurableObjectConstructor {\n\tconst globalState = new CloudflareDurableObjectGlobalState();\n\n\t/**\n\t * Startup steps:\n\t * 1. If not already created call `initialize`, otherwise check KV to ensure it's initialized\n\t * 2. Load actor\n\t * 3. Start service requests\n\t */\n\treturn class ActorHandler\n\t\textends DurableObject<Bindings>\n\t\timplements ActorHandlerInterface\n\t{\n\t\t#initialized?: InitializedData;\n\t\t#initializedPromise?: PromiseWithResolvers<void>;\n\n\t\t#actor?: LoadedActor;\n\n\t\tasync #loadActor(): Promise<LoadedActor> {\n\t\t\t// This is always called from another context using CF_AMBIENT_ENV\n\n\t\t\t// Wait for init\n\t\t\tif (!this.#initialized) {\n\t\t\t\t// Wait for init\n\t\t\t\tif (this.#initializedPromise) {\n\t\t\t\t\tawait this.#initializedPromise.promise;\n\t\t\t\t} else {\n\t\t\t\t\tthis.#initializedPromise = Promise.withResolvers();\n\t\t\t\t\tconst res = await this.ctx.storage.get([\n\t\t\t\t\t\tKEYS.NAME,\n\t\t\t\t\t\tKEYS.KEY,\n\t\t\t\t\t\tKEYS.PERSIST_DATA,\n\t\t\t\t\t]);\n\t\t\t\t\tif (res.get(KEYS.PERSIST_DATA)) {\n\t\t\t\t\t\tconst name = res.get(KEYS.NAME) as string;\n\t\t\t\t\t\tif (!name) throw new Error(\"missing actor name\");\n\t\t\t\t\t\tconst key = res.get(KEYS.KEY) as ActorKey;\n\t\t\t\t\t\tif (!key) throw new Error(\"missing actor key\");\n\n\t\t\t\t\t\tlogger().debug(\"already initialized\", { name, key });\n\n\t\t\t\t\t\tthis.#initialized = { name, key };\n\t\t\t\t\t\tthis.#initializedPromise.resolve();\n\t\t\t\t\t} else {\n\t\t\t\t\t\tlogger().debug(\"waiting to initialize\");\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Check if already loaded\n\t\t\tif (this.#actor) {\n\t\t\t\treturn this.#actor;\n\t\t\t}\n\n\t\t\tif (!this.#initialized) throw new Error(\"Not initialized\");\n\n\t\t\t// Configure actor driver\n\t\t\trunConfig.driver.actor = new CloudflareActorsActorDriver(globalState);\n\n\t\t\tconst actorTopology = new PartitionTopologyActor(\n\t\t\t\tregistry.config,\n\t\t\t\trunConfig,\n\t\t\t);\n\n\t\t\t// Register DO with global state\n\t\t\t// HACK: This leaks the DO context, but DO does not provide a native way\n\t\t\t// of knowing when the DO shuts down. We're making a broad assumption\n\t\t\t// that DO will boot a new isolate frequenlty enough that this is not an issue.\n\t\t\tconst actorId = this.ctx.id.toString();\n\t\t\tglobalState.setDOState(actorId, { ctx: this.ctx, env: this.env });\n\n\t\t\t// Save actor\n\t\t\tthis.#actor = {\n\t\t\t\tactorTopology,\n\t\t\t};\n\n\t\t\t// Start actor\n\t\t\tawait actorTopology.start(\n\t\t\t\tactorId,\n\t\t\t\tthis.#initialized.name,\n\t\t\t\tthis.#initialized.key,\n\t\t\t\t// TODO:\n\t\t\t\t\"unknown\",\n\t\t\t);\n\n\t\t\treturn this.#actor;\n\t\t}\n\n\t\t/** RPC called by the service that creates the DO to initialize it. */\n\t\tasync initialize(req: ActorInitRequest) {\n\t\t\t// TODO: Need to add this to a core promise that needs to be resolved before start\n\n\t\t\treturn await CF_AMBIENT_ENV.run(this.env, async () => {\n\t\t\t\tawait this.ctx.storage.put({\n\t\t\t\t\t[KEYS.NAME]: req.name,\n\t\t\t\t\t[KEYS.KEY]: req.key,\n\t\t\t\t\t[KEYS.PERSIST_DATA]: serializeEmptyPersistData(req.input),\n\t\t\t\t});\n\t\t\t\tthis.#initialized = {\n\t\t\t\t\tname: req.name,\n\t\t\t\t\tkey: req.key,\n\t\t\t\t};\n\n\t\t\t\tlogger().debug(\"initialized actor\", { key: req.key });\n\n\t\t\t\t// Preemptively actor so the lifecycle hooks are called\n\t\t\t\tawait this.#loadActor();\n\t\t\t});\n\t\t}\n\n\t\tasync fetch(request: Request): Promise<Response> {\n\t\t\treturn await CF_AMBIENT_ENV.run(this.env, async () => {\n\t\t\t\tconst { actorTopology } = await this.#loadActor();\n\n\t\t\t\tconst ctx = this.ctx;\n\t\t\t\treturn await actorTopology.router.fetch(\n\t\t\t\t\trequest,\n\t\t\t\t\tthis.env,\n\t\t\t\t\t// Implement execution context so we can wait on requests\n\t\t\t\t\t{\n\t\t\t\t\t\twaitUntil(promise: Promise<unknown>) {\n\t\t\t\t\t\t\tctx.waitUntil(promise);\n\t\t\t\t\t\t},\n\t\t\t\t\t\tpassThroughOnException() {\n\t\t\t\t\t\t\t// Do nothing\n\t\t\t\t\t\t},\n\t\t\t\t\t\tprops: {},\n\t\t\t\t\t} satisfies ExecutionContext,\n\t\t\t\t);\n\t\t\t});\n\t\t}\n\n\t\tasync alarm(): Promise<void> {\n\t\t\treturn await CF_AMBIENT_ENV.run(this.env, async () => {\n\t\t\t\tconst { actorTopology } = await this.#loadActor();\n\t\t\t\tawait actorTopology.actor.onAlarm();\n\t\t\t});\n\t\t}\n\t};\n}\n","import type {\n\tActorDriver,\n\tAnyActorInstance,\n} from \"@rivetkit/core/driver-helpers\";\nimport invariant from \"invariant\";\nimport { KEYS } from \"./actor-handler-do\";\n\ninterface DurableObjectGlobalState {\n\tctx: DurableObjectState;\n\tenv: unknown;\n}\n\n/**\n * Cloudflare DO can have multiple DO running within the same global scope.\n *\n * This allows for storing the actor context globally and looking it up by ID in `CloudflareActorsActorDriver`.\n */\nexport class CloudflareDurableObjectGlobalState {\n\t// Single map for all actor state\n\t#dos: Map<string, DurableObjectGlobalState> = new Map();\n\n\tgetDOState(actorId: string): DurableObjectGlobalState {\n\t\tconst state = this.#dos.get(actorId);\n\t\tinvariant(state !== undefined, \"durable object state not in global state\");\n\t\treturn state;\n\t}\n\n\tsetDOState(actorId: string, state: DurableObjectGlobalState) {\n\t\tthis.#dos.set(actorId, state);\n\t}\n}\n\nexport interface ActorDriverContext {\n\tctx: DurableObjectState;\n\tenv: unknown;\n}\n\nexport class CloudflareActorsActorDriver implements ActorDriver {\n\t#globalState: CloudflareDurableObjectGlobalState;\n\n\tconstructor(globalState: CloudflareDurableObjectGlobalState) {\n\t\tthis.#globalState = globalState;\n\t}\n\n\t#getDOCtx(actorId: string) {\n\t\treturn this.#globalState.getDOState(actorId).ctx;\n\t}\n\n\tgetContext(actorId: string): ActorDriverContext {\n\t\tconst state = this.#globalState.getDOState(actorId);\n\t\treturn { ctx: state.ctx, env: state.env };\n\t}\n\n\tasync readPersistedData(actorId: string): Promise<Uint8Array | undefined> {\n\t\treturn await this.#getDOCtx(actorId).storage.get(KEYS.PERSIST_DATA);\n\t}\n\n\tasync writePersistedData(actorId: string, data: Uint8Array): Promise<void> {\n\t\tawait this.#getDOCtx(actorId).storage.put(KEYS.PERSIST_DATA, data);\n\t}\n\n\tasync setAlarm(actor: AnyActorInstance, timestamp: number): Promise<void> {\n\t\tawait this.#getDOCtx(actor.id).storage.setAlarm(timestamp);\n\t}\n\n\tasync getDatabase(actorId: string): Promise<unknown | undefined> {\n\t\treturn this.#getDOCtx(actorId).storage.sql;\n\t}\n}\n","import { getLogger } from \"@rivetkit/core/log\";\n\nexport const LOGGER_NAME = \"driver-cloudflare-workers\";\n\nexport function logger() {\n\treturn getLogger(LOGGER_NAME);\n}\n","import { RunConfigSchema } from \"@rivetkit/core/driver-helpers\";\nimport type { Hono } from \"hono\";\nimport { z } from \"zod\";\n\nexport const ConfigSchema = RunConfigSchema.removeDefault()\n\t.omit({ driver: true, getUpgradeWebSocket: true })\n\t.extend({\n\t\tapp: z.custom<Hono>().optional(),\n\t})\n\t.default({});\nexport type InputConfig = z.input<typeof ConfigSchema>;\nexport type Config = z.infer<typeof ConfigSchema>;\n","import type { Encoding } from \"@rivetkit/core\";\nimport {\n\ttype ActorOutput,\n\ttype ConnRoutingHandler,\n\ttype CreateInput,\n\ttype GetForIdInput,\n\ttype GetOrCreateWithKeyInput,\n\ttype GetWithKeyInput,\n\tHEADER_AUTH_DATA,\n\tHEADER_CONN_PARAMS,\n\tHEADER_ENCODING,\n\tHEADER_EXPOSE_INTERNAL_ERROR,\n\ttype ManagerDriver,\n} from \"@rivetkit/core/driver-helpers\";\nimport { ActorAlreadyExists, InternalError } from \"@rivetkit/core/errors\";\nimport { getCloudflareAmbientEnv } from \"./handler\";\nimport { logger } from \"./log\";\nimport type { Bindings } from \"./mod\";\nimport { serializeKey, serializeNameAndKey } from \"./util\";\n\n// Actor metadata structure\ninterface ActorData {\n\tname: string;\n\tkey: string[];\n}\n\n// Key constants similar to Redis implementation\nconst KEYS = {\n\tACTOR: {\n\t\t// Combined key for actor metadata (name and key)\n\t\tmetadata: (actorId: string) => `actor:${actorId}:metadata`,\n\n\t\t// Key index function for actor lookup\n\t\tkeyIndex: (name: string, key: string[] = []) => {\n\t\t\t// Use serializeKey for consistent handling of all keys\n\t\t\treturn `actor_key:${serializeKey(key)}`;\n\t\t},\n\t},\n};\n\nconst STANDARD_WEBSOCKET_HEADERS = [\n\t\"connection\",\n\t\"upgrade\",\n\t\"sec-websocket-key\",\n\t\"sec-websocket-version\",\n\t\"sec-websocket-protocol\",\n\t\"sec-websocket-extensions\",\n];\n\nexport class CloudflareActorsManagerDriver implements ManagerDriver {\n\tconnRoutingHandler: ConnRoutingHandler;\n\n\tconstructor() {\n\t\tthis.connRoutingHandler = {\n\t\t\tcustom: {\n\t\t\t\tsendRequest: async (actorId, actorRequest): Promise<Response> => {\n\t\t\t\t\tconst env = getCloudflareAmbientEnv();\n\n\t\t\t\t\tlogger().debug(\"sending request to durable object\", {\n\t\t\t\t\t\tactorId,\n\t\t\t\t\t\tmethod: actorRequest.method,\n\t\t\t\t\t\turl: actorRequest.url,\n\t\t\t\t\t});\n\n\t\t\t\t\tconst id = env.ACTOR_DO.idFromString(actorId);\n\t\t\t\t\tconst stub = env.ACTOR_DO.get(id);\n\n\t\t\t\t\treturn await stub.fetch(actorRequest);\n\t\t\t\t},\n\n\t\t\t\topenWebSocket: async (\n\t\t\t\t\tactorId,\n\t\t\t\t\tencodingKind: Encoding,\n\t\t\t\t\tparams: unknown,\n\t\t\t\t): Promise<WebSocket> => {\n\t\t\t\t\tconst env = getCloudflareAmbientEnv();\n\n\t\t\t\t\tlogger().debug(\"opening websocket to durable object\", { actorId });\n\n\t\t\t\t\t// Make a fetch request to the Durable Object with WebSocket upgrade\n\t\t\t\t\tconst id = env.ACTOR_DO.idFromString(actorId);\n\t\t\t\t\tconst stub = env.ACTOR_DO.get(id);\n\n\t\t\t\t\tconst headers: Record<string, string> = {\n\t\t\t\t\t\tUpgrade: \"websocket\",\n\t\t\t\t\t\tConnection: \"Upgrade\",\n\t\t\t\t\t\t[HEADER_EXPOSE_INTERNAL_ERROR]: \"true\",\n\t\t\t\t\t\t[HEADER_ENCODING]: encodingKind,\n\t\t\t\t\t};\n\t\t\t\t\tif (params) {\n\t\t\t\t\t\theaders[HEADER_CONN_PARAMS] = JSON.stringify(params);\n\t\t\t\t\t}\n\t\t\t\t\t// HACK: See packages/platforms/cloudflare-workers/src/websocket.ts\n\t\t\t\t\theaders[\"sec-websocket-protocol\"] = \"rivetkit\";\n\n\t\t\t\t\tconst response = await stub.fetch(\"http://actor/connect/websocket\", {\n\t\t\t\t\t\theaders,\n\t\t\t\t\t});\n\t\t\t\t\tconst webSocket = response.webSocket;\n\n\t\t\t\t\tif (!webSocket) {\n\t\t\t\t\t\tthrow new InternalError(\n\t\t\t\t\t\t\t\"missing websocket connection in response from DO\",\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\n\t\t\t\t\tlogger().debug(\"durable object websocket connection open\", {\n\t\t\t\t\t\tactorId,\n\t\t\t\t\t});\n\n\t\t\t\t\twebSocket.accept();\n\n\t\t\t\t\t// TODO: Is this still needed?\n\t\t\t\t\t// HACK: Cloudflare does not call onopen automatically, so we need\n\t\t\t\t\t// to call this on the next tick\n\t\t\t\t\tsetTimeout(() => {\n\t\t\t\t\t\t(webSocket as any).onopen?.(new Event(\"open\"));\n\t\t\t\t\t}, 0);\n\n\t\t\t\t\treturn webSocket as unknown as WebSocket;\n\t\t\t\t},\n\n\t\t\t\tproxyRequest: async (c, actorRequest, actorId): Promise<Response> => {\n\t\t\t\t\tlogger().debug(\"forwarding request to durable object\", {\n\t\t\t\t\t\tactorId,\n\t\t\t\t\t\tmethod: actorRequest.method,\n\t\t\t\t\t\turl: actorRequest.url,\n\t\t\t\t\t});\n\n\t\t\t\t\tconst id = c.env.ACTOR_DO.idFromString(actorId);\n\t\t\t\t\tconst stub = c.env.ACTOR_DO.get(id);\n\n\t\t\t\t\treturn await stub.fetch(actorRequest);\n\t\t\t\t},\n\t\t\t\tproxyWebSocket: async (\n\t\t\t\t\tc,\n\t\t\t\t\tpath,\n\t\t\t\t\tactorId,\n\t\t\t\t\tencoding,\n\t\t\t\t\tparams,\n\t\t\t\t\tauthData,\n\t\t\t\t) => {\n\t\t\t\t\tlogger().debug(\"forwarding websocket to durable object\", {\n\t\t\t\t\t\tactorId,\n\t\t\t\t\t\tpath,\n\t\t\t\t\t});\n\n\t\t\t\t\t// Validate upgrade\n\t\t\t\t\tconst upgradeHeader = c.req.header(\"Upgrade\");\n\t\t\t\t\tif (!upgradeHeader || upgradeHeader !== \"websocket\") {\n\t\t\t\t\t\treturn new Response(\"Expected Upgrade: websocket\", {\n\t\t\t\t\t\t\tstatus: 426,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\n\t\t\t\t\t// TODO: strip headers\n\t\t\t\t\tconst newUrl = new URL(`http://actor${path}`);\n\t\t\t\t\tconst actorRequest = new Request(newUrl, c.req.raw);\n\n\t\t\t\t\t// Always build fresh request to prevent forwarding unwanted headers\n\t\t\t\t\t// HACK: Since we can't build a new request, we need to remove\n\t\t\t\t\t// non-standard headers manually\n\t\t\t\t\tconst headerKeys: string[] = [];\n\t\t\t\t\tactorRequest.headers.forEach((v, k) => headerKeys.push(k));\n\t\t\t\t\tfor (const k of headerKeys) {\n\t\t\t\t\t\tif (!STANDARD_WEBSOCKET_HEADERS.includes(k)) {\n\t\t\t\t\t\t\tactorRequest.headers.delete(k);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\t// Add RivetKit headers\n\t\t\t\t\tactorRequest.headers.set(HEADER_EXPOSE_INTERNAL_ERROR, \"true\");\n\t\t\t\t\tactorRequest.headers.set(HEADER_ENCODING, encoding);\n\t\t\t\t\tif (params) {\n\t\t\t\t\t\tactorRequest.headers.set(\n\t\t\t\t\t\t\tHEADER_CONN_PARAMS,\n\t\t\t\t\t\t\tJSON.stringify(params),\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tif (authData) {\n\t\t\t\t\t\tactorRequest.headers.set(\n\t\t\t\t\t\t\tHEADER_AUTH_DATA,\n\t\t\t\t\t\t\tJSON.stringify(authData),\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\n\t\t\t\t\tconst id = c.env.ACTOR_DO.idFromString(actorId);\n\t\t\t\t\tconst stub = c.env.ACTOR_DO.get(id);\n\n\t\t\t\t\treturn await stub.fetch(actorRequest);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t}\n\n\tasync getForId({\n\t\tc,\n\t\tactorId,\n\t}: GetForIdInput<{ Bindings: Bindings }>): Promise<ActorOutput | undefined> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\t// Get actor metadata from KV (combined name and key)\n\t\tconst actorData = (await env.ACTOR_KV.get(KEYS.ACTOR.metadata(actorId), {\n\t\t\ttype: \"json\",\n\t\t})) as ActorData | null;\n\n\t\t// If the actor doesn't exist, return undefined\n\t\tif (!actorData) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\treturn {\n\t\t\tactorId,\n\t\t\tname: actorData.name,\n\t\t\tkey: actorData.key,\n\t\t};\n\t}\n\n\tasync getWithKey({\n\t\tc,\n\t\tname,\n\t\tkey,\n\t}: GetWithKeyInput<{ Bindings: Bindings }>): Promise<\n\t\tActorOutput | undefined\n\t> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\tlogger().debug(\"getWithKey: searching for actor\", { name, key });\n\n\t\t// Generate deterministic ID from the name and key\n\t\t// This is aligned with how createActor generates IDs\n\t\tconst nameKeyString = serializeNameAndKey(name, key);\n\t\tconst actorId = env.ACTOR_DO.idFromName(nameKeyString).toString();\n\n\t\t// Check if the actor metadata exists\n\t\tconst actorData = await env.ACTOR_KV.get(KEYS.ACTOR.metadata(actorId), {\n\t\t\ttype: \"json\",\n\t\t});\n\n\t\tif (!actorData) {\n\t\t\tlogger().debug(\"getWithKey: no actor found with matching name and key\", {\n\t\t\t\tname,\n\t\t\t\tkey,\n\t\t\t\tactorId,\n\t\t\t});\n\t\t\treturn undefined;\n\t\t}\n\n\t\tlogger().debug(\"getWithKey: found actor with matching name and key\", {\n\t\t\tactorId,\n\t\t\tname,\n\t\t\tkey,\n\t\t});\n\t\treturn this.#buildActorOutput(c, actorId);\n\t}\n\n\tasync getOrCreateWithKey(\n\t\tinput: GetOrCreateWithKeyInput,\n\t): Promise<ActorOutput> {\n\t\t// TODO: Prevent race condition here\n\t\tconst getOutput = await this.getWithKey(input);\n\t\tif (getOutput) {\n\t\t\treturn getOutput;\n\t\t} else {\n\t\t\treturn await this.createActor(input);\n\t\t}\n\t}\n\n\tasync createActor({\n\t\tc,\n\t\tname,\n\t\tkey,\n\t\tinput,\n\t}: CreateInput<{ Bindings: Bindings }>): Promise<ActorOutput> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\t// Check if actor with the same name and key already exists\n\t\tconst existingActor = await this.getWithKey({ c, name, key });\n\t\tif (existingActor) {\n\t\t\tthrow new ActorAlreadyExists(name, key);\n\t\t}\n\n\t\t// Create a deterministic ID from the actor name and key\n\t\t// This ensures that actors with the same name and key will have the same ID\n\t\tconst nameKeyString = serializeNameAndKey(name, key);\n\t\tconst doId = env.ACTOR_DO.idFromName(nameKeyString);\n\t\tconst actorId = doId.toString();\n\n\t\t// Init actor\n\t\tconst actor = env.ACTOR_DO.get(doId);\n\t\tawait actor.initialize({\n\t\t\tname,\n\t\t\tkey,\n\t\t\tinput,\n\t\t});\n\n\t\t// Store combined actor metadata (name and key)\n\t\tconst actorData: ActorData = { name, key };\n\t\tawait env.ACTOR_KV.put(\n\t\t\tKEYS.ACTOR.metadata(actorId),\n\t\t\tJSON.stringify(actorData),\n\t\t);\n\n\t\t// Add to key index for lookups by name and key\n\t\tawait env.ACTOR_KV.put(KEYS.ACTOR.keyIndex(name, key), actorId);\n\n\t\treturn {\n\t\t\tactorId,\n\t\t\tname,\n\t\t\tkey,\n\t\t};\n\t}\n\n\t// Helper method to build actor output from an ID\n\tasync #buildActorOutput(\n\t\tc: any,\n\t\tactorId: string,\n\t): Promise<ActorOutput | undefined> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\tconst actorData = (await env.ACTOR_KV.get(KEYS.ACTOR.metadata(actorId), {\n\t\t\ttype: \"json\",\n\t\t})) as ActorData | null;\n\n\t\tif (!actorData) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\treturn {\n\t\t\tactorId,\n\t\t\tname: actorData.name,\n\t\t\tkey: actorData.key,\n\t\t};\n\t}\n}\n","// Constants for key handling\nexport const EMPTY_KEY = \"(none)\";\nexport const KEY_SEPARATOR = \",\";\n\n/**\n * Serializes an array of key strings into a single string for use with idFromName\n *\n * @param name The actor name\n * @param key Array of key strings to serialize\n * @returns A single string containing the serialized name and key\n */\nexport function serializeNameAndKey(name: string, key: string[]): string {\n\t// Escape colons in the name\n\tconst escapedName = name.replace(/:/g, \"\\\\:\");\n\n\t// For empty keys, just return the name and a marker\n\tif (key.length === 0) {\n\t\treturn `${escapedName}:${EMPTY_KEY}`;\n\t}\n\n\t// Serialize the key array\n\tconst serializedKey = serializeKey(key);\n\n\t// Combine name and serialized key\n\treturn `${escapedName}:${serializedKey}`;\n}\n\n/**\n * Serializes an array of key strings into a single string\n *\n * @param key Array of key strings to serialize\n * @returns A single string containing the serialized key\n */\nexport function serializeKey(key: string[]): string {\n\t// Use a special marker for empty key arrays\n\tif (key.length === 0) {\n\t\treturn EMPTY_KEY;\n\t}\n\n\t// Escape each key part to handle the separator and the empty key marker\n\tconst escapedParts = key.map((part) => {\n\t\t// First check if it matches our empty key marker\n\t\tif (part === EMPTY_KEY) {\n\t\t\treturn `\\\\${EMPTY_KEY}`;\n\t\t}\n\n\t\t// Escape backslashes first, then commas\n\t\tlet escaped = part.replace(/\\\\/g, \"\\\\\\\\\");\n\t\tescaped = escaped.replace(/,/g, \"\\\\,\");\n\t\treturn escaped;\n\t});\n\n\treturn escapedParts.join(KEY_SEPARATOR);\n}\n\n/**\n * Deserializes a key string back into an array of key strings\n *\n * @param keyString The serialized key string\n * @returns Array of key strings\n */\nexport function deserializeKey(keyString: string): string[] {\n\t// Handle empty values\n\tif (!keyString) {\n\t\treturn [];\n\t}\n\n\t// Check for special empty key marker\n\tif (keyString === EMPTY_KEY) {\n\t\treturn [];\n\t}\n\n\t// Split by unescaped commas and unescape the escaped characters\n\tconst parts: string[] = [];\n\tlet currentPart = \"\";\n\tlet escaping = false;\n\n\tfor (let i = 0; i < keyString.length; i++) {\n\t\tconst char = keyString[i];\n\n\t\tif (escaping) {\n\t\t\t// This is an escaped character, add it directly\n\t\t\tcurrentPart += char;\n\t\t\tescaping = false;\n\t\t} else if (char === \"\\\\\") {\n\t\t\t// Start of an escape sequence\n\t\t\tescaping = true;\n\t\t} else if (char === KEY_SEPARATOR) {\n\t\t\t// This is a separator\n\t\t\tparts.push(currentPart);\n\t\t\tcurrentPart = \"\";\n\t\t} else {\n\t\t\t// Regular character\n\t\t\tcurrentPart += char;\n\t\t}\n\t}\n\n\t// Add the last part if it exists\n\tif (currentPart || parts.length > 0) {\n\t\tparts.push(currentPart);\n\t}\n\n\treturn parts;\n}\n","// Modified from https://github.com/honojs/hono/blob/40ea0eee58e39b31053a0246c595434f1094ad31/src/adapter/cloudflare-workers/websocket.ts#L17\n//\n// This version calls the open event by default\n\nimport type { UpgradeWebSocket, WSEvents, WSReadyState } from \"hono/ws\";\nimport { defineWebSocketHelper, WSContext } from \"hono/ws\";\n\n// Based on https://github.com/honojs/hono/issues/1153#issuecomment-1767321332\nexport const upgradeWebSocket: UpgradeWebSocket<\n\tWebSocket,\n\t// eslint-disable-next-line @typescript-eslint/no-explicit-any\n\tany,\n\tWSEvents<WebSocket>\n> = defineWebSocketHelper(async (c, events) => {\n\tconst upgradeHeader = c.req.header(\"Upgrade\");\n\tif (upgradeHeader !== \"websocket\") {\n\t\treturn;\n\t}\n\n\tconst webSocketPair = new WebSocketPair();\n\tconst client: WebSocket = webSocketPair[0];\n\tconst server: WebSocket = webSocketPair[1];\n\n\tconst wsContext = new WSContext<WebSocket>({\n\t\tclose: (code, reason) => server.close(code, reason),\n\t\tget protocol() {\n\t\t\treturn server.protocol;\n\t\t},\n\t\traw: server,\n\t\tget readyState() {\n\t\t\treturn server.readyState as WSReadyState;\n\t\t},\n\t\turl: server.url ? new URL(server.url) : null,\n\t\tsend: (source) => server.send(source),\n\t});\n\n\tif (events.onClose) {\n\t\tserver.addEventListener(\"close\", (evt: CloseEvent) =>\n\t\t\tevents.onClose?.(evt, wsContext),\n\t\t);\n\t}\n\tif (events.onMessage) {\n\t\tserver.addEventListener(\"message\", (evt: MessageEvent) =>\n\t\t\tevents.onMessage?.(evt, wsContext),\n\t\t);\n\t}\n\tif (events.onError) {\n\t\tserver.addEventListener(\"error\", (evt: Event) =>\n\t\t\tevents.onError?.(evt, wsContext),\n\t\t);\n\t}\n\n\tserver.accept?.();\n\n\t// note: cloudflare actors doesn't support 'open' event, so we call it immediately with a fake event\n\t//\n\t// we have to do this after `server.accept() is called`\n\tevents.onOpen?.(new Event(\"open\"), wsContext);\n\n\treturn new Response(null, {\n\t\tstatus: 101,\n\t\theaders: {\n\t\t\t// HACK: Required in order for Cloudflare to not error with \"Network connection lost\"\n\t\t\t//\n\t\t\t// This bug undocumented. Cannot easily reproduce outside of RivetKit.\n\t\t\t\"Sec-WebSocket-Protocol\": \"rivetkit\",\n\t\t},\n\t\twebSocket: client,\n\t});\n});\n"],"mappings":";AAAA,SAAS,yBAAyB;AAGlC,SAAS,gCAAgC;AACzC,SAAS,YAAY;AACrB,OAAOA,gBAAe;;;ACLtB,SAAS,qBAAqB;AAE9B,SAAS,iCAAiC;AAC1C,SAAS,8BAA8B;;;ACCvC,OAAO,eAAe;AAaf,IAAM,qCAAN,MAAyC;AAAA;AAAA,EAE/C,OAA8C,oBAAI,IAAI;AAAA,EAEtD,WAAW,SAA2C;AACrD,UAAM,QAAQ,KAAK,KAAK,IAAI,OAAO;AACnC,cAAU,UAAU,QAAW,0CAA0C;AACzE,WAAO;AAAA,EACR;AAAA,EAEA,WAAW,SAAiB,OAAiC;AAC5D,SAAK,KAAK,IAAI,SAAS,KAAK;AAAA,EAC7B;AACD;AAOO,IAAM,8BAAN,MAAyD;AAAA,EAC/D;AAAA,EAEA,YAAY,aAAiD;AAC5D,SAAK,eAAe;AAAA,EACrB;AAAA,EAEA,UAAU,SAAiB;AAC1B,WAAO,KAAK,aAAa,WAAW,OAAO,EAAE;AAAA,EAC9C;AAAA,EAEA,WAAW,SAAqC;AAC/C,UAAM,QAAQ,KAAK,aAAa,WAAW,OAAO;AAClD,WAAO,EAAE,KAAK,MAAM,KAAK,KAAK,MAAM,IAAI;AAAA,EACzC;AAAA,EAEA,MAAM,kBAAkB,SAAkD;AACzE,WAAO,MAAM,KAAK,UAAU,OAAO,EAAE,QAAQ,IAAI,KAAK,YAAY;AAAA,EACnE;AAAA,EAEA,MAAM,mBAAmB,SAAiB,MAAiC;AAC1E,UAAM,KAAK,UAAU,OAAO,EAAE,QAAQ,IAAI,KAAK,cAAc,IAAI;AAAA,EAClE;AAAA,EAEA,MAAM,SAAS,OAAyB,WAAkC;AACzE,UAAM,KAAK,UAAU,MAAM,EAAE,EAAE,QAAQ,SAAS,SAAS;AAAA,EAC1D;AAAA,EAEA,MAAM,YAAY,SAA+C;AAChE,WAAO,KAAK,UAAU,OAAO,EAAE,QAAQ;AAAA,EACxC;AACD;;;ACpEA,SAAS,iBAAiB;AAEnB,IAAM,cAAc;AAEpB,SAAS,SAAS;AACxB,SAAO,UAAU,WAAW;AAC7B;;;AFMO,IAAM,OAAO;AAAA,EACnB,MAAM;AAAA,EACN,KAAK;AAAA,EACL,cAAc;AACf;AAyBO,SAAS,yBACf,UACA,WAC2B;AAC3B,QAAM,cAAc,IAAI,mCAAmC;AAQ3D,SAAO,MAAM,qBACJ,cAET;AAAA,IACC;AAAA,IACA;AAAA,IAEA;AAAA,IAEA,MAAM,aAAmC;AAIxC,UAAI,CAAC,KAAK,cAAc;AAEvB,YAAI,KAAK,qBAAqB;AAC7B,gBAAM,KAAK,oBAAoB;AAAA,QAChC,OAAO;AACN,eAAK,sBAAsB,QAAQ,cAAc;AACjD,gBAAM,MAAM,MAAM,KAAK,IAAI,QAAQ,IAAI;AAAA,YACtC,KAAK;AAAA,YACL,KAAK;AAAA,YACL,KAAK;AAAA,UACN,CAAC;AACD,cAAI,IAAI,IAAI,KAAK,YAAY,GAAG;AAC/B,kBAAM,OAAO,IAAI,IAAI,KAAK,IAAI;AAC9B,gBAAI,CAAC,KAAM,OAAM,IAAI,MAAM,oBAAoB;AAC/C,kBAAM,MAAM,IAAI,IAAI,KAAK,GAAG;AAC5B,gBAAI,CAAC,IAAK,OAAM,IAAI,MAAM,mBAAmB;AAE7C,mBAAO,EAAE,MAAM,uBAAuB,EAAE,MAAM,IAAI,CAAC;AAEnD,iBAAK,eAAe,EAAE,MAAM,IAAI;AAChC,iBAAK,oBAAoB,QAAQ;AAAA,UAClC,OAAO;AACN,mBAAO,EAAE,MAAM,uBAAuB;AAAA,UACvC;AAAA,QACD;AAAA,MACD;AAGA,UAAI,KAAK,QAAQ;AAChB,eAAO,KAAK;AAAA,MACb;AAEA,UAAI,CAAC,KAAK,aAAc,OAAM,IAAI,MAAM,iBAAiB;AAGzD,gBAAU,OAAO,QAAQ,IAAI,4BAA4B,WAAW;AAEpE,YAAM,gBAAgB,IAAI;AAAA,QACzB,SAAS;AAAA,QACT;AAAA,MACD;AAMA,YAAM,UAAU,KAAK,IAAI,GAAG,SAAS;AACrC,kBAAY,WAAW,SAAS,EAAE,KAAK,KAAK,KAAK,KAAK,KAAK,IAAI,CAAC;AAGhE,WAAK,SAAS;AAAA,QACb;AAAA,MACD;AAGA,YAAM,cAAc;AAAA,QACnB;AAAA,QACA,KAAK,aAAa;AAAA,QAClB,KAAK,aAAa;AAAA;AAAA,QAElB;AAAA,MACD;AAEA,aAAO,KAAK;AAAA,IACb;AAAA;AAAA,IAGA,MAAM,WAAW,KAAuB;AAGvC,aAAO,MAAM,eAAe,IAAI,KAAK,KAAK,YAAY;AACrD,cAAM,KAAK,IAAI,QAAQ,IAAI;AAAA,UAC1B,CAAC,KAAK,IAAI,GAAG,IAAI;AAAA,UACjB,CAAC,KAAK,GAAG,GAAG,IAAI;AAAA,UAChB,CAAC,KAAK,YAAY,GAAG,0BAA0B,IAAI,KAAK;AAAA,QACzD,CAAC;AACD,aAAK,eAAe;AAAA,UACnB,MAAM,IAAI;AAAA,UACV,KAAK,IAAI;AAAA,QACV;AAEA,eAAO,EAAE,MAAM,qBAAqB,EAAE,KAAK,IAAI,IAAI,CAAC;AAGpD,cAAM,KAAK,WAAW;AAAA,MACvB,CAAC;AAAA,IACF;AAAA,IAEA,MAAM,MAAM,SAAqC;AAChD,aAAO,MAAM,eAAe,IAAI,KAAK,KAAK,YAAY;AACrD,cAAM,EAAE,cAAc,IAAI,MAAM,KAAK,WAAW;AAEhD,cAAM,MAAM,KAAK;AACjB,eAAO,MAAM,cAAc,OAAO;AAAA,UACjC;AAAA,UACA,KAAK;AAAA;AAAA,UAEL;AAAA,YACC,UAAU,SAA2B;AACpC,kBAAI,UAAU,OAAO;AAAA,YACtB;AAAA,YACA,yBAAyB;AAAA,YAEzB;AAAA,YACA,OAAO,CAAC;AAAA,UACT;AAAA,QACD;AAAA,MACD,CAAC;AAAA,IACF;AAAA,IAEA,MAAM,QAAuB;AAC5B,aAAO,MAAM,eAAe,IAAI,KAAK,KAAK,YAAY;AACrD,cAAM,EAAE,cAAc,IAAI,MAAM,KAAK,WAAW;AAChD,cAAM,cAAc,MAAM,QAAQ;AAAA,MACnC,CAAC;AAAA,IACF;AAAA,EACD;AACD;;;AGvLA,SAAS,uBAAuB;AAEhC,SAAS,SAAS;AAEX,IAAM,eAAe,gBAAgB,cAAc,EACxD,KAAK,EAAE,QAAQ,MAAM,qBAAqB,KAAK,CAAC,EAChD,OAAO;AAAA,EACP,KAAK,EAAE,OAAa,EAAE,SAAS;AAChC,CAAC,EACA,QAAQ,CAAC,CAAC;;;ACRZ;AAAA,EAOC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEM;AACP,SAAS,oBAAoB,qBAAqB;;;ACb3C,IAAM,YAAY;AAClB,IAAM,gBAAgB;AAStB,SAAS,oBAAoB,MAAc,KAAuB;AAExE,QAAM,cAAc,KAAK,QAAQ,MAAM,KAAK;AAG5C,MAAI,IAAI,WAAW,GAAG;AACrB,WAAO,GAAG,WAAW,IAAI,SAAS;AAAA,EACnC;AAGA,QAAM,gBAAgB,aAAa,GAAG;AAGtC,SAAO,GAAG,WAAW,IAAI,aAAa;AACvC;AAQO,SAAS,aAAa,KAAuB;AAEnD,MAAI,IAAI,WAAW,GAAG;AACrB,WAAO;AAAA,EACR;AAGA,QAAM,eAAe,IAAI,IAAI,CAAC,SAAS;AAEtC,QAAI,SAAS,WAAW;AACvB,aAAO,KAAK,SAAS;AAAA,IACtB;AAGA,QAAI,UAAU,KAAK,QAAQ,OAAO,MAAM;AACxC,cAAU,QAAQ,QAAQ,MAAM,KAAK;AACrC,WAAO;AAAA,EACR,CAAC;AAED,SAAO,aAAa,KAAK,aAAa;AACvC;;;AD1BA,IAAMC,QAAO;AAAA,EACZ,OAAO;AAAA;AAAA,IAEN,UAAU,CAAC,YAAoB,SAAS,OAAO;AAAA;AAAA,IAG/C,UAAU,CAAC,MAAc,MAAgB,CAAC,MAAM;AAE/C,aAAO,aAAa,aAAa,GAAG,CAAC;AAAA,IACtC;AAAA,EACD;AACD;AAEA,IAAM,6BAA6B;AAAA,EAClC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACD;AAEO,IAAM,gCAAN,MAA6D;AAAA,EACnE;AAAA,EAEA,cAAc;AACb,SAAK,qBAAqB;AAAA,MACzB,QAAQ;AAAA,QACP,aAAa,OAAO,SAAS,iBAAoC;AAChE,gBAAM,MAAM,wBAAwB;AAEpC,iBAAO,EAAE,MAAM,qCAAqC;AAAA,YACnD;AAAA,YACA,QAAQ,aAAa;AAAA,YACrB,KAAK,aAAa;AAAA,UACnB,CAAC;AAED,gBAAM,KAAK,IAAI,SAAS,aAAa,OAAO;AAC5C,gBAAM,OAAO,IAAI,SAAS,IAAI,EAAE;AAEhC,iBAAO,MAAM,KAAK,MAAM,YAAY;AAAA,QACrC;AAAA,QAEA,eAAe,OACd,SACA,cACA,WACwB;AACxB,gBAAM,MAAM,wBAAwB;AAEpC,iBAAO,EAAE,MAAM,uCAAuC,EAAE,QAAQ,CAAC;AAGjE,gBAAM,KAAK,IAAI,SAAS,aAAa,OAAO;AAC5C,gBAAM,OAAO,IAAI,SAAS,IAAI,EAAE;AAEhC,gBAAM,UAAkC;AAAA,YACvC,SAAS;AAAA,YACT,YAAY;AAAA,YACZ,CAAC,4BAA4B,GAAG;AAAA,YAChC,CAAC,eAAe,GAAG;AAAA,UACpB;AACA,cAAI,QAAQ;AACX,oBAAQ,kBAAkB,IAAI,KAAK,UAAU,MAAM;AAAA,UACpD;AAEA,kBAAQ,wBAAwB,IAAI;AAEpC,gBAAM,WAAW,MAAM,KAAK,MAAM,kCAAkC;AAAA,YACnE;AAAA,UACD,CAAC;AACD,gBAAM,YAAY,SAAS;AAE3B,cAAI,CAAC,WAAW;AACf,kBAAM,IAAI;AAAA,cACT;AAAA,YACD;AAAA,UACD;AAEA,iBAAO,EAAE,MAAM,4CAA4C;AAAA,YAC1D;AAAA,UACD,CAAC;AAED,oBAAU,OAAO;AAKjB,qBAAW,MAAM;AAnHtB;AAoHM,aAAC,eAAkB,WAAlB,mCAA2B,IAAI,MAAM,MAAM;AAAA,UAC7C,GAAG,CAAC;AAEJ,iBAAO;AAAA,QACR;AAAA,QAEA,cAAc,OAAO,GAAG,cAAc,YAA+B;AACpE,iBAAO,EAAE,MAAM,wCAAwC;AAAA,YACtD;AAAA,YACA,QAAQ,aAAa;AAAA,YACrB,KAAK,aAAa;AAAA,UACnB,CAAC;AAED,gBAAM,KAAK,EAAE,IAAI,SAAS,aAAa,OAAO;AAC9C,gBAAM,OAAO,EAAE,IAAI,SAAS,IAAI,EAAE;AAElC,iBAAO,MAAM,KAAK,MAAM,YAAY;AAAA,QACrC;AAAA,QACA,gBAAgB,OACf,GACA,MACA,SACA,UACA,QACA,aACI;AACJ,iBAAO,EAAE,MAAM,0CAA0C;AAAA,YACxD;AAAA,YACA;AAAA,UACD,CAAC;AAGD,gBAAM,gBAAgB,EAAE,IAAI,OAAO,SAAS;AAC5C,cAAI,CAAC,iBAAiB,kBAAkB,aAAa;AACpD,mBAAO,IAAI,SAAS,+BAA+B;AAAA,cAClD,QAAQ;AAAA,YACT,CAAC;AAAA,UACF;AAGA,gBAAM,SAAS,IAAI,IAAI,eAAe,IAAI,EAAE;AAC5C,gBAAM,eAAe,IAAI,QAAQ,QAAQ,EAAE,IAAI,GAAG;AAKlD,gBAAM,aAAuB,CAAC;AAC9B,uBAAa,QAAQ,QAAQ,CAAC,GAAG,MAAM,WAAW,KAAK,CAAC,CAAC;AACzD,qBAAW,KAAK,YAAY;AAC3B,gBAAI,CAAC,2BAA2B,SAAS,CAAC,GAAG;AAC5C,2BAAa,QAAQ,OAAO,CAAC;AAAA,YAC9B;AAAA,UACD;AAGA,uBAAa,QAAQ,IAAI,8BAA8B,MAAM;AAC7D,uBAAa,QAAQ,IAAI,iBAAiB,QAAQ;AAClD,cAAI,QAAQ;AACX,yBAAa,QAAQ;AAAA,cACpB;AAAA,cACA,KAAK,UAAU,MAAM;AAAA,YACtB;AAAA,UACD;AACA,cAAI,UAAU;AACb,yBAAa,QAAQ;AAAA,cACpB;AAAA,cACA,KAAK,UAAU,QAAQ;AAAA,YACxB;AAAA,UACD;AAEA,gBAAM,KAAK,EAAE,IAAI,SAAS,aAAa,OAAO;AAC9C,gBAAM,OAAO,EAAE,IAAI,SAAS,IAAI,EAAE;AAElC,iBAAO,MAAM,KAAK,MAAM,YAAY;AAAA,QACrC;AAAA,MACD;AAAA,IACD;AAAA,EACD;AAAA,EAEA,MAAM,SAAS;AAAA,IACd;AAAA,IACA;AAAA,EACD,GAA4E;AAC3E,UAAM,MAAM,wBAAwB;AAGpC,UAAM,YAAa,MAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,OAAO,GAAG;AAAA,MACvE,MAAM;AAAA,IACP,CAAC;AAGD,QAAI,CAAC,WAAW;AACf,aAAO;AAAA,IACR;AAEA,WAAO;AAAA,MACN;AAAA,MACA,MAAM,UAAU;AAAA,MAChB,KAAK,UAAU;AAAA,IAChB;AAAA,EACD;AAAA,EAEA,MAAM,WAAW;AAAA,IAChB;AAAA,IACA;AAAA,IACA;AAAA,EACD,GAEE;AACD,UAAM,MAAM,wBAAwB;AAEpC,WAAO,EAAE,MAAM,mCAAmC,EAAE,MAAM,IAAI,CAAC;AAI/D,UAAM,gBAAgB,oBAAoB,MAAM,GAAG;AACnD,UAAM,UAAU,IAAI,SAAS,WAAW,aAAa,EAAE,SAAS;AAGhE,UAAM,YAAY,MAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,OAAO,GAAG;AAAA,MACtE,MAAM;AAAA,IACP,CAAC;AAED,QAAI,CAAC,WAAW;AACf,aAAO,EAAE,MAAM,yDAAyD;AAAA,QACvE;AAAA,QACA;AAAA,QACA;AAAA,MACD,CAAC;AACD,aAAO;AAAA,IACR;AAEA,WAAO,EAAE,MAAM,sDAAsD;AAAA,MACpE;AAAA,MACA;AAAA,MACA;AAAA,IACD,CAAC;AACD,WAAO,KAAK,kBAAkB,GAAG,OAAO;AAAA,EACzC;AAAA,EAEA,MAAM,mBACL,OACuB;AAEvB,UAAM,YAAY,MAAM,KAAK,WAAW,KAAK;AAC7C,QAAI,WAAW;AACd,aAAO;AAAA,IACR,OAAO;AACN,aAAO,MAAM,KAAK,YAAY,KAAK;AAAA,IACpC;AAAA,EACD;AAAA,EAEA,MAAM,YAAY;AAAA,IACjB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACD,GAA8D;AAC7D,UAAM,MAAM,wBAAwB;AAGpC,UAAM,gBAAgB,MAAM,KAAK,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC;AAC5D,QAAI,eAAe;AAClB,YAAM,IAAI,mBAAmB,MAAM,GAAG;AAAA,IACvC;AAIA,UAAM,gBAAgB,oBAAoB,MAAM,GAAG;AACnD,UAAM,OAAO,IAAI,SAAS,WAAW,aAAa;AAClD,UAAM,UAAU,KAAK,SAAS;AAG9B,UAAM,QAAQ,IAAI,SAAS,IAAI,IAAI;AACnC,UAAM,MAAM,WAAW;AAAA,MACtB;AAAA,MACA;AAAA,MACA;AAAA,IACD,CAAC;AAGD,UAAM,YAAuB,EAAE,MAAM,IAAI;AACzC,UAAM,IAAI,SAAS;AAAA,MAClBA,MAAK,MAAM,SAAS,OAAO;AAAA,MAC3B,KAAK,UAAU,SAAS;AAAA,IACzB;AAGA,UAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,MAAM,GAAG,GAAG,OAAO;AAE9D,WAAO;AAAA,MACN;AAAA,MACA;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAAA;AAAA,EAGA,MAAM,kBACL,GACA,SACmC;AACnC,UAAM,MAAM,wBAAwB;AAEpC,UAAM,YAAa,MAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,OAAO,GAAG;AAAA,MACvE,MAAM;AAAA,IACP,CAAC;AAED,QAAI,CAAC,WAAW;AACf,aAAO;AAAA,IACR;AAEA,WAAO;AAAA,MACN;AAAA,MACA,MAAM,UAAU;AAAA,MAChB,KAAK,UAAU;AAAA,IAChB;AAAA,EACD;AACD;;;AEzUA,SAAS,uBAAuB,iBAAiB;AAG1C,IAAM,mBAKT,sBAAsB,OAAO,GAAG,WAAW;AAb/C;AAcC,QAAM,gBAAgB,EAAE,IAAI,OAAO,SAAS;AAC5C,MAAI,kBAAkB,aAAa;AAClC;AAAA,EACD;AAEA,QAAM,gBAAgB,IAAI,cAAc;AACxC,QAAM,SAAoB,cAAc,CAAC;AACzC,QAAM,SAAoB,cAAc,CAAC;AAEzC,QAAM,YAAY,IAAI,UAAqB;AAAA,IAC1C,OAAO,CAAC,MAAM,WAAW,OAAO,MAAM,MAAM,MAAM;AAAA,IAClD,IAAI,WAAW;AACd,aAAO,OAAO;AAAA,IACf;AAAA,IACA,KAAK;AAAA,IACL,IAAI,aAAa;AAChB,aAAO,OAAO;AAAA,IACf;AAAA,IACA,KAAK,OAAO,MAAM,IAAI,IAAI,OAAO,GAAG,IAAI;AAAA,IACxC,MAAM,CAAC,WAAW,OAAO,KAAK,MAAM;AAAA,EACrC,CAAC;AAED,MAAI,OAAO,SAAS;AACnB,WAAO;AAAA,MAAiB;AAAA,MAAS,CAAC,QAAiB;AArCrD,YAAAC;AAsCG,gBAAAA,MAAA,OAAO,YAAP,gBAAAA,IAAA,aAAiB,KAAK;AAAA;AAAA,IACvB;AAAA,EACD;AACA,MAAI,OAAO,WAAW;AACrB,WAAO;AAAA,MAAiB;AAAA,MAAW,CAAC,QAAmB;AA1CzD,YAAAA;AA2CG,gBAAAA,MAAA,OAAO,cAAP,gBAAAA,IAAA,aAAmB,KAAK;AAAA;AAAA,IACzB;AAAA,EACD;AACA,MAAI,OAAO,SAAS;AACnB,WAAO;AAAA,MAAiB;AAAA,MAAS,CAAC,QAAY;AA/ChD,YAAAA;AAgDG,gBAAAA,MAAA,OAAO,YAAP,gBAAAA,IAAA,aAAiB,KAAK;AAAA;AAAA,IACvB;AAAA,EACD;AAEA,eAAO,WAAP;AAKA,eAAO,WAAP,gCAAgB,IAAI,MAAM,MAAM,GAAG;AAEnC,SAAO,IAAI,SAAS,MAAM;AAAA,IACzB,QAAQ;AAAA,IACR,SAAS;AAAA;AAAA;AAAA;AAAA,MAIR,0BAA0B;AAAA,IAC3B;AAAA,IACA,WAAW;AAAA,EACZ,CAAC;AACF,CAAC;;;AP3CM,IAAM,iBAAiB,IAAI,kBAA4B;AAEvD,SAAS,0BAAoC;AACnD,QAAM,MAAM,eAAe,SAAS;AACpC,EAAAC,WAAU,KAAK,wBAAwB;AACvC,SAAO;AACR;AAYO,SAAS,oBACf,UACA,aACU;AACV,QAAM,EAAE,cAAc,IAAI,aAAa,UAAU,WAAW;AAC5D,SAAO,cAAc;AACtB;AAEO,SAAS,aACf,UACA,aACiB;AACjB,QAAM,SAAS,aAAa,MAAM,WAAW;AAG7C,QAAM,YAAY;AAAA,IACjB,QAAQ;AAAA,MACP,UAAU;AAAA,MACV,SAAS,IAAI,8BAA8B;AAAA;AAAA,MAE3C,OAAO;AAAA,IACR;AAAA,IACA,qBAAqB,MAAM;AAAA,IAC3B,GAAG;AAAA,EACJ;AAGA,QAAM,eAAe,yBAAyB,UAAU,SAAS;AAEjE,QAAM,kBAAkB,IAAI;AAAA,IAC3B,SAAS;AAAA,IACT;AAAA,EACD;AAEA,SAAO;AAAA,IACN,QAAQ,gBAAgB;AAAA,IACxB,eAAe,CAAC,SAAS;AAExB,YAAM,MAAM,QAAQ,IAAI,KAAK;AAG7B,UAAI,MAAM,aAAa,gBAAgB,MAAM;AAG7C,YAAM,UAAU;AAAA,QACf,OAAO,CAAC,SAAS,KAAK,QAAQ;AAC7B,iBAAO,eAAe,IAAI,KAAK,MAAM,IAAI,MAAM,SAAS,KAAK,GAAG,CAAC;AAAA,QAClE;AAAA,MACD;AAEA,aAAO,EAAE,SAAS,aAAa;AAAA,IAChC;AAAA,EACD;AACD;","names":["invariant","KEYS","_a","invariant"]}
|
|
1
|
+
{"version":3,"sources":["../src/handler.ts","../src/actor-handler-do.ts","../src/actor-driver.ts","../src/log.ts","../src/config.ts","../src/manager-driver.ts","../src/util.ts","../src/websocket.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport type { Registry, RunConfig } from \"@rivetkit/core\";\nimport type { Client } from \"@rivetkit/core/client\";\nimport { Hono } from \"hono\";\nimport invariant from \"invariant\";\nimport {\n\ttype ActorHandlerInterface,\n\tcreateActorDurableObject,\n\ttype DurableObjectConstructor,\n} from \"./actor-handler-do\";\nimport { ConfigSchema, type InputConfig } from \"./config\";\nimport { CloudflareActorsManagerDriver } from \"./manager-driver\";\nimport { upgradeWebSocket } from \"./websocket\";\n\n/** Cloudflare Workers env */\nexport interface Bindings {\n\tACTOR_KV: KVNamespace;\n\tACTOR_DO: DurableObjectNamespace<ActorHandlerInterface>;\n}\n\n/**\n * Stores the env for the current request. Required since some contexts like the inline client driver does not have access to the Hono context.\n *\n * Use getCloudflareAmbientEnv unless using CF_AMBIENT_ENV.run.\n */\nexport const CF_AMBIENT_ENV = new AsyncLocalStorage<Bindings>();\n\nexport function getCloudflareAmbientEnv(): Bindings {\n\tconst env = CF_AMBIENT_ENV.getStore();\n\tinvariant(env, \"missing CF_AMBIENT_ENV\");\n\treturn env;\n}\n\ninterface Handler {\n\thandler: ExportedHandler<Bindings>;\n\tActorHandler: DurableObjectConstructor;\n}\n\ninterface SetupOutput<A extends Registry<any>> {\n\tclient: Client<A>;\n\tcreateHandler: (hono?: Hono) => Handler;\n}\n\nexport function createServerHandler<R extends Registry<any>>(\n\tregistry: R,\n\tinputConfig?: InputConfig,\n): Handler {\n\tconst { createHandler } = createServer(registry, inputConfig);\n\treturn createHandler();\n}\n\nexport function createServer<R extends Registry<any>>(\n\tregistry: R,\n\tinputConfig?: InputConfig,\n): SetupOutput<R> {\n\tconst config = ConfigSchema.parse(inputConfig);\n\n\t// Create config\n\tconst runConfig = {\n\t\tdriver: {\n\t\t\tmanager: () => new CloudflareActorsManagerDriver(),\n\t\t\t// HACK: We can't build the actor driver until we're inside the Durable Object\n\t\t\tactor: undefined as any,\n\t\t},\n\t\tgetUpgradeWebSocket: () => upgradeWebSocket,\n\t\t...config,\n\t} satisfies RunConfig;\n\n\t// Create Durable Object\n\tconst ActorHandler = createActorDurableObject(registry, runConfig);\n\n\t// Create server\n\tconst serverOutput = registry.createServer(runConfig);\n\n\treturn {\n\t\tclient: serverOutput.client as Client<R>,\n\t\tcreateHandler: (hono) => {\n\t\t\t// Build base router\n\t\t\tconst app = hono ?? new Hono();\n\n\t\t\t// Mount registry router at /registry\n\t\t\tif (!hono) {\n\t\t\t\tapp.route(\"/registry\", serverOutput.hono);\n\t\t\t}\n\n\t\t\t// Create Cloudflare handler\n\t\t\tconst handler = {\n\t\t\t\tfetch: (request, env, ctx) => {\n\t\t\t\t\treturn CF_AMBIENT_ENV.run(env, () => app.fetch(request, env, ctx));\n\t\t\t\t},\n\t\t\t} satisfies ExportedHandler<Bindings>;\n\n\t\t\treturn { handler, ActorHandler };\n\t\t},\n\t};\n}\n","import { DurableObject } from \"cloudflare:workers\";\nimport type {\n\tActorKey,\n\tActorRouter,\n\tRegistry,\n\tRunConfig,\n} from \"@rivetkit/core\";\nimport {\n\tcreateActorRouter,\n\tcreateClientWithDriver,\n\tcreateInlineClientDriver,\n} from \"@rivetkit/core\";\nimport { serializeEmptyPersistData } from \"@rivetkit/core/driver-helpers\";\nimport type { ExecutionContext } from \"hono\";\nimport {\n\tCloudflareDurableObjectGlobalState,\n\tcreateCloudflareActorsActorDriverBuilder,\n} from \"./actor-driver\";\nimport { type Bindings, CF_AMBIENT_ENV } from \"./handler\";\nimport { logger } from \"./log\";\n\nexport const KEYS = {\n\tNAME: \"rivetkit:name\",\n\tKEY: \"rivetkit:key\",\n\tPERSIST_DATA: \"rivetkit:data\",\n};\n\nexport interface ActorHandlerInterface extends DurableObject {\n\tinitialize(req: ActorInitRequest): Promise<void>;\n}\n\nexport interface ActorInitRequest {\n\tname: string;\n\tkey: ActorKey;\n\tinput?: unknown;\n}\n\ninterface InitializedData {\n\tname: string;\n\tkey: ActorKey;\n}\n\nexport type DurableObjectConstructor = new (\n\t...args: ConstructorParameters<typeof DurableObject<Bindings>>\n) => DurableObject<Bindings>;\n\ninterface LoadedActor {\n\tactorRouter: ActorRouter;\n}\n\nexport function createActorDurableObject(\n\tregistry: Registry<any>,\n\trunConfig: RunConfig,\n): DurableObjectConstructor {\n\tconst globalState = new CloudflareDurableObjectGlobalState();\n\n\t/**\n\t * Startup steps:\n\t * 1. If not already created call `initialize`, otherwise check KV to ensure it's initialized\n\t * 2. Load actor\n\t * 3. Start service requests\n\t */\n\treturn class ActorHandler\n\t\textends DurableObject<Bindings>\n\t\timplements ActorHandlerInterface\n\t{\n\t\t#initialized?: InitializedData;\n\t\t#initializedPromise?: PromiseWithResolvers<void>;\n\n\t\t#actor?: LoadedActor;\n\n\t\tasync #loadActor(): Promise<LoadedActor> {\n\t\t\t// This is always called from another context using CF_AMBIENT_ENV\n\n\t\t\t// Wait for init\n\t\t\tif (!this.#initialized) {\n\t\t\t\t// Wait for init\n\t\t\t\tif (this.#initializedPromise) {\n\t\t\t\t\tawait this.#initializedPromise.promise;\n\t\t\t\t} else {\n\t\t\t\t\tthis.#initializedPromise = Promise.withResolvers();\n\t\t\t\t\tconst res = await this.ctx.storage.get([\n\t\t\t\t\t\tKEYS.NAME,\n\t\t\t\t\t\tKEYS.KEY,\n\t\t\t\t\t\tKEYS.PERSIST_DATA,\n\t\t\t\t\t]);\n\t\t\t\t\tif (res.get(KEYS.PERSIST_DATA)) {\n\t\t\t\t\t\tconst name = res.get(KEYS.NAME) as string;\n\t\t\t\t\t\tif (!name) throw new Error(\"missing actor name\");\n\t\t\t\t\t\tconst key = res.get(KEYS.KEY) as ActorKey;\n\t\t\t\t\t\tif (!key) throw new Error(\"missing actor key\");\n\n\t\t\t\t\t\tlogger().debug(\"already initialized\", { name, key });\n\n\t\t\t\t\t\tthis.#initialized = { name, key };\n\t\t\t\t\t\tthis.#initializedPromise.resolve();\n\t\t\t\t\t} else {\n\t\t\t\t\t\tlogger().debug(\"waiting to initialize\");\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Check if already loaded\n\t\t\tif (this.#actor) {\n\t\t\t\treturn this.#actor;\n\t\t\t}\n\n\t\t\tif (!this.#initialized) throw new Error(\"Not initialized\");\n\n\t\t\t// Register DO with global state first\n\t\t\t// HACK: This leaks the DO context, but DO does not provide a native way\n\t\t\t// of knowing when the DO shuts down. We're making a broad assumption\n\t\t\t// that DO will boot a new isolate frequenlty enough that this is not an issue.\n\t\t\tconst actorId = this.ctx.id.toString();\n\t\t\tglobalState.setDOState(actorId, { ctx: this.ctx, env: this.env });\n\n\t\t\t// Configure actor driver\n\t\t\trunConfig.driver.actor =\n\t\t\t\tcreateCloudflareActorsActorDriverBuilder(globalState);\n\n\t\t\t// Create manager driver (we need this for the actor router)\n\t\t\tconst managerDriver = runConfig.driver.manager(\n\t\t\t\tregistry.config,\n\t\t\t\trunConfig,\n\t\t\t);\n\n\t\t\t// Create inline client\n\t\t\tconst inlineClient = createClientWithDriver(\n\t\t\t\tcreateInlineClientDriver(managerDriver),\n\t\t\t);\n\n\t\t\t// Create actor driver\n\t\t\tconst actorDriver = runConfig.driver.actor(\n\t\t\t\tregistry.config,\n\t\t\t\trunConfig,\n\t\t\t\tmanagerDriver,\n\t\t\t\tinlineClient,\n\t\t\t);\n\n\t\t\t// Create actor router\n\t\t\tconst actorRouter = createActorRouter(runConfig, actorDriver);\n\n\t\t\t// Save actor\n\t\t\tthis.#actor = {\n\t\t\t\tactorRouter,\n\t\t\t};\n\n\t\t\t// Initialize the actor instance with proper metadata\n\t\t\t// This ensures the actor driver knows about this actor\n\t\t\tawait actorDriver.loadActor(actorId);\n\n\t\t\treturn this.#actor;\n\t\t}\n\n\t\t/** RPC called by the service that creates the DO to initialize it. */\n\t\tasync initialize(req: ActorInitRequest) {\n\t\t\t// TODO: Need to add this to a core promise that needs to be resolved before start\n\n\t\t\treturn await CF_AMBIENT_ENV.run(this.env, async () => {\n\t\t\t\tawait this.ctx.storage.put({\n\t\t\t\t\t[KEYS.NAME]: req.name,\n\t\t\t\t\t[KEYS.KEY]: req.key,\n\t\t\t\t\t[KEYS.PERSIST_DATA]: serializeEmptyPersistData(req.input),\n\t\t\t\t});\n\t\t\t\tthis.#initialized = {\n\t\t\t\t\tname: req.name,\n\t\t\t\t\tkey: req.key,\n\t\t\t\t};\n\n\t\t\t\tlogger().debug(\"initialized actor\", { key: req.key });\n\n\t\t\t\t// Preemptively actor so the lifecycle hooks are called\n\t\t\t\tawait this.#loadActor();\n\t\t\t});\n\t\t}\n\n\t\tasync fetch(request: Request): Promise<Response> {\n\t\t\treturn await CF_AMBIENT_ENV.run(this.env, async () => {\n\t\t\t\tconst { actorRouter } = await this.#loadActor();\n\n\t\t\t\tconst actorId = this.ctx.id.toString();\n\t\t\t\treturn await actorRouter.fetch(request, {\n\t\t\t\t\tactorId,\n\t\t\t\t});\n\t\t\t});\n\t\t}\n\n\t\tasync alarm(): Promise<void> {\n\t\t\treturn await CF_AMBIENT_ENV.run(this.env, async () => {\n\t\t\t\tawait this.#loadActor();\n\t\t\t\tconst actorId = this.ctx.id.toString();\n\n\t\t\t\t// Get the actor driver\n\t\t\t\tconst managerDriver = runConfig.driver.manager(\n\t\t\t\t\tregistry.config,\n\t\t\t\t\trunConfig,\n\t\t\t\t);\n\t\t\t\tconst inlineClient = createClientWithDriver(\n\t\t\t\t\tcreateInlineClientDriver(managerDriver),\n\t\t\t\t);\n\t\t\t\tconst actorDriver = runConfig.driver.actor(\n\t\t\t\t\tregistry.config,\n\t\t\t\t\trunConfig,\n\t\t\t\t\tmanagerDriver,\n\t\t\t\t\tinlineClient,\n\t\t\t\t);\n\n\t\t\t\t// Load the actor instance and trigger alarm\n\t\t\t\tconst actor = await actorDriver.loadActor(actorId);\n\t\t\t\tawait actor.onAlarm();\n\t\t\t});\n\t\t}\n\t};\n}\n","import type {\n\tAnyActorInstance as CoreAnyActorInstance,\n\tRegistryConfig,\n\tRunConfig,\n} from \"@rivetkit/core\";\nimport {\n\tcreateGenericConnDrivers,\n\tGenericConnGlobalState,\n\tlookupInRegistry,\n} from \"@rivetkit/core\";\nimport type { Client } from \"@rivetkit/core/client\";\nimport type {\n\tActorDriver,\n\tAnyActorInstance,\n\tManagerDriver,\n} from \"@rivetkit/core/driver-helpers\";\nimport invariant from \"invariant\";\nimport { KEYS } from \"./actor-handler-do\";\n\ninterface DurableObjectGlobalState {\n\tctx: DurableObjectState;\n\tenv: unknown;\n}\n\n/**\n * Cloudflare DO can have multiple DO running within the same global scope.\n *\n * This allows for storing the actor context globally and looking it up by ID in `CloudflareActorsActorDriver`.\n */\nexport class CloudflareDurableObjectGlobalState {\n\t// Single map for all actor state\n\t#dos: Map<string, DurableObjectGlobalState> = new Map();\n\n\tgetDOState(actorId: string): DurableObjectGlobalState {\n\t\tconst state = this.#dos.get(actorId);\n\t\tinvariant(state !== undefined, \"durable object state not in global state\");\n\t\treturn state;\n\t}\n\n\tsetDOState(actorId: string, state: DurableObjectGlobalState) {\n\t\tthis.#dos.set(actorId, state);\n\t}\n}\n\nexport interface ActorDriverContext {\n\tctx: DurableObjectState;\n\tenv: unknown;\n}\n\n// Actor handler to track running instances\nclass ActorHandler {\n\tactor?: AnyActorInstance;\n\tactorPromise?: PromiseWithResolvers<void> = Promise.withResolvers();\n\tgenericConnGlobalState = new GenericConnGlobalState();\n}\n\nexport class CloudflareActorsActorDriver implements ActorDriver {\n\t#registryConfig: RegistryConfig;\n\t#runConfig: RunConfig;\n\t#managerDriver: ManagerDriver;\n\t#inlineClient: Client<any>;\n\t#globalState: CloudflareDurableObjectGlobalState;\n\t#actors: Map<string, ActorHandler> = new Map();\n\n\tconstructor(\n\t\tregistryConfig: RegistryConfig,\n\t\trunConfig: RunConfig,\n\t\tmanagerDriver: ManagerDriver,\n\t\tinlineClient: Client<any>,\n\t\tglobalState: CloudflareDurableObjectGlobalState,\n\t) {\n\t\tthis.#registryConfig = registryConfig;\n\t\tthis.#runConfig = runConfig;\n\t\tthis.#managerDriver = managerDriver;\n\t\tthis.#inlineClient = inlineClient;\n\t\tthis.#globalState = globalState;\n\t}\n\n\t#getDOCtx(actorId: string) {\n\t\treturn this.#globalState.getDOState(actorId).ctx;\n\t}\n\n\tasync loadActor(actorId: string): Promise<AnyActorInstance> {\n\t\t// Check if actor is already loaded\n\t\tlet handler = this.#actors.get(actorId);\n\t\tif (handler) {\n\t\t\tif (handler.actorPromise) await handler.actorPromise.promise;\n\t\t\tif (!handler.actor) throw new Error(\"Actor should be loaded\");\n\t\t\treturn handler.actor;\n\t\t}\n\n\t\t// Create new actor handler\n\t\thandler = new ActorHandler();\n\t\tthis.#actors.set(actorId, handler);\n\n\t\t// Get the actor metadata from Durable Object storage\n\t\tconst doState = this.#globalState.getDOState(actorId);\n\t\tconst storage = doState.ctx.storage;\n\n\t\t// Load actor metadata\n\t\tconst [name, key] = await Promise.all([\n\t\t\tstorage.get<string>(KEYS.NAME),\n\t\t\tstorage.get<string[]>(KEYS.KEY),\n\t\t]);\n\n\t\tif (!name) {\n\t\t\tthrow new Error(`Actor ${actorId} is not initialized - missing name`);\n\t\t}\n\t\tif (!key) {\n\t\t\tthrow new Error(`Actor ${actorId} is not initialized - missing key`);\n\t\t}\n\n\t\t// Create actor instance\n\t\tconst definition = lookupInRegistry(this.#registryConfig, name);\n\t\thandler.actor = definition.instantiate();\n\n\t\t// Start actor\n\t\tconst connDrivers = createGenericConnDrivers(\n\t\t\thandler.genericConnGlobalState,\n\t\t);\n\t\tawait handler.actor.start(\n\t\t\tconnDrivers,\n\t\t\tthis,\n\t\t\tthis.#inlineClient,\n\t\t\tactorId,\n\t\t\tname,\n\t\t\tkey,\n\t\t\t\"unknown\", // TODO: Support regions in Cloudflare\n\t\t);\n\n\t\t// Finish\n\t\thandler.actorPromise?.resolve();\n\t\thandler.actorPromise = undefined;\n\n\t\treturn handler.actor;\n\t}\n\n\tgetGenericConnGlobalState(actorId: string): GenericConnGlobalState {\n\t\tconst handler = this.#actors.get(actorId);\n\t\tif (!handler) {\n\t\t\tthrow new Error(`Actor ${actorId} not loaded`);\n\t\t}\n\t\treturn handler.genericConnGlobalState;\n\t}\n\n\tgetContext(actorId: string): ActorDriverContext {\n\t\tconst state = this.#globalState.getDOState(actorId);\n\t\treturn { ctx: state.ctx, env: state.env };\n\t}\n\n\tasync readPersistedData(actorId: string): Promise<Uint8Array | undefined> {\n\t\treturn await this.#getDOCtx(actorId).storage.get(KEYS.PERSIST_DATA);\n\t}\n\n\tasync writePersistedData(actorId: string, data: Uint8Array): Promise<void> {\n\t\tawait this.#getDOCtx(actorId).storage.put(KEYS.PERSIST_DATA, data);\n\t}\n\n\tasync setAlarm(actor: AnyActorInstance, timestamp: number): Promise<void> {\n\t\tawait this.#getDOCtx(actor.id).storage.setAlarm(timestamp);\n\t}\n\n\tasync getDatabase(actorId: string): Promise<unknown | undefined> {\n\t\treturn this.#getDOCtx(actorId).storage.sql;\n\t}\n}\n\nexport function createCloudflareActorsActorDriverBuilder(\n\tglobalState: CloudflareDurableObjectGlobalState,\n) {\n\treturn (\n\t\tregistryConfig: RegistryConfig,\n\t\trunConfig: RunConfig,\n\t\tmanagerDriver: ManagerDriver,\n\t\tinlineClient: Client<any>,\n\t) => {\n\t\treturn new CloudflareActorsActorDriver(\n\t\t\tregistryConfig,\n\t\t\trunConfig,\n\t\t\tmanagerDriver,\n\t\t\tinlineClient,\n\t\t\tglobalState,\n\t\t);\n\t};\n}\n","import { getLogger } from \"@rivetkit/core/log\";\n\nexport const LOGGER_NAME = \"driver-cloudflare-workers\";\n\nexport function logger() {\n\treturn getLogger(LOGGER_NAME);\n}\n","import { RunConfigSchema } from \"@rivetkit/core/driver-helpers\";\nimport type { Hono } from \"hono\";\nimport { z } from \"zod\";\n\nexport const ConfigSchema = RunConfigSchema.removeDefault()\n\t.omit({ driver: true, getUpgradeWebSocket: true })\n\t.extend({\n\t\tapp: z.custom<Hono>().optional(),\n\t})\n\t.default({});\nexport type InputConfig = z.input<typeof ConfigSchema>;\nexport type Config = z.infer<typeof ConfigSchema>;\n","import type { Encoding } from \"@rivetkit/core\";\nimport {\n\ttype ActorOutput,\n\ttype CreateInput,\n\ttype GetForIdInput,\n\ttype GetOrCreateWithKeyInput,\n\ttype GetWithKeyInput,\n\tHEADER_AUTH_DATA,\n\tHEADER_CONN_PARAMS,\n\tHEADER_ENCODING,\n\tHEADER_EXPOSE_INTERNAL_ERROR,\n\ttype ManagerDriver,\n} from \"@rivetkit/core/driver-helpers\";\nimport { ActorAlreadyExists, InternalError } from \"@rivetkit/core/errors\";\nimport type { Context as HonoContext } from \"hono\";\nimport { getCloudflareAmbientEnv } from \"./handler\";\nimport { logger } from \"./log\";\nimport type { Bindings } from \"./mod\";\nimport { serializeKey, serializeNameAndKey } from \"./util\";\n\n// Actor metadata structure\ninterface ActorData {\n\tname: string;\n\tkey: string[];\n}\n\n// Key constants similar to Redis implementation\nconst KEYS = {\n\tACTOR: {\n\t\t// Combined key for actor metadata (name and key)\n\t\tmetadata: (actorId: string) => `actor:${actorId}:metadata`,\n\n\t\t// Key index function for actor lookup\n\t\tkeyIndex: (name: string, key: string[] = []) => {\n\t\t\t// Use serializeKey for consistent handling of all keys\n\t\t\treturn `actor_key:${serializeKey(key)}`;\n\t\t},\n\t},\n};\n\nconst STANDARD_WEBSOCKET_HEADERS = [\n\t\"connection\",\n\t\"upgrade\",\n\t\"sec-websocket-key\",\n\t\"sec-websocket-version\",\n\t\"sec-websocket-protocol\",\n\t\"sec-websocket-extensions\",\n];\n\nexport class CloudflareActorsManagerDriver implements ManagerDriver {\n\tasync sendRequest(actorId: string, actorRequest: Request): Promise<Response> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\tlogger().debug(\"sending request to durable object\", {\n\t\t\tactorId,\n\t\t\tmethod: actorRequest.method,\n\t\t\turl: actorRequest.url,\n\t\t});\n\n\t\tconst id = env.ACTOR_DO.idFromString(actorId);\n\t\tconst stub = env.ACTOR_DO.get(id);\n\n\t\treturn await stub.fetch(actorRequest);\n\t}\n\n\tasync openWebSocket(\n\t\tpath: string,\n\t\tactorId: string,\n\t\tencoding: Encoding,\n\t\tparams: unknown,\n\t): Promise<WebSocket> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\tlogger().debug(\"opening websocket to durable object\", { actorId, path });\n\n\t\t// Make a fetch request to the Durable Object with WebSocket upgrade\n\t\tconst id = env.ACTOR_DO.idFromString(actorId);\n\t\tconst stub = env.ACTOR_DO.get(id);\n\n\t\tconst headers: Record<string, string> = {\n\t\t\tUpgrade: \"websocket\",\n\t\t\tConnection: \"Upgrade\",\n\t\t\t[HEADER_EXPOSE_INTERNAL_ERROR]: \"true\",\n\t\t\t[HEADER_ENCODING]: encoding,\n\t\t};\n\t\tif (params) {\n\t\t\theaders[HEADER_CONN_PARAMS] = JSON.stringify(params);\n\t\t}\n\t\t// HACK: See packages/drivers/cloudflare-workers/src/websocket.ts\n\t\theaders[\"sec-websocket-protocol\"] = \"rivetkit\";\n\n\t\t// Use the path parameter to determine the URL\n\t\tconst url = `http://actor${path}`;\n\n\t\tlogger().debug(\"rewriting websocket url\", {\n\t\t\tfrom: path,\n\t\t\tto: url,\n\t\t});\n\n\t\tconst response = await stub.fetch(url, {\n\t\t\theaders,\n\t\t});\n\t\tconst webSocket = response.webSocket;\n\n\t\tif (!webSocket) {\n\t\t\tthrow new InternalError(\n\t\t\t\t\"missing websocket connection in response from DO\",\n\t\t\t);\n\t\t}\n\n\t\tlogger().debug(\"durable object websocket connection open\", {\n\t\t\tactorId,\n\t\t});\n\n\t\twebSocket.accept();\n\n\t\t// TODO: Is this still needed?\n\t\t// HACK: Cloudflare does not call onopen automatically, so we need\n\t\t// to call this on the next tick\n\t\tsetTimeout(() => {\n\t\t\tconst event = new Event(\"open\");\n\t\t\t(webSocket as any).onopen?.(event);\n\t\t\t(webSocket as any).dispatchEvent(event);\n\t\t}, 0);\n\n\t\treturn webSocket as unknown as WebSocket;\n\t}\n\n\tasync proxyRequest(\n\t\tc: HonoContext<{ Bindings: Bindings }>,\n\t\tactorRequest: Request,\n\t\tactorId: string,\n\t): Promise<Response> {\n\t\tlogger().debug(\"forwarding request to durable object\", {\n\t\t\tactorId,\n\t\t\tmethod: actorRequest.method,\n\t\t\turl: actorRequest.url,\n\t\t});\n\n\t\tconst id = c.env.ACTOR_DO.idFromString(actorId);\n\t\tconst stub = c.env.ACTOR_DO.get(id);\n\n\t\treturn await stub.fetch(actorRequest);\n\t}\n\n\tasync proxyWebSocket(\n\t\tc: HonoContext<{ Bindings: Bindings }>,\n\t\tpath: string,\n\t\tactorId: string,\n\t\tencoding: Encoding,\n\t\tparams: unknown,\n\t\tauthData: unknown,\n\t): Promise<Response> {\n\t\tlogger().debug(\"forwarding websocket to durable object\", {\n\t\t\tactorId,\n\t\t\tpath,\n\t\t});\n\n\t\t// Validate upgrade\n\t\tconst upgradeHeader = c.req.header(\"Upgrade\");\n\t\tif (!upgradeHeader || upgradeHeader !== \"websocket\") {\n\t\t\treturn new Response(\"Expected Upgrade: websocket\", {\n\t\t\t\tstatus: 426,\n\t\t\t});\n\t\t}\n\n\t\t// TODO: strip headers\n\t\tconst newUrl = new URL(`http://actor${path}`);\n\t\tconst actorRequest = new Request(newUrl, c.req.raw);\n\n\t\tlogger().debug(\"rewriting websocket url\", {\n\t\t\tfrom: c.req.url,\n\t\t\tto: actorRequest.url,\n\t\t});\n\n\t\t// Always build fresh request to prevent forwarding unwanted headers\n\t\t// HACK: Since we can't build a new request, we need to remove\n\t\t// non-standard headers manually\n\t\tconst headerKeys: string[] = [];\n\t\tactorRequest.headers.forEach((v, k) => headerKeys.push(k));\n\t\tfor (const k of headerKeys) {\n\t\t\tif (!STANDARD_WEBSOCKET_HEADERS.includes(k)) {\n\t\t\t\tactorRequest.headers.delete(k);\n\t\t\t}\n\t\t}\n\n\t\t// Add RivetKit headers\n\t\tactorRequest.headers.set(HEADER_EXPOSE_INTERNAL_ERROR, \"true\");\n\t\tactorRequest.headers.set(HEADER_ENCODING, encoding);\n\t\tif (params) {\n\t\t\tactorRequest.headers.set(HEADER_CONN_PARAMS, JSON.stringify(params));\n\t\t}\n\t\tif (authData) {\n\t\t\tactorRequest.headers.set(HEADER_AUTH_DATA, JSON.stringify(authData));\n\t\t}\n\n\t\tconst id = c.env.ACTOR_DO.idFromString(actorId);\n\t\tconst stub = c.env.ACTOR_DO.get(id);\n\n\t\treturn await stub.fetch(actorRequest);\n\t}\n\n\tasync getForId({\n\t\tc,\n\t\tactorId,\n\t}: GetForIdInput<{ Bindings: Bindings }>): Promise<ActorOutput | undefined> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\t// Get actor metadata from KV (combined name and key)\n\t\tconst actorData = (await env.ACTOR_KV.get(KEYS.ACTOR.metadata(actorId), {\n\t\t\ttype: \"json\",\n\t\t})) as ActorData | null;\n\n\t\t// If the actor doesn't exist, return undefined\n\t\tif (!actorData) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\treturn {\n\t\t\tactorId,\n\t\t\tname: actorData.name,\n\t\t\tkey: actorData.key,\n\t\t};\n\t}\n\n\tasync getWithKey({\n\t\tc,\n\t\tname,\n\t\tkey,\n\t}: GetWithKeyInput<{ Bindings: Bindings }>): Promise<\n\t\tActorOutput | undefined\n\t> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\tlogger().debug(\"getWithKey: searching for actor\", { name, key });\n\n\t\t// Generate deterministic ID from the name and key\n\t\t// This is aligned with how createActor generates IDs\n\t\tconst nameKeyString = serializeNameAndKey(name, key);\n\t\tconst actorId = env.ACTOR_DO.idFromName(nameKeyString).toString();\n\n\t\t// Check if the actor metadata exists\n\t\tconst actorData = await env.ACTOR_KV.get(KEYS.ACTOR.metadata(actorId), {\n\t\t\ttype: \"json\",\n\t\t});\n\n\t\tif (!actorData) {\n\t\t\tlogger().debug(\"getWithKey: no actor found with matching name and key\", {\n\t\t\t\tname,\n\t\t\t\tkey,\n\t\t\t\tactorId,\n\t\t\t});\n\t\t\treturn undefined;\n\t\t}\n\n\t\tlogger().debug(\"getWithKey: found actor with matching name and key\", {\n\t\t\tactorId,\n\t\t\tname,\n\t\t\tkey,\n\t\t});\n\t\treturn this.#buildActorOutput(c, actorId);\n\t}\n\n\tasync getOrCreateWithKey(\n\t\tinput: GetOrCreateWithKeyInput,\n\t): Promise<ActorOutput> {\n\t\t// TODO: Prevent race condition here\n\t\tconst getOutput = await this.getWithKey(input);\n\t\tif (getOutput) {\n\t\t\treturn getOutput;\n\t\t} else {\n\t\t\treturn await this.createActor(input);\n\t\t}\n\t}\n\n\tasync createActor({\n\t\tc,\n\t\tname,\n\t\tkey,\n\t\tinput,\n\t}: CreateInput<{ Bindings: Bindings }>): Promise<ActorOutput> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\t// Check if actor with the same name and key already exists\n\t\tconst existingActor = await this.getWithKey({ c, name, key });\n\t\tif (existingActor) {\n\t\t\tthrow new ActorAlreadyExists(name, key);\n\t\t}\n\n\t\t// Create a deterministic ID from the actor name and key\n\t\t// This ensures that actors with the same name and key will have the same ID\n\t\tconst nameKeyString = serializeNameAndKey(name, key);\n\t\tconst doId = env.ACTOR_DO.idFromName(nameKeyString);\n\t\tconst actorId = doId.toString();\n\n\t\t// Init actor\n\t\tconst actor = env.ACTOR_DO.get(doId);\n\t\tawait actor.initialize({\n\t\t\tname,\n\t\t\tkey,\n\t\t\tinput,\n\t\t});\n\n\t\t// Store combined actor metadata (name and key)\n\t\tconst actorData: ActorData = { name, key };\n\t\tawait env.ACTOR_KV.put(\n\t\t\tKEYS.ACTOR.metadata(actorId),\n\t\t\tJSON.stringify(actorData),\n\t\t);\n\n\t\t// Add to key index for lookups by name and key\n\t\tawait env.ACTOR_KV.put(KEYS.ACTOR.keyIndex(name, key), actorId);\n\n\t\treturn {\n\t\t\tactorId,\n\t\t\tname,\n\t\t\tkey,\n\t\t};\n\t}\n\n\t// Helper method to build actor output from an ID\n\tasync #buildActorOutput(\n\t\tc: any,\n\t\tactorId: string,\n\t): Promise<ActorOutput | undefined> {\n\t\tconst env = getCloudflareAmbientEnv();\n\n\t\tconst actorData = (await env.ACTOR_KV.get(KEYS.ACTOR.metadata(actorId), {\n\t\t\ttype: \"json\",\n\t\t})) as ActorData | null;\n\n\t\tif (!actorData) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\treturn {\n\t\t\tactorId,\n\t\t\tname: actorData.name,\n\t\t\tkey: actorData.key,\n\t\t};\n\t}\n}\n","// Constants for key handling\nexport const EMPTY_KEY = \"(none)\";\nexport const KEY_SEPARATOR = \",\";\n\n/**\n * Serializes an array of key strings into a single string for use with idFromName\n *\n * @param name The actor name\n * @param key Array of key strings to serialize\n * @returns A single string containing the serialized name and key\n */\nexport function serializeNameAndKey(name: string, key: string[]): string {\n\t// Escape colons in the name\n\tconst escapedName = name.replace(/:/g, \"\\\\:\");\n\n\t// For empty keys, just return the name and a marker\n\tif (key.length === 0) {\n\t\treturn `${escapedName}:${EMPTY_KEY}`;\n\t}\n\n\t// Serialize the key array\n\tconst serializedKey = serializeKey(key);\n\n\t// Combine name and serialized key\n\treturn `${escapedName}:${serializedKey}`;\n}\n\n/**\n * Serializes an array of key strings into a single string\n *\n * @param key Array of key strings to serialize\n * @returns A single string containing the serialized key\n */\nexport function serializeKey(key: string[]): string {\n\t// Use a special marker for empty key arrays\n\tif (key.length === 0) {\n\t\treturn EMPTY_KEY;\n\t}\n\n\t// Escape each key part to handle the separator and the empty key marker\n\tconst escapedParts = key.map((part) => {\n\t\t// First check if it matches our empty key marker\n\t\tif (part === EMPTY_KEY) {\n\t\t\treturn `\\\\${EMPTY_KEY}`;\n\t\t}\n\n\t\t// Escape backslashes first, then commas\n\t\tlet escaped = part.replace(/\\\\/g, \"\\\\\\\\\");\n\t\tescaped = escaped.replace(/,/g, \"\\\\,\");\n\t\treturn escaped;\n\t});\n\n\treturn escapedParts.join(KEY_SEPARATOR);\n}\n\n/**\n * Deserializes a key string back into an array of key strings\n *\n * @param keyString The serialized key string\n * @returns Array of key strings\n */\nexport function deserializeKey(keyString: string): string[] {\n\t// Handle empty values\n\tif (!keyString) {\n\t\treturn [];\n\t}\n\n\t// Check for special empty key marker\n\tif (keyString === EMPTY_KEY) {\n\t\treturn [];\n\t}\n\n\t// Split by unescaped commas and unescape the escaped characters\n\tconst parts: string[] = [];\n\tlet currentPart = \"\";\n\tlet escaping = false;\n\n\tfor (let i = 0; i < keyString.length; i++) {\n\t\tconst char = keyString[i];\n\n\t\tif (escaping) {\n\t\t\t// This is an escaped character, add it directly\n\t\t\tcurrentPart += char;\n\t\t\tescaping = false;\n\t\t} else if (char === \"\\\\\") {\n\t\t\t// Start of an escape sequence\n\t\t\tescaping = true;\n\t\t} else if (char === KEY_SEPARATOR) {\n\t\t\t// This is a separator\n\t\t\tparts.push(currentPart);\n\t\t\tcurrentPart = \"\";\n\t\t} else {\n\t\t\t// Regular character\n\t\t\tcurrentPart += char;\n\t\t}\n\t}\n\n\t// Add the last part if it exists\n\tif (currentPart || parts.length > 0) {\n\t\tparts.push(currentPart);\n\t}\n\n\treturn parts;\n}\n","// Modified from https://github.com/honojs/hono/blob/40ea0eee58e39b31053a0246c595434f1094ad31/src/adapter/cloudflare-workers/websocket.ts#L17\n//\n// This version calls the open event by default\n\nimport type { UpgradeWebSocket, WSEvents, WSReadyState } from \"hono/ws\";\nimport { defineWebSocketHelper, WSContext } from \"hono/ws\";\n\n// Based on https://github.com/honojs/hono/issues/1153#issuecomment-1767321332\nexport const upgradeWebSocket: UpgradeWebSocket<\n\tWebSocket,\n\t// eslint-disable-next-line @typescript-eslint/no-explicit-any\n\tany,\n\tWSEvents<WebSocket>\n> = defineWebSocketHelper(async (c, events) => {\n\tconst upgradeHeader = c.req.header(\"Upgrade\");\n\tif (upgradeHeader !== \"websocket\") {\n\t\treturn;\n\t}\n\n\tconst webSocketPair = new WebSocketPair();\n\tconst client: WebSocket = webSocketPair[0];\n\tconst server: WebSocket = webSocketPair[1];\n\n\tconst wsContext = new WSContext<WebSocket>({\n\t\tclose: (code, reason) => server.close(code, reason),\n\t\tget protocol() {\n\t\t\treturn server.protocol;\n\t\t},\n\t\traw: server,\n\t\tget readyState() {\n\t\t\treturn server.readyState as WSReadyState;\n\t\t},\n\t\turl: server.url ? new URL(server.url) : null,\n\t\tsend: (source) => server.send(source),\n\t});\n\n\tif (events.onClose) {\n\t\tserver.addEventListener(\"close\", (evt: CloseEvent) =>\n\t\t\tevents.onClose?.(evt, wsContext),\n\t\t);\n\t}\n\tif (events.onMessage) {\n\t\tserver.addEventListener(\"message\", (evt: MessageEvent) =>\n\t\t\tevents.onMessage?.(evt, wsContext),\n\t\t);\n\t}\n\tif (events.onError) {\n\t\tserver.addEventListener(\"error\", (evt: Event) =>\n\t\t\tevents.onError?.(evt, wsContext),\n\t\t);\n\t}\n\n\tserver.accept?.();\n\n\t// note: cloudflare actors doesn't support 'open' event, so we call it immediately with a fake event\n\t//\n\t// we have to do this after `server.accept() is called`\n\tevents.onOpen?.(new Event(\"open\"), wsContext);\n\n\treturn new Response(null, {\n\t\tstatus: 101,\n\t\theaders: {\n\t\t\t// HACK: Required in order for Cloudflare to not error with \"Network connection lost\"\n\t\t\t//\n\t\t\t// This bug undocumented. Cannot easily reproduce outside of RivetKit.\n\t\t\t\"Sec-WebSocket-Protocol\": \"rivetkit\",\n\t\t},\n\t\twebSocket: client,\n\t});\n});\n"],"mappings":";AAAA,SAAS,yBAAyB;AAGlC,SAAS,YAAY;AACrB,OAAOA,gBAAe;;;ACJtB,SAAS,qBAAqB;AAO9B;AAAA,EACC;AAAA,EACA;AAAA,EACA;AAAA,OACM;AACP,SAAS,iCAAiC;;;ACP1C;AAAA,EACC;AAAA,EACA;AAAA,EACA;AAAA,OACM;AAOP,OAAO,eAAe;AAaf,IAAM,qCAAN,MAAyC;AAAA;AAAA,EAE/C,OAA8C,oBAAI,IAAI;AAAA,EAEtD,WAAW,SAA2C;AACrD,UAAM,QAAQ,KAAK,KAAK,IAAI,OAAO;AACnC,cAAU,UAAU,QAAW,0CAA0C;AACzE,WAAO;AAAA,EACR;AAAA,EAEA,WAAW,SAAiB,OAAiC;AAC5D,SAAK,KAAK,IAAI,SAAS,KAAK;AAAA,EAC7B;AACD;AAQA,IAAM,eAAN,MAAmB;AAAA,EAClB;AAAA,EACA,eAA4C,QAAQ,cAAc;AAAA,EAClE,yBAAyB,IAAI,uBAAuB;AACrD;AAEO,IAAM,8BAAN,MAAyD;AAAA,EAC/D;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,UAAqC,oBAAI,IAAI;AAAA,EAE7C,YACC,gBACA,WACA,eACA,cACA,aACC;AACD,SAAK,kBAAkB;AACvB,SAAK,aAAa;AAClB,SAAK,iBAAiB;AACtB,SAAK,gBAAgB;AACrB,SAAK,eAAe;AAAA,EACrB;AAAA,EAEA,UAAU,SAAiB;AAC1B,WAAO,KAAK,aAAa,WAAW,OAAO,EAAE;AAAA,EAC9C;AAAA,EAEA,MAAM,UAAU,SAA4C;AAlF7D;AAoFE,QAAI,UAAU,KAAK,QAAQ,IAAI,OAAO;AACtC,QAAI,SAAS;AACZ,UAAI,QAAQ,aAAc,OAAM,QAAQ,aAAa;AACrD,UAAI,CAAC,QAAQ,MAAO,OAAM,IAAI,MAAM,wBAAwB;AAC5D,aAAO,QAAQ;AAAA,IAChB;AAGA,cAAU,IAAI,aAAa;AAC3B,SAAK,QAAQ,IAAI,SAAS,OAAO;AAGjC,UAAM,UAAU,KAAK,aAAa,WAAW,OAAO;AACpD,UAAM,UAAU,QAAQ,IAAI;AAG5B,UAAM,CAAC,MAAM,GAAG,IAAI,MAAM,QAAQ,IAAI;AAAA,MACrC,QAAQ,IAAY,KAAK,IAAI;AAAA,MAC7B,QAAQ,IAAc,KAAK,GAAG;AAAA,IAC/B,CAAC;AAED,QAAI,CAAC,MAAM;AACV,YAAM,IAAI,MAAM,SAAS,OAAO,oCAAoC;AAAA,IACrE;AACA,QAAI,CAAC,KAAK;AACT,YAAM,IAAI,MAAM,SAAS,OAAO,mCAAmC;AAAA,IACpE;AAGA,UAAM,aAAa,iBAAiB,KAAK,iBAAiB,IAAI;AAC9D,YAAQ,QAAQ,WAAW,YAAY;AAGvC,UAAM,cAAc;AAAA,MACnB,QAAQ;AAAA,IACT;AACA,UAAM,QAAQ,MAAM;AAAA,MACnB;AAAA,MACA;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA;AAAA,IACD;AAGA,kBAAQ,iBAAR,mBAAsB;AACtB,YAAQ,eAAe;AAEvB,WAAO,QAAQ;AAAA,EAChB;AAAA,EAEA,0BAA0B,SAAyC;AAClE,UAAM,UAAU,KAAK,QAAQ,IAAI,OAAO;AACxC,QAAI,CAAC,SAAS;AACb,YAAM,IAAI,MAAM,SAAS,OAAO,aAAa;AAAA,IAC9C;AACA,WAAO,QAAQ;AAAA,EAChB;AAAA,EAEA,WAAW,SAAqC;AAC/C,UAAM,QAAQ,KAAK,aAAa,WAAW,OAAO;AAClD,WAAO,EAAE,KAAK,MAAM,KAAK,KAAK,MAAM,IAAI;AAAA,EACzC;AAAA,EAEA,MAAM,kBAAkB,SAAkD;AACzE,WAAO,MAAM,KAAK,UAAU,OAAO,EAAE,QAAQ,IAAI,KAAK,YAAY;AAAA,EACnE;AAAA,EAEA,MAAM,mBAAmB,SAAiB,MAAiC;AAC1E,UAAM,KAAK,UAAU,OAAO,EAAE,QAAQ,IAAI,KAAK,cAAc,IAAI;AAAA,EAClE;AAAA,EAEA,MAAM,SAAS,OAAyB,WAAkC;AACzE,UAAM,KAAK,UAAU,MAAM,EAAE,EAAE,QAAQ,SAAS,SAAS;AAAA,EAC1D;AAAA,EAEA,MAAM,YAAY,SAA+C;AAChE,WAAO,KAAK,UAAU,OAAO,EAAE,QAAQ;AAAA,EACxC;AACD;AAEO,SAAS,yCACf,aACC;AACD,SAAO,CACN,gBACA,WACA,eACA,iBACI;AACJ,WAAO,IAAI;AAAA,MACV;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACD;AAAA,EACD;AACD;;;ACxLA,SAAS,iBAAiB;AAEnB,IAAM,cAAc;AAEpB,SAAS,SAAS;AACxB,SAAO,UAAU,WAAW;AAC7B;;;AFeO,IAAM,OAAO;AAAA,EACnB,MAAM;AAAA,EACN,KAAK;AAAA,EACL,cAAc;AACf;AAyBO,SAAS,yBACf,UACA,WAC2B;AAC3B,QAAM,cAAc,IAAI,mCAAmC;AAQ3D,SAAO,MAAM,qBACJ,cAET;AAAA,IACC;AAAA,IACA;AAAA,IAEA;AAAA,IAEA,MAAM,aAAmC;AAIxC,UAAI,CAAC,KAAK,cAAc;AAEvB,YAAI,KAAK,qBAAqB;AAC7B,gBAAM,KAAK,oBAAoB;AAAA,QAChC,OAAO;AACN,eAAK,sBAAsB,QAAQ,cAAc;AACjD,gBAAM,MAAM,MAAM,KAAK,IAAI,QAAQ,IAAI;AAAA,YACtC,KAAK;AAAA,YACL,KAAK;AAAA,YACL,KAAK;AAAA,UACN,CAAC;AACD,cAAI,IAAI,IAAI,KAAK,YAAY,GAAG;AAC/B,kBAAM,OAAO,IAAI,IAAI,KAAK,IAAI;AAC9B,gBAAI,CAAC,KAAM,OAAM,IAAI,MAAM,oBAAoB;AAC/C,kBAAM,MAAM,IAAI,IAAI,KAAK,GAAG;AAC5B,gBAAI,CAAC,IAAK,OAAM,IAAI,MAAM,mBAAmB;AAE7C,mBAAO,EAAE,MAAM,uBAAuB,EAAE,MAAM,IAAI,CAAC;AAEnD,iBAAK,eAAe,EAAE,MAAM,IAAI;AAChC,iBAAK,oBAAoB,QAAQ;AAAA,UAClC,OAAO;AACN,mBAAO,EAAE,MAAM,uBAAuB;AAAA,UACvC;AAAA,QACD;AAAA,MACD;AAGA,UAAI,KAAK,QAAQ;AAChB,eAAO,KAAK;AAAA,MACb;AAEA,UAAI,CAAC,KAAK,aAAc,OAAM,IAAI,MAAM,iBAAiB;AAMzD,YAAM,UAAU,KAAK,IAAI,GAAG,SAAS;AACrC,kBAAY,WAAW,SAAS,EAAE,KAAK,KAAK,KAAK,KAAK,KAAK,IAAI,CAAC;AAGhE,gBAAU,OAAO,QAChB,yCAAyC,WAAW;AAGrD,YAAM,gBAAgB,UAAU,OAAO;AAAA,QACtC,SAAS;AAAA,QACT;AAAA,MACD;AAGA,YAAM,eAAe;AAAA,QACpB,yBAAyB,aAAa;AAAA,MACvC;AAGA,YAAM,cAAc,UAAU,OAAO;AAAA,QACpC,SAAS;AAAA,QACT;AAAA,QACA;AAAA,QACA;AAAA,MACD;AAGA,YAAM,cAAc,kBAAkB,WAAW,WAAW;AAG5D,WAAK,SAAS;AAAA,QACb;AAAA,MACD;AAIA,YAAM,YAAY,UAAU,OAAO;AAEnC,aAAO,KAAK;AAAA,IACb;AAAA;AAAA,IAGA,MAAM,WAAW,KAAuB;AAGvC,aAAO,MAAM,eAAe,IAAI,KAAK,KAAK,YAAY;AACrD,cAAM,KAAK,IAAI,QAAQ,IAAI;AAAA,UAC1B,CAAC,KAAK,IAAI,GAAG,IAAI;AAAA,UACjB,CAAC,KAAK,GAAG,GAAG,IAAI;AAAA,UAChB,CAAC,KAAK,YAAY,GAAG,0BAA0B,IAAI,KAAK;AAAA,QACzD,CAAC;AACD,aAAK,eAAe;AAAA,UACnB,MAAM,IAAI;AAAA,UACV,KAAK,IAAI;AAAA,QACV;AAEA,eAAO,EAAE,MAAM,qBAAqB,EAAE,KAAK,IAAI,IAAI,CAAC;AAGpD,cAAM,KAAK,WAAW;AAAA,MACvB,CAAC;AAAA,IACF;AAAA,IAEA,MAAM,MAAM,SAAqC;AAChD,aAAO,MAAM,eAAe,IAAI,KAAK,KAAK,YAAY;AACrD,cAAM,EAAE,YAAY,IAAI,MAAM,KAAK,WAAW;AAE9C,cAAM,UAAU,KAAK,IAAI,GAAG,SAAS;AACrC,eAAO,MAAM,YAAY,MAAM,SAAS;AAAA,UACvC;AAAA,QACD,CAAC;AAAA,MACF,CAAC;AAAA,IACF;AAAA,IAEA,MAAM,QAAuB;AAC5B,aAAO,MAAM,eAAe,IAAI,KAAK,KAAK,YAAY;AACrD,cAAM,KAAK,WAAW;AACtB,cAAM,UAAU,KAAK,IAAI,GAAG,SAAS;AAGrC,cAAM,gBAAgB,UAAU,OAAO;AAAA,UACtC,SAAS;AAAA,UACT;AAAA,QACD;AACA,cAAM,eAAe;AAAA,UACpB,yBAAyB,aAAa;AAAA,QACvC;AACA,cAAM,cAAc,UAAU,OAAO;AAAA,UACpC,SAAS;AAAA,UACT;AAAA,UACA;AAAA,UACA;AAAA,QACD;AAGA,cAAM,QAAQ,MAAM,YAAY,UAAU,OAAO;AACjD,cAAM,MAAM,QAAQ;AAAA,MACrB,CAAC;AAAA,IACF;AAAA,EACD;AACD;;;AGrNA,SAAS,uBAAuB;AAEhC,SAAS,SAAS;AAEX,IAAM,eAAe,gBAAgB,cAAc,EACxD,KAAK,EAAE,QAAQ,MAAM,qBAAqB,KAAK,CAAC,EAChD,OAAO;AAAA,EACP,KAAK,EAAE,OAAa,EAAE,SAAS;AAChC,CAAC,EACA,QAAQ,CAAC,CAAC;;;ACRZ;AAAA,EAMC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEM;AACP,SAAS,oBAAoB,qBAAqB;;;ACZ3C,IAAM,YAAY;AAClB,IAAM,gBAAgB;AAStB,SAAS,oBAAoB,MAAc,KAAuB;AAExE,QAAM,cAAc,KAAK,QAAQ,MAAM,KAAK;AAG5C,MAAI,IAAI,WAAW,GAAG;AACrB,WAAO,GAAG,WAAW,IAAI,SAAS;AAAA,EACnC;AAGA,QAAM,gBAAgB,aAAa,GAAG;AAGtC,SAAO,GAAG,WAAW,IAAI,aAAa;AACvC;AAQO,SAAS,aAAa,KAAuB;AAEnD,MAAI,IAAI,WAAW,GAAG;AACrB,WAAO;AAAA,EACR;AAGA,QAAM,eAAe,IAAI,IAAI,CAAC,SAAS;AAEtC,QAAI,SAAS,WAAW;AACvB,aAAO,KAAK,SAAS;AAAA,IACtB;AAGA,QAAI,UAAU,KAAK,QAAQ,OAAO,MAAM;AACxC,cAAU,QAAQ,QAAQ,MAAM,KAAK;AACrC,WAAO;AAAA,EACR,CAAC;AAED,SAAO,aAAa,KAAK,aAAa;AACvC;;;AD1BA,IAAMC,QAAO;AAAA,EACZ,OAAO;AAAA;AAAA,IAEN,UAAU,CAAC,YAAoB,SAAS,OAAO;AAAA;AAAA,IAG/C,UAAU,CAAC,MAAc,MAAgB,CAAC,MAAM;AAE/C,aAAO,aAAa,aAAa,GAAG,CAAC;AAAA,IACtC;AAAA,EACD;AACD;AAEA,IAAM,6BAA6B;AAAA,EAClC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACD;AAEO,IAAM,gCAAN,MAA6D;AAAA,EACnE,MAAM,YAAY,SAAiB,cAA0C;AAC5E,UAAM,MAAM,wBAAwB;AAEpC,WAAO,EAAE,MAAM,qCAAqC;AAAA,MACnD;AAAA,MACA,QAAQ,aAAa;AAAA,MACrB,KAAK,aAAa;AAAA,IACnB,CAAC;AAED,UAAM,KAAK,IAAI,SAAS,aAAa,OAAO;AAC5C,UAAM,OAAO,IAAI,SAAS,IAAI,EAAE;AAEhC,WAAO,MAAM,KAAK,MAAM,YAAY;AAAA,EACrC;AAAA,EAEA,MAAM,cACL,MACA,SACA,UACA,QACqB;AACrB,UAAM,MAAM,wBAAwB;AAEpC,WAAO,EAAE,MAAM,uCAAuC,EAAE,SAAS,KAAK,CAAC;AAGvE,UAAM,KAAK,IAAI,SAAS,aAAa,OAAO;AAC5C,UAAM,OAAO,IAAI,SAAS,IAAI,EAAE;AAEhC,UAAM,UAAkC;AAAA,MACvC,SAAS;AAAA,MACT,YAAY;AAAA,MACZ,CAAC,4BAA4B,GAAG;AAAA,MAChC,CAAC,eAAe,GAAG;AAAA,IACpB;AACA,QAAI,QAAQ;AACX,cAAQ,kBAAkB,IAAI,KAAK,UAAU,MAAM;AAAA,IACpD;AAEA,YAAQ,wBAAwB,IAAI;AAGpC,UAAM,MAAM,eAAe,IAAI;AAE/B,WAAO,EAAE,MAAM,2BAA2B;AAAA,MACzC,MAAM;AAAA,MACN,IAAI;AAAA,IACL,CAAC;AAED,UAAM,WAAW,MAAM,KAAK,MAAM,KAAK;AAAA,MACtC;AAAA,IACD,CAAC;AACD,UAAM,YAAY,SAAS;AAE3B,QAAI,CAAC,WAAW;AACf,YAAM,IAAI;AAAA,QACT;AAAA,MACD;AAAA,IACD;AAEA,WAAO,EAAE,MAAM,4CAA4C;AAAA,MAC1D;AAAA,IACD,CAAC;AAED,cAAU,OAAO;AAKjB,eAAW,MAAM;AAvHnB;AAwHG,YAAM,QAAQ,IAAI,MAAM,MAAM;AAC9B,OAAC,eAAkB,WAAlB,mCAA2B;AAC5B,MAAC,UAAkB,cAAc,KAAK;AAAA,IACvC,GAAG,CAAC;AAEJ,WAAO;AAAA,EACR;AAAA,EAEA,MAAM,aACL,GACA,cACA,SACoB;AACpB,WAAO,EAAE,MAAM,wCAAwC;AAAA,MACtD;AAAA,MACA,QAAQ,aAAa;AAAA,MACrB,KAAK,aAAa;AAAA,IACnB,CAAC;AAED,UAAM,KAAK,EAAE,IAAI,SAAS,aAAa,OAAO;AAC9C,UAAM,OAAO,EAAE,IAAI,SAAS,IAAI,EAAE;AAElC,WAAO,MAAM,KAAK,MAAM,YAAY;AAAA,EACrC;AAAA,EAEA,MAAM,eACL,GACA,MACA,SACA,UACA,QACA,UACoB;AACpB,WAAO,EAAE,MAAM,0CAA0C;AAAA,MACxD;AAAA,MACA;AAAA,IACD,CAAC;AAGD,UAAM,gBAAgB,EAAE,IAAI,OAAO,SAAS;AAC5C,QAAI,CAAC,iBAAiB,kBAAkB,aAAa;AACpD,aAAO,IAAI,SAAS,+BAA+B;AAAA,QAClD,QAAQ;AAAA,MACT,CAAC;AAAA,IACF;AAGA,UAAM,SAAS,IAAI,IAAI,eAAe,IAAI,EAAE;AAC5C,UAAM,eAAe,IAAI,QAAQ,QAAQ,EAAE,IAAI,GAAG;AAElD,WAAO,EAAE,MAAM,2BAA2B;AAAA,MACzC,MAAM,EAAE,IAAI;AAAA,MACZ,IAAI,aAAa;AAAA,IAClB,CAAC;AAKD,UAAM,aAAuB,CAAC;AAC9B,iBAAa,QAAQ,QAAQ,CAAC,GAAG,MAAM,WAAW,KAAK,CAAC,CAAC;AACzD,eAAW,KAAK,YAAY;AAC3B,UAAI,CAAC,2BAA2B,SAAS,CAAC,GAAG;AAC5C,qBAAa,QAAQ,OAAO,CAAC;AAAA,MAC9B;AAAA,IACD;AAGA,iBAAa,QAAQ,IAAI,8BAA8B,MAAM;AAC7D,iBAAa,QAAQ,IAAI,iBAAiB,QAAQ;AAClD,QAAI,QAAQ;AACX,mBAAa,QAAQ,IAAI,oBAAoB,KAAK,UAAU,MAAM,CAAC;AAAA,IACpE;AACA,QAAI,UAAU;AACb,mBAAa,QAAQ,IAAI,kBAAkB,KAAK,UAAU,QAAQ,CAAC;AAAA,IACpE;AAEA,UAAM,KAAK,EAAE,IAAI,SAAS,aAAa,OAAO;AAC9C,UAAM,OAAO,EAAE,IAAI,SAAS,IAAI,EAAE;AAElC,WAAO,MAAM,KAAK,MAAM,YAAY;AAAA,EACrC;AAAA,EAEA,MAAM,SAAS;AAAA,IACd;AAAA,IACA;AAAA,EACD,GAA4E;AAC3E,UAAM,MAAM,wBAAwB;AAGpC,UAAM,YAAa,MAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,OAAO,GAAG;AAAA,MACvE,MAAM;AAAA,IACP,CAAC;AAGD,QAAI,CAAC,WAAW;AACf,aAAO;AAAA,IACR;AAEA,WAAO;AAAA,MACN;AAAA,MACA,MAAM,UAAU;AAAA,MAChB,KAAK,UAAU;AAAA,IAChB;AAAA,EACD;AAAA,EAEA,MAAM,WAAW;AAAA,IAChB;AAAA,IACA;AAAA,IACA;AAAA,EACD,GAEE;AACD,UAAM,MAAM,wBAAwB;AAEpC,WAAO,EAAE,MAAM,mCAAmC,EAAE,MAAM,IAAI,CAAC;AAI/D,UAAM,gBAAgB,oBAAoB,MAAM,GAAG;AACnD,UAAM,UAAU,IAAI,SAAS,WAAW,aAAa,EAAE,SAAS;AAGhE,UAAM,YAAY,MAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,OAAO,GAAG;AAAA,MACtE,MAAM;AAAA,IACP,CAAC;AAED,QAAI,CAAC,WAAW;AACf,aAAO,EAAE,MAAM,yDAAyD;AAAA,QACvE;AAAA,QACA;AAAA,QACA;AAAA,MACD,CAAC;AACD,aAAO;AAAA,IACR;AAEA,WAAO,EAAE,MAAM,sDAAsD;AAAA,MACpE;AAAA,MACA;AAAA,MACA;AAAA,IACD,CAAC;AACD,WAAO,KAAK,kBAAkB,GAAG,OAAO;AAAA,EACzC;AAAA,EAEA,MAAM,mBACL,OACuB;AAEvB,UAAM,YAAY,MAAM,KAAK,WAAW,KAAK;AAC7C,QAAI,WAAW;AACd,aAAO;AAAA,IACR,OAAO;AACN,aAAO,MAAM,KAAK,YAAY,KAAK;AAAA,IACpC;AAAA,EACD;AAAA,EAEA,MAAM,YAAY;AAAA,IACjB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACD,GAA8D;AAC7D,UAAM,MAAM,wBAAwB;AAGpC,UAAM,gBAAgB,MAAM,KAAK,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC;AAC5D,QAAI,eAAe;AAClB,YAAM,IAAI,mBAAmB,MAAM,GAAG;AAAA,IACvC;AAIA,UAAM,gBAAgB,oBAAoB,MAAM,GAAG;AACnD,UAAM,OAAO,IAAI,SAAS,WAAW,aAAa;AAClD,UAAM,UAAU,KAAK,SAAS;AAG9B,UAAM,QAAQ,IAAI,SAAS,IAAI,IAAI;AACnC,UAAM,MAAM,WAAW;AAAA,MACtB;AAAA,MACA;AAAA,MACA;AAAA,IACD,CAAC;AAGD,UAAM,YAAuB,EAAE,MAAM,IAAI;AACzC,UAAM,IAAI,SAAS;AAAA,MAClBA,MAAK,MAAM,SAAS,OAAO;AAAA,MAC3B,KAAK,UAAU,SAAS;AAAA,IACzB;AAGA,UAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,MAAM,GAAG,GAAG,OAAO;AAE9D,WAAO;AAAA,MACN;AAAA,MACA;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAAA;AAAA,EAGA,MAAM,kBACL,GACA,SACmC;AACnC,UAAM,MAAM,wBAAwB;AAEpC,UAAM,YAAa,MAAM,IAAI,SAAS,IAAIA,MAAK,MAAM,SAAS,OAAO,GAAG;AAAA,MACvE,MAAM;AAAA,IACP,CAAC;AAED,QAAI,CAAC,WAAW;AACf,aAAO;AAAA,IACR;AAEA,WAAO;AAAA,MACN;AAAA,MACA,MAAM,UAAU;AAAA,MAChB,KAAK,UAAU;AAAA,IAChB;AAAA,EACD;AACD;;;AEhVA,SAAS,uBAAuB,iBAAiB;AAG1C,IAAM,mBAKT,sBAAsB,OAAO,GAAG,WAAW;AAb/C;AAcC,QAAM,gBAAgB,EAAE,IAAI,OAAO,SAAS;AAC5C,MAAI,kBAAkB,aAAa;AAClC;AAAA,EACD;AAEA,QAAM,gBAAgB,IAAI,cAAc;AACxC,QAAM,SAAoB,cAAc,CAAC;AACzC,QAAM,SAAoB,cAAc,CAAC;AAEzC,QAAM,YAAY,IAAI,UAAqB;AAAA,IAC1C,OAAO,CAAC,MAAM,WAAW,OAAO,MAAM,MAAM,MAAM;AAAA,IAClD,IAAI,WAAW;AACd,aAAO,OAAO;AAAA,IACf;AAAA,IACA,KAAK;AAAA,IACL,IAAI,aAAa;AAChB,aAAO,OAAO;AAAA,IACf;AAAA,IACA,KAAK,OAAO,MAAM,IAAI,IAAI,OAAO,GAAG,IAAI;AAAA,IACxC,MAAM,CAAC,WAAW,OAAO,KAAK,MAAM;AAAA,EACrC,CAAC;AAED,MAAI,OAAO,SAAS;AACnB,WAAO;AAAA,MAAiB;AAAA,MAAS,CAAC,QAAiB;AArCrD,YAAAC;AAsCG,gBAAAA,MAAA,OAAO,YAAP,gBAAAA,IAAA,aAAiB,KAAK;AAAA;AAAA,IACvB;AAAA,EACD;AACA,MAAI,OAAO,WAAW;AACrB,WAAO;AAAA,MAAiB;AAAA,MAAW,CAAC,QAAmB;AA1CzD,YAAAA;AA2CG,gBAAAA,MAAA,OAAO,cAAP,gBAAAA,IAAA,aAAmB,KAAK;AAAA;AAAA,IACzB;AAAA,EACD;AACA,MAAI,OAAO,SAAS;AACnB,WAAO;AAAA,MAAiB;AAAA,MAAS,CAAC,QAAY;AA/ChD,YAAAA;AAgDG,gBAAAA,MAAA,OAAO,YAAP,gBAAAA,IAAA,aAAiB,KAAK;AAAA;AAAA,IACvB;AAAA,EACD;AAEA,eAAO,WAAP;AAKA,eAAO,WAAP,gCAAgB,IAAI,MAAM,MAAM,GAAG;AAEnC,SAAO,IAAI,SAAS,MAAM;AAAA,IACzB,QAAQ;AAAA,IACR,SAAS;AAAA;AAAA;AAAA;AAAA,MAIR,0BAA0B;AAAA,IAC3B;AAAA,IACA,WAAW;AAAA,EACZ,CAAC;AACF,CAAC;;;AP5CM,IAAM,iBAAiB,IAAI,kBAA4B;AAEvD,SAAS,0BAAoC;AACnD,QAAM,MAAM,eAAe,SAAS;AACpC,EAAAC,WAAU,KAAK,wBAAwB;AACvC,SAAO;AACR;AAYO,SAAS,oBACf,UACA,aACU;AACV,QAAM,EAAE,cAAc,IAAI,aAAa,UAAU,WAAW;AAC5D,SAAO,cAAc;AACtB;AAEO,SAAS,aACf,UACA,aACiB;AACjB,QAAM,SAAS,aAAa,MAAM,WAAW;AAG7C,QAAM,YAAY;AAAA,IACjB,QAAQ;AAAA,MACP,SAAS,MAAM,IAAI,8BAA8B;AAAA;AAAA,MAEjD,OAAO;AAAA,IACR;AAAA,IACA,qBAAqB,MAAM;AAAA,IAC3B,GAAG;AAAA,EACJ;AAGA,QAAMC,gBAAe,yBAAyB,UAAU,SAAS;AAGjE,QAAM,eAAe,SAAS,aAAa,SAAS;AAEpD,SAAO;AAAA,IACN,QAAQ,aAAa;AAAA,IACrB,eAAe,CAAC,SAAS;AAExB,YAAM,MAAM,QAAQ,IAAI,KAAK;AAG7B,UAAI,CAAC,MAAM;AACV,YAAI,MAAM,aAAa,aAAa,IAAI;AAAA,MACzC;AAGA,YAAM,UAAU;AAAA,QACf,OAAO,CAAC,SAAS,KAAK,QAAQ;AAC7B,iBAAO,eAAe,IAAI,KAAK,MAAM,IAAI,MAAM,SAAS,KAAK,GAAG,CAAC;AAAA,QAClE;AAAA,MACD;AAEA,aAAO,EAAE,SAAS,cAAAA,cAAa;AAAA,IAChC;AAAA,EACD;AACD;","names":["invariant","KEYS","_a","invariant","ActorHandler"]}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@rivetkit/cloudflare-workers",
|
|
3
|
-
"version": "0.9.
|
|
3
|
+
"version": "0.9.2",
|
|
4
4
|
"keywords": [
|
|
5
5
|
"rivetkit",
|
|
6
6
|
"cloudflare",
|
|
@@ -37,15 +37,16 @@
|
|
|
37
37
|
"@cloudflare/workers-types": "^4.20250129.0",
|
|
38
38
|
"@types/invariant": "^2",
|
|
39
39
|
"@types/node": "^24.0.3",
|
|
40
|
+
"hono": "4.8.3",
|
|
40
41
|
"tsup": "^8.4.0",
|
|
41
42
|
"typescript": "^5.5.2",
|
|
42
43
|
"vitest": "^3.1.1",
|
|
43
44
|
"wrangler": "^4.22.0",
|
|
44
|
-
"@rivetkit/core": "0.9.
|
|
45
|
+
"@rivetkit/core": "0.9.2"
|
|
45
46
|
},
|
|
46
47
|
"dependencies": {
|
|
47
48
|
"invariant": "^2.2.4",
|
|
48
|
-
"zod": "^3.25.
|
|
49
|
+
"zod": "^3.25.76"
|
|
49
50
|
},
|
|
50
51
|
"stableVersion": "0.8.0",
|
|
51
52
|
"scripts": {
|
package/src/actor-driver.ts
CHANGED
|
@@ -1,6 +1,18 @@
|
|
|
1
|
+
import type {
|
|
2
|
+
AnyActorInstance as CoreAnyActorInstance,
|
|
3
|
+
RegistryConfig,
|
|
4
|
+
RunConfig,
|
|
5
|
+
} from "@rivetkit/core";
|
|
6
|
+
import {
|
|
7
|
+
createGenericConnDrivers,
|
|
8
|
+
GenericConnGlobalState,
|
|
9
|
+
lookupInRegistry,
|
|
10
|
+
} from "@rivetkit/core";
|
|
11
|
+
import type { Client } from "@rivetkit/core/client";
|
|
1
12
|
import type {
|
|
2
13
|
ActorDriver,
|
|
3
14
|
AnyActorInstance,
|
|
15
|
+
ManagerDriver,
|
|
4
16
|
} from "@rivetkit/core/driver-helpers";
|
|
5
17
|
import invariant from "invariant";
|
|
6
18
|
import { KEYS } from "./actor-handler-do";
|
|
@@ -35,10 +47,32 @@ export interface ActorDriverContext {
|
|
|
35
47
|
env: unknown;
|
|
36
48
|
}
|
|
37
49
|
|
|
50
|
+
// Actor handler to track running instances
|
|
51
|
+
class ActorHandler {
|
|
52
|
+
actor?: AnyActorInstance;
|
|
53
|
+
actorPromise?: PromiseWithResolvers<void> = Promise.withResolvers();
|
|
54
|
+
genericConnGlobalState = new GenericConnGlobalState();
|
|
55
|
+
}
|
|
56
|
+
|
|
38
57
|
export class CloudflareActorsActorDriver implements ActorDriver {
|
|
58
|
+
#registryConfig: RegistryConfig;
|
|
59
|
+
#runConfig: RunConfig;
|
|
60
|
+
#managerDriver: ManagerDriver;
|
|
61
|
+
#inlineClient: Client<any>;
|
|
39
62
|
#globalState: CloudflareDurableObjectGlobalState;
|
|
63
|
+
#actors: Map<string, ActorHandler> = new Map();
|
|
40
64
|
|
|
41
|
-
constructor(
|
|
65
|
+
constructor(
|
|
66
|
+
registryConfig: RegistryConfig,
|
|
67
|
+
runConfig: RunConfig,
|
|
68
|
+
managerDriver: ManagerDriver,
|
|
69
|
+
inlineClient: Client<any>,
|
|
70
|
+
globalState: CloudflareDurableObjectGlobalState,
|
|
71
|
+
) {
|
|
72
|
+
this.#registryConfig = registryConfig;
|
|
73
|
+
this.#runConfig = runConfig;
|
|
74
|
+
this.#managerDriver = managerDriver;
|
|
75
|
+
this.#inlineClient = inlineClient;
|
|
42
76
|
this.#globalState = globalState;
|
|
43
77
|
}
|
|
44
78
|
|
|
@@ -46,6 +80,69 @@ export class CloudflareActorsActorDriver implements ActorDriver {
|
|
|
46
80
|
return this.#globalState.getDOState(actorId).ctx;
|
|
47
81
|
}
|
|
48
82
|
|
|
83
|
+
async loadActor(actorId: string): Promise<AnyActorInstance> {
|
|
84
|
+
// Check if actor is already loaded
|
|
85
|
+
let handler = this.#actors.get(actorId);
|
|
86
|
+
if (handler) {
|
|
87
|
+
if (handler.actorPromise) await handler.actorPromise.promise;
|
|
88
|
+
if (!handler.actor) throw new Error("Actor should be loaded");
|
|
89
|
+
return handler.actor;
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
// Create new actor handler
|
|
93
|
+
handler = new ActorHandler();
|
|
94
|
+
this.#actors.set(actorId, handler);
|
|
95
|
+
|
|
96
|
+
// Get the actor metadata from Durable Object storage
|
|
97
|
+
const doState = this.#globalState.getDOState(actorId);
|
|
98
|
+
const storage = doState.ctx.storage;
|
|
99
|
+
|
|
100
|
+
// Load actor metadata
|
|
101
|
+
const [name, key] = await Promise.all([
|
|
102
|
+
storage.get<string>(KEYS.NAME),
|
|
103
|
+
storage.get<string[]>(KEYS.KEY),
|
|
104
|
+
]);
|
|
105
|
+
|
|
106
|
+
if (!name) {
|
|
107
|
+
throw new Error(`Actor ${actorId} is not initialized - missing name`);
|
|
108
|
+
}
|
|
109
|
+
if (!key) {
|
|
110
|
+
throw new Error(`Actor ${actorId} is not initialized - missing key`);
|
|
111
|
+
}
|
|
112
|
+
|
|
113
|
+
// Create actor instance
|
|
114
|
+
const definition = lookupInRegistry(this.#registryConfig, name);
|
|
115
|
+
handler.actor = definition.instantiate();
|
|
116
|
+
|
|
117
|
+
// Start actor
|
|
118
|
+
const connDrivers = createGenericConnDrivers(
|
|
119
|
+
handler.genericConnGlobalState,
|
|
120
|
+
);
|
|
121
|
+
await handler.actor.start(
|
|
122
|
+
connDrivers,
|
|
123
|
+
this,
|
|
124
|
+
this.#inlineClient,
|
|
125
|
+
actorId,
|
|
126
|
+
name,
|
|
127
|
+
key,
|
|
128
|
+
"unknown", // TODO: Support regions in Cloudflare
|
|
129
|
+
);
|
|
130
|
+
|
|
131
|
+
// Finish
|
|
132
|
+
handler.actorPromise?.resolve();
|
|
133
|
+
handler.actorPromise = undefined;
|
|
134
|
+
|
|
135
|
+
return handler.actor;
|
|
136
|
+
}
|
|
137
|
+
|
|
138
|
+
getGenericConnGlobalState(actorId: string): GenericConnGlobalState {
|
|
139
|
+
const handler = this.#actors.get(actorId);
|
|
140
|
+
if (!handler) {
|
|
141
|
+
throw new Error(`Actor ${actorId} not loaded`);
|
|
142
|
+
}
|
|
143
|
+
return handler.genericConnGlobalState;
|
|
144
|
+
}
|
|
145
|
+
|
|
49
146
|
getContext(actorId: string): ActorDriverContext {
|
|
50
147
|
const state = this.#globalState.getDOState(actorId);
|
|
51
148
|
return { ctx: state.ctx, env: state.env };
|
|
@@ -67,3 +164,22 @@ export class CloudflareActorsActorDriver implements ActorDriver {
|
|
|
67
164
|
return this.#getDOCtx(actorId).storage.sql;
|
|
68
165
|
}
|
|
69
166
|
}
|
|
167
|
+
|
|
168
|
+
export function createCloudflareActorsActorDriverBuilder(
|
|
169
|
+
globalState: CloudflareDurableObjectGlobalState,
|
|
170
|
+
) {
|
|
171
|
+
return (
|
|
172
|
+
registryConfig: RegistryConfig,
|
|
173
|
+
runConfig: RunConfig,
|
|
174
|
+
managerDriver: ManagerDriver,
|
|
175
|
+
inlineClient: Client<any>,
|
|
176
|
+
) => {
|
|
177
|
+
return new CloudflareActorsActorDriver(
|
|
178
|
+
registryConfig,
|
|
179
|
+
runConfig,
|
|
180
|
+
managerDriver,
|
|
181
|
+
inlineClient,
|
|
182
|
+
globalState,
|
|
183
|
+
);
|
|
184
|
+
};
|
|
185
|
+
}
|
package/src/actor-handler-do.ts
CHANGED
|
@@ -1,11 +1,20 @@
|
|
|
1
1
|
import { DurableObject } from "cloudflare:workers";
|
|
2
|
-
import type {
|
|
2
|
+
import type {
|
|
3
|
+
ActorKey,
|
|
4
|
+
ActorRouter,
|
|
5
|
+
Registry,
|
|
6
|
+
RunConfig,
|
|
7
|
+
} from "@rivetkit/core";
|
|
8
|
+
import {
|
|
9
|
+
createActorRouter,
|
|
10
|
+
createClientWithDriver,
|
|
11
|
+
createInlineClientDriver,
|
|
12
|
+
} from "@rivetkit/core";
|
|
3
13
|
import { serializeEmptyPersistData } from "@rivetkit/core/driver-helpers";
|
|
4
|
-
import { PartitionTopologyActor } from "@rivetkit/core/topologies/partition";
|
|
5
14
|
import type { ExecutionContext } from "hono";
|
|
6
15
|
import {
|
|
7
|
-
CloudflareActorsActorDriver,
|
|
8
16
|
CloudflareDurableObjectGlobalState,
|
|
17
|
+
createCloudflareActorsActorDriverBuilder,
|
|
9
18
|
} from "./actor-driver";
|
|
10
19
|
import { type Bindings, CF_AMBIENT_ENV } from "./handler";
|
|
11
20
|
import { logger } from "./log";
|
|
@@ -36,7 +45,7 @@ export type DurableObjectConstructor = new (
|
|
|
36
45
|
) => DurableObject<Bindings>;
|
|
37
46
|
|
|
38
47
|
interface LoadedActor {
|
|
39
|
-
|
|
48
|
+
actorRouter: ActorRouter;
|
|
40
49
|
}
|
|
41
50
|
|
|
42
51
|
export function createActorDurableObject(
|
|
@@ -98,34 +107,47 @@ export function createActorDurableObject(
|
|
|
98
107
|
|
|
99
108
|
if (!this.#initialized) throw new Error("Not initialized");
|
|
100
109
|
|
|
110
|
+
// Register DO with global state first
|
|
111
|
+
// HACK: This leaks the DO context, but DO does not provide a native way
|
|
112
|
+
// of knowing when the DO shuts down. We're making a broad assumption
|
|
113
|
+
// that DO will boot a new isolate frequenlty enough that this is not an issue.
|
|
114
|
+
const actorId = this.ctx.id.toString();
|
|
115
|
+
globalState.setDOState(actorId, { ctx: this.ctx, env: this.env });
|
|
116
|
+
|
|
101
117
|
// Configure actor driver
|
|
102
|
-
runConfig.driver.actor =
|
|
118
|
+
runConfig.driver.actor =
|
|
119
|
+
createCloudflareActorsActorDriverBuilder(globalState);
|
|
103
120
|
|
|
104
|
-
|
|
121
|
+
// Create manager driver (we need this for the actor router)
|
|
122
|
+
const managerDriver = runConfig.driver.manager(
|
|
105
123
|
registry.config,
|
|
106
124
|
runConfig,
|
|
107
125
|
);
|
|
108
126
|
|
|
109
|
-
//
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
127
|
+
// Create inline client
|
|
128
|
+
const inlineClient = createClientWithDriver(
|
|
129
|
+
createInlineClientDriver(managerDriver),
|
|
130
|
+
);
|
|
131
|
+
|
|
132
|
+
// Create actor driver
|
|
133
|
+
const actorDriver = runConfig.driver.actor(
|
|
134
|
+
registry.config,
|
|
135
|
+
runConfig,
|
|
136
|
+
managerDriver,
|
|
137
|
+
inlineClient,
|
|
138
|
+
);
|
|
139
|
+
|
|
140
|
+
// Create actor router
|
|
141
|
+
const actorRouter = createActorRouter(runConfig, actorDriver);
|
|
115
142
|
|
|
116
143
|
// Save actor
|
|
117
144
|
this.#actor = {
|
|
118
|
-
|
|
145
|
+
actorRouter,
|
|
119
146
|
};
|
|
120
147
|
|
|
121
|
-
//
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
this.#initialized.name,
|
|
125
|
-
this.#initialized.key,
|
|
126
|
-
// TODO:
|
|
127
|
-
"unknown",
|
|
128
|
-
);
|
|
148
|
+
// Initialize the actor instance with proper metadata
|
|
149
|
+
// This ensures the actor driver knows about this actor
|
|
150
|
+
await actorDriver.loadActor(actorId);
|
|
129
151
|
|
|
130
152
|
return this.#actor;
|
|
131
153
|
}
|
|
@@ -154,30 +176,38 @@ export function createActorDurableObject(
|
|
|
154
176
|
|
|
155
177
|
async fetch(request: Request): Promise<Response> {
|
|
156
178
|
return await CF_AMBIENT_ENV.run(this.env, async () => {
|
|
157
|
-
const {
|
|
158
|
-
|
|
159
|
-
const
|
|
160
|
-
return await
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
// Implement execution context so we can wait on requests
|
|
164
|
-
{
|
|
165
|
-
waitUntil(promise: Promise<unknown>) {
|
|
166
|
-
ctx.waitUntil(promise);
|
|
167
|
-
},
|
|
168
|
-
passThroughOnException() {
|
|
169
|
-
// Do nothing
|
|
170
|
-
},
|
|
171
|
-
props: {},
|
|
172
|
-
} satisfies ExecutionContext,
|
|
173
|
-
);
|
|
179
|
+
const { actorRouter } = await this.#loadActor();
|
|
180
|
+
|
|
181
|
+
const actorId = this.ctx.id.toString();
|
|
182
|
+
return await actorRouter.fetch(request, {
|
|
183
|
+
actorId,
|
|
184
|
+
});
|
|
174
185
|
});
|
|
175
186
|
}
|
|
176
187
|
|
|
177
188
|
async alarm(): Promise<void> {
|
|
178
189
|
return await CF_AMBIENT_ENV.run(this.env, async () => {
|
|
179
|
-
|
|
180
|
-
|
|
190
|
+
await this.#loadActor();
|
|
191
|
+
const actorId = this.ctx.id.toString();
|
|
192
|
+
|
|
193
|
+
// Get the actor driver
|
|
194
|
+
const managerDriver = runConfig.driver.manager(
|
|
195
|
+
registry.config,
|
|
196
|
+
runConfig,
|
|
197
|
+
);
|
|
198
|
+
const inlineClient = createClientWithDriver(
|
|
199
|
+
createInlineClientDriver(managerDriver),
|
|
200
|
+
);
|
|
201
|
+
const actorDriver = runConfig.driver.actor(
|
|
202
|
+
registry.config,
|
|
203
|
+
runConfig,
|
|
204
|
+
managerDriver,
|
|
205
|
+
inlineClient,
|
|
206
|
+
);
|
|
207
|
+
|
|
208
|
+
// Load the actor instance and trigger alarm
|
|
209
|
+
const actor = await actorDriver.loadActor(actorId);
|
|
210
|
+
await actor.onAlarm();
|
|
181
211
|
});
|
|
182
212
|
}
|
|
183
213
|
};
|
package/src/handler.ts
CHANGED
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
import { AsyncLocalStorage } from "node:async_hooks";
|
|
2
2
|
import type { Registry, RunConfig } from "@rivetkit/core";
|
|
3
3
|
import type { Client } from "@rivetkit/core/client";
|
|
4
|
-
import { PartitionTopologyManager } from "@rivetkit/core/topologies/partition";
|
|
5
4
|
import { Hono } from "hono";
|
|
6
5
|
import invariant from "invariant";
|
|
7
6
|
import {
|
|
@@ -59,9 +58,8 @@ export function createServer<R extends Registry<any>>(
|
|
|
59
58
|
// Create config
|
|
60
59
|
const runConfig = {
|
|
61
60
|
driver: {
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
// HACK: We can't build the actor driver until we're inside the Druable Object
|
|
61
|
+
manager: () => new CloudflareActorsManagerDriver(),
|
|
62
|
+
// HACK: We can't build the actor driver until we're inside the Durable Object
|
|
65
63
|
actor: undefined as any,
|
|
66
64
|
},
|
|
67
65
|
getUpgradeWebSocket: () => upgradeWebSocket,
|
|
@@ -71,19 +69,19 @@ export function createServer<R extends Registry<any>>(
|
|
|
71
69
|
// Create Durable Object
|
|
72
70
|
const ActorHandler = createActorDurableObject(registry, runConfig);
|
|
73
71
|
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
runConfig,
|
|
77
|
-
);
|
|
72
|
+
// Create server
|
|
73
|
+
const serverOutput = registry.createServer(runConfig);
|
|
78
74
|
|
|
79
75
|
return {
|
|
80
|
-
client:
|
|
76
|
+
client: serverOutput.client as Client<R>,
|
|
81
77
|
createHandler: (hono) => {
|
|
82
78
|
// Build base router
|
|
83
79
|
const app = hono ?? new Hono();
|
|
84
80
|
|
|
85
|
-
// Mount registry
|
|
86
|
-
|
|
81
|
+
// Mount registry router at /registry
|
|
82
|
+
if (!hono) {
|
|
83
|
+
app.route("/registry", serverOutput.hono);
|
|
84
|
+
}
|
|
87
85
|
|
|
88
86
|
// Create Cloudflare handler
|
|
89
87
|
const handler = {
|