From 20acc3c459d3ec753fabd3955b1d24d8563b89da Mon Sep 17 00:00:00 2001
From: Him188 <Him188@mamoe.net>
Date: Tue, 28 Apr 2020 15:25:43 +0800
Subject: [PATCH] Support parsing `VipFace`

---
 .../mirai/qqandroid/message/convension.kt     |  11 ++
 .../message/data/HummerMessage.kt             | 102 +++++++++++++++---
 2 files changed, 100 insertions(+), 13 deletions(-)

diff --git a/mirai-core-qqandroid/src/commonMain/kotlin/net/mamoe/mirai/qqandroid/message/convension.kt b/mirai-core-qqandroid/src/commonMain/kotlin/net/mamoe/mirai/qqandroid/message/convension.kt
index c254dd7fa..86fe0e627 100644
--- a/mirai-core-qqandroid/src/commonMain/kotlin/net/mamoe/mirai/qqandroid/message/convension.kt
+++ b/mirai-core-qqandroid/src/commonMain/kotlin/net/mamoe/mirai/qqandroid/message/convension.kt
@@ -245,6 +245,13 @@ private fun MessageChain.cleanupRubbishMessageElements(): MessageChain {
                     return@forEach
                 }
             }
+            if (last is VipFace && element is PlainText) {
+                val l = last as VipFace
+                if (element.length == 4 + (l.count / 10) + l.kind.name.length) {
+                    last = element
+                    return@forEach
+                }
+            }
             if (last is FlashImage && element is PlainText) {
                 if (element == UNSUPPORTED_FLASH_MESSAGE_PLAIN) {
                     last = element
@@ -383,6 +390,10 @@ internal fun List<ImMsgBody.Elem>.joinToMessageChain(groupIdOrZero: Long, bot: B
             }
             element.commonElem != null -> {
                 when (element.commonElem.serviceType) {
+                    23 -> {
+                        val proto = element.commonElem.pbElem.loadAs(HummerCommelem.MsgElemInfoServtype23.serializer())
+                        list.add(VipFace(VipFace.Kind(proto.faceType, proto.faceSummary), proto.faceBubbleCount))
+                    }
                     2 -> {
                         val proto = element.commonElem.pbElem.loadAs(HummerCommelem.MsgElemInfoServtype2.serializer())
                         list.add(PokeMessage(
diff --git a/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/message/data/HummerMessage.kt b/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/message/data/HummerMessage.kt
index 7a56273f9..97ccfb170 100644
--- a/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/message/data/HummerMessage.kt
+++ b/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/message/data/HummerMessage.kt
@@ -14,6 +14,7 @@
 package net.mamoe.mirai.message.data
 
 import net.mamoe.mirai.message.data.PokeMessage.Types
+import net.mamoe.mirai.message.data.VipFace.Companion
 import net.mamoe.mirai.utils.MiraiExperimentalAPI
 import net.mamoe.mirai.utils.MiraiInternalAPI
 import net.mamoe.mirai.utils.SinceMirai
@@ -145,26 +146,101 @@ data class PokeMessage internal constructor(
         stringValue.subSequence(startIndex, endIndex)
 
     override fun contentToString(): String = "[戳一戳]"
-
-    @OptIn(MiraiExperimentalAPI::class)
-    override fun equals(other: Any?): Boolean = other is PokeMessage && other.type == this.type && other.id == this.id
-
     override fun compareTo(other: String): Int = stringValue.compareTo(other)
-
-    @OptIn(MiraiExperimentalAPI::class)
-    override fun hashCode(): Int {
-        var result = type
-        result = 31 * result + id
-        result = 31 * result + stringValue.hashCode()
-        return result
-    }
-
     //businessType=0x00000001(1)
     //pbElem=08 01 18 00 20 FF FF FF FF 0F 2A 00 32 00 38 00 50 00
     //serviceType=0x00000002(2)
 }
 
 
+////////////////////////////////////
+///////////// VIP FACE /////////////
+////////////////////////////////////
+
+/**
+ * VIP 表情. 可以发送给好友或群.
+ *
+ * 不支持发送.
+ *
+ * @see Types 使用伴生对象中的常量
+ */
+@SinceMirai("0.40.0")
+@OptIn(MiraiInternalAPI::class)
+data class VipFace internal constructor(
+    /**
+     * 使用 [Companion] 中常量.
+     */
+    val kind: Kind,
+    val count: Int
+) : HummerMessage() {
+    data class Kind(
+        val id: Int,
+        val name: String
+    )
+
+    @Suppress("DEPRECATION_ERROR", "DEPRECATION", "INVISIBLE_MEMBER", "INVISIBLE_REFERENCE")
+    companion object : Message.Key<VipFace> {
+        override val typeName: String get() = "VipFace"
+
+        @JvmStatic
+        val LiuLian = 9 to "榴莲"
+
+        @JvmStatic
+        val PingDiGuo = 1 to "平底锅"
+
+        @JvmStatic
+        val ChaoPiao = 12 to "钞票"
+
+        @JvmStatic
+        val LueLueLue = 10 to "略略略"
+
+        @JvmStatic
+        val ZhuTou = 4 to "猪头"
+
+        @JvmStatic
+        val BianBian = 6 to "便便"
+
+        @JvmStatic
+        val ZhaDan = 5 to "炸弹"
+
+        @JvmStatic
+        val AiXin = 2 to "爱心"
+
+        @JvmStatic
+        val HaHa = 3 to "哈哈"
+
+        @JvmStatic
+        val DianZan = 1 to "点赞"
+
+        @JvmStatic
+        val QinQin = 7 to "亲亲"
+
+        @JvmStatic
+        val YaoWan = 8 to "药丸"
+
+        @JvmStatic
+        val values: Array<Kind> = arrayOf(
+            LiuLian, PingDiGuo, ChaoPiao, LueLueLue, ZhuTou,
+            BianBian, ZhaDan, AiXin, HaHa, DianZan, QinQin, YaoWan
+        )
+
+        private infix fun Int.to(name: String): Kind = Kind(this, name)
+    }
+
+    @OptIn(MiraiExperimentalAPI::class)
+    private val stringValue = "[mirai:vipface:$kind,$count]"
+
+    override fun toString(): String = stringValue
+    override val length: Int get() = stringValue.length
+    override fun get(index: Int): Char = stringValue[index]
+    override fun subSequence(startIndex: Int, endIndex: Int): CharSequence =
+        stringValue.subSequence(startIndex, endIndex)
+
+    override fun contentToString(): String = "[${kind.name}]x$count"
+    override fun compareTo(other: String): Int = stringValue.compareTo(other)
+}
+
+
 ///////////////////////////////////////
 ///////////// FLASH IMAGE /////////////
 ///////////////////////////////////////