使用React Router实现前端的权限访问控制

前段时间学习了React Router,发现没有Vue里面的路由功能强大,没有直接提供路由中间件,不能像Vue里面一样在路由配置上设置任意的额外属性,但是可以通过一些技巧来实现这些功能。

1、配置菜单

后台管理系统一般都会在左侧显示菜单,右侧显示页面,本例中使用Ant Design组件当然也不例外。虽然umi里面已经集成了很多功能,但是有些地方用起来不够灵活,比如路由配置高阶组件,不能传递prop;每一个权限码都要配置相同的函数,等等。所以,我更喜欢用Vite+React来搭建项目。
废话不多说,菜单配置的代码如下

javascript 复制代码
export type MenuConfig = {
  computedMatch?: match<any>;
  route?: MenuDataItem
  location: {
    pathname?: string;
  };
}

export type MenuDataItem = {
  /** @name 子菜单 */
  children?: MenuDataItem[];
  routes?: MenuDataItem[];
  /** @name 在菜单中隐藏子节点 */
  hideChildrenInMenu?: boolean;
  /** @name 在菜单中隐藏自己和子节点 */
  hideInMenu?: boolean;
  /** @name 菜单的icon */
  icon?: React.ReactNode;
  /** @name 自定义菜单的国际化 key */
  locale?: string | false;
  /** @name 菜单的名字 */
  name?: string;
  /** @name 用于标定选中的值,默认是 path */
  key?: string;
  /** @name disable 菜单选项 */
  disabled?: boolean;
  /** @name disable menu 的 tooltip 菜单选项 */
  disabledTooltip?: boolean;
  /** @name 路径,可以设定为网页链接 */
  path?: string;
  /**
   * 当此节点被选中的时候也会选中 parentKeys 的节点
   *
   * @name 自定义父节点
   */
  parentKeys?: string[];
  /** @name 隐藏自己,并且将子节点提升到与自己平级 */
  flatMenu?: boolean;
  /** @name 指定外链打开形式,同a标签 */
  target?: string;
  /**
   * menuItem 的 tooltip 显示的路径
   */
  tooltip?: string;
  /**
   * 组件
   */
  component?: Promise<{ default: React.ComponentType<any> }>;
  /**
   * 权限码
   */
  access?: string;

}

const menuConfig: MenuConfig = {
  route: {
    path: '/',
    routes: [
      {
        key: '1',
        name: '首页',
        path: '/home',
        icon: <HomeFilled />,
        component: import('@/pages/home')
      },
      {
        name: '系统管理',
        path: '/system',
        access: 'system:view',
        icon: <SettingFilled />,
        routes: [
          {
            name: '用户管理',
            path: '/system/user',
            icon: <ContactsFilled />,
            access: 'user:view',
            component: import('@/pages/system/user')
          },
          {
            name: '角色管理',
            path: '/system/role',
            icon: <SmileFilled />,
            access: 'role:view',
            component: import('@/pages/system/role')
          },
          {
            name: '权限管理',
            path: '/system/authority',
            access: 'ahthority:view',
            routes: [
              {
                name: '菜单按钮管理',
                path: '/system/authority/menu',
                icon: <GoldenFilled />,
                access: 'menuBtn:view',
                component: import('@/pages/system/authority/menuBtn')
              },
              {
                name: '接口权限管理',
                path: '/system/authority/interface',
                icon: <SecurityScanFilled />,
                access: 'interface:view',
                component: import('@/pages/system/authority/interface')
              }
            ]
          }
        ]
      }
    ],
  },
  location: {
    pathname: '/',
  }
}

export default menuConfig

这里使用import()函数,动态导入组件,避免将来路由组件多了以后,在开头写大量的import语句,access表示权限码,用来控制菜单的隐藏和显示,并且可以将权限码传递给路由,设置路由的访问权限,后面会说到。

2、定义用户的全局状态

javascript 复制代码
export type UserInfo = {
    userName?: string | null,
    avatar?: string | null,
    authCodes?: Set<string> | null,
}

const userInfo: UserInfo = {
    userName: '',
    avatar: '',
    authCodes: undefined,
}

const user = {
    data: userInfo,
    async requestUserInfo() {
        const userInfoFromDB = await getUserInfo()
        this.data = { ...userInfoFromDB, authCodes: new Set(userInfoFromDB.menuBtnCodes) }
        localStorage.setItem(USER_INFO,JSON.stringify(userInfoFromDB))
    }
}

export default {
    user,
    tab,
    menu
}

这里将用户的用户名、头像、权限码,保存到了全局变量当中,其中authCodes代表权限码的Set集合,里面包含了菜单和按钮的权限码,方便后面进行校验,requestUserInfo函数用来请求后台接口,获取用户信息,并保存到全局变量和本地缓存当中,然后将user对象导出,tab和menu涉及到其他的功能,这里先不讨论。

3、获取用户信息

为主页所在的路由组件定义一个函数,作为loader,并在loader函数里面获取用户信息

javascript 复制代码
export const loader = async ({request}:LoaderFunctionArgs) => {
  const url=new URL(request.url)
  if (url.pathname==import.meta.env.VITE_BASE_NAME) {
    return redirect('/home')
  }
  if (currentAction==Action.INIT) {
    await store.user.requestUserInfo()
    store.menu.filterMenuConfig()
  }else{
    currentAction=Action.INIT
  }
  return { userInfo: store.user.data, tabsData: store.tab.data,menuConfig:store.menu.data }
}

其中,store对象保存了当前的全局状态,调用requestUserInfo()函数请求后台获取用户信息,并保存,然后需要将store对象里面的数据返回。代码中的其他逻辑涉及其他功能,这里先不讨论。

然后,使用useLoaderData()函数,在主页的路由组件里面获取到这些信息即可,后面可以进行显示和调用。

pro components组件库有很多高级组件,只要调用ProLayout组件,将loader获取的数据传入对应的prop即可,由于代码量庞大,这里不展开讨论。

javascript 复制代码
const loaderData = useLoaderData() as { userInfo: UserInfo, tabsData: TabsData,menuConfig:MenuConfig}
const {userInfo,tabsData,menuConfig}=loaderData

4、定义权限校验函数

javascript 复制代码
/**
 * 检查权限
 * @param access 权限码
 * @returns true 有权限
 *          false 没有权限
 */
export function checkAuth(access?:string){
    if (access) {
        const authCodes=store.user.data.authCodes
        if (!authCodes) {
            const userStr=localStorage.getItem(USER_INFO)
            if (userStr) {
                const userInfo:SYSTEM_API.UserInfo =JSON.parse(userStr)
                const menuCodes=new Set(userInfo.menuBtnCodes)
                return checkAccess(access,menuCodes)
            }else{
                store.user.requestUserInfo().then(() => {
                    const menuCodes=store.user.data.authCodes
                    checkAccess(access,menuCodes)
                })
            }
        }else{
            return checkAccess(access,authCodes)
        }
    }else{
        return true
    }
}

function checkAccess(access:string,authCodes?:Set<string>|null){
    if (authCodes?.has(access)) {
        return true
    }else{
        return false
    }
}

checkAuth是一个权限校验的函数,首先从全局状态当中获取用户权限码,如果为空,就从本地缓存中获取,如果本地缓存为空,就请求后台去获取,然后判断权限码的Set集合里面是否包含当前所需权限,返回true代表验证通过,返回false代表没有权限。

5、生成路由

javascript 复制代码
let key=1

const createRoutes = (menus: MenuDataItem[] | undefined) => {
    if (menus) {
        const routes: RouteObject[] = []
        for (const menu of menus) {
            if (menu.path) {
                const route: RouteObject = {path: menu.path}
                if (menu.component) {
                    const module = menu.component
                    const Component = React.lazy(() => module)
                    route.element=(
                        <Suspense fallback={<ProSkeleton type="list"></ProSkeleton>}>
                            <MiddleWare tabKey={String(key)} title={menu.name} path={menu.path}><Component /></MiddleWare>
                        </Suspense>
                    )
                    menu.key=String(key)
                    key++
                }else if (!menu.routes && !menu.children) {
                    route.element=(
                        <MiddleWare tabKey={String(key)} title={menu.name} path={menu.path}></MiddleWare>
                    )
                    menu.key=String(key)
                    key++
                }
                const children = createRoutes(menu.routes || menu.children)
                if (children) {
                    route.children = children
                }
                route.loader=() => {
                    if (!checkAuth(menu.access)) {
                        throw new Response('Forbidden',{status:403})
                    }
                    return null
                }
                routes.push(route)
            }
        }
        return routes
    }
}

const menus = menuConfig.route?.routes

const routes = createRoutes(menus);

export {menuConfig}

const router = createBrowserRouter([
    {
        path: '/',
        element: <Main />,
        loader: mainLoader,
        action:mainAction,
        errorElement:<ErrorBoundary/>,
        children: [
            ...(routes || [])
        ]
    },
    {
        path: '/login',
        element: <Login />,
        errorElement:<ErrorBoundary/>
    },
], {
    basename: import.meta.env.VITE_BASE_NAME
})

export default router

这里面的逻辑比较复杂。

createRoutes是一个递归函数,用来循环递归遍历菜单,通过调用React.lazy()函数得到菜单中的组件对象,使用Suspense组件进行包裹才能正常显示。MiddleWare是我自定义的一个高阶组件,用来获取菜单信息,控制tab页的显示状态,这里不展开讨论。这里还为菜单对应的路由创建了loader,在loader函数里面调用前面定义的checkAuth,判断是否有权限访问对应的路由,如果没有权限,就抛出异常,显示错误页,也就是403页面。

在createBrowserRouter函数里面配置了主页和登录页的错误页,并将前面定义的loader函数在主页的路由配置当中进行配置,用于获取用户信息,将菜单生成的路由在主页的children配置中展开。

当然,也要在main.tsx中调用路由对象,才能使它生效

javascript 复制代码
createRoot(document.getElementById('root')!).render(
  <StrictMode>
    <RouterProvider router={router}/>
  </StrictMode>,
)

6、菜单权限校验

javascript 复制代码
//检查菜单的权限,并做一次深拷贝,得到新的对象
function filterMenuConfig(){
    const menus=menuConfig.route?.routes
    if (menus) {
        const menuConfigCopy:MenuConfig={
            route:{
                path:'/',
                routes:filterMenus(menus)
            },
            location:{
                pathname:'/'
            }
        }
        menu.data=menuConfigCopy
    }
}

function filterMenus(menus: MenuDataItem[]) {
    const menusCopy:MenuDataItem[]=[]
    for (const menu of menus) {
        const menuCopy={...menu}
        menusCopy.push(menuCopy)
        if (!checkAuth(menuCopy.access)) {
            menuCopy.hideInMenu = true
        }
        const children = menuCopy.routes || menuCopy.children
        if (children) {
            menuCopy.routes=filterMenus(children)
        }
    }
    return menusCopy
}

const menu = {
    data: menuConfig,
    filterMenuConfig
}

export default {
    user,
    tab,
    menu
}

这里定义了filterMenuConfig函数,对菜单的配置对象做了一次深拷贝,通过循环遍历和递归拷贝了里面的每一个对象,并且在这过程中调用前面定义的checkAuth,来检查每个菜单的权限,如果没有权限,就隐藏对应的菜单。之所以要做深拷贝,就是为了不破坏原先菜单配置里面的数据,方面用户退出的时候恢复菜单数据。

然后,可以在主页的loader里面调用filterMenuConfig函数,代码如第3步所示。

7、按钮权限校验

javascript 复制代码
export function Access({children,auth}:{
    children?:ReactNode
    auth?:string
}){
    if (checkAuth(auth)) {
        return (
            <>{children}</>
        )
    }
}

这里定义了一个高阶组件,用于对按钮的权限进行校验,组件内调用了前面定义的checkAuth函数,如果用户没有权限,就不会显示对应的按钮。

调用示例如下

javascript 复制代码
<Access auth="user:save">
    <Button type="primary" icon={<PlusCircleOutlined />} onClick={() => {
        dialogRef.current?.openDialog('新增用户')
    }
    }>新增</Button>
</Access>

其中,user:save代码按钮的权限码,只有拥有这个权限的用户才能看到这个按钮。

相关推荐
花海少爷几秒前
第十章 JavaScript的应用课后习题
开发语言·javascript·ecmascript
Amd7944 分钟前
Nuxt.js 应用中的 webpack:compiled 事件钩子
前端·webpack·开发·编译·nuxt.js·事件·钩子
生椰拿铁You13 分钟前
09 —— Webpack搭建开发环境
前端·webpack·node.js
狸克先生24 分钟前
如何用AI写小说(二):Gradio 超简单的网页前端交互
前端·人工智能·chatgpt·交互
sinat_3842410927 分钟前
在有网络连接的机器上打包 electron 及其依赖项,在没有网络连接的机器上安装这些离线包
javascript·arcgis·electron
baiduopenmap39 分钟前
百度世界2024精选公开课:基于地图智能体的导航出行AI应用创新实践
前端·人工智能·百度地图
loooseFish1 小时前
小程序webview我爱死你了 小程序webview和H5通讯
前端
小牛itbull1 小时前
ReactPress vs VuePress vs WordPress
开发语言·javascript·reactpress
菜牙买菜1 小时前
让安卓也能玩出Element-Plus的表格效果
前端
请叫我欧皇i1 小时前
html本地离线引入vant和vue2(详细步骤)
开发语言·前端·javascript