rivetkit 2.0.20 → 2.0.21

package/src/manager/router-schema.ts

@@ -1,7 +1,9 @@
 import { z } from "zod";
 
 export const ServerlessStartHeadersSchema = z.object({
-	endpoint: z.string({ required_error: "x-rivet-endpoint header is required" }),
+	endpoint: z.string({
+		required_error: "x-rivet-endpoint header is required",
+	}),
 	token: z
 		.string({ invalid_type_error: "x-rivet-token header must be a string" })
 		.optional(),
@@ -15,6 +17,6 @@ export const ServerlessStartHeadersSchema = z.object({
 		required_error: "x-rivet-runner-name header is required",
 	}),
 	namespace: z.string({
-		required_error: "x-rivet-namespace-id header is required",
+		required_error: "x-rivet-namespace-name header is required",
 	}),
 });

package/src/manager/router.ts

@@ -166,7 +166,7 @@ function addServerlessRoutes(
 			token: c.req.header("x-rivet-token") ?? undefined,
 			totalSlots: c.req.header("x-rivet-total-slots"),
 			runnerName: c.req.header("x-rivet-runner-name"),
-			namespace: c.req.header("x-rivet-namespace-id"),
+			namespace: c.req.header("x-rivet-namespace-name"),
 		});
 		if (!parseResult.success) {
 			throw new InvalidRequest(
@@ -262,7 +262,7 @@ function addManagerRoutes(
 			path: "/actors",
 			request: {
 				query: z.object({
-					name: z.string(),
+					name: z.string().optional(),
 					actor_ids: z.string().optional(),
 					key: z.string().optional(),
 				}),
@@ -282,6 +282,36 @@ function addManagerRoutes(
 
 			const actors: ActorOutput[] = [];
 
+			// Validate: cannot provide both actor_ids and (name or key)
+			if (actorIdsParsed && (name || key)) {
+				return c.json(
+					{
+						error: "Cannot provide both actor_ids and (name + key). Use either actor_ids or (name + key).",
+					},
+					400,
+				);
+			}
+
+			// Validate: when key is provided, name must also be provided
+			if (key && !name) {
+				return c.json(
+					{
+						error: "When providing 'key', 'name' must also be provided.",
+					},
+					400,
+				);
+			}
+
+			// Validate: must provide either actor_ids or (name + key)
+			if (!actorIdsParsed && !key) {
+				return c.json(
+					{
+						error: "Must provide either 'actor_ids' or both 'name' and 'key'.",
+					},
+					400,
+				);
+			}
+
 			if (actorIdsParsed) {
 				if (actorIdsParsed.length > 32) {
 					return c.json(
@@ -298,8 +328,10 @@ function addManagerRoutes(
 					});
 				}
 
+				// Fetch actors by ID
 				for (const actorId of actorIdsParsed) {
 					if (name) {
+						// If name is provided, use it directly
 						const actorOutput = await managerDriver.getForId({
 							c,
 							name,
@@ -308,12 +340,29 @@ function addManagerRoutes(
 						if (actorOutput) {
 							actors.push(actorOutput);
 						}
+					} else {
+						// If no name is provided, try all registered actor types
+						// Actor IDs are globally unique, so we'll find it in one of them
+						for (const actorName of Object.keys(
+							registryConfig.use,
+						)) {
+							const actorOutput = await managerDriver.getForId({
+								c,
+								name: actorName,
+								actorId,
+							});
+							if (actorOutput) {
+								actors.push(actorOutput);
+								break; // Found the actor, no need to check other names
+							}
+						}
 					}
 				}
 			} else if (key) {
+				// At this point, name is guaranteed to be defined due to validation above
 				const actorOutput = await managerDriver.getWithKey({
 					c,
-					name,
+					name: name!,
 					key: [key], // Convert string to ActorKey array
 				});
 				if (actorOutput) {
@@ -449,8 +498,14 @@ function addManagerRoutes(
 		router.post(".test/inline-driver/call", async (c) => {
 			// TODO: use openapi instead
 			const buffer = await c.req.arrayBuffer();
-			const { encoding, transport, method, args }: TestInlineDriverCallRequest =
-				cbor.decode(new Uint8Array(buffer));
+			const {
+				encoding,
+				transport,
+				method,
+				args,
+			}: TestInlineDriverCallRequest = cbor.decode(
+				new Uint8Array(buffer),
+			);
 
 			logger().debug({
 				msg: "received inline request",
@@ -463,7 +518,9 @@ function addManagerRoutes(
 			// Forward inline driver request
 			let response: TestInlineDriverCallResponse<unknown>;
 			try {
-				const output = await ((managerDriver as any)[method] as any)(...args);
+				const output = await ((managerDriver as any)[method] as any)(
+					...args,
+				);
 				response = { ok: output };
 			} catch (rawErr) {
 				const err = deconstructError(rawErr, logger(), {}, true);
@@ -476,11 +533,15 @@ function addManagerRoutes(
 
 		router.get(".test/inline-driver/connect-websocket/*", async (c) => {
 			const upgradeWebSocket = runConfig.getUpgradeWebSocket?.();
-			invariant(upgradeWebSocket, "websockets not supported on this platform");
+			invariant(
+				upgradeWebSocket,
+				"websockets not supported on this platform",
+			);
 
 			return upgradeWebSocket(async (c: any) => {
 				// Extract information from sec-websocket-protocol header
-				const protocolHeader = c.req.header("sec-websocket-protocol") || "";
+				const protocolHeader =
+					c.req.header("sec-websocket-protocol") || "";
 				const protocols = protocolHeader.split(/,\s*/);
 
 				// Parse protocols to extract connection info
@@ -515,7 +576,9 @@ function addManagerRoutes(
 					} else if (protocol.startsWith(WS_PROTOCOL_CONN_ID)) {
 						connId = protocol.substring(WS_PROTOCOL_CONN_ID.length);
 					} else if (protocol.startsWith(WS_PROTOCOL_CONN_TOKEN)) {
-						connToken = protocol.substring(WS_PROTOCOL_CONN_TOKEN.length);
+						connToken = protocol.substring(
+							WS_PROTOCOL_CONN_TOKEN.length,
+						);
 					}
 				}
 
@@ -619,9 +682,12 @@ function addManagerRoutes(
 				// Send a special request to the actor to force disconnect the connection
 				const response = await managerDriver.sendRequest(
 					actorId,
-					new Request(`http://actor/.test/force-disconnect?conn=${connId}`, {
-						method: "POST",
-					}),
+					new Request(
+						`http://actor/.test/force-disconnect?conn=${connId}`,
+						{
+							method: "POST",
+						},
+					),
 				);
 
 				if (!response.ok) {

package/src/manager-api/actors.ts

@@ -23,6 +23,7 @@ export type ActorsListResponse = z.infer<typeof ActorsListResponseSchema>;
 
 // MARK: POST /actors
 export const ActorsCreateRequestSchema = z.object({
+	datacenter: z.string().optional(),
 	name: z.string(),
 	runner_name_selector: z.string(),
 	crash_policy: z.string(),
@@ -38,6 +39,7 @@ export type ActorsCreateResponse = z.infer<typeof ActorsCreateResponseSchema>;
 
 // MARK: PUT /actors
 export const ActorsGetOrCreateRequestSchema = z.object({
+	datacenter: z.string().optional(),
 	name: z.string(),
 	key: z.string(),
 	runner_name_selector: z.string(),

package/src/registry/mod.ts

@@ -62,7 +62,10 @@ export class Registry<A extends RegistryActors> {
 		const config = RunnerConfigSchema.parse(inputConfig);
 
 		// Validate autoConfigureServerless is only used with serverless runner
-		if (config.autoConfigureServerless && config.runnerKind !== "serverless") {
+		if (
+			config.autoConfigureServerless &&
+			config.runnerKind !== "serverless"
+		) {
 			throw new Error(
 				"autoConfigureServerless can only be configured when runnerKind is 'serverless'",
 			);
@@ -171,7 +174,10 @@ export class Registry<A extends RegistryActors> {
 			...driverLog,
 		});
 		if (isInspectorEnabled(config, "manager") && managerDriver.inspector) {
-			logger().info({ msg: "inspector ready", url: getInspectorUrl(config) });
+			logger().info({
+				msg: "inspector ready",
+				url: getInspectorUrl(config),
+			});
 		}
 
 		// Print welcome information
@@ -182,7 +188,9 @@ export class Registry<A extends RegistryActors> {
 			if (!config.disableDefaultServer) {
 				console.log(`  - Endpoint:     ${getEndpoint(config)}`);
 			} else if (config.overrideServerAddress) {
-				console.log(`  - Endpoint:     ${config.overrideServerAddress}`);
+				console.log(
+					`  - Endpoint:     ${config.overrideServerAddress}`,
+				);
 			}
 			if (config.runEngine) {
 				const padding = " ".repeat(Math.max(0, 13 - "Engine".length));
@@ -192,7 +200,10 @@ export class Registry<A extends RegistryActors> {
 				const padding = " ".repeat(Math.max(0, 13 - k.length));
 				console.log(`  - ${k}:${padding}${v}`);
 			}
-			if (isInspectorEnabled(config, "manager") && managerDriver.inspector) {
+			if (
+				isInspectorEnabled(config, "manager") &&
+				managerDriver.inspector
+			) {
 				console.log(`  - Inspector:    ${getInspectorUrl(config)}`);
 			}
 			console.log();
@@ -210,7 +221,10 @@ export class Registry<A extends RegistryActors> {
 		}
 
 		// Configure serverless runner if enabled when actor driver is disabled
-		if (config.runnerKind === "serverless" && config.autoConfigureServerless) {
+		if (
+			config.runnerKind === "serverless" &&
+			config.autoConfigureServerless
+		) {
 			Promise.all(readyPromises).then(async () => {
 				await configureServerlessRunner(config);
 			});
@@ -245,13 +259,19 @@ async function configureServerlessRunner(config: RunnerConfig): Promise<void> {
 	try {
 		// Ensure we have required config values
 		if (!config.runnerName) {
-			throw new Error("runnerName is required for serverless configuration");
+			throw new Error(
+				"runnerName is required for serverless configuration",
+			);
 		}
 		if (!config.namespace) {
-			throw new Error("namespace is required for serverless configuration");
+			throw new Error(
+				"namespace is required for serverless configuration",
+			);
 		}
 		if (!config.endpoint) {
-			throw new Error("endpoint is required for serverless configuration");
+			throw new Error(
+				"endpoint is required for serverless configuration",
+			);
 		}
 
 		// Prepare the configuration
@@ -288,7 +308,9 @@ async function configureServerlessRunner(config: RunnerConfig): Promise<void> {
 		});
 		const serverlessConfig = {
 			serverless: {
-				url: customConfig.url || `http://localhost:${config.defaultServerPort}`,
+				url:
+					customConfig.url ||
+					`http://localhost:${config.defaultServerPort}`,
 				headers: customConfig.headers || {},
 				max_runners: customConfig.maxRunners ?? 100,
 				min_runners: customConfig.minRunners ?? 0,

package/src/registry/run-config.ts

@@ -50,7 +50,9 @@ export const RunnerConfigSchema = z
 		runEngineVersion: z
 			.string()
 			.optional()
-			.default(() => getEnvUniversal("RIVET_RUN_ENGINE_VERSION") ?? "25.8.1"),
+			.default(
+				() => getEnvUniversal("RIVET_RUN_ENGINE_VERSION") ?? "25.8.2",
+			),
 
 		/** @experimental */
 		overrideServerAddress: z.string().optional(),

package/src/remote-manager-driver/api-endpoints.ts

@@ -15,13 +15,13 @@ import { apiCall } from "./api-utils";
 // MARK: Get actor
 export async function getActor(
 	config: ClientConfig,
-	name: string,
+	_: string,
 	actorId: RivetId,
 ): Promise<ActorsListResponse> {
 	return apiCall<never, ActorsListResponse>(
 		config,
 		"GET",
-		`/actors?name=${name}&actor_ids=${encodeURIComponent(actorId)}`,
+		`/actors?actor_ids=${encodeURIComponent(actorId)}`,
 	);
 }
 

package/src/remote-manager-driver/mod.ts

@@ -60,7 +60,9 @@ export class RemoteManagerDriver implements ManagerDriver {
 		//
 		// See https://github.com/vercel/next.js/blob/5e6b008b561caf2710ab7be63320a3d549474a5b/packages/next/shared/lib/constants.ts#L19-L23
 		if (getEnvUniversal("NEXT_PHASE") === "phase-production-build") {
-			logger().info("detected next.js build phase, disabling health check");
+			logger().info(
+				"detected next.js build phase, disabling health check",
+			);
 			runConfig.disableHealthCheck = true;
 		}
 
@@ -70,10 +72,10 @@ export class RemoteManagerDriver implements ManagerDriver {
 		if (!runConfig.disableHealthCheck) {
 			this.#metadataPromise = this.#performMetadataCheck(runConfig);
 			this.#metadataPromise.catch((error) => {
-				// TODO: Promot to error after metadata endpoint merged on prod
-				logger().info({
+				logger().error({
 					msg: "metadata check failed",
-					error: error instanceof Error ? error.message : String(error),
+					error:
+						error instanceof Error ? error.message : String(error),
 				});
 			});
 		}
@@ -219,6 +221,7 @@ export class RemoteManagerDriver implements ManagerDriver {
 		});
 
 		const { actor, created } = await getOrCreateActor(this.#config, {
+			datacenter: region,
 			name,
 			key: serializeActorKey(key),
 			runner_name_selector: this.#config.runnerName,
@@ -250,6 +253,7 @@ export class RemoteManagerDriver implements ManagerDriver {
 		name,
 		key,
 		input,
+		region,
 	}: CreateInput): Promise<ActorOutput> {
 		// Wait for metadata check to complete if in progress
 		if (this.#metadataPromise) {
@@ -260,6 +264,7 @@ export class RemoteManagerDriver implements ManagerDriver {
 
 		// Create actor via engine API
 		const result = await createActor(this.#config, {
+			datacenter: region,
 			name,
 			runner_name_selector: this.#config.runnerName,
 			key: serializeActorKey(key),
@@ -290,13 +295,20 @@ export class RemoteManagerDriver implements ManagerDriver {
 		logger().info({ msg: "actor destroyed", actorId });
 	}
 
-	async sendRequest(actorId: string, actorRequest: Request): Promise<Response> {
+	async sendRequest(
+		actorId: string,
+		actorRequest: Request,
+	): Promise<Response> {
 		// Wait for metadata check to complete if in progress
 		if (this.#metadataPromise) {
 			await this.#metadataPromise;
 		}
 
-		return await sendHttpRequestToActor(this.#config, actorId, actorRequest);
+		return await sendHttpRequestToActor(
+			this.#config,
+			actorId,
+			actorRequest,
+		);
 	}
 
 	async openWebSocket(
@@ -333,7 +345,11 @@ export class RemoteManagerDriver implements ManagerDriver {
 			await this.#metadataPromise;
 		}
 
-		return await sendHttpRequestToActor(this.#config, actorId, actorRequest);
+		return await sendHttpRequestToActor(
+			this.#config,
+			actorId,
+			actorRequest,
+		);
 	}
 
 	async proxyWebSocket(

package/src/remote-manager-driver/ws-proxy.ts

@@ -42,7 +42,10 @@ export async function createWebSocketProxy(
 			// Setup connection promise
 			state.connectPromise = new Promise<void>((resolve, reject) => {
 				targetWs.addEventListener("open", () => {
-					logger().debug({ msg: "target websocket connected", targetUrl });
+					logger().debug({
+						msg: "target websocket connected",
+						targetUrl,
+					});
 
 					if (clientWs.readyState !== 1) {
 						logger().warn({
@@ -102,7 +105,10 @@ export async function createWebSocketProxy(
 
 		onMessage: async (event: any, clientWs: WSContext) => {
 			if (!state.targetWs || !state.connectPromise) {
-				logger().error({ msg: "websocket state not initialized", targetUrl });
+				logger().error({
+					msg: "websocket state not initialized",
+					targetUrl,
+				});
 				return;
 			}
 
@@ -123,7 +129,11 @@ export async function createWebSocketProxy(
 					targetUrl,
 					error,
 				});
-				closeWebSocketIfOpen(clientWs, 1011, "Failed to connect to target");
+				closeWebSocketIfOpen(
+					clientWs,
+					1011,
+					"Failed to connect to target",
+				);
 			}
 		},
 
@@ -141,7 +151,10 @@ export async function createWebSocketProxy(
 					state.targetWs.readyState === WebSocket.OPEN ||
 					state.targetWs.readyState === WebSocket.CONNECTING
 				) {
-					state.targetWs.close(1000, event.reason || "Client disconnected");
+					state.targetWs.close(
+						1000,
+						event.reason || "Client disconnected",
+					);
 				}
 			}
 		},
@@ -167,7 +180,10 @@ function closeWebSocketIfOpen(
 ): void {
 	if (ws.readyState === 1) {
 		ws.close(code, reason);
-	} else if ("close" in ws && (ws as WebSocket).readyState === WebSocket.OPEN) {
+	} else if (
+		"close" in ws &&
+		(ws as WebSocket).readyState === WebSocket.OPEN
+	) {
 		ws.close(code, reason);
 	}
 }

package/src/serde.ts

@@ -63,7 +63,9 @@ export function serializeWithEncoding<T>(
 		return cbor.encode(value);
 	} else if (encoding === "bare") {
 		if (!versionedDataHandler) {
-			throw new Error("VersionedDataHandler is required for 'bare' encoding");
+			throw new Error(
+				"VersionedDataHandler is required for 'bare' encoding",
+			);
 		}
 		return versionedDataHandler.serializeWithEmbeddedVersion(value);
 	} else {
@@ -96,7 +98,9 @@ export function deserializeWithEncoding<T>(
 			"buffer cannot be string for bare encoding",
 		);
 		if (!versionedDataHandler) {
-			throw new Error("VersionedDataHandler is required for 'bare' encoding");
+			throw new Error(
+				"VersionedDataHandler is required for 'bare' encoding",
+			);
 		}
 		return versionedDataHandler.deserializeWithEmbeddedVersion(buffer);
 	} else {

package/src/test/mod.ts

@@ -94,7 +94,8 @@ export async function setupTest<A extends Registry<any>>(
 	const port = await getPort();
 	const server = serve(registry, { port });
 	c.onTestFinished(
-		async () => await new Promise((resolve) => server.close(() => resolve())),
+		async () =>
+			await new Promise((resolve) => server.close(() => resolve())),
 	);
 
 	throw "TODO: Fix engine port";

package/src/utils.ts

@@ -19,7 +19,8 @@ export function httpUserAgent(): string {
 	let userAgent = `RivetKit/${VERSION}`;
 
 	// Navigator
-	const navigatorObj = typeof navigator !== "undefined" ? navigator : undefined;
+	const navigatorObj =
+		typeof navigator !== "undefined" ? navigator : undefined;
 	if (navigatorObj?.userAgent) userAgent += ` ${navigatorObj.userAgent}`;
 
 	_userAgent = userAgent;
@@ -62,7 +63,10 @@ export function toUint8Array(data: ArrayBuffer | ArrayBufferView): Uint8Array {
 	} else if (ArrayBuffer.isView(data)) {
 		// Handle other ArrayBufferView types (Int8Array, Uint16Array, DataView, etc.)
 		return new Uint8Array(
-			data.buffer.slice(data.byteOffset, data.byteOffset + data.byteLength),
+			data.buffer.slice(
+				data.byteOffset,
+				data.byteOffset + data.byteLength,
+			),
 		);
 	} else {
 		throw new TypeError("Input must be ArrayBuffer or ArrayBufferView");

package/dist/tsup/chunk-2I6L3VRO.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/nathan/rivetkit/packages/rivetkit/dist/tsup/chunk-2I6L3VRO.cjs","../../src/test/mod.ts","../../src/test/config.ts","../../src/test/log.ts"],"names":["server"],"mappings":"AAAA;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;AClBA,0BAA6B;AAC7B,+CAAoD;AACpD,uCAAoC;AACpC,gCAAqC;ADoBrC;AACA;AExBA,0BAAkB;AAGX,IAAM,aAAA,EAAe,oCAAA,CAAmB,aAAA,CAAc,CAAA,CAC3D,MAAA,CAAO;AAAA,EACP,QAAA,EAAU,MAAA,CACR,MAAA,CAAO,CAAA,CACP,QAAA,CAAS,CAAA,CACT,OAAA,kBAAQ,OAAA,CAAQ,GAAA,CAAI,QAAA,UAAY,aAAW,CAAA;AAAA,EAC7C,IAAA,EAAM,MAAA,CACJ,MAAA,CAAO,CAAA,CACP,QAAA,CAAS,CAAA,CACT,OAAA,CAAQ,MAAA,CAAO,QAAA,kBAAS,OAAA,CAAQ,GAAA,CAAI,IAAA,UAAQ,QAAM,CAAC;AACtD,CAAC,CAAA,CACA,OAAA,CAAQ,CAAC,CAAC,CAAA;AFgBZ;AACA;AG7BO,SAAS,MAAA,CAAA,EAAS;AACxB,EAAA,OAAO,yCAAA,MAAgB,CAAA;AACxB;AH+BA;AACA;ACjBA,SAAS,KAAA,CAAM,QAAA,EAAyB,WAAA,EAAuC;AAE9E,EAAA,YAAA,IAAgB,CAAC,CAAA;AAEjB,EAAA,MAAM,OAAA,EAAS,YAAA,CAAa,KAAA,CAAM,WAAW,CAAA;AAE7C,EAAA,IAAI,gBAAA;AACJ,EAAA,GAAA,CAAI,CAAC,MAAA,CAAO,mBAAA,EAAqB;AAChC,IAAA,MAAA,CAAO,oBAAA,EAAsB,CAAA,EAAA,GAAM,gBAAA;AAAA,EACpC;AAGA,EAAA,MAAM,UAAA,EAAY,oCAAA,CAAmB,KAAA,CAAM,WAAW,CAAA;AACtD,EAAA,MAAM,OAAA,mBAAS,WAAA,CAAY,MAAA,UAAU,8DAAA,KAAoC,GAAA;AACzE,EAAA,MAAM,cAAA,EAAgB,MAAA,CAAO,OAAA,CAAQ,QAAA,CAAS,MAAA,EAAQ,MAAM,CAAA;AAC5D,EAAA,MAAM,OAAA,EAAS,sDAAA;AAAA,IACd,aAAA;AAAA,IACA,oCAAA,CAAmB,KAAA,CAAM,CAAC,CAAC;AAAA,EAC5B,CAAA;AACA,EAAA,6DAAA,MAA8B,EAAQ,aAAa,CAAA;AACnD,EAAA,MAAM,EAAE,OAAO,EAAA,EAAI,mDAAA;AAAA,IAClB,QAAA,CAAS,MAAA;AAAA,IACT,SAAA;AAAA,IACA,aAAA;AAAA,IACA,MAAA;AAAA,IACA;AAAA,EACD,CAAA;AAGA,EAAA,MAAM,cAAA,EAAgB,yCAAA,EAAsB,GAAA,EAAK,OAAO,CAAC,CAAA;AACzD,EAAA,iBAAA,EAAmB,aAAA,CAAc,gBAAA;AAEjC,EAAA,MAAM,OAAA,EAAS,+BAAA;AAAU,IACxB,KAAA,EAAO,MAAA,CAAO,KAAA;AAAA,IACd,QAAA,EAAU,MAAA,CAAO,QAAA;AAAA,IACjB,IAAA,EAAM,MAAA,CAAO;AAAA,EACd,CAAC,CAAA;AACD,EAAA,aAAA,CAAc,eAAA,CAAgB,MAAM,CAAA;AAEpC,EAAA,MAAA,CAAO,CAAA,CAAE,IAAA,CAAK;AAAA,IACb,GAAA,EAAK,kBAAA;AAAA,IACL,QAAA,EAAU,MAAA,CAAO,QAAA;AAAA,IACjB,IAAA,EAAM,MAAA,CAAO,IAAA;AAAA,IACb,WAAA,EAAa,MAAA,CAAO,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,GAAG,CAAA,CAAE;AAAA,EAC/C,CAAC,CAAA;AAED,EAAA,OAAO,MAAA;AACR;AAYA,MAAA,SAAsB,SAAA,CACrB,CAAA,EACA,QAAA,EAC8B;AAC9B,EAAA,UAAA,CAAG,aAAA,CAAc,CAAA;AAGjB,EAAA,MAAM,kBAAA,EAAyB,CAAC,CAAA;AAChC,EAAA,MAAM,mBAAA,EAAqB,CAAC,GAAA,EAAA,GAAa;AACxC,IAAA,iBAAA,CAAkB,QAAA,EAAU,GAAA;AAAA,EAC7B,CAAA;AAKA,EAAA,MAAM,KAAA,EAAO,MAAM,OAAA,CAAQ,CAAA;AAC3B,EAAA,MAAM,OAAA,EAAS,KAAA,CAAM,QAAA,EAAU,EAAE,KAAK,CAAC,CAAA;AACvC,EAAA,CAAA,CAAE,cAAA;AAAA,IACD,MAAA,CAAA,EAAA,GAAY,MAAM,IAAI,OAAA,CAAQ,CAAC,OAAA,EAAA,GAAY,MAAA,CAAO,KAAA,CAAM,CAAA,EAAA,GAAM,OAAA,CAAQ,CAAC,CAAC;AAAA,EACzE,CAAA;AAEA,EAAA,MAAM,uBAAA;AAeP;AAEA,MAAA,SAAsB,OAAA,CAAA,EAA2B;AAEhD,EAAA,MAAM,SAAA,EAAW,GAAA;AACjB,EAAA,MAAM,SAAA,EAAW,KAAA;AACjB,EAAA,MAAM,cAAA,EAAgB,CAAA,EAAA,GACrB,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,MAAA,CAAO,EAAA,EAAA,CAAK,SAAA,EAAW,SAAA,EAAW,CAAA,CAAE,EAAA,EAAI,QAAA;AAEzD,EAAA,IAAI,KAAA,EAAO,aAAA,CAAc,CAAA;AACzB,EAAA,IAAI,YAAA,EAAc,EAAA;AAElB,EAAA,MAAA,CAAO,YAAA,EAAc,CAAA,EAAG;AACvB,IAAA,IAAI;AAEH,MAAA,MAAM,OAAA,EAAS,MAAM,IAAI,OAAA,CAAa,CAAC,OAAA,EAAS,MAAA,EAAA,GAAW;AAC1D,QAAA,MAAMA,QAAAA,EAAS,+BAAA,CAAa;AAE5B,QAAAA,OAAAA,CAAO,IAAA,CAAK,OAAA,EAAS,CAAC,GAAA,EAAA,GAAmC;AACxD,UAAA,GAAA,CAAI,GAAA,CAAI,KAAA,IAAS,YAAA,EAAc;AAC9B,YAAA,MAAA,CAAO,IAAI,KAAA,CAAM,CAAA,KAAA,EAAQ,IAAI,CAAA,UAAA,CAAY,CAAC,CAAA;AAAA,UAC3C,EAAA,KAAO;AACN,YAAA,MAAA,CAAO,GAAG,CAAA;AAAA,UACX;AAAA,QACD,CAAC,CAAA;AAED,QAAAA,OAAAA,CAAO,IAAA,CAAK,WAAA,EAAa,CAAA,EAAA,GAAM;AAC9B,UAAA,OAAA,CAAQA,OAAM,CAAA;AAAA,QACf,CAAC,CAAA;AAED,QAAAA,OAAAA,CAAO,MAAA,CAAO,IAAI,CAAA;AAAA,MACnB,CAAC,CAAA;AAGD,MAAA,MAAM,IAAI,OAAA,CAAc,CAAC,OAAA,EAAA,GAAY;AACpC,QAAA,MAAA,CAAO,KAAA,CAAM,CAAA,EAAA,GAAM,OAAA,CAAQ,CAAC,CAAA;AAAA,MAC7B,CAAC,CAAA;AAED,MAAA,OAAO,IAAA;AAAA,IACR,EAAA,MAAA,CAAS,GAAA,EAAK;AAEb,MAAA,WAAA,EAAA;AACA,MAAA,GAAA,CAAI,YAAA,GAAe,CAAA,EAAG;AACrB,QAAA,KAAA;AAAA,MACD;AACA,MAAA,KAAA,EAAO,aAAA,CAAc,CAAA;AAAA,IACtB;AAAA,EACD;AAEA,EAAA,MAAM,IAAI,KAAA,CAAM,0DAA0D,CAAA;AAC3E;ADxCA;AACA;AACE;AACA;AACF,yDAAC","file":"/Users/nathan/rivetkit/packages/rivetkit/dist/tsup/chunk-2I6L3VRO.cjs","sourcesContent":[null,"import { createServer } from \"node:net\";\nimport { serve as honoServe, type ServerType } from \"@hono/node-server\";\nimport { createNodeWebSocket } from \"@hono/node-ws\";\nimport { type TestContext, vi } from \"vitest\";\nimport { ClientConfigSchema } from \"@/client/config\";\nimport { type Client, createClient } from \"@/client/mod\";\nimport { chooseDefaultDriver } from \"@/drivers/default\";\nimport { createFileSystemOrMemoryDriver } from \"@/drivers/file-system/mod\";\nimport {\n\tconfigureInspectorAccessToken,\n\tgetInspectorUrl,\n} from \"@/inspector/utils\";\nimport { createManagerRouter } from \"@/manager/router\";\nimport { createClientWithDriver } from \"@/mod\";\nimport type { Registry } from \"@/registry/mod\";\nimport { RunnerConfigSchema } from \"@/registry/run-config\";\nimport { ConfigSchema, type InputConfig } from \"./config\";\nimport { logger } from \"./log\";\n\nfunction serve(registry: Registry<any>, inputConfig?: InputConfig): ServerType {\n\t// Configure default configuration\n\tinputConfig ??= {};\n\n\tconst config = ConfigSchema.parse(inputConfig);\n\n\tlet upgradeWebSocket: any;\n\tif (!config.getUpgradeWebSocket) {\n\t\tconfig.getUpgradeWebSocket = () => upgradeWebSocket!;\n\t}\n\n\t// Create router\n\tconst runConfig = RunnerConfigSchema.parse(inputConfig);\n\tconst driver = inputConfig.driver ?? createFileSystemOrMemoryDriver(false);\n\tconst managerDriver = driver.manager(registry.config, config);\n\tconst client = createClientWithDriver(\n\t\tmanagerDriver,\n\t\tClientConfigSchema.parse({}),\n\t);\n\tconfigureInspectorAccessToken(config, managerDriver);\n\tconst { router } = createManagerRouter(\n\t\tregistry.config,\n\t\trunConfig,\n\t\tmanagerDriver,\n\t\tdriver,\n\t\tclient,\n\t);\n\n\t// Inject WebSocket\n\tconst nodeWebSocket = createNodeWebSocket({ app: router });\n\tupgradeWebSocket = nodeWebSocket.upgradeWebSocket;\n\n\tconst server = honoServe({\n\t\tfetch: router.fetch,\n\t\thostname: config.hostname,\n\t\tport: config.port,\n\t});\n\tnodeWebSocket.injectWebSocket(server);\n\n\tlogger().info({\n\t\tmsg: \"rivetkit started\",\n\t\thostname: config.hostname,\n\t\tport: config.port,\n\t\tdefinitions: Object.keys(registry.config.use).length,\n\t});\n\n\treturn server;\n}\n\nexport interface SetupTestResult<A extends Registry<any>> {\n\tclient: Client<A>;\n\tmockDriver: {\n\t\tactorDriver: {\n\t\t\tsetCreateVarsContext: (ctx: any) => void;\n\t\t};\n\t};\n}\n\n// Must use `TestContext` since global hooks do not work when running concurrently\nexport async function setupTest<A extends Registry<any>>(\n\tc: TestContext,\n\tregistry: A,\n): Promise<SetupTestResult<A>> {\n\tvi.useFakeTimers();\n\n\t// Set up mock driver for testing createVars context\n\tconst mockDriverContext: any = {};\n\tconst setDriverContextFn = (ctx: any) => {\n\t\tmockDriverContext.current = ctx;\n\t};\n\n\t// We don't need to modify the driver context anymore since we're testing with the actual context\n\n\t// Start server with a random port\n\tconst port = await getPort();\n\tconst server = serve(registry, { port });\n\tc.onTestFinished(\n\t\tasync () => await new Promise((resolve) => server.close(() => resolve())),\n\t);\n\n\tthrow \"TODO: Fix engine port\";\n\n\t// // TODO: Figure out how to make this the correct endpoint\n\t// // Create client\n\t// const client = createClient<A>(`http://127.0.0.1:${port}`);\n\t// c.onTestFinished(async () => await client.dispose());\n\t//\n\t// return {\n\t// \tclient,\n\t// \tmockDriver: {\n\t// \t\tactorDriver: {\n\t// \t\t\tsetCreateVarsContext: setDriverContextFn,\n\t// \t\t},\n\t// \t},\n\t// };\n}\n\nexport async function getPort(): Promise<number> {\n\t// Pick random port between 10000 and 65535 (avoiding well-known and registered ports)\n\tconst MIN_PORT = 10000;\n\tconst MAX_PORT = 65535;\n\tconst getRandomPort = () =>\n\t\tMath.floor(Math.random() * (MAX_PORT - MIN_PORT + 1)) + MIN_PORT;\n\n\tlet port = getRandomPort();\n\tlet maxAttempts = 10;\n\n\twhile (maxAttempts > 0) {\n\t\ttry {\n\t\t\t// Try to create a server on the port to check if it's available\n\t\t\tconst server = await new Promise<any>((resolve, reject) => {\n\t\t\t\tconst server = createServer();\n\n\t\t\t\tserver.once(\"error\", (err: Error & { code?: string }) => {\n\t\t\t\t\tif (err.code === \"EADDRINUSE\") {\n\t\t\t\t\t\treject(new Error(`Port ${port} is in use`));\n\t\t\t\t\t} else {\n\t\t\t\t\t\treject(err);\n\t\t\t\t\t}\n\t\t\t\t});\n\n\t\t\t\tserver.once(\"listening\", () => {\n\t\t\t\t\tresolve(server);\n\t\t\t\t});\n\n\t\t\t\tserver.listen(port);\n\t\t\t});\n\n\t\t\t// Close the server since we're just checking availability\n\t\t\tawait new Promise<void>((resolve) => {\n\t\t\t\tserver.close(() => resolve());\n\t\t\t});\n\n\t\t\treturn port;\n\t\t} catch (err) {\n\t\t\t// If port is in use, try a different one\n\t\t\tmaxAttempts--;\n\t\t\tif (maxAttempts <= 0) {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tport = getRandomPort();\n\t\t}\n\t}\n\n\tthrow new Error(\"Could not find an available port after multiple attempts\");\n}\n","import { z } from \"zod\";\nimport { RunnerConfigSchema } from \"@/registry/run-config\";\n\nexport const ConfigSchema = RunnerConfigSchema.removeDefault()\n\t.extend({\n\t\thostname: z\n\t\t\t.string()\n\t\t\t.optional()\n\t\t\t.default(process.env.HOSTNAME ?? \"127.0.0.1\"),\n\t\tport: z\n\t\t\t.number()\n\t\t\t.optional()\n\t\t\t.default(Number.parseInt(process.env.PORT ?? \"8080\")),\n\t})\n\t.default({});\nexport type InputConfig = z.input<typeof ConfigSchema>;\n","import { getLogger } from \"@/common/log\";\n\nexport function logger() {\n\treturn getLogger(\"test\");\n}\n"]}