mirror of
https://github.com/mamoe/mirai.git
synced 2025-03-26 23:50:16 +08:00
111 lines
4.0 KiB
Kotlin
111 lines
4.0 KiB
Kotlin
/*
|
|
* Copyright 2019-2020 Mamoe Technologies and contributors.
|
|
*
|
|
* 此源代码的使用受 GNU AFFERO GENERAL PUBLIC LICENSE version 3 许可证的约束, 可以在以下链接找到该许可证.
|
|
* Use of this source code is governed by the GNU AGPLv3 license that can be found through the following link.
|
|
*
|
|
* https://github.com/mamoe/mirai/blob/master/LICENSE
|
|
*/
|
|
|
|
@file:JvmMultifileClass
|
|
@file:JvmName("MessageEventKt")
|
|
@file:Suppress("unused")
|
|
|
|
package net.mamoe.mirai.message
|
|
|
|
import kotlinx.coroutines.Deferred
|
|
import kotlinx.coroutines.async
|
|
import net.mamoe.mirai.event.EventPriority
|
|
import net.mamoe.mirai.event.Listener
|
|
import net.mamoe.mirai.event.events.*
|
|
import net.mamoe.mirai.event.syncFromEvent
|
|
import net.mamoe.mirai.event.syncFromEventOrNull
|
|
import net.mamoe.mirai.message.data.MessageChain
|
|
import kotlin.coroutines.CoroutineContext
|
|
import kotlin.coroutines.EmptyCoroutineContext
|
|
|
|
|
|
/**
|
|
* 判断两个 [MessageEvent] 的 [MessageEvent.sender] 和 [MessageEvent.subject] 是否相同
|
|
*/
|
|
public fun MessageEvent.isContextIdenticalWith(another: MessageEvent): Boolean {
|
|
return this.sender == another.sender && this.subject == another.subject
|
|
}
|
|
|
|
|
|
/**
|
|
* 挂起当前协程, 等待下一条 [MessageEvent.sender] 和 [MessageEvent.subject] 与 [this] 相同且通过 [筛选][filter] 的 [MessageEvent]
|
|
*
|
|
* 若 [filter] 抛出了一个异常, 本函数会立即抛出这个异常.
|
|
*
|
|
* @param timeoutMillis 超时. 单位为毫秒. `-1` 为不限制
|
|
* @param filter 过滤器. 返回非 null 则代表得到了需要的值. [syncFromEvent] 会返回这个值
|
|
*
|
|
* @see syncFromEvent 实现原理
|
|
*/
|
|
@JvmSynthetic
|
|
public suspend inline fun <reified P : MessageEvent> P.nextMessage(
|
|
timeoutMillis: Long = -1,
|
|
priority: Listener.EventPriority = EventPriority.MONITOR,
|
|
noinline filter: suspend P.(P) -> Boolean = { true }
|
|
): MessageChain {
|
|
return syncFromEvent<P, P>(timeoutMillis, priority) {
|
|
takeIf { this.isContextIdenticalWith(this@nextMessage) }?.takeIf { filter(it, it) }
|
|
}.message
|
|
}
|
|
|
|
/**
|
|
* 挂起当前协程, 等待下一条 [MessageEvent.sender] 和 [MessageEvent.subject] 与 [this] 相同且通过 [筛选][filter] 的 [MessageEvent]
|
|
*
|
|
* 若 [filter] 抛出了一个异常, 本函数会立即抛出这个异常.
|
|
*
|
|
* @param timeoutMillis 超时. 单位为毫秒.
|
|
* @param filter 过滤器. 返回非 null 则代表得到了需要的值. [syncFromEvent] 会返回这个值
|
|
* @return 消息链. 超时时返回 `null`
|
|
*
|
|
* @see syncFromEventOrNull 实现原理
|
|
*/
|
|
@JvmSynthetic
|
|
public suspend inline fun <reified P : MessageEvent> P.nextMessageOrNull(
|
|
timeoutMillis: Long,
|
|
priority: Listener.EventPriority = EventPriority.MONITOR,
|
|
noinline filter: suspend P.(P) -> Boolean = { true }
|
|
): MessageChain? {
|
|
require(timeoutMillis > 0) { "timeoutMillis must be > 0" }
|
|
return syncFromEventOrNull<P, P>(timeoutMillis, priority) {
|
|
takeIf { this.isContextIdenticalWith(this@nextMessageOrNull) }?.takeIf { filter(it, it) }
|
|
}?.message
|
|
}
|
|
|
|
/**
|
|
* @see nextMessage
|
|
*/
|
|
@JvmSynthetic
|
|
public inline fun <reified P : MessageEvent> P.nextMessageAsync(
|
|
timeoutMillis: Long = -1,
|
|
coroutineContext: CoroutineContext = EmptyCoroutineContext,
|
|
priority: Listener.EventPriority = EventPriority.MONITOR,
|
|
noinline filter: suspend P.(P) -> Boolean = { true }
|
|
): Deferred<MessageChain> {
|
|
return this.bot.async(coroutineContext) {
|
|
nextMessage(timeoutMillis, priority, filter)
|
|
}
|
|
}
|
|
|
|
/**
|
|
* [nextMessageOrNull] 的异步版本
|
|
*
|
|
* @see nextMessageOrNull
|
|
*/
|
|
@JvmSynthetic
|
|
public inline fun <reified P : MessageEvent> P.nextMessageOrNullAsync(
|
|
timeoutMillis: Long,
|
|
coroutineContext: CoroutineContext = EmptyCoroutineContext,
|
|
priority: Listener.EventPriority = EventPriority.MONITOR,
|
|
noinline filter: suspend P.(P) -> Boolean = { true }
|
|
): Deferred<MessageChain?> {
|
|
require(timeoutMillis > 0) { "timeoutMillis must be > 0" }
|
|
return this.bot.async(coroutineContext) {
|
|
nextMessageOrNull(timeoutMillis, priority, filter)
|
|
}
|
|
} |