const stringReplaceAsync = require('string-replace-async'); const async = require('async'); class SpecialConfigs { constructor(embark, options) { this.logger = embark.logger; this.events = embark.events; this.buildDir = options.buildDir; this.embark = embark; this.config = embark.config; this.registerAfterDeployAction(); this.registerOnDeployAction(); this.registerDeployIfAction(); } replaceWithENSAddress(cmd, cb) { const self = this; let regex = /\'[a-zA-Z0-9.]+\.eth\'/g; return stringReplaceAsync.seq(cmd, regex, (ensDomain) => { ensDomain = ensDomain.slice(1, ensDomain.length - 1); return (new Promise((resolve, reject) => { self.events.request("ens:resolve", ensDomain, (err, address) => { if(err) { return reject(new Error(err)); } address = `'${address}'`; return resolve(address); }); })); }).then((address) => { cb(null, address); }).catch(cb); } replaceWithAddresses(cmd, cb) { const self = this; let regex = /\$\w+/g; stringReplaceAsync.seq(cmd, regex, (match) => { return (new Promise((resolve, reject) => { let referedContractName = match.slice(1); self.events.request('contracts:contract', referedContractName, (referedContract) => { if (!referedContract) { self.logger.error(referedContractName + ' does not exist'); self.logger.error("error running cmd: " + cmd); return reject(new Error("ReferedContractDoesNotExist")); } if (referedContract && referedContract.deploy === false) { self.logger.error(referedContractName + " exists but has been set to not deploy"); self.logger.error("error running cmd: " + cmd); return reject(new Error("ReferedContracSetToNotdeploy")); } if (referedContract && !referedContract.deployedAddress) { self.logger.error("couldn't find a valid address for " + referedContractName + ". has it been deployed?"); self.logger.error("error running cmd: " + cmd); return reject(new Error("ReferedContractAddressNotFound")); } return resolve(referedContract.deployedAddress); }); })); }).then((address) => { cb(null, address); }).catch(cb); } registerAfterDeployAction() { const self = this; this.embark.registerActionForEvent("contracts:deploy:afterAll", (cb) => { let afterDeployCmds = self.config.contractsConfig.afterDeploy || []; async.mapLimit(afterDeployCmds, 1, (cmd, nextMapCb) => { async.waterfall([ function replaceWithAddresses(next) { self.replaceWithAddresses(cmd, next); }, self.replaceWithENSAddress.bind(self) ], nextMapCb); }, (err, onDeployCode) => { if (err) { self.logger.trace(err); return cb(new Error("error running afterDeploy")); } self.runOnDeployCode(onDeployCode, cb); }); }); } runOnDeployCode(onDeployCode, callback, silent) { const self = this; const logFunction = silent ? self.logger.trace.bind(self.logger) : self.logger.info.bind(self.logger); async.each(onDeployCode, (cmd, eachCb) => { if (!cmd) { return eachCb(); } logFunction("==== executing: " + cmd); self.events.request('runcode:eval', cmd, (err) => { if (err && err.message.indexOf("invalid opcode") >= 0) { self.logger.error('the transaction was rejected; this usually happens due to a throw or a require, it can also happen due to an invalid operation'); } eachCb(err); }); }, callback); } registerOnDeployAction() { const self = this; this.embark.registerActionForEvent("deploy:contract:deployed", (params, cb) => { let contract = params.contract; if (!contract.onDeploy || contract.deploy === false) { return cb(); } if (!contract.silent) { self.logger.info(__('executing onDeploy commands')); } let onDeployCmds = contract.onDeploy; async.mapLimit(onDeployCmds, 1, (cmd, nextMapCb) => { async.waterfall([ function replaceWithAddresses(next) { self.replaceWithAddresses(cmd, next); }, self.replaceWithENSAddress.bind(self) ], (err, code) => { if (err) { self.logger.error(err.message || err); return nextMapCb(); // Don't return error as we just skip the failing command } nextMapCb(null, code); }); }, (err, onDeployCode) => { if (err) { return cb(new Error("error running onDeploy for " + contract.className.cyan)); } self.runOnDeployCode(onDeployCode, cb, contract.silent); }); }); } registerDeployIfAction() { const self = this; self.embark.registerActionForEvent("deploy:contract:shouldDeploy", (params, cb) => { let cmd = params.contract.deployIf; if (!cmd) { return cb(params); } self.events.request('runcode:eval', cmd, (err, result) => { if (err) { self.logger.error(params.contract.className + ' deployIf directive has an error; contract will not deploy'); self.logger.error(err); params.shouldDeploy = false; } else if (!result) { self.logger.info(params.contract.className + ' deployIf directive returned false; contract will not deploy'); params.shouldDeploy = false; } cb(params); }); }); } } module.exports = SpecialConfigs;