menu.tsx 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. import React from 'react';
  2. import { useNavigate } from 'react-router';
  3. import { useAuth } from './hooks/AuthProvider';
  4. import type { MenuProps } from 'antd';
  5. import {
  6. UserOutlined,
  7. DashboardOutlined,
  8. TeamOutlined,
  9. InfoCircleOutlined,
  10. } from '@ant-design/icons';
  11. export interface MenuItem {
  12. key: string;
  13. label: string;
  14. icon?: React.ReactNode;
  15. children?: MenuItem[];
  16. path?: string;
  17. permission?: string;
  18. }
  19. /**
  20. * 菜单搜索 Hook
  21. * 封装菜单搜索相关逻辑
  22. */
  23. export const useMenuSearch = (menuItems: MenuItem[]) => {
  24. const [searchText, setSearchText] = React.useState('');
  25. // 过滤菜单项
  26. const filteredMenuItems = React.useMemo(() => {
  27. if (!searchText) return menuItems;
  28. const filterItems = (items: MenuItem[]): MenuItem[] => {
  29. return items
  30. .map(item => {
  31. // 克隆对象避免修改原数据
  32. const newItem = { ...item };
  33. if (newItem.children) {
  34. newItem.children = filterItems(newItem.children);
  35. }
  36. return newItem;
  37. })
  38. .filter(item => {
  39. // 保留匹配项或其子项匹配的项
  40. const match = item.label.toLowerCase().includes(searchText.toLowerCase());
  41. if (match) return true;
  42. if (item.children?.length) return true;
  43. return false;
  44. });
  45. };
  46. return filterItems(menuItems);
  47. }, [menuItems, searchText]);
  48. // 清除搜索
  49. const clearSearch = () => {
  50. setSearchText('');
  51. };
  52. return {
  53. searchText,
  54. setSearchText,
  55. filteredMenuItems,
  56. clearSearch
  57. };
  58. };
  59. export const useMenu = () => {
  60. const navigate = useNavigate();
  61. const { logout: handleLogout } = useAuth();
  62. const [collapsed, setCollapsed] = React.useState(false);
  63. const [openKeys, setOpenKeys] = React.useState<string[]>([]);
  64. // 基础菜单项配置
  65. const menuItems: MenuItem[] = [
  66. {
  67. key: 'dashboard',
  68. label: '控制台',
  69. icon: <DashboardOutlined />,
  70. path: '/admin/dashboard'
  71. },
  72. {
  73. key: 'users',
  74. label: '用户管理',
  75. icon: <TeamOutlined />,
  76. path: '/admin/users',
  77. permission: 'user:manage'
  78. },
  79. ];
  80. // 用户菜单项
  81. const userMenuItems: MenuProps['items'] = [
  82. {
  83. key: 'profile',
  84. label: '个人资料',
  85. icon: <UserOutlined />
  86. },
  87. {
  88. key: 'logout',
  89. label: '退出登录',
  90. icon: <InfoCircleOutlined />,
  91. danger: true,
  92. onClick: () => handleLogout()
  93. }
  94. ];
  95. // 处理菜单点击
  96. const handleMenuClick = (item: MenuItem) => {
  97. if (item.path) {
  98. navigate(item.path);
  99. }
  100. };
  101. // 处理菜单展开变化
  102. const onOpenChange = (keys: string[]) => {
  103. const latestOpenKey = keys.find(key => openKeys.indexOf(key) === -1);
  104. setOpenKeys(latestOpenKey ? [latestOpenKey] : []);
  105. };
  106. return {
  107. menuItems,
  108. userMenuItems,
  109. openKeys,
  110. collapsed,
  111. setCollapsed,
  112. handleMenuClick,
  113. onOpenChange
  114. };
  115. };