From e8b1159cdac40503769fab83a57319d9e3bcc767 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Tue, 10 Jun 2025 03:55:52 +0800
Subject: [PATCH] add connection log

---
 packages/hydrooj/src/handler/connection.ts    | 36 ++++++++-----------
 .../components/message/index.page.ts          |  6 ++--
 2 files changed, 18 insertions(+), 24 deletions(-)

diff --git a/packages/hydrooj/src/handler/connection.ts b/packages/hydrooj/src/handler/connection.ts
index f9e01b6f1..8605b5ac5 100644
--- a/packages/hydrooj/src/handler/connection.ts
+++ b/packages/hydrooj/src/handler/connection.ts
@@ -13,7 +13,8 @@ class WebsocketEventsConnectionManagerHandler extends ConnectionHandler {
 
     async prepare() {
         const secret = SystemModel.get('websocket.secret');
-        if (this.request.headers['x-hydro-websocket-gateway'] === secret) {
+        if (secret && this.request.headers['x-hydro-websocket-gateway'] === secret) {
+            this.ctx.logger('connection').info('Gateway websocket connection from %s', this.request.ip);
             this.privileged = true;
             this.ctx.on('user/message', async (uid, mdoc) => {
                 await this.notifyMessage(uid, mdoc);
@@ -31,7 +32,7 @@ class WebsocketEventsConnectionManagerHandler extends ConnectionHandler {
         });
     }
 
-    async accept(channel: string) {
+    accept(channel: string) {
         if (this.channels.has(channel)) return;
         this.channels.add(channel);
         if (this.privileged) return;
@@ -45,8 +46,8 @@ class WebsocketEventsConnectionManagerHandler extends ConnectionHandler {
     }
 
     async message(payload: any) {
-        const success = [];
-        const failed = [];
+        const accept = [];
+        const reject = [];
         const session = payload?.credential
             ? await TokenModel.get(payload.credential, TokenModel.TYPE_SESSION)
             : null;
@@ -54,27 +55,18 @@ class WebsocketEventsConnectionManagerHandler extends ConnectionHandler {
         const user = await UserModel.getById('system', session.uid);
         for (const channel of payload.channels) {
             if (channel === 'message' && user.hasPriv(PRIV.PRIV_USER_PROFILE)) {
-                success.push(`message:${user._id}`);
-                // eslint-disable-next-line no-await-in-loop
-                await this.accept(`message:${user._id}`);
+                accept.push(`message:${user._id}`);
+                this.accept(`message:${user._id}`);
                 continue;
             }
-            failed.push(channel);
-        }
-        if (success.length) {
-            this.send({
-                operation: 'accept',
-                channels: success,
-                connection_id: payload.connection_id,
-            });
-        }
-        if (failed.length) {
-            this.send({
-                operation: 'reject',
-                channels: failed,
-                connection_id: payload.connection_id,
-            });
+            reject.push(channel);
         }
+        this.send({
+            operation: 'verify',
+            accept,
+            reject,
+            connection_id: payload.connection_id,
+        });
     }
 }
 
diff --git a/packages/ui-default/components/message/index.page.ts b/packages/ui-default/components/message/index.page.ts
index e6af1bb25..9a3727e85 100644
--- a/packages/ui-default/components/message/index.page.ts
+++ b/packages/ui-default/components/message/index.page.ts
@@ -93,7 +93,7 @@ const messagePage = new AutoloadPage('messagePage', (pagename) => {
   const endpoint = url.toString().replace('http', 'ws');
   if (window.SharedWorker) {
     try {
-      initWorkerMode(url);
+      initWorkerMode(endpoint);
       return;
     } catch (e) {
       console.error('SharedWorker init fail: ', e.message);
@@ -137,7 +137,9 @@ const messagePage = new AutoloadPage('messagePage', (pagename) => {
     };
     sock.onmessage = async (message) => {
       const payload = JSON.parse(message.data);
-      masterChannel.postMessage({ type: 'message', payload });
+      if (payload.operation === 'event') {
+        masterChannel.postMessage({ type: 'message', payload: payload.payload });
+      }
     };
   }