From 43ef5575b931d3c54c99a44826b4f9936d5a1d57 Mon Sep 17 00:00:00 2001
From: Yangcheng He <107044023+hyc1230@users.noreply.github.com>
Date: Sat, 23 Dec 2023 14:35:11 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=AD=E5=BC=80=E8=BF=9E=E6=8E=A5=20?=
 =?UTF-8?q?/=20=E9=95=BF=E6=97=B6=E9=97=B4=E6=9C=AA=E6=8E=A5=E6=94=B6?=
 =?UTF-8?q?=E5=88=B0=E6=B6=88=E6=81=AF=E8=87=AA=E5=8A=A8=E9=87=8D=E8=BF=9E?=
 =?UTF-8?q?=20(#1)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 index.cjs | 57 ++++++++++++++++++++++++++++++++++++++-----------------
 1 file changed, 40 insertions(+), 17 deletions(-)

diff --git a/index.cjs b/index.cjs
index 3395e0b..c147d98 100644
--- a/index.cjs
+++ b/index.cjs
@@ -7,9 +7,6 @@ const dl = require("download");
 const opn = require("opn");
 
 const rli = rl.createInterface({input: process.stdin, output: process.stdout});
-rli.on("close", () => {
-    process.exit();
-});
 
 let cache = {};
 if (fs.existsSync("cache/cache.json")) {
@@ -28,6 +25,7 @@ async function notify(msg) {
         cache[uid] = new Date().getTime();
         fs.writeFileSync("cache/cache.json", JSON.stringify(cache));
     }
+    console.log("Send notification");
     notifier.notify(
         {
             title: `来自 ${msg.sender.name} 的洛谷私信`,
@@ -38,6 +36,7 @@ async function notify(msg) {
         },
         (er, rp, md) => {
             if (!er) {
+                console.log("Notification callback", rp, md);
                 if (rp === "activate" && md.activationType === "clicked") {
                     opn(`https://www.luogu.com.cn/chat?uid=${uid}`);
                 }
@@ -47,7 +46,16 @@ async function notify(msg) {
         }
     );
 }
-function chatnotify(uid, cid) {
+
+var lastmsg = 0;
+function chatnotify(uid, cid, fst = true) {
+    if (fst) {
+        console.log("Connecting to server");
+    } else {
+        console.log("Reconnecting");
+    }
+    console.log("Connect info: uid", uid);
+    console.log("Connect info: cid", cid);
     const ws = new WebSocket("wss://ws.luogu.com.cn/ws", {
         headers: {
             "Cookie": `__client_id=${cid}; _uid=${uid}`,
@@ -58,15 +66,17 @@ function chatnotify(uid, cid) {
         if (data._ws_type === "join_result") {
             if (data.result === "success") {
                 console.log("Connected, channel join success");
-                notifier.notify({
-                    title: "已经开始监听洛谷私信",
-                    message: "当收到私信时，会发送提示",
-                    sound: true,
-                    wait: true,
-                });
+                if (fst) {
+                    notifier.notify({
+                        title: "已经开始监听洛谷私信",
+                        message: "当收到私信时，会发送提示",
+                        sound: true,
+                        wait: true,
+                    });
+                }
             } else {
-                console.log("Connected, channel join fail:", data.result);
-                process.exit();
+                console.log("Connected, channel join fail", data.result);
+                ws.close();
             }
         } else if (data._ws_type === "heartbeat") {
             console.log("Received heartbeat");
@@ -78,11 +88,20 @@ function chatnotify(uid, cid) {
                     notify(data.message);
                 }
             } else {
-                console.log("Cannot parse as message");
+                console.log("Broadcast cannot parse as message", data);
             }
         } else {
-            console.log("Received data, but cannot parse:", data);
+            console.log("Received unknown data", data);
         }
+        lastmsg = new Date().getTime();
+        console.log("Message time", lastmsg);
+        setTimeout(() => {
+            if ((new Date().getTime()) - lastmsg >= 100000) {
+                console.log("No message for 100s, closing connection to retry");
+                ws.close();
+                chatnotify(uid, cid, false);
+            }
+        }, 101000);
     });
     ws.on("open", () => {
         ws.send(JSON.stringify({
@@ -92,9 +111,12 @@ function chatnotify(uid, cid) {
             exclusive_key: null,
         }));
     });
-    ws.on("close", () => {
-        console.log("Connection closed");
-        process.exit();
+    ws.on("close", (code, reason) => {
+        console.log("Connection closed", code, reason.toString());
+        chatnotify(uid, cid, false);
+    });
+    ws.on("error", (err) => {
+        console.log("ERROR", err.message);
     });
 }
 
@@ -103,6 +125,7 @@ rli.question("Input Luogu UID: ", (data) => {
     uid = data;
     rli.question("Input Luogu cookie __client_id: ", (data) => {
         cid = data;
+        rli.close();
         chatnotify(uid, cid);
     });
 });
\ No newline at end of file