mirror of https://github.com/embarklabs/embark.git
122 lines
4.1 KiB
JavaScript
122 lines
4.1 KiB
JavaScript
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.addCheck = options.addCheck;
|
|
this.embark = embark;
|
|
this.contractsConfig = embark.config.contractsConfig;
|
|
|
|
this.registerAfterDeployAction();
|
|
this.registerOnDeployAction();
|
|
}
|
|
|
|
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((result) => {
|
|
cb(null, result);
|
|
}).catch(cb);
|
|
}
|
|
|
|
registerAfterDeployAction() {
|
|
const self = this;
|
|
|
|
this.embark.registerActionForEvent("contracts:deploy:afterAll", (cb) => {
|
|
let afterDeployCmds = self.contractsConfig.afterDeploy || [];
|
|
|
|
async.mapLimit(afterDeployCmds, 1, (cmd, nextMapCb) => {
|
|
self.replaceWithAddresses(cmd, nextMapCb);
|
|
}, (err, result) => {
|
|
if (err) {
|
|
self.logger.trace(err);
|
|
return cb(new Error("error running afterDeploy"));
|
|
}
|
|
let onDeployCode = result;
|
|
|
|
// TODO: convert to for to avoid repeated callback
|
|
for(let cmd of onDeployCode) {
|
|
self.logger.info("==== executing: " + cmd);
|
|
try {
|
|
self.events.request('runcode:eval', cmd);
|
|
} catch(e) {
|
|
if (e.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');
|
|
}
|
|
return cb(new Error(e));
|
|
}
|
|
}
|
|
cb();
|
|
});
|
|
});
|
|
}
|
|
|
|
registerOnDeployAction() {
|
|
const self = this;
|
|
|
|
this.embark.registerActionForEvent("deploy:contract:deployed", (params, cb) => {
|
|
let contract = params.contract;
|
|
|
|
if (!contract.onDeploy) {
|
|
return cb();
|
|
}
|
|
|
|
self.logger.info(__('executing onDeploy commands'));
|
|
|
|
let onDeployCmds = contract.onDeploy;
|
|
|
|
async.mapLimit(onDeployCmds, 1, (cmd, nextMapCb) => {
|
|
self.replaceWithAddresses(cmd, nextMapCb);
|
|
}, (err, result) => {
|
|
if (err) {
|
|
return cb(new Error("error running onDeploy for " + contract.className.cyan));
|
|
}
|
|
let onDeployCode = result;
|
|
|
|
// TODO: convert to for to avoid repeated callback
|
|
for(let cmd of onDeployCode) {
|
|
self.logger.info("==== executing: " + cmd);
|
|
try {
|
|
self.events.request('runcode:eval', cmd);
|
|
} catch(e) {
|
|
if (e.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');
|
|
}
|
|
return cb(new Error(e));
|
|
}
|
|
}
|
|
cb();
|
|
});
|
|
});
|
|
}
|
|
|
|
}
|
|
|
|
module.exports = SpecialConfigs;
|