From 942e63654ea71286880501a2ab6ed3cdf6971b41 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=9B=AE=E6=A3=83?= <BT-Muli@outlook.com>
Date: Sat, 27 Jul 2024 20:41:39 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=E9=AA=8C=E8=AF=81=E7=A0=81?=
 =?UTF-8?q?=E7=99=BB=E5=BD=95=E6=88=90=E5=8A=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

close #118
---
 src/components/config/tc-userBadge.vue | 114 ++++++++++++++++++++++++-
 src/pages/common/Test.vue              |  51 +----------
 2 files changed, 112 insertions(+), 53 deletions(-)

diff --git a/src/components/config/tc-userBadge.vue b/src/components/config/tc-userBadge.vue
index 35c3e893..e0653b64 100644
--- a/src/components/config/tc-userBadge.vue
+++ b/src/components/config/tc-userBadge.vue
@@ -9,6 +9,12 @@
     <template #text>{{ userInfo.desc }}</template>
     <template #actions>
       <v-spacer />
+      <v-btn
+        variant="outlined"
+        @click="tryCaptchaLogin()"
+        icon="mdi-cellphone"
+        title="验证码登录"
+      />
       <v-btn variant="outlined" @click="scan = true" icon="mdi-qrcode-scan" title="扫码登录" />
       <v-btn
         variant="outlined"
@@ -22,10 +28,11 @@
   </v-card>
 </template>
 <script lang="ts" setup>
-import type { UnlistenFn } from "@tauri-apps/api/event";
+import { UnlistenFn } from "@tauri-apps/api/event";
 import { storeToRefs } from "pinia";
 import { onMounted, onUnmounted, ref, watch } from "vue";
 
+import Mys from "../../plugins/Mys/index.js";
 import TGSqlite from "../../plugins/Sqlite/index.js";
 import { useAppStore } from "../../store/modules/app.js";
 import { useUserStore } from "../../store/modules/user.js";
@@ -33,6 +40,7 @@ import TGLogger from "../../utils/TGLogger.js";
 import { getDeviceFp } from "../../web/request/getDeviceFp.js";
 import TGRequest from "../../web/request/TGRequest.js";
 import showConfirm from "../func/confirm.js";
+import showGeetest from "../func/geetest.js";
 import showSnackbar from "../func/snackbar.js";
 import ToGameLogin from "../overlay/to-gameLogin.vue";
 
@@ -68,6 +76,65 @@ watch(userStore.briefInfo, (v) => {
   }
 });
 
+async function tryCaptchaLogin(): Promise<void> {
+  const phone = await showConfirm({
+    mode: "input",
+    title: "请输入手机号",
+    text: "+86",
+  });
+  if (!phone) {
+    showSnackbar({
+      color: "cancel",
+      text: "已取消验证码登录",
+    });
+    return;
+  }
+  const phoneReg = /^1[3-9]\d{9}$/;
+  if (!phoneReg.test(phone)) {
+    showSnackbar({
+      color: "error",
+      text: "请输入正确的手机号",
+    });
+    return;
+  }
+  const actionType = await tryGetCaptcha(phone);
+  if (!actionType) return;
+  showSnackbar({
+    text: `已发送验证码到 ${phone}`,
+  });
+  const captcha = await showConfirm({
+    mode: "input",
+    title: "请输入验证码",
+    text: "验证码：",
+    otcancel: false,
+  });
+  if (!captcha) {
+    showSnackbar({
+      color: "error",
+      text: "输入验证码为空",
+    });
+    return;
+  }
+  const loginResp = await tryLoginByCaptcha(phone, captcha, actionType);
+  if (!loginResp) return;
+  userStore.cookie.value = {
+    account_id: loginResp.user_info.aid,
+    ltuid: loginResp.user_info.aid,
+    stuid: loginResp.user_info.aid,
+    mid: loginResp.user_info.mid,
+    cookie_token: "",
+    stoken: loginResp.token.token,
+    ltoken: "",
+  };
+  showSnackbar({
+    text: "登录成功，即将刷新用户信息",
+    color: "success",
+  });
+  setTimeout(() => {
+    refreshUser();
+  }, 1000);
+}
+
 async function refreshUser() {
   const ck = userStore.cookie.value;
   if (ck === undefined || JSON.stringify(ck) === "{}") {
@@ -154,8 +221,8 @@ async function refreshUser() {
   emits("loadOuter", { show: false });
 }
 
-async function refreshUserInfo(cnt: number = 0): Promise<number> {
-  let failCount = cnt;
+async function refreshUserInfo(): Promise<number> {
+  let failCount = 0;
   const ck = userStore.cookie.value;
   if (ck === undefined) {
     showSnackbar({
@@ -252,6 +319,47 @@ async function confirmCopyCookie(): Promise<void> {
   });
 }
 
+async function tryGetCaptcha(phone: string, aigis?: string): Promise<string | false> {
+  const captchaResp = await Mys.User.getCaptcha(phone, aigis);
+  if ("retcode" in captchaResp) {
+    if (!captchaResp.data || captchaResp.data === "") {
+      showSnackbar({
+        text: `[${captchaResp.retcode}] ${captchaResp.message}`,
+        color: "error",
+      });
+      return false;
+    }
+    const aigisResp: TGApp.Plugins.Mys.CaptchaLogin.CaptchaAigis = JSON.parse(captchaResp.data);
+    const resp = await showGeetest(JSON.parse(aigisResp.data));
+    const aigisStr = `${aigisResp.session_id};${btoa(JSON.stringify(resp))}`;
+    return await tryGetCaptcha(phone, aigisStr);
+  }
+  return captchaResp.action_type;
+}
+
+async function tryLoginByCaptcha(
+  phone: string,
+  captcha: string,
+  actionType: string,
+  aigis?: string,
+): Promise<TGApp.Plugins.Mys.CaptchaLogin.LoginData | false> {
+  const loginResp = await Mys.User.login(phone, captcha, actionType, aigis);
+  if ("retcode" in loginResp) {
+    if (!loginResp.data || loginResp.data === "") {
+      showSnackbar({
+        text: `[${loginResp.retcode}] ${loginResp.message}`,
+        color: "error",
+      });
+      return false;
+    }
+    const aigisResp: TGApp.Plugins.Mys.CaptchaLogin.CaptchaAigis = JSON.parse(loginResp.data);
+    const resp = await showGeetest(JSON.parse(aigisResp.data));
+    const aigisStr = `${aigisResp.session_id};${btoa(JSON.stringify(resp))}`;
+    return await tryLoginByCaptcha(phone, captcha, actionType, aigisStr);
+  }
+  return loginResp;
+}
+
 onUnmounted(() => {
   if (signListener) signListener();
 });
diff --git a/src/pages/common/Test.vue b/src/pages/common/Test.vue
index a232f909..d47da50f 100644
--- a/src/pages/common/Test.vue
+++ b/src/pages/common/Test.vue
@@ -13,58 +13,9 @@
         </div>
       </div>
     </div>
-    <h1>验证码登录测试</h1>
-    <div class="test-item">
-      <div class="btn-list">
-        <button class="test-btn" @click="tryCaptchaLogin()">获取验证码</button>
-      </div>
-    </div>
   </div>
 </template>
-<script lang="ts" setup>
-import showConfirm from "../../components/func/confirm.js";
-import showGeetest from "../../components/func/geetest.js";
-import showSnackbar from "../../components/func/snackbar.js";
-import Mys from "../../plugins/Mys/index.js";
-
-async function tryCaptchaLogin(): Promise<void> {
-  const phone = await showConfirm({
-    mode: "input",
-    title: "请输入手机号",
-    text: "+86",
-  });
-  if (!phone) return;
-  const action_type = await tryGetCaptcha(phone);
-  if (!action_type) return;
-  const captcha = await showConfirm({
-    mode: "input",
-    title: "请输入验证码",
-    text: "验证码：",
-  });
-  if (!captcha) return;
-  const loginResp = await Mys.User.login(phone, captcha, action_type);
-  console.log("[loginResp]", loginResp);
-}
-
-async function tryGetCaptcha(phone: string, aigis?: string): Promise<string | false> {
-  const captchaResp = await Mys.User.getCaptcha(phone, aigis);
-  console.log("[captchaResp]", captchaResp);
-  if ("retcode" in captchaResp) {
-    if (!captchaResp.data || captchaResp.data === "") {
-      showSnackbar({
-        text: `[${captchaResp.retcode}] ${captchaResp.message}`,
-        color: "error",
-      });
-      return false;
-    }
-    const aigis: TGApp.Plugins.Mys.CaptchaLogin.CaptchaAigis = JSON.parse(captchaResp.data);
-    const resp = await showGeetest(aigis.data);
-    const aigisStr = btoa(`${aigis.session_id};${JSON.stringify(resp)}`);
-    return await tryGetCaptcha(phone, aigisStr);
-  }
-  return captchaResp.action_type;
-}
-</script>
+<script lang="ts" setup></script>
 <style lang="css" scoped>
 .test-box {
   display: flex;