mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-09 23:51:17 +08:00
317 lines
8.9 KiB
TypeScript
317 lines
8.9 KiB
TypeScript
/* eslint-disable react/button-has-type */
|
|
import React, {
|
|
Children,
|
|
createRef,
|
|
forwardRef,
|
|
useContext,
|
|
useEffect,
|
|
useMemo,
|
|
useState,
|
|
} from 'react';
|
|
import classNames from 'classnames';
|
|
import omit from 'rc-util/lib/omit';
|
|
import { composeRef } from 'rc-util/lib/ref';
|
|
|
|
import { devUseWarning } from '../_util/warning';
|
|
import Wave from '../_util/wave';
|
|
import { ConfigContext } from '../config-provider';
|
|
import DisabledContext from '../config-provider/DisabledContext';
|
|
import useSize from '../config-provider/hooks/useSize';
|
|
import type { SizeType } from '../config-provider/SizeContext';
|
|
import { useCompactItemContext } from '../space/Compact';
|
|
import Group, { GroupSizeContext } from './button-group';
|
|
import type { ButtonHTMLType, ButtonShape, ButtonType } from './buttonHelpers';
|
|
import { isTwoCNChar, isUnBorderedButtonType, spaceChildren } from './buttonHelpers';
|
|
import IconWrapper from './IconWrapper';
|
|
import LoadingIcon from './LoadingIcon';
|
|
import useStyle from './style';
|
|
import CompactCmp from './style/compactCmp';
|
|
|
|
export type LegacyButtonType = ButtonType | 'danger';
|
|
|
|
export interface BaseButtonProps {
|
|
type?: ButtonType;
|
|
icon?: React.ReactNode;
|
|
shape?: ButtonShape;
|
|
size?: SizeType;
|
|
disabled?: boolean;
|
|
loading?: boolean | { delay?: number };
|
|
prefixCls?: string;
|
|
className?: string;
|
|
rootClassName?: string;
|
|
ghost?: boolean;
|
|
danger?: boolean;
|
|
block?: boolean;
|
|
children?: React.ReactNode;
|
|
[key: `data-${string}`]: string;
|
|
classNames?: { icon: string };
|
|
styles?: { icon: React.CSSProperties };
|
|
}
|
|
|
|
type MergedHTMLAttributes = Omit<
|
|
React.HTMLAttributes<HTMLElement> &
|
|
React.ButtonHTMLAttributes<HTMLElement> &
|
|
React.AnchorHTMLAttributes<HTMLElement>,
|
|
'type'
|
|
>;
|
|
|
|
export interface ButtonProps extends BaseButtonProps, MergedHTMLAttributes {
|
|
href?: string;
|
|
htmlType?: ButtonHTMLType;
|
|
}
|
|
|
|
type CompoundedComponent = React.ForwardRefExoticComponent<
|
|
ButtonProps & React.RefAttributes<HTMLElement>
|
|
> & {
|
|
Group: typeof Group;
|
|
/** @internal */
|
|
__ANT_BUTTON: boolean;
|
|
};
|
|
|
|
type LoadingConfigType = {
|
|
loading: boolean;
|
|
delay: number;
|
|
};
|
|
|
|
function getLoadingConfig(loading: BaseButtonProps['loading']): LoadingConfigType {
|
|
if (typeof loading === 'object' && loading) {
|
|
let delay = loading?.delay;
|
|
delay = !Number.isNaN(delay) && typeof delay === 'number' ? delay : 0;
|
|
return {
|
|
loading: delay <= 0,
|
|
delay,
|
|
};
|
|
}
|
|
|
|
return {
|
|
loading: !!loading,
|
|
delay: 0,
|
|
};
|
|
}
|
|
|
|
const InternalButton: React.ForwardRefRenderFunction<
|
|
HTMLButtonElement | HTMLAnchorElement,
|
|
ButtonProps
|
|
> = (props, ref) => {
|
|
const {
|
|
loading = false,
|
|
prefixCls: customizePrefixCls,
|
|
type = 'default',
|
|
danger,
|
|
shape = 'default',
|
|
size: customizeSize,
|
|
styles,
|
|
disabled: customDisabled,
|
|
className,
|
|
rootClassName,
|
|
children,
|
|
icon,
|
|
ghost = false,
|
|
block = false,
|
|
// React does not recognize the `htmlType` prop on a DOM element. Here we pick it out of `rest`.
|
|
htmlType = 'button',
|
|
classNames: customClassNames,
|
|
style: customStyle = {},
|
|
...rest
|
|
} = props;
|
|
|
|
const { getPrefixCls, autoInsertSpaceInButton, direction, button } = useContext(ConfigContext);
|
|
const prefixCls = getPrefixCls('btn', customizePrefixCls);
|
|
|
|
const [wrapCSSVar, hashId, cssVarCls] = useStyle(prefixCls);
|
|
|
|
const disabled = useContext(DisabledContext);
|
|
const mergedDisabled = customDisabled ?? disabled;
|
|
|
|
const groupSize = useContext(GroupSizeContext);
|
|
|
|
const loadingOrDelay = useMemo<LoadingConfigType>(() => getLoadingConfig(loading), [loading]);
|
|
|
|
const [innerLoading, setLoading] = useState<boolean>(loadingOrDelay.loading);
|
|
|
|
const [hasTwoCNChar, setHasTwoCNChar] = useState<boolean>(false);
|
|
|
|
const internalRef = createRef<HTMLButtonElement | HTMLAnchorElement>();
|
|
|
|
const buttonRef = composeRef(ref, internalRef);
|
|
|
|
const needInserted = Children.count(children) === 1 && !icon && !isUnBorderedButtonType(type);
|
|
|
|
useEffect(() => {
|
|
let delayTimer: ReturnType<typeof setTimeout> | null = null;
|
|
if (loadingOrDelay.delay > 0) {
|
|
delayTimer = setTimeout(() => {
|
|
delayTimer = null;
|
|
setLoading(true);
|
|
}, loadingOrDelay.delay);
|
|
} else {
|
|
setLoading(loadingOrDelay.loading);
|
|
}
|
|
|
|
function cleanupTimer() {
|
|
if (delayTimer) {
|
|
clearTimeout(delayTimer);
|
|
delayTimer = null;
|
|
}
|
|
}
|
|
|
|
return cleanupTimer;
|
|
}, [loadingOrDelay]);
|
|
|
|
useEffect(() => {
|
|
// FIXME: for HOC usage like <FormatMessage />
|
|
if (!buttonRef || !(buttonRef as any).current || autoInsertSpaceInButton === false) {
|
|
return;
|
|
}
|
|
const buttonText = (buttonRef as any).current.textContent;
|
|
if (needInserted && isTwoCNChar(buttonText)) {
|
|
if (!hasTwoCNChar) {
|
|
setHasTwoCNChar(true);
|
|
}
|
|
} else if (hasTwoCNChar) {
|
|
setHasTwoCNChar(false);
|
|
}
|
|
}, [buttonRef]);
|
|
|
|
const handleClick = (e: React.MouseEvent<HTMLButtonElement | HTMLAnchorElement, MouseEvent>) => {
|
|
const { onClick } = props;
|
|
// FIXME: https://github.com/ant-design/ant-design/issues/30207
|
|
if (innerLoading || mergedDisabled) {
|
|
e.preventDefault();
|
|
return;
|
|
}
|
|
(onClick as React.MouseEventHandler<HTMLButtonElement | HTMLAnchorElement>)?.(e);
|
|
};
|
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
const warning = devUseWarning('Button');
|
|
|
|
warning(
|
|
!(typeof icon === 'string' && icon.length > 2),
|
|
'breaking',
|
|
`\`icon\` is using ReactNode instead of string naming in v4. Please check \`${icon}\` at https://ant.design/components/icon`,
|
|
);
|
|
|
|
warning(
|
|
!(ghost && isUnBorderedButtonType(type)),
|
|
'usage',
|
|
"`link` or `text` button can't be a `ghost` button.",
|
|
);
|
|
}
|
|
|
|
const autoInsertSpace = autoInsertSpaceInButton !== false;
|
|
const { compactSize, compactItemClassnames } = useCompactItemContext(prefixCls, direction);
|
|
|
|
const sizeClassNameMap = { large: 'lg', small: 'sm', middle: undefined };
|
|
|
|
const sizeFullName = useSize((ctxSize) => customizeSize ?? compactSize ?? groupSize ?? ctxSize);
|
|
|
|
const sizeCls = sizeFullName ? sizeClassNameMap[sizeFullName] || '' : '';
|
|
|
|
const iconType = innerLoading ? 'loading' : icon;
|
|
|
|
const linkButtonRestProps = omit(rest as ButtonProps & { navigate: any }, ['navigate']);
|
|
|
|
const classes = classNames(
|
|
prefixCls,
|
|
hashId,
|
|
cssVarCls,
|
|
{
|
|
[`${prefixCls}-${shape}`]: shape !== 'default' && shape,
|
|
[`${prefixCls}-${type}`]: type,
|
|
[`${prefixCls}-${sizeCls}`]: sizeCls,
|
|
[`${prefixCls}-icon-only`]: !children && children !== 0 && !!iconType,
|
|
[`${prefixCls}-background-ghost`]: ghost && !isUnBorderedButtonType(type),
|
|
[`${prefixCls}-loading`]: innerLoading,
|
|
[`${prefixCls}-two-chinese-chars`]: hasTwoCNChar && autoInsertSpace && !innerLoading,
|
|
[`${prefixCls}-block`]: block,
|
|
[`${prefixCls}-dangerous`]: !!danger,
|
|
[`${prefixCls}-rtl`]: direction === 'rtl',
|
|
},
|
|
compactItemClassnames,
|
|
className,
|
|
rootClassName,
|
|
button?.className,
|
|
);
|
|
|
|
const fullStyle: React.CSSProperties = { ...button?.style, ...customStyle };
|
|
|
|
const iconClasses = classNames(customClassNames?.icon, button?.classNames?.icon);
|
|
const iconStyle: React.CSSProperties = {
|
|
...(styles?.icon || {}),
|
|
...(button?.styles?.icon || {}),
|
|
};
|
|
|
|
const iconNode =
|
|
icon && !innerLoading ? (
|
|
<IconWrapper prefixCls={prefixCls} className={iconClasses} style={iconStyle}>
|
|
{icon}
|
|
</IconWrapper>
|
|
) : (
|
|
<LoadingIcon existIcon={!!icon} prefixCls={prefixCls} loading={!!innerLoading} />
|
|
);
|
|
|
|
const kids =
|
|
children || children === 0 ? spaceChildren(children, needInserted && autoInsertSpace) : null;
|
|
|
|
if (linkButtonRestProps.href !== undefined) {
|
|
return wrapCSSVar(
|
|
<a
|
|
{...linkButtonRestProps}
|
|
className={classNames(classes, {
|
|
[`${prefixCls}-disabled`]: mergedDisabled,
|
|
})}
|
|
href={mergedDisabled ? undefined : linkButtonRestProps.href}
|
|
style={fullStyle}
|
|
onClick={handleClick}
|
|
ref={buttonRef as React.Ref<HTMLAnchorElement>}
|
|
tabIndex={mergedDisabled ? -1 : 0}
|
|
>
|
|
{iconNode}
|
|
{kids}
|
|
</a>,
|
|
);
|
|
}
|
|
|
|
let buttonNode = (
|
|
<button
|
|
{...rest}
|
|
type={htmlType}
|
|
className={classes}
|
|
style={fullStyle}
|
|
onClick={handleClick}
|
|
disabled={mergedDisabled}
|
|
ref={buttonRef as React.Ref<HTMLButtonElement>}
|
|
>
|
|
{iconNode}
|
|
{kids}
|
|
|
|
{/* Styles: compact */}
|
|
{!!compactItemClassnames && <CompactCmp key="compact" prefixCls={prefixCls} />}
|
|
</button>
|
|
);
|
|
|
|
if (!isUnBorderedButtonType(type)) {
|
|
buttonNode = (
|
|
<Wave component="Button" disabled={!!innerLoading}>
|
|
{buttonNode}
|
|
</Wave>
|
|
);
|
|
}
|
|
|
|
return wrapCSSVar(buttonNode);
|
|
};
|
|
|
|
const Button = forwardRef<HTMLButtonElement | HTMLAnchorElement, ButtonProps>(
|
|
InternalButton,
|
|
) as CompoundedComponent;
|
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
Button.displayName = 'Button';
|
|
}
|
|
|
|
Button.Group = Group;
|
|
Button.__ANT_BUTTON = true;
|
|
|
|
export default Button;
|