From 7bc843d440212d54c9a773fac16e40acc77652c0 Mon Sep 17 00:00:00 2001
From: CrescentLeaf <crescentleaf@noreply.codeberg.org>
Date: Fri, 31 Oct 2025 21:55:54 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E9=80=9A=E8=BF=87=E8=AE=BE=E7=BD=AE=20?=
 =?UTF-8?q?token=20=E5=88=B0=20Headers=20=E8=8E=B7=E5=8F=96=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/main.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/server/main.ts b/server/main.ts
index 75c919d..b9eea0a 100644
--- a/server/main.ts
+++ b/server/main.ts
@@ -35,8 +35,8 @@ app.get('/uploaded_files/:hash', (req, res) => {
     }
 
     if (file.getChatId() != null) {
-        const userToken = TokenManager.decode(req.cookies.token)
-        if (!TokenManager.checkToken(userToken, req.cookies.device_id)) {
+        const userToken = TokenManager.decode(req.headers['token'] || req.cookies.token)
+        if (!TokenManager.checkToken(userToken, req.headers['device_id'] || req.cookies.device_id)) {
             res.status(401).send("401 UnAuthorized")
             return
         }