From 5f0cf952e89ca4d0973c40a20777c2c6995e8f12 Mon Sep 17 00:00:00 2001
From: WhiteChi <whitechi73@outlook.com>
Date: Thu, 23 Nov 2023 09:48:30 +0800
Subject: [PATCH] `Shamrock`: support `/switch_account`

---
 .../src/main/java/mqq/app/AppRuntime.java     |  5 +++
 .../shamrock/remote/action/ActionManager.kt   |  1 +
 .../remote/action/handlers/SwitchAccount.kt   | 35 +++++++++++++++++++
 .../shamrock/remote/api/UserAction.kt         |  5 +++
 .../service/api/GlobalEventTransmitter.kt     |  2 +-
 5 files changed, 47 insertions(+), 1 deletion(-)
 create mode 100644 xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/handlers/SwitchAccount.kt

diff --git a/qqinterface/src/main/java/mqq/app/AppRuntime.java b/qqinterface/src/main/java/mqq/app/AppRuntime.java
index fc8d304..5562807 100644
--- a/qqinterface/src/main/java/mqq/app/AppRuntime.java
+++ b/qqinterface/src/main/java/mqq/app/AppRuntime.java
@@ -5,6 +5,7 @@ import android.text.TextUtils;
 
 import androidx.annotation.NonNull;
 
+import com.tencent.qphone.base.remote.SimpleAccount;
 import com.tencent.qphone.base.remote.ToServiceMsg;
 
 import mqq.app.api.IRuntimeService;
@@ -72,6 +73,10 @@ public abstract class AppRuntime {
     public <T extends IRuntimeService> T getRuntimeServiceIPCSync(@NonNull Class<T> cls, String str) {
         throw new UnsupportedOperationException();
     }
+
+    public void switchAccount(SimpleAccount simpleAccount, String process) {
+
+    }
     public String getAccount() {
         return "";
     }
diff --git a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/ActionManager.kt b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/ActionManager.kt
index fb57d24..cb7e0e0 100644
--- a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/ActionManager.kt
+++ b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/ActionManager.kt
@@ -18,6 +18,7 @@ internal object ActionManager {
         arrayOf(
             // Framework Info
             TestHandler, GetLatestEvents, GetSupportedActions, GetStatus, GetVersionInfo, GetSelfInfo, GetLoginInfo,
+            SwitchAccount,
 
             // UserActions
             GetProfileCard, GetFriendList, SendLike, GetUid, GetUinByUid, ScanQRCode, SetProfileCard,
diff --git a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/handlers/SwitchAccount.kt b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/handlers/SwitchAccount.kt
new file mode 100644
index 0000000..187e93d
--- /dev/null
+++ b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/handlers/SwitchAccount.kt
@@ -0,0 +1,35 @@
+package moe.fuqiuluo.shamrock.remote.action.handlers
+
+import kotlinx.serialization.json.JsonElement
+import moe.fuqiuluo.shamrock.remote.action.ActionSession
+import moe.fuqiuluo.shamrock.remote.action.IActionHandler
+import moe.fuqiuluo.shamrock.tools.EmptyJsonString
+import moe.fuqiuluo.shamrock.xposed.helper.AppRuntimeFetcher
+import mqq.app.MobileQQ
+
+internal object SwitchAccount: IActionHandler() {
+    override suspend fun internalHandle(session: ActionSession): String {
+        val userId = session.getString("user_id")
+        return invoke(userId, session.echo)
+    }
+
+    operator fun invoke(
+        userId: String,
+        echo: JsonElement = EmptyJsonString
+    ): String {
+        val account = MobileQQ.getMobileQQ().allAccounts.firstOrNull { it.uin == userId }
+            ?: return error("账号不存在", echo)
+        val runtime = AppRuntimeFetcher.appRuntime
+        val result = kotlin.runCatching {
+            runtime.switchAccount(account, null)
+        }
+        if (result.isFailure) {
+            return error(result.exceptionOrNull()?.message ?: "切换账号失败", echo)
+        }
+        return ok("切换成功", echo)
+    }
+
+    override val requiredParams: Array<String> = arrayOf("user_id")
+
+    override fun path(): String = "switch_account"
+}
\ No newline at end of file
diff --git a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/api/UserAction.kt b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/api/UserAction.kt
index 5b43b28..44a7c6c 100644
--- a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/api/UserAction.kt
+++ b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/api/UserAction.kt
@@ -12,6 +12,11 @@ import moe.fuqiuluo.shamrock.tools.*
 import moe.fuqiuluo.shamrock.utils.PlatformUtils
 
 fun Routing.userAction() {
+    getOrPost("/switch_account") {
+        val userId = fetchOrThrow("user_id")
+        call.respondText(SwitchAccount(userId), ContentType.Application.Json)
+    }
+
     getOrPost("/set_group_leave") {
         val group = fetchOrThrow("group_id")
         call.respondText(LeaveTroop(group), ContentType.Application.Json)
diff --git a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/service/api/GlobalEventTransmitter.kt b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/service/api/GlobalEventTransmitter.kt
index 9757a3b..5b7f157 100644
--- a/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/service/api/GlobalEventTransmitter.kt
+++ b/xposed/src/main/java/moe/fuqiuluo/shamrock/remote/service/api/GlobalEventTransmitter.kt
@@ -104,7 +104,7 @@ internal object GlobalEventTransmitter: BaseSvc() {
             var nickName = record.sendNickName
             if (nickName.isNullOrBlank()) {
                 CardSvc.getProfileCard(record.senderUin.toString()).onSuccess {
-                    nickName = it.strNick ?: ""
+                    nickName = it.strNick ?: record.peerName
                 }
             }
             transMessageEvent(record,