更新首页角色选择

src/views/index.vue

@@ -1,10 +1,10 @@
 <template>
   <div class="container">
-    <div class="item" @click="handleJump('LeaderIndex', 'leader')">
+    <div class="item" v-if="role_is_leader" @click="handleJump('LeaderIndex', 'leader')">
       <i class="icon1" />
       <span>指挥端</span>
     </div>
-    <div class="item2" @click="handleJump('WorkerIndex', 'worker')">
+    <div class="item2" v-if="role_is_worker" @click="handleJump('WorkerIndex', 'worker')">
       <i class="icon2" />
       <span>工作人员端</span>
     </div>
@@ -14,12 +14,15 @@
 <script lang="ts" setup>
 import {useRouter} from "vue-router";
 import useUserStore from "@/store/modules/user";
-import {onMounted} from "vue";
-
+import {getInfo} from "@/api/login"
+import {onMounted, ref} from "vue";
 const router = useRouter();
 const useUser = useUserStore();
 const env = import.meta.env.VITE_APP_ENV;
 
+const role_is_leader = ref(false);
+const role_is_worker = ref(false);
+
 const handleJump = (path: string, role: string) => {
   useUser.setRoles(role);
   router.push({ name: path });
@@ -30,8 +33,27 @@ onMounted(() => {
   if(env === 'development') {
     localStorage.setItem('Admin-Token', 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJzdWIiOiIxIiwiZXhwIjoyMDM5Njk2ODMzfQ.Rhd38oo_S1odjg0xnT4n31cCWCAAPXGb8y_V2XcgqzQ');
   }
-  useUser.setRoles('');
-  localStorage.setItem('role', '');
+  getInfo().then((res)=>{
+    const roles = res.data.roles;
+    console.log('roles', roles);
+    if(roles.includes("superadmin")) {
+      role_is_leader.value = true;
+      role_is_worker.value = true;
+    }
+    if(roles.includes("super_ld")) {
+      role_is_leader.value = true;
+    }
+    if(roles.includes("super_worker")) {
+      role_is_worker.value = true;
+    }
+    // auto jump....
+    if(role_is_leader.value && !role_is_worker.value) {
+      handleJump('LeaderIndex', 'leader');
+    }
+    if(!role_is_leader.value && role_is_worker.value) {
+      handleJump('WorkerIndex', 'worker');
+    }
+  })
 })
 </script>