Merge pull request #7 from rwjblue/migrate-inject

Migrate `this.inject.service` usage to `this.owner.lookup`.

Author: rwjblue

__testfixtures__/ember-qunit-codemod/inject.input.js

@@ -0,0 +1,14 @@
+import { moduleFor, test } from 'ember-qunit';
+ 
+moduleFor('service:foo-bar', 'Unit | Service | FooBar', {
+});
+
+test('it exists', function(assert) {
+  this.inject.service('foo');
+  this.inject.service('foo', { as: 'bar' });
+}); 
+
+test('it works for controllers', function(assert) {
+  this.inject.controller('foo');
+  this.inject.controller('foo', { as: 'bar' });
+});

__testfixtures__/ember-qunit-codemod/inject.output.js

@@ -0,0 +1,16 @@
+import { module, test } from 'qunit';
+import { setupTest } from 'ember-qunit';
+
+module('Unit | Service | FooBar', function(hooks) {
+  setupTest(hooks);
+
+  test('it exists', function(assert) {
+    this.foo = this.owner.lookup('service:foo');
+    this.bar = this.owner.lookup('service:foo');
+  });
+
+  test('it works for controllers', function(assert) {
+    this.foo = this.owner.lookup('controller:foo');
+    this.bar = this.owner.lookup('controller:foo');
+  });
+});

ember-qunit-codemod.js

@@ -281,6 +281,48 @@ function updateRegisterCalls(j, root) {
     });
 }
 
+function updateInjectCalls(j, root) {
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'MemberExpression',
+        object: {
+          object: {
+            type: 'ThisExpression',
+          },
+          property: {
+            name: 'inject',
+          },
+        },
+      },
+    })
+    .forEach(p => {
+      let injectType = p.node.callee.property.name;
+      let injectedName = p.node.arguments[0].value;
+      let localName = injectedName;
+      if (p.node.arguments[1]) {
+        let options = p.node.arguments[1];
+        let as = options.properties.find(property => property.key.name === 'as');
+        if (as) {
+          localName = as.value.value;
+        }
+      }
+      let assignment = j.assignmentExpression(
+        '=',
+        j.memberExpression(j.thisExpression(), j.identifier(localName)),
+        j.callExpression(
+          j.memberExpression(
+            j.memberExpression(j.thisExpression(), j.identifier('owner')),
+            j.identifier('lookup')
+          ),
+          [j.literal(`${injectType}:${injectedName}`)]
+        )
+      );
+
+      p.replace(assignment);
+    });
+}
+
 module.exports = function(file, api, options) {
   const j = api.jscodeshift;
 
@@ -298,6 +340,7 @@ module.exports = function(file, api, options) {
   updateModuleForToNestedModule(j, root);
   updateLookupCalls(j, root);
   updateRegisterCalls(j, root);
+  updateInjectCalls(j, root);
 
   return root.toSource(printOptions);
 };