rt-sim-training-client/src/permission.js

import Vue from 'vue';
import store from '@/store';
import router from './router';
import NProgress from 'nprogress'; // Progress 进度条
import 'nprogress/nprogress.css';// Progress 进度条样式
import { admin, userDesign} from './router';
import { getToken, getDesignToken} from '@/utils/auth'; // 验权
import { LoginParams } from '@/utils/login';
import { getSessionStorage } from '@/utils/auth';
import localStore from 'storejs';

function hasPermission(roles, permissionRoles) {
	if (roles.indexOf(admin) >= 0) return true;
	if (!permissionRoles) return true;
	return roles.some(role => permissionRoles.indexOf(role) >= 0);
}

const whiteList = ['/login', '/design/login', '/login/xty']; // 不重定向白名单

const loginPage = whiteList[0];

const loginDesignPage = whiteList[1];

// 获取路径数据
function getRouteInfo(to) {
	let loginPath = '/';
	let getTokenInfo = () => { };
	let clientId = '';
	const toRoutePath = to.redirectedFrom || to.path;
	if (/^\/design/.test(toRoutePath) || /^\/scriptDisplay/.test(toRoutePath) || /^\/publish/.test(toRoutePath) || /^\/orderauthor/.test(toRoutePath) || /^\/system/.test(toRoutePath)|| /^\/display\/record/.test(toRoutePath) || /^\/display\/manage/.test(toRoutePath) || /^\/apply/.test(toRoutePath)) {
		loginPath = loginDesignPage;
		getTokenInfo = getDesignToken;
		clientId = LoginParams.Design.clientId;
	} else if (/^\/plan/.test(toRoutePath) || /^\/display\/plan/.test(toRoutePath)) {
		if (getSessionStorage('project')==='design') {
			loginPath = loginDesignPage;
			getTokenInfo = getDesignToken;
			clientId = LoginParams.Design.clientId;
		} else {
			loginPath = loginPage;
			getTokenInfo = getToken;
			clientId = null;
		}
	} else {
		loginPath = loginPage;
		getTokenInfo = getToken;
		clientId = null;
	}

	return { clientId, loginPath, getTokenInfo };
}

function handleRoute(to, from, next, routeInfo) {
	if (store.getters.roles.length === 0) {
		// 拉取用户信息
		store.dispatch('GetInfo', routeInfo.getTokenInfo).then(res => {
			// 根据roles权限生成可访问的路由表
			const roles = res.roles;
			if (getSessionStorage('project')==='design') {
				roles.push(userDesign);
			}
			store.dispatch('GenerateRoutes', { roles, clientId: routeInfo.clientId }).then(() => {
				// 动态添加可访问路由表
				router.addRoutes(store.getters.addRouters);
				// router.addRoutes(asyncRouter1);
				if (to.redirectedFrom) {
					next({ path: to.redirectedFrom, replace: true });
				} else {
					next({ ...to, replace: true });
				}
			});

		}).catch(() => {
			store.dispatch('FedLogOut', routeInfo.clientId).then(() => {
				Vue.prototype.$messageBox('验证失败，请重新登录！');
				next({ path: routeInfo.loginPath });
			});
		});
	} else {
		// 除没有动态改变权限的需求可直接next() 删下方权限判断
		if (hasPermission(store.getters.roles, to.meta.roles)) {
			if (to.path==='/404' && to.redirectedFrom==='/') {
				if (getSessionStorage('project') === 'design') {
					next('/design/home');
				} else {
					next(localStore.get('trainingPlatformRoute'+store.getters.id) ||'/trainingPlatform');
				}
			} else {
				next();
			}
		} else {
			next({ path: '/401', replace: true, query: { noGoBack: true } });
		}
	}
}

router.beforeEach((to, from, next) => {
	NProgress.start();
	const routeInfo = getRouteInfo(to);
	if (routeInfo.getTokenInfo()) {
		// 已登录
		if (to.path === routeInfo.loginPath) {
			// 登录页面不拦截
			next();
		} else {
			// 进入系统重新计算路由
			handleRoute(to, from, next, routeInfo);
		}
	} else {
		// 未登录情况下
		if (whiteList.indexOf(to.path) !== -1) {
			// 在免登录白名单，直接进入
			next();
		} else {
			// 否则全部重定向到登录页
			next(routeInfo.loginPath);
		}
	}
});

router.afterEach(() => {
	// 结束Progress
	NProgress.done();
});