Set environment variable internally during the build

Author: jamesopstad

packages/vite-plugin-cloudflare/playground/prerendering/vite.config.ts

@@ -24,8 +24,6 @@ export default defineConfig({
 			buildApp: {
 				order: "post",
 				async handler(builder) {
-					process.env.IS_VITE_PRERENDER = "true";
-
 					const previewServer = await preview({
 						root: path.dirname(fileURLToPath(import.meta.url)),
 						logLevel: "silent",
@@ -56,8 +54,6 @@ export default defineConfig({
 
 					await previewServer.close();
 
-					delete process.env.IS_VITE_PRERENDER;
-
 					const prerenderOutputDirectory =
 						builder.environments.prerender?.config.build.outDir;
 

packages/vite-plugin-cloudflare/playground/vitest-setup.ts

@@ -309,6 +309,11 @@ export async function startDefaultServe(): Promise<
 		const builder = await createBuilder(buildConfig);
 		await builder.buildApp();
 
+		// This environment variable is used to indicate to the preview server that it is being run during a build
+		// We need to delete it here as, during testing, preview also runs in the same process after the build completes
+		// eslint-disable-next-line turbo/no-undeclared-env-vars
+		delete process.env.CLOUDFLARE_VITE_BUILD;
+
 		const previewConfig = await loadConfig({
 			command: "serve",
 			mode: "development",

packages/vite-plugin-cloudflare/src/index.ts

@@ -52,6 +52,10 @@ export function cloudflare(pluginConfig: PluginConfig = {}): vite.Plugin[] {
 				ctx.setResolvedPluginConfig(
 					resolvePluginConfig(pluginConfig, userConfig, env)
 				);
+
+				if (env.command === "build") {
+					process.env.CLOUDFLARE_VITE_BUILD = "true";
+				}
 			},
 			async configureServer(viteDevServer) {
 				// Patch the `server.restart` method to track whether the server is restarting or not.

packages/vite-plugin-cloudflare/src/plugin-config.ts

@@ -273,7 +273,7 @@ export function resolvePluginConfig(
 			...shared,
 			remoteBindings: pluginConfig.remoteBindings ?? true,
 			type: "preview",
-			workers: getWorkerConfigs(root, !!process.env.IS_VITE_PRERENDER),
+			workers: getWorkerConfigs(root, !!process.env.CLOUDFLARE_VITE_BUILD),
 		};
 	}