| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- import React from 'react';
- import { useNavigate } from 'react-router';
- import type { MenuProps } from 'antd';
- import {
- UserOutlined,
- DashboardOutlined,
- TeamOutlined,
- InfoCircleOutlined,
- } from '@ant-design/icons';
- export interface MenuItem {
- key: string;
- label: string;
- icon?: React.ReactNode;
- children?: MenuItem[];
- path?: string;
- permission?: string;
- }
- /**
- * 菜单搜索 Hook
- * 封装菜单搜索相关逻辑
- */
- export const useMenuSearch = (menuItems: MenuItem[]) => {
- const [searchText, setSearchText] = React.useState('');
- // 过滤菜单项
- const filteredMenuItems = React.useMemo(() => {
- if (!searchText) return menuItems;
-
- const filterItems = (items: MenuItem[]): MenuItem[] => {
- return items
- .map(item => {
- // 克隆对象避免修改原数据
- const newItem = { ...item };
- if (newItem.children) {
- newItem.children = filterItems(newItem.children);
- }
- return newItem;
- })
- .filter(item => {
- // 保留匹配项或其子项匹配的项
- const match = item.label.toLowerCase().includes(searchText.toLowerCase());
- if (match) return true;
- if (item.children?.length) return true;
- return false;
- });
- };
-
- return filterItems(menuItems);
- }, [menuItems, searchText]);
- // 清除搜索
- const clearSearch = () => {
- setSearchText('');
- };
- return {
- searchText,
- setSearchText,
- filteredMenuItems,
- clearSearch
- };
- };
- export const useMenu = () => {
- const navigate = useNavigate();
- const [collapsed, setCollapsed] = React.useState(false);
- const [openKeys, setOpenKeys] = React.useState<string[]>([]);
- // 基础菜单项配置
- const menuItems: MenuItem[] = [
- {
- key: 'dashboard',
- label: '控制台',
- icon: <DashboardOutlined />,
- path: '/admin/dashboard'
- },
- {
- key: 'users',
- label: '用户管理',
- icon: <TeamOutlined />,
- path: '/admin/users',
- permission: 'user:manage'
- },
- ];
- // 用户菜单项
- const userMenuItems: MenuProps['items'] = [
- {
- key: 'profile',
- label: '个人资料',
- icon: <UserOutlined />
- },
- {
- key: 'logout',
- label: '退出登录',
- icon: <InfoCircleOutlined />,
- danger: true
- }
- ];
- // 处理菜单点击
- const handleMenuClick = (item: MenuItem) => {
- if (item.path) {
- navigate(item.path);
- }
- };
- // 处理菜单展开变化
- const onOpenChange = (keys: string[]) => {
- const latestOpenKey = keys.find(key => openKeys.indexOf(key) === -1);
- setOpenKeys(latestOpenKey ? [latestOpenKey] : []);
- };
- return {
- menuItems,
- userMenuItems,
- openKeys,
- collapsed,
- setCollapsed,
- handleMenuClick,
- onOpenChange
- };
- };
|