From 790c4f589d0afba94149a5e26bcb752196226491 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=89=8B=E7=93=9C=E4=B8=80=E5=8D=81=E9=9B=AA?=
 <nanaeonn@outlook.com>
Date: Wed, 8 May 2024 15:07:20 +0800
Subject: [PATCH] feat: Login HeartBeat

---
 src/webui/src/router/QQLogin.ts |  3 ++-
 static/QQLogin.html             | 33 +++++++++++++++++++++++++++++----
 static/login.html               |  2 +-
 3 files changed, 32 insertions(+), 6 deletions(-)

diff --git a/src/webui/src/router/QQLogin.ts b/src/webui/src/router/QQLogin.ts
index 7370d578..82635b28 100644
--- a/src/webui/src/router/QQLogin.ts
+++ b/src/webui/src/router/QQLogin.ts
@@ -1,7 +1,8 @@
 import { Router } from 'express';
-import { QQCheckLoginStatusHandler, QQGetQuickLoginListHandler } from '../api/QQLogin';
+import { QQCheckLoginStatusHandler, QQGetQRcodeHandler, QQGetQuickLoginListHandler } from '../api/QQLogin';
 
 const router = Router();
 router.all('/GetQuickLoginList', QQGetQuickLoginListHandler)
 router.post('/CheckLoginStatus', QQCheckLoginStatusHandler);
+router.post('/GetQQLoginQrcode', QQGetQRcodeHandler);
 export { router as QQLoginRouter };
\ No newline at end of file
diff --git a/static/QQLogin.html b/static/QQLogin.html
index 3447b797..b02419f6 100644
--- a/static/QQLogin.html
+++ b/static/QQLogin.html
@@ -158,7 +158,26 @@
         <p id="message"></p>
     </div>
     <script>
-
+        async function GetQQLoginQrcode(retCredential) {
+            let QQLoginResponse = await fetch('/api/QQLogin/GetQQLoginQrcode', {
+                method: 'POST',
+                headers: {
+                    'Authorization': "Bearer " + retCredential,
+                    'Content-Type': 'application/json'
+                }
+            });
+            if (QQLoginResponse.status == 200) {
+                let QQLoginResponseJson = await QQLoginResponse.json();
+                if (QQLoginResponseJson.code == 0) {
+                    if (QQLoginResponseJson.data.qrcode) {
+                        return QQLoginResponseJson.data.qrcode;
+                    } else {
+                        return "";
+                    }
+                }
+            }
+            return "";
+        }
         async function CheckQQLoginStatus(retCredential) {
             let QQLoginResponse = await fetch('/api/QQLogin/CheckLoginStatus', {
                 method: 'POST',
@@ -195,6 +214,13 @@
             }
             return [];
         }
+        //心跳轮询CheckQQLoginStatus 是否已经进入登录状态，如果为true进入./config.html
+        async function HeartBeat() {
+            let isLogined = await CheckQQLoginStatus(localStorage.getItem('auth'));
+            if (isLogined) {
+                window.location.href = './config.html';
+            }
+        }
         async function InitPages() {
             let QuickLists = await GetQQQucickLoginList(localStorage.getItem('auth'));
             let QuickListSelect = document.querySelector("#quick-login-select");
@@ -204,6 +230,8 @@
                 optionUinEle.value = QuickUin;
                 QuickListSelect.appendChild(optionUinEle);
             });
+            generateQrCode(await GetQQLoginQrcode(), document.querySelector('#qrcode-canvas'));
+            setTimeout(HeartBeat, 3000);
         }
         document.getElementById('quick-login').addEventListener('click', function () {
             let quickLoginOptions = document.querySelector('#quick-login-dropdown');
@@ -214,7 +242,6 @@
 
         function selectAccount(accountName) {
             alert(`Logging in with ${accountName}...`);
-            // Implement your login logic here
             document.getElementById('quick-login-options').classList.remove('show');
         }
         document.getElementById('qrcode-login').addEventListener('click', function () {
@@ -222,8 +249,6 @@
             let qrcode = document.querySelector('#qrcode');
             loginForm.style.display = 'none';
             qrcode.style.display = 'flex';
-            let qrData = 'https://yourapp.com/login?code=123456';
-            generateQrCode(qrData, document.querySelector('#qrcode-canvas'));
         });
 
         function generateQrCode(data, canvas) {
diff --git a/static/login.html b/static/login.html
index 96d446c3..6149bf87 100644
--- a/static/login.html
+++ b/static/login.html
@@ -109,7 +109,7 @@
                         if (QQLoginResponse.status == 200) {
                             let QQLoginResponseJson = await QQLoginResponse.json();
                             if (QQLoginResponseJson.code == 0) {
-                                alert(QQLoginResponseJson.data.isLogin.toString());
+                                //alert(QQLoginResponseJson.data.isLogin.toString());
                                 if (QQLoginResponseJson.data.isLogin) {
                                     window.location.href = './config.html';
                                 } else {