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

import * as fsp from "node:fs/promises";
import * as path from "node:path";
import { fileURLToPath } from "node:url";
import { cloudflare } from "@cloudflare/vite-plugin";
import { defineConfig, preview } from "vite";
import { satisfiesMinimumViteVersion } from "../__test-utils__/vite-version";

export function createConfig(assetsOnly: boolean) {
	return defineConfig(
		// These playground tests don't run in Vite 6 because the feature is not compatible
		// We return an empty Vite config so that the preview server can still start
		!satisfiesMinimumViteVersion("7.0.0")
			? {}
			: {
					plugins: [
						cloudflare({
							inspectorPort: false,
							persistState: false,
							viteEnvironment: { name: "ssr" },
							assetsOnly: () => assetsOnly,
							experimental: {
								prerenderWorker: {
									config(_, { entryWorkerConfig }) {
										return {
											...entryWorkerConfig,
											name: "prerender",
											main: "./src/prerender.ts",
										};
									},
								},
							},
						}),
						{
							name: "prerender-plugin",
							enforce: "post",
							buildApp: {
								order: "post",
								async handler(builder) {
									const previewServer = await preview({
										// Needed because the tests are run from a different directory
										root: path.dirname(fileURLToPath(import.meta.url)),
										logLevel: "silent",
										preview: {
											port: 0,
										},
									});

									const baseUrl = previewServer.resolvedUrls?.local[0];
									const clientOutputDirectory =
										builder.environments.client?.config.build.outDir;

									if (baseUrl && clientOutputDirectory) {
										const response = await fetch(
											new URL("/prerendered", baseUrl)
										);
										const html = await response.text();

										await fsp.writeFile(
											path.resolve(
												builder.config.root,
												clientOutputDirectory,
												"prerendered.html"
											),
											html
										);

										console.log("Pre-rendered /prerendered");
									}

									await previewServer.close();
								},
							},
						},
					],
				}
	);
}

export default createConfig(false);