Update transform file if a custom transform is selected

.changeset/rich-ladybugs-ring.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": minor
+---
+
+Send updated transform file if a custom transform is selected

src/cli.ts

@@ -5,7 +5,13 @@ import { spawn } from "child_process";
 // eslint-disable-next-line @typescript-eslint/ban-ts-comment
 // @ts-ignore: package.json will be imported from dist folders
 import { version } from "../package.json"; // eslint-disable-line
-import { getHelpParagraph, getTransforms } from "./utils";
+import {
+  getArgsWithUpdatedTransformFile,
+  getHelpParagraph,
+  getTransformFileFromArgs,
+  getTransforms,
+  getUpdatedTransformFile,
+} from "./utils";
 
 export const run = async (args: string[]): Promise<void> => {
   const transforms = getTransforms();
@@ -14,6 +20,12 @@ export const run = async (args: string[]): Promise<void> => {
     process.stdout.write(`aws-sdk-js-codemod: ${version}\n\n`);
   } else if (args[0] === "--help" || args[0] === "-h") {
     process.stdout.write(getHelpParagraph(transforms));
+  } else if (args.includes("-t") || args.some((arg) => arg.startsWith("--transform="))) {
+    const transformFile = getTransformFileFromArgs(args);
+    if (transforms.map(({ name }) => name).includes(transformFile)) {
+      const updatedTransformFile = getUpdatedTransformFile(transformFile);
+      args = getArgsWithUpdatedTransformFile(args, updatedTransformFile);
+    }
   }
   spawn("npm", ["exec", "jscodeshift", "--", ...args], {
     stdio: "inherit",

src/utils/getArgsWithUpdatedTransformFile.ts

@@ -0,0 +1,18 @@
+export const getArgsWithUpdatedTransformFile = (
+  args: string[],
+  updatedTransformFile: string
+): string[] => {
+  if (args.includes("-t")) {
+    const transformIndex = args.indexOf("-t");
+    args[transformIndex + 1] = updatedTransformFile;
+    return args;
+  }
+
+  const transformArg = args.find((arg) => arg.startsWith("--transform="));
+  if (!transformArg) {
+    throw new Error("No transform file specified in -t or --transform.");
+  }
+  const transformIndex = args.indexOf(transformArg);
+  args[transformIndex] = `--transform=${updatedTransformFile}`;
+  return args;
+};

src/utils/getTransformFileFromArgs.ts

@@ -0,0 +1,12 @@
+export const getTransformFileFromArgs = (args: string[]): string => {
+  if (args.includes("-t")) {
+    const transformIndex = args.indexOf("-t");
+    return args[transformIndex + 1];
+  }
+
+  const transformArg = args.find((arg) => arg.startsWith("--transform="));
+  if (!transformArg) {
+    throw new Error("No transform file specified in -t or --transform.");
+  }
+  return transformArg.split("=")[1];
+};

src/utils/getUpdatedTransformFile.ts

@@ -0,0 +1,4 @@
+import { resolve } from "path";
+
+export const getUpdatedTransformFile = (transformFolder: string) =>
+  resolve(__dirname, "..", "transforms", transformFolder, "transformer.js");

src/utils/index.ts

@@ -1,2 +1,5 @@
+export * from "./getArgsWithUpdatedTransformFile";
 export * from "./getHelpParagraph";
+export * from "./getTransformFileFromArgs";
 export * from "./getTransforms";
+export * from "./getUpdatedTransformFile";