From c168a1da0c0becb5caa01cef69710b0867940761 Mon Sep 17 00:00:00 2001
From: kubbo <390378816@qq.com>
Date: Thu, 16 Feb 2023 12:01:46 +0800
Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=96=B0=E9=80=82=E9=85=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/websocket/custom/getZip.js | 2 +-
 src/websocket/index.js         | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/websocket/custom/getZip.js b/src/websocket/custom/getZip.js
index b8ce7d4..b3498ba 100644
--- a/src/websocket/custom/getZip.js
+++ b/src/websocket/custom/getZip.js
@@ -72,7 +72,7 @@ const handleZip = (id, ws) => {
 
 module.exports = {
   action: "/custom/getZip", execute: (ws, request) => {
-    const {id, cid} = request.params
+    const {id, cid} = request
     if (cid) {
       controller.abort()
       controller = new AbortController()
diff --git a/src/websocket/index.js b/src/websocket/index.js
index 290a06e..ed58891 100644
--- a/src/websocket/index.js
+++ b/src/websocket/index.js
@@ -18,8 +18,9 @@ module.exports = {
           actions[ws.action] = ws.execute
         }
       }).then(() => ws.on('message', res => {
-        if (res?.action) {
-          actions[res.action]?.(ws, JSON.parse(res))
+        if (res?.data) {
+          const data = JSON.parse(res.data)
+          !!data?.action && actions[res.action]?.(ws, data)
         }
       }))
     })