feat(core): support bot.supports() for permission API

satorijs · Jul 31, 2023 · 0f373ba · 0f373ba
1 parent 88f6f23
commit 0f373ba
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 1 deletion.
diff --git a/packages/axios/src/index.ts b/packages/axios/src/index.ts
@@ -72,7 +72,7 @@ export class Quester {
     return this('PATCH', url, { ...config, data })
   }
 
-  async head(url: string, config?: AxiosRequestConfig): Promise<Dict<string>> {
+  async head(url: string, config?: AxiosRequestConfig): Promise<Dict<any>> {
     const response = await this.axios(url, { ...config, method: 'HEAD' })
     return response.headers
   }

diff --git a/packages/core/src/bot.ts b/packages/core/src/bot.ts
@@ -120,6 +120,10 @@ export abstract class Bot<T extends Bot.Config = Bot.Config> {
     const { MessageEncoder } = this.constructor as typeof Bot
     return new MessageEncoder(this, channelId, null, options).send(content)
   }
+
+  async supports(name: string, session: Partial<Session> = {}) {
+    return !!this[Universal.Methods[name]]
+  }
 }
 
 export namespace Bot {

diff --git a/packages/core/src/universal.ts b/packages/core/src/universal.ts
@@ -3,6 +3,31 @@ import { SendOptions } from './session'
 import { Dict } from 'cosmokit'
 
 export namespace Universal {
+  export const Methods = {
+    'message.get': 'getMessage',
+    'message.list': 'getMessageList',
+    'message.update': 'editMessage',
+    'message.delete': 'deleteMessage',
+    'reaction.create': 'createReaction',
+    'reaction.delete': 'deleteReaction',
+    'reaction.clear': 'clearReaction',
+    'reaction.list': 'getReactions',
+    'guild.get': 'getGuild',
+    'guild.list': 'getGuildList',
+    'guild.member.get': 'getGuildMember',
+    'guild.member.list': 'getGuildMemberList',
+    'guild.member.kick': 'kickGuildMember',
+    'guild.member.mute': 'muteGuildMember',
+    'guild.member.role': 'setGuildMemberRole',
+    'guild.role.list': 'getGuildRoles',
+    'guild.role.create': 'createGuildRole',
+    'guild.role.update': 'modifyGuildRole',
+    'guild.role.delete': 'deleteGuildRole',
+    'channel.get': 'getChannel',
+    'channel.list': 'getChannelList',
+    'channel.mute': 'muteChannel',
+  }
+
   export interface Methods {
     // message
     sendMessage(channelId: string, content: segment.Fragment, guildId?: string, options?: SendOptions): Promise<string[]>