From c715ebd481259d96a12b3cde8f1265568d4e9989 Mon Sep 17 00:00:00 2001
From: "Captain.B" <bin@fit2cloud.com>
Date: Thu, 20 Feb 2020 22:27:48 +0800
Subject: [PATCH] current user

---
 frontend/src/common/constants.js              |   6 ++
 .../src/performance/components/HeaderUser.vue |  25 +++--
 .../components/settings/CurrentUser.vue       |  36 ++++---
 .../components/settings/SettingMenu.vue       | 102 +++++++++---------
 4 files changed, 96 insertions(+), 73 deletions(-)

diff --git a/frontend/src/common/constants.js b/frontend/src/common/constants.js
index 7bdfc87370..2c173d792e 100644
--- a/frontend/src/common/constants.js
+++ b/frontend/src/common/constants.js
@@ -1 +1,7 @@
 export const TokenKey = 'Admin-Token';
+
+export const ROLE_ADMIN = 'admin';
+export const ROLE_ORG_ADMIN = 'org_admin';
+export const ROLE_TEST_MANAGER = 'test_manager';
+export const ROLE_TEST_USER = 'test_user';
+export const ROLE_TEST_VIEWER = 'test_viewer';
diff --git a/frontend/src/performance/components/HeaderUser.vue b/frontend/src/performance/components/HeaderUser.vue
index 758741c1c8..0abba4a06a 100644
--- a/frontend/src/performance/components/HeaderUser.vue
+++ b/frontend/src/performance/components/HeaderUser.vue
@@ -6,7 +6,7 @@
                class="header-user-menu"
                background-color="rgb(44, 42, 72)"
                text-color="#fff">
-        <el-submenu index="1" popper-class="submenu">
+        <el-submenu index="1" popper-class="submenu" v-permission="['org_admin']">
           <template slot="title">组织</template>
           <label v-for="(item,index) in organizationList" :key="index">
             <el-menu-item @click="clickMenu(item)">{{item.name}}
@@ -15,7 +15,7 @@
             </el-menu-item>
           </label>
         </el-submenu>
-        <el-submenu index="2" popper-class="submenu">
+        <el-submenu index="2" popper-class="submenu" v-permission="['test_manager', 'test_user', 'test_viewer']">
           <template slot="title">工作空间</template>
           <label v-for="(item,index) in workspaceList" :key="index">
             <el-menu-item @click="clickMenu(item)">
@@ -43,7 +43,7 @@
 
 <script>
   import Cookies from 'js-cookie';
-  import {TokenKey} from '../../common/constants';
+  import {ROLE_ORG_ADMIN, ROLE_TEST_MANAGER, ROLE_TEST_USER, ROLE_TEST_VIEWER, TokenKey} from '../../common/constants';
 
   export default {
     name: "MsUser",
@@ -98,13 +98,18 @@
         }
       },
       initMenuData() {
-        this.$get("/organization/list/userorg/" + this.currentUserId, response => {
-          this.organizationList = response.data;
-        })
-        this.$get("/workspace/list/userworkspace/" + this.currentUserId, response => {
-          this.workspaceList = response.data;
-          this.workspaceIds = response.data.map(r => r.id);
-        })
+        let roles = this.currentUser.roles.map(r => r.id);
+        if (roles.indexOf(ROLE_ORG_ADMIN) > 0) {
+          this.$get("/organization/list/userorg/" + this.currentUserId, response => {
+            this.organizationList = response.data;
+          });
+        }
+        if (roles.indexOf(ROLE_TEST_MANAGER) > 0 || roles.indexOf(ROLE_TEST_USER) > 0 || roles.indexOf(ROLE_TEST_VIEWER) > 0) {
+          this.$get("/workspace/list/userworkspace/" + this.currentUserId, response => {
+            this.workspaceList = response.data;
+            this.workspaceIds = response.data.map(r => r.id);
+          });
+        }
       },
       getCurrentUserInfo() {
         this.$get("/user/info/" + this.currentUserId, response => {
diff --git a/frontend/src/performance/components/settings/CurrentUser.vue b/frontend/src/performance/components/settings/CurrentUser.vue
index e9c16e0b05..d69758242a 100644
--- a/frontend/src/performance/components/settings/CurrentUser.vue
+++ b/frontend/src/performance/components/settings/CurrentUser.vue
@@ -1,10 +1,10 @@
 <template>
   <el-row type="flex" align="middle" class="current-user">
     <el-avatar shape="square" size="small" :src="squareUrl"/>
-    <span class="username">kun@fit2cloud.com</span>
+    <span class="username">{{currentUser.name}}</span>
     <el-button class="edit" type="primary" icon="el-icon-edit" size="mini"
-               circle @click="editVisible = true"/>
-    <el-dialog :title="title" :visible.sync="editVisible" width="30%">
+               circle @click="edit"/>
+    <el-dialog :title="currentUser.name" :visible.sync="editVisible" width="30%">
       <el-form :model="form" label-position="top" size="small">
         <el-form-item label="姓名">
           <el-input v-model="form.name" autocomplete="off"/>
@@ -14,28 +14,40 @@
         </el-form-item>
       </el-form>
       <span slot="footer" class="dialog-footer">
-                <el-button type="primary" @click="editVisible = false" size="medium">更新</el-button>
-            </span>
+        <el-button type="primary" @click="submit" size="medium">更新</el-button>
+      </span>
     </el-dialog>
   </el-row>
 </template>
 
 <script>
+  import Cookies from 'js-cookie';
+  import {TokenKey} from "../../../common/constants";
+
   export default {
     name: "MsCurrentUser",
+
     data() {
       return {
         editVisible: false,
         id: "123456",
         squareUrl: "https://cube.elemecdn.com/9/c2/f0ee8a3c7c9638a54940382568c9dpng.png",
-        form: {
-          name: "kun@fit2cloud.com",
-          mobile: ""
-        }
+        form: {}
       }
-    }, computed: {
-      title: function () {
-        return "编辑账号(id: " + this.id + ")";
+    },
+    methods: {
+      edit() {
+        this.editVisible = true;
+        this.form = Object.assign({}, this.currentUser);
+      },
+      submit() {
+        this.editVisible = false;
+      }
+    },
+    computed: {
+      currentUser: () => {
+        let user = Cookies.get(TokenKey);
+        return JSON.parse(user);
       }
     }
   }
diff --git a/frontend/src/performance/components/settings/SettingMenu.vue b/frontend/src/performance/components/settings/SettingMenu.vue
index 147ec2f87f..82c6392fa9 100644
--- a/frontend/src/performance/components/settings/SettingMenu.vue
+++ b/frontend/src/performance/components/settings/SettingMenu.vue
@@ -1,63 +1,63 @@
 <template>
-    <el-menu menu-trigger="click" :default-active="$route.path" router>
-        <el-submenu index="1" v-permission="['admin']">
-            <template slot="title">
-                <font-awesome-icon class="icon account" :icon="['far', 'address-card']" size="lg"/>
-                <span>账号</span>
-            </template>
-            <el-menu-item index="/setting/user">用户</el-menu-item>
-            <el-menu-item index="/setting/testresourcepool" v-permission="['test']">测试资源池</el-menu-item>
-            <el-menu-item index="/setting/organization">组织</el-menu-item>
-            <el-menu-item index="/setting/workspace">工作空间</el-menu-item>
-            <el-menu-item>API Keys</el-menu-item>
-        </el-submenu>
-        <el-submenu index=2 v-permission="['admin','org_admin']">
-            <template slot="title">
-                <font-awesome-icon class="icon workspace" :icon="['far', 'clone']" size="lg"/>
-                <span>工作空间</span>
-            </template>
-            <el-menu-item index="/setting/member">成员</el-menu-item>
-            <el-menu-item>证书</el-menu-item>
-            <el-menu-item>测试计划</el-menu-item>
-            <el-menu-item>警告</el-menu-item>
-        </el-submenu>
-        <el-submenu index="3">
-            <template slot="title">
-                <font-awesome-icon class="icon" :icon="['far', 'user']" size="lg"/>
-                <span>个人</span>
-            </template>
-            <el-menu-item>个人设置</el-menu-item>
-            <el-menu-item>API Keys</el-menu-item>
-        </el-submenu>
-    </el-menu>
+  <el-menu menu-trigger="click" :default-active="$route.path" router>
+    <el-submenu index="1" v-permission="['admin']">
+      <template slot="title">
+        <font-awesome-icon class="icon account" :icon="['far', 'address-card']" size="lg"/>
+        <span>账号</span>
+      </template>
+      <el-menu-item index="/setting/user" v-permission="['admin']">用户</el-menu-item>
+      <el-menu-item index="/setting/testresourcepool" v-permission="['test']">测试资源池</el-menu-item>
+      <el-menu-item index="/setting/organization" v-permission="['admin']">组织</el-menu-item>
+      <el-menu-item index="/setting/workspace" v-permission="['org_admin']">工作空间</el-menu-item>
+      <el-menu-item>API Keys</el-menu-item>
+    </el-submenu>
+    <el-submenu index=2 v-permission="['org_admin']">
+      <template slot="title">
+        <font-awesome-icon class="icon workspace" :icon="['far', 'clone']" size="lg"/>
+        <span>工作空间</span>
+      </template>
+      <el-menu-item index="/setting/member">成员</el-menu-item>
+      <el-menu-item>证书</el-menu-item>
+      <el-menu-item>测试计划</el-menu-item>
+      <el-menu-item>警告</el-menu-item>
+    </el-submenu>
+    <el-submenu index="3">
+      <template slot="title">
+        <font-awesome-icon class="icon" :icon="['far', 'user']" size="lg"/>
+        <span>个人</span>
+      </template>
+      <el-menu-item>个人设置</el-menu-item>
+      <el-menu-item>API Keys</el-menu-item>
+    </el-submenu>
+  </el-menu>
 </template>
 
 <script>
-    export default {
-        name: "MsSettingMenu"
-    }
+  export default {
+    name: "MsSettingMenu"
+  }
 </script>
 
 <style scoped>
-    .el-menu {
-        border-right: 0;
-    }
+  .el-menu {
+    border-right: 0;
+  }
 
-    .el-menu-item {
-        height: 40px;
-        line-height: 40px;
-    }
+  .el-menu-item {
+    height: 40px;
+    line-height: 40px;
+  }
 
-    .icon {
-        width: 24px;
-        margin-right: 10px;
-    }
+  .icon {
+    width: 24px;
+    margin-right: 10px;
+  }
 
-    .account {
-        color: #5a78f0;
-    }
+  .account {
+    color: #5a78f0;
+  }
 
-    .workspace {
-        color: #44b349;
-    }
+  .workspace {
+    color: #44b349;
+  }
 </style>