better node register

lib/6to5/node.js

@@ -5,18 +5,8 @@ var _         = require("lodash");
 exports.browserify = require("./browserify");
 exports.middleware = require("./middleware");
 
-exports.register = function (includeNodeModules) {
-  var old = require.extensions[".js"];
-
-  require.extensions[".js"] = function (m, filename) {
-    if (!includeNodeModules && filename.indexOf("node_modules") >= 0) {
-      return old.apply(this, arugments);
-    }
-
-    m._compile(exports.transformFileSync(filename, {
-      sourceMap: true
-    }), filename);
-  };
+exports.register = function () {
+  require("./register");
 };
 
 exports.transform = transform;

lib/6to5/register.js

@@ -0,0 +1,16 @@
+var maps = {};
+var old  = require.extensions[".js"];
+
+require.extensions[".js"] = function (m, filename) {
+  if (filename.indexOf("node_modules") >= 0) {
+    return old.apply(this, arguments);
+  }
+
+  var result = exports.transformFileSync(filename, {
+    sourceMapObject: true
+  });
+
+  maps[flename] = result.map;
+
+  m._compile(result.code, filename);
+};