Close #103 PR: Replace babel require hook with transformFileSync.

Author: floatdrop

lib/babel.js

@@ -1,26 +1,28 @@
 'use strict';
 var resolveFrom = require('resolve-from');
 var createEspowerPlugin = require('babel-plugin-espower/create');
+var requireFromString = require('require-from-string');
 var hasGenerators = parseInt(process.version.slice(1), 10) > 0;
 var path = process.argv[2];
 
 var options = {
-	only: path,
 	blacklist: hasGenerators ? ['regenerator'] : [],
-	optional: hasGenerators ? ['asyncToGenerator'] : [],
+	optional: hasGenerators ? ['asyncToGenerator', 'runtime'] : ['runtime'],
 	plugins: [
 		createEspowerPlugin(require('babel-core'), {
 			patterns: require('./enhance-assert').PATTERNS
 		})
 	]
 };
 
-try {
-	var localBabel = resolveFrom('.', 'babel-core/register') || resolveFrom('.', 'babel/register');
+var babel;
 
-	require(localBabel)(options);
+try {
+	var localBabel = resolveFrom('.', 'babel-core') || resolveFrom('.', 'babel');
+	babel = require(localBabel);
 } catch (err) {
-	require('babel-core/register')(options);
+	babel = require('babel-core');
 }
 
-require(path);
+var transpiled = babel.transformFileSync(path, options);
+requireFromString(transpiled.code, path);

package.json

@@ -64,9 +64,10 @@
     "ava-init": "^0.1.0",
     "babel-core": "^5.8.23",
     "babel-plugin-espower": "^1.0.0",
+    "babel-runtime": "^5.8.29",
     "bluebird": "^3.0.0",
     "chalk": "^1.0.0",
-    "co": "^4.6.0",
+    "co": "floatdrop/co#e6c2a50",
     "core-assert": "^0.1.0",
     "empower": "^1.0.2",
     "figures": "^1.4.0",
@@ -79,6 +80,7 @@
     "power-assert-formatter": "^1.1.0",
     "power-assert-renderers": "^0.1.0",
     "pretty-ms": "^2.0.0",
+    "require-from-string": "^1.0.0",
     "resolve-from": "^1.0.0",
     "serialize-error": "^1.0.0",
     "set-immediate-shim": "^1.0.1",

test/fork.js

@@ -35,7 +35,7 @@ test('rejects on error and streams output', function (t) {
 			buffer += data;
 		})
 		.catch(function () {
-			t.ok(/Cannot find module/.test(buffer));
+			t.ok(/no such file or directory/.test(buffer));
 			t.end();
 		});
 });