var _ = require('lodash'); var askedForHistory = false; var askedForHistoryTime = 0; 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 env = 'production'; if( process.env.NODE_ENV !== 'production' ) { var express = require('express'); var app = express(); var path = require('path'); var favicon = require('serve-favicon'); var bodyParser = require('body-parser'); // view engine setup app.set('views', path.join(__dirname, 'views')); app.set('view engine', 'jade'); app.use(bodyParser.json()); app.use(bodyParser.urlencoded({ extended: false })); app.use(express.static(path.join(__dirname, 'dist'))); app.get('/', function(req, res) { res.render('index'); }); // 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 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: {} }); }); var server = require('http').createServer(app); } else { var server = require('http').createServer(); } 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( _.isUndefined(data.secret) || data.secret !== WS_SECRET ) { spark.end(undefined, { reconnect: false }); return false; } if( !_.isUndefined(data.id) && !_.isUndefined(data.info) ) { 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 }); client.write({ action: 'charts', data: Nodes.getCharts() }); } }); spark.on('update', function(data) { if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) ) { data.stats.latency = spark.latency; var stats = Nodes.update(data.id, data.stats); if(stats !== false) { client.write({ action: 'update', data: stats }); client.write({ action: 'charts', data: Nodes.getCharts() }); } } }); spark.on('history', function(data) { console.log("got history from " + data.id); client.write({ action: 'charts', data: Nodes.addHistory(data.id, data.history) }); askedForHistory = false; }); spark.on('node-ping', function(data) { spark.emit('node-pong'); }); spark.on('latency', function(data) { if( !_.isUndefined(data.id) ) { var stats = Nodes.updateLatency(data.id, data.latency); client.write({ action: 'latency', data: stats }); if( Nodes.requiresUpdate(data.id) && (!askedForHistory || _.now() - askedForHistoryTime > 200000) ) { var range = Nodes.getHistory().getHistoryRequestRange(); console.log("asked " + data.id + " for history: " + range.min + " - " + range.max); spark.emit('history', range); askedForHistory = true; askedForHistoryTime = _.now(); } } }); spark.on('end', function(data) { var stats = Nodes.inactive(spark.id); client.write({ action: 'inactive', data: stats }); }); }); client.on('connection', function(clientSpark) { clientSpark.on('ready', function(data) { clientSpark.emit('init', { nodes: Nodes.all() }); clientSpark.write({ action: 'charts', data: Nodes.getCharts() }); }); clientSpark.on('client-pong', function(data) { var latency = Math.ceil( (_.now() - clientLatency) / 2 ); clientSpark.emit('client-latency', { latency: latency }); }); }); var latencyTimeout = setInterval( function () { clientLatency = _.now(); client.write({ action: 'client-ping' }); }, 5000); // Cleanup old inactive nodes var nodeCleanupTimeout = setInterval( function () { client.write({ action: 'init', data: Nodes.all() }); client.write({ action: 'charts', data: Nodes.getCharts() }); }, 1000*60*60); server.listen(process.env.PORT || 3000); module.exports = server;