chore: formatted

Author: sourabpramanik

src/lua-scripts/single.ts

@@ -45,7 +45,6 @@ export const slidingWindowScript = `
   return tokens - ( newValue + requestsInPreviousWindow )
   `;
 
-
 export const tokenBucketScript = `
         local key         = KEYS[1]           -- identifier including prefixes
         local maxTokens   = tonumber(ARGV[1]) -- maximum number of tokens
@@ -84,4 +83,4 @@ export const tokenBucketScript = `
         redis.call("HSET", key, "refilledAt", refilledAt, "tokens", remaining)
         redis.call("PEXPIRE", key, expireAt)
         return {remaining, refilledAt + interval}
-       `;
+       `;

src/multi.ts

@@ -147,7 +147,7 @@ export class MultiRegionRatelimit extends Ratelimit<MultiRegionContext> {
     return members
 `;
 
-    return async function (ctx: MultiRegionContext, identifier: string) {
+    return async (ctx: MultiRegionContext, identifier: string) => {
       if (ctx.cache) {
         const { blocked, reset } = ctx.cache.isBlocked(identifier);
         if (blocked) {
@@ -281,7 +281,7 @@ export class MultiRegionRatelimit extends Ratelimit<MultiRegionContext> {
       `;
     const windowDuration = ms(window);
 
-    return async function (ctx: MultiRegionContext, identifier: string) {
+    return async (ctx: MultiRegionContext, identifier: string) => {
       // if (ctx.cache) {
       //   const { blocked, reset } = ctx.cache.isBlocked(identifier);
       //   if (blocked) {

src/ratelimit.test.ts

@@ -43,7 +43,9 @@ function run<TContext extends Context>(builder: (tc: TestCase) => Ratelimit<TCon
   for (const tc of testcases) {
     const name = `${tc.rps.toString().padStart(4, " ")}/s - Load: ${(tc.load * 100)
       .toString()
-      .padStart(3, " ")}% -> Sending ${(tc.rps * tc.load).toString().padStart(4, " ")}req/s at the rate of ${tc.rate ?? 1}`;
+      .padStart(3, " ")}% -> Sending ${(tc.rps * tc.load)
+      .toString()
+      .padStart(4, " ")}req/s at the rate of ${tc.rate ?? 1}`;
     const ratelimit = builder(tc);
 
     const limits = {
@@ -65,13 +67,13 @@ function run<TContext extends Context>(builder: (tc: TestCase) => Ratelimit<TCon
             ", blocked:",
             harness.metrics.rejected,
             "out of:",
-            harness.metrics.requests
+            harness.metrics.requests,
           );
 
           expect(harness.metrics.success).toBeLessThanOrEqual(limits.lte);
           expect(harness.metrics.success).toBeGreaterThanOrEqual(limits.gte);
         },
-        attackDuration * 1000 * 2
+        attackDuration * 1000 * 2,
       );
     });
   }
@@ -157,5 +159,7 @@ describe("slidingWindow", () => {
 
 describe("tokenBucket", () => {
   describe("region", () =>
-    run((tc) => newRegion(RegionRatelimit.tokenBucket(tc.rps, windowString, tc.rps * (tc.rate ?? 1)))));
+    run((tc) =>
+      newRegion(RegionRatelimit.tokenBucket(tc.rps, windowString, tc.rps * (tc.rate ?? 1))),
+    ));
 });

src/ratelimit.ts

@@ -129,7 +129,10 @@ export abstract class Ratelimit<TContext extends Context> {
    *  return "Yes"
    * ```
    */
-  public limit = async (identifier: string, req?: { geo?: Geo, rate?: number }): Promise<RatelimitResponse> => {
+  public limit = async (
+    identifier: string,
+    req?: { geo?: Geo; rate?: number },
+  ): Promise<RatelimitResponse> => {
     const key = [this.prefix, identifier].join(":");
     let timeoutId: any = null;
     try {

src/single.ts

@@ -1,9 +1,9 @@
 import type { Duration } from "./duration";
 import { ms } from "./duration";
+import { fixedWindowScript, slidingWindowScript, tokenBucketScript } from "./lua-scripts/single";
+import { Ratelimit } from "./ratelimit";
 import type { Algorithm, RegionContext } from "./types";
 import type { Redis } from "./types";
-import { Ratelimit } from "./ratelimit";
-import { fixedWindowScript, slidingWindowScript, tokenBucketScript } from "./lua-scripts/single";
 
 export type RegionRatelimitConfig = {
   /**
@@ -128,8 +128,7 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
     // payloadLimit?: number,
   ): Algorithm<RegionContext> {
     const windowDuration = ms(window);
-    return async function (ctx: RegionContext, identifier: string, rate?: number) {
-
+    return async (ctx: RegionContext, identifier: string, rate?: number) => {
       const bucket = Math.floor(Date.now() / windowDuration);
       const key = [identifier, bucket].join(":");
       if (ctx.cache) {
@@ -153,9 +152,9 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
         [windowDuration, incrementBy],
       )) as number;
 
-      let success = usedTokensAfterUpdate <= tokens;
+      const success = usedTokensAfterUpdate <= tokens;
 
-      let remainingTokens = Math.max(0, tokens - usedTokensAfterUpdate)
+      const remainingTokens = Math.max(0, tokens - usedTokensAfterUpdate);
 
       const reset = (bucket + 1) * windowDuration;
       if (ctx.cache && !success) {
@@ -198,9 +197,8 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
      */
     window: Duration,
   ): Algorithm<RegionContext> {
-
     const windowSize = ms(window);
-    return async function (ctx: RegionContext, identifier: string, rate?: number) {
+    return async (ctx: RegionContext, identifier: string, rate?: number) => {
       const now = Date.now();
 
       const currentWindow = Math.floor(now / windowSize);
@@ -229,7 +227,7 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
         [tokens, now, windowSize, incrementBy],
       )) as number;
 
-      let success = remainingTokens >= 0;
+      const success = remainingTokens >= 0;
 
       const reset = (currentWindow + 1) * windowSize;
       if (ctx.cache && !success) {
@@ -277,7 +275,7 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
     maxTokens: number,
   ): Algorithm<RegionContext> {
     const intervalDuration = ms(interval);
-    return async function (ctx: RegionContext, identifier: string, rate?: number) {
+    return async (ctx: RegionContext, identifier: string, rate?: number) => {
       if (ctx.cache) {
         const { blocked, reset } = ctx.cache.isBlocked(identifier);
         if (blocked) {
@@ -366,7 +364,7 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
       return r
       `;
 
-    return async function (ctx: RegionContext, identifier: string) {
+    return async (ctx: RegionContext, identifier: string) => {
       if (!ctx.cache) {
         throw new Error("This algorithm requires a cache");
       }
@@ -381,8 +379,8 @@ export class RegionRatelimit extends Ratelimit<RegionContext> {
 
         const pending = success
           ? ctx.redis.eval(script, [key], [windowDuration]).then((t) => {
-            ctx.cache!.set(key, t as number);
-          })
+              ctx.cache!.set(key, t as number);
+            })
           : Promise.resolve();
 
         return {

src/test_utils.ts

@@ -44,7 +44,7 @@ export class TestHarness<TContext extends Context> {
           this.ratelimit.limit(this.id, { rate }).then((res) => {
             this.latencies[id].end = Date.now();
             return res;
-          })
+          }),
         );
         await new Promise((r) => setTimeout(r, 500 / rps));
       }
@@ -59,7 +59,7 @@ export class TestHarness<TContext extends Context> {
         } else {
           this.metrics.rejected++;
         }
-      })
+      }),
     );
   }
 }