127 lines
3.8 KiB
TypeScript
127 lines
3.8 KiB
TypeScript
import React, { useCallback, useEffect } from 'react';
|
|
import { useRouter } from 'next/router';
|
|
import { useSystemStore } from '@/web/common/system/useSystemStore';
|
|
import type { ResLogin } from '@/global/support/api/userRes.d';
|
|
import { useUserStore } from '@/web/support/user/useUserStore';
|
|
import { clearToken } from '@/web/support/user/auth';
|
|
import { oauthLogin } from '@/web/support/user/api';
|
|
import { useToast } from '@fastgpt/web/hooks/useToast';
|
|
import Loading from '@fastgpt/web/components/common/MyLoading';
|
|
import { serviceSideProps } from '@/web/common/i18n/utils';
|
|
import { getErrText } from '@fastgpt/global/common/error/utils';
|
|
import { useTranslation } from 'next-i18next';
|
|
import { OAuthEnum } from '@fastgpt/global/support/user/constant';
|
|
|
|
let isOauthLogging = false;
|
|
|
|
const provider = () => {
|
|
const { t } = useTranslation();
|
|
const { initd, loginStore, setLoginStore } = useSystemStore();
|
|
const { setUserInfo } = useUserStore();
|
|
const router = useRouter();
|
|
const { code, state, error } = router.query as { code: string; state: string; error?: string };
|
|
const { toast } = useToast();
|
|
|
|
const loginSuccess = useCallback(
|
|
(res: ResLogin) => {
|
|
setUserInfo(res.user);
|
|
|
|
router.push(loginStore?.lastRoute ? decodeURIComponent(loginStore?.lastRoute) : '/app/list');
|
|
},
|
|
[setUserInfo, router, loginStore?.lastRoute]
|
|
);
|
|
|
|
const authCode = useCallback(
|
|
async (code: string) => {
|
|
try {
|
|
const res = await oauthLogin({
|
|
type: loginStore?.provider || OAuthEnum.sso,
|
|
code,
|
|
callbackUrl: `${location.origin}/login/provider`,
|
|
inviterId: localStorage.getItem('inviterId') || undefined,
|
|
bd_vid: sessionStorage.getItem('bd_vid') || undefined,
|
|
fastgpt_sem: (() => {
|
|
try {
|
|
return sessionStorage.getItem('fastgpt_sem')
|
|
? JSON.parse(sessionStorage.getItem('fastgpt_sem')!)
|
|
: undefined;
|
|
} catch {
|
|
return undefined;
|
|
}
|
|
})(),
|
|
sourceDomain: sessionStorage.getItem('sourceDomain') || undefined
|
|
});
|
|
|
|
if (!res) {
|
|
toast({
|
|
status: 'warning',
|
|
title: t('common:support.user.login.error')
|
|
});
|
|
return setTimeout(() => {
|
|
router.replace('/login');
|
|
}, 1000);
|
|
}
|
|
loginSuccess(res);
|
|
} catch (error) {
|
|
toast({
|
|
status: 'warning',
|
|
title: getErrText(error, t('common:support.user.login.error'))
|
|
});
|
|
setTimeout(() => {
|
|
router.replace('/login');
|
|
}, 1000);
|
|
}
|
|
setLoginStore(undefined);
|
|
},
|
|
[loginStore?.provider, loginSuccess, router, setLoginStore, t, toast]
|
|
);
|
|
|
|
useEffect(() => {
|
|
if (error) {
|
|
toast({
|
|
status: 'warning',
|
|
title: t('common:support.user.login.Provider error')
|
|
});
|
|
router.replace('/login');
|
|
return;
|
|
}
|
|
|
|
console.log('SSO', { initd, loginStore, code, state });
|
|
if (!code || !initd) return;
|
|
|
|
if (isOauthLogging) return;
|
|
|
|
isOauthLogging = true;
|
|
|
|
(async () => {
|
|
await clearToken();
|
|
router.prefetch('/app/list');
|
|
|
|
if (loginStore && loginStore.provider !== 'sso' && state !== loginStore.state) {
|
|
toast({
|
|
status: 'warning',
|
|
title: t('common:support.user.login.security_failed')
|
|
});
|
|
setTimeout(() => {
|
|
router.replace('/login');
|
|
}, 1000);
|
|
return;
|
|
} else {
|
|
authCode(code);
|
|
}
|
|
})();
|
|
}, [initd, authCode, code, error, loginStore, loginStore?.state, router, state, t, toast]);
|
|
|
|
return <Loading />;
|
|
};
|
|
|
|
export default provider;
|
|
|
|
export async function getServerSideProps(context: any) {
|
|
return {
|
|
props: {
|
|
...(await serviceSideProps(context))
|
|
}
|
|
};
|
|
}
|