Merge branch 'develop'

Author: iurimatias

bin/embark

@@ -15,17 +15,24 @@ var run = function(cmd) {
 }
 
 var deploy = function(env, embarkConfig) {
-  contractFiles = grunt.file.expand(embarkConfig.contracts);
-  destFile = embarkConfig.output
-  Embark.init()
-  Embark.blockchainConfig.loadConfigFile(embarkConfig.blockchainConfig)
-  Embark.contractsConfig.loadConfigFile(embarkConfig.contractsConfig)
-  abi = Embark.deployContracts(env, contractFiles, destFile)
+  var contractFiles = grunt.file.expand(embarkConfig.contracts);
+  var destFile = embarkConfig.output;
+  var chainFile = embarkConfig.chains;
+
+  Embark.init();
+  Embark.blockchainConfig.loadConfigFile(embarkConfig.blockchainConfig);
+  Embark.contractsConfig.loadConfigFile(embarkConfig.contractsConfig);
+
+  if (chainFile === undefined) {
+    chainFile = './chains.json';
+  }
+
+  abi = Embark.deployContracts(env, contractFiles, destFile, chainFile);
   grunt.file.write(destFile, abi);
 }
 
 program
-  .version('0.7.3')
+  .version('0.8.0')
 
 program.command('new [name]').description('New application').action(function(name) {
   if (name === undefined) {

boilerplate/chains.json

@@ -0,0 +1 @@
+{}

boilerplate/package.json

@@ -10,8 +10,8 @@
   "license": "ISC",
   "homepage": "",
   "devDependencies": {
-    "embark-framework": "^0.7.3",
-    "grunt-embark": "^0.2.0",
+    "embark-framework": "^0.8.0",
+    "grunt-embark": "^0.3.0",
     "grunt-contrib-clean": "^0.6.0",
     "grunt-contrib-coffee": "^0.13.0",
     "grunt-contrib-concat": "^0.5.1",

lib/chain_manager.js

@@ -0,0 +1,54 @@
+var fs = require('fs');
+var web3 = require('web3');
+var sha3_256 = require('js-sha3').sha3_256;
+
+ChainManager = function() {
+  this.currentChain = {};
+  this.file = "";
+}
+
+ChainManager.prototype.loadConfigFile = function(filename) {
+  try {
+    var obj = JSON.parse(fs.readFileSync(filename));
+    this.file = filename;
+    this.chainManagerConfig = obj;
+  } catch (e) {
+    throw new Error("error reading " + filename);
+  }
+  return this;
+};
+
+ChainManager.prototype.loadConfig = function(config) {
+  this.chainManagerConfig = config;
+  return this;
+};
+
+ChainManager.prototype.init = function(env, config) {
+  web3.setProvider(new web3.providers.HttpProvider("http://" + config.rpcHost + ":" + config.rpcPort));
+
+  var chainId = web3.eth.getBlock(0).hash;
+
+  if (this.chainManagerConfig[chainId] === undefined) {
+    this.chainManagerConfig[chainId] = {contracts: {}};
+  }
+
+  this.currentChain = this.chainManagerConfig[chainId];
+}
+
+ChainManager.prototype.addContract = function(contractName, code, address) {
+  this.currentChain.contracts[sha3_256(code)] = {
+    name: contractName,
+    address: address
+  }
+}
+
+ChainManager.prototype.getContract = function(code) {
+  return this.currentChain.contracts[sha3_256(code)];
+}
+
+ChainManager.prototype.save = function() {
+  fs.writeFileSync(this.file, JSON.stringify(this.chainManagerConfig));
+}
+
+module.exports = ChainManager;
+

lib/config/contracts.js

@@ -122,6 +122,7 @@ ContractsConfig.prototype.compileContracts = function(env) {
     contract.gasLimit = contractConfig.gas_limit || contract.gasLimit;
     contract.args     = contractConfig.args      || [];
     contract.address  = contractConfig.address;
+    contract.onDeploy = contractConfig.onDeploy  || [];
 
     if (contractConfig.instanceOf !== undefined) {
       contract.types.push('instance');

lib/deploy.js

@@ -10,9 +10,11 @@ sleep = function sleep(ms) {
   while (new Date().getTime() < start + ms);
 }
 
-Deploy = function(env, contractFiles, blockchainConfig, contractsConfig) {
+Deploy = function(env, contractFiles, blockchainConfig, contractsConfig, chainManager) {
   //this.blockchainConfig = (new Config.Blockchain()).loadConfigFile('config/blockchain.yml').config(env);
   this.blockchainConfig = blockchainConfig;
+  this.chainManager = chainManager;
+  this.chainManager.init(env, this.blockchainConfig);
 
   //this.contractsManager = (new Config.Contracts(contractFiles,  blockchainConfig)).loadConfigFile('config/contracts.yml');
   this.contractsManager = contractsConfig;
@@ -56,59 +58,90 @@ Deploy.prototype.deploy_contracts = function(env) {
     className = all_contracts[k];
     contract = this.contractDB[className];
 
+
     if (contract.address !== undefined) {
       this.deployedContracts[className] = contract.address;
 
       //console.log("contract " + className + " at " + contractAddress);
       console.log("contract " + className + " at " + contract.address);
     }
     else {
-      contractObject = web3.eth.contract(contract.compiled.info.abiDefinition);
-
-      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);
-        }
+      var chainContract = this.chainManager.getContract(contract.compiled.code);
+
+      if (chainContract != undefined) {
+        console.log("contract " + className + " is unchanged and already deployed at " + chainContract.address);
       }
+      else {
 
-      contractParams = realArgs;
-      contractParams.push({
-        from: primaryAddress,
-        data: contract.compiled.code,
-        gas: contract.gasLimit,
-        gasPrice: contract.gasPrice
-      });
+        contractObject = web3.eth.contract(contract.compiled.info.abiDefinition);
 
-      console.log('trying to obtain ' + className + ' address...');
+        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);
+          }
+        }
 
-      while((receipt = this.deploy_contract(contractObject, contractParams)) === false) {
-        console.log("timeout... failed to deploy contract.. retrying...");
-      }
+        contractParams = realArgs;
+        contractParams.push({
+          from: primaryAddress,
+          data: contract.compiled.code,
+          gas: contract.gasLimit,
+          gasPrice: contract.gasPrice
+        });
 
-      var contractAddress = receipt.contractAddress;
+        console.log('trying to obtain ' + className + ' address...');
 
-      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);
-      }
+        while((receipt = this.deploy_contract(contractObject, contractParams)) === false) {
+          console.log("timeout... failed to deploy contract.. retrying...");
+        }
 
-      this.deployedContracts[className] = contractAddress;
+        var contractAddress = receipt.contractAddress;
 
-      console.log("deployed " + className + " at " + 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.deployedContracts[className] = contractAddress;
+        this.chainManager.addContract(className, contract.compiled.code, contractAddress);
+        this.chainManager.save();
+
+        console.log("deployed " + className + " at " + contractAddress);
+        this.execute_cmds(contract.onDeploy);
+      }
     }
   }
 
 };
 
+Deploy.prototype.execute_cmds = function(cmds) {
+  if (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_abi_file = function() {
   var result;
 

lib/index.js

@@ -16,12 +16,14 @@ var Deploy = require('./deploy.js');
 var Release = require('./ipfs.js');
 var Config = require('./config/config.js');
 var Compiler = require('./compiler.js');
+var ChainManager = require('./chain_manager.js');
 
 Embark = {
   init: function() {
     this.blockchainConfig = (new Config.Blockchain());
     this.compiler         = (new Compiler(this.blockchainConfig));
     this.contractsConfig  = (new Config.Contracts(this.blockchainConfig, this.compiler));
+    this.chainManager     = (new ChainManager());
   },
 
   tests: function(contractFiles) {
@@ -33,9 +35,11 @@ Embark = {
     chain.startChain(use_tmp);
   },
 
-  deployContracts: function(env, contractFiles, destFile) {
+  deployContracts: function(env, contractFiles, destFile, chainFile) {
     this.contractsConfig.init(contractFiles, env);
-    var deploy = new Deploy(env, contractFiles, this.blockchainConfig.config(env), this.contractsConfig);
+
+    this.chainManager.loadConfigFile(chainFile)
+    var deploy = new Deploy(env, contractFiles, this.blockchainConfig.config(env), this.contractsConfig, this.chainManager);
     deploy.deploy_contracts(env);
     return deploy.generate_abi_file(destFile);
   },

package.json

@@ -1,6 +1,6 @@
 {
   "name": "embark-framework",
-  "version": "0.7.3",
+  "version": "0.8.0",
   "description": "",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"
@@ -17,6 +17,7 @@
     "grunt": "^0.4.5",
     "hashmerge": "^1.0.2",
     "jasmine": "^2.3.1",
+    "js-sha3": "^0.3.1",
     "meteor-build-client": "^0.1.6",
     "methodmissing": "^0.0.3",
     "mkdirp": "^0.5.1",

test/chain_manager.js

@@ -0,0 +1,60 @@
+var ChainManager = require('../lib/chain_manager.js');
+var Config = require('../lib/config/config.js');
+var Blockchain = require('../lib/blockchain.js');
+var assert = require('assert');
+var fs = require('fs');
+
+describe('embark.chain_manager', function() {
+  var chainFile = './test/support/chain_manager.json';
+  fs.writeFileSync(chainFile, '{}');
+
+  var chainManager = (new ChainManager()).loadConfigFile(chainFile);
+  var blockchainConfig = (new Config.Blockchain()).loadConfigFile('test/support/blockchain.yml').config('development');
+
+  describe('#init', function() {
+    chainManager.init('development', blockchainConfig);
+
+    it('should initialize chain', function() {
+      var chain = chainManager.chainManagerConfig['0x629e768beb87dc8c54a475d310a7196e86c97d0006e5a6d34a8217726c90223f']
+      assert.equal(chain != undefined, true);
+    });
+  });
+
+  describe('#addContract', function() {
+
+    it('should register a contract in the chain', function() {
+      chainManager.addContract("Foo", "123456", "0x123");
+
+      var chain = chainManager.chainManagerConfig['0x629e768beb87dc8c54a475d310a7196e86c97d0006e5a6d34a8217726c90223f']
+      var contract = chain.contracts["d7190eb194ff9494625514b6d178c87f99c5973e28c398969d2233f2960a573e"]
+
+      assert.equal(contract.name, "Foo");
+      assert.equal(contract.address, "0x123");
+    });
+
+  });
+
+  describe('#getContract', function() {
+
+    it('should a contract in the chain', function() {
+      var contract = chainManager.getContract("123456");
+
+      assert.equal(contract.name, "Foo");
+      assert.equal(contract.address, "0x123");
+    });
+
+  });
+
+  describe('#save', function() {
+
+    it('should save changes in the chain', function() {
+      chainManager.save();
+
+      var chainFile = './test/support/chain_manager.json';
+      var content = fs.readFileSync(chainFile).toString();
+      assert.equal(content, '{"0x629e768beb87dc8c54a475d310a7196e86c97d0006e5a6d34a8217726c90223f":{"contracts":{"d7190eb194ff9494625514b6d178c87f99c5973e28c398969d2233f2960a573e":{"name":"Foo","address":"0x123"}}}}');
+    });
+
+  });
+
+});