Update literal paths

__testfixtures__/literal.input.js

@@ -0,0 +1,5 @@
+import Model, { attr } from 'ember-data/model';
+import { hasMany, belongsTo } from 'ember-data/relationships';
+import JSONAPIAdapter from 'ember-data/adapters/json-api';
+import { InvalidError, ServerError, TimeoutError, NotFoundError  } from 'ember-data/adapter/error';
+import Transform from '@ember-data/serializer/transform';

__testfixtures__/literal.output.js

@@ -0,0 +1,4 @@
+import Model, { attr, belongsTo, hasMany } from '@ember-data/model';
+import JSONAPIAdapter from '@ember-data/adapter/json-api';
+import { InvalidError, ServerError, TimeoutError, NotFoundError  } from '@ember-data/adapter/error';
+import Transform from '@ember-data/serializer/transform';

bin/ember-data-codemod.js

@@ -89,7 +89,7 @@ function buildReport() {
 
   // Find all of the temporary logs from the worker processes, which contain a
   // serialized JSON array on each line.
-  glob('ember-modules-codemod.tmp.*', (err, logs) => {
+  glob('ember-data-codemod.tmp.*', (err, logs) => {
     // If no worker found an unexpected value, nothing to report.
     if (!logs) {
       return;
@@ -143,7 +143,7 @@ function buildReport() {
       );
     } else {
       console.log(
-        chalk.green('\nDone! All uses of the Ember global have been updated.')
+        chalk.green('\nDone! All uses of the Ember global and Ember Data imports have been updated.')
       );
     }
   });

transforms/globals-to-ember-data-imports.js

@@ -67,7 +67,7 @@ function transform(file, api /*, options*/) {
     // Now that we've identified all of the replacements that we need to do, we'll
     // make sure to either add new `import` declarations, or update existing ones
     // to add new named exports or the default export.
-    updateOrCreateImportDeclarations(root, modules);
+    updateOrCreateImportDeclarations(root, modules, mappings);
 
     // Actually go through and replace each usage of `DS.whatever` with the
     // imported binding (`whatever`).
@@ -157,6 +157,30 @@ function transform(file, api /*, options*/) {
     return dsUsages.filter(isDSGlobal(globalDS)).paths();
   }
 
+  /*
+   * loops through all modules and replaces literal path if necessary
+   * 'ember-data/model' -> '@ember-data/model'
+   */
+  function updateLiteralPaths(root, module, mappings) {
+    let foundMapping = mappings[module.local];
+    if (foundMapping) {
+      let newSource = foundMapping.source;
+      if (module.source !== newSource) {
+        root
+          .find(j.ImportDeclaration, {
+            source: {
+              type: 'Literal',
+              value: module.source
+            }
+          })
+          .find(j.Literal)
+          .forEach(importLiteral => {
+            j(importLiteral).replaceWith(j.literal(newSource));
+          });
+      }
+    }
+  }
+
   // Find destructured global aliases for fields on the DS global
   function findGlobalDSAliases(root, globalDS, mappings) {
     let aliases = {};
@@ -436,7 +460,7 @@ function transform(file, api /*, options*/) {
     return parent.node.id.name === local;
   }
 
-  function updateOrCreateImportDeclarations(root, registry) {
+  function updateOrCreateImportDeclarations(root, registry, mappings) {
     let body = root.get().value.program.body;
 
     registry.modules.forEach(mod => {
@@ -448,7 +472,6 @@ function transform(file, api /*, options*/) {
         let declaration = root.find(j.ImportDeclaration, {
           source: { value: mod.source }
         });
-
         if (declaration.size() > 0) {
           let specifier;
 
@@ -471,6 +494,9 @@ function transform(file, api /*, options*/) {
           delete body[1].comments;
           mod.node = importStatement;
         }
+      } else {
+        // Update literal paths based on mappings from 'ember-data/model' to '@ember-data/model'
+        updateLiteralPaths(root, mod, mappings);
       }
     });
   }