mirror of
https://github.com/JonasunderscoreJones/api.jonasjones.dev.git
synced 2025-10-23 20:19:19 +02:00
59 lines
1.8 KiB
TypeScript
59 lines
1.8 KiB
TypeScript
// @ts-nocheck
|
|
/// <reference path="middleware-multiworker-dev.d.ts"/>
|
|
|
|
import { workers } from "config:middleware/multiworker-dev";
|
|
import type { WorkerRegistry } from "../../src/dev-registry";
|
|
|
|
export function wrap(env: Record<string, unknown>) {
|
|
const facadeEnv = { ...env };
|
|
// For every Worker definition that's available,
|
|
// create a fetcher for it on the facade env.
|
|
// for const [name, binding] of env
|
|
// if Workers[name]
|
|
// const details = Workers[name];
|
|
|
|
for (const [name, details] of Object.entries(workers as WorkerRegistry)) {
|
|
if (details) {
|
|
facadeEnv[name] = {
|
|
async fetch(...reqArgs: Parameters<Fetcher["fetch"]>) {
|
|
const reqFromArgs = new Request(...reqArgs);
|
|
if (details.headers) {
|
|
for (const [key, value] of Object.entries(details.headers)) {
|
|
// In remote mode, you need to add a couple of headers
|
|
// to make sure it's talking to the 'dev' preview session
|
|
// (much like wrangler dev already does via proxy.ts)
|
|
reqFromArgs.headers.set(key, value);
|
|
}
|
|
return (env[name] as Fetcher).fetch(reqFromArgs);
|
|
}
|
|
|
|
const url = new URL(reqFromArgs.url);
|
|
if (details.protocol !== undefined) {
|
|
url.protocol = details.protocol;
|
|
}
|
|
if (details.host !== undefined) {
|
|
url.host = details.host;
|
|
}
|
|
if (details.port !== undefined) {
|
|
url.port = details.port.toString();
|
|
}
|
|
|
|
const request = new Request(url.toString(), reqFromArgs);
|
|
return fetch(request);
|
|
},
|
|
};
|
|
} else {
|
|
// This means there's no dev binding available.
|
|
// Let's use whatever's available, or put a shim with a message.
|
|
facadeEnv[name] = facadeEnv[name] || {
|
|
async fetch() {
|
|
return new Response(
|
|
`You should start up wrangler dev --local on the ${name} worker`,
|
|
{ status: 404 }
|
|
);
|
|
},
|
|
};
|
|
}
|
|
}
|
|
return facadeEnv;
|
|
}
|