embark/lib/chain_manager.js

var fs = require('fs');
var web3 = require('web3');
var sha3_256 = require('js-sha3').sha3_256;

ChainManager = function() {
  this.chainManagerConfig = {};
  this.currentChain = {};
  this.file = "";
  this.web3 = null;
}

ChainManager.prototype.loadConfigFile = function(filename) {
  this.file = filename;
  try {
    var obj = JSON.parse(fs.readFileSync(filename));
    this.chainManagerConfig = obj;
  } catch (e) {
    console.warn("error reading " + filename + "; defaulting to empty set");
  }
  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];
  this.web3 = web3;
}

ChainManager.prototype.addContract = function(contractName, code, args, address) {
  this.currentChain.contracts[sha3_256(code + contractName + args.join(','))] = {
    name: contractName,
    address: address
  }
}

ChainManager.prototype.getContract = function(contractName, code, args) {
  return this.currentChain.contracts[sha3_256(code + contractName + args.join(','))];
}

ChainManager.prototype.save = function() {
  fs.writeFileSync(this.file, JSON.stringify(this.chainManagerConfig));
}

module.exports = ChainManager;