61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import router from "@/router";
|
||
import { useUserStore } from "@/store/modules/user";
|
||
import { usePermissionStore } from "@/store/modules/permission";
|
||
import NProgress from "@/utils/nprogress";
|
||
|
||
export function setupPermission() {
|
||
// 白名单路由
|
||
const whiteList = ["/login"];
|
||
|
||
router.beforeEach(async (to, from, next) => {
|
||
NProgress.start();
|
||
const hasToken = localStorage.getItem("token");
|
||
if (hasToken) {
|
||
if (to.path === "/login") {
|
||
// 如果已登录，跳转首页
|
||
next({ path: "/" });
|
||
NProgress.done();
|
||
} else {
|
||
const userStore = useUserStore();
|
||
const hasRoles =
|
||
userStore.user.roles && userStore.user.roles.length > 0;
|
||
if (hasRoles) {
|
||
// 未匹配到任何路由，跳转404
|
||
if (to.matched.length === 0) {
|
||
from.name ? next({ name: from.name }) : next("/404");
|
||
} else {
|
||
next();
|
||
}
|
||
} else {
|
||
const permissionStore = usePermissionStore();
|
||
try {
|
||
const { roles } = await userStore.getUserInfo();
|
||
const accessRoutes = await permissionStore.generateRoutes(roles);
|
||
accessRoutes.forEach((route) => {
|
||
router.addRoute(route);
|
||
});
|
||
next({ ...to, replace: true });
|
||
} catch (error) {
|
||
// 移除 token 并跳转登录页
|
||
await userStore.resetToken();
|
||
next(`/login?redirect=${to.path}`);
|
||
NProgress.done();
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
// 未登录可以访问白名单页面
|
||
if (whiteList.indexOf(to.path) !== -1) {
|
||
next();
|
||
} else {
|
||
next(`/login?redirect=${to.path}`);
|
||
NProgress.done();
|
||
}
|
||
}
|
||
});
|
||
|
||
router.afterEach(() => {
|
||
NProgress.done();
|
||
});
|
||
}
|