@upstash/ratelimit 2.0.8 → 2.1.0-rc

package/dist/index.mjs

@@ -156,7 +156,7 @@ var safeEval = async (ctx, script, keys, args) => {
 };
 
 // src/lua-scripts/single.ts
-var fixedWindowLimitScript = `
+var fixedWindowLimitScript = `#!lua flags=allow-key-locking
   local key           = KEYS[1]
   local dynamicLimitKey = KEYS[2]  -- optional: key for dynamic limit in redis
   local tokens        = tonumber(ARGV[1])  -- default limit
@@ -181,7 +181,7 @@ var fixedWindowLimitScript = `
 
   return {r, effectiveLimit}
 `;
-var fixedWindowRemainingTokensScript = `
+var fixedWindowRemainingTokensScript = `#!lua flags=allow-key-locking
   local key = KEYS[1]
   local dynamicLimitKey = KEYS[2]  -- optional: key for dynamic limit in redis
   local tokens = tonumber(ARGV[1])  -- default limit
@@ -203,7 +203,7 @@ var fixedWindowRemainingTokensScript = `
   
   return {effectiveLimit - usedTokens, effectiveLimit}
 `;
-var slidingWindowLimitScript = `
+var slidingWindowLimitScript = `#!lua flags=allow-key-locking
   local currentKey  = KEYS[1]           -- identifier including prefixes
   local previousKey = KEYS[2]           -- key of the previous bucket
   local dynamicLimitKey = KEYS[3]       -- optional: key for dynamic limit in redis
@@ -247,7 +247,7 @@ var slidingWindowLimitScript = `
   end
   return {effectiveLimit - ( newValue + requestsInPreviousWindow ), effectiveLimit}
 `;
-var slidingWindowRemainingTokensScript = `
+var slidingWindowRemainingTokensScript = `#!lua flags=allow-key-locking
   local currentKey  = KEYS[1]           -- identifier including prefixes
   local previousKey = KEYS[2]           -- key of the previous bucket
   local dynamicLimitKey = KEYS[3]       -- optional: key for dynamic limit in redis
@@ -281,7 +281,7 @@ var slidingWindowRemainingTokensScript = `
   local usedTokens = requestsInPreviousWindow + requestsInCurrentWindow
   return {effectiveLimit - usedTokens, effectiveLimit}
 `;
-var tokenBucketLimitScript = `
+var tokenBucketLimitScript = `#!lua flags=allow-key-locking
   local key         = KEYS[1]           -- identifier including prefixes
   local dynamicLimitKey = KEYS[2]       -- optional: key for dynamic limit in redis
   local maxTokens   = tonumber(ARGV[1]) -- default maximum number of tokens
@@ -335,7 +335,7 @@ var tokenBucketLimitScript = `
   return {remaining, refilledAt + interval, effectiveLimit}
 `;
 var tokenBucketIdentifierNotFound = -1;
-var tokenBucketRemainingTokensScript = `
+var tokenBucketRemainingTokensScript = `#!lua flags=allow-key-locking
   local key         = KEYS[1]
   local dynamicLimitKey = KEYS[2]       -- optional: key for dynamic limit in redis
   local maxTokens   = tonumber(ARGV[1]) -- default maximum number of tokens
@@ -357,7 +357,7 @@ var tokenBucketRemainingTokensScript = `
         
   return {tonumber(bucket[2]), tonumber(bucket[1]), effectiveLimit}
 `;
-var cachedFixedWindowLimitScript = `
+var cachedFixedWindowLimitScript = `#!lua flags=allow-key-locking
   local key     = KEYS[1]
   local window  = ARGV[1]
   local incrementBy   = ARGV[2] -- increment rate per request at a given value, default is 1
@@ -371,7 +371,7 @@ var cachedFixedWindowLimitScript = `
       
   return r
 `;
-var cachedFixedWindowRemainingTokenScript = `
+var cachedFixedWindowRemainingTokenScript = `#!lua flags=allow-key-locking
   local key = KEYS[1]
   local tokens = 0
 
@@ -383,7 +383,7 @@ var cachedFixedWindowRemainingTokenScript = `
 `;
 
 // src/lua-scripts/multi.ts
-var fixedWindowLimitScript2 = `
+var fixedWindowLimitScript2 = `#!lua flags=allow-key-locking
 	local key           = KEYS[1]
 	local id            = ARGV[1]
 	local window        = ARGV[2]
@@ -399,7 +399,7 @@ var fixedWindowLimitScript2 = `
 
 	return fields
 `;
-var fixedWindowRemainingTokensScript2 = `
+var fixedWindowRemainingTokensScript2 = `#!lua flags=allow-key-locking
       local key = KEYS[1]
       local tokens = 0
 
@@ -407,7 +407,7 @@ var fixedWindowRemainingTokensScript2 = `
 
       return fields
     `;
-var slidingWindowLimitScript2 = `
+var slidingWindowLimitScript2 = `#!lua flags=allow-key-locking
 	local currentKey    = KEYS[1]           -- identifier including prefixes
 	local previousKey   = KEYS[2]           -- key of the previous bucket
 	local tokens        = tonumber(ARGV[1]) -- tokens per window
@@ -444,7 +444,7 @@ var slidingWindowLimitScript2 = `
 	end
 	return {currentFields, previousFields, true}
 `;
-var slidingWindowRemainingTokensScript2 = `
+var slidingWindowRemainingTokensScript2 = `#!lua flags=allow-key-locking
 	local currentKey    = KEYS[1]           -- identifier including prefixes
 	local previousKey   = KEYS[2]           -- key of the previous bucket
 	local now         	= ARGV[1]           -- current timestamp in milliseconds
@@ -499,41 +499,41 @@ var SCRIPTS = {
     fixedWindow: {
       limit: {
         script: fixedWindowLimitScript,
-        hash: "472e55443b62f60d0991028456c57815a387066d"
+        hash: "4ef8749b9e927b157546ebad13061fa86b9ffc2e"
       },
       getRemaining: {
         script: fixedWindowRemainingTokensScript,
-        hash: "40515c9dd0a08f8584f5f9b593935f6a87c1c1c3"
+        hash: "e62252cb05b4676b27a5d396da02f0ca0d375d88"
       }
     },
     slidingWindow: {
       limit: {
         script: slidingWindowLimitScript,
-        hash: "977fb636fb5ceb7e98a96d1b3a1272ba018efdae"
+        hash: "5cd9665be7533e4dfabb3581021737fc69b41ae8"
       },
       getRemaining: {
         script: slidingWindowRemainingTokensScript,
-        hash: "ee3a3265fad822f83acad23f8a1e2f5c0b156b03"
+        hash: "1bdbbdb2082bb557084518c64486d7d5a29d1bfe"
       }
     },
     tokenBucket: {
       limit: {
         script: tokenBucketLimitScript,
-        hash: "b35c5bc0b7fdae7dd0573d4529911cabaf9d1d89"
+        hash: "8f7286b9b0f65d631f760ba615f7b3c598a569a3"
       },
       getRemaining: {
         script: tokenBucketRemainingTokensScript,
-        hash: "deb03663e8af5a968deee895dd081be553d2611b"
+        hash: "08d3b3381c507b6f7bfec1bbfa0a6053434b16bd"
       }
     },
     cachedFixedWindow: {
       limit: {
         script: cachedFixedWindowLimitScript,
-        hash: "c26b12703dd137939b9a69a3a9b18e906a2d940f"
+        hash: "1861a700bafe96c833a483af6b1c28a8897cfdb0"
       },
       getRemaining: {
         script: cachedFixedWindowRemainingTokenScript,
-        hash: "8e8f222ccae68b595ee6e3f3bf2199629a62b91a"
+        hash: "eb82f0e853d2fc9a236fa9bfbe704fda6ac2fc36"
       }
     }
   },
@@ -541,21 +541,21 @@ var SCRIPTS = {
     fixedWindow: {
       limit: {
         script: fixedWindowLimitScript2,
-        hash: "a8c14f3835aa87bd70e5e2116081b81664abcf5c"
+        hash: "e04b753a75909b7f99aae4c04cf8869a8de02a9e"
       },
       getRemaining: {
         script: fixedWindowRemainingTokensScript2,
-        hash: "8ab8322d0ed5fe5ac8eb08f0c2e4557f1b4816fd"
+        hash: "e066c8ce3eaca142b0894ad1541e0a58c9924819"
       }
     },
     slidingWindow: {
       limit: {
         script: slidingWindowLimitScript2,
-        hash: "1e7ca8dcd2d600a6d0124a67a57ea225ed62921b"
+        hash: "aeb4d8f381e8dafc8e69041baae1324d254ded44"
       },
       getRemaining: {
         script: slidingWindowRemainingTokensScript2,
-        hash: "558c9306b7ec54abb50747fe0b17e5d44bd24868"
+        hash: "87828c1088f6a8d1f512a434ea330189982a1c0a"
       }
     }
   }
@@ -1036,6 +1036,27 @@ var Ratelimit = class {
   };
 };
 
+// src/version.ts
+var VERSION = "2.1.0-rc";
+
+// src/telemetry.ts
+var taggedClients = /* @__PURE__ */ new WeakSet();
+var getSafeEnv = () => typeof process === "object" && process && typeof process.env === "object" ? process.env : {};
+var addTelemetry = (redis, enableTelemetry = true) => {
+  if (!enableTelemetry || getSafeEnv().UPSTASH_DISABLE_TELEMETRY)
+    return;
+  if (!redis || typeof redis !== "object")
+    return;
+  if (taggedClients.has(redis))
+    return;
+  taggedClients.add(redis);
+  try {
+    const client = redis;
+    client.addTelemetry?.({ sdk: `@upstash/ratelimit@${VERSION}` });
+  } catch {
+  }
+};
+
 // src/multi.ts
 function randomId() {
   let result = "";
@@ -1065,6 +1086,9 @@ var MultiRegionRatelimit = class extends Ratelimit {
         cache: config.ephemeralCache ? new Cache(config.ephemeralCache) : void 0
       }
     });
+    for (const redis of config.redis) {
+      addTelemetry(redis, config.enableTelemetry);
+    }
     if (config.dynamicLimits) {
       console.warn(
         "Warning: Dynamic limits are not yet supported for multi-region rate limiters. The dynamicLimits option will be ignored."
@@ -1435,6 +1459,7 @@ var RegionRatelimit = class extends Ratelimit {
       denyListThreshold: config.denyListThreshold,
       dynamicLimits: config.dynamicLimits
     });
+    addTelemetry(config.redis, config.enableTelemetry);
   }
   /**
    * Each request inside a fixed time increases a counter.