126
projects/app/src/web/support/user/hooks/useSendCode.tsx
Normal file
126
projects/app/src/web/support/user/hooks/useSendCode.tsx
Normal file
@@ -0,0 +1,126 @@
|
||||
import { useState, useMemo } from 'react';
|
||||
import { sendAuthCode } from '@/web/support/user/api';
|
||||
import { UserAuthTypeEnum } from '@fastgpt/global/support/user/auth/constants';
|
||||
import { useTranslation } from 'next-i18next';
|
||||
import { useSystemStore } from '@/web/common/system/useSystemStore';
|
||||
import { useRequest2 } from '@fastgpt/web/hooks/useRequest';
|
||||
import { Box, BoxProps, useDisclosure } from '@chakra-ui/react';
|
||||
import SendCodeAuthModal from '@/components/support/user/safe/SendCodeAuthModal';
|
||||
import { useMemoizedFn } from 'ahooks';
|
||||
import { useToast } from '@fastgpt/web/hooks/useToast';
|
||||
let timer: NodeJS.Timeout;
|
||||
|
||||
export const useSendCode = ({ type }: { type: `${UserAuthTypeEnum}` }) => {
|
||||
const { t } = useTranslation();
|
||||
const { feConfigs } = useSystemStore();
|
||||
const { toast } = useToast();
|
||||
const [codeCountDown, setCodeCountDown] = useState(0);
|
||||
|
||||
const { runAsync: sendCode, loading: codeSending } = useRequest2(
|
||||
async ({ username, captcha }: { username: string; captcha: string }) => {
|
||||
if (codeCountDown > 0) return;
|
||||
const googleToken = await getClientToken(feConfigs.googleClientVerKey);
|
||||
await sendAuthCode({ username, type, googleToken, captcha });
|
||||
|
||||
setCodeCountDown(60);
|
||||
|
||||
timer = setInterval(() => {
|
||||
setCodeCountDown((val) => {
|
||||
if (val <= 0) {
|
||||
clearInterval(timer);
|
||||
}
|
||||
return val - 1;
|
||||
});
|
||||
}, 1000);
|
||||
},
|
||||
{
|
||||
successToast: t('user:password.code_sended'),
|
||||
errorToast: t('user:password.code_send_error'),
|
||||
refreshDeps: [codeCountDown, type, feConfigs?.googleClientVerKey]
|
||||
}
|
||||
);
|
||||
|
||||
const sendCodeText = useMemo(() => {
|
||||
if (codeSending) return t('common:support.user.auth.Sending Code');
|
||||
if (codeCountDown >= 10) {
|
||||
return `${codeCountDown}${t('user:password.get_code_again')}`;
|
||||
}
|
||||
if (codeCountDown > 0) {
|
||||
return `0${codeCountDown}${t('user:password.get_code_again')}`;
|
||||
}
|
||||
return t('user:password.get_code');
|
||||
}, [codeCountDown, codeSending, t]);
|
||||
|
||||
const {
|
||||
isOpen: openCodeAuthModal,
|
||||
onOpen: onOpenCodeAuthModal,
|
||||
onClose: onCloseCodeAuthModal
|
||||
} = useDisclosure();
|
||||
|
||||
const SendCodeBox = useMemoizedFn(({ username, ...styles }: BoxProps & { username: string }) => {
|
||||
return (
|
||||
<>
|
||||
<Box
|
||||
position={'absolute'}
|
||||
right={3}
|
||||
zIndex={1}
|
||||
fontSize={'sm'}
|
||||
{...styles}
|
||||
{...(codeCountDown > 0
|
||||
? {
|
||||
color: 'myGray.500'
|
||||
}
|
||||
: {
|
||||
color: 'primary.700',
|
||||
cursor: 'pointer',
|
||||
onClick: () => {
|
||||
if (!username) {
|
||||
toast({
|
||||
status: 'warning',
|
||||
title: t('common:error.username_empty')
|
||||
});
|
||||
} else {
|
||||
onOpenCodeAuthModal();
|
||||
}
|
||||
}
|
||||
})}
|
||||
>
|
||||
{sendCodeText}
|
||||
</Box>
|
||||
{openCodeAuthModal && (
|
||||
<SendCodeAuthModal
|
||||
onClose={onCloseCodeAuthModal}
|
||||
username={username}
|
||||
onSending={codeSending}
|
||||
onSendCode={sendCode}
|
||||
/>
|
||||
)}
|
||||
</>
|
||||
);
|
||||
});
|
||||
|
||||
return {
|
||||
codeSending,
|
||||
sendCode,
|
||||
sendCodeText,
|
||||
codeCountDown,
|
||||
SendCodeBox
|
||||
};
|
||||
};
|
||||
|
||||
export function getClientToken(googleClientVerKey?: string) {
|
||||
if (!googleClientVerKey || typeof window.grecaptcha === 'undefined' || !window.grecaptcha?.ready)
|
||||
return '';
|
||||
return new Promise<string>((resolve, reject) => {
|
||||
window.grecaptcha.ready(async () => {
|
||||
try {
|
||||
const token = await window.grecaptcha.execute(googleClientVerKey, {
|
||||
action: 'submit'
|
||||
});
|
||||
resolve(token);
|
||||
} catch (error) {
|
||||
reject(error);
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user