github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-02-10 05:57:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add containsAny and containsAll

Browse Source
This commit is contained in:
Him188 2020-02-29 13:47:47 +08:00
parent 756e8b537e
commit 0cbc0f0e29
1 changed files with 71 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
mirai-core/src/commonMain/kotlin/net.mamoe.mirai/event/subscribeMessages.kt
View File

@ -324,7 +324,7 @@ class MessageSubscribersBuilder<T : MessagePacket<*, *>>(
content { sub in it }
/**
* 如果消息内容包含 [sub]
* 如果消息内容包含 [sub] 中的任意一个元素
*/
@MessageDsl
inline fun contains(
@ -345,6 +345,72 @@ class MessageSubscribersBuilder<T : MessagePacket<*, *>>(
}
}
/**
* 如果消息内容包含 [sub]
*/
@MessageDsl
fun containsAny(vararg sub: String): ListeningFilter =
content { sub.any { item -> item in it } }
/**
* 如果消息内容包含 [sub] 中的任意一个元素
*/
@MessageDsl
inline fun containsAny(
vararg sub: String,
ignoreCase: Boolean = false,
trim: Boolean = true,
crossinline onEvent: MessageListener<T>
): Listener<T> {
return if (trim) {
val list = sub.map { it.trim() }
content({
list.any { toCheck -> it.contains(toCheck, ignoreCase = ignoreCase) }
}, {
onEvent(this, this.message.toString().trim())
})
} else {
content({
sub.any { toCheck -> it.contains(toCheck, ignoreCase = ignoreCase) }
}, {
onEvent(this, this.message.toString())
})
}
}
/**
* 如果消息内容包含 [sub]
*/
@MessageDsl
fun containsAll(vararg sub: String): ListeningFilter =
content { sub.all { item -> item in it } }
/**
* 如果消息内容包含 [sub] 中的任意一个元素
*/
@MessageDsl
inline fun containsAll(
vararg sub: String,
ignoreCase: Boolean = false,
trim: Boolean = true,
crossinline onEvent: MessageListener<T>
): Listener<T> {
return if (trim) {
val list = sub.map { it.trim() }
content({
list.all { toCheck -> it.contains(toCheck, ignoreCase = ignoreCase) }
}, {
onEvent(this, this.message.toString().trim())
})
} else {
content({
sub.all { toCheck -> it.contains(toCheck, ignoreCase = ignoreCase) }
}, {
onEvent(this, this.message.toString())
})
}
}
/**
* 如果消息的前缀是 [prefix]
*/
@ -538,7 +604,10 @@ class MessageSubscribersBuilder<T : MessagePacket<*, *>>(
* 如果 [filter] 返回 `true` 就执行 `onEvent`
*/
@MessageDsl
inline fun content(crossinline filter: T.(String) -> Boolean, crossinline onEvent: MessageListener<T>): Listener<T> =
inline fun content(
crossinline filter: T.(String) -> Boolean,
crossinline onEvent: MessageListener<T>
): Listener<T> =
subscriber {
if (filter(this, it)) onEvent(this, it)
}