refactor: convert require.resolve patches to astgrep

Author: vicb

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

@@ -16,6 +16,7 @@ import { inlineFindDir } from "./patches/plugins/find-dir.js";
 import { patchInstrumentation } from "./patches/plugins/instrumentation.js";
 import { inlineLoadManifest } from "./patches/plugins/load-manifest.js";
 import { patchNextServer } from "./patches/plugins/next-server.js";
+import { patchResolveCache } from "./patches/plugins/open-next.js";
 import { handleOptionalDependencies } from "./patches/plugins/optional-deps.js";
 import { patchPagesRouterContext } from "./patches/plugins/pages-router-context.js";
 import { patchDepdDeprecations } from "./patches/plugins/patch-depd-deprecations.js";
@@ -99,6 +100,7 @@ export async function bundleServer(buildOpts: BuildOptions): Promise<void> {
       inlineLoadManifest(updater, buildOpts),
       patchNextServer(updater, buildOpts),
       patchDepdDeprecations(updater),
+      patchResolveCache(updater, buildOpts),
       // Apply updater updates, must be the last plugin
       updater.plugin,
     ] as Plugin[],
@@ -152,7 +154,7 @@ export async function bundleServer(buildOpts: BuildOptions): Promise<void> {
 
   fs.writeFileSync(openNextServerBundle + ".meta.json", JSON.stringify(result.metafile, null, 2));
 
-  await updateWorkerBundledCode(openNextServerBundle, buildOpts);
+  await updateWorkerBundledCode(openNextServerBundle);
 
   const isMonorepo = monorepoRoot !== appPath;
   if (isMonorepo) {
@@ -170,28 +172,9 @@ export async function bundleServer(buildOpts: BuildOptions): Promise<void> {
 /**
  * This function applies patches required for the code to run on workers.
  */
-export async function updateWorkerBundledCode(
-  workerOutputFile: string,
-  buildOpts: BuildOptions
-): Promise<void> {
+export async function updateWorkerBundledCode(workerOutputFile: string): Promise<void> {
   const code = await readFile(workerOutputFile, "utf8");
-
-  const patchedCode = await patchCodeWithValidations(code, [
-    ["require", patches.patchRequire],
-    ["cacheHandler", (code) => patches.patchCache(code, buildOpts)],
-    ["composableCache", (code) => patches.patchComposableCache(code, buildOpts), { isOptional: true }],
-    [
-      "`require.resolve` call",
-      // workers do not support dynamic require nor require.resolve
-      (code) => code.replace('require.resolve("./cache.cjs")', '"unused"'),
-    ],
-    [
-      "`require.resolve composable cache` call",
-      // workers do not support dynamic require nor require.resolve
-      (code) => code.replace('require.resolve("./composable-cache.cjs")', '"unused"'),
-    ],
-  ]);
-
+  const patchedCode = await patchCodeWithValidations(code, [["require", patches.patchRequire]]);
   await writeFile(workerOutputFile, patchedCode);
 }
 

packages/cloudflare/src/cli/build/patches/plugins/next-server.ts

@@ -8,7 +8,7 @@
  */
 
 import { existsSync, readFileSync } from "node:fs";
-import { join } from "node:path";
+import path from "node:path";
 
 import { type BuildOptions, getPackagePath } from "@opennextjs/aws/build/helper.js";
 import { patchCode } from "@opennextjs/aws/build/patch/astCodePatcher.js";
@@ -26,7 +26,9 @@ export function patchNextServer(updater: ContentUpdater, buildOpts: BuildOptions
         callback: async ({ contents }) => {
           const { outputDir } = buildOpts;
 
-          const manifestPath = join(
+          contents = patchCode(contents, buildIdRule);
+
+          const manifestPath = path.join(
             outputDir,
             "server-functions/default",
             getPackagePath(buildOpts),

packages/cloudflare/src/cli/build/patches/plugins/open-next.ts

@@ -0,0 +1,51 @@
+/**
+ * Removed unused `require.resolve` calls in Open Next.
+ */
+
+import path from "node:path";
+
+import { type BuildOptions, getPackagePath } from "@opennextjs/aws/build/helper.js";
+import { patchCode } from "@opennextjs/aws/build/patch/astCodePatcher.js";
+import type { ContentUpdater, Plugin } from "@opennextjs/aws/plugins/content-updater.js";
+import { getCrossPlatformPathRegex } from "@opennextjs/aws/utils/regex.js";
+
+export function patchResolveCache(updater: ContentUpdater, buildOpts: BuildOptions): Plugin {
+  const { outputDir } = buildOpts;
+  const packagePath = getPackagePath(buildOpts);
+  const outputPath = path.join(outputDir, "server-functions/default");
+
+  const indexPath = path.relative(
+    buildOpts.appBuildOutputPath,
+    path.join(outputPath, packagePath, `index.mjs`)
+  );
+
+  console.error({ index: indexPath });
+
+  return updater.updateContent("patch-resolve-cache", [
+    {
+      field: {
+        filter: getCrossPlatformPathRegex(indexPath),
+        contentFilter: /cacheHandlerPath/,
+        callback: async ({ contents }) => {
+          contents = patchCode(contents, cacheHandlerRule);
+          contents = patchCode(contents, compositeCacheHandlerRule);
+          return contents;
+        },
+      },
+    },
+  ]);
+}
+
+export const cacheHandlerRule = `
+rule:
+  pattern: var cacheHandlerPath = __require.resolve("./cache.cjs");
+fix: |-
+  var cacheHandlerPath = "";
+`;
+
+export const compositeCacheHandlerRule = `
+rule:
+  pattern: var composableCacheHandlerPath = __require.resolve("./composable-cache.cjs");
+fix: |-
+  var composableCacheHandlerPath = "";
+`;