mirror of
https://github.com/koreanbots/core.git
synced 2025-12-15 14:10:22 +00:00
81 lines
2.4 KiB
TypeScript
81 lines
2.4 KiB
TypeScript
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'
|
|
import { sendWebhook } from '@utils/Webhook'
|
|
import { WebhookType } from '@types'
|
|
|
|
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) {
|
|
get.bot.clear(req.query.id)
|
|
sendWebhook(bot, {
|
|
type: 'bot',
|
|
data: {
|
|
botId: bot.id,
|
|
type: WebhookType.HeartChange,
|
|
before: bot.votes,
|
|
after: bot.votes + 1,
|
|
userId: user,
|
|
},
|
|
timestamp: Date.now(),
|
|
})
|
|
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
|