support config proxy

package.json

@@ -53,9 +53,11 @@
     "express-validator": "^7.0.1",
     "got": "^13.0.0",
     "http-errors": "^2.0.0",
+    "https-proxy-agent": "^7.0.1",
     "node-html-parser": "^6.1.6",
     "probe-image-size": "^7.2.3",
-    "qs": "^6.11.2"
+    "qs": "^6.11.2",
+    "socks-proxy-agent": "^8.0.1"
   },
   "type": "module"
 }

src/env.d.ts

@@ -3,5 +3,7 @@ declare namespace NodeJS {
     PORT?: string;
     SSL_CERT?: string;
     SSL_KEY?: string;
+    HTTP_PROXY?: string;
+    HTTPS_PROXY?: string;
   }
 }

src/router/v1/client.ts

@@ -0,0 +1,34 @@
+import { Agent as HttpsAgent } from 'https';
+import { HttpsProxyAgent } from 'https-proxy-agent';
+import { SocksProxyAgent } from 'socks-proxy-agent';
+import got, { type ExtendOptions } from 'got';
+import { JAVBUS_TIMEOUT, USER_AGENT } from './constants.js';
+
+const PROXY_URL = process.env.HTTP_PROXY || process.env.HTTPS_PROXY;
+
+export let agent: HttpsAgent | undefined = undefined;
+
+if (PROXY_URL) {
+  if (/^https?:\/\//.test(PROXY_URL)) {
+    agent = new HttpsProxyAgent(PROXY_URL);
+  } else if (/^socks/.test(PROXY_URL)) {
+    agent = new SocksProxyAgent(PROXY_URL);
+  }
+}
+
+const extendOptions: ExtendOptions = {
+  headers: {
+    'User-Agent': USER_AGENT,
+  },
+  timeout: {
+    request: JAVBUS_TIMEOUT,
+  },
+};
+
+if (agent) {
+  extendOptions.agent = { http: agent, https: agent };
+}
+
+const client = got.extend(extendOptions);
+
+export default client;

src/router/v1/javbusParser.ts

@@ -1,9 +1,9 @@
-import got from 'got';
 import bytes from 'bytes';
 import { parse } from 'node-html-parser';
 import type { HTMLElement } from 'node-html-parser';
 import probe from 'probe-image-size';
-import { JAVBUS_TIMEOUT, JAVBUS, USER_AGENT } from './constants.js';
+import client, { agent } from './client.js';
+import { JAVBUS } from './constants.js';
 import type {
   FilterType,
   ImageSize,
@@ -24,15 +24,6 @@ type StarInfoRequiredKey = 'avatar' | 'id' | 'name';
 
 type StarInfoOptionalKey = Exclude<keyof StarInfo, StarInfoRequiredKey>;
 
-const client = got.extend({
-  headers: {
-    'User-Agent': USER_AGENT,
-  },
-  timeout: {
-    request: JAVBUS_TIMEOUT,
-  },
-});
-
 const starInfoMap: Record<StarInfoOptionalKey, string> = {
   birthday: '生日: ',
   age: '年齡: ',
@@ -291,7 +282,7 @@ export async function getMovieDetail(id: string): Promise<MovieDetail> {
   let imageSize: ImageSize | null = null;
 
   if (img) {
-    const { width, height } = await probe(img);
+    const { width, height } = await probe(img, { agent });
     imageSize = { width, height };
   }