Make luogu url configurable (#29)

immccn123 · web-flow · commit 5cdd85486b97 · 2024-10-06T22:50:24.000+08:00
diff --git a/packages/archive/package.json b/packages/archive/package.json
@@ -19,6 +19,7 @@
     "@logtail/pino": "^0.4.22",
     "@prisma/client": "^5.15.0",
     "@socket.io/admin-ui": "^0.5.1",
+    "dotenv": "^16.4.5",
     "fastify": "^4.28.0",
     "fastify-plugin": "^4.5.1",
     "jsdom": "^22.1.0",
diff --git a/packages/archive/src/lib/activity.ts b/packages/archive/src/lib/activity.ts
@@ -5,6 +5,7 @@ import type { PrismaClient, PrismaPromise } from "@prisma/client";
 import { getResponse } from "./parser";
 import type { UserSummary } from "./user";
 import { upsertUserSnapshot } from "./user";
+import lgUrl from "../utils/url";
 
 export interface Activity {
   content: string;
@@ -30,7 +31,7 @@ export async function saveActivityPage(
 ) {
   const res = await getResponse(
     logger,
-    `https://www.luogu.com.cn/api/feed/list?page=${page}`,
+    lgUrl(`/api/feed/list?page=${page}`, false),
     false,
   ).then((response): Promise<Body> => response.json());
 
diff --git a/packages/archive/src/lib/judgement.ts b/packages/archive/src/lib/judgement.ts
@@ -2,6 +2,7 @@ import type { PrismaClient, PrismaPromise } from "@prisma/client";
 import type { BaseLogger } from "pino";
 import { getResponse } from "./parser";
 import { UserSummary, upsertUserSnapshot } from "./user";
+import lgUrl from "../utils/url";
 
 interface JudgementBody {
   user: UserSummary;
@@ -23,7 +24,7 @@ export default async function saveJudgements(
 ) {
   const res = await getResponse(
     logger,
-    `https://www.luogu.com.cn/judgement?_contentOnly`,
+    lgUrl(`/judgement?_contentOnly`, false),
     false,
   ).then((response): Promise<JudgementResponse> => response.json());
 
diff --git a/packages/archive/src/lib/list.ts b/packages/archive/src/lib/list.ts
@@ -3,6 +3,7 @@ import type { PrismaClient } from "@prisma/client";
 import type { UserSummary } from "./user";
 import type { ForumData, ReplyContent } from "./post";
 import { getResponse } from "./parser";
+import lgUrl from "../utils/url";
 
 interface PostData {
   id: number;
@@ -40,7 +41,7 @@ export default async function getPostList(
 ) {
   const response = await getResponse(
     logger,
-    `https://www.luogu.com/discuss?_contentOnly&page=${page}`,
+    lgUrl(`/discuss?_contentOnly&page=${page}`),
     false,
   );
   const {
diff --git a/packages/archive/src/lib/paste.ts b/packages/archive/src/lib/paste.ts
@@ -3,6 +3,7 @@ import type { PrismaClient } from "@prisma/client";
 import { getResponse } from "./parser";
 import { type UserSummary } from "./user";
 import { upsertUserSnapshot } from "./user";
+import lgUrl from "../utils/url";
 
 interface Paste {
   data: string;
@@ -25,7 +26,7 @@ export default async function savePaste(
 ) {
   const response = await getResponse(
     logger,
-    `https://www.luogu.com.cn/paste/${id}?_contentOnly`,
+    lgUrl(`/paste/${id}?_contentOnly`, false),
   );
   const json = (await response.json()) as
     | { code: 403 | 404; currentData: LuoguError }
diff --git a/packages/archive/src/lib/post.ts b/packages/archive/src/lib/post.ts
@@ -5,6 +5,7 @@ import type { PostSnapshot, PrismaClient } from "@prisma/client";
 import { getResponse } from "./parser";
 import type { ServerToClientEvents } from "../plugins/socket.io";
 import { UserSummary, upsertUserSnapshot } from "./user";
+import lgUrl from "../utils/url";
 
 const PAGES_PER_SAVE = parseInt(process.env.PAGES_PER_SAVE ?? "64", 10);
 export const emitters: Record<number, EventEmitter> = {};
@@ -65,7 +66,7 @@ export async function savePost(
   const fetchPage = (page: number) =>
     getResponse(
       logger,
-      `https://www.luogu.com/discuss/${id}?_contentOnly&page=${page}`,
+      lgUrl(`/discuss/${id}?_contentOnly&page=${page}`),
       false,
     ).then((response): Promise<ResponseBody> => response.json());
 
diff --git a/packages/archive/src/server.ts b/packages/archive/src/server.ts
@@ -6,6 +6,8 @@ import routes from "./plugins/routes";
 import io from "./plugins/socket.io";
 import cron from "./plugins/cron";
 
+import "dotenv/config";
+
 const fastify = Fastify({
   logger: process.env.SOURCE_TOKEN
     ? pino(
diff --git a/packages/archive/src/utils/url.ts b/packages/archive/src/utils/url.ts
@@ -0,0 +1,7 @@
+const backendGlobal = process.env.LUOGU_URL ?? "https://www.luogu.com";
+const backendCN = process.env.LUOGU_CN_URL ?? "https://www.luogu.com.cn";
+
+export default function lgUrl(path: string, global = true): string {
+  if (global) return `${backendGlobal}${path}`;
+  return `${backendCN}${path}`;
+}
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
