turbopack plugins

Author: conico974

packages/cloudflare/src/cli/build/open-next/createServerBundle.ts

@@ -30,6 +30,7 @@ import { patchResRevalidate } from "../patches/plugins/res-revalidate.js";
 import { patchUseCacheIO } from "../patches/plugins/use-cache.js";
 import { normalizePath } from "../utils/index.js";
 import { copyWorkerdPackages } from "../utils/workerd.js";
+import { inlineChunksPatch } from "../patches/plugins/turbopack.js";
 
 interface CodeCustomization {
 	// These patches are meant to apply on user and next generated code
@@ -210,6 +211,7 @@ async function generateBundle(
 		// Cloudflare specific patches
 		patchResRevalidate,
 		patchUseCacheIO,
+		inlineChunksPatch,
 		...additionalCodePatches,
 	]);
 
@@ -287,8 +289,8 @@ async function generateBundle(
 			alias: {
 				...(isBundled
 					? {
-							"next/dist/server/next-server.js": "./next-server.runtime.prod.js",
-						}
+						"next/dist/server/next-server.js": "./next-server.runtime.prod.js",
+					}
 					: {}),
 			},
 		},

packages/cloudflare/src/cli/build/patches/plugins/turbopack.ts

@@ -0,0 +1,59 @@
+import { patchCode } from "@opennextjs/aws/build/patch/astCodePatcher.js";
+import type { CodePatcher } from "@opennextjs/aws/build/patch/codePatcher";
+import { getCrossPlatformPathRegex } from "@opennextjs/aws/utils/regex.js";
+
+const inlineChunksRule = `
+rule:
+  kind: call_expression
+  pattern: require(resolved)
+fix:
+  requireChunk(chunkPath)
+`;
+
+export const inlineChunksPatch: CodePatcher = {
+    name: "inline-turbopack-chunks",
+    patches: [
+        {
+            versions: ">=16.0.0",
+            pathFilter: getCrossPlatformPathRegex(
+                String.raw`\[turbopack\]_runtime\.js$`,
+                {
+                    escape: false,
+                }
+            ),
+            contentFilter: /loadRuntimeChunkPath/,
+            patchCode: async ({ code, tracedFiles }) => {
+                const patched = patchCode(code, inlineChunksRule);
+
+                return `${patched}\n${inlineChunksFn(tracedFiles)}`;
+            },
+        },
+    ],
+};
+
+function getInlinableChunks(tracedFiles: string[]) {
+    const chunks = new Set<string>();
+    for (const file of tracedFiles) {
+        if (file.includes(".next/server/chunks/") && !file.includes("[turbopack]_runtime.js")) {
+            chunks.add(file);
+        }
+    }
+    return chunks;
+}
+
+function inlineChunksFn(tracedFiles: string[]) {
+    // From the outputs, we extract every chunks
+    const chunks = getInlinableChunks(tracedFiles);
+    return `
+  function requireChunk(chunkPath) {
+    switch(chunkPath) {
+${Array.from(chunks).map(chunk => `      case "${
+        // we only want the path after /path/to/.next/
+        chunk.replace(/.*\.next\//, "")
+        }": return require("${chunk}");`).join("\n")}
+      default:
+        throw new Error(\`Not found \${chunkPath}\`);
+    }
+  }
+`;
+}