70 lines
2.1 KiB
TypeScript
70 lines
2.1 KiB
TypeScript
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import React, { forwardRef, useState } from 'react';
|
|
|
|
import { isString } from 'lodash-es';
|
|
import { type ToastReactProps } from '@coze-arch/bot-semi/Toast';
|
|
import { type ButtonProps } from '@coze-arch/bot-semi/Button';
|
|
import { UIButton, Toast, Spin } from '@coze-arch/bot-semi';
|
|
|
|
export type LoadingButtonProps = ButtonProps & {
|
|
/** 加载中的 toast 文案 */
|
|
loadingToast?: string | Omit<ToastReactProps, 'type'>;
|
|
};
|
|
|
|
export const LoadingButton: React.ForwardRefExoticComponent<
|
|
ButtonProps & {
|
|
loadingToast?: string | Omit<ToastReactProps, 'type'> | undefined;
|
|
} & React.RefAttributes<UIButton>
|
|
> = forwardRef<UIButton, LoadingButtonProps>(
|
|
({ loadingToast, ...buttonProps }, ref) => {
|
|
const [loading, setLoading] = useState(false);
|
|
const onClick: React.MouseEventHandler<HTMLButtonElement> = async event => {
|
|
let toastId = '';
|
|
try {
|
|
if (loadingToast) {
|
|
toastId = Toast.info({
|
|
icon: <Spin />,
|
|
showClose: false,
|
|
duration: 0,
|
|
...(isString(loadingToast)
|
|
? { content: loadingToast as string }
|
|
: (loadingToast as Omit<ToastReactProps, 'type'>)),
|
|
});
|
|
}
|
|
setLoading(true);
|
|
if (buttonProps.onClick) {
|
|
await buttonProps.onClick(event);
|
|
}
|
|
} finally {
|
|
setLoading(false);
|
|
if (toastId) {
|
|
Toast.close(toastId);
|
|
}
|
|
}
|
|
};
|
|
return (
|
|
<UIButton
|
|
ref={ref}
|
|
loading={loading}
|
|
{...buttonProps}
|
|
onClick={onClick}
|
|
/>
|
|
);
|
|
},
|
|
);
|