const RunCode = require('./runCode.js'); const EmbarkJS = require('embarkjs'); class CodeRunner { constructor(options) { this.plugins = options.plugins; this.logger = options.logger; this.events = options.events; this.ipc = options.ipc; this.commands = []; this.runCode = new RunCode(); let self = this; if (this.ipc.isServer()) { this.ipc.on('runcode:getCommands', (_err, callback) => { let result = {web3Config: self.runCode.getWeb3Config(), commands: self.commands}; callback(null, result); }); } if (this.ipc.isClient() && this.ipc.connected) { this.ipc.listenTo('runcode:newCommand', function (command) { if (command.varName) { self.events.emit("runcode:register", command.varName, command.code); } else { self.events.request("runcode:eval", command.code); } }); } else { this.runCode.registerVar('EmbarkJS', EmbarkJS); this.events.on('code-generator-ready', () => { this.events.request('code-generator:embarkjs:initialization-code', (code) => { this.runCode.doEval(code); }); }) } this.events.on("runcode:register", (varName, code) => { if (self.ipc.isServer() && varName !== 'web3') { self.commands.push({varName, code}); self.ipc.broadcast("runcode:newCommand", {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) => { if (!cb) { cb = function() {}; } const awaitIdx = code.indexOf('await'); if (awaitIdx > -1) { if (awaitIdx < 2) { let end = code.length; if (code[end - 1] === ';') { end--; // Remove the `;` because we add function calls } code = code.substring(5, end); // remove await keyword } else { code = `(async function() {${code}})();`; } } 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); }); } } module.exports = CodeRunner;