feat: EmberData Packages Polyfill

README.md

@@ -120,6 +120,7 @@ interface EmberCLIBabelConfig {
     disableDebugTooling?: boolean;
     disablePresetEnv?: boolean;
     disableEmberModulesAPIPolyfill?: boolean;
+    disableEmberDataPackagesPolyfill?: boolean;
     disableDecoratorTransforms?: boolean;
     extensions?: string[];
   };

index.js

@@ -301,6 +301,7 @@ module.exports = {
       userPlugins,
       this._getDebugMacroPlugins(config),
       this._getEmberModulesAPIPolyfill(config),
+      this._getEmberDataPackagesPolyfill(config),
       shouldCompileModules && this._getModulesPlugin(),
       userPostTransformPlugins
     ).filter(Boolean);
@@ -444,6 +445,16 @@ module.exports = {
     }
   },
 
+  _getEmberDataPackagesPolyfill(config) {
+    let addonOptions = config['ember-cli-babel'] || {};
+
+    if (addonOptions.disableEmberDataPackagesPolyfill) { return; }
+
+    if (this._emberDataVersionRequiresPackagesPolyfill()) {
+      return [[require.resolve('babel-plugin-ember-data-packages-polyfill')]];
+    }
+  },
+
   _getPresetEnv(config) {
     let options = config.options;
 
@@ -511,6 +522,20 @@ module.exports = {
     return true;
   },
 
+  _emberDataVersionRequiresPackagesPolyfill() {
+    let checker = new VersionChecker(this.project);
+    let dep = checker.for('ember-data');
+    let hasEmberData = dep.exists();
+
+    if (hasEmberData) {
+      if (!dep.version) {
+        throw new Error('EmberData missing version');
+      }
+      return semver.lt(dep.version, '3.12.0-alpha.0');
+    }
+    return false;
+  },
+
   _getEmberModulesAPIBlacklist() {
     const blacklist = {
       '@ember/debug': ['assert', 'deprecate', 'warn'],

node-tests/addon-test.js

@@ -8,10 +8,23 @@ const CoreObject = require('core-object');
 const AddonMixin = require('../index');
 const CommonTags = require('common-tags');
 const stripIndent = CommonTags.stripIndent;
+const FixturifyProject = require('fixturify-project');
+const EmberProject = require('ember-cli/lib/models/project');
+const MockCLI = require('ember-cli/tests/helpers/mock-cli');
 const BroccoliTestHelper = require('broccoli-test-helper');
 const createBuilder = BroccoliTestHelper.createBuilder;
 const createTempDir = BroccoliTestHelper.createTempDir;
 const terminateWorkerPool = require('./utils/terminate-workers');
+const path = require('path');
+const fs = require('fs');
+
+function prepareAddon(addon) {
+  addon.pkg.keywords.push('ember-addon');
+  addon.pkg['ember-addon'] = {};
+  addon.files['index.js'] = 'module.exports = { name: require("./package").name };';
+
+  return addon;
+}
 
 let Addon = CoreObject.extend(AddonMixin);
 
@@ -22,6 +35,8 @@ describe('ember-cli-babel', function() {
   beforeEach(function() {
     this.ui = new MockUI();
     let project = {
+      isEmberCLIProject: () => true,
+      _addonsInitialized: true,
       root: __dirname,
       emberCLIVersion: () => '2.16.2',
       dependencies() { return {}; },
@@ -1214,3 +1229,134 @@ describe('ember-cli-babel', function() {
     });
   });
 });
+
+describe('EmberData Packages Polyfill', function() {
+  this.timeout(0);
+
+  let input;
+  let output;
+  let subject;
+  let setupForVersion;
+  let project;
+  let unlink;
+
+  beforeEach(function() {
+    setupForVersion = async (v) => {
+      let fixturifyProject = new FixturifyProject('whatever', '0.0.1');
+      fixturifyProject.addDependency('ember-data', v, addon => {
+        return prepareAddon(addon);
+      });
+      fixturifyProject.addDependency('ember-cli-babel', 'babel/ember-cli-babel#master');
+      fixturifyProject.addDependency('random-addon', '0.0.1', addon => {
+        return prepareAddon(addon);
+      });
+      let pkg = JSON.parse(fixturifyProject.toJSON('package.json'));
+      fixturifyProject.writeSync();
+
+      let linkPath = path.join(fixturifyProject.root, '/whatever/node_modules/ember-cli-babel');
+      let addonPath = path.resolve(__dirname, '../');
+      fs.rmdirSync(linkPath, { recursive: true });
+      fs.symlinkSync(addonPath, linkPath);
+      unlink = () => {
+        fs.unlinkSync(linkPath);
+      };
+
+      let cli = new MockCLI();
+      let root = path.join(fixturifyProject.root, 'whatever');
+      project = new EmberProject(root, pkg, cli.ui, cli);
+      project.initializeAddons();
+
+      this.addon = project.addons.find(a => { return a.name === 'ember-cli-babel'; });
+
+      input = await createTempDir();
+    };
+  });
+
+  afterEach(async function() {
+    unlink();
+    await input.dispose();
+    await output.dispose();
+    // shut down workers after the tests are run so that mocha doesn't hang
+    await terminateWorkerPool();
+  });
+
+  it("does not convert when _emberDataVersionRequiresPackagesPolyfill returns false", async function() {
+    await setupForVersion('3.12.0-alpha.0');
+    input.write({
+      "foo.js": `export { default } from '@ember-data/store';`,
+      "bar.js": `import Model, { attr } from '@ember-data/model';\nexport var User = Model;\nexport var name = attr;`,
+    });
+
+    subject = this.addon.transpileTree(input.path(), {
+      'ember-cli-babel': {
+        compileModules: false,
+        disableDebugTooling: true,
+        disableEmberDataPackagesPolyfill: true
+      }
+    });
+
+    output = createBuilder(subject);
+
+    await output.build();
+
+    expect(
+      output.read()
+    ).to.deep.equal({
+      "foo.js": `export { default } from '@ember-data/store';`,
+      "bar.js": `import Model, { attr } from '@ember-data/model';\nexport var User = Model;\nexport var name = attr;`,
+    });
+  });
+
+  it("does not convert for EmberData when _emberDataVersionRequiresPackagesPolyfill returns true and disableEmberDataPackagesPolyfill is true", async function() {
+    await setupForVersion('3.11.0');
+    input.write({
+      "foo.js": `export { default } from '@ember-data/store';`,
+      "bar.js": `import Model, { attr } from '@ember-data/model';\nexport var User = Model;\nexport var name = attr;`,
+    });
+
+    subject = this.addon.transpileTree(input.path(), {
+      'ember-cli-babel': {
+        compileModules: false,
+        disableDebugTooling: true,
+        disableEmberDataPackagesPolyfill: true
+      }
+    });
+
+    output = createBuilder(subject);
+
+    await output.build();
+
+    expect(
+      output.read()
+    ).to.deep.equal({
+      "foo.js": `export { default } from '@ember-data/store';`,
+      "bar.js": `import Model, { attr } from '@ember-data/model';\nexport var User = Model;\nexport var name = attr;`,
+    });
+  });
+
+  it("it does convert for EmberData when _emberDataVersionRequiresPackagesPolyfill returns true", async function() {
+    await setupForVersion('3.11.99');
+    input.write({
+      "foo.js": `export { default } from '@ember-data/store';`,
+      "bar.js": `import Model, { attr } from '@ember-data/model';\nexport var User = Model;export var name = attr;`,
+    });
+
+    subject = this.addon.transpileTree(input.path(), {
+      'ember-cli-babel': {
+        compileModules: false,
+        disableDebugTooling: true,
+      }
+    });
+
+    output = createBuilder(subject);
+
+    await output.build();
+
+    expect(
+      output.read()
+    ).to.deep.equal({
+      "foo.js": `import DS from "ember-data";\nexport default DS.Store;`,
+      "bar.js": `import DS from "ember-data";\nexport var User = DS.Model;\nexport var name = DS.attr;`,
+    });
+  });
+});

package.json

@@ -50,6 +50,7 @@
     "@babel/runtime": "^7.8.3",
     "amd-name-resolver": "^1.2.1",
     "babel-plugin-debug-macros": "^0.3.0",
+    "babel-plugin-ember-data-packages-polyfill": "^0.1.0",
     "babel-plugin-ember-modules-api-polyfill": "^2.12.0",
     "babel-plugin-module-resolver": "^3.1.1",
     "broccoli-babel-transpiler": "^7.4.0",
@@ -58,8 +59,9 @@
     "broccoli-source": "^1.1.0",
     "clone": "^2.1.2",
     "ember-cli-babel-plugin-helpers": "^1.1.0",
-    "ember-cli-version-checker": "^2.1.2",
+    "ember-cli-version-checker": "^4.1.0",
     "ensure-posix-path": "^1.0.2",
+    "fixturify-project": "^1.10.0",
     "semver": "^5.5.0"
   },
   "devDependencies": {