#609 WIP Add responder menus to taks list

TheHive-Project · Jul 11, 2018 · aa77164 · aa77164
1 parent e1aba4b
commit aa77164
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 4 deletions.
diff --git a/ui/app/scripts/controllers/case/CaseTasksCtrl.js b/ui/app/scripts/controllers/case/CaseTasksCtrl.js
@@ -4,7 +4,7 @@
         .controller('CaseTaskDeleteCtrl', CaseTaskDeleteCtrl)
         .controller('CaseTasksCtrl', CaseTasksCtrl);
 
-    function CaseTasksCtrl($scope, $state, $stateParams, $q, $uibModal, CaseTabsSrv, PSearchSrv, CaseTaskSrv, UserInfoSrv, NotificationSrv) {
+    function CaseTasksCtrl($scope, $state, $stateParams, $q, $uibModal, CaseTabsSrv, PSearchSrv, CaseTaskSrv, UserInfoSrv, NotificationSrv, CortexSrv) {
 
         CaseTabsSrv.activateTab($state.current.data.tab);
 
@@ -15,6 +15,7 @@
         $scope.newTask = {
             status: 'Waiting'
         };
+        $scope.taskResponders = null;
 
         $scope.tasks = PSearchSrv($scope.caseId, 'case_task', {
             scope: $scope,
@@ -132,6 +133,29 @@
             return defer.promise;
         };
 
+        $scope.getTaskResponders = function(taskId, force) {
+            if(!force && $scope.taskResponders !== null) {
+               return;
+            }
+
+            CortexSrv.getResponders('case_task', taskId)
+              .then(function(responders) {
+                  $scope.taskResponders = responders;
+              })
+              .catch(function(err) {
+                  NotificationSrv.error('taskList', response.data, response.status);
+              })
+        };
+
+        $scope.runResponder = function(responderId, task) {
+            CortexSrv.runResponder(responderId, 'case_task', _.pick(task, 'id'))
+              .then(function(response) {
+                  console.log(response);
+              })
+              .catch(function(err) {
+                  console.log(err);
+              });
+        };
     }
 
     function CaseTaskDeleteCtrl($uibModalInstance, title) {

diff --git a/ui/app/scripts/controllers/case/CaseTasksItemCtrl.js b/ui/app/scripts/controllers/case/CaseTasksItemCtrl.js
@@ -155,7 +155,7 @@
                    return;
                 }
 
-                CortexSrv.getResponders('case_task', $scope.caseId)
+                CortexSrv.getResponders('case_task', $scope.task.id)
                   .then(function(responders) {
                       $scope.taskResponders = responders;
                   })

diff --git a/ui/app/views/directives/flow/case.html b/ui/app/views/directives/flow/case.html
@@ -27,7 +27,7 @@
             </div>
             <div ng-switch-when="tags">
                 {{k}}:
-                <span ng-repeat="tag in v">
+                <span ng-repeat="tag in v track by $index">
                     <span class="label label-primary">{{tag}}</span>
                 </span>
                 <span ng-if="v.length === 0">

diff --git a/ui/app/views/partials/case/case.tasks.html b/ui/app/views/partials/case/case.tasks.html
@@ -55,7 +55,7 @@
                     <th>Task</th>
                     <th style="width: 240px">Date</th>
                     <th style="width: 200px">Assignee</th>
-                    <th style="width: 200px" class="text-right">Actions</th>
+                    <th style="width: 250px" class="text-right">Actions</th>
                 </tr>
             </thead>
             <tbody>
@@ -102,10 +102,32 @@
                             <i class="fa fa-play"></i>
                             Start
                         </span>
+
+                        <span class="ml-xs" uib-dropdown ng-if="appConfig.connectors.cortex.enabled">
+                            <a href class="text-primary noline nowrap" ng-click="getTaskResponders(task.id)" uib-dropdown-toggle>
+                                <i class="text-primary fa fa-cog"></i>
+                                <!-- Responders
+                                <i class="text-primary fa fa-angle-down"></i> -->
+                            </a>
+                            <ul class="dropdown-menu align-right" uib-dropdown-menu>
+                              <li ng-if="taskResponders && taskResponders.length === 0">
+                                <a href ng-click="getTaskResponders(task.id, true)">
+                                  <strong><i class="fa fa-refresh mr-xxs"></i> No responders available</strong>
+                                </a>
+                              </li>
+                              <li ng-repeat="responder in taskResponders">
+                                <a href ng-click="runResponder(responder.id)">
+                                  <strong>{{responder.name}}</strong>
+                                  <br>
+                                  <small>{{responder.description}}</small></a>
+                              </li>
+                            </ul>
+                        </span>
                     </td>
                 </tr>
             </tbody>
         </table>
         <psearch control="tasks"></psearch>
+        <pre>{{taskResponders | json}}</pre>
     </div>
 </div>