#165 Add a severity column and filter in case list page

TheHive-Project · Apr 7, 2017 · 9631f52 · 9631f52
1 parent 4264bd2
commit 9631f52
Show file tree

Hide file tree

Showing 5 changed files with 70 additions and 5 deletions.
diff --git a/ui/app/scripts/controllers/case/CaseListCtrl.js b/ui/app/scripts/controllers/case/CaseListCtrl.js
@@ -3,7 +3,7 @@
     angular.module('theHiveControllers')
         .controller('CaseListCtrl', CaseListCtrl);
 
-    function CaseListCtrl($scope, $q, $state, $window, CasesUISrv, StreamStatSrv, PSearchSrv, EntitySrv, UserInfoSrv, TagSrv, UserSrv, AuthenticationSrv, CaseResolutionStatus) {
+    function CaseListCtrl($scope, $q, $state, $window, CasesUISrv, StreamStatSrv, PSearchSrv, EntitySrv, UserInfoSrv, TagSrv, UserSrv, AuthenticationSrv, CaseResolutionStatus, Severity) {
         var self = this;
 
         this.showFlow = true;
@@ -133,6 +133,25 @@
             ]);
         };
 
+        this.getSeverities = function(query) {
+            var defer = $q.defer();
+
+            $q.resolve(_.map(Severity.keys, function(value, key) {
+                return {text: key};
+            })).then(function(response) {
+                var severities = [];
+
+                severities = _.filter(response, function(sev) {
+                    var regex = new RegExp(query, 'gi');
+                    return regex.test(sev.text);
+                });
+
+                defer.resolve(severities);
+            });
+
+            return defer.promise;
+        };
+
         this.getTags = function(query) {
             return TagSrv.fromCases(query);
         };
@@ -182,6 +201,10 @@
                 });
         };
 
+        this.filterBySeverity = function(numericSev) {
+            self.addFilterValue('severity', Severity.values[numericSev]);
+        };
+
         this.sortBy = function(sort) {
             this.list.sort = sort;
             this.list.update();

diff --git a/ui/app/scripts/services/CasesUISrv.js b/ui/app/scripts/services/CasesUISrv.js
@@ -1,7 +1,7 @@
 (function() {
     'use strict';
     angular.module('theHiveServices')
-        .factory('CasesUISrv', function($q, localStorageService) {
+        .factory('CasesUISrv', function($q, localStorageService, Severity) {
             var defaultFilter = {
                 status: {
                     field: 'status',
@@ -29,6 +29,15 @@
                         type: 'list',
                         defaultValue: []
                     },
+                    severity: {
+                        field: 'severity',
+                        type: 'list',
+                        defaultValue: [],
+                        convert: function(value) {
+                            // Convert the text value to its numeric representation
+                            return Severity.keys[value];
+                        }
+                    },
                     resolutionStatus: {
                         field: 'resolutionStatus',
                         type: 'list',
@@ -147,14 +156,15 @@
 
                 addFilter: function(field, value) {
                     var query,
-                        filterDef = factory.filterDefs[field];
+                        filterDef = factory.filterDefs[field],
+                        convertFn = filterDef.convert || angular.identity;
 
                     // Prepare the filter value
                     if (field === 'keyword') {
                         query = value;
                     } else if (angular.isArray(value) && value.length > 0) {
                         query = _.map(value, function(val) {
-                            return field + ':"' + val.text + '"';
+                            return field + ':"' + convertFn(val.text) + '"';
                         }).join(' OR ');
                         query = '(' + query + ')';
                     } else if (filterDef.type === 'date') {
@@ -164,7 +174,7 @@
                         query = field + ':[ ' + fromDate + ' TO ' + toDate + ' ]';
 
                     } else {
-                        query = field + ':' + value;
+                        query = field + ':' + convertFn(value);
                     }
 
                     factory.filters[field] = {

diff --git a/ui/app/scripts/services/Constants.js b/ui/app/scripts/services/Constants.js
@@ -12,5 +12,19 @@
             NoImpact: 'No Impact',
             WithImpact: 'With impact',
             NotApplicable: 'Not Applicable'
+        })
+        .value('Severity', {
+            keys: {
+                High: 3,
+                Medium: 2,
+                Low: 1
+            },
+            values: ['Unknown', 'Low', 'Medium', 'High']
+        })
+        .value('Tlp', {
+            RED: 3,
+            AMBER: 2,
+            GREEN: 1,
+            WHITE: 1
         });
 })();
diff --git a/ui/app/views/partials/case/case.list.html b/ui/app/views/partials/case/case.list.html
@@ -48,6 +48,7 @@ <h3 class="box-title">List of cases ({{$vm.list.total || 0}} of {{$vm.caseStats.
                                 <tr>
                                     <th style="width: 10px;" class="p-0"></th>
                                     <th>Title</th>
+                                    <th style="width: 100px;text-align:center;">Severity</th>
                                     <th style="width: 100px;">Tasks</th>
                                     <th style="width: 100px;">Observables</th>
                                     <th style="width: 60px;">Assignee</th>
@@ -82,6 +83,11 @@ <h3 class="box-title">List of cases ({{$vm.list.total || 0}} of {{$vm.caseStats.
                                             </small>
                         				</div>
                                     </td>
+                                    <td align="center">
+                                        <div class="clickable" ng-click="$vm.filterBySeverity(currentCase.severity)">
+                                            <severity active="false" value="currentCase.severity"></severity>
+                                        </div>
+                                    </td>
                                     <td>
                                         <a ui-sref="app.case.tasks({caseId: currentCase.id})">
                                             <ng-pluralize count="currentCase.stats.tasks.total" when="{'0': 'No Tasks', '1': '1 Task', 'other': '{} Tasks'}"></ng-pluralize>

diff --git a/ui/app/views/partials/case/list/filters.html b/ui/app/views/partials/case/list/filters.html
@@ -63,6 +63,18 @@ <h4>Filters</h4>
                     </div>
                 </div>
                 <div class="col-md-6">
+                    <div class="form-group">
+                        <label class="col-sm-4 control-label">Severity</label>
+                        <div class="col-sm-8">
+                            <tags-input class="form-control form-control-wrapper"
+                                min-length="2"
+                                ng-model="$vm.uiSrv.activeFilters.severity.value"
+                                placeholder="ex: High, Medium, Low"
+                                replace-spaces-with-dashes="false">
+                                    <auto-complete load-on-focus="true" min-length="1" source="$vm.getSeverities($query)"></auto-complete>
+                                </tags-input>
+                        </div>
+                    </div>
                     <div class="form-group">
                         <label class="col-sm-4 control-label">Title</label>
                         <div class="col-sm-8">