ethstats-server/public/js/controllers.js

'use strict';

/* Controllers */

function StatsCtrl($scope, $filter, socket, _) {

	// Main Stats init
	// ---------------

	$scope.nodesTotal = 0;
	$scope.nodesActive = 0;
	$scope.bestBlock = 0;
	$scope.lastBlock = 0;
	$scope.upTimeTotal = 0;

	$scope.map = [];

	// Socket listeners
	// ----------------

	socket = new Primus();

	socket.on('open', function open() {
		socket.emit('ready');
		console.log('The connection has been opened.');
	})
	.on('end', function end() {
		self._socket = false;
	})
	.on('error', function error(err) {
		console.log(err);
	})
	.on('reconnecting', function reconnecting(opts) {
		console.log('We are scheduling a reconnect operation', opts);
	})
	.on('data', function incoming(data) {
		console.log('Received some data', data);
		socketAction(data.action, data.data);
	});

	socket.on('init', function(data)
	{
		console.log(data);

		$scope.nodes = data.nodes;

		updateStats();
	});

	function socketAction(action, data)
	{
		switch(action) {
			case "add":
				console.log(data);
				$scope.nodes.push(data);
				break;

			case "update":
				console.log(data);
				$scope.nodes[findIndex({id: data.id})].stats = data.stats;
				break;

			case "info":
				console.log(data);
				$scope.nodes[findIndex({id: data.id})].info = data.info;
		}

		updateStats();
	}

	function findIndex(search)
	{
		return _.findIndex($scope.nodes, search);
	}

	function updateStats()
	{
		if($scope.nodes.length)
		{
			$scope.nodesTotal = $scope.nodes.length;

			$scope.nodesActive = _.filter($scope.nodes, function(node) {
				return node.stats.active == true;
			}).length;

			$scope.bestBlock = _.max($scope.nodes, function(node) {
				return parseInt(node.stats.block.number);
			}).stats.block.number;

			$scope.lastBlock = _.max($scope.nodes, function(node) {
				return parseInt(node.stats.block.timestamp);
			}).stats.block.timestamp;

			$scope.upTimeTotal = _.reduce($scope.nodes, function(total, node) {
				return total + node.stats.uptime;
			}, 0) / $scope.nodes.length;

			$scope.map = _.map($scope.nodes, function(node) {
				if(node.geo != null)
					return {
						radius: 3,
						latitude: node.geo.ll[0],
						longitude: node.geo.ll[1],
						fillKey: $filter('bubbleClass')(node, $scope.bestBlock)
					};
				else
					return {
						radius: 0,
						latitude: 0,
						longitude: 0
					};
			});
		}

		$scope.$apply();
	}
}