chore(rivetkit): add support for conn params on onRequest and onWebSocket

Author: NathanFlurry

rivetkit-typescript/packages/rivetkit/src/actor/router-endpoints.ts

@@ -335,19 +335,21 @@ export async function handleAction(
 }
 
 export async function handleRawRequest(
+	c: HonoContext,
 	req: Request,
 	actorDriver: ActorDriver,
 	actorId: string,
 ): Promise<Response> {
 	const actor = await actorDriver.loadActor(actorId);
+	const parameters = getRequestConnParams(c.req);
 
 	// Track connection outside of scope for cleanup
 	let createdConn: AnyConn | undefined;
 
 	try {
 		const conn = await actor.connectionManager.prepareAndConnectConn(
 			createRawRequestSocket(),
-			{},
+			parameters,
 			req,
 		);
 
@@ -368,6 +370,7 @@ export async function handleRawWebSocket(
 	actorDriver: ActorDriver,
 	actorId: string,
 	requestIdBuf: ArrayBuffer | undefined,
+	connParams: unknown | undefined,
 ): Promise<UpgradeWebSocketArgs> {
 	const exposeInternalError = req
 		? getRequestExposeInternalError(req)
@@ -414,7 +417,7 @@ export async function handleRawWebSocket(
 		);
 		const conn = await actor.connectionManager.prepareAndConnectConn(
 			driver,
-			{},
+			connParams ?? {},
 			newRequest,
 		);
 		createdConn = conn;

rivetkit-typescript/packages/rivetkit/src/actor/router.ts

@@ -110,7 +110,6 @@ export function createActorRouter(
 		const upgradeWebSocket = runConfig.getUpgradeWebSocket?.();
 		if (upgradeWebSocket) {
 			return upgradeWebSocket(async (c) => {
-				// Parse configuration from Sec-WebSocket-Protocol header
 				const protocols = c.req.header("sec-websocket-protocol");
 				const { encoding, connParams } =
 					parseWebSocketProtocols(protocols);
@@ -165,6 +164,7 @@ export function createActorRouter(
 		});
 
 		return await handleRawRequest(
+			c,
 			correctedRequest,
 			actorDriver,
 			c.env.actorId,
@@ -178,12 +178,16 @@ export function createActorRouter(
 				const url = new URL(c.req.url);
 				const pathWithQuery = c.req.path + url.search;
 
+				const protocols = c.req.header("sec-websocket-protocol");
+				const { connParams } = parseWebSocketProtocols(protocols);
+
 				loggerWithoutContext().debug({
 					msg: "actor router raw websocket",
 					path: c.req.path,
 					url: c.req.url,
 					search: url.search,
 					pathWithQuery,
+					connParams,
 				});
 
 				return await handleRawWebSocket(
@@ -192,6 +196,7 @@ export function createActorRouter(
 					actorDriver,
 					c.env.actorId,
 					undefined,
+					connParams,
 				);
 			})(c, noopNext());
 		} else {

rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts

@@ -562,6 +562,7 @@ export class EngineActorDriver implements ActorDriver {
 				this,
 				actorId,
 				requestIdBuf,
+				connParams,
 			);
 		} else {
 			throw new Error(`Unreachable path: ${url.pathname}`);

rivetkit-typescript/packages/rivetkit/src/drivers/file-system/manager.ts

@@ -189,6 +189,7 @@ export class FileSystemManagerDriver implements ManagerDriver {
 				this.#actorDriver,
 				actorId,
 				undefined,
+				params,
 			);
 			return new InlineWebSocketAdapter2(wsHandler);
 		} else {
@@ -246,6 +247,7 @@ export class FileSystemManagerDriver implements ManagerDriver {
 				this.#actorDriver,
 				actorId,
 				undefined,
+				connParams,
 			);
 			return upgradeWebSocket(() => wsHandler)(c, noopNext());
 		} else {