es module support

Author: guybedford

README.md

@@ -71,6 +71,7 @@ require("babel-core").transform("code", {
       },
       moduleName: "foo",
       systemGlobal: "SystemJS",
+      esModule: true
     }]
   ]
 });

src/index.js

@@ -16,6 +16,18 @@ export default function ({ types: t }) {
     })
   `);
 
+  const buildFactoryEs = template(`
+    (function ($__require, $__exports, $__module) {
+      var _retrieveGlobal = SYSTEM_GLOBAL.registry.get("@@global-helpers").prepareGlobal($__module.id, EXPORT_NAME, GLOBALS);
+      (BODY)(this)
+      var $__moduleValue = _retrieveGlobal();
+      Object.defineProperty($__moduleValue, '__esModule', {
+        value: true
+      });
+      return $__moduleValue;
+    })
+  `);
+
   const buildGlobal = template(`
     $__global[NAME] = VALUE;
   `);
@@ -74,7 +86,7 @@ export default function ({ types: t }) {
 
           const systemGlobal = t.identifier(opts.systemGlobal || "System");
 
-          const factory = buildFactory({
+          const factory = (opts.esModule ? buildFactoryEs : buildFactory)({
             SYSTEM_GLOBAL: systemGlobal,
             EXPORT_NAME: exportName,
             GLOBALS: globals || t.nullLiteral(),

test/fixtures/es-module/actual.js

@@ -0,0 +1,6 @@
+var foo = function bar() {
+};
+
+function baz() {
+  return "qux";
+}

test/fixtures/es-module/expected.js

@@ -0,0 +1,21 @@
+System.registerDynamic([], false, function ($__require, $__exports, $__module) {
+  var _retrieveGlobal = System.registry.get("@@global-helpers").prepareGlobal($__module.id, null, null);
+
+  (function ($__global) {
+    $__global["baz"] = baz;
+    var foo = $__global["foo"];
+    var foo = function bar() {};
+
+    function baz() {
+      return "qux";
+    }
+    $__global["foo"] = foo;
+  })(this);
+
+  var $__moduleValue = _retrieveGlobal();
+
+  Object.defineProperty($__moduleValue, '__esModule', {
+    value: true
+  });
+  return $__moduleValue;
+});

test/fixtures/es-module/options.js

@@ -0,0 +1,3 @@
+module.exports = {
+  esModule: true
+};