add rate limit

.example.env

@@ -1,4 +1,6 @@
 OPENAI_API_KEY=sk-xxx
 BILIBILI_SESSION_TOKEN=
 UPSTASH_REDIS_REST_URL=
-UPSTASH_REDIS_REST_TOKEN=
+UPSTASH_REDIS_REST_TOKEN=
+UPSTASH_RATE_REDIS_REST_URL=
+UPSTASH_RATE_REDIS_REST_TOKEN=

lib/upstash.ts

@@ -0,0 +1,13 @@
+import { Ratelimit } from "@upstash/ratelimit";
+import { Redis } from "@upstash/redis";
+import { RATE_LIMIT_COUNT } from "../utils/constants";
+
+export const ratelimit = new Ratelimit({
+  redis: new Redis({
+    url: process.env.UPSTASH_RATE_REDIS_REST_URL,
+    token: process.env.UPSTASH_RATE_REDIS_REST_TOKEN,
+  }),
+  // 速率限制算法 https://github.com/upstash/ratelimit#ratelimiting-algorithms
+  limiter: Ratelimit.fixedWindow(RATE_LIMIT_COUNT, "1 d"),
+  analytics: true, // <- Enable analytics
+});

middleware.ts

@@ -0,0 +1,42 @@
+import { Redis } from "@upstash/redis";
+import type { NextFetchEvent, NextRequest } from "next/server";
+import { NextResponse } from "next/server";
+// import { validateLicenseKey } from "./lib/lemon";
+import { checkOpenaiApiKeys } from "./lib/openai/openai";
+import { ratelimit } from "./lib/upstash"
+import { isDev } from "./utils/env";
+
+const redis = Redis.fromEnv();
+
+export async function middleware(req: NextRequest, context: NextFetchEvent) {
+  const { apiKey } = await req.json();
+//   const result = await redis.get<string>(bvId);
+//   if (!isDev && result) {
+//     console.log("hit cache for ", bvId);
+//     return NextResponse.json(result);
+//   }
+
+  // licenseKeys
+  if (apiKey) {
+    if (checkOpenaiApiKeys(apiKey)) {
+      return NextResponse.next();
+    }
+
+    // // 3. something-invalid-sdalkjfasncs-key
+    // if (!(await validateLicenseKey(apiKey, bvId))) {
+    //   return NextResponse.redirect(new URL("/shop", req.url));
+    // }
+  }
+  // TODO: unique to a user (userid, email etc) instead of IP
+  const identifier = req.ip ?? "127.0.0.7";
+  const { success, remaining } = await ratelimit.limit("trans-" + identifier);
+  console.log(`======== ip ${identifier}, remaining: ${remaining} ========`);
+  if (!apiKey && !success) {
+    // return NextResponse.redirect(new URL("/shop", req.url));
+    return NextResponse.error();
+  }
+}
+
+export const config = {
+  matcher: "/api/translate",
+};

package.json

@@ -13,6 +13,7 @@
     "@types/node": "18.14.4",
     "@types/react": "18.0.28",
     "@types/react-dom": "18.0.11",
+    "@upstash/ratelimit": "^0.3.9",
     "@upstash/redis": "^1.20.1",
     "detect-file-encoding-and-language": "^2.3.2",
     "eslint": "8.35.0",