From 382476cb36df7cc2878d84f39a5863cfaea1303d Mon Sep 17 00:00:00 2001
From: Nicolas <nicolascamara29@gmail.com>
Date: Sat, 18 Jan 2025 23:16:25 -0300
Subject: [PATCH] Nick: auth extract

---
 apps/api/src/controllers/auth.ts      | 4 +---
 apps/api/src/routes/v1.ts             | 2 +-
 apps/api/src/services/rate-limiter.ts | 4 ++++
 apps/api/src/types.ts                 | 1 +
 4 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/apps/api/src/controllers/auth.ts b/apps/api/src/controllers/auth.ts
index d87f59d6..c9ff86b0 100644
--- a/apps/api/src/controllers/auth.ts
+++ b/apps/api/src/controllers/auth.ts
@@ -94,7 +94,7 @@ export async function getACUC(
     let retries = 0;
     const maxRetries = 5;
 
-    let rpcName = mode === RateLimiterMode.Extract ? "auth_credit_usage_chunk_extract" : "auth_credit_usage_chunk_test_21_credit_pack_n_extract";
+    let rpcName = mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus ? "auth_credit_usage_chunk_extract" : "auth_credit_usage_chunk_test_21_credit_pack_n_extract";
     while (retries < maxRetries) {
       ({ data, error } = await supabase_service.rpc(
         rpcName,
@@ -129,8 +129,6 @@ export async function getACUC(
       setCachedACUC(api_key, chunk);
     }
 
-    // console.log(chunk);
-
     return chunk;
   } else {
     return null;
diff --git a/apps/api/src/routes/v1.ts b/apps/api/src/routes/v1.ts
index 4aacfe18..f92fe024 100644
--- a/apps/api/src/routes/v1.ts
+++ b/apps/api/src/routes/v1.ts
@@ -228,7 +228,7 @@ v1Router.post(
 
 v1Router.get(
   "/extract/:jobId",
-  authMiddleware(RateLimiterMode.CrawlStatus),
+  authMiddleware(RateLimiterMode.ExtractStatus),
   wrap(extractStatusController),
 );
 
diff --git a/apps/api/src/services/rate-limiter.ts b/apps/api/src/services/rate-limiter.ts
index 4631fcfc..09c8d749 100644
--- a/apps/api/src/services/rate-limiter.ts
+++ b/apps/api/src/services/rate-limiter.ts
@@ -100,6 +100,10 @@ const RATE_LIMITS = {
     free: 500,
     default: 5000,
   },
+  extractStatus: {
+    free: 500,
+    default: 5000,
+  },
   testSuite: {
     free: 10000,
     default: 10000,
diff --git a/apps/api/src/types.ts b/apps/api/src/types.ts
index d1e163ef..ce535ede 100644
--- a/apps/api/src/types.ts
+++ b/apps/api/src/types.ts
@@ -134,6 +134,7 @@ export enum RateLimiterMode {
   Search = "search",
   Map = "map",
   Extract = "extract",
+  ExtractStatus = "extractStatus",
 }
 
 export type AuthResponse =