Initial commit (by create-cloudflare CLI)

1 files changed, 59 insertions, 0 deletions

diff --git a/node_modules/wrangler/templates/middleware/middleware-multiworker-dev.ts b/node_modules/wrangler/templates/middleware/middleware-multiworker-dev.ts
new file mode 100644
index 0000000..fee33d5
--- /dev/null
+++ b/node_modules/wrangler/templates/middleware/middleware-multiworker-dev.ts
@@ -0,0 +1,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;
+}