import { NextApiResponse } from 'next'
import ResponseWrapper from './ResponseWrapper'

export default function RateLimitHandler(
	res: NextApiResponse, ratelimit: RateLimit
) {
	res.setHeader('x-ratelimit-limit', ratelimit.limit)
	res.setHeader('x-ratelimit-remaining', 600 - (ratelimit.used > ratelimit.limit ? ratelimit.limit : ratelimit.used))
	res.setHeader('x-ratelimit-reset', Math.round(ratelimit.reset / 1000))
	if(ratelimit.limit < ratelimit.used) {
		if(ratelimit.onLimitExceed) ratelimit.onLimitExceed(res)
		else ResponseWrapper(res, { code: 429 })
		return true
	}
}


interface RateLimit {
	used: number
	limit: number
  reset: number,
  onLimitExceed: (res: NextApiResponse)=> void  | Promise<void>
}