Shumiland/src/lib/rateLimit.ts

type Bucket = {
  tokens: number
  lastRefill: number
}

const buckets = new Map<string, Bucket>()

const MAX_TOKENS = 5
const REFILL_INTERVAL_MS = 15 * 60 * 1000
const REFILL_AMOUNT = 5

export function checkRateLimit(ip: string): boolean {
  const now = Date.now()
  let bucket = buckets.get(ip)

  if (!bucket) {
    bucket = { tokens: MAX_TOKENS - 1, lastRefill: now }
    buckets.set(ip, bucket)
    return true
  }

  const elapsed = now - bucket.lastRefill
  if (elapsed >= REFILL_INTERVAL_MS) {
    bucket.tokens = REFILL_AMOUNT
    bucket.lastRefill = now
  }

  if (bucket.tokens <= 0) {
    return false
  }

  bucket.tokens -= 1
  return true
}