@rivetkit/rivetkit-native 0.0.0-pr.4652.130bc98 → 0.0.0-pr.4653.7179278

package/index.d.ts

@@ -48,6 +48,10 @@ export interface JsKvEntry {
 export interface HibernatingRequestEntry {
   gatewayId: Buffer
   requestId: Buffer
+  envoyMessageIndex: number
+  rivetMessageIndex: number
+  path: string
+  headers?: Record<string, string>
 }
 /**
  * Start the native envoy client synchronously.

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/rivetkit-native",
-	"version": "0.0.0-pr.4652.130bc98",
+	"version": "0.0.0-pr.4653.7179278",
 	"description": "Native N-API addon for RivetKit providing envoy client and SQLite access",
 	"license": "Apache-2.0",
 	"main": "index.js",
@@ -49,15 +49,15 @@
 	},
 	"dependencies": {
 		"@napi-rs/cli": "^2.18.4",
-		"@rivetkit/engine-envoy-protocol": "0.0.0-pr.4652.130bc98"
+		"@rivetkit/engine-envoy-protocol": "0.0.0-pr.4653.7179278"
 	},
 	"optionalDependencies": {
-		"@rivetkit/rivetkit-native-darwin-arm64": "0.0.0-pr.4652.130bc98",
-		"@rivetkit/rivetkit-native-darwin-x64": "0.0.0-pr.4652.130bc98",
-		"@rivetkit/rivetkit-native-linux-arm64-gnu": "0.0.0-pr.4652.130bc98",
-		"@rivetkit/rivetkit-native-linux-arm64-musl": "0.0.0-pr.4652.130bc98",
-		"@rivetkit/rivetkit-native-linux-x64-gnu": "0.0.0-pr.4652.130bc98",
-		"@rivetkit/rivetkit-native-linux-x64-musl": "0.0.0-pr.4652.130bc98",
-		"@rivetkit/rivetkit-native-win32-x64-msvc": "0.0.0-pr.4652.130bc98"
+		"@rivetkit/rivetkit-native-darwin-arm64": "0.0.0-pr.4653.7179278",
+		"@rivetkit/rivetkit-native-darwin-x64": "0.0.0-pr.4653.7179278",
+		"@rivetkit/rivetkit-native-linux-arm64-gnu": "0.0.0-pr.4653.7179278",
+		"@rivetkit/rivetkit-native-linux-arm64-musl": "0.0.0-pr.4653.7179278",
+		"@rivetkit/rivetkit-native-linux-x64-gnu": "0.0.0-pr.4653.7179278",
+		"@rivetkit/rivetkit-native-linux-x64-musl": "0.0.0-pr.4653.7179278",
+		"@rivetkit/rivetkit-native-win32-x64-msvc": "0.0.0-pr.4653.7179278"
 	}
 }

package/wrapper.js

@@ -98,6 +98,10 @@ function wrapHandle(jsHandle) {
 			const requests = (metaEntries || []).map((e) => ({
 				gatewayId: Buffer.from(e.gatewayId),
 				requestId: Buffer.from(e.requestId),
+				envoyMessageIndex: e.envoyMessageIndex ?? 0,
+				rivetMessageIndex: e.rivetMessageIndex ?? 0,
+				path: e.path ?? "",
+				headers: e.headers ?? {},
 			}));
 			jsHandle.restoreHibernatingRequests(actorId, requests);
 		},
@@ -308,6 +312,46 @@ function handleEvent(event, config, wrappedHandle) {
 			);
 			break;
 		}
+		case "websocket_can_hibernate": {
+			const gatewayId = Buffer.from(event.gatewayId);
+			const requestId = Buffer.from(event.requestId);
+			const headers = new Headers(event.headers || {});
+			headers.set("Upgrade", "websocket");
+			headers.set("Connection", "Upgrade");
+			const request = new Request(`http://actor${event.path}`, {
+				method: event.method,
+				headers,
+			});
+
+			Promise.resolve(
+				config.hibernatableWebSocket
+					? config.hibernatableWebSocket.canHibernate(
+						event.actorId,
+						gatewayId,
+						requestId,
+						request,
+					)
+					: false,
+			).then(
+				async (canHibernate) => {
+					if (handle._raw) {
+						await handle._raw.respondCallback(event.responseId, {
+							canHibernate: Boolean(canHibernate),
+						});
+					}
+				},
+				async (err) => {
+					console.error("canHibernate error:", err);
+					if (handle._raw) {
+						await handle._raw.respondCallback(event.responseId, {
+							canHibernate: false,
+							error: String(err),
+						});
+					}
+				},
+			);
+			break;
+		}
 		case "websocket_open": {
 			if (config.websocket) {
 				const messageId = Buffer.from(event.messageId);