55 lines
1.4 KiB
JavaScript
55 lines
1.4 KiB
JavaScript
var fs = require('fs');
|
|
var web3 = require('web3');
|
|
var sha3_256 = require('js-sha3').sha3_256;
|
|
|
|
ChainManager = function() {
|
|
this.chainManagerConfig = {};
|
|
this.currentChain = {};
|
|
this.file = "";
|
|
}
|
|
|
|
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];
|
|
}
|
|
|
|
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;
|