blob: fee33d5a8c4e50e165c6849aeba2a376d7ea0cd1 (
plain)
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
|
// @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;
}
|
