rivetkit 2.3.6 → 2.3.8

package/dist/browser/client.d.ts

@@ -1765,6 +1765,12 @@ interface RuntimeHttpRequest {
     headers?: Record<string, string>;
     body?: RuntimeBytes;
 }
+interface RuntimeHttpResponseBodyStream {
+    cancelled(): Promise<void>;
+    write(chunk: Uint8Array): Promise<void>;
+    end(): Promise<void>;
+    error(message: string): Promise<void>;
+}
 interface RuntimeStateDeltaPayload {
     state?: RuntimeBytes;
     connHibernation: Array<{
@@ -1987,6 +1993,7 @@ interface RuntimeListenerConfig {
     port: number;
     host?: string;
     publicDir?: string;
+    application?: RuntimeApplicationFetch;
 }
 interface RuntimeServerlessRequest {
     method: string;
@@ -1994,6 +2001,19 @@ interface RuntimeServerlessRequest {
     headers: Record<string, string>;
     body: RuntimeBytes;
 }
+interface RuntimeApplicationRequest extends RuntimeServerlessRequest {
+    cancelToken?: CancellationTokenHandle;
+}
+interface RuntimeApplicationResponse {
+    status: number;
+    headers: Record<string, string>;
+    body?: RuntimeBytes;
+    stream?: boolean;
+}
+type RuntimeApplicationFetch = (request: RuntimeApplicationRequest, responseBodyStream?: RuntimeHttpResponseBodyStream) => Promise<RuntimeApplicationResponse>;
+interface RuntimeApplicationListenerConfig extends RuntimeListenerConfig {
+    application: RuntimeApplicationFetch;
+}
 interface RuntimeServerlessResponseHead {
     status: number;
     headers: Record<string, string>;
@@ -2036,6 +2056,7 @@ interface CoreRuntime {
     registryActorStopThresholdMs?(registry: RegistryHandle): Promise<number | undefined>;
     handleServerlessRequest(registry: RegistryHandle, req: RuntimeServerlessRequest, onStreamEvent: RuntimeServerlessStreamCallback, cancelToken: CancellationTokenHandle, config: RuntimeServeConfig): Promise<RuntimeServerlessResponseHead>;
     serveListener(registry: RegistryHandle, listener: RuntimeListenerConfig, config: RuntimeServeConfig): Promise<void>;
+    serveApplicationListener(registry: RegistryHandle, listener: RuntimeApplicationListenerConfig, config: RuntimeServeConfig): Promise<void>;
     registryHealth?(registry: RegistryHandle): Promise<RuntimeRegistryRouteResponse>;
     registryMetadata?(registry: RegistryHandle): Promise<RuntimeRegistryRouteResponse>;
     registryMetrics?(registry: RegistryHandle): Promise<RuntimeRegistryRouteResponse>;
@@ -2146,6 +2167,15 @@ type FetchHandler = (request: Request, ...args: any) => Response | Promise<Respo
 interface ServerlessHandler {
     fetch: FetchHandler;
 }
+interface RegistryApplication {
+    fetch: FetchHandler;
+}
+interface RegistryListenOptions {
+    port?: number;
+    host?: string;
+    publicDir?: string;
+    application?: RegistryApplication;
+}
 interface RegistryRoutes {
     health(): Promise<Response>;
     metadata(): Promise<Response>;
@@ -2195,18 +2225,16 @@ declare class Registry<A extends RegistryActors> {
      * @param opts.host      Address to bind. Defaults to `0.0.0.0`.
      * @param opts.publicDir If set, serves static files from this directory
      *                       as a fallback below the framework routes.
+     * @param opts.application If set, handles requests that do not match a
+     *                         framework route.
      *
      * @example
      * ```ts
      * await registry.listen();
-     * await registry.listen({ port: 8080, publicDir: "./public" });
+     * await registry.listen({ application: app });
      * ```
      */
-    listen(opts?: {
-        port?: number;
-        host?: string;
-        publicDir?: string;
-    }): Promise<void>;
+    listen(opts?: RegistryListenOptions): Promise<void>;
     /**
      * Gracefully drains all live registries.
      *

package/dist/browser/client.js

@@ -194,7 +194,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "2.3.6",
+  version: "2.3.8",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -729,7 +729,10 @@ function jsonStringifyCompat(input, space) {
         return [JSON_COMPAT_BIGINT, value.toString()];
       }
       if (value instanceof ArrayBuffer) {
-        return [JSON_COMPAT_ARRAY_BUFFER, base64EncodeArrayBuffer(value)];
+        return [
+          JSON_COMPAT_ARRAY_BUFFER,
+          base64EncodeArrayBuffer(value)
+        ];
       }
       if (value instanceof Uint8Array) {
         return [JSON_COMPAT_UINT8_ARRAY, base64EncodeUint8Array(value)];
@@ -1109,6 +1112,7 @@ var PATH_WEBSOCKET_BASE = "/websocket";
 var PATH_WEBSOCKET_PREFIX = "/websocket/";
 var HEADER_ENCODING = "x-rivet-encoding";
 var HEADER_CONN_PARAMS = "x-rivet-conn-params";
+var HEADER_ORIGINAL_REQUEST_URL = "x-rivet-internal-original-request-url";
 var HEADER_RIVET_TOKEN = "x-rivet-token";
 var HEADER_RIVET_TARGET = "x-rivet-target";
 var HEADER_RIVET_ACTOR = "x-rivet-actor";
@@ -1135,23 +1139,19 @@ async function sendHttpRequestToGateway(runConfig, gatewayUrl, actorRequest, opt
     if (actorRequest.bodyUsed) {
       throw new Error("Request body has already been consumed");
     }
-    const reqBody = await actorRequest.arrayBuffer();
-    if (reqBody.byteLength !== 0) {
-      bodyToSend = reqBody;
+    if (actorRequest.body) {
+      bodyToSend = actorRequest.body;
       guardHeaders.delete("transfer-encoding");
       guardHeaders.delete("content-length");
     }
   }
-  const guardRequest = new Request(gatewayUrl, {
+  return fetch(gatewayUrl, {
     method: actorRequest.method,
     headers: guardHeaders,
     body: bodyToSend,
-    signal: actorRequest.signal
+    signal: actorRequest.signal,
+    ...bodyToSend ? { duplex: "half" } : {}
   });
-  return mutableResponse(await fetch(guardRequest));
-}
-function mutableResponse(fetchRes) {
-  return new Response(fetchRes.body, fetchRes);
 }
 function buildGuardHeaders(runConfig, actorRequest, options) {
   const headers = new Headers();
@@ -5824,6 +5824,17 @@ var ActorConnRaw = class {
   }
 };
 
+// src/common/fetch-like.ts
+function isUrlLike(value) {
+  return typeof value === "object" && value !== null && typeof value.href === "string" && typeof value.pathname === "string" && typeof value.search === "string";
+}
+function isRequestLike(value) {
+  return typeof value === "object" && value !== null && typeof value.url === "string" && typeof value.method === "string" && isHeadersLike(value.headers);
+}
+function isHeadersLike(value) {
+  return typeof value === "object" && value !== null && typeof value.entries === "function";
+}
+
 // src/common/database/shared.ts
 var MIGRATION_TRANSACTION_TIMEOUT_MS = 5 * 6e4;
 var AsyncMutex = class {
@@ -5855,14 +5866,17 @@ var AsyncMutex = class {
 // src/client/raw-utils.ts
 async function rawHttpFetch(driver, target, params, input, init, options = {}) {
   let path;
+  let originalUrl;
   let mergedInit = init || {};
   if (typeof input === "string") {
     path = input;
-  } else if (input instanceof URL) {
+  } else if (isUrlLike(input)) {
     path = input.pathname + input.search;
-  } else if (input instanceof Request) {
+    originalUrl = input.toString();
+  } else if (isRequestLike(input)) {
     const url = new URL(input.url);
     path = url.pathname + url.search;
+    originalUrl = url.toString();
     const requestHeaders = new Headers(input.headers);
     const initHeaders = new Headers(init?.headers || {});
     const mergedHeaders = new Headers(requestHeaders);
@@ -5904,6 +5918,10 @@ async function rawHttpFetch(driver, target, params, input, init, options = {}) {
     const normalizedPath = path.startsWith("/") ? path.slice(1) : path;
     const url = new URL(`http://actor/request/${normalizedPath}`);
     const proxyRequestHeaders = new Headers(mergedInit.headers);
+    proxyRequestHeaders.delete(HEADER_ORIGINAL_REQUEST_URL);
+    if (originalUrl) {
+      proxyRequestHeaders.set(HEADER_ORIGINAL_REQUEST_URL, originalUrl);
+    }
     if (params) {
       proxyRequestHeaders.set(HEADER_CONN_PARAMS, JSON.stringify(params));
     }
@@ -6376,6 +6394,7 @@ var ActorHandleRaw = class {
   async #fetchWithResolvedActor(input, init) {
     const maxAttempts = this.#getDynamicQueryMaxAttempts();
     let useQueryTarget = false;
+    const retryableRequest = isRequestLike(input) ? input.clone() : void 0;
     const { skipReadyWait, ...requestInit } = init ?? {};
     const gatewayOptions = resolveActorGatewayOptions(
       this.#gatewayOptions,
@@ -6395,7 +6414,7 @@ var ActorHandleRaw = class {
           this.#driver,
           target,
           this.#params,
-          input,
+          retryableRequest ? retryableRequest.clone() : input,
           requestInit,
           gatewayOptions
         );