embark-area-51/lib/contracts.js

var Compiler = require('./compiler.js');
var toposort = require('toposort');

// 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() {
  this.compiledContracts = this.compileContracts();

  // go through config file first
  for(var className in this.contractsConfig.contracts) {
    var contract = this.contractsConfig.contracts[className];

    contract.className = className;
    contract.args = contract.args || [];

    this.contracts[className] = contract;
  }

  // compile contracts
  for(var className in this.compiledContracts) {
    var compiledContract = this.compiledContracts[className];
    var contractConfig   = this.contractsConfig.contracts[className];

    var contract = this.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;

    if (this.contractsConfig.gas === 'auto') {
      var maxGas;
      if (contract.deploy) {
        maxGas = Math.max(contract.gasEstimates.creation[0], contract.gasEstimates.creation[1], 500000);
      } else {
        maxGas = 500000;
      }
      var adjustedGas = Math.round(maxGas * 1.01);
      contract.gas = adjustedGas;
    } else {
      contract.gas = this.contractsConfig.gas;
    }
    contract.gasPrice = this.contractsConfig.gasPrice;
    contract.type = 'file';
    contract.className = className;

    this.contracts[className] = contract;
  }

  // deal with special configs
  for(var className in this.contracts) {
    var contract = this.contracts[className];

    // if deploy intention is not specified default is true
    if (contract.deploy === undefined) {
      contract.deploy = true;
    }

    if (contract.instanceOf !== undefined) {
      var parentContractName = contract.instanceOf;
      var parentContract = this.contracts[parentContractName];

      if (parentContract === className) {
        this.logger.error(className + ": instanceOf is set to itself");
        continue;
      }

      if (parentContract === undefined) {
        this.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) {
        this.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;
    }
  }

  // remove contracts that don't have code
  for(var className in this.contracts) {
    var contract = this.contracts[className];

    if (contract.code === undefined) {
        this.logger.error(className + " has no code associated");
        delete this.contracts[className];
    }
  }

  this.logger.trace(this.contracts);

  // determine dependencies
  for(var className in this.contracts) {
    var contract = this.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] === "$") {
        if (this.contractDependencies[className] === void 0) {
          this.contractDependencies[className] = [];
        }
        this.contractDependencies[className].push(arg.substr(1));
      }
    }
  }

};

ContractsManager.prototype.getContract = function(className) {
  return this.compiledContracts[className];
};

ContractsManager.prototype.sortContracts = function(contractList) {
  var converted_dependencies = [], i;

  for(var contract in this.contractDependencies) {
    var dependencies = this.contractDependencies[contract];
    for(var 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];

    data.push([
      className.green,
      (contract.deployedAddress || '...').green,
      ((contract.deployedAddress !== undefined) ? "\t\tDeployed".green : "\t\tPending".magenta)
    ]);
  }

  return data;
};

module.exports = ContractsManager;