From dff0c8a0082aa41b96c9c64f2afd09e24138246e Mon Sep 17 00:00:00 2001
From: shiziyuan9527 <yuhao.li@fit2cloud.com>
Date: Mon, 11 May 2020 15:33:40 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9util.js?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../business/components/common/head/HeaderUser.vue   |  8 +++-----
 .../src/business/components/project/MsProject.vue    |  5 ++---
 .../src/business/components/settings/CurrentUser.vue |  5 ++---
 .../settings/organization/OrganizationMember.vue     |  5 ++---
 .../settings/organization/OrganizationWorkspace.vue  |  7 +++----
 .../components/settings/personal/PersonSetting.vue   |  4 ++--
 .../settings/workspace/WorkspaceMember.vue           |  5 ++---
 frontend/src/common/js/utils.js                      | 12 ++++++------
 8 files changed, 22 insertions(+), 29 deletions(-)

diff --git a/frontend/src/business/components/common/head/HeaderUser.vue b/frontend/src/business/components/common/head/HeaderUser.vue
index 685bfb86fb..59e63f8500 100644
--- a/frontend/src/business/components/common/head/HeaderUser.vue
+++ b/frontend/src/business/components/common/head/HeaderUser.vue
@@ -51,7 +51,7 @@
     TokenKey,
     WORKSPACE_ID
   } from '../../../../common/js/constants';
-  import {hasRoles, saveLocalStorage} from "../../../../common/js/utils";
+  import {getCurrentUser, hasRoles, saveLocalStorage} from "../../../../common/js/utils";
 
   export default {
     name: "MsUser",
@@ -68,7 +68,7 @@
           {index: '2-1', name: '无工作空间'},
         ],
         currentUserInfo: {},
-        currentUserId: JSON.parse(localStorage.getItem(TokenKey)).id,
+        currentUserId: getCurrentUser().id,
         workspaceIds: [],
         currentOrganizationName: '选择组织',
         currentWorkspaceName: '选择工作空间'
@@ -76,9 +76,7 @@
     },
     computed: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        // window.console.log(user);
-        return JSON.parse(user);
+        return getCurrentUser();
       }
     },
     methods: {
diff --git a/frontend/src/business/components/project/MsProject.vue b/frontend/src/business/components/project/MsProject.vue
index 5877d731f5..e95ce0728d 100644
--- a/frontend/src/business/components/project/MsProject.vue
+++ b/frontend/src/business/components/project/MsProject.vue
@@ -44,11 +44,11 @@
 <script>
   import MsCreateBox from "../settings/CreateBox";
   import {Message} from "element-ui";
-  import {TokenKey} from "../../../common/js/constants";
   import MsTablePagination from "../common/pagination/TablePagination";
   import MsTableHeader from "../common/components/MsTableHeader";
   import MsTableOperator from "../common/components/MsTableOperator";
   import MsDialogFooter from "../common/components/MsDialogFooter";
+  import {getCurrentUser} from "../../../common/js/utils";
 
   export default {
     name: "MsProject",
@@ -97,8 +97,7 @@
     },
     computed: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        return JSON.parse(user);
+        return getCurrentUser();
       }
     },
     destroyed() {
diff --git a/frontend/src/business/components/settings/CurrentUser.vue b/frontend/src/business/components/settings/CurrentUser.vue
index 0ecfb9058f..9b62a8d600 100644
--- a/frontend/src/business/components/settings/CurrentUser.vue
+++ b/frontend/src/business/components/settings/CurrentUser.vue
@@ -6,7 +6,7 @@
 </template>
 
 <script>
-  import {TokenKey} from "../../../common/js/constants";
+  import {getCurrentUser} from "../../../common/js/utils";
 
   export default {
     name: "MsCurrentUser",
@@ -30,8 +30,7 @@
     },
     computed: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        return JSON.parse(user);
+        return getCurrentUser();
       }
     }
   }
diff --git a/frontend/src/business/components/settings/organization/OrganizationMember.vue b/frontend/src/business/components/settings/organization/OrganizationMember.vue
index cc2b830134..1096908a50 100644
--- a/frontend/src/business/components/settings/organization/OrganizationMember.vue
+++ b/frontend/src/business/components/settings/organization/OrganizationMember.vue
@@ -95,12 +95,12 @@
 
 <script>
   import MsCreateBox from "../CreateBox";
-  import {TokenKey} from "../../../../common/js/constants";
   import MsTablePagination from "../../common/pagination/TablePagination";
   import MsTableHeader from "../../common/components/MsTableHeader";
   import MsRolesTag from "../../common/components/MsRolesTag";
   import MsTableOperator from "../../common/components/MsTableOperator";
   import MsDialogFooter from "../../common/components/MsDialogFooter";
+  import {getCurrentUser} from "../../../../common/js/utils";
 
   export default {
     name: "MsOrganizationMember",
@@ -134,8 +134,7 @@
     },
     methods: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        return JSON.parse(user);
+        return getCurrentUser();
       },
       initTableData() {
         let param = {
diff --git a/frontend/src/business/components/settings/organization/OrganizationWorkspace.vue b/frontend/src/business/components/settings/organization/OrganizationWorkspace.vue
index 743a9b5065..c682e12590 100644
--- a/frontend/src/business/components/settings/organization/OrganizationWorkspace.vue
+++ b/frontend/src/business/components/settings/organization/OrganizationWorkspace.vue
@@ -145,13 +145,13 @@
 <script>
   import MsCreateBox from "../CreateBox";
   import {Message} from "element-ui";
-  import {DEFAULT, TokenKey} from "../../../../common/js/constants";
+  import {DEFAULT} from "../../../../common/js/constants";
   import MsTablePagination from "../../common/pagination/TablePagination";
   import MsTableHeader from "../../common/components/MsTableHeader";
   import MsRolesTag from "../../common/components/MsRolesTag";
   import MsTableOperator from "../../common/components/MsTableOperator";
   import MsDialogFooter from "../../common/components/MsDialogFooter";
-  import {getCurrentWorkspaceId, refreshSessionAndCookies} from "../../../../common/js/utils";
+  import {getCurrentUser, getCurrentWorkspaceId, refreshSessionAndCookies} from "../../../../common/js/utils";
 
   export default {
     name: "MsOrganizationWorkspace",
@@ -161,8 +161,7 @@
     },
     computed: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        return JSON.parse(user);
+        return getCurrentUser();
       }
     },
     methods: {
diff --git a/frontend/src/business/components/settings/personal/PersonSetting.vue b/frontend/src/business/components/settings/personal/PersonSetting.vue
index e4627660d8..0ec98416c2 100644
--- a/frontend/src/business/components/settings/personal/PersonSetting.vue
+++ b/frontend/src/business/components/settings/personal/PersonSetting.vue
@@ -82,6 +82,7 @@
 <script>
   import {TokenKey} from "../../../../common/js/constants";
   import MsDialogFooter from "../../common/components/MsDialogFooter";
+  import {getCurrentUser} from "../../../../common/js/utils";
 
   export default {
     data() {
@@ -152,8 +153,7 @@
     },
     methods: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        return JSON.parse(user);
+        return getCurrentUser();
       },
       edit(row) {
         this.updateVisible = true;
diff --git a/frontend/src/business/components/settings/workspace/WorkspaceMember.vue b/frontend/src/business/components/settings/workspace/WorkspaceMember.vue
index 188f74fafd..65374f59fe 100644
--- a/frontend/src/business/components/settings/workspace/WorkspaceMember.vue
+++ b/frontend/src/business/components/settings/workspace/WorkspaceMember.vue
@@ -97,12 +97,12 @@
 
 <script>
   import MsCreateBox from "../CreateBox";
-  import {TokenKey} from "../../../../common/js/constants";
   import MsTablePagination from "../../common/pagination/TablePagination";
   import MsTableHeader from "../../common/components/MsTableHeader";
   import MsRolesTag from "../../common/components/MsRolesTag";
   import MsTableOperator from "../../common/components/MsTableOperator";
   import MsDialogFooter from "../../common/components/MsDialogFooter";
+  import {getCurrentUser} from "../../../../common/js/utils";
 
   export default {
     name: "MsMember",
@@ -136,8 +136,7 @@
     },
     methods: {
       currentUser: () => {
-        let user = localStorage.getItem(TokenKey);
-        return JSON.parse(user);
+        return getCurrentUser();
       },
       initTableData() {
         if (this.currentUser().lastWorkspaceId === null) {
diff --git a/frontend/src/common/js/utils.js b/frontend/src/common/js/utils.js
index 9935e1610c..f70fbc971b 100644
--- a/frontend/src/common/js/utils.js
+++ b/frontend/src/common/js/utils.js
@@ -9,13 +9,13 @@ import {
 import axios from "axios";
 
 export function hasRole(role) {
-  let user = JSON.parse(localStorage.getItem(TokenKey));
+  let user = getCurrentUser();
   let roles = user.roles.map(r => r.id);
   return roles.indexOf(role) > -1;
 }
 
 export function hasRoles(...roles) {
-  let user = JSON.parse(localStorage.getItem(TokenKey));
+  let user = getCurrentUser();
   let rs = user.roles.map(r => r.id);
   for (let item of roles) {
     if (rs.indexOf(item) > -1) {
@@ -26,24 +26,24 @@ export function hasRoles(...roles) {
 }
 
 export function checkoutCurrentOrganization() {
-  let user = JSON.parse(localStorage.getItem(TokenKey));
+  let user = getCurrentUser();
   // 查看当前用户是否是 lastOrganizationId 的组织管理员
   return user.userRoles.filter(ur => hasRole(ROLE_ORG_ADMIN) && user.lastOrganizationId === ur.sourceId).length > 0;
 }
 
 export function checkoutCurrentWorkspace() {
-  let user = JSON.parse(localStorage.getItem(TokenKey));
+  let user = getCurrentUser();
   // 查看当前用户是否是 lastWorkspaceId 的工作空间用户
   return user.userRoles.filter(ur => hasRoles(ROLE_TEST_MANAGER, ROLE_TEST_USER, ROLE_TEST_VIEWER) && user.lastWorkspaceId === ur.sourceId).length > 0;
 }
 
 export function getCurrentOrganizationId() {
-  let user = JSON.parse(localStorage.getItem(TokenKey));
+  let user = getCurrentUser();
   return user.lastOrganizationId;
 }
 
 export function getCurrentWorkspaceId() {
-  let user = JSON.parse(localStorage.getItem(TokenKey));
+  let user = getCurrentUser();
   return user.lastWorkspaceId;
 }