feat(runner): add child process support for specific codemod execution

.changeset/ninety-candles-lie.md

@@ -0,0 +1,6 @@
+---
+"@codemod-com/runner": patch
+"codemod": patch
+---
+
+fix: add child process support for specific codemod execution

apps/cli/package.json

@@ -68,7 +68,7 @@
     "yargs": "catalog:"
   },
   "scripts": {
-    "build": "tsc && node ./build.js",
+    "build": "tsc && node ./build.js -- --prod",
     "build:local": "pnpm build -- --local",
     "build:prod": "pnpm build -- --prod",
     "build:publish": "pnpm build -- --publish",
@@ -91,10 +91,11 @@
   "dependencies": {
     "@ast-grep/cli": "catalog:",
     "@ast-grep/napi": "catalog:",
-    "keytar": "catalog:",
+    "@codemod.com/workflow": "workspace:*",
+    "@nodejs-loaders/alias": "^1.1.1",
     "@octokit/rest": "catalog:",
     "blessed": "catalog:",
-    "@codemod.com/workflow": "workspace:*",
-    "esbuild": "^0.23.0"
+    "esbuild": "^0.23.0",
+    "keytar": "catalog:"
   }
 }

packages/runner/src/runner.ts

@@ -1,9 +1,8 @@
+import { spawn } from "node:child_process";
 import * as fs from "node:fs";
 import { readFile } from "node:fs/promises";
 import { dirname, join } from "node:path";
 import { join as joinPosix } from "node:path/posix";
-import { glob, globStream } from "glob";
-import * as yaml from "js-yaml";
 
 import type { Filemod } from "@codemod-com/filemod";
 import { Printer, boxen, chalk, colorLongString } from "@codemod-com/printer";
@@ -21,12 +20,15 @@ import {
   isRecipeCodemod,
 } from "@codemod-com/utilities";
 import type { AuthServiceInterface } from "@codemod.com/workflow";
+import { glob, globStream } from "glob";
+import * as yaml from "js-yaml";
 import type { RunnerServiceInterface } from "#runner-service.js";
 import {
   getCodemodExecutable,
   getRunConfig,
   temporaryLoadedModules,
 } from "#source-code.js";
+import { getRandomMjs, shouldRunAsChildNodeProcess } from "#utils.js";
 import { astGrepLanguageToPatterns } from "./engines/ast-grep.js";
 import type { Dependencies } from "./engines/filemod.js";
 import { runFilemod } from "./engines/filemod.js";
@@ -492,6 +494,56 @@ export class Runner {
       return await onSuccess?.({ codemod, output: "", commands: [] });
     }
 
+    if (shouldRunAsChildNodeProcess(codemod.config.name)) {
+      const randomMjs = join(process.cwd(), getRandomMjs());
+      await fs.promises.copyFile(
+        join(codemod.path, "cdmd_dist", "index.js"),
+        randomMjs,
+      );
+      const cliNodeModules = join(dirname(__dirname), "node_modules");
+      const symlinkedPackages: string[] = [];
+      const cliNodeModulesDir = await fs.promises.readdir(cliNodeModules);
+      const targetNodeModulesDir = join(process.cwd(), "node_modules");
+
+      for (const dir of cliNodeModulesDir) {
+        if (fs.existsSync(join(targetNodeModulesDir, dir)) || dir === ".bin") {
+          continue;
+        }
+
+        await fs.promises.symlink(
+          join(cliNodeModules, dir),
+          join(targetNodeModulesDir, dir),
+        );
+
+        symlinkedPackages.push(dir);
+      }
+
+      return new Promise<void>((resolve, reject) => {
+        const child = spawn(process.execPath, [randomMjs], {
+          stdio: "inherit",
+          cwd: process.cwd(),
+        });
+
+        child.on("close", async (code) => {
+          await fs.promises.unlink(randomMjs);
+          for (const dir of symlinkedPackages) {
+            await fs.promises.unlink(join(targetNodeModulesDir, dir));
+          }
+
+          if (code === 0) {
+            await onSuccess?.({ codemod, output: "", commands: [] });
+            resolve();
+          } else {
+            reject(new Error(`Process exited with code ${code}`));
+          }
+        });
+
+        child.on("error", (err) => {
+          reject(err);
+        });
+      });
+    }
+
     const codemodSource = await getCodemodExecutable(
       codemod.path,
       flowSettings.esm,

packages/runner/src/utils.ts

@@ -25,3 +25,11 @@ export const isTheSameData = (oldData: string, newData: string) => {
     return false;
   }
 };
+
+export const getRandomMjs = () => {
+  return `${Math.random().toString(36).substring(2, 15)}_${Math.random().toString(36).substring(2, 15)}.mjs`;
+};
+
+export const shouldRunAsChildNodeProcess = (codemodName: string) => {
+  return codemodName === "correct-ts-specifiers";
+};