hlt_gis_admin/src/router/guard/permissionGuard.ts

import type { Router, RouteRecordRaw } from 'vue-router';
import { usePermissionStoreWithOut } from '/@/store/modules/permission';
import { PageEnum } from '/@/enums/pageEnum';
import { useUserStore, useUserStoreWithOut } from '/@/store/modules/user';
import { PAGE_NOT_FOUND_ROUTE } from '/@/router/routes/basic';
import { RootRoute } from '/@/router/routes';
import { getMenuList } from '/@/api/sys/menu';
import { USER_INFO_KEY } from '/@/enums/cacheEnum';
import { getAuthCache } from '/@/utils/auth';
import { session } from '/@/utils/Memory';

const LOGIN_PATH = PageEnum.BASE_LOGIN;
const DEVICE_LIST = PageEnum.DEVICE_LIST;
const ROOT_PATH = RootRoute.path;
const whitePathList: string[] = [LOGIN_PATH, DEVICE_LIST];

export async function createPermissionGuard(router: Router) {
  const userStore = useUserStoreWithOut();
  const permissionStore = usePermissionStoreWithOut();
  const userStore2 = useUserStore();
  var token = session.getItem('token')
  var tokenV2 = session.getItem('tokenV2')
  if ((location.href.indexOf('refreshToken') > -1) && (!token || !tokenV2)) {
    var o = session.getItem('loginParams');

    await userStore2.login({
      // password: 'Tofly@028',
      // username: 'ADMIN2',
      password: o.password,
      username: o.username,
      mode: 'modal', //不要默认的错误提示
    }).catch((data) => {
      console.log("自动登录:", data);
    });
    //修改地址栏链接
    window.history.pushState('', '', location.href.split('?')[0]);
  }


  router.beforeEach(async (to, from, next) => {
    const userInfo = await getAuthCache(USER_INFO_KEY);
    if (
      from.path === ROOT_PATH &&
      to.path === PageEnum.BASE_HOME &&
      userStore.getUserInfo.homePath &&
      userStore.getUserInfo.homePath !== PageEnum.BASE_HOME
    ) {
      next(userStore.getUserInfo.homePath);
      return;
    }

    const token = userStore.getJwtToken;
    // Whitelist can be directly entered
    // 路由守卫拦截, 如果是已经登陆情况, 就不要回到登陆页面了;
    if (whitePathList.includes(to.path as PageEnum)) {
      if (to.path === LOGIN_PATH && token) {
        const isSessionTimeout = userStore.getSessionTimeout;
        try {
          // await userStore.afterLoginAction();
          if (!isSessionTimeout) {
            next((to.query?.redirect as string) || '/');
            return;
          }
        } catch { }
      } else {
        // window.location.href = '/admin/index.html';//地灾项目使用这个
        // return;
      }
      next();
      return;
    }

    // token does not exist
    if (!token) {
      // You can access without permission. You need to set the routing meta.ignoreAuth to true
      if (to.meta.ignoreAuth) {
        next();
        return;
      }

      // redirect login page
      const redirectData: { path: string; replace: boolean; query?: Recordable<string> } = {
        path: LOGIN_PATH,
        replace: true,
      };
      if (to.path) {
        redirectData.query = {
          ...redirectData.query,
          redirect: to.path,
        };
      }
      console.log('redirectData', redirectData);

      next(redirectData);
      return;
    }

    if (from.path === LOGIN_PATH && userInfo?.needSetPwd == false) {
      const getMenuListData = getAuthCache('MENU_LIST') || (await getMenuList());
      // const getMenuListData = await getMenuList();
      const getHomePage = getMenuListData.find((f) => {
        return f.path == '/dashboard/workbench';
      });
      if (getHomePage?.path == '/dashboard/workbench') {
        setTimeout(() => {
          router.push('/dashboard/workbench');
        }, 1);
      } else {
        const routeF = getMenuListData[0]?.children[0]?.path || getMenuListData[0]?.path;
        if (routeF) {
          setTimeout(() => {
            router.push(routeF);
          }, 1);
        }
      }
    }

    if (from.path === LOGIN_PATH && userInfo?.needSetPwd == true) {
      setTimeout(() => {
        router.push('/system/changePassword');
      }, 1);
    }

    if (
      from.path === LOGIN_PATH &&
      to.name === PAGE_NOT_FOUND_ROUTE.name &&
      to.fullPath !== (userStore.getUserInfo.homePath || PageEnum.BASE_HOME)
    ) {
      next(userStore.getUserInfo.homePath || PageEnum.BASE_HOME);
      return;
    }

    if (permissionStore.getIsDynamicAddedRoute) {
      next();
      return;
    }

    const routes = await permissionStore.buildRoutesAction();

    routes?.forEach((route) => {
      if (!route.path || (route.path && route.path.indexOf('/') > -1)) {
        router.addRoute(route as unknown as RouteRecordRaw);
      }
    });

    router.addRoute(PAGE_NOT_FOUND_ROUTE as unknown as RouteRecordRaw);

    permissionStore.setDynamicAddedRoute(true);
    // console.log(to.query);

    if (to.name === PAGE_NOT_FOUND_ROUTE.name) {
      // 动态添加路由后，此处应当重定向到fullPath，否则会加载404页面内容
      var tk = session.getItem("tokenV2");
      if (tk && token) {
        next({ path: to.fullPath, replace: true, query: to.query });
      } else {
        userStore.setToken(undefined)
        window.localStorage.clear();
        window.sessionStorage.clear();
        next({ path: PageEnum.BASE_LOGIN, replace: true, query: to.query });
      }
    } else {
      const redirectPath = (from.query.redirect || to.path) as string;
      const redirect = decodeURIComponent(redirectPath);
      const nextData = to.path === redirect ? { ...to, replace: true } : { path: redirect };
      next(nextData);
    }
  });
}