Isolate the code runner
This commit is contained in:
parent
3fe1cd4a2c
commit
de57e48cff
|
@ -44,6 +44,10 @@ class REPL {
|
|||
writer: this.enhancedWriter.bind(this)
|
||||
});
|
||||
|
||||
this.events.request("runcode:getContext", (context) => {
|
||||
this.replServer.context = context;
|
||||
});
|
||||
|
||||
this.replServer.on("exit", () => {
|
||||
process.exit();
|
||||
});
|
||||
|
|
|
@ -8,13 +8,12 @@ class CodeRunner {
|
|||
this.events = options.events;
|
||||
this.ipc = options.ipc;
|
||||
this.commands = [];
|
||||
this.runCode = new RunCode();
|
||||
let self = this;
|
||||
// necessary to init the context
|
||||
RunCode.initContext();
|
||||
|
||||
if (this.ipc.isServer()) {
|
||||
this.ipc.on('runcode:getCommands', (_err, callback) => {
|
||||
let result = {web3Config: RunCode.getWeb3Config(), commands: self.commands};
|
||||
let result = {web3Config: self.runCode.getWeb3Config(), commands: self.commands};
|
||||
callback(null, result);
|
||||
});
|
||||
}
|
||||
|
@ -34,7 +33,11 @@ class CodeRunner {
|
|||
self.commands.push({varName, code});
|
||||
self.ipc.broadcast("runcode:newCommand", {varName, code});
|
||||
}
|
||||
RunCode.registerVar(varName, code);
|
||||
self.runCode.registerVar(varName, code);
|
||||
});
|
||||
|
||||
this.events.setCommandHandler('runcode:getContext', (cb) => {
|
||||
cb(self.runCode.context);
|
||||
});
|
||||
|
||||
this.events.setCommandHandler('runcode:eval', (code, cb, forConsoleOnly = false) => {
|
||||
|
@ -53,14 +56,17 @@ class CodeRunner {
|
|||
code = `(async function() {${code}})();`;
|
||||
}
|
||||
}
|
||||
let result = RunCode.doEval(code);
|
||||
if (result instanceof Promise) {
|
||||
return result.then((value) => cb(null, value)).catch(cb);
|
||||
}
|
||||
let result = self.runCode.doEval(code);
|
||||
|
||||
if (forConsoleOnly && self.ipc.isServer()) {
|
||||
self.commands.push({code});
|
||||
self.ipc.broadcast("runcode:newCommand", {code});
|
||||
}
|
||||
|
||||
if (result instanceof Promise) {
|
||||
return result.then((value) => cb(null, value)).catch(cb);
|
||||
}
|
||||
|
||||
cb(null, result);
|
||||
});
|
||||
}
|
||||
|
|
|
@ -1,31 +1,27 @@
|
|||
/*eslint no-unused-vars: off*/
|
||||
let __mainContext = this;
|
||||
const vm = require('vm');
|
||||
|
||||
function initContext() {
|
||||
doEval("__mainContext = this");
|
||||
class RunCode {
|
||||
constructor() {
|
||||
this.context = Object.assign({}, global.this);
|
||||
}
|
||||
|
||||
doEval(code) {
|
||||
return vm.runInNewContext(code, this.context);
|
||||
}
|
||||
|
||||
registerVar(varName, code) {
|
||||
// TODO: Update all the code being dependent of web3
|
||||
// To identify, look at the top of the file for something like:
|
||||
// /*global web3*/
|
||||
if (varName === 'web3') {
|
||||
global.web3 = code;
|
||||
}
|
||||
this.context[varName] = code;
|
||||
}
|
||||
|
||||
getWeb3Config() {
|
||||
return {defaultAccount: this.context.web3.eth.defaultAccount, provider: this.context.web3.currentProvider};
|
||||
}
|
||||
}
|
||||
|
||||
// ======================
|
||||
// the eval is used for evaluating some of the contact calls for different purposes
|
||||
// this should be at least moved to a different process and scope
|
||||
// for now it is defined here
|
||||
// ======================
|
||||
function doEval(code) {
|
||||
// TODO: add trace log here
|
||||
return eval(code);
|
||||
}
|
||||
|
||||
function registerVar(varName, code) {
|
||||
__mainContext[varName] = code;
|
||||
}
|
||||
|
||||
function getWeb3Config() {
|
||||
return {defaultAccount:__mainContext.web3.eth.defaultAccount, provider: __mainContext.web3.currentProvider};
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
doEval,
|
||||
registerVar,
|
||||
initContext,
|
||||
getWeb3Config
|
||||
};
|
||||
module.exports = RunCode;
|
||||
|
|
Loading…
Reference in New Issue