ant-design/components/tooltip/index.tsx
lijianan d840f6a87a
🦄️ refactor: use React.useMemo replace function (#41533)
* 🦄️refactor: use React.useMemo replace function

* rename
2023-03-29 20:30:50 +08:00

428 lines
13 KiB
TypeScript

import type { BuildInPlacements } from '@rc-component/trigger';
import classNames from 'classnames';
import RcTooltip from 'rc-tooltip';
import type {
TooltipProps as RcTooltipProps,
TooltipRef as RcTooltipRef,
} from 'rc-tooltip/lib/Tooltip';
import type { placements as Placements } from 'rc-tooltip/lib/placements';
import type { AlignType } from 'rc-trigger/lib/interface';
import useMergedState from 'rc-util/lib/hooks/useMergedState';
import type { CSSProperties } from 'react';
import * as React from 'react';
import type { PresetColorType } from '../_util/colors';
import { getTransitionName } from '../_util/motion';
import type { AdjustOverflow, PlacementsConfig } from '../_util/placements';
import getPlacements from '../_util/placements';
import { cloneElement, isFragment, isValidElement } from '../_util/reactNode';
import type { LiteralUnion } from '../_util/type';
import warning from '../_util/warning';
import { ConfigContext } from '../config-provider';
import theme from '../theme';
import PurePanel from './PurePanel';
import useStyle from './style';
import { parseColor } from './util';
const { useToken } = theme;
export type { AdjustOverflow, PlacementsConfig };
export interface TooltipRef {
/** @deprecated Please use `forceAlign` instead */
forcePopupAlign: VoidFunction;
forceAlign: VoidFunction;
}
export type TooltipPlacement =
| 'top'
| 'left'
| 'right'
| 'bottom'
| 'topLeft'
| 'topRight'
| 'bottomLeft'
| 'bottomRight'
| 'leftTop'
| 'leftBottom'
| 'rightTop'
| 'rightBottom';
// https://github.com/react-component/tooltip
// https://github.com/yiminghe/dom-align
export interface TooltipAlignConfig {
points?: [string, string];
offset?: [number | string, number | string];
targetOffset?: [number | string, number | string];
overflow?: { adjustX: boolean; adjustY: boolean };
useCssRight?: boolean;
useCssBottom?: boolean;
useCssTransform?: boolean;
}
// remove this after RcTooltip switch visible to open.
interface LegacyTooltipProps
extends Partial<
Omit<
RcTooltipProps,
| 'children'
| 'visible'
| 'defaultVisible'
| 'onVisibleChange'
| 'afterVisibleChange'
| 'destroyTooltipOnHide'
>
> {
open?: RcTooltipProps['visible'];
defaultOpen?: RcTooltipProps['defaultVisible'];
onOpenChange?: RcTooltipProps['onVisibleChange'];
afterOpenChange?: RcTooltipProps['afterVisibleChange'];
// Legacy
/** @deprecated Please use `open` instead. */
visible?: RcTooltipProps['visible'];
/** @deprecated Please use `defaultOpen` instead. */
defaultVisible?: RcTooltipProps['defaultVisible'];
/** @deprecated Please use `onOpenChange` instead. */
onVisibleChange?: RcTooltipProps['onVisibleChange'];
/** @deprecated Please use `afterOpenChange` instead. */
afterVisibleChange?: RcTooltipProps['afterVisibleChange'];
}
export interface AbstractTooltipProps extends LegacyTooltipProps {
style?: React.CSSProperties;
className?: string;
rootClassName?: string;
color?: LiteralUnion<PresetColorType>;
placement?: TooltipPlacement;
builtinPlacements?: typeof Placements;
openClassName?: string;
/** @deprecated Please use `arrow` instead. */
arrowPointAtCenter?: boolean;
arrow?:
| boolean
| {
/** @deprecated Please use `pointAtCenter` instead. */
arrowPointAtCenter?: boolean;
pointAtCenter?: boolean;
};
autoAdjustOverflow?: boolean | AdjustOverflow;
getPopupContainer?: (triggerNode: HTMLElement) => HTMLElement;
children?: React.ReactNode;
destroyTooltipOnHide?: boolean | { keepParent?: boolean };
}
export type RenderFunction = () => React.ReactNode;
export interface TooltipPropsWithOverlay extends AbstractTooltipProps {
title?: React.ReactNode | RenderFunction;
overlay?: React.ReactNode | RenderFunction;
}
export interface TooltipPropsWithTitle extends AbstractTooltipProps {
title: React.ReactNode | RenderFunction;
overlay?: React.ReactNode | RenderFunction;
}
export declare type TooltipProps = TooltipPropsWithTitle | TooltipPropsWithOverlay;
const splitObject = <T extends CSSProperties>(
obj: T,
keys: (keyof T)[],
): Record<'picked' | 'omitted', T> => {
const picked: T = {} as T;
const omitted: T = { ...obj };
keys.forEach((key) => {
if (obj && key in obj) {
picked[key] = obj[key];
delete omitted[key];
}
});
return { picked, omitted };
};
// Fix Tooltip won't hide at disabled button
// mouse events don't trigger at disabled button in Chrome
// https://github.com/react-component/tooltip/issues/18
function getDisabledCompatibleChildren(element: React.ReactElement<any>, prefixCls: string) {
const elementType = element.type as any;
if (
((elementType.__ANT_BUTTON === true || element.type === 'button') && element.props.disabled) ||
(elementType.__ANT_SWITCH === true && (element.props.disabled || element.props.loading)) ||
(elementType.__ANT_RADIO === true && element.props.disabled)
) {
// Pick some layout related style properties up to span
// Prevent layout bugs like https://github.com/ant-design/ant-design/issues/5254
const { picked, omitted } = splitObject(element.props.style, [
'position',
'left',
'right',
'top',
'bottom',
'float',
'display',
'zIndex',
]);
const spanStyle: React.CSSProperties = {
display: 'inline-block', // default inline-block is important
...picked,
cursor: 'not-allowed',
width: element.props.block ? '100%' : undefined,
};
const buttonStyle: React.CSSProperties = {
...omitted,
pointerEvents: 'none',
};
const child = cloneElement(element, {
style: buttonStyle,
className: null,
});
return (
<span
style={spanStyle}
className={classNames(element.props.className, `${prefixCls}-disabled-compatible-wrapper`)}
>
{child}
</span>
);
}
return element;
}
const Tooltip = React.forwardRef<TooltipRef, TooltipProps>((props, ref) => {
const {
prefixCls: customizePrefixCls,
openClassName,
getTooltipContainer,
overlayClassName,
color,
overlayInnerStyle,
children,
afterOpenChange,
afterVisibleChange,
destroyTooltipOnHide,
arrow = true,
title,
overlay,
builtinPlacements,
arrowPointAtCenter = false,
autoAdjustOverflow = true,
} = props;
const mergedShowArrow = !!arrow;
const { token } = useToken();
const {
getPopupContainer: getContextPopupContainer,
getPrefixCls,
direction,
} = React.useContext(ConfigContext);
// ============================== Ref ===============================
const tooltipRef = React.useRef<RcTooltipRef>(null);
const forceAlign = () => {
tooltipRef.current?.forceAlign();
};
React.useImperativeHandle(ref, () => ({
forceAlign,
forcePopupAlign: () => {
warning(false, 'Tooltip', '`forcePopupAlign` is align to `forceAlign` instead.');
forceAlign();
},
}));
// ============================== Warn ==============================
if (process.env.NODE_ENV !== 'production') {
[
['visible', 'open'],
['defaultVisible', 'defaultOpen'],
['onVisibleChange', 'onOpenChange'],
['afterVisibleChange', 'afterOpenChange'],
['arrowPointAtCenter', 'arrow'],
].forEach(([deprecatedName, newName]) => {
warning(
!(deprecatedName in props),
'Tooltip',
`\`${deprecatedName}\` is deprecated, please use \`${newName}\` instead.`,
);
});
warning(
!destroyTooltipOnHide || typeof destroyTooltipOnHide === 'boolean',
'Tooltip',
'`destroyTooltipOnHide` no need config `keepParent` anymore. Please use `boolean` value directly.',
);
warning(
!arrow || typeof arrow === 'boolean' || !('arrowPointAtCenter' in arrow),
'Tooltip',
'`arrowPointAtCenter` in `arrow` is deprecated, please use `pointAtCenter` instead.',
);
}
// ============================== Open ==============================
const [open, setOpen] = useMergedState(false, {
value: props.open ?? props.visible,
defaultValue: props.defaultOpen ?? props.defaultVisible,
});
const noTitle = !title && !overlay && title !== 0; // overlay for old version compatibility
const onOpenChange = (vis: boolean) => {
setOpen(noTitle ? false : vis);
if (!noTitle) {
props.onOpenChange?.(vis);
props.onVisibleChange?.(vis);
}
};
const tooltipPlacements = React.useMemo<BuildInPlacements>(() => {
let mergedArrowPointAtCenter = arrowPointAtCenter;
if (typeof arrow === 'object') {
mergedArrowPointAtCenter =
arrow.pointAtCenter ?? arrow.arrowPointAtCenter ?? arrowPointAtCenter;
}
return (
builtinPlacements ||
getPlacements({
arrowPointAtCenter: mergedArrowPointAtCenter,
autoAdjustOverflow,
arrowWidth: mergedShowArrow ? token.sizePopupArrow : 0,
borderRadius: token.borderRadius,
offset: token.marginXXS,
})
);
}, [arrowPointAtCenter, arrow, builtinPlacements, token]);
// 动态设置动画点
const onPopupAlign = (domNode: HTMLElement, align: AlignType) => {
// 当前返回的位置
const placement = Object.keys(tooltipPlacements).find(
(key) =>
tooltipPlacements[key].points![0] === align.points?.[0] &&
tooltipPlacements[key].points![1] === align.points?.[1],
);
if (placement) {
// 根据当前坐标设置动画点
const rect = domNode.getBoundingClientRect();
const transformOrigin: React.CSSProperties = { top: '50%', left: '50%' };
if (/top|Bottom/.test(placement)) {
transformOrigin.top = `${rect.height - align.offset![1]}px`;
} else if (/Top|bottom/.test(placement)) {
transformOrigin.top = `${-align.offset![1]}px`;
}
if (/left|Right/.test(placement)) {
transformOrigin.left = `${rect.width - align.offset![0]}px`;
} else if (/right|Left/.test(placement)) {
transformOrigin.left = `${-align.offset![0]}px`;
}
domNode.style.transformOrigin = `${transformOrigin.left} ${transformOrigin.top}`;
}
};
const memoOverlay = React.useMemo<TooltipProps['overlay']>(() => {
if (title === 0) {
return title;
}
return overlay || title || '';
}, [overlay, title]);
const {
getPopupContainer,
placement = 'top',
mouseEnterDelay = 0.1,
mouseLeaveDelay = 0.1,
overlayStyle,
rootClassName,
...otherProps
} = props;
const prefixCls = getPrefixCls('tooltip', customizePrefixCls);
const rootPrefixCls = getPrefixCls();
const injectFromPopover = (props as any)['data-popover-inject'];
let tempOpen = open;
// Hide tooltip when there is no title
if (!('open' in props) && !('visible' in props) && noTitle) {
tempOpen = false;
}
// ============================= Render =============================
const child = getDisabledCompatibleChildren(
isValidElement(children) && !isFragment(children) ? children : <span>{children}</span>,
prefixCls,
);
const childProps = child.props;
const childCls =
!childProps.className || typeof childProps.className === 'string'
? classNames(childProps.className, {
[openClassName || `${prefixCls}-open`]: true,
})
: childProps.className;
// Style
const [wrapSSR, hashId] = useStyle(prefixCls, !injectFromPopover);
// Color
const colorInfo = parseColor(prefixCls, color);
const formattedOverlayInnerStyle = { ...overlayInnerStyle, ...colorInfo.overlayStyle };
const arrowContentStyle = colorInfo.arrowStyle;
const customOverlayClassName = classNames(
overlayClassName,
{
[`${prefixCls}-rtl`]: direction === 'rtl',
},
colorInfo.className,
rootClassName,
hashId,
);
return wrapSSR(
<RcTooltip
{...otherProps}
showArrow={mergedShowArrow}
placement={placement}
mouseEnterDelay={mouseEnterDelay}
mouseLeaveDelay={mouseLeaveDelay}
prefixCls={prefixCls}
overlayClassName={customOverlayClassName}
overlayStyle={{ ...arrowContentStyle, ...overlayStyle }}
getTooltipContainer={getPopupContainer || getTooltipContainer || getContextPopupContainer}
ref={tooltipRef}
builtinPlacements={tooltipPlacements}
overlay={memoOverlay}
visible={tempOpen}
onVisibleChange={onOpenChange}
afterVisibleChange={afterOpenChange ?? afterVisibleChange}
onPopupAlign={onPopupAlign}
overlayInnerStyle={formattedOverlayInnerStyle}
arrowContent={<span className={`${prefixCls}-arrow-content`} />}
motion={{
motionName: getTransitionName(rootPrefixCls, 'zoom-big-fast', props.transitionName),
motionDeadline: 1000,
}}
destroyTooltipOnHide={!!destroyTooltipOnHide}
>
{tempOpen ? cloneElement(child, { className: childCls }) : child}
</RcTooltip>,
);
}) as React.ForwardRefExoticComponent<
React.PropsWithoutRef<TooltipProps> & React.RefAttributes<unknown>
> & {
_InternalPanelDoNotUseOrYouWillBeFired: typeof PurePanel;
};
if (process.env.NODE_ENV !== 'production') {
Tooltip.displayName = 'Tooltip';
}
Tooltip._InternalPanelDoNotUseOrYouWillBeFired = PurePanel;
export default Tooltip;