funcunit
Version:
<!-- @hide title
175 lines (130 loc) • 4.92 kB
JavaScript
"use strict";
// istanbul ignore next
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj["default"] = obj; return newObj; } }
// istanbul ignore next
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
var _pathIsAbsolute = require("path-is-absolute");
var _pathIsAbsolute2 = _interopRequireDefault(_pathIsAbsolute);
var _commander = require("commander");
var _commander2 = _interopRequireDefault(_commander);
var _module2 = require("module");
var _module3 = _interopRequireDefault(_module2);
var _util = require("util");
var _path = require("path");
var _path2 = _interopRequireDefault(_path);
var _repl = require("repl");
var _repl2 = _interopRequireDefault(_repl);
var _babelCore = require("babel-core");
var babel = _interopRequireWildcard(_babelCore);
var _vm = require("vm");
var _vm2 = _interopRequireDefault(_vm);
var _lodash = require("lodash");
var _lodash2 = _interopRequireDefault(_lodash);
var program = new _commander2["default"].Command("babel-node");
program.option("-e, --eval [script]", "Evaluate script");
program.option("-p, --print [code]", "Evaluate script and print result");
program.option("-i, --ignore [regex]", "Ignore all files that match this regex when using the require hook");
program.option("-x, --extensions [extensions]", "List of extensions to hook into [.es6,.js,.es,.jsx]");
program.option("-r, --stage [stage]", "Enable support for specific ECMAScript stages");
program.option("-w, --whitelist [whitelist]", "Whitelist of transformers separated by comma to ONLY use", _babelCore.util.list);
program.option("-b, --blacklist [blacklist]", "Blacklist of transformers separated by comma to NOT use", _babelCore.util.list);
program.option("-o, --optional [optional]", "List of optional transformers separated by comma to enable", _babelCore.util.list);
var pkg = require("../package.json");
program.version(pkg.version);
program.usage("[options] [ -e script | script.js ] [arguments]");
program.parse(process.argv);
//
babel.register({
extensions: program.extensions,
blacklist: program.blacklist,
whitelist: program.whitelist,
optional: program.optional,
ignore: program.ignore,
stage: program.stage
});
//
var _eval = function _eval(code, filename) {
code = code.trim();
if (!code) return undefined;
code = babel.transform(code, {
filename: filename,
blacklist: program.blacklist,
whitelist: program.whitelist,
optional: program.optional,
stage: program.stage
}).code;
return _vm2["default"].runInThisContext(code, {
filename: filename
});
};
if (program.eval || program.print) {
var code = program.eval;
if (!code || code === true) code = program.print;
global.__filename = "[eval]";
global.__dirname = process.cwd();
var _module = new _module3["default"](global.__filename);
_module.filename = global.__filename;
_module.paths = _module3["default"]._nodeModulePaths(global.__dirname);
global.exports = _module.exports;
global.module = _module;
global.require = _module.require.bind(_module);
var result = _eval(code, global.__filename);
if (program.print) {
var output = _lodash2["default"].isString(result) ? result : _util.inspect(result);
process.stdout.write(output + "\n");
}
} else {
if (program.args.length) {
// slice all arguments up to the first filename since they're babel args that we handle
var args = process.argv.slice(2);
var i = 0;
var ignoreNext = false;
_lodash2["default"].each(args, function (arg, i2) {
if (ignoreNext) {
ignoreNext = false;
return;
}
if (arg[0] === "-") {
var parsedArg = program[arg.slice(2)];
if (parsedArg && parsedArg !== true) {
ignoreNext = true;
}
} else {
i = i2;
return false;
}
});
args = args.slice(i);
// make the filename absolute
var filename = args[0];
if (!_pathIsAbsolute2["default"](filename)) args[0] = _path2["default"].join(process.cwd(), filename);
// add back on node and concat the sliced args
process.argv = ["node"].concat(args);
process.execArgv.unshift(__filename);
_module3["default"].runMain();
} else {
replStart();
}
}
function replStart() {
_repl2["default"].start({
prompt: "> ",
input: process.stdin,
output: process.stdout,
eval: replEval,
useGlobal: true
});
}
function replEval(code, context, filename, callback) {
var err;
var result;
try {
if (code[0] === "(" && code[code.length - 1] === ")") {
code = code.slice(1, -1); // remove "(" and ")"
}
result = _eval(code, filename);
} catch (e) {
err = e;
}
callback(err, result);
}