@longzai-intelligence-auth/rate-limit 0.0.1

package/dist/index.d.ts

@@ -0,0 +1,15 @@
+import type { RateLimitConfig } from "@longzai-intelligence-auth/core";
+type MemoryRateLimitRecord = {
+    count: number;
+    windowStart: number;
+};
+type MemoryRateLimiter = {
+    check(key: string): Promise<boolean>;
+    getCount(key: string): number;
+    getResetTime(key: string): Date | null;
+    cleanup(): void;
+    getConfig(): RateLimitConfig;
+};
+declare function createMemoryRateLimiter(config?: RateLimitConfig): MemoryRateLimiter;
+export type { MemoryRateLimitRecord, MemoryRateLimiter };
+export { createMemoryRateLimiter };

package/dist/index.js

@@ -0,0 +1,56 @@
+const DEFAULT_RATE_LIMIT_CONFIG = {
+    windowSeconds: 60,
+    maxRequests: 100,
+};
+function createMemoryRateLimiter(config = DEFAULT_RATE_LIMIT_CONFIG) {
+    const records = new Map();
+    return {
+        async check(key) {
+            const now = Date.now();
+            const windowMs = config.windowSeconds * 1000;
+            const record = records.get(key);
+            if (!record || now - record.windowStart >= windowMs) {
+                records.set(key, {
+                    count: 1,
+                    windowStart: now,
+                });
+                return true;
+            }
+            if (record.count >= config.maxRequests) {
+                return false;
+            }
+            record.count++;
+            return true;
+        },
+        getCount(key) {
+            const now = Date.now();
+            const windowMs = config.windowSeconds * 1000;
+            const record = records.get(key);
+            if (!record || now - record.windowStart >= windowMs) {
+                return 0;
+            }
+            return record.count;
+        },
+        getResetTime(key) {
+            const record = records.get(key);
+            if (!record) {
+                return null;
+            }
+            const windowMs = config.windowSeconds * 1000;
+            return new Date(record.windowStart + windowMs);
+        },
+        cleanup() {
+            const now = Date.now();
+            const windowMs = config.windowSeconds * 1000;
+            for (const [key, record] of records.entries()) {
+                if (now - record.windowStart >= windowMs) {
+                    records.delete(key);
+                }
+            }
+        },
+        getConfig() {
+            return config;
+        },
+    };
+}
+export { createMemoryRateLimiter };

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "@longzai-intelligence-auth/rate-limit",
+  "version": "0.0.1",
+  "license": "MIT",
+  "type": "module",
+  "sideEffects": false,
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/longzai/longzai-intelligence-auth",
+    "directory": "packages/rate-limit"
+  },
+  "dependencies": {
+    "@longzai-intelligence-auth/core": "0.0.1"
+  },
+  "scripts": {
+    "build": "tsgo --build",
+    "typecheck": "tsgo --noEmit",
+    "lint": "oxlint && oxfmt --check",
+    "lint:fix": "oxlint --fix && oxfmt",
+    "test": "bun test",
+    "clean": "rm -rf dist out .cache"
+  }
+}