import * as React from 'react'; import Button from '../button'; import { LegacyButtonType, ButtonProps, convertLegacyProps } from '../button/button'; import useDestroyed from './hooks/useDestroyed'; export interface ActionButtonProps { type?: LegacyButtonType; actionFn?: (...args: any[]) => any | PromiseLike; close: Function; autoFocus?: boolean; prefixCls: string; buttonProps?: ButtonProps; emitEvent?: boolean; quitOnNullishReturnValue?: boolean; } function isThenable(thing?: PromiseLike): boolean { return !!(thing && !!thing.then); } const ActionButton: React.FC = props => { const clickedRef = React.useRef(false); const ref = React.useRef(); const isDestroyed = useDestroyed(); const [loading, setLoading] = React.useState(false); React.useEffect(() => { let timeoutId: any; if (props.autoFocus) { const $this = ref.current as HTMLInputElement; timeoutId = setTimeout(() => $this.focus()); } return () => { if (timeoutId) { clearTimeout(timeoutId); } }; }, []); const handlePromiseOnOk = (returnValueOfOnOk?: PromiseLike) => { const { close } = props; if (!isThenable(returnValueOfOnOk)) { return; } setLoading(true); returnValueOfOnOk!.then( (...args: any[]) => { if (!isDestroyed()) { setLoading(false); } close(...args); clickedRef.current = false; }, (e: Error) => { // Emit error when catch promise reject // eslint-disable-next-line no-console console.error(e); // See: https://github.com/ant-design/ant-design/issues/6183 if (!isDestroyed()) { setLoading(false); } clickedRef.current = false; }, ); }; const onClick = (e: React.MouseEvent) => { const { actionFn, close } = props; if (clickedRef.current) { return; } clickedRef.current = true; if (!actionFn) { close(); return; } let returnValueOfOnOk; if (props.emitEvent) { returnValueOfOnOk = actionFn(e); if (props.quitOnNullishReturnValue && !isThenable(returnValueOfOnOk)) { clickedRef.current = false; close(e); return; } } else if (actionFn.length) { returnValueOfOnOk = actionFn(close); // https://github.com/ant-design/ant-design/issues/23358 clickedRef.current = false; } else { returnValueOfOnOk = actionFn(); if (!returnValueOfOnOk) { close(); return; } } handlePromiseOnOk(returnValueOfOnOk); }; const { type, children, prefixCls, buttonProps } = props; return ( ); }; export default ActionButton;