var web3 = require('web3'); var fs = require('fs'); var grunt = require('grunt'); var readYaml = require('read-yaml'); var Config = require('./config/config.js'); var BigNumber = require('bignumber.js'); Deploy = function(env, contractFiles, blockchainConfig, contractsConfig, chainManager, withProvider, withChain, _web3) { if (_web3 !== undefined) { web3 = _web3; } this.contractsManager = contractsConfig; this.contractsConfig = this.contractsManager.config(env); this.deployedContracts = {}; this.blockchainConfig = blockchainConfig; try { if (withProvider) { web3.setProvider(new web3.providers.HttpProvider("http://" + this.blockchainConfig.rpcHost + ":" + this.blockchainConfig.rpcPort)); } primaryAddress = web3.eth.coinbase; web3.eth.defaultAccount = primaryAddress; } catch (e) { throw new Error("==== can't connect to " + this.blockchainConfig.rpcHost + ":" + this.blockchainConfig.rpcPort + " check if an ethereum node is running"); } this.chainManager = chainManager; this.chainManager.init(env, this.blockchainConfig, web3); this.withChain = withChain; console.log("primary account address is : " + primaryAddress); }; Deploy.waitForContract = function(transactionHash, cb) { web3.eth.getTransactionReceipt(transactionHash, function(e, receipt) { if (!e && receipt && receipt.contractAddress !== undefined) { cb(receipt.contractAddress); } else { Deploy.waitForContract(transactionHash, cb); } }); }; Deploy.prototype.deploy_contract = function(contractObject, contractParams, cb) { var callback = function(e, contract) { if(!e && contract.address !== undefined) { cb(contract.address); } else { Deploy.waitForContract(contract.transactionHash, cb); } }; contractParams.push(callback); contractObject["new"].apply(contractObject, contractParams); } Deploy.prototype.deploy_contracts = function(env, cb) { this.contractsManager.compileContracts(env); var all_contracts = this.contractsManager.all_contracts; this.contractDB = this.contractsManager.contractDB; this.deployedContracts = {}; if(this.blockchainConfig.deploy_synchronously) this.deploy_contract_list_synchronously(env, all_contracts, cb); else this.deploy_contract_list(all_contracts.length, env, all_contracts, cb); } Deploy.prototype.deploy_contract_list = function(index, env, all_contracts, cb) { if(index === 0) { cb(); } else { var _this = this; this.deploy_contract_list(index - 1, env, all_contracts, function() { var className = all_contracts[index - 1]; _this.deploy_a_contract(env, className, cb); }); } } Deploy.prototype.deploy_contract_list_synchronously = function(env, all_contracts, cb) { var _this = this ,deployed_contracts_count = 0 all_contracts.forEach(function(className){ _this.deploy_a_contract(env, className, function(){ mark_contract_as_deployed() }); }) function mark_contract_as_deployed(){ deployed_contracts_count ++; if(deployed_contracts_count === all_contracts.length) cb() } } Deploy.prototype.deploy_a_contract = function(env, className, cb) { var contractDependencies = this.contractsManager.contractDependencies; var contract = this.contractDB[className]; if (contract.deploy === false) { console.log("skipping " + className); cb(); return; } var realArgs = []; for (var l = 0; l < contract.args.length; l++) { arg = contract.args[l]; if (arg[0] === "$") { realArgs.push(this.deployedContracts[arg.substr(1)]); } else { realArgs.push(arg); } } if (contract.address !== undefined) { this.deployedContracts[className] = contract.address; console.log("contract " + className + " at " + contract.address); cb(); } else { var chainContract = this.chainManager.getContract(className, contract.compiled.code, realArgs); if (chainContract != undefined && web3.eth.getCode(chainContract.address) !== "0x") { console.log("contract " + className + " is unchanged and already deployed at " + chainContract.address); this.deployedContracts[className] = chainContract.address; this.execute_cmds(contract.onDeploy); cb(); } else { contractObject = web3.eth.contract(contract.compiled.info.abiDefinition); contractParams = realArgs.slice(); contractParams.push({ from: primaryAddress, data: contract.compiled.code, gas: contract.gasLimit, gasPrice: contract.gasPrice }); console.log('trying to obtain ' + className + ' address...'); var _this = this; this.deploy_contract(contractObject, contractParams, function(contractAddress) { if (web3.eth.getCode(contractAddress) === "0x") { console.log("========="); console.log("contract was deployed at " + contractAddress + " but doesn't seem to be working"); console.log("try adjusting your gas values"); console.log("========="); } else { console.log("deployed " + className + " at " + contractAddress); _this.chainManager.addContract(className, contract.compiled.code, realArgs, contractAddress); if (_this.withChain) { _this.chainManager.save(); } } _this.deployedContracts[className] = contractAddress; _this.execute_cmds(contract.onDeploy); cb(); }); } } }; Deploy.prototype.execute_cmds = function(cmds) { if (cmds == undefined || cmds.length === 0) return; eval(this.generate_abi_file()); for (var i = 0; i < cmds.length; i++) { var cmd = cmds[i]; for(className in this.deployedContracts) { var contractAddress = this.deployedContracts[className]; var re = new RegExp("\\$" + className, 'g'); cmd = cmd.replace(re, '"' + contractAddress + '"'); } console.log("executing: " + cmd); eval(cmd); } } Deploy.prototype.generate_provider_file = function() { var result = ""; result += "var web3 = new Web3();"; result += "web3.setProvider(new web3.providers.HttpProvider('http://" + this.blockchainConfig.rpcHost + ":" + this.blockchainConfig.rpcPort + "'));"; result += "web3.eth.defaultAccount = web3.eth.accounts[0];"; return result; } Deploy.prototype.generate_abi_file = function() { var result = ""; result += 'blockchain = '+JSON.stringify(this.blockchainConfig)+';'; for(className in this.deployedContracts) { var deployedContract = this.deployedContracts[className]; var contract = this.contractDB[className]; var abi = JSON.stringify(contract.compiled.info.abiDefinition); var contractAddress = deployedContract; console.log('address is ' + contractAddress); result += className + "Abi = " + abi + ";"; result += className + "Contract = web3.eth.contract(" + className + "Abi);"; result += className + " = " + className + "Contract.at('" + contractAddress + "');"; } result += 'contractDB = '+JSON.stringify(this.contractDB)+';' return result; }; Deploy.prototype.generate_and_write_abi_file = function(destFile) { var result = this.generate_abi_file(); grunt.file.write(destFile, result); }; module.exports = Deploy;