From f03fb3f488d3bd52c94d0a274b2a574f1237c741 Mon Sep 17 00:00:00 2001
From: Nabil Adouani <nabil.adouani@gmail.com>
Date: Fri, 15 May 2020 16:08:16 +0200
Subject: [PATCH] #1316 Add UI to switch organisation for users belongs to more
 than one org

---
 frontend/app/index.html                       |  1 +
 .../components/organisation/OrgSwitchCtrl.js  | 18 ++++++++++
 frontend/app/scripts/controllers/RootCtrl.js  | 34 +++++++++++++++++++
 .../scripts/services/api/AuthenticationSrv.js | 12 +++++--
 .../views/components/header.component.html    |  7 +++-
 .../views/components/org/orgSwitch.modal.html | 30 ++++++++++++++++
 6 files changed, 99 insertions(+), 3 deletions(-)
 create mode 100644 frontend/app/scripts/components/organisation/OrgSwitchCtrl.js
 create mode 100644 frontend/app/views/components/org/orgSwitch.modal.html

diff --git a/frontend/app/index.html b/frontend/app/index.html
index f6b081c997..ec9828b2ee 100644
--- a/frontend/app/index.html
+++ b/frontend/app/index.html
@@ -133,6 +133,7 @@
     <script src="scripts/components/header.component.js"></script>
     <script src="scripts/components/main-sidebar.component.js"></script>
     <script src="scripts/components/organisation/OrgCaseTemplateListCmp.js"></script>
+    <script src="scripts/components/organisation/OrgSwitchCtrl.js"></script>
     <script src="scripts/components/organisation/OrgUserListCmp.js"></script>
     <script src="scripts/components/search/filters-preview.component.js"></script>
     <script src="scripts/components/sharing/SharingListCmp.js"></script>
diff --git a/frontend/app/scripts/components/organisation/OrgSwitchCtrl.js b/frontend/app/scripts/components/organisation/OrgSwitchCtrl.js
new file mode 100644
index 0000000000..39873534ca
--- /dev/null
+++ b/frontend/app/scripts/components/organisation/OrgSwitchCtrl.js
@@ -0,0 +1,18 @@
+(function() {
+    'use strict';
+
+    angular.module('theHiveControllers').controller('OrgSwitchCtrl',
+        function($uibModalInstance, currentUser) {
+            //var self = this;
+
+            this.currentUser = currentUser;
+
+            this.selectOrg = function(selected) {
+                $uibModalInstance.close(selected);
+            };
+
+            this.cancel = function() {
+                $uibModalInstance.dismiss();
+            };        
+        });
+})();
diff --git a/frontend/app/scripts/controllers/RootCtrl.js b/frontend/app/scripts/controllers/RootCtrl.js
index bf76b3a7a2..15c3f71ec2 100644
--- a/frontend/app/scripts/controllers/RootCtrl.js
+++ b/frontend/app/scripts/controllers/RootCtrl.js
@@ -133,6 +133,40 @@ angular.module('theHiveControllers').controller('RootCtrl',
             });
         };
 
+        $scope.switchOrg = function() {
+            var modal = $uibModal.open({
+                templateUrl: 'views/components/org/orgSwitch.modal.html',
+                controller: 'OrgSwitchCtrl',
+                controllerAs: '$dialog',
+                resolve: {
+                    currentUser: $scope.currentUser
+                }
+            });
+
+            modal.result
+                .then(function(organisation) {
+                    console.log(organisation);
+
+                    $rootScope.isLoading = true;
+
+                    AuthenticationSrv.current(organisation)
+                        .then(function(/*userData*/) {
+                            $rootScope.isLoading = false;
+                            $state.go('app.index');
+                        })
+                        .catch( function(err) {
+                            NotificationSrv.error('App', err.data, err.status);
+                            $rootScope.isLoading = false;
+                        });
+                })
+                .catch(function(err) {
+                    if(err && !_.isString(err)) {
+                        NotificationSrv.error('Switch organisation', err.data, err.status);
+                    }
+                });
+
+        };
+
         $scope.createNewCase = function(template) {
             var modal = $uibModal.open({
                 templateUrl: 'views/partials/case/case.creation.html',
diff --git a/frontend/app/scripts/services/api/AuthenticationSrv.js b/frontend/app/scripts/services/api/AuthenticationSrv.js
index 88c8269e62..9fd29e3e71 100644
--- a/frontend/app/scripts/services/api/AuthenticationSrv.js
+++ b/frontend/app/scripts/services/api/AuthenticationSrv.js
@@ -33,10 +33,18 @@
                             }
                         });
                 },
-                current: function() {
+                current: function(organisation) {
                     var result = {};
+
+                    var options = {};
+                    if(organisation) {
+                        options.headers = {
+                            'X-Organisation': organisation
+                        };
+                    }
+
                     return $http
-                        .get('./api/v1/user/current')
+                        .get('./api/v1/user/current', options)
                         .then(function(response) {
                             self.currentUser = response.data;
                             UtilsSrv.shallowClearAndCopy(response.data, result);
diff --git a/frontend/app/views/components/header.component.html b/frontend/app/views/components/header.component.html
index 24320f477d..a47c51cfee 100644
--- a/frontend/app/views/components/header.component.html
+++ b/frontend/app/views/components/header.component.html
@@ -133,7 +133,7 @@
                                 <i class="fa fa-file-text"></i>
                                 <span class="hpad5">Analyzer templates</span>
                             </a>
-                        </li>                        
+                        </li>
                         <!-- <li>
                             <a ui-sref="app.administration.ui-settings">
                                 <i class="fa fa-cogs"></i>
@@ -175,6 +175,11 @@
                         </li>
                     </ul>
                 </li>
+                <li ng-if="currentUser.organisations.length > 0">
+                    <a href ng-click="switchOrg()">
+                      <i class="fa fa-exchange"></i>
+                    </a>
+                </li>
             </ul>
         </div>
 
diff --git a/frontend/app/views/components/org/orgSwitch.modal.html b/frontend/app/views/components/org/orgSwitch.modal.html
new file mode 100644
index 0000000000..1c710a4f1e
--- /dev/null
+++ b/frontend/app/views/components/org/orgSwitch.modal.html
@@ -0,0 +1,30 @@
+<form ng-submit="$modal.confirm()">
+    <div class="modal-header bg-primary">
+        <h3 class="modal-title">Switch organisation</h3>
+    </div>
+    <div class="modal-body">
+        <p class="mb-xs">
+            Please select the organisation you want to switch to:
+        </p>
+
+        <div class="form-group has-feedback has-feedback-left" ng-if="$dialog.currentUser.organisations.length > 5">
+            <input type="text" class="input input-lg form-control" placeholder="Filter organisations" ng-model="$modal.filter" autocomplete="off">
+            <i class="form-control-feedback glyphicon glyphicon-search"></i>
+        </div>
+
+
+        <ul class="nav nav-pills nav-stacked">
+            <div class="list-group">
+                <a href ng-click="$dialog.selectOrg(item.organisation)" class="list-group-item"
+                    ng-class="{'active': $dialog.currentUser.organisation === item.organisation}"
+                    ng-repeat="item in $dialog.currentUser.organisations | filter:$modal.filter track by $index">
+                    <h4 class="list-group-item-heading">{{item.organisation}}</h4>
+                    <p class="list-group-item-text">Profile: <strong>{{item.role}}</strong></p>
+                </a>
+            </div>
+        </ul>
+    </div>
+    <div class="modal-footer text-left">
+        <button class="btn btn-default" ng-click="$dialog.cancel()" type="button">Cancel</button>
+    </div>
+</form>