'use strict'; /* Controllers */ function StatsCtrl($scope, $filter, socket, _, toastr) { // Main Stats init // --------------- $scope.nodesTotal = 0; $scope.nodesActive = 0; $scope.bestBlock = 0; $scope.lastBlock = 0; $scope.upTimeTotal = 0; $scope.nodes = []; $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() { console.log('Socket connection ended.') }) .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) { socketAction(data.action, data.data); }); socket.on('init', function(data) { socketAction("init", data.nodes); }); function socketAction(action, data) { console.log('Action: ', action); console.log('Data: ', data); switch(action) { case "init": $scope.nodes = data; if($scope.nodes.length > 0) toastr['success']("Got nodes list", "Got nodes!"); break; case "add": if(addNewNode(data)) toastr['success']("New node connected!", "New node!"); else toastr['info']("Node reconnected!", "Node is back!"); break; case "update": $scope.nodes[findIndex({id: data.id})].stats = data.stats; break; case "info": $scope.nodes[findIndex({id: data.id})].info = data.info; break; } updateStats(); } function findIndex(search) { return _.findIndex($scope.nodes, search); } function addNewNode(data) { var index = findIndex({id: data.id}); if(index < 0) { $scope.nodes.push(data); return true; } $scope.nodes[index] = data; return false; } 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(); } }