File size: 1,672 Bytes
0ad74ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import type { WorkerProxy } from "@gradio/wasm";
import { is_self_host } from "@gradio/wasm/network";
/**
* A fetch() function that proxies HTTP requests to the worker,
* which also falls back to the original fetch() for external resource requests.
*/
/**
*
* @param { WorkerProxy } workerProxy The worker proxy
* @param { RequestInfo | URL} input The request info
* @param {RequestInit} init The request init
* @returns {Promise<Response>} The response
*/
export async function wasm_proxied_fetch(
workerProxy: WorkerProxy,
input: RequestInfo | URL,
init?: RequestInit
): Promise<Response> {
console.debug("wasm_proxied_fetch", input, init);
const request = new Request(input, init);
const url = new URL(request.url);
if (!is_self_host(url)) {
console.debug("Fallback to original fetch");
return fetch(input, init);
}
const method = request.method;
if (
method !== "GET" &&
method !== "POST" &&
method !== "PUT" &&
method !== "DELETE"
) {
throw new Error(`Unsupported method: ${method}`);
}
const headers: Parameters<WorkerProxy["httpRequest"]>[0]["headers"] = {};
request.headers.forEach((value, key) => {
headers[key] = value;
});
const body = request.body ?? new Uint8Array(await request.arrayBuffer()); // request.body can't be read on FireFox, so fallback to arrayBuffer().
const response = await workerProxy.httpRequest({
path: url.pathname,
query_string: url.searchParams.toString(), // The `query_string` field in the ASGI spec must not contain the leading `?`.
method,
headers,
body
});
return new Response(response.body, {
status: response.status,
headers: new Headers(response.headers)
});
}
|