patch node-module-loader dynamic require

Author: HyperKiko

packages/cloudflare/src/cli/build/bundle-server.ts

@@ -16,6 +16,7 @@ import { patchFetchCacheSetMissingWaitUntil } from "./patches/plugins/fetch-cach
 import { inlineFindDir } from "./patches/plugins/find-dir.js";
 import { patchInstrumentation } from "./patches/plugins/instrumentation.js";
 import { inlineLoadManifest } from "./patches/plugins/load-manifest.js";
+import { inlineNodeModuleLoader } from "./patches/plugins/node-module-loader.js";
 import { handleOptionalDependencies } from "./patches/plugins/optional-deps.js";
 import { patchDepdDeprecations } from "./patches/plugins/patch-depd-deprecations.js";
 import { fixRequire } from "./patches/plugins/require.js";
@@ -89,6 +90,7 @@ export async function bundleServer(buildOpts: BuildOptions): Promise<void> {
     plugins: [
       shimRequireHook(buildOpts),
       inlineRequirePage(updater, buildOpts),
+      inlineNodeModuleLoader(updater, buildOpts),
       setWranglerExternal(),
       fixRequire(updater),
       handleOptionalDependencies(optionalDependencies),

packages/cloudflare/src/cli/build/patches/plugins/node-module-loader.ts

@@ -0,0 +1,66 @@
+import { readFile } from "node:fs/promises";
+import { join } from "node:path";
+
+import { type BuildOptions, getPackagePath } from "@opennextjs/aws/build/helper.js";
+import { getCrossPlatformPathRegex } from "@opennextjs/aws/utils/regex.js";
+
+import { normalizePath } from "../../utils/normalize-path.js";
+import { patchCode, type RuleConfig } from "../ast/util.js";
+import type { ContentUpdater } from "./content-updater.js";
+
+export function inlineNodeModuleLoader(updater: ContentUpdater, buildOpts: BuildOptions) {
+  return updater.updateContent(
+    "inline-node-module-loader",
+    {
+      filter: getCrossPlatformPathRegex(
+        String.raw`/next/dist/server/lib/module-loader/node-module-loader\.js$`,
+        { escape: false }
+      ),
+      contentFilter: /class NodeModuleLoader {/,
+    },
+    async ({ contents }) => patchCode(contents, await getRule(buildOpts))
+  );
+}
+
+async function getRule(buildOpts: BuildOptions) {
+  const { outputDir } = buildOpts;
+  const serverDir = join(outputDir, "server-functions/default", getPackagePath(buildOpts), ".next/server");
+
+  const pagesManifestFile = join(serverDir, "pages-manifest.json");
+
+  let pagesManifests: string[] = [];
+  try {
+    pagesManifests = Object.values(JSON.parse(await readFile(pagesManifestFile, "utf-8")));
+  } catch {
+    // The file does not exist
+    pagesManifests = [];
+  }
+
+  const files = pagesManifests.map((path) => normalizePath(path));
+
+  // Inline fs access and dynamic requires that are not supported by workerd.
+  const fnBody = `
+${files
+  .map(
+    (file) => `if (id.endsWith("${file}")) {
+  return require(${JSON.stringify(join(serverDir, file))});
+}`
+  )
+  .join("\n")}
+`;
+
+  return {
+    rule: {
+      pattern: `class NodeModuleLoader {
+    async load($ID) {
+        $$$_BODY
+    }
+}`,
+    },
+    fix: `class NodeModuleLoader {
+    async load($ID) {
+        ${fnBody}
+    }
+}`,
+  } satisfies RuleConfig;
+}