react-router v6实现权限管理+自动替换页面标题的案例

// 在这里进行判断用户权限使用不同的页面
import React from "react";
import Merchant from "../pages/merchant";
import Admin from "../pages/admin";
import User from "../pages/user";
export default function Index() {
//这里使用power来判断是否登录，以及权限
  let power = localStorage.getItem("power");
  return (
    <div>
      {power ? (
        power === "1" ? (
          <Merchant></Merchant>
        ) : (
          <Admin></Admin>
        )
      ) : (
        <User></User>
      )}
    </div>
  );
}

// 引入路由
import Login from "../pages/user/login";
import Register from "../pages/user/register";
import NoPower from "../pages/error/noPower";
import NoPage from "../pages/error/noPage";
import { myRouteObj } from "../type";
//这里的myRouteObj是我自定义的一个类型用于拓展title实现页面标题的自动切换
export const userRoutes: myRouteObj[] = [
  {
    path: "/404",
    element: <NoPage></NoPage>,
    title: '404'
  },
  {
    path: "/login",
    element: <Login></Login>,
    title: '登录'
  },
  {
    path: "/register",
    element: <Register></Register>,
    title: '注册'
  },
];
const routes = [
  ...userRoutes,
  {
    path: "*",
    element: <NoPower></NoPower>,
    title: '没有权限'
  },
];
export default routes;

import { NonIndexRouteObject } from "react-router";
export interface myRouteObj extends NonIndexRouteObject {
  title?: string;
}

import React from "react";
import userRoutes from "../../router/user";
import { useMyRoutes } from "../../hooks/route";
export default function User() {
//这里的useMyRoutes是我自己封装的一个钩子函数
  const element = useMyRoutes(userRoutes);
  return <div>{element}</div>;
}

import { useEffect } from "react";
import { useLocation, useNavigate, useRoutes } from "react-router";
import { myRouteObj } from "../type";
function flatDeep(children: any, findArr: any): string {
  // 判断children数组中是否有想要的元素
  let tempArr = children.map((item: any) => item.path);
  let index = tempArr.indexOf(findArr[0]);
  if (index !== -1) {
    return findArr.length > 1
      ? flatDeep(children[index], findArr.splice(1, findArr.length))
      : children[index].title;
     ? children[index].title
     : "商城";
  }
  return "商城";
}
export function useMyRoutes(routes: myRouteObj[]) {
  // 判断路由表中是否有对应路由，如果没有就返回到404
  const location = useLocation();
  const navigate = useNavigate();
  useEffect(() => {
  /*
这里有一个小问题，困扰了我半天，就是原本我获取路径是写在useEffect外边的，页面每次刷新就会显示成错误的标题，最后输出是因为报存的值没有更新，所以就改到useEffect函数里边了
*/
    // 非空
    const firstPathName = location.pathname
      .split("/")
      .filter((item) => item !== "");
    const firstPath = routes.map((item) => item.path?.split("/")[1]);
    let index = firstPath.indexOf(firstPathName[0]);
    if (!index) {
      navigate("/404");
      return;
    }
    // 判断是否是首页
    if (location.pathname === "/") {
      document.title = "5ZWG5Z+O";
      return;
    }
    // 判断是几级路由
    if (firstPathName.length > 1) {
      // 多级
      document.title = flatDeep(
        routes[index].children,
        firstPathName.splice(1, firstPathName.length)
      );
      return;
    }
    // 一级
    document.title = routes[index].title
      ? (routes[index].title as string)
      : "商城";
  }, [location, navigate, routes]);
  return useRoutes(routes);
}

import { Navigate } from "react-router";
import Merchant from "../pages/merchant";
import Commodity from "../pages/merchant/commodity";
import { userRoutes } from "./user";
import { getRoutesObj } from "../utils";
import { myRouteObj } from "../type";
const children: myRouteObj[] = [
  {
    path: "commodity",
    element: <Commodity></Commodity>,
  },
];
const primitiveObj = {
  path: "/merchant/*",
  element: <Merchant></Merchant>,
  title: '商店后台'
};
// 有拦截器版本
export const [routes, defaultRoutes] = getRoutesObj(
  userRoutes,
  primitiveObj,
  children,
  {
    path: "*",
    element: <Navigate to="/merchant/commodity"></Navigate>,
  }
);

import { myRouteObj } from "../type";
// 第一个参数是复用的数组对象,第二个参数是有拦截器的对象
/**
 *生成路由表配置（加拦截器）
 * @export
 * @param {myRouteObj[]} reuseObj 复用的数组对象
 * @param {myRouteObj} targetObj 目标对象
 * @param {myRouteObj[]} childrenObj 子级路由
 * @param {myRouteObj} globalinterceptorObj 全局拦截路由
 * @param {myRouteObj} interceptor 可选拦截器对象(添加子级路由中)
 */
export function getRoutesObj(
  reuseObj: myRouteObj[],
  targetObj: myRouteObj,
  childrenObj: myRouteObj[],
  interceptorObj?: myRouteObj
) {
  //  先复制一份children
  const children = [...childrenObj];
  //   判断是否有值
  if (interceptorObj) {
    children.push(interceptorObj);
  }
  return [
    [
      ...reuseObj,
      {
        ...targetObj,
        children: children,
      },
    ],
    [
      ...reuseObj,
      {
        ...targetObj,
        children: [...childrenObj],
      },
    ],
  ];
}

import React, { Fragment } from "react";
import { Outlet } from "react-router";
import { routes } from "../../router/merchant";
import { useMyRoutes } from "../../hooks/route";
import { myRouteObj } from "../../type";
export default function Merchant() {
  const element = useMyRoutes(routes as myRouteObj[]);
  return (
    <Fragment>
      {element}
      <Outlet></Outlet>
    </Fragment>
  );
}