From 4b1916d32401b92e6ca8c9ebf16f62ce70f8a5e0 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Tue, 29 Mar 2022 22:45:40 +0800
Subject: [PATCH] core: support bearer token auth

---
 packages/hydrooj/package.json          | 2 +-
 packages/hydrooj/src/service/server.ts | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/hydrooj/package.json b/packages/hydrooj/package.json
index 91e9c88b9..923aa7c5e 100644
--- a/packages/hydrooj/package.json
+++ b/packages/hydrooj/package.json
@@ -1,6 +1,6 @@
 {
     "name": "hydrooj",
-    "version": "3.11.1",
+    "version": "3.11.2",
     "bin": "bin/hydrooj.js",
     "main": "src/loader",
     "module": "src/loader",
diff --git a/packages/hydrooj/src/service/server.ts b/packages/hydrooj/src/service/server.ts
index 1b7a49308..9a8ec0435 100644
--- a/packages/hydrooj/src/service/server.ts
+++ b/packages/hydrooj/src/service/server.ts
@@ -509,7 +509,10 @@ export class Handler extends HandlerCommon {
     }
 
     async getSession() {
-        const sid = this.request.cookies.get('sid');
+        const header = this.request.headers['authorization'];
+        const sid = header
+            ? header.split(' ')[1] // Just accept bearer token
+            : this.request.cookies.get('sid');
         this.session = await token.get(sid, token.TYPE_SESSION);
         if (!this.session) this.session = { uid: 0 };
     }