mirror of
https://github.com/ant-design/ant-design.git
synced 2024-11-24 19:19:57 +08:00
117 lines
3.6 KiB
TypeScript
117 lines
3.6 KiB
TypeScript
import classNames from 'classnames';
|
|
import type { MenuItemProps as RcMenuItemProps } from 'rc-menu';
|
|
import { Item } from 'rc-menu';
|
|
import toArray from 'rc-util/lib/Children/toArray';
|
|
import omit from 'rc-util/lib/omit';
|
|
import * as React from 'react';
|
|
import type { SiderContextProps } from '../layout/Sider';
|
|
import { SiderContext } from '../layout/Sider';
|
|
import type { TooltipProps } from '../tooltip';
|
|
import Tooltip from '../tooltip';
|
|
import { cloneElement, isValidElement } from '../_util/reactNode';
|
|
import type { MenuContextProps } from './MenuContext';
|
|
import MenuContext from './MenuContext';
|
|
|
|
export interface MenuItemProps extends Omit<RcMenuItemProps, 'title'> {
|
|
icon?: React.ReactNode;
|
|
danger?: boolean;
|
|
title?: React.ReactNode;
|
|
}
|
|
|
|
type MenuItemComponent = React.FC<MenuItemProps>;
|
|
|
|
type RestArgs<T> = T extends (arg: any, ...args: infer P) => any ? P : never;
|
|
|
|
type GenericProps<T = unknown> = T extends infer U extends MenuItemProps
|
|
? unknown extends U
|
|
? MenuItemProps
|
|
: U
|
|
: MenuItemProps;
|
|
|
|
type GenericComponent = Omit<MenuItemComponent, ''> & {
|
|
<T extends MenuItemProps>(
|
|
props: GenericProps<T>,
|
|
...args: RestArgs<MenuItemComponent>
|
|
): ReturnType<MenuItemComponent>;
|
|
};
|
|
|
|
const MenuItem: GenericComponent = (props) => {
|
|
const { className, children, icon, title, danger } = props;
|
|
const {
|
|
prefixCls,
|
|
firstLevel,
|
|
direction,
|
|
disableMenuItemTitleTooltip,
|
|
inlineCollapsed: isInlineCollapsed,
|
|
} = React.useContext<MenuContextProps>(MenuContext);
|
|
const renderItemChildren = (inlineCollapsed: boolean) => {
|
|
const wrapNode = <span className={`${prefixCls}-title-content`}>{children}</span>;
|
|
// inline-collapsed.md demo 依赖 span 来隐藏文字,有 icon 属性，则内部包裹一个 span
|
|
// ref: https://github.com/ant-design/ant-design/pull/23456
|
|
if (!icon || (isValidElement(children) && children.type === 'span')) {
|
|
if (children && inlineCollapsed && firstLevel && typeof children === 'string') {
|
|
return <div className={`${prefixCls}-inline-collapsed-noicon`}>{children.charAt(0)}</div>;
|
|
}
|
|
}
|
|
return wrapNode;
|
|
};
|
|
|
|
const { siderCollapsed } = React.useContext<SiderContextProps>(SiderContext);
|
|
|
|
let tooltipTitle = title;
|
|
|
|
if (typeof title === 'undefined') {
|
|
tooltipTitle = firstLevel ? children : '';
|
|
} else if (title === false) {
|
|
tooltipTitle = '';
|
|
}
|
|
|
|
const tooltipProps: TooltipProps = { title: tooltipTitle };
|
|
|
|
if (!siderCollapsed && !isInlineCollapsed) {
|
|
tooltipProps.title = null;
|
|
// Reset `open` to fix control mode tooltip display not correct
|
|
// ref: https://github.com/ant-design/ant-design/issues/16742
|
|
tooltipProps.open = false;
|
|
}
|
|
|
|
const childrenLength = toArray(children).length;
|
|
|
|
let returnNode = (
|
|
<Item
|
|
{...omit(props, ['title', 'icon', 'danger'])}
|
|
className={classNames(
|
|
{
|
|
[`${prefixCls}-item-danger`]: danger,
|
|
[`${prefixCls}-item-only-child`]: (icon ? childrenLength + 1 : childrenLength) === 1,
|
|
},
|
|
className,
|
|
)}
|
|
title={typeof title === 'string' ? title : undefined}
|
|
>
|
|
{cloneElement(icon, {
|
|
className: classNames(
|
|
isValidElement(icon) ? icon.props?.className : '',
|
|
`${prefixCls}-item-icon`,
|
|
),
|
|
})}
|
|
{renderItemChildren(isInlineCollapsed)}
|
|
</Item>
|
|
);
|
|
|
|
if (!disableMenuItemTitleTooltip) {
|
|
returnNode = (
|
|
<Tooltip
|
|
{...tooltipProps}
|
|
placement={direction === 'rtl' ? 'left' : 'right'}
|
|
overlayClassName={`${prefixCls}-inline-collapsed-tooltip`}
|
|
>
|
|
{returnNode}
|
|
</Tooltip>
|
|
);
|
|
}
|
|
return returnNode;
|
|
};
|
|
|
|
export default MenuItem;
|