CustomButton.tsx

const Button = ({onClick, ...rest}) => {
  const [loading, setLoading] = useState(false);
  const isMounted = useMounted();
  const onClicked = async (e) => {
    setLoading(true);
    await onClick(e);
    isMounted.current && setLoading(false);
  }
  return <button onClick={onClicked} {...rest}/>
}