payment-platform/frontend/src/plugins/permission.ts

import type { NavigationGuardNext, RouteLocationNormalized, RouteRecordRaw } from "vue-router";
import NProgress from "@/utils/nprogress";
import { Auth } from "@/utils/auth";
import router from "@/router";
import { usePermissionStore, useUserStore, useEnterpriseStore } from "@/store";

export function setupPermission() {
  // 白名单路由
  const whiteList = ["/login"];

  router.beforeEach(async (to, from, next) => {
    NProgress.start();

    try {
      const isLoggedIn = Auth.isLoggedIn();

      if (isLoggedIn) {
        // 如果已登录但访问登录页，重定向到首页
        if (to.path === "/login") {
          next({ path: "/" });
          return;
        }

        // 处理已登录用户的路由访问
        await handleAuthenticatedUser(to, from, next);
      } else {
        // 未登录用户的处理
        if (whiteList.includes(to.path)) {
          next();
        } else {
          next(`/login?redirect=${encodeURIComponent(to.fullPath)}`);
          NProgress.done();
        }
      }
    } catch (error) {
      // 错误处理：重置状态并跳转登录（保留 redirect 与未登录分支一致）
      console.error("Route guard error:", error);
      await useUserStore().resetAllState();
      next(`/login?redirect=${encodeURIComponent(to.fullPath)}`);
      NProgress.done();
    }
  });

  // 后置守卫，确保进度条关闭
  router.afterEach(() => {
    NProgress.done();
  });
}

/**
 * 处理已登录用户的路由访问
 */
async function handleAuthenticatedUser(
  to: RouteLocationNormalized,
  from: RouteLocationNormalized,
  next: NavigationGuardNext
) {
  const permissionStore = usePermissionStore();
  const userStore = useUserStore();
  const enterpriseStore = useEnterpriseStore();

  try {
    // 检查路由是否已生成
    if (!permissionStore.isRouteGenerated) {
      // 如果用户信息为空，获取用户信息
      if (!userStore.basicInfo?.roles?.length) {
        await userStore.getUserInfo();
      }

      // 如果是租户用户，获取企业列表
      if (!userStore.is_platform_user) {
        await enterpriseStore.loadEnterpriseList();
      }

      const dynamicRoutes = await permissionStore.generateRoutes();
      // 添加路由到路由器
      dynamicRoutes.forEach((route: RouteRecordRaw) => {
        router.addRoute('/', route);
      });

      // 路由生成完成后，重新导航到目标路由
      next({ ...to, replace: true });
      return;
    }

    // 路由已加载，检查路由是否存在
    if (to.matched.length === 0) {
      next("/404");
      return;
    }

    // 动态设置页面标题（仅做简单净化，避免 query 注入标签/过长字符串影响标签栏展示）
    const rawTitle = (to.params.title as string) || (to.query.title as string);
    if (rawTitle && typeof rawTitle === "string") {
      const safe = rawTitle.replace(/[<>]/g, "").trim().slice(0, 64);
      if (safe) {
        to.meta.title = safe;
      }
    }

    next();
  } catch (error) {
    console.error("❌ Route guard error:", error);
    await useUserStore().resetAllState();
    next(`/login?redirect=${encodeURIComponent(to.fullPath)}`);
    NProgress.done();
  }
}