var express = require('express'); var app = express(); var path = require('path'); var favicon = require('serve-favicon'); var bodyParser = require('body-parser'); var Primus = require('primus'), api, client; var WS_SECRET = process.env.WS_SECRET || "eth-net-stats-has-a-secret"; var Collection = require('./models/collection'); var Nodes = new Collection(); var server = require('http').createServer(app); api = new Primus(server, { transformer: 'websockets', pathname: '/api', parser: 'JSON' }); api.use('emit', require('primus-emit')); api.use('spark-latency', require('primus-spark-latency')); var client = new Primus(server, { transformer: 'websockets', pathname: '/primus', parser: 'JSON' }); var clientLatency = 0; client.use('emit', require('primus-emit')); api.on('connection', function(spark) { console.log('Latency: ', spark.latency); console.log(spark.id); console.log(spark.address); console.log(spark.query); spark.on('hello', function(data) { console.log('Latency: ', spark.latency); console.log('got hello data from ', spark.id); console.log(data); if(typeof data.secret === 'undefined' || data.secret !== WS_SECRET) { spark.end(undefined, { reconnect: false }); return false; } if(typeof data.id !== 'undefined' && typeof data.info !== 'undefined') { data.ip = spark.address.ip; data.spark = spark.id; data.latency = spark.latency; var info = Nodes.add(data); spark.emit('ready'); client.write({action: 'add', data: info}); var blockPropagationChart = Nodes.blockPropagationChart(); client.write({action: 'blockPropagationChart', data: blockPropagationChart}); } }); spark.on('update', function(data) { console.log('Latency: ', spark.latency); console.log('got update from ' + spark.id); if(typeof data.id !== 'undefined' && typeof data.stats !== 'undefined') { data.stats.latency = spark.latency; var stats = Nodes.update(data.id, data.stats); client.write({action: 'update', data: stats}); var blockPropagationChart = Nodes.blockPropagationChart(); client.write({action: 'blockPropagationChart', data: blockPropagationChart}); } }); spark.on('node-ping', function(data){ spark.emit('node-pong'); }); spark.on('latency', function(data) { console.log('Latency: ', data.latency); console.log('got ping from ' + spark.id); if(typeof data.id !== 'undefined') { var stats = Nodes.updateLatency(data.id, data.latency); client.write({action: 'latency', data: stats}); } }); spark.on('end', function(data) { var stats = Nodes.inactive(spark.id); client.write({action: 'inactive', data: stats}); }); }); client.on('connection', function(spark) { console.log(spark.id); console.log(spark.address); console.log(spark.query); spark.on('ready', function(data){ console.log('got hello data from ' + spark.id); console.log(data); spark.emit('init', {nodes: Nodes.all()}); var blockPropagationChart = Nodes.blockPropagationChart(); spark.write({action: 'blockPropagationChart', data: blockPropagationChart}); }); spark.on('client-pong', function(data) { var latency = Math.ceil(((new Date()).getTime() - clientLatency)/2); spark.emit('client-latency', { latency: latency }); }); }); var latencyTimeout = setInterval(function(){ clientLatency = (new Date()).getTime(); client.write({action: 'client-ping'}); }, 5000); // view engine setup app.set('views', path.join(__dirname, 'views')); app.set('view engine', 'jade'); app.use(favicon(path.join(__dirname, '/public/images/favicon.png'))); app.use(bodyParser.json()); app.use(bodyParser.urlencoded({ extended: false })); app.use(express.static(path.join(__dirname, 'public'))); app.get('/', function(req, res) { res.render('index', { title: 'Ethereum Network Status' }); }); // catch 404 and forward to error handler app.use(function(req, res, next) { var err = new Error('Not Found'); err.status = 404; next(err); }); // error handlers if (app.get('env') === 'development') { app.use(function(err, req, res, next) { res.status(err.status || 500); res.render('error', { message: err.message, error: err }); }); } // production error handler app.use(function(err, req, res, next) { res.status(err.status || 500); res.render('error', { message: err.message, error: {} }); }); server.listen(process.env.PORT || 3000); module.exports = app;