From db71f2928831867a7ebfed3063cb04791a92318c Mon Sep 17 00:00:00 2001
From: MythEclipse <asepharyanalm001@gmail.com>
Date: Mon, 18 May 2026 07:01:32 +0700
Subject: [PATCH] feat: add in-memory rate limiter

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
---
 src/utils/rateLimit.js | 42 +++++++++++++++++++++
 test/rateLimit.test.js | 86 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 128 insertions(+)
 create mode 100644 src/utils/rateLimit.js
 create mode 100644 test/rateLimit.test.js

diff --git a/src/utils/rateLimit.js b/src/utils/rateLimit.js
new file mode 100644
index 0000000..f80a848
--- /dev/null
+++ b/src/utils/rateLimit.js
@@ -0,0 +1,42 @@
+import logger from './logger.js';
+
+const rateLimitMap = new Map();
+
+export const checkRateLimit = (key) => {
+  const now = Date.now();
+  const windowMs = parseInt(process.env.RATE_LIMIT_WINDOW_MS, 10) || 60000;
+  const maxRequests = parseInt(process.env.RATE_LIMIT_MAX_REQUESTS, 10) || 30;
+
+  if (!rateLimitMap.has(key)) {
+    rateLimitMap.set(key, { count: 0, reset: now + windowMs });
+  }
+
+  const record = rateLimitMap.get(key);
+
+  if (now > record.reset) {
+    record.count = 0;
+    record.reset = now + windowMs;
+  }
+
+  if (record.count >= maxRequests) {
+    logger.warn('Rate limit exceeded', { key, count: record.count, reset: record.reset });
+    return false;
+  }
+
+  record.count++;
+  return true;
+};
+
+export const cleanupRateLimitCache = () => {
+  const now = Date.now();
+  const windowMs = parseInt(process.env.RATE_LIMIT_WINDOW_MS, 10) || 60000;
+  const keysToDelete = [];
+
+  for (const [key, record] of rateLimitMap.entries()) {
+    if (now > record.reset) {
+      keysToDelete.push(key);
+    }
+  }
+
+  keysToDelete.forEach(key => rateLimitMap.delete(key));
+};
diff --git a/test/rateLimit.test.js b/test/rateLimit.test.js
new file mode 100644
index 0000000..bc37d00
--- /dev/null
+++ b/test/rateLimit.test.js
@@ -0,0 +1,86 @@
+import { describe, it, expect, beforeEach, afterEach, mock, spyOn } from "bun:test";
+import { checkRateLimit, cleanupRateLimitCache } from "../src/utils/rateLimit.js";
+import logger from "../src/utils/logger.js";
+
+// Spy on logger.warn
+const warnSpy = spyOn(logger, "warn");
+
+describe("Rate Limiter", () => {
+  beforeEach(() => {
+    warnSpy.mockClear();
+    // Set custom env variables for predictable tests
+    process.env.RATE_LIMIT_WINDOW_MS = "100"; // 100ms window
+    process.env.RATE_LIMIT_MAX_REQUESTS = "3"; // max 3 requests
+  });
+
+  afterEach(() => {
+    delete process.env.RATE_LIMIT_WINDOW_MS;
+    delete process.env.RATE_LIMIT_MAX_REQUESTS;
+  });
+
+  it("should allow requests under the limit", () => {
+    const key = "user-1";
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(true);
+    expect(warnSpy).not.toHaveBeenCalled();
+  });
+
+  it("should block requests exceeding the limit and log a warning", () => {
+    const key = "user-2";
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(true);
+
+    // 4th request exceeds limit of 3
+    expect(checkRateLimit(key)).toBe(false);
+    expect(warnSpy).toHaveBeenCalled();
+    const callArgs = warnSpy.mock.calls[0];
+    expect(callArgs[0]).toBe("Rate limit exceeded");
+    expect(callArgs[1].key).toBe(key);
+  });
+
+  it("should reset request count after the window passes", async () => {
+    const key = "user-3";
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(true);
+    expect(checkRateLimit(key)).toBe(false); // blocked
+
+    // Wait for window to expire (100ms)
+    await new Promise((resolve) => setTimeout(resolve, 110));
+
+    // Should be allowed again
+    expect(checkRateLimit(key)).toBe(true);
+  });
+
+  it("should cleanup rate limit cache of expired keys", async () => {
+    const key1 = "cleanup-1";
+    const key2 = "cleanup-2";
+
+    // Populate keys
+    expect(checkRateLimit(key1)).toBe(true);
+    expect(checkRateLimit(key2)).toBe(true);
+
+    // Run cleanup immediately (none should be expired yet as 100ms hasn't passed)
+    cleanupRateLimitCache();
+
+    // Verify still tracked (counts shouldn't reset, e.g., if we consume remaining limits)
+    expect(checkRateLimit(key1)).toBe(true); // request 2
+    expect(checkRateLimit(key1)).toBe(true); // request 3
+    expect(checkRateLimit(key1)).toBe(false); // request 4 (blocked)
+
+    // Wait for window to expire
+    await new Promise((resolve) => setTimeout(resolve, 110));
+
+    // Run cleanup
+    cleanupRateLimitCache();
+
+    // Since they were deleted from the map, they should be initialized as new records
+    // If they were cleaned up, we should be able to do 3 requests again
+    expect(checkRateLimit(key1)).toBe(true); // 1
+    expect(checkRateLimit(key1)).toBe(true); // 2
+    expect(checkRateLimit(key1)).toBe(true); // 3
+    expect(checkRateLimit(key1)).toBe(false); // 4 (blocked)
+  });
+});