core/pages/bots/[id]/vote.tsx

import { NextPage } from 'next'
import Link from 'next/link'
import dynamic from 'next/dynamic'
import { useRouter } from 'next/router'

import { Bot, CsrfContext, ResponseProps, Theme, User } from '@types'
import { get } from '@utils/Query'
import { makeBotURL, parseCookie, checkBotFlag } from '@utils/Tools'

import { ParsedUrlQuery } from 'querystring'

import NotFound from 'pages/404'
import { getToken } from '@utils/Csrf'
import Captcha from '@components/Captcha'
import { useState } from 'react'
import Fetch from '@utils/Fetch'
import Day from '@utils/Day'
import { getJosaPicker } from 'josa'
import { KoreanbotsEndPoints } from '@utils/Constants'


const Container = dynamic(() => import('@components/Container'))
const DiscordAvatar = dynamic(() => import('@components/DiscordAvatar'))
const Button = dynamic(() => import('@components/Button'))
const Tag = dynamic(() => import('@components/Tag'))
const Segment = dynamic(() => import('@components/Segment'))
const SEO = dynamic(() => import('@components/SEO'))
const Advertisement = dynamic(() => import('@components/Advertisement'))
const Login = dynamic(() => import('@components/Login'))

const VoteBot: NextPage<VoteBotProps> = ({ data, user, csrfToken, theme }) => {
	const [ votingStatus, setVotingStatus ] = useState(0)
	const [ result, setResult ] = useState<ResponseProps<{retryAfter?: number}>>(null)
	const router = useRouter()
	if(!data?.id) return <NotFound />
	if(csrfToken !== router.query.csrfToken) {
		router.push(`/bots/${data.id}`)
		return <></>
	}
	if(!user) return <Login>
		<SEO title={data.name} description={`한국 디스코드봇 리스트에서 ${data.name}에 투표하세요`} image={KoreanbotsEndPoints.CDN.avatar(data.id, { format: 'png', size: 256 })} />
	</Login>
	
	if((checkBotFlag(data.flags, 'trusted') || checkBotFlag(data.flags, 'partnered')) && data.vanity && data.vanity !== router.query.id) router.push(`/bots/${data.vanity}/vote?csrfToken=${csrfToken}`)
	return <Container paddingTop className='py-10'>
		<SEO title={data.name} description={`한국 디스코드봇 리스트에서 ${data.name}에 투표하세요`} image={KoreanbotsEndPoints.CDN.avatar(data.id, { format: 'png', size: 256 })} />
		<Advertisement />
		<Link href={makeBotURL(data)}>
			<a className='text-blue-500 hover:opacity-80'><i className='fas fa-arrow-left mt-3 mb-3' /> <strong>{data.name}</strong>{getJosaPicker('로')(data.name)} 돌아가기</a>
		</Link>
		<Segment className='mb-10 py-8'>
			<div className='text-center'>
				<DiscordAvatar userID={data.id} className='mx-auto w-52 h-52 bg-white mb-4' />
				<Tag text={<span><i className='fas fa-heart text-red-600' /> {data.votes}</span>} dark />
				<h1 className='text-3xl font-bold mt-3'>{data.name}</h1>
				<h4 className='text-md mt-1'>12시간 마다 다시 투표하실 수 있습니다.</h4>
				<div className='inline-block mt-2'>
					{
						votingStatus === 0 ? <Button onClick={()=> setVotingStatus(1)}>
							<><i className='far fa-heart text-red-600'/> 하트 추가</>
						</Button> 
							: votingStatus === 1 ? <Captcha dark={theme === 'dark'} onVerify={async (key) => {
								const res = await Fetch<{ retryAfter: number }|unknown>(`/bots/${data.id}/vote`, { method: 'POST', body: JSON.stringify({ _csrf: csrfToken, _captcha: key }) })
								setResult(res)
								setVotingStatus(2)
							}}
							/>
								: result.code === 200 ? <h2 className='text-2xl font-bold'>해당 봇에 투표했습니다!</h2>
									: result.code === 429 ? <>
										<h2 className='text-2xl font-bold'>이미 해당 봇에 투표하였습니다.</h2>
										<h4 className='text-md mt-1'>{Day(+new Date() + result.data?.retryAfter).fromNow()} 다시 투표하실 수 있습니다.</h4>
									</>
										: <p>{result.message}</p>
					}
				</div>
				
			</div>
		</Segment>
		<Advertisement />
	</Container>
}

export const getServerSideProps = async (ctx: Context) => {
	const parsed = parseCookie(ctx.req)
	const data = await get.bot.load(ctx.query.id)
	const user = await get.Authorization(parsed?.token)
	
	return {
		props: {
			csrfToken: getToken(ctx.req, ctx.res),
			data,
			user: await get.user.load(user || '')
		},
	}
}

interface VoteBotProps {
	csrfToken: string
  vote: boolean
  data: Bot
  user: User
	theme: Theme
}

interface Context extends CsrfContext {
  query: URLQuery
}

interface URLQuery extends ParsedUrlQuery {
	id: string
}

export default VoteBot