wrangler 0.0.0-e6e1daba → 0.0.0-e72a5794

package/src/pages/dev.ts

@@ -1,716 +0,0 @@
-import { execSync, spawn } from "node:child_process";
-import { existsSync, readFileSync } from "node:fs";
-import { homedir, tmpdir } from "node:os";
-import { join, resolve } from "node:path";
-import { watch } from "chokidar";
-import * as esbuild from "esbuild";
-import { unstable_dev } from "../api";
-import { esbuildAliasExternalPlugin } from "../bundle";
-import { FatalError } from "../errors";
-import { logger } from "../logger";
-import * as metrics from "../metrics";
-import { getBasePath } from "../paths";
-import { buildFunctions } from "./buildFunctions";
-import { ROUTES_SPEC_VERSION, SECONDS_TO_WAIT_FOR_PROXY } from "./constants";
-import { FunctionsNoRoutesError, getFunctionsNoRoutesWarning } from "./errors";
-import { buildRawWorker, checkRawWorker } from "./functions/buildWorker";
-import { validateRoutes } from "./functions/routes-validation";
-import { CLEANUP, CLEANUP_CALLBACKS, pagesBetaWarning } from "./utils";
-import type { AdditionalDevProps } from "../dev";
-import type {
-	CommonYargsArgv,
-	StrictYargsOptionsToInterface,
-} from "../yargs-types";
-import type { RoutesJSONSpec } from "./functions/routes-transformation";
-
-const DURABLE_OBJECTS_BINDING_REGEXP = new RegExp(
-	/^(?<binding>[^=]+)=(?<className>[^@\s]+)(@(?<scriptName>.*)$)?$/
-);
-
-export function Options(yargs: CommonYargsArgv) {
-	return yargs
-		.positional("directory", {
-			type: "string",
-			demandOption: undefined,
-			description: "The directory of static assets to serve",
-		})
-		.positional("command", {
-			type: "string",
-			demandOption: undefined,
-			description: "The proxy command to run",
-		})
-		.options({
-			local: {
-				type: "boolean",
-				default: true,
-				description: "Run on my machine",
-			},
-			"compatibility-date": {
-				describe: "Date to use for compatibility checks",
-				type: "string",
-			},
-			"compatibility-flags": {
-				describe: "Flags to use for compatibility checks",
-				alias: "compatibility-flag",
-				type: "string",
-				array: true,
-			},
-			ip: {
-				type: "string",
-				default: "0.0.0.0",
-				description: "The IP address to listen on",
-			},
-			port: {
-				type: "number",
-				default: 8788,
-				description: "The port to listen on (serve from)",
-			},
-			"inspector-port": {
-				type: "number",
-				describe: "Port for devtools to connect to",
-			},
-			proxy: {
-				type: "number",
-				description: "The port to proxy (where the static assets are served)",
-			},
-			"script-path": {
-				type: "string",
-				default: "_worker.js",
-				description:
-					"The location of the single Worker script if not using functions",
-			},
-			bundle: {
-				type: "boolean",
-				default: undefined,
-				hidden: true,
-			},
-			"no-bundle": {
-				type: "boolean",
-				default: true,
-				description: "Whether to run bundling on `_worker.js`",
-			},
-			"experimental-worker-bundle": {
-				type: "boolean",
-				default: false,
-				hidden: true,
-				description:
-					"Whether to process non-JS module imports or not, such as wasm/text/binary, when we run bundling on `functions` or `_worker.js`",
-			},
-			binding: {
-				type: "array",
-				description: "Bind variable/secret (KEY=VALUE)",
-				alias: "b",
-			},
-			kv: {
-				type: "array",
-				description: "KV namespace to bind (--kv KV_BINDING)",
-				alias: "k",
-			},
-			d1: {
-				type: "array",
-				description: "D1 database to bind",
-			},
-			do: {
-				type: "array",
-				description: "Durable Object to bind (--do NAME=CLASS)",
-				alias: "o",
-			},
-			r2: {
-				type: "array",
-				description: "R2 bucket to bind (--r2 R2_BINDING)",
-			},
-			"live-reload": {
-				type: "boolean",
-				default: false,
-				description: "Auto reload HTML pages when change is detected",
-			},
-			"local-protocol": {
-				describe: "Protocol to listen to requests on, defaults to http.",
-				choices: ["http", "https"] as const,
-			},
-			"experimental-enable-local-persistence": {
-				describe:
-					"Enable persistence for local mode (deprecated, use --persist)",
-				type: "boolean",
-				deprecated: true,
-				hidden: true,
-			},
-			persist: {
-				describe:
-					"Enable persistence for local mode, using default path: .wrangler/state",
-				type: "boolean",
-			},
-			"persist-to": {
-				describe:
-					"Specify directory to use for local persistence (implies --persist)",
-				type: "string",
-				requiresArg: true,
-			},
-			"node-compat": {
-				describe: "Enable node.js compatibility",
-				default: false,
-				type: "boolean",
-				hidden: true,
-			},
-			"experimental-local": {
-				describe: "Run on my machine using the Cloudflare Workers runtime",
-				type: "boolean",
-				default: false,
-			},
-			config: {
-				describe: "Pages does not support wrangler.toml",
-				type: "string",
-				hidden: true,
-			},
-			"log-level": {
-				choices: ["debug", "info", "log", "warn", "error", "none"] as const,
-				describe: "Specify logging level",
-			},
-		})
-		.epilogue(pagesBetaWarning);
-}
-
-export const Handler = async ({
-	local,
-	directory,
-	compatibilityDate,
-	compatibilityFlags,
-	ip,
-	port,
-	inspectorPort,
-	proxy: requestedProxyPort,
-	bundle,
-	noBundle,
-	experimentalWorkerBundle,
-	scriptPath: singleWorkerScriptPath,
-	binding: bindings = [],
-	kv: kvs = [],
-	do: durableObjects = [],
-	d1: d1s = [],
-	r2: r2s = [],
-	liveReload,
-	localProtocol,
-	experimentalEnableLocalPersistence,
-	persist,
-	persistTo,
-	nodeCompat,
-	experimentalLocal,
-	config: config,
-	_: [_pages, _dev, ...remaining],
-	logLevel,
-}: StrictYargsOptionsToInterface<typeof Options>) => {
-	// Beta message for `wrangler pages <commands>` usage
-	logger.log(pagesBetaWarning);
-
-	if (logLevel) {
-		logger.loggerLevel = logLevel;
-	}
-
-	if (!local) {
-		throw new FatalError("Only local mode is supported at the moment.", 1);
-	}
-
-	if (config) {
-		throw new FatalError("Pages does not support wrangler.toml", 1);
-	}
-
-	const command = remaining;
-
-	let proxyPort: number | undefined;
-
-	if (directory !== undefined && command.length > 0) {
-		throw new FatalError(
-			"Specify either a directory OR a proxy command, not both.",
-			1
-		);
-	} else if (directory === undefined) {
-		proxyPort = await spawnProxyProcess({
-			port: requestedProxyPort,
-			command,
-		});
-		if (proxyPort === undefined) return undefined;
-	} else {
-		directory = resolve(directory);
-	}
-
-	if (!compatibilityDate) {
-		const currentDate = new Date().toISOString().substring(0, 10);
-		logger.warn(
-			`No compatibility_date was specified. Using today's date: ${currentDate}.\n` +
-				"Pass it in your terminal:\n" +
-				"```\n" +
-				`--compatibility-date=${currentDate}\n` +
-				"```\n" +
-				"See https://developers.cloudflare.com/workers/platform/compatibility-dates/ for more information."
-		);
-		compatibilityDate = currentDate;
-	}
-
-	if (experimentalEnableLocalPersistence) {
-		logger.warn(
-			`--experimental-enable-local-persistence is deprecated.\n` +
-				`Move any existing data to .wrangler/state and use --persist, or\n` +
-				`use --persist-to=./wrangler-local-state to keep using the old path.`
-		);
-	}
-
-	let scriptReadyResolve: () => void;
-	const scriptReadyPromise = new Promise<void>(
-		(promiseResolve) => (scriptReadyResolve = promiseResolve)
-	);
-
-	const workerScriptPath =
-		directory !== undefined
-			? join(directory, singleWorkerScriptPath)
-			: singleWorkerScriptPath;
-	const usingWorkerScript = existsSync(workerScriptPath);
-
-	const functionsDirectory = "./functions";
-	let usingFunctions = !usingWorkerScript && existsSync(functionsDirectory);
-
-	let scriptPath = "";
-
-	if (usingWorkerScript) {
-		scriptPath = workerScriptPath;
-		let runBuild = async () => {
-			await checkRawWorker(workerScriptPath, () => scriptReadyResolve());
-		};
-
-		// TODO: Here lies a known bug. If you specify both `--bundle` and `--no-bundle`, this behavior is undefined and you will get unexpected results.
-		// There is no sane way to get the true value out of yargs, so here we are.
-		const enableBundling = (bundle ?? !noBundle) || experimentalWorkerBundle;
-		if (enableBundling) {
-			// We want to actually run the `_worker.js` script through the bundler
-			// So update the final path to the script that will be uploaded and
-			// change the `runBuild()` function to bundle the `_worker.js`.
-			scriptPath = join(tmpdir(), `./bundledWorker-${Math.random()}.mjs`);
-			runBuild = async () => {
-				try {
-					await buildRawWorker({
-						workerScriptPath,
-						outfile: scriptPath,
-						directory: directory ?? ".",
-						local: true,
-						sourcemap: true,
-						watch: true,
-						onEnd: () => scriptReadyResolve(),
-						experimentalWorkerBundle,
-					});
-				} catch (e: unknown) {
-					logger.warn("Failed to bundle _worker.js.", e);
-				}
-			};
-		}
-
-		await runBuild();
-		watch([scriptPath], {
-			persistent: true,
-			ignoreInitial: true,
-		}).on("all", async () => {
-			await runBuild();
-		});
-	} else if (usingFunctions) {
-		// Try to use Functions
-		scriptPath = join(tmpdir(), `./functionsWorker-${Math.random()}.mjs`);
-
-		if (nodeCompat) {
-			console.warn(
-				"Enabling node.js compatibility mode for builtins and globals. This is experimental and has serious tradeoffs. Please see https://github.com/ionic-team/rollup-plugin-node-polyfills/ for more details."
-			);
-		}
-
-		logger.log(`Compiling worker to "${scriptPath}"...`);
-		const onEnd = () => scriptReadyResolve();
-		try {
-			const buildFn = async () => {
-				await buildFunctions({
-					outfile: scriptPath,
-					functionsDirectory,
-					sourcemap: true,
-					watch: true,
-					onEnd,
-					buildOutputDirectory: directory,
-					nodeCompat,
-					local: true,
-					experimentalWorkerBundle,
-				});
-				await metrics.sendMetricsEvent("build pages functions");
-			};
-
-			await buildFn();
-			// If Functions found routes, continue using Functions
-			watch([functionsDirectory], {
-				persistent: true,
-				ignoreInitial: true,
-			}).on("all", async () => {
-				try {
-					await buildFn();
-				} catch (e) {
-					if (e instanceof FunctionsNoRoutesError) {
-						logger.warn(
-							getFunctionsNoRoutesWarning(functionsDirectory, "skipping")
-						);
-					} else {
-						throw e;
-					}
-				}
-			});
-		} catch (e) {
-			// If there are no Functions, then Pages will only serve assets.
-			if (e instanceof FunctionsNoRoutesError) {
-				logger.warn(
-					getFunctionsNoRoutesWarning(functionsDirectory, "skipping")
-				);
-				// Resolve anyway and run without Functions
-				onEnd();
-				// Turn off Functions
-				usingFunctions = false;
-			} else {
-				throw e;
-			}
-		}
-	}
-
-	// Depending on the result of building Functions, we may not actually be using
-	// Functions even if the directory exists.
-	if (!usingFunctions && !usingWorkerScript) {
-		// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-		scriptReadyResolve!();
-
-		logger.log("No functions. Shimming...");
-		scriptPath = resolve(getBasePath(), "templates/pages-shim.ts");
-	}
-
-	await scriptReadyPromise;
-
-	if (scriptPath === "") {
-		// Failed to get a script with or without Functions,
-		// something really bad must have happened.
-		throw new FatalError(
-			"Failed to start wrangler pages dev due to an unknown error",
-			1
-		);
-	}
-
-	let entrypoint = scriptPath;
-
-	// custom _routes.json apply only to Functions or Advanced Mode Pages projects
-	if (directory && (usingFunctions || usingWorkerScript)) {
-		const routesJSONPath = join(directory, "_routes.json");
-
-		if (existsSync(routesJSONPath)) {
-			let routesJSONContents: string;
-			const runBuild = async (
-				entrypointFile: string,
-				outfile: string,
-				routes: string
-			) => {
-				await esbuild.build({
-					entryPoints: [
-						resolve(getBasePath(), "templates/pages-dev-pipeline.ts"),
-					],
-					bundle: true,
-					sourcemap: true,
-					format: "esm",
-					plugins: [
-						esbuildAliasExternalPlugin({
-							__ENTRY_POINT__: entrypointFile,
-							"./pages-dev-util": resolve(
-								getBasePath(),
-								"templates/pages-dev-util.ts"
-							),
-						}),
-					],
-					outfile,
-					define: {
-						__ROUTES__: routes,
-					},
-				});
-			};
-
-			try {
-				// always run the routes validation first. If _routes.json is invalid we
-				// want to throw accordingly and exit.
-				routesJSONContents = readFileSync(routesJSONPath, "utf-8");
-				validateRoutes(JSON.parse(routesJSONContents), directory);
-
-				entrypoint = join(
-					tmpdir(),
-					`${Math.random().toString(36).slice(2)}.js`
-				);
-				await runBuild(scriptPath, entrypoint, routesJSONContents);
-			} catch (err) {
-				if (err instanceof FatalError) {
-					throw err;
-				} else {
-					throw new FatalError(
-						`Could not validate _routes.json at ${directory}: ${err}`,
-						1
-					);
-				}
-			}
-
-			watch([routesJSONPath], {
-				persistent: true,
-				ignoreInitial: true,
-			}).on("all", async () => {
-				try {
-					/**
-					 * Watch for _routes.json file changes and validate file each time.
-					 * If file is valid proceed to running the build.
-					 */
-					routesJSONContents = readFileSync(routesJSONPath, "utf-8");
-					validateRoutes(JSON.parse(routesJSONContents), directory as string);
-					await runBuild(scriptPath, entrypoint, routesJSONContents);
-				} catch (err) {
-					/**
-					 * If _routes.json is invalid, don't exit but instead fallback to a sensible default
-					 * and continue to serve the assets. At the same time make sure we warn users that we
-					 * we detected an invalid file and that we'll be using a default.
-					 * This basically equates to serving a Functions or _worker.js project as is,
-					 * without applying any additional routing rules on top.
-					 */
-					const error =
-						err instanceof FatalError
-							? err
-							: `Could not validate _routes.json at ${directory}: ${err}`;
-					const defaultRoutesJSONSpec: RoutesJSONSpec = {
-						version: ROUTES_SPEC_VERSION,
-						include: ["/*"],
-						exclude: [],
-					};
-
-					logger.error(error);
-					logger.warn(
-						`Falling back to the following _routes.json default: ${JSON.stringify(
-							defaultRoutesJSONSpec,
-							null,
-							2
-						)}`
-					);
-
-					routesJSONContents = JSON.stringify(defaultRoutesJSONSpec);
-					await runBuild(scriptPath, entrypoint, routesJSONContents);
-				}
-			});
-		}
-	}
-
-	const { stop, waitUntilExit } = await unstable_dev(entrypoint, {
-		ip,
-		port,
-		inspectorPort,
-		localProtocol,
-		compatibilityDate,
-		compatibilityFlags,
-		nodeCompat,
-		vars: Object.fromEntries(
-			bindings
-				.map((binding) => binding.toString().split("="))
-				.map(([key, ...values]) => [key, values.join("=")])
-		),
-		kv: kvs.map((binding) => ({
-			binding: binding.toString(),
-			id: "",
-		})),
-		durableObjects: durableObjects
-			.map((durableObject) => {
-				const { binding, className, scriptName } =
-					DURABLE_OBJECTS_BINDING_REGEXP.exec(durableObject.toString())
-						?.groups || {};
-
-				if (!binding || !className) {
-					logger.warn(
-						"Could not parse Durable Object binding:",
-						durableObject.toString()
-					);
-					return;
-				}
-
-				return {
-					name: binding,
-					class_name: className,
-					script_name: scriptName,
-				};
-			})
-			.filter(Boolean) as AdditionalDevProps["durableObjects"],
-		r2: r2s.map((binding) => {
-			return { binding: binding.toString(), bucket_name: "" };
-		}),
-		persist,
-		persistTo,
-		inspect: undefined,
-		logPrefix: "pages",
-		logLevel: logLevel ?? "warn",
-		experimental: {
-			d1Databases: d1s.map((binding) => ({
-				binding: binding.toString(),
-				database_id: "", // Required for types, but unused by dev
-				database_name: `local-${binding}`,
-			})),
-			disableExperimentalWarning: true,
-			enablePagesAssetsServiceBinding: {
-				proxyPort,
-				directory,
-			},
-			experimentalLocal,
-			liveReload,
-			forceLocal: true,
-			showInteractiveDevSession: undefined,
-			testMode: false,
-			watch: true,
-		},
-	});
-	await metrics.sendMetricsEvent("run pages dev");
-
-	CLEANUP_CALLBACKS.push(stop);
-
-	void waitUntilExit().then(() => {
-		CLEANUP();
-		process.exit(0);
-	});
-
-	process.on("exit", CLEANUP);
-	process.on("SIGINT", CLEANUP);
-	process.on("SIGTERM", CLEANUP);
-};
-
-function isWindows() {
-	return process.platform === "win32";
-}
-
-async function sleep(ms: number) {
-	await new Promise((promiseResolve) => setTimeout(promiseResolve, ms));
-}
-
-function getPids(pid: number) {
-	const pids: number[] = [pid];
-	let command: string, regExp: RegExp;
-
-	if (isWindows()) {
-		command = `wmic process where (ParentProcessId=${pid}) get ProcessId`;
-		regExp = new RegExp(/(\d+)/);
-	} else {
-		command = `pgrep -P ${pid}`;
-		regExp = new RegExp(/(\d+)/);
-	}
-
-	try {
-		const newPids = (
-			execSync(command)
-				.toString()
-				.split("\n")
-				.map((line) => line.match(regExp))
-				.filter((line) => line !== null) as RegExpExecArray[]
-		).map((match) => parseInt(match[1]));
-
-		pids.push(...newPids.map(getPids).flat());
-	} catch {}
-
-	return pids;
-}
-
-function getPort(pid: number) {
-	let command: string, regExp: RegExp;
-
-	if (isWindows()) {
-		const drive = homedir().split(":\\")[0];
-		command = drive + ":\\windows\\system32\\netstat.exe -nao";
-		regExp = new RegExp(`TCP\\s+.*:(\\d+)\\s+.*:\\d+\\s+LISTENING\\s+${pid}`);
-	} else {
-		command = "lsof -nPi";
-		regExp = new RegExp(`${pid}\\s+.*TCP\\s+.*:(\\d+)\\s+\\(LISTEN\\)`);
-	}
-
-	try {
-		const matches = execSync(command)
-			.toString()
-			.split("\n")
-			.map((line) => line.match(regExp))
-			.filter((line) => line !== null) as RegExpExecArray[];
-
-		const match = matches[0];
-		if (match) return parseInt(match[1]);
-	} catch (thrown) {
-		logger.error(
-			`Error scanning for ports of process with PID ${pid}: ${thrown}`
-		);
-	}
-}
-
-async function spawnProxyProcess({
-	port,
-	command,
-}: {
-	port?: number;
-	command: (string | number)[];
-}): Promise<undefined | number> {
-	if (command.length === 0) {
-		if (port !== undefined) {
-			return port;
-		}
-
-		CLEANUP();
-		throw new FatalError(
-			"Must specify a directory of static assets to serve or a command to run or a proxy port.",
-			1
-		);
-	}
-
-	logger.log(`Running ${command.join(" ")}...`);
-	const proxy = spawn(
-		command[0].toString(),
-		command.slice(1).map((value) => value.toString()),
-		{
-			shell: isWindows(),
-			env: {
-				BROWSER: "none",
-				...process.env,
-			},
-		}
-	);
-	CLEANUP_CALLBACKS.push(() => {
-		proxy.kill();
-	});
-
-	proxy.stdout.on("data", (data) => {
-		logger.log(`[proxy]: ${data}`);
-	});
-
-	proxy.stderr.on("data", (data) => {
-		logger.error(`[proxy]: ${data}`);
-	});
-
-	proxy.on("close", (code) => {
-		logger.error(`Proxy exited with status ${code}.`);
-		CLEANUP();
-		process.exitCode = code ?? 0;
-	});
-
-	// Wait for proxy process to start...
-	while (!proxy.pid) {}
-
-	if (port === undefined) {
-		logger.log(
-			`Sleeping ${SECONDS_TO_WAIT_FOR_PROXY} seconds to allow proxy process to start before attempting to automatically determine port...`
-		);
-		logger.log("To skip, specify the proxy port with --proxy.");
-		await sleep(SECONDS_TO_WAIT_FOR_PROXY * 1000);
-
-		port = getPids(proxy.pid)
-			.map(getPort)
-			.filter((nr) => nr !== undefined)[0];
-
-		if (port === undefined) {
-			CLEANUP();
-			throw new FatalError(
-				"Could not automatically determine proxy port. Please specify the proxy port with --proxy.",
-				1
-			);
-		} else {
-			logger.log(`Automatically determined the proxy port to be ${port}.`);
-		}
-	}
-
-	return port;
-}