embark-area-51/old_lib/config/contracts.js

165 lines
4.6 KiB
JavaScript
Raw Normal View History

var readYaml = require('read-yaml');
var fs = require('fs');
var toposort = require('toposort');
2015-07-04 03:23:21 +00:00
ContractsConfig = function(blockchainConfig, compiler) {
2015-07-03 11:59:35 +00:00
this.blockchainConfig = blockchainConfig;
2015-07-04 03:23:21 +00:00
this.compiler = compiler;
2015-07-03 11:59:35 +00:00
this.contractFiles = [];
}
2015-08-01 15:19:21 +00:00
ContractsConfig.prototype.init = function(files, env) {
this.all_contracts = [];
this.contractDB = {};
this.contractFiles = files;
this.contractDependencies = {};
2015-08-01 02:50:51 +00:00
this.contractStubs = {};
2015-07-04 20:52:05 +00:00
//TODO: have to specify environment otherwise wouldn't work with staging
2015-07-04 02:27:17 +00:00
if (this.blockchainConfig.config != undefined) {
2015-08-01 15:19:21 +00:00
//this.blockchainConfig = this.blockchainConfig.config('development');
this.blockchainConfig = this.blockchainConfig.config(env);
2015-07-04 02:27:17 +00:00
}
};
ContractsConfig.prototype.loadConfigFile = function(filename) {
try {
this.contractConfig = readYaml.sync(filename);
} catch (e) {
throw new Error("error reading " + filename);
}
return this;
2015-07-04 02:41:39 +00:00
};
ContractsConfig.prototype.loadConfig = function(config) {
this.contractConfig = config;
return this;
2015-07-04 02:41:39 +00:00
};
ContractsConfig.prototype.config = function(env) {
return this.contractConfig[env];
2015-07-04 02:41:39 +00:00
};
ContractsConfig.prototype.is_a_interface = function(target, className) {
if (this.contractStubs[className] && this.contractStubs[className].indexOf(target) >= 0)
return true;
2015-08-01 02:50:51 +00:00
return false;
};
ContractsConfig.prototype.compileContracts = function(env) {
var contractFile, source, j;
var contractsConfig = this.config(env);
2015-07-04 03:23:21 +00:00
this.compiler.init(env);
2015-07-04 20:52:05 +00:00
// determine dependencies
if (contractsConfig != null) {
for (className in contractsConfig) {
options = contractsConfig[className];
if (options.args == null) continue;
2015-10-25 18:37:57 +00:00
ref = options.args; //get arguments
for (j = 0; j < ref.length; j++) {
arg = ref[j];
2015-10-25 18:37:57 +00:00
if (arg[0] === "$") { //check if they are a contract dependency
if (this.contractDependencies[className] === void 0) {
this.contractDependencies[className] = [];
}
this.contractDependencies[className].push(arg.substr(1));
}
}
2015-08-01 02:50:51 +00:00
this.contractStubs[className] = options.stubs;
}
}
compiled_contracts = this.compiler.compile(this.contractFiles); //compile and push to contract DB
2015-10-25 18:37:57 +00:00
for (var className in compiled_contracts) {
2015-08-01 02:50:51 +00:00
if (this.is_a_interface(className, compiled_contracts)) {
2015-10-25 18:37:57 +00:00
continue;
}
2015-08-01 02:50:51 +00:00
2015-10-25 18:37:57 +00:00
this.all_contracts.push(className);
this.contractDB[className] = {
args: [],
types: ['file'],
gasPrice: this.blockchainConfig.gasPrice,
gasLimit: this.blockchainConfig.gasLimit,
compiled: compiled_contracts[className]
2015-07-04 20:52:05 +00:00
}
}
this.configureContractsParameters(contractsConfig);
this.sortContracts();
};
ContractsConfig.prototype.configureContractsParameters = function(contractsConfig) {
2015-07-04 20:52:05 +00:00
for(className in contractsConfig) {
var contractConfig = contractsConfig[className];
var contract;
contract = this.contractDB[className];
if (contract === undefined) {
contract = {
args: [],
types: ['file'],
gasPrice: this.blockchainConfig.gasPrice,
gasLimit: this.blockchainConfig.gasLimit,
compiled: contract
}
this.contractDB[className] = contract;
}
2015-07-04 20:52:05 +00:00
contract.gasPrice = contractConfig.gas_price || contract.gasPrice;
contract.gasLimit = contractConfig.gas_limit || contract.gasLimit;
contract.args = contractConfig.args || [];
2015-07-06 12:19:25 +00:00
contract.address = contractConfig.address;
contract.onDeploy = contractConfig.onDeploy || [];
2015-07-04 20:52:05 +00:00
if (contractConfig.instanceOf !== undefined) {
2015-07-04 20:52:05 +00:00
contract.types.push('instance');
contract.instanceOf = contractConfig.instanceOf;
contract.compiled = compiled_contracts[contractConfig.instanceOf];
2015-07-04 20:52:05 +00:00
}
if (contractConfig.address !== undefined) {
contract.types.push('static');
}
2015-07-04 20:52:05 +00:00
2015-09-01 00:53:17 +00:00
contract.deploy = contractConfig.deploy;
if (contractConfig.deploy === undefined) {
contract.deploy = true;
}
2015-07-04 20:52:05 +00:00
if (this.all_contracts.indexOf(className) < 0) {
this.all_contracts.push(className);
}
}
}
ContractsConfig.prototype.sortContracts = function() {
var converted_dependencies = [], i;
for(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();
this.all_contracts = this.all_contracts.sort(function(a,b) {
var order_a = orderedDependencies.indexOf(a);
var order_b = orderedDependencies.indexOf(b);
return order_a - order_b;
});;
2015-07-04 02:41:39 +00:00
};
2015-07-04 02:41:39 +00:00
module.exports = ContractsConfig;