@alemonjs/qq-bot 2.1.0-alpha.10 → 2.1.0-alpha.11

package/README.md

@@ -46,6 +46,13 @@ qq-bot:
   # ws: 'ws://[your ip]:17157/websocket'
   # 3) 自定义模式 (用于连接类官网连接方式的指定服务端)
   # gatewayURL: 'ws://[your ip]:8080'
+  # 其他
+  # 使用 user_key
+  master_key:
+     - 'xxx'
+  # 使用 user_id
+  master_id:
+     - 'yyy'
 ```
 
 ```conf

package/lib/config.js

@@ -1,9 +1,20 @@
-import { getConfigValue } from 'alemonjs';
+import { getConfigValue, useUserHashKey } from 'alemonjs';
 
 const platform = 'qq-bot';
 const getQQBotConfig = () => {
     const value = getConfigValue() || {};
     return value[platform] || {};
 };
+const getMaster = (UserId) => {
+    const config = getQQBotConfig();
+    const master_key = config.master_key || [];
+    const master_id = config.master_id || [];
+    const UserKey = useUserHashKey({
+        Platform: platform,
+        UserId: UserId
+    });
+    const is = master_key.includes(UserKey) || master_id.includes(UserId);
+    return [is, UserKey];
+};
 
-export { getQQBotConfig, platform };
+export { getMaster, getQQBotConfig, platform };

package/lib/register.js

@@ -1,6 +1,6 @@
-import { cbpPlatform, useUserHashKey, ResultCode, createResult } from 'alemonjs';
+import { cbpPlatform, ResultCode, createResult } from 'alemonjs';
 import { GROUP_AT_MESSAGE_CREATE, C2C_MESSAGE_CREATE, DIRECT_MESSAGE_CREATE, AT_MESSAGE_CREATE, MESSAGE_CREATE } from './sends.js';
-import { getQQBotConfig, platform } from './config.js';
+import { getQQBotConfig, getMaster, platform } from './config.js';
 
 const register = (client) => {
     const config = getQQBotConfig();
@@ -22,14 +22,9 @@ const register = (client) => {
     };
     // 监听消息
     client.on('GROUP_AT_MESSAGE_CREATE', async (event) => {
-        const master_key = config?.master_key ?? [];
-        const isMaster = master_key.includes(event.author.id);
-        const UserAvatar = createUserAvatarURL(event.author.id);
         const UserId = event.author.id;
-        const UserKey = useUserHashKey({
-            Platform: platform,
-            UserId: UserId
-        });
+        const [isMaster, UserKey] = getMaster(UserId);
+        const UserAvatar = createUserAvatarURL(event.author.id);
         // 定义消
         const e = {
             name: 'message.create',
@@ -55,14 +50,9 @@ const register = (client) => {
         cbp.send(e);
     });
     client.on('C2C_MESSAGE_CREATE', async (event) => {
-        const master_key = config?.master_key ?? [];
-        const isMaster = master_key.includes(event.author.id);
-        const UserAvatar = createUserAvatarURL(event.author.id);
         const UserId = event.author.id;
-        const UserKey = useUserHashKey({
-            Platform: platform,
-            UserId: UserId
-        });
+        const [isMaster, UserKey] = getMaster(UserId);
+        const UserAvatar = createUserAvatarURL(event.author.id);
         // 定义消
         const e = {
             name: 'private.message.create',
@@ -92,15 +82,10 @@ const register = (client) => {
         // 屏蔽其他机器人的消息
         if (event?.author?.bot)
             return;
-        const master_key = config?.master_key ?? [];
-        const isMaster = master_key.includes(event.author.id);
         let msg = event?.content ?? '';
         const UserAvatar = event?.author?.avatar;
         const UserId = event.author.id;
-        const UserKey = useUserHashKey({
-            Platform: platform,
-            UserId: UserId
-        });
+        const [isMaster, UserKey] = getMaster(UserId);
         // 定义消
         const e = {
             name: 'private.message.create',
@@ -129,15 +114,10 @@ const register = (client) => {
         // 屏蔽其他机器人的消息
         if (event?.author?.bot)
             return;
-        const master_key = config?.master_key ?? [];
-        const isMaster = master_key.includes(event.author.id);
         let msg = getMessageContent(event);
         const UserAvatar = event?.author?.avatar;
         const UserId = event.author.id;
-        const UserKey = useUserHashKey({
-            Platform: platform,
-            UserId: UserId
-        });
+        const [isMaster, UserKey] = getMaster(UserId);
         // 定义消
         const e = {
             name: 'message.create',
@@ -195,15 +175,10 @@ const register = (client) => {
         // 撤回消息
         if (new RegExp(/DELETE$/).test(event.eventType))
             return;
-        const master_key = config?.master_key ?? [];
         const UserId = event.author.id;
-        const isMaster = master_key.includes(UserId);
         const msg = getMessageContent(event);
         const UserAvatar = event?.author?.avatar;
-        const UserKey = useUserHashKey({
-            Platform: platform,
-            UserId: UserId
-        });
+        const [isMaster, UserKey] = getMaster(UserId);
         // 定义消
         const e = {
             name: 'message.create',
@@ -241,16 +216,10 @@ const register = (client) => {
         // } catch (err) {
         //   createResult(ResultCode.Fail, err?.response?.data ?? err?.message ?? err, null)
         // }
-        // 如何区分私聊/群聊
         if (event.scene === 'group') {
-            const master_key = config?.master_key ?? [];
-            const isMaster = master_key.includes(event.group_member_openid);
             const UserAvatar = createUserAvatarURL(event.group_member_openid);
             const UserId = event.group_member_openid;
-            const UserKey = useUserHashKey({
-                Platform: platform,
-                UserId: UserId
-            });
+            const [isMaster, UserKey] = getMaster(UserId);
             const MessageText = event.data.resolved.button_data?.trim() || '';
             const e = {
                 name: 'interaction.create',
@@ -276,14 +245,9 @@ const register = (client) => {
             cbp.send(e);
         }
         else if (event.scene === 'c2c') {
-            const master_key = config?.master_key ?? [];
-            const isMaster = master_key.includes(event.user_openid);
             const UserAvatar = createUserAvatarURL(event.user_openid);
             const UserId = event.user_openid;
-            const UserKey = useUserHashKey({
-                Platform: platform,
-                UserId: UserId
-            });
+            const [isMaster, UserKey] = getMaster(UserId);
             const MessageText = event.data.resolved.button_data?.trim() || '';
             // 处理消息
             const e = {
@@ -306,14 +270,9 @@ const register = (client) => {
             cbp.send(e);
         }
         else if (event.scene === 'guild') {
-            const master_key = config?.master_key ?? [];
-            const isMaster = master_key.includes(event.data.resolved.user_id);
             const UserAvatar = createUserAvatarURL(event.data.resolved.user_id);
             const UserId = event.data.resolved.user_id;
-            const UserKey = useUserHashKey({
-                Platform: platform,
-                UserId: UserId
-            });
+            const [isMaster, UserKey] = getMaster(UserId);
             const MessageText = event.data.resolved.button_data?.trim() || '';
             // 处理消息
             const e = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@alemonjs/qq-bot",
-  "version": "2.1.0-alpha.10",
+  "version": "2.1.0-alpha.11",
   "description": "阿柠檬qqbot平台连接",
   "author": "lemonade",
   "license": "MIT",