From 1322990e739b17d21bebb108bbb33728b1c50db0 Mon Sep 17 00:00:00 2001
From: davidjumani <dj.davidjumani1994@gmail.com>
Date: Fri, 14 Aug 2020 13:43:48 +0530
Subject: [PATCH] saml: Adding Account Switcher for SAML Accounts (#575)

Adding Account Switcher for SAML Accounts in multiple domains matching the same saml user

Co-authored-by: Rohit Yadav <rohit.yadav@shapeblue.com>
Signed-off-by: Rohit Yadav <rohit.yadav@shapeblue.com>
---
 ui/src/components/header/ProjectMenu.vue      |   2 +-
 .../components/header/SamlDomainSwitcher.vue  | 124 ++++++++++++++++++
 ui/src/components/page/GlobalHeader.vue       |   4 +
 3 files changed, 129 insertions(+), 1 deletion(-)
 create mode 100644 ui/src/components/header/SamlDomainSwitcher.vue

diff --git a/ui/src/components/header/ProjectMenu.vue b/ui/src/components/header/ProjectMenu.vue
index 11d72dd77ea..7751e050a01 100644
--- a/ui/src/components/header/ProjectMenu.vue
+++ b/ui/src/components/header/ProjectMenu.vue
@@ -109,7 +109,7 @@ export default {
 <style lang="less" scoped>
 .project {
   &-select {
-    width: 40vw;
+    width: 30vw;
   }
 
   &-icon {
diff --git a/ui/src/components/header/SamlDomainSwitcher.vue b/ui/src/components/header/SamlDomainSwitcher.vue
new file mode 100644
index 00000000000..edce64a873b
--- /dev/null
+++ b/ui/src/components/header/SamlDomainSwitcher.vue
@@ -0,0 +1,124 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+<template>
+  <span class="header-notice-opener" v-if="showSwitcher">
+    <a-select
+      class="domain-select"
+      :loading="loading"
+      :defaultValue="currentAccount"
+      :value="currentAccount"
+      @change="changeAccount"
+      @focus="fetchData" >
+
+      <a-tooltip placement="bottom" slot="suffixIcon">
+        <template slot="title">
+          <span>{{ $t('label.domain') }}</span>
+        </template>
+        <span style="font-size: 20px; color: #999; margin-top: -5px">
+          <a-icon v-if="!loading" type="block" />
+          <a-icon v-else type="loading" />
+        </span>
+      </a-tooltip>
+
+      <a-select-option v-for="(account, index) in samlAccounts" :key="index">
+        {{ `${account.accountName} (${account.domainName})` }}
+      </a-select-option>
+    </a-select>
+  </span>
+</template>
+
+<script>
+import store from '@/store'
+import { api } from '@/api'
+import _ from 'lodash'
+
+export default {
+  name: 'SamlDomainSwitcher',
+  data () {
+    return {
+      showSwitcher: false,
+      samlAccounts: [],
+      currentAccount: '',
+      loading: false
+    }
+  },
+  mounted () {
+    this.fetchData()
+  },
+  methods: {
+    fetchData () {
+      var page = 1
+      const samlAccounts = []
+      const getNextPage = () => {
+        this.loading = true
+        api('listAndSwitchSamlAccount', { listAll: true, details: 'min', page: page, pageSize: 500 }).then(json => {
+          if (json && json.listandswitchsamlaccountresponse && json.listandswitchsamlaccountresponse.samluseraccount) {
+            samlAccounts.push(...json.listandswitchsamlaccountresponse.samluseraccount)
+          }
+          if (samlAccounts.length < json.listandswitchsamlaccountresponse.count) {
+            page++
+            getNextPage()
+          }
+        }).catch(error => {
+          console.log(error)
+        }).finally(() => {
+          if (samlAccounts.length < 2) {
+            this.showSwitcher = false
+            return
+          }
+          this.samlAccounts = _.orderBy(samlAccounts, ['domainPath'], ['asc'])
+          const currentAccount = this.samlAccounts.filter(x => {
+            return x.userId === store.getters.userInfo.id
+          })[0]
+          this.currentAccount = `${currentAccount.accountName} (${currentAccount.domainName})`
+          this.loading = false
+          this.showSwitcher = true
+        })
+      }
+      getNextPage()
+    },
+    changeAccount (index) {
+      const account = this.samlAccounts[index]
+      api('listAndSwitchSamlAccount', {}, 'POST', {
+        userid: account.userId,
+        domainid: account.domainId
+      }).then(response => {
+        store.dispatch('GetInfo').then(() => {
+          this.$message.success(`Switched to "${account.accountName} (${account.domainPath})"`)
+          this.$router.go()
+        })
+      })
+    }
+  }
+}
+</script>
+
+<style lang="less" scoped>
+.domain {
+  &-select {
+    width: 20vw;
+  }
+
+  &-icon {
+    font-size: 20px;
+    line-height: 1;
+    padding-top: 5px;
+    padding-right: 5px;
+  }
+}
+</style>
diff --git a/ui/src/components/page/GlobalHeader.vue b/ui/src/components/page/GlobalHeader.vue
index d06882cf462..c06651a93ad 100644
--- a/ui/src/components/page/GlobalHeader.vue
+++ b/ui/src/components/page/GlobalHeader.vue
@@ -29,6 +29,7 @@
         :type="collapsed ? 'menu-unfold' : 'menu-fold'"
         @click="toggle"/>
       <project-menu v-if="device !== 'mobile'" />
+      <saml-domain-switcher style="margin-left: 20px" />
       <user-menu></user-menu>
     </div>
     <div v-else :class="['top-nav-header-index', theme]">
@@ -48,6 +49,7 @@
             @click="toggle"></a-icon>
         </div>
         <project-menu v-if="device !== 'mobile'" />
+        <saml-domain-switcher style="margin-left: 20px" />
         <user-menu></user-menu>
       </div>
     </div>
@@ -60,6 +62,7 @@ import Breadcrumb from '@/components/widgets/Breadcrumb'
 import Logo from '../header/Logo'
 import SMenu from '../menu/'
 import ProjectMenu from '../header/ProjectMenu'
+import SamlDomainSwitcher from '../header/SamlDomainSwitcher'
 import UserMenu from '../header/UserMenu'
 
 import { mixin } from '@/utils/mixin.js'
@@ -71,6 +74,7 @@ export default {
     Logo,
     SMenu,
     ProjectMenu,
+    SamlDomainSwitcher,
     UserMenu
   },
   mixins: [mixin],