mirror of
https://github.com/whitechi73/OpenShamrock.git
synced 2024-08-14 13:12:17 +08:00
264 lines
10 KiB
Kotlin
264 lines
10 KiB
Kotlin
@file:OptIn(DelicateCoroutinesApi::class)
|
||
|
||
package moe.fuqiuluo.qqinterface.servlet
|
||
|
||
import com.tencent.mobileqq.qroute.QRoute
|
||
import com.tencent.mobileqq.troop.api.ITroopMemberNameService
|
||
import com.tencent.qqnt.kernel.api.IKernelService
|
||
import com.tencent.qqnt.kernel.nativeinterface.IOperateCallback
|
||
import com.tencent.qqnt.kernel.nativeinterface.MsgConstant
|
||
import com.tencent.qqnt.kernel.nativeinterface.MsgRecord
|
||
import com.tencent.qqnt.kernel.nativeinterface.TempChatGameSession
|
||
import com.tencent.qqnt.kernel.nativeinterface.TempChatPrepareInfo
|
||
import com.tencent.qqnt.msg.api.IMsgService
|
||
import kotlinx.coroutines.DelicateCoroutinesApi
|
||
import kotlinx.coroutines.GlobalScope
|
||
import kotlinx.coroutines.delay
|
||
import kotlinx.coroutines.launch
|
||
import kotlinx.coroutines.suspendCancellableCoroutine
|
||
import kotlinx.coroutines.withTimeoutOrNull
|
||
import kotlinx.serialization.json.JsonArray
|
||
import moe.fuqiuluo.shamrock.helper.ContactHelper
|
||
import moe.fuqiuluo.shamrock.helper.Level
|
||
import moe.fuqiuluo.shamrock.helper.LogCenter
|
||
import moe.fuqiuluo.shamrock.helper.MessageHelper
|
||
import moe.fuqiuluo.shamrock.helper.SendMsgException
|
||
import moe.fuqiuluo.shamrock.tools.EMPTY_BYTE_ARRAY
|
||
import moe.fuqiuluo.shamrock.xposed.helper.NTServiceFetcher
|
||
import moe.fuqiuluo.shamrock.xposed.helper.msgService
|
||
import java.util.UUID
|
||
import kotlin.coroutines.resume
|
||
import kotlin.coroutines.suspendCoroutine
|
||
|
||
internal object MsgSvc: BaseSvc() {
|
||
fun uploadForwardMsg(): Result<String> {
|
||
return Result.failure(Exception("Not implemented"))
|
||
}
|
||
|
||
suspend fun prepareTempChatFromGroup(
|
||
groupId: String,
|
||
peerId: String
|
||
): Result<Unit> {
|
||
LogCenter.log("主动临时消息，创建临时会话。", Level.INFO)
|
||
val msgService = app.getRuntimeService(IKernelService::class.java, "all").msgService
|
||
?: return Result.failure(Exception("获取消息服务失败"))
|
||
msgService.prepareTempChat(TempChatPrepareInfo(
|
||
MsgConstant.KCHATTYPETEMPC2CFROMGROUP,
|
||
ContactHelper.getUidByUinAsync(peerId = peerId.toLong()),
|
||
app.getRuntimeService(ITroopMemberNameService::class.java, "all")
|
||
.getTroopMemberNameRemarkFirst(groupId, peerId),
|
||
groupId, EMPTY_BYTE_ARRAY, app.currentUid, "", TempChatGameSession()
|
||
)) { code, reason ->
|
||
if (code != 0) {
|
||
LogCenter.log("临时会话创建失败: $code, $reason", Level.ERROR)
|
||
}
|
||
}
|
||
return Result.success(Unit)
|
||
}
|
||
|
||
/**
|
||
* 正常获取
|
||
*/
|
||
suspend fun getMsg(hash: Int): Result<MsgRecord> {
|
||
val mapping = MessageHelper.getMsgMappingByHash(hash)
|
||
?: return Result.failure(Exception("没有对应消息映射，消息获取失败"))
|
||
|
||
val peerId = mapping.peerId
|
||
val contact = MessageHelper.generateContact(mapping.chatType, peerId)
|
||
|
||
val msg = withTimeoutOrNull(5000) {
|
||
val service = QRoute.api(IMsgService::class.java)
|
||
suspendCancellableCoroutine { continuation ->
|
||
service.getMsgsByMsgId(contact, arrayListOf(mapping.qqMsgId)) { code, _, msgRecords ->
|
||
if (code == 0 && msgRecords.isNotEmpty()) {
|
||
continuation.resume(msgRecords.first())
|
||
} else {
|
||
continuation.resume(null)
|
||
}
|
||
}
|
||
continuation.invokeOnCancellation {
|
||
continuation.resume(null)
|
||
} // 貌似不会被取消，写了也没什么鸟用啊？
|
||
}
|
||
}
|
||
|
||
return if (msg != null) {
|
||
Result.success(msg)
|
||
} else {
|
||
Result.failure(Exception("获取消息失败"))
|
||
}
|
||
}
|
||
|
||
suspend fun getMsgByQMsgId(
|
||
chatType: Int,
|
||
peerId: String,
|
||
qqMsgId: Long
|
||
): Result<MsgRecord> {
|
||
val contact = MessageHelper.generateContact(chatType, peerId)
|
||
val service = QRoute.api(IMsgService::class.java) ?:
|
||
return Result.failure(Exception("获取消息服务"))
|
||
|
||
val msg = withTimeoutOrNull(5000) {
|
||
suspendCoroutine { continuation ->
|
||
service.getMsgsByMsgId(contact, arrayListOf(qqMsgId)) { code, _, msgRecords ->
|
||
if (code == 0 && msgRecords.isNotEmpty()) {
|
||
continuation.resume(msgRecords.first())
|
||
} else {
|
||
continuation.resume(null)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return if (msg != null) {
|
||
Result.success(msg)
|
||
} else {
|
||
Result.failure(Exception("获取消息失败"))
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 什么鸟屎都获取不到
|
||
*/
|
||
suspend fun getMsgBySeq(
|
||
chatType: Int,
|
||
peerId: String,
|
||
seq: Long
|
||
): Result<MsgRecord> {
|
||
val contact = MessageHelper.generateContact(chatType, peerId)
|
||
val msg = withTimeoutOrNull(1000) {
|
||
val service = QRoute.api(IMsgService::class.java)
|
||
suspendCancellableCoroutine { continuation ->
|
||
service.getMsgsBySeqs(contact, arrayListOf(seq)) { code, _, msgRecords ->
|
||
continuation.resume(msgRecords?.firstOrNull())
|
||
}
|
||
continuation.invokeOnCancellation {
|
||
continuation.resume(null)
|
||
}
|
||
}
|
||
}
|
||
return if (msg != null) {
|
||
Result.success(msg)
|
||
} else {
|
||
Result.failure(Exception("获取消息失败"))
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 撤回消息 同步 HTTP API
|
||
*/
|
||
suspend fun recallMsg(msgHash: Int): Pair<Int, String> {
|
||
val kernelService = NTServiceFetcher.kernelService
|
||
val sessionService = kernelService.wrapperSession
|
||
val msgService = sessionService.msgService
|
||
|
||
val mapping = MessageHelper.getMsgMappingByHash(msgHash)
|
||
?: return -1 to "无法找到消息映射"
|
||
|
||
val contact = MessageHelper.generateContact(mapping.chatType, mapping.peerId)
|
||
|
||
return suspendCancellableCoroutine { continuation ->
|
||
msgService.recallMsg(contact, arrayListOf(mapping.qqMsgId)) { code, why ->
|
||
continuation.resume(code to why)
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 发送消息
|
||
*
|
||
* Aio 腾讯内部命名 All In One
|
||
*/
|
||
suspend fun sendToAio(
|
||
chatType: Int,
|
||
peedId: String,
|
||
message: JsonArray,
|
||
fromId: String = peedId,
|
||
retryCnt: Int = 3
|
||
): Result<Pair<Long, Int>> {
|
||
//LogCenter.log(message.toString(), Level.ERROR)
|
||
//callback.msgHash = result.second 什么垃圾代码，万一cb比你快，你不就寄了？
|
||
|
||
// 主动临时消息
|
||
when (chatType) {
|
||
MsgConstant.KCHATTYPETEMPC2CFROMGROUP -> {
|
||
prepareTempChatFromGroup(fromId, peedId).onFailure {
|
||
LogCenter.log("主动临时消息，创建临时会话失败。", Level.ERROR)
|
||
return Result.failure(Exception("主动临时消息，创建临时会话失败。"))
|
||
}
|
||
}
|
||
}
|
||
val result = MessageHelper.sendMessageWithoutMsgId(
|
||
chatType,
|
||
peedId,
|
||
message,
|
||
fromId,
|
||
MessageCallback(peedId, 0)
|
||
)
|
||
return if (result.isFailure
|
||
&& result.exceptionOrNull()?.javaClass == SendMsgException::class.java
|
||
&& retryCnt > 0) {
|
||
// 发送失败，可能网络问题出现红色感叹号，重试
|
||
// 例如 rich media transfer failed
|
||
delay(100)
|
||
sendToAio(chatType, peedId, message, fromId, retryCnt - 1)
|
||
} else {
|
||
result
|
||
}
|
||
}
|
||
|
||
suspend fun getMultiMsg(resId: String): Result<List<MsgRecord>> {
|
||
val kernelService = NTServiceFetcher.kernelService
|
||
val sessionService = kernelService.wrapperSession
|
||
val msgService = sessionService.msgService
|
||
val contact = MessageHelper.generateContact(MsgConstant.KCHATTYPEC2C, TicketSvc.getUin())
|
||
|
||
val content = "{\"app\":\"com.tencent.multimsg\",\"config\":{\"autosize\":1,\"forward\":1,\"round\":1,\"type\":\"normal\",\"width\":300},\"desc\":\"[聊天记录]\",\"extra\":\"\",\"meta\":{\"detail\":{\"news\":[{\"text\":\"Shamrock: 这是条假消息！\"}],\"resid\":\"$resId\",\"source\":\"聊天记录\",\"summary\":\"转发消息\",\"uniseq\":\"${UUID.randomUUID()}\"}},\"prompt\":\"[聊天记录]\",\"ver\":\"0.0.0.5\",\"view\":\"contact\"}"
|
||
val msgId = PacketSvc.fakeSelfRecvJsonMsg(msgService, content)
|
||
if (msgId < 0) {
|
||
return Result.failure(Exception("获取合并转发消息ID失败"))
|
||
}
|
||
val msgList = withTimeoutOrNull(5000L) {
|
||
suspendCancellableCoroutine<ArrayList<MsgRecord>> {
|
||
val job = GlobalScope.launch {
|
||
var hasResult = false
|
||
while (!hasResult) {
|
||
msgService.getMultiMsg(contact, msgId, msgId) { code, why, msgList ->
|
||
if (code == 0) {
|
||
it.resume(msgList)
|
||
hasResult = true
|
||
} else {
|
||
LogCenter.log("获取合并转发消息失败: $code($why): $msgId", Level.ERROR)
|
||
}
|
||
}
|
||
delay(200)
|
||
}
|
||
}
|
||
it.invokeOnCancellation {
|
||
job.cancel()
|
||
}
|
||
}
|
||
} ?: return Result.failure(Exception("获取合并转发消息失败"))
|
||
|
||
//msgService.deleteMsg(contact, arrayListOf(msgId), null)
|
||
|
||
return Result.success(msgList)
|
||
}
|
||
|
||
class MessageCallback(
|
||
private val peerId: String,
|
||
var msgHash: Int
|
||
): IOperateCallback {
|
||
override fun onResult(code: Int, reason: String?) {
|
||
if (code != 0 && msgHash != 0) {
|
||
MessageHelper.removeMsgByHashCode(msgHash)
|
||
}
|
||
when (code) {
|
||
110 -> LogCenter.log("消息发送: $peerId, 无该联系人无法发送。")
|
||
120 -> LogCenter.log("消息发送: $peerId, 禁言状态无法发送。")
|
||
5 -> LogCenter.log("消息发送: $peerId, 当前不支持该消息类型。")
|
||
else -> LogCenter.log("消息发送: $peerId, code: $code $reason")
|
||
}
|
||
}
|
||
}
|
||
} |