var Compiler = require('./compiler.js'); var toposort = require('toposort'); var async = require('async'); // TODO: create a contract object var ContractsManager = function(options) { this.contractFiles = options.contractFiles; this.contractsConfig = options.contractsConfig; this.contracts = {}; this.logger = options.logger; this.contractDependencies = {}; }; ContractsManager.prototype.compileContracts = function() { var compiler = new Compiler(); return compiler.compile_solidity(this.contractFiles); }; ContractsManager.prototype.build = function(done) { var self = this; async.waterfall([ function compileContracts(callback) { var compiler = new Compiler(); try { self.compiledContracts = compiler.compile_solidity(self.contractFiles); return callback(); } catch(err) { return callback(new Error(err.message)); } }, function prepareContractsFromConfig(callback) { var className, contract; for(className in self.contractsConfig.contracts) { contract = self.contractsConfig.contracts[className]; contract.className = className; contract.args = contract.args || []; self.contracts[className] = contract; } callback(); }, function prepareContractsFromCompilation(callback) { var className, compiledContract, contractConfig, contract; var maxGas, adjustedGas; for(className in self.compiledContracts) { compiledContract = self.compiledContracts[className]; contractConfig = self.contractsConfig.contracts[className]; contract = self.contracts[className] || {className: className, args: []}; contract.code = compiledContract.code; contract.runtimeBytecode = compiledContract.runtimeBytecode; contract.gasEstimates = compiledContract.gasEstimates; contract.functionHashes = compiledContract.functionHashes; contract.abiDefinition = compiledContract.abiDefinition; contract.gas = (contractConfig && contractConfig.gas) || self.contractsConfig.gas; if (contract.deploy === undefined) { contract.deploy = true; } if (contract.gas === 'auto') { if (contract.deploy) { maxGas = Math.max(contract.gasEstimates.creation[0], contract.gasEstimates.creation[1], 500000); } else { maxGas = 500000; } // TODO: put a check so it doesn't go over the block limit adjustedGas = Math.round(maxGas * 1.40); contract.gas = adjustedGas; } contract.gasPrice = contract.gasPrice || self.contractsConfig.gasPrice; contract.type = 'file'; contract.className = className; self.contracts[className] = contract; } callback(); }, function setDeployIntention(callback) { var className, contract; for(className in self.contracts) { contract = self.contracts[className]; contract.deploy = (contract.deploy === undefined) || contract.deploy; } }, function dealWithSpecialConfigs(callback) { var className, contract, parentContractName, parentContract; for(className in self.contracts) { contract = self.contracts[className]; if (contract.instanceOf === undefined) { continue; } parentContractName = contract.instanceOf; parentContract = self.contracts[parentContractName]; if (parentContract === className) { self.logger.error(className + ": instanceOf is set to itself"); continue; } if (parentContract === undefined) { self.logger.error(className + ": couldn't find instanceOf contract " + parentContractName); continue; } if (parentContract.args && parentContract.args.length > 0 && contract.args === []) { contract.args = parentContract.args; } if (contract.code !== undefined) { self.logger.error(className + " has code associated to it but it's configured as an instanceOf " + parentContractName); } contract.code = parentContract.code; contract.runtimeBytecode = parentContract.runtimeBytecode; contract.gasEstimates = parentContract.gasEstimates; contract.functionHashes = parentContract.functionHashes; contract.abiDefinition = parentContract.abiDefinition; contract.gas = contract.gas || parentContract.gas; contract.gasPrice = contract.gasPrice || parentContract.gasPrice; } callback(); }, function removeContractsWithNoCode(callback) { var className, contract; for(className in self.contracts) { contract = self.contracts[className]; if (contract.code === undefined) { self.logger.error(className + " has no code associated"); delete self.contracts[className]; } } self.logger.trace(self.contracts); callback(); }, function determineDependencies(callback) { var className, contract; for(className in self.contracts) { contract = self.contracts[className]; if (contract.args === []) continue; var ref = contract.args; for (var j = 0; j < ref.length; j++) { var arg = ref[j]; if (arg[0] === "$") { self.contractDependencies[className] = self.contractDependencies[className] || []; self.contractDependencies[className].push(arg.substr(1)); } } } callback(); } ], function(err, result) { self.logger.trace("finished".underline); if (err) { //self.logger.debug(err.stack); done(err); } else { done(null, self); } }); }; ContractsManager.prototype.getContract = function(className) { return this.contracts[className]; }; ContractsManager.prototype.sortContracts = function(contractList) { var converted_dependencies = [], i; for(var contract in this.contractDependencies) { var dependencies = this.contractDependencies[contract]; for(i=0; i < dependencies.length; i++) { converted_dependencies.push([contract, dependencies[i]]); } } var orderedDependencies = toposort(converted_dependencies).reverse(); var newList = contractList.sort(function(a,b) { var order_a = orderedDependencies.indexOf(a.className); var order_b = orderedDependencies.indexOf(b.className); return order_a - order_b; }); return newList; }; // TODO: should be built contracts ContractsManager.prototype.listContracts = function() { var contracts = []; for(var className in this.contracts) { var contract = this.contracts[className]; contracts.push(contract); } return this.sortContracts(contracts); }; ContractsManager.prototype.contractsState = function() { var data = []; for(var className in this.contracts) { var contract = this.contracts[className]; var contractData; if (contract.deploy === false) { contractData = [ className.green, 'Interface or set to not deploy'.green, "\t\tn/a".green ]; } else if (contract.error) { contractData = [ className.green, (contract.error).red, '\t\tError'.red ]; } else { contractData = [ className.green, (contract.deployedAddress || '...').green, ((contract.deployedAddress !== undefined) ? "\t\tDeployed".green : "\t\tPending".magenta) ]; } data.push(contractData); } return data; }; module.exports = ContractsManager;