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', '/xty/login', '/designxty/login']; // 不重定向白名单

const loginPage = whiteList[0];

const loginDesignPage = whiteList[1];

const loginXtyPage = whiteList[2];

const loginDesignXtyPage = whiteList[3];

// 获取路径数据
function getRouteInfo(to) {
    let loginPath = '/';
    let getTokenInfo = () => { };
    let clientId = '';
    const toRoutePath = to.redirectedFrom || to.path;
    if (/^\/designxty/.test(toRoutePath)) {
        loginPath = loginDesignXtyPage;
        getTokenInfo = getDesignToken;
        clientId = LoginParams.Design.clientId;
    } else 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 = getSessionStorage('project') === 'designxty' ? loginDesignXtyPage : loginDesignPage;
        getTokenInfo = getDesignToken;
        clientId = LoginParams.Design.clientId;
    } else if (/^\/plan/.test(toRoutePath) || /^\/display\/plan/.test(toRoutePath)) {
        if (getSessionStorage('project').startsWith('design')) {
            loginPath = getSessionStorage('project') === 'designxty' ? loginDesignXtyPage : loginDesignPage;
            getTokenInfo = getDesignToken;
            clientId = LoginParams.Design.clientId;
        } else {
            loginPath = getSessionStorage('project') === 'xty' ? loginXtyPage : loginPage;
            getTokenInfo = getToken;
            clientId = null;
        }
    } else if ( /^\/xty/.test(toRoutePath)) {
        loginPath = loginXtyPage;
        getTokenInfo = getToken;
        clientId = null;
    } else {
        loginPath = getSessionStorage('project') === 'xty' ? loginXtyPage : 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').startsWith('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 === '/') {
                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();
});