diff --git a/src/pages/index.tsx b/src/pages/index.tsx index e7119e384..8e35ee43b 100644 --- a/src/pages/index.tsx +++ b/src/pages/index.tsx @@ -1,11 +1,19 @@ -import React from 'react'; +import React, { useEffect } from 'react'; import { Card } from '@chakra-ui/react'; import Markdown from '@/components/Markdown'; import { useMarkdown } from '@/hooks/useMarkdown'; +import { useRouter } from 'next/router'; const Home = () => { + const { inviterId } = useRouter().query as { inviterId: string }; const { data } = useMarkdown({ url: '/intro.md' }); + useEffect(() => { + if (inviterId) { + localStorage.setItem('inviterId', inviterId); + } + }, [inviterId]); + return ( diff --git a/src/pages/login/components/RegisterForm.tsx b/src/pages/login/components/RegisterForm.tsx index 1d04da2f5..b239058ee 100644 --- a/src/pages/login/components/RegisterForm.tsx +++ b/src/pages/login/components/RegisterForm.tsx @@ -57,7 +57,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => { username, code, password, - inviterId + inviterId: inviterId || localStorage.getItem('inviterId') || '' }) ); toast({