Merge pull request #23 from maraoz/feature/commander-p2p
added commander support for p2p.js
This commit is contained in:
commit
5855efa566
96
p2p.js
96
p2p.js
|
@ -1,3 +1,4 @@
|
|||
#! /usr/bin/env node
|
||||
'use strict';
|
||||
|
||||
process.env.NODE_ENV = process.env.NODE_ENV || 'development';
|
||||
|
@ -9,21 +10,30 @@ var CoinConst = require('bitcore/const');
|
|||
var coinUtil = require('bitcore/util/util');
|
||||
var networks = require('bitcore/networks');
|
||||
var Parser = require('bitcore/util/BinaryParser').class();
|
||||
var async = require('async');
|
||||
var Sync = require('./lib/Sync').class();
|
||||
var Peer = require('bitcore/Peer').class();
|
||||
|
||||
var peerdb_fn = 'peerdb.json';
|
||||
|
||||
var peerdb = undefined;
|
||||
var hdrdb = undefined;
|
||||
var network = networks.testnet;
|
||||
var config = {
|
||||
network: network.name
|
||||
};
|
||||
var PeerManager = require('bitcore/PeerManager').createClass({
|
||||
config: config
|
||||
|
||||
var PROGRAM_VERSION = '0.1';
|
||||
var program = require('commander');
|
||||
|
||||
program
|
||||
.version(PROGRAM_VERSION)
|
||||
.option('-N --network [testnet]', 'Set bitcoin network [testnet]', 'testnet')
|
||||
.parse(process.argv);
|
||||
|
||||
var sync = new Sync({
|
||||
networkName: program.network
|
||||
});
|
||||
sync.init();
|
||||
|
||||
var PeerManager = require('bitcore/PeerManager').createClass({
|
||||
config: {
|
||||
network: program.network
|
||||
}
|
||||
});
|
||||
var Peer = require('bitcore/Peer').class();
|
||||
|
||||
function peerdb_load() {
|
||||
try {
|
||||
|
@ -40,74 +50,23 @@ function peerdb_load() {
|
|||
}
|
||||
}
|
||||
|
||||
function hdrdb_load() {
|
||||
hdrdb = new HeaderDB({
|
||||
network: network
|
||||
});
|
||||
}
|
||||
|
||||
function get_more_headers(info) {
|
||||
var conn = info.conn;
|
||||
var loc = hdrdb.locator();
|
||||
conn.sendGetHeaders(loc, coinUtil.NULL_HASH);
|
||||
}
|
||||
|
||||
function add_header(info, block) {
|
||||
var hashStr = coinUtil.formatHashFull(block.calcHash());
|
||||
|
||||
try {
|
||||
hdrdb.add(block);
|
||||
} catch(e) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
function handle_headers(info) {
|
||||
console.log('handle headers');
|
||||
var headers = info.message.headers;
|
||||
|
||||
headers.forEach(function(hdr) {
|
||||
add_header(info, hdr);
|
||||
});
|
||||
|
||||
// We persist the header DB after each batch
|
||||
//hdrdb.writeFile(hdrdb_fn);
|
||||
// Only one request per batch of headers we receive.
|
||||
get_more_headers(info);
|
||||
}
|
||||
|
||||
function handle_verack(info) {
|
||||
var inv = {
|
||||
type: CoinConst.MSG.BLOCK,
|
||||
hash: network.genesisBlock.hash,
|
||||
};
|
||||
var invs = [inv];
|
||||
|
||||
// Asks for the genesis block
|
||||
// console.log('p2psync: Asking for the genesis block');
|
||||
// info.conn.sendGetData(invs);
|
||||
}
|
||||
|
||||
function handle_inv(info) {
|
||||
// TODO: should limit the invs to objects we haven't seen yet
|
||||
var invs = info.message.invs;
|
||||
invs.forEach(function(inv) {
|
||||
console.log('Handle inv for a ' + CoinConst.MSG.to_str(inv.type));
|
||||
});
|
||||
// this is not needed right now, but it's left in case
|
||||
// we need to store more info in the future
|
||||
info.conn.sendGetData(invs);
|
||||
}
|
||||
|
||||
var sync = new Sync({
|
||||
networkName: networks.testnet
|
||||
});
|
||||
sync.init();
|
||||
|
||||
function handle_tx(info) {
|
||||
var tx = info.message.tx.getStandardizedObject();
|
||||
console.log('Handle tx: ' + tx.hash);
|
||||
sync.storeTxs([tx.hash], function(err) {
|
||||
if (err) {
|
||||
console.log('error in handle TX: ' + err);
|
||||
console.log('Error in handle TX: ' + err);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
@ -123,8 +82,9 @@ function handle_block(info) {
|
|||
},
|
||||
function(err) {
|
||||
if (err) {
|
||||
console.log('error in handle Block: ' + err);
|
||||
console.log('Error in handle Block: ' + err);
|
||||
} else {
|
||||
// if no errors importing block, import the transactions
|
||||
var hashes = block.txs.map(function(tx) {
|
||||
return coinUtil.formatHashFull(tx.hash);
|
||||
});
|
||||
|
@ -149,10 +109,8 @@ function p2psync() {
|
|||
});
|
||||
|
||||
peerman.on('connection', function(conn) {
|
||||
conn.on('verack', handle_verack);
|
||||
conn.on('block', handle_block);
|
||||
conn.on('headers', handle_headers);
|
||||
conn.on('inv', handle_inv);
|
||||
conn.on('block', handle_block);
|
||||
conn.on('tx', handle_tx);
|
||||
});
|
||||
peerman.on('connect', handle_connected);
|
||||
|
@ -162,8 +120,6 @@ function p2psync() {
|
|||
|
||||
function main() {
|
||||
peerdb_load();
|
||||
hdrdb_load();
|
||||
|
||||
p2psync();
|
||||
}
|
||||
|
||||
|
|
|
@ -11,7 +11,14 @@ var program = require('commander');
|
|||
var Sync = require('../lib/Sync').class();
|
||||
var async = require('async');
|
||||
|
||||
program.version(SYNC_VERSION).option('-N --network [livenet]', 'Set bitcoin network [testnet]', 'testnet').option('-R --reindex', 'Force reindexing', '0').option('-D --destroy', 'Remove current DB', '0').option('--skip_blocks', 'Sync blocks').option('--skip_txs', 'Sync transactions').parse(process.argv);
|
||||
program
|
||||
.version(SYNC_VERSION)
|
||||
.option('-N --network [livenet]', 'Set bitcoin network [testnet]', 'testnet')
|
||||
.option('-R --reindex', 'Force reindexing', '0')
|
||||
.option('-D --destroy', 'Remove current DB', '0')
|
||||
.option('--skip_blocks', 'Sync blocks')
|
||||
.option('--skip_txs', 'Sync transactions')
|
||||
.parse(process.argv);
|
||||
|
||||
var sync = new Sync({
|
||||
networkName: program.network
|
||||
|
|
Loading…
Reference in New Issue