Add --codemod-dir command line flag

Author: Robert Jackson

commands/local/generate/codemod.js

@@ -6,6 +6,11 @@ module.exports.builder = function builder(yargs) {
     .positional('codemod-name', {
       describe: 'the name of the codemod to generate',
     })
+    .option('codemod-dir', {
+      type: 'string',
+      describe: 'the path to the transform directory',
+      default: `./transforms/`,
+    })
     .option('type', {
       alias: 't',
       describe: 'choose the transform type',
@@ -16,13 +21,14 @@ module.exports.builder = function builder(yargs) {
 
 function jsHandler(options) {
   const fs = require('fs-extra');
+  const path = require('path');
   const { stripIndent } = require('common-tags');
   const importCwd = require('import-cwd');
   const generateFixture = require('./fixture').handler;
 
   let { codemodName } = options;
   let projectName = importCwd('./package.json').name;
-  let codemodDir = `${process.cwd()}/transforms/${codemodName}`;
+  let codemodDir = path.join(options.codemodDir, codemodName);
 
   fs.outputFileSync(
     `${codemodDir}/index.js`,
@@ -94,7 +100,12 @@ function jsHandler(options) {
     'utf8'
   );
 
-  generateFixture({ codemodName, fixtureName: 'basic', type: options.type });
+  generateFixture({
+    codemodName,
+    codemodDir: options.codemodDir,
+    fixtureName: 'basic',
+    type: options.type,
+  });
 }
 
 function hbsHandler(options) {
@@ -173,7 +184,12 @@ function hbsHandler(options) {
     'utf8'
   );
 
-  generateFixture({ codemodName, fixtureName: 'basic', type: options.type });
+  generateFixture({
+    codemodName,
+    codemodDir: options.codemodDir,
+    fixtureName: 'basic',
+    type: options.type,
+  });
 }
 
 module.exports.handler = function handler(options) {

commands/local/generate/fixture.js

@@ -8,15 +8,21 @@ module.exports.builder = function builder(yargs) {
     })
     .positional('fixture-name', {
       describe: 'the name of the fixture to generate',
+    })
+    .option('codemod-dir', {
+      type: 'string',
+      describe: 'the path to the transform directory',
+      default: `./transforms/`,
     });
 };
 
 module.exports.handler = function handler(options) {
   const fs = require('fs-extra');
+  const path = require('path');
   const { getTransformType } = require('../../../src/transform-support');
 
   let { codemodName, fixtureName } = options;
-  let codemodDir = `${process.cwd()}/transforms/${codemodName}`;
+  let codemodDir = path.resolve(process.cwd(), path.join(options.codemodDir, codemodName));
   let fixturePath = `${codemodDir}/__testfixtures__/${fixtureName}`;
 
   let transformType = getTransformType(codemodDir);

tests/cli-test.js

@@ -158,6 +158,27 @@ QUnit.module('codemod-cli', function (hooks) {
         ]);
       });
 
+      QUnit.test('should generate a codemod in a custom directory', async function(assert) {
+        let result = await execa(EXECUTABLE_PATH, [
+          'generate',
+          'codemod',
+          'main',
+          '--codemod-dir',
+          'other-dir',
+        ]);
+
+        assert.equal(result.exitCode, 0, 'exited with zero');
+        assert.deepEqual(walkSync(codemodProject.path('other-dir')), [
+          'main/',
+          'main/README.md',
+          'main/__testfixtures__/',
+          'main/__testfixtures__/basic.input.js',
+          'main/__testfixtures__/basic.output.js',
+          'main/index.js',
+          'main/test.js',
+        ]);
+      });
+
       QUnit.test('should generate a hbs codemod', async function (assert) {
         let result = await execa(EXECUTABLE_PATH, ['generate', 'codemod', 'main', '--type', 'hbs']);