import React, { PureComponent } from 'react'; import { Menu, Icon } from 'antd'; import { Link } from 'dva/router'; import pathToRegexp from 'path-to-regexp'; import { urlToList } from '../_utils/pathTools'; import styles from './index.less'; const { SubMenu } = Menu; // Allow menu.js config icon as string or ReactNode // icon: 'setting', // icon: 'http://demo.com/icon.png', // icon: , const getIcon = icon => { if (typeof icon === 'string' && icon.indexOf('http') === 0) { return icon; } if (typeof icon === 'string') { return ; } return icon; }; export const getMenuMatches = (flatMenuKeys, path) => { return flatMenuKeys.filter(item => { return pathToRegexp(item).test(path); }); }; export default class BaseMenu extends PureComponent { constructor(props) { super(props); this.flatMenuKeys = this.getFlatMenuKeys(props.menuData); } /** * Recursively flatten the data * [{path:string},{path:string}] => {path,path2} * @param menus */ getFlatMenuKeys(menus) { let keys = []; menus.forEach(item => { if (item.children) { keys = keys.concat(this.getFlatMenuKeys(item.children)); } keys.push(item.path); }); return keys; } /** * 获得菜单子节点 * @memberof SiderMenu */ getNavMenuItems = menusData => { if (!menusData) { return []; } return menusData .filter(item => item.name && !item.hideInMenu) .map(item => { // make dom const ItemDom = this.getSubMenuOrItem(item); return this.checkPermissionItem(item.authority, ItemDom); }) .filter(item => item); }; // Get the currently selected menu getSelectedMenuKeys = () => { const { location: { pathname }, } = this.props; return urlToList(pathname).map(itemPath => getMenuMatches(this.flatMenuKeys, itemPath).pop()); }; /** * get SubMenu or Item */ getSubMenuOrItem = item => { if (item.children && item.children.some(child => child.name)) { return ( {getIcon(item.icon)} {item.name} ) : ( item.name ) } key={item.path} > {this.getNavMenuItems(item.children)} ); } else { return {this.getMenuItemPath(item)}; } }; /** * 判断是否是http链接.返回 Link 或 a * Judge whether it is http link.return a or Link * @memberof SiderMenu */ getMenuItemPath = item => { const itemPath = this.conversionPath(item.path); const icon = getIcon(item.icon); const { target, name } = item; // Is it a http link if (/^https?:\/\//.test(itemPath)) { return ( {icon} {name} ); } const { location, isMobile, onCollapse } = this.props; return ( { onCollapse(true); } : undefined } > {icon} {name} ); }; // permission to check checkPermissionItem = (authority, ItemDom) => { const { Authorized } = this.props; if (Authorized && Authorized.check) { const { check } = Authorized; return check(authority, ItemDom); } return ItemDom; }; conversionPath = path => { if (path && path.indexOf('http') === 0) { return path; } else { return `/${path || ''}`.replace(/\/+/g, '/'); } }; render() { const { openKeys, theme, mode } = this.props; // if pathname can't match, use the nearest parent's key let selectedKeys = this.getSelectedMenuKeys(); if (!selectedKeys.length && openKeys) { selectedKeys = [openKeys[openKeys.length - 1]]; } let props = {}; if (openKeys) { props = { openKeys, }; } const { handleOpenChange, style, menuData } = this.props; return ( {this.getNavMenuItems(menuData)} ); } }