From 5b3c75b06e3756bfc09a469ee9f029582bbc16c7 Mon Sep 17 00:00:00 2001
From: Nicolas <nicolascamara29@gmail.com>
Date: Sat, 20 Apr 2024 14:10:29 -0700
Subject: [PATCH] Nick:

---
 apps/api/src/index.ts                 |  2 +-
 apps/api/src/services/rate-limiter.ts | 13 +++++++------
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/apps/api/src/index.ts b/apps/api/src/index.ts
index 271d96d..0fbd91e 100644
--- a/apps/api/src/index.ts
+++ b/apps/api/src/index.ts
@@ -70,7 +70,7 @@ async function authenticateUser(req, res, mode?: RateLimiterMode): Promise<{ suc
     return { success: false, error: "Rate limit exceeded. Too many requests, try again in 1 minute.", status: 429 };
   }
 
-  if (token === "this_is_just_a_preview_token" && mode === "scrape") {
+  if (token === "this_is_just_a_preview_token" && (mode === RateLimiterMode.Scrape || mode === RateLimiterMode.Preview)) {
     return { success: true, team_id: "preview" };
   }
 
diff --git a/apps/api/src/services/rate-limiter.ts b/apps/api/src/services/rate-limiter.ts
index dcd05da..b1ee562 100644
--- a/apps/api/src/services/rate-limiter.ts
+++ b/apps/api/src/services/rate-limiter.ts
@@ -70,11 +70,12 @@ export function crawlRateLimit(plan: string){
 
 
 export function getRateLimiter(mode: RateLimiterMode){
-  if(mode === RateLimiterMode.Preview){
-    return previewRateLimiter;
-  }else if(mode === RateLimiterMode.CrawlStatus){
-    return crawlStatusRateLimiter;
-  }else{
-    return serverRateLimiter;
+  switch(mode) {
+    case RateLimiterMode.Preview:
+      return previewRateLimiter;
+    case RateLimiterMode.CrawlStatus:
+      return crawlStatusRateLimiter;
+    default:
+      return serverRateLimiter;
   }
 }