mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-06 02:15:35 +08:00
80 lines
1.9 KiB
TypeScript
80 lines
1.9 KiB
TypeScript
import * as React from 'react';
|
|
import * as ReactDOM from 'react-dom';
|
|
import Button from '../button';
|
|
import { ButtonType, NativeButtonProps } from '../button/button';
|
|
|
|
export interface ActionButtonProps {
|
|
type?: ButtonType;
|
|
actionFn?: (...args: any[]) => any | PromiseLike<any>;
|
|
closeModal: Function;
|
|
autoFocus?: boolean;
|
|
buttonProps?: NativeButtonProps;
|
|
}
|
|
|
|
export interface ActionButtonState {
|
|
loading: boolean;
|
|
}
|
|
|
|
export default class ActionButton extends React.Component<ActionButtonProps, ActionButtonState> {
|
|
timeoutId: number;
|
|
|
|
constructor(props: ActionButtonProps) {
|
|
super(props);
|
|
this.state = {
|
|
loading: false,
|
|
};
|
|
}
|
|
|
|
componentDidMount() {
|
|
if (this.props.autoFocus) {
|
|
const $this = ReactDOM.findDOMNode(this) as HTMLInputElement;
|
|
this.timeoutId = setTimeout(() => $this.focus());
|
|
}
|
|
}
|
|
|
|
componentWillUnmount() {
|
|
clearTimeout(this.timeoutId);
|
|
}
|
|
|
|
onClick = () => {
|
|
const { actionFn, closeModal } = this.props;
|
|
if (actionFn) {
|
|
let ret;
|
|
if (actionFn.length) {
|
|
ret = actionFn(closeModal);
|
|
} else {
|
|
ret = actionFn();
|
|
if (!ret) {
|
|
closeModal();
|
|
}
|
|
}
|
|
if (ret && ret.then) {
|
|
this.setState({ loading: true });
|
|
ret.then(
|
|
(...args: any[]) => {
|
|
// It's unnecessary to set loading=false, for the Modal will be unmounted after close.
|
|
// this.setState({ loading: false });
|
|
closeModal(...args);
|
|
},
|
|
() => {
|
|
// See: https://github.com/ant-design/ant-design/issues/6183
|
|
this.setState({ loading: false });
|
|
},
|
|
);
|
|
}
|
|
} else {
|
|
closeModal();
|
|
}
|
|
};
|
|
|
|
render() {
|
|
const { type, children, buttonProps } = this.props;
|
|
const loading = this.state.loading;
|
|
return (
|
|
<Button type={type} onClick={this.onClick} loading={loading} {...buttonProps}>
|
|
{children}
|
|
</Button>
|
|
);
|
|
}
|
|
}
|