feat: provide experimental flags to the defineCommand handler

packages/wrangler/src/core/register-yargs-command.ts

@@ -2,6 +2,7 @@ import { fetchResult } from "../cfetch";
 import { readConfig } from "../config";
 import { defaultWranglerConfig } from "../config/config";
 import { FatalError, UserError } from "../errors";
+import { run } from "../experimental-flags";
 import { logger } from "../logger";
 import { printWranglerBanner } from "../update-check";
 import { demandSingleValue } from "./helpers";
@@ -78,18 +79,27 @@ function createHandler(def: CommandDefinition) {
 			// TODO(telemetry): send command started event
 
 			await def.validateArgs?.(args);
+			const experimentalFlags = def.behaviour?.overrideExperimentalFlags
+				? def.behaviour?.overrideExperimentalFlags(args)
+				: {
+						FILE_BASED_REGISTRY: false,
+						MULTIWORKER: false,
+						RESOURCES_PROVISION: args.experimentalProvision ?? false,
+					};
 
-			await def.handler(args, {
-				config:
-					def.behaviour?.provideConfig ?? true
-						? readConfig(args.config, args, {
-								hideWarnings: !(def.behaviour?.printConfigWarnings ?? true),
-							})
-						: defaultWranglerConfig,
-				errors: { UserError, FatalError },
-				logger,
-				fetchResult,
-			});
+			await run(experimentalFlags, () =>
+				def.handler(args, {
+					config:
+						def.behaviour?.provideConfig ?? true
+							? readConfig(args.config, args, {
+									hideWarnings: !(def.behaviour?.printConfigWarnings ?? true),
+								})
+							: defaultWranglerConfig,
+					errors: { UserError, FatalError },
+					logger,
+					fetchResult,
+				})
+			);
 
 			// TODO(telemetry): send command completed event
 		} catch (err) {

packages/wrangler/src/core/types.ts

@@ -2,6 +2,7 @@ import type { fetchResult } from "../cfetch";
 import type { Config } from "../config";
 import type { OnlyCamelCase } from "../config/config";
 import type { FatalError, UserError } from "../errors";
+import type { ExperimentalFlags } from "../experimental-flags";
 import type { Logger } from "../logger";
 import type { CommonYargsOptions, RemoveIndex } from "../yargs-types";
 import type { Teams } from "./teams";
@@ -106,6 +107,15 @@ export type CommandDefinition<
 		 * Set this value to `false` to skip this.
 		 */
 		provideConfig?: boolean;
+
+		/**
+		 * By default, wrangler will provide experimental flags in the handler context,
+		 * according to the default values in register-yargs.command.ts
+		 * Use this to override those defaults per command.
+		 */
+		overrideExperimentalFlags?: (
+			args: HandlerArgs<NamedArgDefs>
+		) => ExperimentalFlags;
 	};
 
 	/**

packages/wrangler/src/deploy/index.ts

@@ -234,6 +234,7 @@ export function deployOptions(yargs: CommonYargsArgv) {
 
 export type DeployArgs = StrictYargsOptionsToInterface<typeof deployOptions>;
 
+// TODO: move this into defineCommand behaviour
 export async function deployHandler(args: DeployArgs) {
 	await run(
 		{

packages/wrangler/src/dev.ts

@@ -23,7 +23,7 @@
 import registerDevHotKeys from "./dev/hotkeys";
 import { maybeRegisterLocalWorker } from "./dev/local";
 import { UserError } from "./errors";
 import { run } from "./experimental-flags";
 import isInteractive from "./is-interactive";
 import { logger } from "./logger";
 import { getLegacyAssetPaths, getSiteAssetPaths } from "./sites";
@@ -62,6 +62,11 @@
 export const dev = createCommand({
 	behaviour: {
 		provideConfig: false,
+		overrideExperimentalFlags: (args) => ({
+			FILE_BASED_REGISTRY: args.experimentalRegistry,
+			MULTIWORKER: Array.isArray(args.config),
+			RESOURCES_PROVISION: args.experimentalProvision ?? false,
+		}),
 	},
 	metadata: {
 		description: "👂 Start a local server for developing your Worker",
@@ -376,14 +381,7 @@
 		}
 	},
 	async handler(args) {
-		const devInstance = await run(
-			{
-				FILE_BASED_REGISTRY: args.experimentalRegistry,
-				MULTIWORKER: Array.isArray(args.config),
-				RESOURCES_PROVISION: false,
-			},
-			() => startDev(args)
-		);
+		const devInstance = await startDev(args);
 		assert(devInstance.devEnv !== undefined);
 		await events.once(devInstance.devEnv, "teardown");
 		await Promise.all(devInstance.secondary.map((d) => d.teardown()));

packages/wrangler/src/experimental-flags.ts

@@ -1,7 +1,7 @@
 import { AsyncLocalStorage } from "async_hooks";
 import { logger } from "./logger";
 
-type ExperimentalFlags = {
+export type ExperimentalFlags = {
 	FILE_BASED_REGISTRY: boolean;
 	MULTIWORKER: boolean;
 	RESOURCES_PROVISION: boolean;