87 lines
2.1 KiB
JavaScript
87 lines
2.1 KiB
JavaScript
'use strict';
|
|
|
|
var util = require('util');
|
|
var EventEmitter = require('eventemitter2').EventEmitter2;
|
|
|
|
var bitcore = require('bitcore');
|
|
var Networks = bitcore.Networks;
|
|
var $ = bitcore.util.preconditions;
|
|
var _ = bitcore.deps._;
|
|
var p2p = require('bitcore-p2p');
|
|
var Peer = p2p.Peer;
|
|
var messages = new p2p.Messages();
|
|
|
|
function NetworkMonitor(eventBus, peer) {
|
|
$.checkArgument(eventBus);
|
|
$.checkArgument(peer);
|
|
this.bus = eventBus;
|
|
this.peer = peer;
|
|
this.setupPeer(peer);
|
|
}
|
|
util.inherits(NetworkMonitor, EventEmitter);
|
|
|
|
NetworkMonitor.create = function(eventBus, opts) {
|
|
opts = opts || {};
|
|
var host = opts.host || 'localhost';
|
|
var port = opts.port || Networks.defaultNetwork.port;
|
|
var peer = new Peer({
|
|
host: host,
|
|
port: port,
|
|
network: Networks.defaultNetwork
|
|
});
|
|
return new NetworkMonitor(eventBus, peer);
|
|
};
|
|
|
|
NetworkMonitor.prototype.setupPeer = function(peer) {
|
|
var self = this;
|
|
|
|
peer.on('ready', function() {
|
|
self.emit('ready');
|
|
});
|
|
peer.on('inv', function(m) {
|
|
self.emit('inv', m.inventory);
|
|
// TODO only ask for data if tx or block is unknown
|
|
peer.sendMessage(messages.GetData(m.inventory));
|
|
});
|
|
peer.on('tx', function(m) {
|
|
self.bus.process(m.transaction)
|
|
.catch(function(err) {
|
|
self.abort(err);
|
|
});
|
|
});
|
|
peer.on('block', function(m) {
|
|
self.bus.process(m.block)
|
|
.catch(function(err) {
|
|
self.abort(err);
|
|
});
|
|
});
|
|
peer.on('error', function(err) {
|
|
self.emit('error', err);
|
|
self.abort(err);
|
|
});
|
|
peer.on('disconnect', function() {
|
|
self.emit('disconnect');
|
|
});
|
|
|
|
};
|
|
|
|
NetworkMonitor.prototype.requestBlocks = function(start) {
|
|
$.checkArgument(_.isArray(start) ||
|
|
_.isString(start), 'start must be a block hash string or array');
|
|
this.peer.sendMessage(messages.GetBlocks(_.isArray(start) ? start : [start]));
|
|
};
|
|
|
|
NetworkMonitor.prototype.start = function() {
|
|
this.peer.connect();
|
|
};
|
|
NetworkMonitor.prototype.stop = function(reason) {
|
|
this.peer.disconnect();
|
|
console.log('Stopping network, reason:', reason);
|
|
};
|
|
NetworkMonitor.prototype.abort = function(reason) {
|
|
this.peer.disconnect();
|
|
throw reason;
|
|
};
|
|
|
|
module.exports = NetworkMonitor;
|