import RightOutlined from '@ant-design/icons/RightOutlined'; import classNames from 'classnames'; import RcDropdown from 'rc-dropdown'; import useEvent from 'rc-util/lib/hooks/useEvent'; import useMergedState from 'rc-util/lib/hooks/useMergedState'; import * as React from 'react'; import { ConfigContext } from '../config-provider'; import { OverrideProvider } from '../menu/OverrideContext'; import getPlacements from '../_util/placements'; import { cloneElement } from '../_util/reactNode'; import { tuple } from '../_util/type'; import warning from '../_util/warning'; import DropdownButton from './dropdown-button'; const Placements = tuple( 'topLeft', 'topCenter', 'topRight', 'bottomLeft', 'bottomCenter', 'bottomRight', 'top', 'bottom', ); type Placement = typeof Placements[number]; type OverlayFunc = () => React.ReactElement; type Align = { points?: [string, string]; offset?: [number, number]; targetOffset?: [number, number]; overflow?: { adjustX?: boolean; adjustY?: boolean; }; useCssRight?: boolean; useCssBottom?: boolean; useCssTransform?: boolean; }; export type DropdownArrowOptions = { pointAtCenter?: boolean; }; export interface DropdownProps { autoFocus?: boolean; arrow?: boolean | DropdownArrowOptions; trigger?: ('click' | 'hover' | 'contextMenu')[]; overlay: React.ReactElement | OverlayFunc; /** * @deprecated `onVisibleChange` is deprecated which will be removed in next major version. Please * use `onOpenChange` instead. */ onVisibleChange?: (visible: boolean) => void; onOpenChange?: (open: boolean) => void; /** * @deprecated `visible` is deprecated which will be removed in next major version. Please use * `open` instead. */ visible?: boolean; open?: boolean; disabled?: boolean; destroyPopupOnHide?: boolean; align?: Align; getPopupContainer?: (triggerNode: HTMLElement) => HTMLElement; prefixCls?: string; className?: string; transitionName?: string; placement?: Placement; overlayClassName?: string; overlayStyle?: React.CSSProperties; forceRender?: boolean; mouseEnterDelay?: number; mouseLeaveDelay?: number; openClassName?: string; children?: React.ReactNode; } interface DropdownInterface extends React.FC { Button: typeof DropdownButton; } const Dropdown: DropdownInterface = props => { const { getPopupContainer: getContextPopupContainer, getPrefixCls, direction, } = React.useContext(ConfigContext); // Warning for deprecated usage if (process.env.NODE_ENV !== 'production') { [ ['visible', 'open'], ['onVisibleChange', 'onOpenChange'], ].forEach(([deprecatedName, newName]) => { warning( !(deprecatedName in props), 'Dropdown', `\`${deprecatedName}\` is deprecated which will be removed in next major version, please use \`${newName}\` instead.`, ); }); } const getTransitionName = () => { const rootPrefixCls = getPrefixCls(); const { placement = '', transitionName } = props; if (transitionName !== undefined) { return transitionName; } if (placement.includes('top')) { return `${rootPrefixCls}-slide-down`; } return `${rootPrefixCls}-slide-up`; }; const getPlacement = () => { const { placement } = props; if (!placement) { return direction === 'rtl' ? 'bottomRight' : 'bottomLeft'; } if (placement.includes('Center')) { const newPlacement = placement.slice(0, placement.indexOf('Center')); warning( !placement.includes('Center'), 'Dropdown', `You are using '${placement}' placement in Dropdown, which is deprecated. Try to use '${newPlacement}' instead.`, ); return newPlacement; } return placement; }; const { arrow, prefixCls: customizePrefixCls, children, trigger, disabled, getPopupContainer, overlayClassName, visible, open, onVisibleChange, onOpenChange, mouseEnterDelay = 0.15, mouseLeaveDelay = 0.1, } = props; const prefixCls = getPrefixCls('dropdown', customizePrefixCls); const child = React.Children.only(children) as React.ReactElement; const dropdownTrigger = cloneElement(child, { className: classNames( `${prefixCls}-trigger`, { [`${prefixCls}-rtl`]: direction === 'rtl', }, child.props.className, ), disabled, }); const triggerActions = disabled ? [] : trigger; let alignPoint: boolean; if (triggerActions && triggerActions.includes('contextMenu')) { alignPoint = true; } // =========================== Visible ============================ const [mergedOpen, setOpen] = useMergedState(false, { value: open !== undefined ? open : visible, }); const onInnerOpenChange = useEvent((nextOpen: boolean) => { onVisibleChange?.(nextOpen); onOpenChange?.(nextOpen); setOpen(nextOpen); }); // =========================== Overlay ============================ const overlayClassNameCustomized = classNames(overlayClassName, { [`${prefixCls}-rtl`]: direction === 'rtl', }); const builtinPlacements = getPlacements({ arrowPointAtCenter: typeof arrow === 'object' && arrow.pointAtCenter, autoAdjustOverflow: true, }); const onMenuClick = React.useCallback(() => { setOpen(false); }, []); const renderOverlay = () => { // rc-dropdown already can process the function of overlay, but we have check logic here. // So we need render the element to check and pass back to rc-dropdown. const { overlay } = props; let overlayNode: React.ReactNode; if (typeof overlay === 'function') { overlayNode = overlay(); } else { overlayNode = overlay; } overlayNode = React.Children.only( typeof overlayNode === 'string' ? {overlayNode} : overlayNode, ); return ( } mode="vertical" selectable={false} onClick={onMenuClick} validator={({ mode }) => { // Warning if use other mode warning( !mode || mode === 'vertical', 'Dropdown', `mode="${mode}" is not supported for Dropdown's Menu.`, ); }} > {overlayNode} ); }; // ============================ Render ============================ return ( {dropdownTrigger} ); }; Dropdown.Button = DropdownButton; export default Dropdown;