feat(adapter): add basic support for bot.kickGuildMember()

also bot.muteGuildMember() and bot.muteChannel() but not widely supported
koishijs · May 3, 2022 · b85397f · b85397f
1 parent de9b6cf
commit b85397f
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 1 deletion.
diff --git a/packages/core/src/bot.ts b/packages/core/src/bot.ts
@@ -171,10 +171,13 @@ export namespace Bot {
     // guild member
     getGuildMember(guildId: string, userId: string): Promise<GuildMember>
     getGuildMemberList(guildId: string): Promise<GuildMember[]>
+    kickGuildMember(guildId: string, userId: string, permanent?: boolean): Promise<void>
+    muteGuildMember(guildId: string, userId: string, duration: number, reason?: string): Promise<void>
 
     // channel
     getChannel(channelId: string, guildId?: string): Promise<Channel>
     getChannelList(guildId: string): Promise<Channel[]>
+    muteChannel(channelId: string, guildId?: string, enable?: boolean): Promise<void>
 
     // request
     handleFriendRequest(messageId: string, approve: boolean, comment?: string): Promise<void>

diff --git a/plugins/adapter/discord/src/bot.ts b/plugins/adapter/discord/src/bot.ts
@@ -153,6 +153,10 @@ export class DiscordBot extends Bot<BotConfig> {
     }
   }
 
+  async kickGuildMember(guildId: string, userId: string) {
+    return this.internal.removeGuildMember(guildId, userId)
+  }
+
   async getGuild(guildId: string) {
     const data = await this.internal.getGuild(guildId)
     return adaptGuild(data)

diff --git a/plugins/adapter/onebot/src/bot.ts b/plugins/adapter/onebot/src/bot.ts
@@ -143,6 +143,18 @@ export class OneBotBot extends Bot<BotConfig> {
     return data.map(OneBot.adaptGuildMember)
   }
 
+  async kickGuildMember(guildId: string, userId: string, permanent?: boolean) {
+    return this.internal.setGroupKick(guildId, userId, permanent)
+  }
+
+  async muteGuildMember(guildId: string, userId: string, duration: number) {
+    return this.internal.setGroupBan(guildId, userId, duration / 1000)
+  }
+
+  async muteChannel(channelId: string, guildId?: string, enable?: boolean) {
+    return this.internal.setGroupWholeBan(channelId, enable)
+  }
+
   protected async sendGuildMessage(guildId: string, channelId: string, content: string) {
     const session = await this.session({ content, subtype: 'group', guildId, channelId })
     if (!session?.content) return []

diff --git a/plugins/adapter/telegram/src/bot.ts b/plugins/adapter/telegram/src/bot.ts
@@ -1,4 +1,4 @@
-import { Adapter, assertProperty, Bot, Dict, Logger, Quester, renameProperty, Schema } from 'koishi'
+import { Adapter, assertProperty, Bot, Dict, Logger, Quester, renameProperty, Schema, Time } from 'koishi'
 import * as Telegram from './types'
 import { AdapterConfig, adaptGuildMember, adaptUser } from './utils'
 import { Sender } from './sender'
@@ -137,6 +137,16 @@ export class TelegramBot extends Bot<BotConfig> {
     return data.map(adaptGuildMember)
   }
 
+  async kickGuildMember(chat_id: string, user_id: string | number, permanent?: boolean) {
+    user_id = +user_id
+    await this.internal.banChatMember({
+      chat_id,
+      user_id,
+      until_date: Date.now() + (permanent ? 0 : Time.minute),
+      revoke_messages: true,
+    })
+  }
+
   setGroupLeave(chat_id: string) {
     return this.internal.leaveChat({ chat_id })
   }