46 lines
1.2 KiB
JavaScript
46 lines
1.2 KiB
JavaScript
const vm = require('vm');
|
|
|
|
class RunCode {
|
|
constructor({logger}) {
|
|
this.logger = logger;
|
|
this.context = Object.assign({}, {
|
|
global, console, exports, require, module, __filename, __dirname, process,
|
|
setTimeout, setInterval, clearTimeout, clearInterval
|
|
});
|
|
}
|
|
|
|
doEval(code) {
|
|
try {
|
|
return vm.runInNewContext(code, this.context);
|
|
} catch(e) {
|
|
this.logger.error(e.message);
|
|
return e.message;
|
|
}
|
|
}
|
|
|
|
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["global"][varName] = code;
|
|
this.context[varName] = code;
|
|
}
|
|
|
|
getWeb3Config() {
|
|
const Web3 = require('web3');
|
|
const provider = this.context.web3.currentProvider;
|
|
let providerUrl;
|
|
if(provider instanceof Web3.providers.HttpProvider){
|
|
providerUrl = provider.host;
|
|
} else if (provider instanceof Web3.providers.WebsocketProvider) {
|
|
providerUrl = provider.connection._url;
|
|
}
|
|
return {defaultAccount: this.context.web3.eth.defaultAccount, providerUrl: providerUrl};
|
|
}
|
|
}
|
|
|
|
module.exports = RunCode;
|