101 lines
2.5 KiB
JavaScript
101 lines
2.5 KiB
JavaScript
"use client";
|
|
|
|
import * as React from 'react';
|
|
import useState from "@rc-component/util/es/hooks/useState";
|
|
import Button from '../button/Button';
|
|
import { convertLegacyProps } from '../button/buttonHelpers';
|
|
import { isThenable } from './is';
|
|
const ActionButton = props => {
|
|
const {
|
|
type,
|
|
children,
|
|
prefixCls,
|
|
buttonProps,
|
|
close,
|
|
autoFocus,
|
|
emitEvent,
|
|
isSilent,
|
|
quitOnNullishReturnValue,
|
|
actionFn
|
|
} = props;
|
|
const clickedRef = React.useRef(false);
|
|
const buttonRef = React.useRef(null);
|
|
const [loading, setLoading] = useState(false);
|
|
const onInternalClose = (...args) => {
|
|
close?.(...args);
|
|
};
|
|
React.useEffect(() => {
|
|
let timeoutId = null;
|
|
if (autoFocus) {
|
|
timeoutId = setTimeout(() => {
|
|
buttonRef.current?.focus({
|
|
preventScroll: true
|
|
});
|
|
});
|
|
}
|
|
return () => {
|
|
if (timeoutId) {
|
|
clearTimeout(timeoutId);
|
|
}
|
|
};
|
|
}, [autoFocus]);
|
|
const handlePromiseOnOk = returnValueOfOnOk => {
|
|
if (!isThenable(returnValueOfOnOk)) {
|
|
return;
|
|
}
|
|
setLoading(true);
|
|
returnValueOfOnOk.then((...args) => {
|
|
setLoading(false, true);
|
|
onInternalClose.apply(void 0, args);
|
|
clickedRef.current = false;
|
|
}, e => {
|
|
// See: https://github.com/ant-design/ant-design/issues/6183
|
|
setLoading(false, true);
|
|
clickedRef.current = false;
|
|
// Do not throw if is `await` mode
|
|
if (isSilent?.()) {
|
|
return;
|
|
}
|
|
return Promise.reject(e);
|
|
});
|
|
};
|
|
const onClick = e => {
|
|
if (clickedRef.current) {
|
|
return;
|
|
}
|
|
clickedRef.current = true;
|
|
if (!actionFn) {
|
|
onInternalClose();
|
|
return;
|
|
}
|
|
let returnValueOfOnOk;
|
|
if (emitEvent) {
|
|
returnValueOfOnOk = actionFn(e);
|
|
if (quitOnNullishReturnValue && !isThenable(returnValueOfOnOk)) {
|
|
clickedRef.current = false;
|
|
onInternalClose(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 (!isThenable(returnValueOfOnOk)) {
|
|
onInternalClose();
|
|
return;
|
|
}
|
|
}
|
|
handlePromiseOnOk(returnValueOfOnOk);
|
|
};
|
|
return /*#__PURE__*/React.createElement(Button, {
|
|
...convertLegacyProps(type),
|
|
onClick: onClick,
|
|
loading: loading,
|
|
prefixCls: prefixCls,
|
|
...buttonProps,
|
|
ref: buttonRef
|
|
}, children);
|
|
};
|
|
export default ActionButton; |