rivetkit 2.0.23 → 2.0.24

package/dist/tsup/chunk-YAYNBR37.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/inspector/protocol/actor.ts","../../src/inspector/protocol/manager.ts","../../src/inspector/utils.ts","../../src/inspector/log.ts"],"sourcesContent":["import { hc } from \"hono/client\";\nimport type { createActorInspectorRouter } from \"../actor\";\n\ntype ActorInspectorRouter = ReturnType<typeof createActorInspectorRouter>;\nconst client = hc<ActorInspectorRouter>(\"\");\nexport type ActorInspectorClient = typeof client;\n\nexport const createActorInspectorClient = (\n\t...args: Parameters<typeof hc>\n): ActorInspectorClient => hc<ActorInspectorRouter>(...args);\n","import { hc } from \"hono/client\";\nimport type { createManagerInspectorRouter } from \"../manager\";\n\ntype ManagerInspectorRouter = ReturnType<typeof createManagerInspectorRouter>;\nconst client = hc<ManagerInspectorRouter>(\"\");\nexport type ManagerInspectorClient = typeof client;\n\nexport const createManagerInspectorClient = (\n\t...args: Parameters<typeof hc>\n): ManagerInspectorClient => hc<ManagerInspectorRouter>(...args);\n","import crypto from \"node:crypto\";\nimport { createMiddleware } from \"hono/factory\";\nimport type { ManagerDriver } from \"@/driver-helpers/mod\";\nimport type { RunConfig } from \"@/mod\";\nimport type { RunnerConfigInput } from \"@/registry/run-config\";\nimport { inspectorLogger } from \"./log\";\n\nexport function compareSecrets(providedSecret: string, validSecret: string) {\n\t// Early length check to avoid unnecessary processing\n\tif (providedSecret.length !== validSecret.length) {\n\t\treturn false;\n\t}\n\n\tconst encoder = new TextEncoder();\n\n\tconst a = encoder.encode(providedSecret);\n\tconst b = encoder.encode(validSecret);\n\n\tif (a.byteLength !== b.byteLength) {\n\t\treturn false;\n\t}\n\n\t// Perform timing-safe comparison\n\tif (!crypto.timingSafeEqual(a, b)) {\n\t\treturn false;\n\t}\n\treturn true;\n}\n\nexport const secureInspector = (runConfig: RunConfig) =>\n\tcreateMiddleware(async (c, next) => {\n\t\tconst userToken = c.req.header(\"Authorization\")?.replace(\"Bearer \", \"\");\n\t\tif (!userToken) {\n\t\t\treturn c.text(\"Unauthorized\", 401);\n\t\t}\n\n\t\tconst inspectorToken = runConfig.inspector.token?.();\n\t\tif (!inspectorToken) {\n\t\t\treturn c.text(\"Unauthorized\", 401);\n\t\t}\n\n\t\tconst isValid = compareSecrets(userToken, inspectorToken);\n\n\t\tif (!isValid) {\n\t\t\treturn c.text(\"Unauthorized\", 401);\n\t\t}\n\t\tawait next();\n\t});\n\nexport function getInspectorUrl(runConfig: RunnerConfigInput | undefined) {\n\tif (!runConfig?.inspector?.enabled) {\n\t\treturn \"disabled\";\n\t}\n\n\tconst accessToken = runConfig?.inspector?.token?.();\n\n\tif (!accessToken) {\n\t\tinspectorLogger().warn(\n\t\t\t\"Inspector Token is not set, but Inspector is enabled. Please set it in the run configuration `inspector.token` or via `RIVETKIT_INSPECTOR_TOKEN` environment variable. Inspector will not be accessible.\",\n\t\t);\n\t\treturn \"disabled\";\n\t}\n\n\tconst url = new URL(\"https://inspect.rivet.dev\");\n\n\turl.searchParams.set(\"t\", accessToken);\n\n\tconst overrideDefaultEndpoint =\n\t\trunConfig?.inspector?.defaultEndpoint ??\n\t\trunConfig.overrideServerAddress;\n\tif (overrideDefaultEndpoint) {\n\t\turl.searchParams.set(\"u\", overrideDefaultEndpoint);\n\t}\n\n\treturn url.href;\n}\n\nexport const isInspectorEnabled = (\n\trunConfig: RunConfig,\n\tcontext: \"actor\" | \"manager\",\n) => {\n\tif (typeof runConfig.inspector?.enabled === \"boolean\") {\n\t\treturn runConfig.inspector.enabled;\n\t} else if (typeof runConfig.inspector?.enabled === \"object\") {\n\t\treturn runConfig.inspector.enabled[context];\n\t}\n\treturn false;\n};\n\nexport const configureInspectorAccessToken = (\n\trunConfig: RunConfig,\n\tmanagerDriver: ManagerDriver,\n) => {\n\tif (!runConfig.inspector?.token()) {\n\t\tconst token = managerDriver.getOrCreateInspectorAccessToken();\n\t\trunConfig.inspector.token = () => token;\n\t}\n};\n","import { getLogger } from \"@/common/log\";\n\nexport function inspectorLogger() {\n\treturn getLogger(\"inspector\");\n}\n"],"mappings":";;;;;AAAA,SAAS,UAAU;AAInB,IAAM,SAAS,GAAyB,EAAE;AAGnC,IAAM,6BAA6B,IACtC,SACuB,GAAyB,GAAG,IAAI;;;ACT3D,SAAS,MAAAA,WAAU;AAInB,IAAMC,UAASD,IAA2B,EAAE;AAGrC,IAAM,+BAA+B,IACxC,SACyBA,IAA2B,GAAG,IAAI;;;ACT/D,OAAO,YAAY;AACnB,SAAS,wBAAwB;;;ACC1B,SAAS,kBAAkB;AACjC,SAAO,UAAU,WAAW;AAC7B;;;ADGO,SAAS,eAAe,gBAAwB,aAAqB;AAE3E,MAAI,eAAe,WAAW,YAAY,QAAQ;AACjD,WAAO;AAAA,EACR;AAEA,QAAM,UAAU,IAAI,YAAY;AAEhC,QAAM,IAAI,QAAQ,OAAO,cAAc;AACvC,QAAM,IAAI,QAAQ,OAAO,WAAW;AAEpC,MAAI,EAAE,eAAe,EAAE,YAAY;AAClC,WAAO;AAAA,EACR;AAGA,MAAI,CAAC,OAAO,gBAAgB,GAAG,CAAC,GAAG;AAClC,WAAO;AAAA,EACR;AACA,SAAO;AACR;AAEO,IAAM,kBAAkB,CAAC,cAC/B,iBAAiB,OAAO,GAAG,SAAS;AA9BrC;AA+BE,QAAM,aAAY,OAAE,IAAI,OAAO,eAAe,MAA5B,mBAA+B,QAAQ,WAAW;AACpE,MAAI,CAAC,WAAW;AACf,WAAO,EAAE,KAAK,gBAAgB,GAAG;AAAA,EAClC;AAEA,QAAM,kBAAiB,qBAAU,WAAU,UAApB;AACvB,MAAI,CAAC,gBAAgB;AACpB,WAAO,EAAE,KAAK,gBAAgB,GAAG;AAAA,EAClC;AAEA,QAAM,UAAU,eAAe,WAAW,cAAc;AAExD,MAAI,CAAC,SAAS;AACb,WAAO,EAAE,KAAK,gBAAgB,GAAG;AAAA,EAClC;AACA,QAAM,KAAK;AACZ,CAAC;AAEK,SAAS,gBAAgB,WAA0C;AAjD1E;AAkDC,MAAI,GAAC,4CAAW,cAAX,mBAAsB,UAAS;AACnC,WAAO;AAAA,EACR;AAEA,QAAM,eAAc,kDAAW,cAAX,mBAAsB,UAAtB;AAEpB,MAAI,CAAC,aAAa;AACjB,oBAAgB,EAAE;AAAA,MACjB;AAAA,IACD;AACA,WAAO;AAAA,EACR;AAEA,QAAM,MAAM,IAAI,IAAI,2BAA2B;AAE/C,MAAI,aAAa,IAAI,KAAK,WAAW;AAErC,QAAM,4BACL,4CAAW,cAAX,mBAAsB,oBACtB,UAAU;AACX,MAAI,yBAAyB;AAC5B,QAAI,aAAa,IAAI,KAAK,uBAAuB;AAAA,EAClD;AAEA,SAAO,IAAI;AACZ;AAEO,IAAM,qBAAqB,CACjC,WACA,YACI;AAhFL;AAiFC,MAAI,SAAO,eAAU,cAAV,mBAAqB,aAAY,WAAW;AACtD,WAAO,UAAU,UAAU;AAAA,EAC5B,WAAW,SAAO,eAAU,cAAV,mBAAqB,aAAY,UAAU;AAC5D,WAAO,UAAU,UAAU,QAAQ,OAAO;AAAA,EAC3C;AACA,SAAO;AACR;AAEO,IAAM,gCAAgC,CAC5C,WACA,kBACI;AA5FL;AA6FC,MAAI,GAAC,eAAU,cAAV,mBAAqB,UAAS;AAClC,UAAM,QAAQ,cAAc,gCAAgC;AAC5D,cAAU,UAAU,QAAQ,MAAM;AAAA,EACnC;AACD;","names":["hc","client"]}

package/src/actor/action.ts

@@ -1,178 +0,0 @@
-import type { ActorKey } from "@/actor/mod";
-import type { Client } from "@/client/client";
-import type { Logger } from "@/common/log";
-import type { Registry } from "@/registry/mod";
-import type { Conn, ConnId } from "./conn";
-import type { ActorContext } from "./context";
-import type { AnyDatabaseProvider, InferDatabaseClient } from "./database";
-import type { SaveStateOptions } from "./instance";
-import type { Schedule } from "./schedule";
-
-/**
- * Context for a remote procedure call.
- *
- * @typeParam A Actor this action belongs to
- */
-export class ActionContext<
-	TState,
-	TConnParams,
-	TConnState,
-	TVars,
-	TInput,
-	TDatabase extends AnyDatabaseProvider,
-> {
-	#actorContext: ActorContext<
-		TState,
-		TConnParams,
-		TConnState,
-		TVars,
-		TInput,
-		TDatabase
-	>;
-
-	/**
-	 * Should not be called directly.
-	 *
-	 * @param actorContext - The actor context
-	 * @param conn - The connection associated with the action
-	 */
-	constructor(
-		actorContext: ActorContext<
-			TState,
-			TConnParams,
-			TConnState,
-			TVars,
-			TInput,
-			TDatabase
-		>,
-		public readonly conn: Conn<
-			TState,
-			TConnParams,
-			TConnState,
-			TVars,
-			TInput,
-			TDatabase
-		>,
-	) {
-		this.#actorContext = actorContext;
-	}
-
-	/**
-	 * Get the actor state
-	 */
-	get state(): TState {
-		return this.#actorContext.state;
-	}
-
-	/**
-	 * Get the actor variables
-	 */
-	get vars(): TVars {
-		return this.#actorContext.vars;
-	}
-
-	/**
-	 * Broadcasts an event to all connected clients.
-	 */
-	broadcast(name: string, ...args: any[]): void {
-		this.#actorContext.broadcast(name, ...args);
-	}
-
-	/**
-	 * Gets the logger instance.
-	 */
-	get log(): Logger {
-		return this.#actorContext.log;
-	}
-
-	/**
-	 * Gets actor ID.
-	 */
-	get actorId(): string {
-		return this.#actorContext.actorId;
-	}
-
-	/**
-	 * Gets the actor name.
-	 */
-	get name(): string {
-		return this.#actorContext.name;
-	}
-
-	/**
-	 * Gets the actor key.
-	 */
-	get key(): ActorKey {
-		return this.#actorContext.key;
-	}
-
-	/**
-	 * Gets the region.
-	 */
-	get region(): string {
-		return this.#actorContext.region;
-	}
-
-	/**
-	 * Gets the scheduler.
-	 */
-	get schedule(): Schedule {
-		return this.#actorContext.schedule;
-	}
-
-	/**
-	 * Gets the map of connections.
-	 */
-	get conns(): Map<
-		ConnId,
-		Conn<TState, TConnParams, TConnState, TVars, TInput, TDatabase>
-	> {
-		return this.#actorContext.conns;
-	}
-
-	/**
-	 * Returns the client for the given registry.
-	 */
-	client<R extends Registry<any>>(): Client<R> {
-		return this.#actorContext.client<R>();
-	}
-
-	/**
-	 * @experimental
-	 */
-	get db(): InferDatabaseClient<TDatabase> {
-		return this.#actorContext.db;
-	}
-
-	/**
-	 * Forces the state to get saved.
-	 */
-	async saveState(opts: SaveStateOptions): Promise<void> {
-		return this.#actorContext.saveState(opts);
-	}
-
-	/**
-	 * Prevents the actor from sleeping until promise is complete.
-	 */
-	waitUntil(promise: Promise<void>): void {
-		this.#actorContext.waitUntil(promise);
-	}
-
-	/**
-	 * AbortSignal that fires when the actor is stopping.
-	 */
-	get abortSignal(): AbortSignal {
-		return this.#actorContext.abortSignal;
-	}
-
-	/**
-	 * Forces the actor to sleep.
-	 *
-	 * Not supported on all drivers.
-	 *
-	 * @experimental
-	 */
-	sleep() {
-		this.#actorContext.sleep();
-	}
-}

package/src/actor/conn-drivers.ts

@@ -1,216 +0,0 @@
-import type { SSEStreamingApi } from "hono/streaming";
-import type { WSContext } from "hono/ws";
-import type { WebSocket } from "ws";
-import type { AnyConn } from "@/actor/conn";
-import type { AnyActorInstance } from "@/actor/instance";
-import type { CachedSerializer, Encoding } from "@/actor/protocol/serde";
-import { encodeDataToString } from "@/actor/protocol/serde";
-import type { HonoWebSocketAdapter } from "@/manager/hono-websocket-adapter";
-import type * as protocol from "@/schemas/client-protocol/mod";
-import { assertUnreachable, type promiseWithResolvers } from "@/utils";
-
-export enum ConnDriverKind {
-	WEBSOCKET = 0,
-	SSE = 1,
-	HTTP = 2,
-}
-
-export enum ConnReadyState {
-	UNKNOWN = -1,
-	CONNECTING = 0,
-	OPEN = 1,
-	CLOSING = 2,
-	CLOSED = 3,
-}
-
-export interface ConnDriverWebSocketState {
-	encoding: Encoding;
-	websocket: WSContext;
-	closePromise: ReturnType<typeof promiseWithResolvers<void>>;
-}
-
-export interface ConnDriverSseState {
-	encoding: Encoding;
-	stream: SSEStreamingApi;
-}
-
-export type ConnDriverHttpState = Record<never, never>;
-
-export type ConnDriverState =
-	| { [ConnDriverKind.WEBSOCKET]: ConnDriverWebSocketState }
-	| { [ConnDriverKind.SSE]: ConnDriverSseState }
-	| { [ConnDriverKind.HTTP]: ConnDriverHttpState };
-
-export interface ConnDriver<State> {
-	sendMessage?(
-		actor: AnyActorInstance,
-		conn: AnyConn,
-		state: State,
-		message: CachedSerializer<protocol.ToClient>,
-	): void;
-
-	/**
-	 * This returns a promise since we commonly disconnect at the end of a program, and not waiting will cause the socket to not close cleanly.
-	 */
-	disconnect(
-		actor: AnyActorInstance,
-		conn: AnyConn,
-		state: State,
-		reason?: string,
-	): Promise<void>;
-
-	/**
-	 * Returns the ready state of the connection.
-	 * This is used to determine if the connection is ready to send messages, or if the connection is stale.
-	 */
-	getConnectionReadyState(
-		actor: AnyActorInstance,
-		conn: AnyConn,
-		state: State,
-	): ConnReadyState | undefined;
-}
-
-// MARK: WebSocket
-const WEBSOCKET_DRIVER: ConnDriver<ConnDriverWebSocketState> = {
-	sendMessage: (
-		actor: AnyActorInstance,
-		conn: AnyConn,
-		state: ConnDriverWebSocketState,
-		message: CachedSerializer<protocol.ToClient>,
-	) => {
-		if (state.websocket.readyState !== ConnReadyState.OPEN) {
-			actor.rLog.warn({
-				msg: "attempting to send message to closed websocket, this is likely a bug in RivetKit",
-				connId: conn.id,
-				wsReadyState: state.websocket.readyState,
-			});
-			return;
-		}
-
-		const serialized = message.serialize(state.encoding);
-
-		actor.rLog.debug({
-			msg: "sending websocket message",
-			encoding: state.encoding,
-			dataType: typeof serialized,
-			isUint8Array: serialized instanceof Uint8Array,
-			isArrayBuffer: serialized instanceof ArrayBuffer,
-			dataLength:
-				(serialized as any).byteLength || (serialized as any).length,
-		});
-
-		// Convert Uint8Array to ArrayBuffer for proper transmission
-		if (serialized instanceof Uint8Array) {
-			const buffer = serialized.buffer.slice(
-				serialized.byteOffset,
-				serialized.byteOffset + serialized.byteLength,
-			);
-			// Handle SharedArrayBuffer case
-			if (buffer instanceof SharedArrayBuffer) {
-				const arrayBuffer = new ArrayBuffer(buffer.byteLength);
-				new Uint8Array(arrayBuffer).set(new Uint8Array(buffer));
-				actor.rLog.debug({
-					msg: "converted SharedArrayBuffer to ArrayBuffer",
-					byteLength: arrayBuffer.byteLength,
-				});
-				state.websocket.send(arrayBuffer);
-			} else {
-				actor.rLog.debug({
-					msg: "sending ArrayBuffer",
-					byteLength: buffer.byteLength,
-				});
-				state.websocket.send(buffer);
-			}
-		} else {
-			actor.rLog.debug({
-				msg: "sending string data",
-				length: (serialized as string).length,
-			});
-			state.websocket.send(serialized);
-		}
-	},
-
-	disconnect: async (
-		_actor: AnyActorInstance,
-		_conn: AnyConn,
-		state: ConnDriverWebSocketState,
-		reason?: string,
-	) => {
-		// Close socket
-		state.websocket.close(1000, reason);
-
-		// Create promise to wait for socket to close gracefully
-		await state.closePromise.promise;
-	},
-
-	getConnectionReadyState: (
-		_actor: AnyActorInstance,
-		_conn: AnyConn,
-		state: ConnDriverWebSocketState,
-	): ConnReadyState | undefined => {
-		return state.websocket.readyState;
-	},
-};
-
-// MARK: SSE
-const SSE_DRIVER: ConnDriver<ConnDriverSseState> = {
-	sendMessage: (
-		_actor: AnyActorInstance,
-		_conn: AnyConn,
-		state: ConnDriverSseState,
-		message: CachedSerializer<protocol.ToClient>,
-	) => {
-		state.stream.writeSSE({
-			data: encodeDataToString(message.serialize(state.encoding)),
-		});
-	},
-
-	disconnect: async (
-		_actor: AnyActorInstance,
-		_conn: AnyConn,
-		state: ConnDriverSseState,
-		_reason?: string,
-	) => {
-		state.stream.close();
-	},
-
-	getConnectionReadyState: (
-		_actor: AnyActorInstance,
-		_conn: AnyConn,
-		state: ConnDriverSseState,
-	): ConnReadyState | undefined => {
-		if (state.stream.aborted || state.stream.closed) {
-			return ConnReadyState.CLOSED;
-		}
-
-		return ConnReadyState.OPEN;
-	},
-};
-
-// MARK: HTTP
-const HTTP_DRIVER: ConnDriver<ConnDriverHttpState> = {
-	getConnectionReadyState(_actor, _conn) {
-		// TODO: This might not be the correct logic
-		return ConnReadyState.OPEN;
-	},
-	disconnect: async () => {
-		// Noop
-		// TODO: Abort the request
-	},
-};
-
-/** List of all connection drivers. */
-export const CONN_DRIVERS: Record<ConnDriverKind, ConnDriver<unknown>> = {
-	[ConnDriverKind.WEBSOCKET]: WEBSOCKET_DRIVER,
-	[ConnDriverKind.SSE]: SSE_DRIVER,
-	[ConnDriverKind.HTTP]: HTTP_DRIVER,
-};
-
-export function getConnDriverKindFromState(
-	state: ConnDriverState,
-): ConnDriverKind {
-	if (ConnDriverKind.WEBSOCKET in state) return ConnDriverKind.WEBSOCKET;
-	else if (ConnDriverKind.SSE in state) return ConnDriverKind.SSE;
-	else if (ConnDriverKind.HTTP in state) return ConnDriverKind.HTTP;
-	else assertUnreachable(state);
-}

package/src/actor/conn-socket.ts

@@ -1,8 +0,0 @@
-import type { ConnDriverState } from "./conn-drivers";
-
-export interface ConnSocket {
-	requestId: string;
-	requestIdBuf?: ArrayBuffer;
-	hibernatable: boolean;
-	driverState: ConnDriverState;
-}

package/src/actor/conn.ts

@@ -1,272 +0,0 @@
-import * as cbor from "cbor-x";
-import invariant from "invariant";
-import { PersistedHibernatableWebSocket } from "@/schemas/actor-persist/mod";
-import type * as protocol from "@/schemas/client-protocol/mod";
-import { TO_CLIENT_VERSIONED } from "@/schemas/client-protocol/versioned";
-import { arrayBuffersEqual, bufferToArrayBuffer } from "@/utils";
-import {
-	CONN_DRIVERS,
-	ConnDriverKind,
-	type ConnDriverState,
-	ConnReadyState,
-	getConnDriverKindFromState,
-} from "./conn-drivers";
-import type { ConnSocket } from "./conn-socket";
-import type { AnyDatabaseProvider } from "./database";
-import * as errors from "./errors";
-import { type ActorInstance, PERSIST_SYMBOL } from "./instance";
-import type { PersistedConn } from "./persisted";
-import { CachedSerializer } from "./protocol/serde";
-import { generateSecureToken } from "./utils";
-
-export function generateConnId(): string {
-	return crypto.randomUUID();
-}
-
-export function generateConnToken(): string {
-	return generateSecureToken(32);
-}
-
-export function generateConnRequestId(): string {
-	return crypto.randomUUID();
-}
-
-export type ConnId = string;
-
-export type AnyConn = Conn<any, any, any, any, any, any>;
-
-export type ConnectionStatus = "connected" | "reconnecting";
-
-/**
- * Represents a client connection to a actor.
- *
- * Manages connection-specific data and controls the connection lifecycle.
- *
- * @see {@link https://rivet.dev/docs/connections|Connection Documentation}
- */
-export class Conn<S, CP, CS, V, I, DB extends AnyDatabaseProvider> {
-	subscriptions: Set<string> = new Set<string>();
-
-	// TODO: Remove this cyclical reference
-	#actor: ActorInstance<S, CP, CS, V, I, DB>;
-
-	/**
-	 * The proxied state that notifies of changes automatically.
-	 *
-	 * Any data that should be stored indefinitely should be held within this object.
-	 */
-	__persist: PersistedConn<CP, CS>;
-
-	get __driverState(): ConnDriverState | undefined {
-		return this.__socket?.driverState;
-	}
-
-	/**
-	 * Socket connected to this connection.
-	 *
-	 * If undefined, then nothing is connected to this.
-	 */
-	__socket?: ConnSocket;
-
-	get __status(): ConnectionStatus {
-		// TODO: isHibernatible might be true while the actual hibernatable websocket has disconnected
-		if (this.__socket || this.isHibernatable) {
-			return "connected";
-		} else {
-			return "reconnecting";
-		}
-	}
-
-	public get params(): CP {
-		return this.__persist.params;
-	}
-
-	public get __stateEnabled() {
-		return this.#actor.connStateEnabled;
-	}
-
-	/**
-	 * Gets the current state of the connection.
-	 *
-	 * Throws an error if the state is not enabled.
-	 */
-	public get state(): CS {
-		this.#validateStateEnabled();
-		if (!this.__persist.state) throw new Error("state should exists");
-		return this.__persist.state;
-	}
-
-	/**
-	 * Sets the state of the connection.
-	 *
-	 * Throws an error if the state is not enabled.
-	 */
-	public set state(value: CS) {
-		this.#validateStateEnabled();
-		this.__persist.state = value;
-	}
-
-	/**
-	 * Unique identifier for the connection.
-	 */
-	public get id(): ConnId {
-		return this.__persist.connId;
-	}
-
-	/**
-	 * Token used to authenticate this request.
-	 */
-	public get _token(): string {
-		return this.__persist.token;
-	}
-
-	/**
-	 * Status of the connection.
-	 */
-	public get status(): ConnectionStatus {
-		return this.__status;
-	}
-
-	/**
-	 * @experimental
-	 *
-	 * If the underlying connection can hibernate.
-	 */
-	public get isHibernatable(): boolean {
-		if (!this.__persist.hibernatableRequestId) {
-			return false;
-		}
-		return (
-			this.#actor[PERSIST_SYMBOL].hibernatableWebSocket.findIndex((x) =>
-				arrayBuffersEqual(
-					x.requestId,
-					this.__persist.hibernatableRequestId!,
-				),
-			) > -1
-		);
-	}
-
-	/**
-	 * Timestamp of the last time the connection was seen, i.e. the last time the connection was active and checked for liveness.
-	 */
-	public get lastSeen(): number {
-		return this.__persist.lastSeen;
-	}
-
-	/**
-	 * Initializes a new instance of the Connection class.
-	 *
-	 * This should only be constructed by {@link Actor}.
-	 *
-	 * @protected
-	 */
-	public constructor(
-		actor: ActorInstance<S, CP, CS, V, I, DB>,
-		persist: PersistedConn<CP, CS>,
-	) {
-		this.#actor = actor;
-		this.__persist = persist;
-	}
-
-	#validateStateEnabled() {
-		if (!this.__stateEnabled) {
-			throw new errors.ConnStateNotEnabled();
-		}
-	}
-
-	/**
-	 * Sends a WebSocket message to the client.
-	 *
-	 * @param message - The message to send.
-	 *
-	 * @protected
-	 */
-	public _sendMessage(message: CachedSerializer<protocol.ToClient>) {
-		if (this.__driverState) {
-			const driverKind = getConnDriverKindFromState(this.__driverState);
-			const driver = CONN_DRIVERS[driverKind];
-			if (driver.sendMessage) {
-				driver.sendMessage(
-					this.#actor,
-					this,
-					(this.__driverState as any)[driverKind],
-					message,
-				);
-			} else {
-				this.#actor.rLog.debug({
-					msg: "conn driver does not support sending messages",
-					conn: this.id,
-				});
-			}
-		} else {
-			this.#actor.rLog.warn({
-				msg: "missing connection driver state for send message",
-				conn: this.id,
-			});
-		}
-	}
-
-	/**
-	 * Sends an event with arguments to the client.
-	 *
-	 * @param eventName - The name of the event.
-	 * @param args - The arguments for the event.
-	 * @see {@link https://rivet.dev/docs/events|Events Documentation}
-	 */
-	public send(eventName: string, ...args: unknown[]) {
-		this.#actor.inspector.emitter.emit("eventFired", {
-			type: "event",
-			eventName,
-			args,
-			connId: this.id,
-		});
-		this._sendMessage(
-			new CachedSerializer<protocol.ToClient>(
-				{
-					body: {
-						tag: "Event",
-						val: {
-							name: eventName,
-							args: bufferToArrayBuffer(cbor.encode(args)),
-						},
-					},
-				},
-				TO_CLIENT_VERSIONED,
-			),
-		);
-	}
-
-	/**
-	 * Disconnects the client with an optional reason.
-	 *
-	 * @param reason - The reason for disconnection.
-	 */
-	public async disconnect(reason?: string) {
-		if (this.__socket && this.__driverState) {
-			const driverKind = getConnDriverKindFromState(this.__driverState);
-			const driver = CONN_DRIVERS[driverKind];
-			if (driver.disconnect) {
-				driver.disconnect(
-					this.#actor,
-					this,
-					(this.__driverState as any)[driverKind],
-					reason,
-				);
-			} else {
-				this.#actor.rLog.debug({
-					msg: "no disconnect handler for conn driver",
-					conn: this.id,
-				});
-			}
-
-			this.#actor.__connDisconnected(this, true, this.__socket.requestId);
-		} else {
-			this.#actor.rLog.warn({
-				msg: "missing connection driver state for disconnect",
-				conn: this.id,
-			});
-		}
-
-		this.__socket = undefined;
-	}
-}