wonderlandpark/core
2
1
Fork 0
mirror of https://github.com/koreanbots/core.git synced 2025-12-16 14:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: avatar api

Browse Source
This commit is contained in:
원더 2021-01-24 13:52:25 +09:00
parent 5280040a19
commit 2b045439a4
1 changed files with 35 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
pages/api/image/discord/avatars/[id].ts
View File

@ -1,24 +1,44 @@
import { createReadStream } from 'fs'
import { NextApiHandler, NextApiRequest, NextApiResponse } from 'next'
import { NextApiRequest, NextApiResponse } from 'next'
import nc from 'next-connect'
import ResponseWrapper from '@utils/ResponseWrapper'
import { DiscordEnpoints } from '@utils/Constants'
import { get } from '@utils/Query'
import { get, ratelimit } from '@utils/Query'
import RateLimitHandler from '@utils/RateLimitHandler'
const Avatar: NextApiHandler = async(req: ApiRequest, res: NextApiResponse) => {
const splitted = req.query.id.split('.')
const ext= splitted[1]
const id = splitted[0]
if(splitted.length !== 2) return ResponseWrapper(res, { code: 400, message: '올바르지 않은 형식입니다.' })
if(!['webp', 'png', 'gif'].includes(ext)) return ResponseWrapper(res, { code: 400, message: '올바르지 않은 확장자입니다.' })
const Avatar = nc<ApiRequest, NextApiResponse>()
.get(async(req, res) => {
const { imageRateLimit } = await import('@utils/Query')
const rate = ratelimit.image(req.socket.remoteAddress)
// eslint-disable-next-line @typescript-eslint/no-explicit-any
const ratelimited = RateLimitHandler(res, { used: rate, limit: 600, reset: (<any>imageRateLimit).scheduler.get(req.socket.remoteAddress).expiry, onLimitExceed: async(res) => {
const img = await get.images.user.load(DiscordEnpoints.CDN.default(Math.floor(Math.random() * 6), { format: 'png' }))
res.setHeader('Content-Type', 'image/png')
img.pipe(res)
} })
console.log(ratelimited)
if(ratelimited) return
const splitted = req.query.id.split('.')
let ext = splitted[1]
const id = splitted[0]
if(splitted.length !== 2) return ResponseWrapper(res, { code: 400, message: '올바르지 않은 형식입니다.' })
if(!['webp', 'png', 'gif'].includes(ext)) return ResponseWrapper(res, { code: 400, message: '올바르지 않은 확장자입니다.' })
const user = await get.discord.user.load(id)
if(!user) return ResponseWrapper(res, { code: 400, message: '올바르지 않은 유저입니다.' })
const user = await get.discord.user.load(id)
if(!user) return ResponseWrapper(res, { code: 400, message: '올바르지 않은 유저입니다.' })
let img = await get.images.user.load(DiscordEnpoints.CDN.user(id, user.avatar, { format: ext === 'gif' && !user.avatar.startsWith('a_') ? 'png' : (ext as Ext) }))
if(!user.avatar || !img) {
img = await get.images.user.load(DiscordEnpoints.CDN.default(user.discriminator, { format: 'png' }))
ext = 'png'
}
res.setHeader('Content-Type', `image/${ext}`)
img.pipe(res)
})
const image = await get.images.user.load(DiscordEnpoints.CDN.user(id, user.avatar, { format: (ext as Ext) }))
res.setHeader('Content-Type', `image/${ext}`)
image.pipe(res)
}
interface ApiRequest extends NextApiRequest {
query: {