236 lines
5.5 KiB
JavaScript
236 lines
5.5 KiB
JavaScript
'use strict';
|
|
|
|
var path = require('path');
|
|
var socketio = require('socket.io');
|
|
var BitcoreNode = require('../node');
|
|
var index = require('../');
|
|
var bitcore = require('bitcore');
|
|
var _ = bitcore.deps._;
|
|
var log = index.log;
|
|
log.debug = function() {};
|
|
|
|
var count = 0;
|
|
var interval = false;
|
|
|
|
function start(options) {
|
|
/* jshint maxstatements: 100 */
|
|
|
|
var services = [];
|
|
|
|
var configPath = options.path;
|
|
var config = options.config;
|
|
|
|
if (config.services) {
|
|
for (var i = 0; i < config.services.length; i++) {
|
|
var serviceName = config.services[i];
|
|
var service;
|
|
try {
|
|
// first try in the built-in bitcore-node services directory
|
|
service = require(path.resolve(__dirname, '../services/' + serviceName));
|
|
} catch(e) {
|
|
|
|
// check if the package.json specifies a specific file to use
|
|
var servicePackage = require(serviceName + '/package.json');
|
|
var serviceModule = serviceName;
|
|
if (servicePackage.bitcoreNode) {
|
|
serviceModule = serviceName + '/' + servicePackage.bitcoreNode;
|
|
}
|
|
service = require(serviceModule);
|
|
}
|
|
|
|
// check that the service supports expected methods
|
|
if (!service.prototype ||
|
|
!service.dependencies ||
|
|
!service.prototype.start ||
|
|
!service.prototype.stop) {
|
|
throw new Error(
|
|
'Could not load service "' + serviceName + '" as it does not support necessary methods.'
|
|
);
|
|
}
|
|
services.push({
|
|
name: serviceName,
|
|
module: service,
|
|
dependencies: service.dependencies
|
|
});
|
|
|
|
}
|
|
}
|
|
|
|
var fullConfig = _.clone(config);
|
|
|
|
// expand to the full path
|
|
fullConfig.datadir = path.resolve(configPath, config.datadir);
|
|
|
|
// load the services
|
|
fullConfig.services = services;
|
|
|
|
var node = new BitcoreNode(fullConfig);
|
|
|
|
function logSyncStatus() {
|
|
log.info(
|
|
'Sync Status: Tip:', node.services.db.tip.hash,
|
|
'Height:', node.services.db.tip.__height,
|
|
'Rate:', count/10, 'blocks per second'
|
|
);
|
|
}
|
|
|
|
node.on('synced', function() {
|
|
// Stop logging of sync status
|
|
clearInterval(interval);
|
|
interval = false;
|
|
logSyncStatus();
|
|
});
|
|
|
|
node.on('ready', function() {
|
|
|
|
var io = socketio(fullConfig.port);
|
|
|
|
io.on('connection', function(socket) {
|
|
|
|
var bus = node.openBus();
|
|
|
|
var methods = node.getAllAPIMethods();
|
|
var methodsMap = {};
|
|
|
|
methods.forEach(function(data) {
|
|
var name = data[0];
|
|
var instance = data[1];
|
|
var method = data[2];
|
|
var args = data[3];
|
|
methodsMap[name] = {
|
|
fn: function() {
|
|
return method.apply(instance, arguments);
|
|
},
|
|
args: args
|
|
};
|
|
});
|
|
|
|
socket.on('message', function(message, socketCallback) {
|
|
if (methodsMap[message.method]) {
|
|
var params = message.params;
|
|
|
|
if(!params || !params.length) {
|
|
params = [];
|
|
}
|
|
|
|
if(params.length !== methodsMap[message.method].args) {
|
|
return socketCallback({
|
|
error: {
|
|
message: 'Expected ' + methodsMap[message.method].args + ' parameters'
|
|
}
|
|
});
|
|
}
|
|
|
|
var callback = function(err, result) {
|
|
var response = {};
|
|
if(err) {
|
|
response.error = {
|
|
message: err.toString()
|
|
};
|
|
}
|
|
|
|
if(result) {
|
|
response.result = result;
|
|
}
|
|
|
|
socketCallback(response);
|
|
};
|
|
|
|
params = params.concat(callback);
|
|
methodsMap[message.method].fn.apply(this, params);
|
|
} else {
|
|
socketCallback({
|
|
error: {
|
|
message: 'Method Not Found'
|
|
}
|
|
});
|
|
}
|
|
});
|
|
|
|
socket.on('subscribe', function(name, params) {
|
|
bus.subscribe(name, params);
|
|
});
|
|
|
|
socket.on('unsubscribe', function(name, params) {
|
|
bus.unsubscribe(name, params);
|
|
});
|
|
|
|
var events = node.getAllPublishEvents();
|
|
|
|
events.forEach(function(event) {
|
|
bus.on(event.name, function() {
|
|
if(socket.connected) {
|
|
var results = [];
|
|
|
|
for(var i = 0; i < arguments.length; i++) {
|
|
results.push(arguments[i]);
|
|
}
|
|
|
|
var params = [event.name].concat(results);
|
|
socket.emit.apply(socket, params);
|
|
}
|
|
});
|
|
});
|
|
|
|
socket.on('disconnect', function() {
|
|
bus.close();
|
|
});
|
|
|
|
});
|
|
|
|
});
|
|
|
|
node.on('error', function(err) {
|
|
log.error(err);
|
|
});
|
|
|
|
node.on('ready', function() {
|
|
node.services.db.on('addblock', function(block) {
|
|
count++;
|
|
// Initialize logging if not already instantiated
|
|
if (!interval) {
|
|
interval = setInterval(function() {
|
|
logSyncStatus();
|
|
count = 0;
|
|
}, 10000);
|
|
}
|
|
});
|
|
});
|
|
|
|
node.on('stopping', function() {
|
|
clearInterval(interval);
|
|
});
|
|
|
|
function exitHandler(options, err) {
|
|
if (err) {
|
|
log.error('uncaught exception:', err);
|
|
if(err.stack) {
|
|
console.log(err.stack);
|
|
}
|
|
process.exit(-1);
|
|
}
|
|
if (options.sigint) {
|
|
node.stop(function(err) {
|
|
if(err) {
|
|
log.error('Failed to stop services: ' + err);
|
|
return process.exit(1);
|
|
}
|
|
|
|
log.info('Halted');
|
|
process.exit(0);
|
|
});
|
|
}
|
|
}
|
|
|
|
//catches uncaught exceptions
|
|
process.on('uncaughtException', exitHandler.bind(null, {exit:true}));
|
|
|
|
//catches ctrl+c event
|
|
process.on('SIGINT', exitHandler.bind(null, {sigint:true}));
|
|
|
|
return node;
|
|
|
|
}
|
|
|
|
module.exports = start;
|