From 30076181dc133bdee97d98fae80ce9c1d95ebcd7 Mon Sep 17 00:00:00 2001
From: austinkelsay <austinkelsay@yahoo.com>
Date: Mon, 30 Sep 2024 19:51:23 -0500
Subject: [PATCH] Update rate limiter

---
 src/middleware.js | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/middleware.js b/src/middleware.js
index 69f5b5f..dc1077a 100644
--- a/src/middleware.js
+++ b/src/middleware.js
@@ -6,6 +6,8 @@ const ratelimit = new Ratelimit({
   redis: kv,
   // 5 requests from the same IP in 10 seconds
   limiter: Ratelimit.slidingWindow(5, '10 s'),
+  analytics: true,
+  timeout: 1000, // 1 second
 });
 
 // Define which routes you want to rate limit
@@ -14,12 +16,19 @@ export const config = {
 };
 
 export default async function middleware(request) {
-  // You could alternatively limit based on user ID or similar
   const ip = request.ip ?? '127.0.0.1';
   const { success, pending, limit, reset, remaining } = await ratelimit.limit(
-    ip
+    `ratelimit_middleware_${ip}`
   );
-  return success
-    ? NextResponse.next()
-    : NextResponse.redirect(new URL('/blocked', request.url));
+
+  if (!success) {
+    return new NextResponse('Too Many Requests', {
+      status: 429,
+      headers: {
+        'Retry-After': reset.toString(),
+      },
+    });
+  }
+
+  return NextResponse.next();
 }
\ No newline at end of file