84 lines
2.8 KiB
JavaScript
84 lines
2.8 KiB
JavaScript
var semver = require("semver");
|
|
var vm = require("vm");
|
|
|
|
function safe_log(arg, level) {
|
|
if (arg) switch (typeof arg) {
|
|
case "function":
|
|
return arg.toString();
|
|
case "object":
|
|
if (/Error$/.test(arg.name)) return arg.toString();
|
|
arg.constructor.toString();
|
|
if (level--) for (var key in arg) {
|
|
if (!Object.getOwnPropertyDescriptor(arg, key).get) {
|
|
arg[key] = safe_log(arg[key], level);
|
|
}
|
|
}
|
|
}
|
|
return arg;
|
|
}
|
|
|
|
var FUNC_TOSTRING = [
|
|
"Function.prototype.toString = Function.prototype.valueOf = function() {",
|
|
" var id = 100000;",
|
|
" return function() {",
|
|
' if (this === Array) return "[Function: Array]";',
|
|
' if (this === Object) return "[Function: Object]";',
|
|
" var i = this.name;",
|
|
' if (typeof i != "number") {',
|
|
" i = ++id;",
|
|
].concat(Object.getOwnPropertyDescriptor(Function.prototype, "name").configurable ? [
|
|
' Object.defineProperty(this, "name", {',
|
|
" get: function() {",
|
|
" return i;",
|
|
" }",
|
|
" });",
|
|
] : [], [
|
|
" }",
|
|
' return "[Function: " + i + "]";',
|
|
" }",
|
|
"}();",
|
|
'Object.defineProperty(Function.prototype, "valueOf", { enumerable: false });',
|
|
]).join("\n");
|
|
exports.run_code = function(code) {
|
|
var stdout = "";
|
|
var original_write = process.stdout.write;
|
|
process.stdout.write = function(chunk) {
|
|
stdout += chunk;
|
|
};
|
|
try {
|
|
vm.runInNewContext([
|
|
FUNC_TOSTRING,
|
|
"!function() {",
|
|
code,
|
|
"}();",
|
|
].join("\n"), {
|
|
console: {
|
|
log: function(msg) {
|
|
if (arguments.length == 1 && typeof msg == "string") {
|
|
return console.log("%s", msg);
|
|
}
|
|
return console.log.apply(console, [].map.call(arguments, function(arg) {
|
|
return safe_log(arg, 3);
|
|
}));
|
|
}
|
|
}
|
|
}, { timeout: 5000 });
|
|
return stdout;
|
|
} catch (ex) {
|
|
return ex;
|
|
} finally {
|
|
process.stdout.write = original_write;
|
|
}
|
|
};
|
|
exports.same_stdout = semver.satisfies(process.version, "0.12") ? function(expected, actual) {
|
|
if (typeof expected != typeof actual) return false;
|
|
if (typeof expected != "string") {
|
|
if (expected.name != actual.name) return false;
|
|
expected = expected.message.slice(expected.message.lastIndexOf("\n") + 1);
|
|
actual = actual.message.slice(actual.message.lastIndexOf("\n") + 1);
|
|
}
|
|
return expected == actual;
|
|
} : function(expected, actual) {
|
|
return typeof expected == typeof actual && expected.toString() == actual.toString();
|
|
};
|