feat: added USER_LICENSE_KEYS

JimmyLv · Mar 2, 2023 · 58ee334 · 58ee334 · vercel · Mar 2, 2023
1 parent b20f6ce
commit 58ee334
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.example.env b/.example.env
@@ -1,3 +1,4 @@
 OPENAI_API_KEY=
 UPSTASH_REDIS_REST_URL=
 UPSTASH_REDIS_REST_TOKEN=
+USER_LICENSE_KEYS=
diff --git a/middleware.ts b/middleware.ts
@@ -16,6 +16,15 @@ export async function middleware(req: NextRequest, ev: NextFetchEvent) {
   const identifier = req.ip ?? "127.0.0.3";
   const { success, remaining } = await ratelimit.limit(identifier);
   console.log(`======== ip ${identifier}, remaining: ${remaining} ========`);
+
+  // note: not forgot to set USER_LICENSE_KEYS env var
+  if (process.env.USER_LICENSE_KEYS?.includes(apiKey)) {
+    const { remaining } = await ratelimit.limit(apiKey);
+    if (remaining === 0) {
+      return NextResponse.redirect(new URL("/blocked", req.url));
+    }
+  }
+
   if (!apiKey && !success) {
     return NextResponse.redirect(new URL("/blocked", req.url));
   }