import { NextRequest } from "next/server";

// Store counts in a Map
const requestCounts = new Map<string, { count: number; timestamp: number }>();

// Clean up the map every minute
setInterval(() => {
  const currentTimestamp = Date.now();
  requestCounts.forEach((requests, ip) => {
    if (currentTimestamp - requests.timestamp > 60000) {
      requestCounts.delete(ip);
    }
  });
}, 60000);

export const config = {
  matcher: "/api/:path*",
};

export function middleware(request: NextRequest) {
  // Get client IP
  const ip =
    request.headers.get("x-real-ip") ||
    request.headers.get("x-forwarded-for") ||
    request.ip ||
    "unknown";

  if (ip === "unknown") {
    return;
  }

  // Get current timestamp
  const now = Date.now();

  // Get requests from last 15 seconds
  let requests = requestCounts.get(ip) || { count: 0, timestamp: now };

  // Increment count
  requests.count++;

  // Check if past 15 seconds
  if (now - requests.timestamp > 15000) {
    // Reset count
    requests.count = 1;
  }

  // Set timestamp
  requests.timestamp = now;

  // Set back to map
  requestCounts.set(ip, requests);

  // Check if rate limit exceeded
  if (requests.count > 300) {
    return new Response("Too Many Requests, please try again later.", {
      status: 429,
    });
  }

  // Continue request
  return;
}