✨ Improve overal CLI flow.

Author: danieldelcore

.node-version

@@ -0,0 +1 @@
+v14.16.0

packages/cli/package.json

@@ -14,9 +14,11 @@
   },
   "dependencies": {
     "chalk": "^4.1.0",
+    "fs-extra": "^9.1.0",
     "jscodeshift": "^0.12.0",
     "live-plugin-manager": "^0.15.1",
     "meow": "^9.0.0",
+    "semver": "^7.3.5",
     "ts-node": "^9.1.1"
   }
 }

packages/cli/src/cli.ts

@@ -22,9 +22,9 @@ Usage
 
 Options
   --transform, -t the transform to run, will prompt for a transform if not provided and no module is passed
-  --packages, runs transforms for the specified comma separated list of packages, optionally include a version for each package to run all transforms since that version
-  --parser, -p babel|babylon|flow|ts|tsx parser to use for parsing the source files (default: babel)
+  --packages -pkgs, Comma separated list of packages to run transforms for, @scope/package[@version]. If version is supplied, will only run transforms above that version
   --version, -v version number
+  --parser, -p babel|babylon|flow|ts|tsx parser to use for parsing the source files (default: babel)
   --help, 😱
 
 Examples
@@ -43,6 +43,11 @@ Examples
       },
       packages: {
         type: 'string',
+        alias: 'pkgs',
+      },
+      range: {
+        type: 'string',
+        alias: 'r',
       },
       parser: {
         type: 'string',

packages/cli/src/main.ts

@@ -1,8 +1,7 @@
-// import { ParsedPath } from 'path';
-// import semver from 'semver';
-
+import semver from 'semver';
+import fs from 'fs-extra';
 import { PluginManager } from 'live-plugin-manager';
-// @ts-ignore
+// @ts-ignore Run transform(s) on path https://github.com/facebook/jscodeshift/issues/398
 import * as jscodeshift from 'jscodeshift/src/Runner';
 
 import { Flags } from './cli';
@@ -12,51 +11,77 @@ export default async function main(
   paths: string[],
   flags: Flags = { parser: 'babel' },
 ) {
-  console.log('CodeshiftCommunity CLI');
-  console.log(paths, flags);
+  console.log('CodeshiftCommunity CLI', paths, flags);
 
-  const transforms = [];
+  let transforms: string[] = [];
 
   if (!flags.transform && !flags.packages) {
     throw new InvalidUserInputError(
       'No transform provided, please specify a transform with either the --transform or --packages flags',
     );
   }
 
+  if (paths.length === 0) {
+    throw new InvalidUserInputError(
+      'No path provided, please specify which files your codemod should modify',
+    );
+  }
+
   if (flags.transform) {
     transforms.push(flags.transform);
   }
 
+  const packageManager = new PluginManager();
+
   if (flags.packages) {
-    const rawPackageName = flags.packages.replace('@', '').replace('/', '__');
-    const packageName = `@codeshift/mod-${rawPackageName}`;
-    console.log(rawPackageName, packageName);
+    const pkgs = flags.packages.split(',');
 
-    const packageManager = new PluginManager();
-    await packageManager.install(packageName);
+    for (const pkg of pkgs) {
+      const pkgSplit = pkg.split('@').filter(str => !!str);
+      const name = pkgSplit[0].replace('/', '__');
+      const baseVersion = semver.coerce(pkgSplit[pkgSplit.length - 1]);
 
-    const codemod = packageManager.require(packageName);
+      if (!baseVersion && !semver.valid(baseVersion)) {
+        throw new InvalidUserInputError(
+          `Invalid version provided to the --packages flag. Package ${pkg} is missing version. Please try: "@[scope]/[package]@[version]" for example @atlaskit/avatar@10.0.0`,
+        );
+      }
 
-    console.log(
-      // codemod,
-      codemod.transform18_0_0,
-      codemod.transform19_0_0,
-      // packageManager.list(),
-      // packageManager.getInfo(packageName),
-    );
+      const codemodName = `@codeshift/mod-${name}`;
 
-    // TODO: We'll have to uninstall the mod at some point
-    // await packageManager.uninstall('codemod');
+      await packageManager.install(codemodName);
+      const info = await packageManager.getInfo(codemodName);
+
+      /**
+       * TODO: currently jscodeshift only accepts a path to a transform rather than a function or module.
+       * The below logic will need to be refactored once this is fixed
+       */
+      const modulePath = `${info?.location}/src`;
+      const directories = await fs.readdir(modulePath);
+
+      directories
+        .filter(
+          dir => semver.valid(dir) && semver.satisfies(dir, `>=${baseVersion}`),
+        )
+        .forEach(dir => transforms.push(`${modulePath}/${dir}/transform.ts`));
+    }
   }
 
   if (!transforms.length) {
-    throw new Error('Unable to locate transforms from provided flags');
+    throw new InvalidUserInputError(
+      'Unable to locate transforms from provided flags.',
+    );
   }
 
+  // Dedupe transform array
+  transforms = transforms.filter(
+    (transform, i) => transforms.indexOf(transform) === i,
+  );
+
   for (const transform of transforms) {
-    // run transform(s) on path https://github.com/facebook/jscodeshift/issues/398
-    jscodeshift.run('', paths, {
-      transform: transform,
+    console.log('Running transform', transform);
+
+    await jscodeshift.run(transform, paths, {
       verbose: 0,
       dry: false,
       print: true,
@@ -70,4 +95,10 @@ export default async function main(
       stdin: false,
     });
   }
+
+  /**
+   * TODO: uncomment the below when jscodeshift can be used as an async function
+   */
+  // await packageManager.uninstallAll();
+  // console.log('Done!');
 }