feat(core): support experimental bot.checkPermission()

satorijs · Aug 28, 2023 · 84930aa · 84930aa
1 parent 6168c47
commit 84930aa
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/adapters/onebot/src/bot/index.ts b/adapters/onebot/src/bot/index.ts
@@ -1,4 +1,4 @@
-import { Bot, Context, noop, Schema } from '@satorijs/satori'
+import { Bot, Context, noop, Schema, Session } from '@satorijs/satori'
 import { HttpServer } from '../http'
 import { WsClient, WsServer } from '../ws'
 import { QQGuildBot } from './qqguild'
@@ -95,6 +95,15 @@ export class OneBotBot<T extends OneBotBot.Config = OneBotBot.Config> extends Ba
   async muteChannel(channelId: string, guildId?: string, enable?: boolean) {
     return this.internal.setGroupWholeBan(channelId, enable)
   }
+
+  async checkPermission(name: string, session: Partial<Session>) {
+    if (name === 'onebot.group.admin') {
+      return session.author?.roles?.[0] === 'admin'
+    } else if (name === 'onebot.group.owner') {
+      return session.author?.roles?.[0] === 'owner'
+    }
+    return super.checkPermission(name, session)
+  }
 }
 
 export namespace OneBotBot {

diff --git a/packages/core/src/bot.ts b/packages/core/src/bot.ts
@@ -133,6 +133,12 @@ export abstract class Bot<T extends Bot.Config = Bot.Config> {
   async supports(name: string, session: Partial<Session> = {}) {
     return !!this[Universal.Methods[name]]
   }
+
+  async checkPermission(name: string, session: Partial<Session>) {
+    if (name.startsWith('bot.')) {
+      return this.supports(name.slice(4), session)
+    }
+  }
 }
 
 export namespace Bot {