diff --git a/frontend/app/scripts/controllers/RootCtrl.js b/frontend/app/scripts/controllers/RootCtrl.js index bb168af521..93057a7b06 100644 --- a/frontend/app/scripts/controllers/RootCtrl.js +++ b/frontend/app/scripts/controllers/RootCtrl.js @@ -68,8 +68,8 @@ angular.module('theHiveControllers').controller('RootCtrl', name: 'my-tasks.stats' } }, - onUpdate: function(updates) { - $scope.myCurrentTasksCount = updates; + onUpdate: function(data) { + $scope.myCurrentTasksCount = data; } }); @@ -85,13 +85,28 @@ angular.module('theHiveControllers').controller('RootCtrl', name: 'waiting-tasks.stats' } }, - onUpdate: function(updates) { - $scope.waitingTasksCount = updates; + onUpdate: function(data) { + $scope.waitingTasksCount = data; } }); - // Get Alert counts - $scope.alertEvents = AlertingSrv.stats($scope); + StreamQuerySrv('v1', [ + {_name: 'listAlert'}, + {_name: 'filter', _field: 'read', _value: false}, + {_name: 'count'} + ], { + scope: $scope, + rootId: 'any', + objectType: 'alert', + query: { + params: { + name: 'unread-alert-count' + } + }, + onUpdate: function(data) { + $scope.unreadAlertCount = data; + } + }); $scope.$on('templates:refresh', function(){ CaseTemplateSrv.list().then(function(templates) { @@ -104,9 +119,13 @@ angular.module('theHiveControllers').controller('RootCtrl', $scope.initCustomFieldsCache(); }); - $scope.$on('alert:event-imported', function() { - $scope.alertEvents = AlertingSrv.stats($scope); - }); + + // Get Alert counts + //$scope.alertEvents = AlertingSrv.stats($scope); + + // $scope.$on('alert:event-imported', function() { + // $scope.alertEvents = AlertingSrv.stats($scope); + // }); // FIXME // $scope.$on('misp:status-updated', function(event, enabled) { diff --git a/frontend/app/scripts/controllers/alert/AlertListCtrl.js b/frontend/app/scripts/controllers/alert/AlertListCtrl.js index b9b77a2b95..b21bdb7a05 100755 --- a/frontend/app/scripts/controllers/alert/AlertListCtrl.js +++ b/frontend/app/scripts/controllers/alert/AlertListCtrl.js @@ -2,7 +2,7 @@ (function() { 'use strict'; angular.module('theHiveControllers') - .controller('AlertListCtrl', function($rootScope, $scope, $q, $state, $uibModal, TagSrv, CaseTemplateSrv, ModalUtilsSrv, AlertingSrv, NotificationSrv, FilteringSrv, CortexSrv, Severity, VersionSrv) { + .controller('AlertListCtrl', function($rootScope, $scope, $q, $state, $uibModal, TagSrv, StreamQuerySrv, CaseTemplateSrv, ModalUtilsSrv, AlertingSrv, NotificationSrv, FilteringSrv, CortexSrv, Severity, VersionSrv) { var self = this; self.urls = VersionSrv.mispUrls(); @@ -44,6 +44,23 @@ self.filtering.setPageSize(newValue); }); }); + + StreamQuerySrv('v1', [ + {_name: 'listAlert'}, + {_name: 'count'} + ], { + scope: $scope, + rootId: 'any', + objectType: 'alert', + query: { + params: { + name: 'alert-count' + } + }, + onUpdate: function(data) { + self.alertListCount = data; + } + }); }; self.load = function() { diff --git a/frontend/app/views/components/header.component.html b/frontend/app/views/components/header.component.html index b17e500957..497dcd2a85 100644 --- a/frontend/app/views/components/header.component.html +++ b/frontend/app/views/components/header.component.html @@ -27,18 +27,18 @@