From 6664deb3cf3fed5ec4dd393194af59a39a753a11 Mon Sep 17 00:00:00 2001
From: sunlei <yemu2333@outlook.com>
Date: Sat, 16 May 2026 11:43:27 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E6=88=AA=E5=9B=BE?=
 =?UTF-8?q?=E8=B5=84=E6=BA=90=E4=BB=A3=E7=90=86=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/minio/minio.controller.ts | 86 +++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/src/minio/minio.controller.ts b/src/minio/minio.controller.ts
index 82c2fff..128cbe6 100644
--- a/src/minio/minio.controller.ts
+++ b/src/minio/minio.controller.ts
@@ -1,4 +1,5 @@
 import {
+  BadRequestException,
   Controller,
   Body,
   Delete,
@@ -34,6 +35,18 @@ import {
   MinioUploadResultDto,
 } from './minio.dto';
 
+const PROXY_RESOURCE_TIMEOUT = 1000 * 15;
+const PROXY_RESOURCE_CONTENT_TYPES = [
+  'image/',
+  'font/',
+  'text/css',
+  'application/font',
+  'application/x-font',
+  'application/vnd.ms-fontobject',
+];
+const PROXY_RESOURCE_EXTENSION_RE =
+  /\.(avif|bmp|css|eot|gif|ico|jpe?g|otf|png|svg|ttf|webp|woff2?)(?:[?#].*)?$/i;
+
 @Controller('minio')
 @ApiTags('minio')
 export class MinioClientController {
@@ -172,6 +185,47 @@ export class MinioClientController {
     res.send(this.toolsService.res(HttpStatus.OK, '操作成功', result));
   }
 
+  @Get('resource-proxy')
+  @ApiOperation({ summary: '代理截图所需的图片/CSS/字体资源' })
+  @ApiQuery({ name: 'url' })
+  async proxyResource(@Res() res: Response, @Query('url') url: string) {
+    const target = this.getProxyResourceUrl(url);
+    const controller = new AbortController();
+    const timer = setTimeout(() => controller.abort(), PROXY_RESOURCE_TIMEOUT);
+
+    try {
+      const response = await fetch(target, {
+        redirect: 'follow',
+        signal: controller.signal,
+      });
+
+      if (!response.ok) {
+        throw new BadRequestException(`资源代理失败：${response.status}`);
+      }
+
+      const contentType =
+        response.headers.get('content-type') || 'application/octet-stream';
+
+      if (!this.isAllowedProxyResource(contentType, target)) {
+        throw new BadRequestException('仅支持代理图片、CSS 和字体资源');
+      }
+
+      const data = Buffer.from(await response.arrayBuffer());
+
+      res.setHeader('Content-Type', contentType);
+      res.setHeader('Cache-Control', 'public, max-age=300');
+      res.send(data);
+    } catch (err) {
+      if (err instanceof BadRequestException) {
+        throw err;
+      }
+
+      throw new BadRequestException('资源代理失败');
+    } finally {
+      clearTimeout(timer);
+    }
+  }
+
   @Get('download')
   @ApiOperation({ summary: '下载MinIO文件' })
   @ApiQuery({ name: 'objectName' })
@@ -223,4 +277,36 @@ export class MinioClientController {
 
     res.send(this.toolsService.res(HttpStatus.OK, '操作成功', result));
   }
+
+  private getProxyResourceUrl(url: string) {
+    if (!url) {
+      throw new BadRequestException('资源地址不能为空');
+    }
+
+    try {
+      const target = new URL(url);
+
+      if (!['http:', 'https:'].includes(target.protocol)) {
+        throw new BadRequestException('仅支持 http/https 资源');
+      }
+
+      return target.toString();
+    } catch (err) {
+      if (err instanceof BadRequestException) {
+        throw err;
+      }
+
+      throw new BadRequestException('资源地址不合法');
+    }
+  }
+
+  private isAllowedProxyResource(contentType: string, target: string) {
+    const normalizedType = contentType.split(';')[0].trim().toLowerCase();
+
+    return (
+      PROXY_RESOURCE_CONTENT_TYPES.some((type) =>
+        normalizedType.startsWith(type),
+      ) || PROXY_RESOURCE_EXTENSION_RE.test(target)
+    );
+  }
 }