@blocklet/rate-limit 1.16.45-beta-20250630-233332-04c2bd6f → 1.16.45-beta-20250702-072658-6f6c82be

package/lib/index.js

@@ -12,6 +12,7 @@ exports.createRateLimiter = createRateLimiter;
 const db_cache_1 = require("@abtnode/db-cache");
 const express_rate_limit_1 = require("express-rate-limit");
 const node_path_1 = __importDefault(require("node:path"));
+const md5_1 = __importDefault(require("@abtnode/util/lib/md5"));
 /**
  * Store for rate limit
  */
@@ -19,7 +20,7 @@ class RateLimitStore {
     constructor(options) {
         const serverUrls = (0, db_cache_1.getAbtNodeRedisAndSQLiteUrl)();
         this.dbCache = new db_cache_1.DBCache(() => ({
-            prefix: `rate-limit-${options.prefix || 'default'}`,
+            prefix: 'rate',
             ttl: options.windowMs || 1000 * 60,
             redisUrl: options.redisUrl || serverUrls.redisUrl,
             sqlitePath: options.sqlitePath || serverUrls.sqlitePath || node_path_1.default.join(process.env.BLOCKLET_DATA_DIR, 'rate-limit.db'),
@@ -124,7 +125,15 @@ function createRateLimiter(options) {
         windowMs: 1000 * 60,
         limit: 10,
         legacyHeaders: false,
-        keyGenerator: getIp,
         ...options,
+        keyGenerator: (req, res) => {
+            const key = options.keyGenerator ? options.keyGenerator(req, res) : getIp(req);
+            if (!key)
+                return null;
+            // @ts-ignore
+            const did = req.getBlockletDid ? req.getBlockletDid() : 'blocklet';
+            const prefix = options.prefix || 'prefix';
+            return (0, md5_1.default)(`${did}:${prefix}:${key}`);
+        },
     });
 }

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.16.45-beta-20250630-233332-04c2bd6f",
+  "version": "1.16.45-beta-20250702-072658-6f6c82be",
   "description": "rate limit middleware",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
@@ -25,7 +25,8 @@
   "author": "blocklet<blocklet@arcblock.io>",
   "license": "Apache-2.0",
   "dependencies": {
-    "@abtnode/db-cache": "1.16.45-beta-20250630-233332-04c2bd6f",
+    "@abtnode/db-cache": "1.16.45-beta-20250702-072658-6f6c82be",
+    "@abtnode/util": "1.16.45-beta-20250702-072658-6f6c82be",
     "express-rate-limit": "^7.5.1"
   },
   "devDependencies": {
@@ -45,5 +46,5 @@
     "ts-node": "^10.9.1",
     "typescript": "^5.6.3"
   },
-  "gitHead": "04c65d22fbc6d064ed6192addb4e00a2d11bde3b"
+  "gitHead": "e2b7203766cce76f547833530ac0eb463752162c"
 }