Allowed passing newComponentStructure to Migrator class

Author: ijlee2

bin/ember-component-template-colocation-migrator

@@ -14,7 +14,12 @@ let options = {
 let parsed = nopt(options);
 let projectRoot = parsed['project-root'] || process.cwd();
 
-let migrator = new Migrator({ projectRoot });
+let newComponentStructure = 'flat';
+
+let migrator = new Migrator({
+  projectRoot,
+  newComponentStructure
+});
 
 migrator.execute().then(function() {
   console.log('Codemod finished successfully!');

lib/migrator.js

@@ -6,9 +6,10 @@ const { moveFile, removeDirs } = require('./utils/file')
 
 module.exports = class Migrator {
   constructor(options) {
-    const { projectRoot } = options;
+    const { projectRoot, newComponentStructure } = options;
 
     this.projectRoot = projectRoot;
+    this.newComponentStructure = newComponentStructure;
   }
 
   findClassicComponentTemplates() {
@@ -120,7 +121,9 @@ module.exports = class Migrator {
     templateFilePaths = this.skipTemplatesUsedAsLayoutName(templateFilePaths);
     templateFilePaths = this.skipTemplatesUsedAsPartial(templateFilePaths);
 
-    this.changeComponentStructureToFlat(templateFilePaths);
+    if (this.newComponentStructure === 'flat') {
+      this.changeComponentStructureToFlat(templateFilePaths);
+    }
 
     // Clean up
     await this.removeEmptyClassicComponentDirectories();

test/migrator-test.js

@@ -38,7 +38,10 @@ describe("Migrator", function() {
       fixturify.writeSync(tmpPath, input);
 
       var migratorOptions = Object.assign(
-        { projectRoot: tmpPath },
+        {
+          projectRoot: tmpPath,
+          newComponentStructure: 'flat'
+        },
         migratorConfig
       );