core/pages/api/v2/bots/[id]/vote.ts

import { NextApiRequest } from 'next'

import { CaptchaVerify, get, put } from '@utils/Query'
import RequestHandler from '@utils/RequestHandler'
import ResponseWrapper from '@utils/ResponseWrapper'
import { checkToken } from '@utils/Csrf'
import Yup from '@utils/Yup'
import { VOTE_COOLDOWN } from '@utils/Constants'

const BotVote = RequestHandler()
	.get(async (req: GetApiRequest, res) => {
		const bot = await get.BotAuthorization(req.headers.authorization)
		if(!bot) return ResponseWrapper(res, { code: 401 })
		if(req.query.id !== bot) return ResponseWrapper(res, { code: 403 })
		const userID = await Yup.string().required().label('userID').validate(req.query.userID).then(el => el).catch(e => {
			ResponseWrapper(res, { code: 400, errors: e.errors })
			return null
		})
		if(!userID) return ResponseWrapper(res, { code: 400 })
		const result = await get.botVote(userID, bot)
		return ResponseWrapper(res, { code: 200, data: { voted: +new Date() < result + VOTE_COOLDOWN, lastVote: result } })
	})
	.post(async (req: PostApiRequest, res) => {
		const user = await get.Authorization(req.cookies.token)
		if(!user) return ResponseWrapper(res, { code: 401 })
		const bot = await get.bot.load(req.query.id)
		if (!bot) return ResponseWrapper(res, { code: 404, message: '존재하지 않는 봇입니다.' })
		const csrfValidated = checkToken(req, res, req.body._csrf)
		if (!csrfValidated) return
		const captcha = await CaptchaVerify(req.body._captcha)
		if(!captcha) return ResponseWrapper(res, { code: 400, message: '캡챠 검증에 실패하였습니다.' })

		const vote = await put.voteBot(user, bot.id)
		if(vote === null) return ResponseWrapper(res, { code: 401 })
		else if(vote === true) return ResponseWrapper(res, { code: 200 })
		else return ResponseWrapper(res, { code: 429, data: { retryAfter: vote } })
	})

interface ApiRequest extends NextApiRequest {
  query: {
    id: string
  }
}

interface GetApiRequest extends ApiRequest {
	query: {
		id: string
		userID: string
	}
}
interface PostApiRequest extends ApiRequest {
	body: {
    _captcha: string
    _csrf: string
  }
}
export default BotVote