feat(qq): delete message (#252)

satorijs · Apr 1, 2024 · 92e5626 · 92e5626
1 parent 214b4f6
commit 92e5626
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 5 deletions.
diff --git a/adapters/qq/src/bot/index.ts b/adapters/qq/src/bot/index.ts
@@ -111,6 +111,15 @@ export class QQBot<C extends Context = Context> extends Bot<C, QQBot.Config> {
   async createDirectChannel(id: string) {
     return { id, type: Universal.Channel.Type.DIRECT }
   }
+
+  async deleteMessage(channelId: string, messageId: string): Promise<void> {
+    // @TODO: need `private:`
+    try {
+      await this.internal.deleteMessage(channelId, messageId)
+    } catch (e) {
+      await this.internal.deletePrivateMessage(channelId, messageId)
+    }
+  }
 }
 
 export namespace QQBot {

diff --git a/adapters/qq/src/internal/group.ts b/adapters/qq/src/internal/group.ts
@@ -19,16 +19,24 @@ declare module './internal' {
     }): Promise<any>
     getGateway(): Promise<QQ.GetGatewayResponse>
     getGatewayBot(): Promise<QQ.GetGatewayBotResponse>
+    deleteMessage(openid: string, message_id: string): Promise<any>
+    deletePrivateMessage(userid: string, message_id: string): Promise<any>
   }
 }
 
 GroupInternal.define(false, {
   '/v2/groups/{channel.id}/messages': {
     POST: 'sendMessage',
   },
+  '/v2/groups/{channel.id}/messages/{message.id}': {
+    DELETE: 'deleteMessage',
+  },
   '/v2/users/{user.id}/messages': {
     POST: 'sendPrivateMessage',
   },
+  '/v2/users/{user.id}/messages/{message.id}': {
+    DELETE: 'deletePrivateMessage',
+  },
   '/v2/users/{user.id}/files': {
     POST: 'sendFilePrivate',
   },

diff --git a/adapters/qq/src/message.ts b/adapters/qq/src/message.ts
@@ -88,11 +88,9 @@ export class QQGuildMessageEncoder<C extends Context = Context> extends MessageE
     const session = this.bot.session()
     session.type = 'send'
     // await decodeMessage(this.bot, r, session.event.message = {}, session.event)
-    if (isDirect) {
-      session.guildId = this.session.guildId
-      session.channelId = this.channelId
-      session.isDirect = true
-    }
+    session.guildId = this.session.guildId
+    session.channelId = this.channelId
+    session.isDirect = isDirect
 
     // https://bot.q.qq.com/wiki/develop/api/gateway/direct_message.html#%E6%B3%A8%E6%84%8F
     /**
@@ -250,6 +248,8 @@ export class QQMessageEncoder<C extends Context = Context> extends MessageEncode
           if (resp.id) {
             session.messageId = resp.id
             session.timestamp = new Date(resp.timestamp).valueOf()
+            session.channelId = this.session.channelId
+            session.guildId = this.session.guildId
             session.app.emit(session, 'send', session)
             this.results.push(session.event.message)
           } else if (resp.code === 304023 && this.bot.config.intents & QQ.Intents.MESSAGE_AUDIT) {