Add MessageChain.contentEquals

This commit is contained in:
Him188 2020-04-20 22:04:39 +08:00
parent f845b5cded
commit 5af1d0435a
2 changed files with 175 additions and 1 deletions

View File

@ -17,7 +17,6 @@ import net.mamoe.mirai.utils.MiraiInternalAPI
import net.mamoe.mirai.utils.PlannedRemoval
import net.mamoe.mirai.utils.SinceMirai
import kotlin.jvm.JvmName
import kotlin.jvm.JvmOverloads
import kotlin.jvm.JvmSynthetic
/**
@ -133,6 +132,61 @@ interface Message {
@SinceMirai("0.34.0")
fun contentToString(): String
/**
* 判断内容是否与 [another] 相等.
*
* 若本函数返回 `true`, 则表明:
* - `this` [another] [contentToString] 相等
* - `this` [another] 的所有 [MessageContent] [相等][Message.equals] 且有同样的排列顺序.
*
* @sample net.mamoe.mirai.message.data.ContentEqualsTest
*/
@SinceMirai("0.38.0")
fun contentEquals(another: Message, ignoreCase: Boolean = false): Boolean {
if (!this.contentToString().equals(another.contentToString(), ignoreCase = ignoreCase)) return false
return when {
this is SingleMessage && another is SingleMessage -> true
this is SingleMessage && another is MessageChain -> another.all { it is MessageMetadata || it is PlainText }
this is MessageChain && another is SingleMessage -> this.all { it is MessageMetadata || it is PlainText }
this is MessageChain && another is MessageChain -> {
val anotherIterator = another.iterator()
/**
* 逐个判断非 [PlainText] [Message] 是否 [equals]
*/
this.forEachContent { thisElement ->
if (thisElement.isPlain()) return@forEachContent
for (it in anotherIterator) {
if (it.isPlain() || it !is MessageContent) continue
if (thisElement != it) return false
}
}
return true
}
else -> error("shouldn't be reached")
}
}
/**
* 判断内容是否与 [another] 相等.
*
* 若本函数返回 `true`, 则表明:
* - [contentToString] [another] 相等
* - `this` [MessageChain], 则只包含 [MessageMetadata] [PlainText]
*
* @sample net.mamoe.mirai.message.data.ContentEqualsTest
*/
@SinceMirai("0.38.0")
fun contentEquals(another: String, ignoreCase: Boolean = false): Boolean {
if (!this.contentToString().equals(another, ignoreCase = ignoreCase)) return false
return when (this) {
is SingleMessage -> true
is MessageChain -> this.all { it is MessageMetadata || it is PlainText }
else -> error("shouldn't be reached")
}
}
operator fun plus(another: Message): MessageChain = this.followedBy(another)
// don't remove! avoid resolution ambiguity between `CharSequence` and `Message`

View File

@ -0,0 +1,120 @@
/*
* Copyright 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
*/
package net.mamoe.mirai.message.data
import kotlin.test.Test
import kotlin.test.assertFalse
import kotlin.test.assertTrue
internal class ContentEqualsTest {
@Test
fun testContentEquals() {
val mySource = TestConstrainSingleMessage()
val image = Image("{01E9451B-70ED-EAE3-B37C-101F1EEBF5B5}.png")
assertTrue {
buildMessageChain {
+"test"
}.contentEquals(buildMessageChain {
+"te"
+mySource
+"st"
})
}
assertFalse {
buildMessageChain {
+"tests"
}.contentEquals(buildMessageChain {
+"te"
+"st"
})
}
assertTrue {
buildMessageChain {
+mySource
+"test"
+mySource
}.contentEquals("test")
}
assertTrue {
buildMessageChain {
+"test"
}.contentEquals(buildMessageChain {
+"te"
+"st"
+mySource
})
}
assertTrue {
buildMessageChain {
+"test"
+image
}.contentEquals(buildMessageChain {
+"te"
+mySource
+"st"
+image
})
}
assertTrue {
buildMessageChain {
+mySource
+"test"
+mySource
}.contentEquals("test")
}
assertTrue {
buildMessageChain {
+"test"
+image
}.contentEquals(buildMessageChain {
+"te"
+"st"
+image
+mySource
})
}
assertFalse {
buildMessageChain {
+image
+"test"
+mySource
}.contentEquals("test")
}
assertFalse {
buildMessageChain {
+"test"
+image
}.contentEquals("test")
}
assertFalse {
buildMessageChain {
+image
+"test"
}.contentEquals(buildMessageChain {
+"te"
+"st"
+image
+mySource
})
}
}
}