Merge pull request #81 from mamoe/http-event

Http event

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/Session.kt

@@ -13,6 +13,8 @@ import kotlinx.coroutines.*
 import net.mamoe.mirai.Bot
 import net.mamoe.mirai.api.http.queue.MessageQueue
 import net.mamoe.mirai.event.Listener
+import net.mamoe.mirai.event.events.BotEvent
+import net.mamoe.mirai.event.subscribeAlways
 import net.mamoe.mirai.event.subscribeMessages
 import net.mamoe.mirai.message.MessagePacket
 import kotlin.coroutines.CoroutineContext
@@ -102,12 +104,10 @@ class TempSession internal constructor(coroutineContext: CoroutineContext) : Ses
 class AuthedSession internal constructor(val bot: Bot, coroutineContext: CoroutineContext) : Session(coroutineContext) {
 
     val messageQueue = MessageQueue()
-    private val _listener: Listener<MessagePacket<*, *>>
+    private val _listener: Listener<BotEvent>
 
     init {
-        bot.subscribeMessages {
+        _listener = bot.subscribeAlways{ this.run(messageQueue::add) }
-            _listener = always { this.run(messageQueue::add) } // this aka messagePacket
-        }
     }
 
     override fun close() {

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/data/common/BotEventDTO.kt

@@ -9,7 +9,7 @@ import net.mamoe.mirai.message.MessagePacket
 import net.mamoe.mirai.utils.MiraiExperimentalAPI
 
 @Serializable
-open class BotEventDTO : EventDTO()
+sealed class BotEventDTO : EventDTO()
 
 @UseExperimental(MiraiExperimentalAPI::class)
 fun BotEvent.toDTO() = when(this) {

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/data/common/DTO.kt

@@ -17,6 +17,6 @@ abstract class VerifyDTO : DTO {
 }
 
 @Serializable
-open class EventDTO : DTO
+abstract class EventDTO : DTO
 
 object IgnoreEventDTO : EventDTO()

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/data/common/MessageDTO.kt

@@ -32,32 +32,36 @@ data class FriendMessagePacketDTO(val sender: QQDTO) : MessagePacketDTO()
 @SerialName("GroupMessage")
 data class GroupMessagePacketDTO(val sender: MemberDTO) : MessagePacketDTO()
 
-@Serializable
-@SerialName("UnKnownMessage")
-data class UnKnownMessagePacketDTO(val msg: String) : MessagePacketDTO()
 
 // Message
 @Serializable
 @SerialName("Source")
 data class MessageSourceDTO(val uid: Long) : MessageDTO()
+
 @Serializable
 @SerialName("At")
 data class AtDTO(val target: Long, val display: String = "") : MessageDTO()
+
 @Serializable
 @SerialName("AtAll")
 data class AtAllDTO(val target: Long = 0) : MessageDTO() // target为保留字段
+
 @Serializable
 @SerialName("Face")
 data class FaceDTO(val faceId: Int) : MessageDTO()
+
 @Serializable
 @SerialName("Plain")
 data class PlainDTO(val text: String) : MessageDTO()
+
 @Serializable
 @SerialName("Image")
 data class ImageDTO(val imageId: String) : MessageDTO()
+
 @Serializable
 @SerialName("Xml")
 data class XmlDTO(val xml: String) : MessageDTO()
+
 @Serializable
 @SerialName("Unknown")
 data class UnknownMessageDTO(val text: String) : MessageDTO()
@@ -66,7 +70,7 @@ data class UnknownMessageDTO(val text: String) : MessageDTO()
 *   Abstract Class
 * */
 @Serializable
-sealed class MessagePacketDTO : DTO {
+sealed class MessagePacketDTO : EventDTO() {
     lateinit var messageChain: MessageChainDTO
 }
 
@@ -79,11 +83,15 @@ sealed class MessageDTO : DTO
 /*
     Extend function
  */
-fun MessagePacket<*, *>.toDTO(): MessagePacketDTO = when (this) {
+fun MessagePacket<*, *>.toDTO() = when (this) {
     is FriendMessage -> FriendMessagePacketDTO(QQDTO(sender))
     is GroupMessage -> GroupMessagePacketDTO(MemberDTO(sender))
-    else -> UnKnownMessagePacketDTO("UnKnown Message Packet")
+    else -> IgnoreEventDTO
-}.apply { messageChain = mutableListOf<MessageDTO>().also{ ls -> message.foreachContent { ls.add(it.toDTO()) }}}
+}.apply {
+    if (this is MessagePacketDTO) {
+        messageChain = mutableListOf<MessageDTO>().also { ls -> message.foreachContent { ls.add(it.toDTO()) } }
+    }
+}
 
 fun MessageChainDTO.toMessageChain(contact: Contact) =
     MessageChain().apply { this@toMessageChain.forEach { add(it.toMessage(contact)) } }
@@ -111,5 +119,3 @@ fun MessageDTO.toMessage(contact: Contact) = when (this) {
     is MessageSourceDTO, is UnknownMessageDTO -> PlainText("assert cannot reach")
 }
 
-
-

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/queue/MessageQueue.kt

@@ -9,26 +9,35 @@
 
 package net.mamoe.mirai.api.http.queue
 
+import net.mamoe.mirai.api.http.data.common.EventDTO
+import net.mamoe.mirai.api.http.data.common.IgnoreEventDTO
+import net.mamoe.mirai.api.http.data.common.toDTO
+import net.mamoe.mirai.event.events.BotEvent
 import net.mamoe.mirai.message.GroupMessage
-import net.mamoe.mirai.message.MessagePacket
 import net.mamoe.mirai.message.data.MessageSource
 import java.util.concurrent.ConcurrentHashMap
 import java.util.concurrent.ConcurrentLinkedDeque
 
-class MessageQueue : ConcurrentLinkedDeque<MessagePacket<*, *>>() {
+class MessageQueue : ConcurrentLinkedDeque<BotEvent>() {
 
     val quoteCache = ConcurrentHashMap<Long, GroupMessage>()
 
-    fun fetch(size: Int): List<MessagePacket<*, *>> {
+    fun fetch(size: Int): List<EventDTO> {
         var count = size
         quoteCache.clear()
-        val ret = ArrayList<MessagePacket<*, *>>(count)
+        val ret = ArrayList<EventDTO>(count)
-        while (!this.isEmpty() && count-- > 0) {
+        while (!this.isEmpty() && count > 0) {
-            val packet = pop()
+            val event = pop()
-            ret.add(packet)
 
-            if (packet is GroupMessage) {
+            event.toDTO().also {
-                addCache(packet)
+                if (it != IgnoreEventDTO) {
+                    ret.add(it)
+                    count--
+                }
+            }
+
+            if (event is GroupMessage) {
+                addCache(event)
             }
         }
         return ret

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/route/SendMessageRouteModule.kt

@@ -37,9 +37,8 @@ fun Application.messageModule() {
         miraiGet("/fetchMessage") {
             val count: Int = paramOrNull("count")
             val fetch = it.messageQueue.fetch(count)
-            val ls = Array(fetch.size) { index -> fetch[index].toDTO() }
 
-            call.respondJson(ls.toList().toJson())
+            call.respondJson(fetch.toJson())
         }
 
         miraiVerify<SendDTO>("/sendFriendMessage") {

mirai-api-http/src/main/kotlin/net/mamoe/mirai/api/http/util/Json.kt

@@ -13,7 +13,6 @@ import kotlinx.serialization.*
 import kotlinx.serialization.json.Json
 import kotlinx.serialization.modules.SerializersModule
 import net.mamoe.mirai.api.http.data.common.*
-import net.mamoe.mirai.message.data.MessageSource
 
 // 解析失败时直接返回null，由路由判断响应400状态
 @UseExperimental(ImplicitReflectionSerializer::class)
@@ -45,20 +44,46 @@ else MiraiJson.json.stringify(serializer, this)
  */
 object MiraiJson {
     val json = Json(context = SerializersModule {
-        polymorphic(MessagePacketDTO.serializer()) {
+
+        polymorphic(EventDTO.serializer()) {
             GroupMessagePacketDTO::class with GroupMessagePacketDTO.serializer()
             FriendMessagePacketDTO::class with FriendMessagePacketDTO.serializer()
-            UnKnownMessagePacketDTO::class with UnKnownMessagePacketDTO.serializer()
+
-        }
+            BotOnlineEventDTO::class with BotOnlineEventDTO.serializer()
-        polymorphic(MessageDTO.serializer()) {
+            BotOfflineEventActiveDTO::class with BotOfflineEventActiveDTO.serializer()
-            MessageSourceDTO::class with MessageSourceDTO.serializer()
+            BotOfflineEventForceDTO::class with BotOfflineEventForceDTO.serializer()
-            AtDTO::class with AtDTO.serializer()
+            BotOfflineEventDroppedDTO::class with BotOfflineEventDroppedDTO.serializer()
-            AtAllDTO::class with AtAllDTO.serializer()
+            BotReloginEventDTO::class with BotReloginEventDTO.serializer()
-            FaceDTO::class with FaceDTO.serializer()
+            BotGroupPermissionChangeEventDTO::class with BotGroupPermissionChangeEventDTO.serializer()
-            PlainDTO::class with PlainDTO.serializer()
+            BotMuteEventDTO::class with BotMuteEventDTO.serializer()
-            ImageDTO::class with ImageDTO.serializer()
+            BotUnmuteEventDTO::class with BotUnmuteEventDTO.serializer()
-            XmlDTO::class with XmlDTO.serializer()
+            BotJoinGroupEventDTO::class with BotJoinGroupEventDTO.serializer()
-            UnknownMessageDTO::class with UnknownMessageDTO.serializer()
+            GroupNameChangeEventDTO::class with GroupNameChangeEventDTO.serializer()
+            GroupEntranceAnnouncementChangeEventDTO::class with GroupEntranceAnnouncementChangeEventDTO.serializer()
+            GroupMuteAllEventDTO::class with GroupMuteAllEventDTO.serializer()
+            GroupAllowAnonymousChatEventDTO::class with GroupAllowAnonymousChatEventDTO.serializer()
+            GroupAllowConfessTalkEventDTO::class with GroupAllowConfessTalkEventDTO.serializer()
+            GroupAllowMemberInviteEventDTO::class with GroupAllowMemberInviteEventDTO.serializer()
+            MemberJoinEventDTO::class with MemberJoinEventDTO.serializer()
+            MemberLeaveEventKickDTO::class with MemberLeaveEventKickDTO.serializer()
+            MemberLeaveEventQuitDTO::class with MemberLeaveEventQuitDTO.serializer()
+            MemberCardChangeEventDTO::class with MemberCardChangeEventDTO.serializer()
+            MemberSpecialTitleChangeEventDTO::class with MemberSpecialTitleChangeEventDTO.serializer()
+            MemberPermissionChangeEventDTO::class with MemberPermissionChangeEventDTO.serializer()
+            MemberMuteEventDTO::class with MemberMuteEventDTO.serializer()
+            MemberUnmuteEventDTO::class with MemberUnmuteEventDTO.serializer()
         }
+
+        // Message Polymorphic
+//        polymorphic(MessageDTO.serializer()) {
+//            MessageSourceDTO::class with MessageSourceDTO.serializer()
+//            AtDTO::class with AtDTO.serializer()
+//            AtAllDTO::class with AtAllDTO.serializer()
+//            FaceDTO::class with FaceDTO.serializer()
+//            PlainDTO::class with PlainDTO.serializer()
+//            ImageDTO::class with ImageDTO.serializer()
+//            XmlDTO::class with XmlDTO.serializer()
+//            UnknownMessageDTO::class with UnknownMessageDTO.serializer()
+//        }
     })
 }