github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-22 13:46:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ContactOrBot.inferMessageSourceKind

Browse Source
This commit is contained in:
Him188 2022-04-30 19:01:28 +01:00
parent e826c94546
commit 05aa3c8026
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mirai-core/src/commonMain/kotlin/contact/SendMessageHandler.kt
View File

@ -13,6 +13,7 @@ import kotlinx.coroutines.CompletableDeferred
import kotlinx.coroutines.Deferred import kotlinx.coroutines.Deferred
import kotlinx.coroutines.async import kotlinx.coroutines.async
import kotlinx.coroutines.withTimeoutOrNull import kotlinx.coroutines.withTimeoutOrNull
import net.mamoe.mirai.Bot
import net.mamoe.mirai.contact.* import net.mamoe.mirai.contact.*
import net.mamoe.mirai.event.EventPriority import net.mamoe.mirai.event.EventPriority
import net.mamoe.mirai.event.GlobalEventChannel import net.mamoe.mirai.event.GlobalEventChannel
@ -51,6 +52,17 @@ import net.mamoe.mirai.message.data.*
import net.mamoe.mirai.utils.castOrNull import net.mamoe.mirai.utils.castOrNull
import net.mamoe.mirai.utils.currentTimeSeconds import net.mamoe.mirai.utils.currentTimeSeconds
internal fun ContactOrBot.inferMessageSourceKind(): MessageSourceKind {
return when (this) {
is Group -> MessageSourceKind.GROUP
is Friend -> MessageSourceKind.FRIEND
is Member -> MessageSourceKind.TEMP
is Stranger -> MessageSourceKind.STRANGER
is Bot -> MessageSourceKind.FRIEND
else -> error("Unsupported contact: $this")
}
}
/** /**
* 处理 mirai 消息系统 `Message` 到协议数据结构的转换. * 处理 mirai 消息系统 `Message` 到协议数据结构的转换.
* *
@ -62,13 +74,7 @@ internal abstract class SendMessageHandler<C : Contact> {
val messageSourceKind: MessageSourceKind val messageSourceKind: MessageSourceKind
get() { get() {
return when (contact) { return contact.inferMessageSourceKind()
is Group -> MessageSourceKind.GROUP
is Friend -> MessageSourceKind.FRIEND
is Member -> MessageSourceKind.TEMP
is Stranger -> MessageSourceKind.STRANGER
else -> error("Unsupported contact: $contact")
}
} }
val bot get() = contact.bot.asQQAndroidBot() val bot get() = contact.bot.asQQAndroidBot()