@alemonjs/qq-bot 2.1.0-alpha.3 → 2.1.0-alpha.30

package/lib/register.js

@@ -1,326 +1,450 @@
-import { getConfigValue, cbpPlatform, useUserHashKey, createResult, ResultCode } 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, getMaster } from './config.js';
 
-const platform = 'qq-bot';
-const getQQBotConfig = () => {
-    let value = getConfigValue();
-    if (!value)
-        value = {};
-    return value[platform] || {};
-};
 const register = (client) => {
     const config = getQQBotConfig();
-    /**
-     * 连接 alemonjs 服务器。
-     * 向 alemonjs 推送标准信息
-     */
     const port = process.env?.port || config?.port || 17117;
     const url = `ws://127.0.0.1:${port}`;
     const cbp = cbpPlatform(url);
-    /**
-     * group
-     *
-     * GROUP_AT_MESSAGE_CREATE
-     * C2C_MESSAGE_CREATE
-     */
     const createUserAvatarURL = (author_id) => {
         return `https://q.qlogo.cn/qqapp/${config.app_id}/${author_id}/640`;
     };
-    // 监听消息
-    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({
+    client.on('GROUP_ADD_ROBOT', event => {
+        const e = {
+            name: 'guild.join',
             Platform: platform,
-            UserId: UserId
-        });
-        // 定义消
+            GuildId: event.group_openid,
+            ChannelId: event.group_openid,
+            SpaceId: `GROUP:${event.group_openid}`,
+            UserId: event.op_member_openid,
+            UserKey: event.op_member_openid,
+            UserAvatar: createUserAvatarURL(event.op_member_openid),
+            IsMaster: false,
+            IsBot: false,
+            OpenId: `C2C:${event.op_member_openid}`,
+            MessageId: '',
+            CreateAt: Date.now(),
+            tag: 'GROUP_ADD_ROBOT',
+            value: event
+        };
+        cbp.send(e);
+    });
+    client.on('GROUP_DEL_ROBOT', event => {
+        const e = {
+            name: 'guild.exit',
+            Platform: platform,
+            GuildId: event.group_openid,
+            ChannelId: event.group_openid,
+            SpaceId: `GROUP:${event.group_openid}`,
+            UserId: event.op_member_openid,
+            UserKey: event.op_member_openid,
+            UserAvatar: createUserAvatarURL(event.op_member_openid),
+            IsMaster: false,
+            IsBot: false,
+            OpenId: `C2C:${event.op_member_openid}`,
+            MessageId: '',
+            CreateAt: Date.now(),
+            tag: 'GROUP_DEL_ROBOT',
+            value: event
+        };
+        cbp.send(e);
+    });
+    client.on('GROUP_AT_MESSAGE_CREATE', event => {
+        const UserId = event.author.id;
+        const [isMaster, UserKey] = getMaster(UserId);
+        const UserAvatar = createUserAvatarURL(event.author.id);
         const e = {
             name: 'message.create',
-            // 事件类型
             Platform: platform,
-            // guild
             GuildId: event.group_id,
             ChannelId: event.group_id,
-            // 用户Id
+            SpaceId: `GROUP:${event.group_id}`,
             UserId: event.author.id,
             UserKey,
             UserAvatar: UserAvatar,
             IsMaster: isMaster,
             IsBot: false,
-            // 格式化数据
             MessageId: event.id,
             MessageText: event.content?.trim(),
-            OpenId: event.author.member_openid,
+            OpenId: `C2C:${event.author.member_openid}`,
             CreateAt: Date.now(),
             tag: 'GROUP_AT_MESSAGE_CREATE',
             value: event
         };
         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);
+    client.on('C2C_MESSAGE_CREATE', event => {
         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',
-            // 事件类型
             Platform: platform,
-            // 用户Id
             UserId: event.author.id,
             UserKey,
             UserAvatar: UserAvatar,
             IsMaster: isMaster,
             IsBot: false,
-            // 格式化数据
             MessageId: event.id,
             MessageText: event.content?.trim(),
             CreateAt: Date.now(),
-            OpenId: event.author.user_openid,
-            //
+            OpenId: `C2C:${event.author.user_openid}`,
             tag: 'C2C_MESSAGE_CREATE',
             value: event
         };
         cbp.send(e);
     });
-    /**
-     * guild
-     */
-    client.on('DIRECT_MESSAGE_CREATE', async (event) => {
-        // 屏蔽其他机器人的消息
-        if (event?.author?.bot)
+    client.on('DIRECT_MESSAGE_CREATE', event => {
+        if (event?.author?.bot) {
             return;
-        const master_key = config?.master_key ?? [];
-        const isMaster = master_key.includes(event.author.id);
-        let msg = event?.content ?? '';
+        }
+        const 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',
-            // 事件类型
             Platform: platform,
-            //
-            // GuildId: event.guild_id,
-            // ChannelId: event.channel_id,
-            // 用户Id
             UserId: event?.author?.id ?? '',
             UserKey,
             UserName: event?.author?.username ?? '',
             UserAvatar: UserAvatar,
             IsMaster: isMaster,
             IsBot: event.author?.bot,
-            // message
             MessageId: event.id,
             MessageText: msg?.trim(),
-            OpenId: event.guild_id,
+            OpenId: `DIRECT:${event.guild_id}`,
             CreateAt: Date.now(),
-            //
             tag: 'DIRECT_MESSAGE_CREATE',
             value: event
         };
         cbp.send(e);
     });
-    // 监听消息
-    client.on('AT_MESSAGE_CREATE', async (event) => {
-        // 屏蔽其他机器人的消息
-        if (event?.author?.bot)
+    client.on('AT_MESSAGE_CREATE', event => {
+        if (event?.author?.bot) {
             return;
-        const master_key = config?.master_key ?? [];
-        const isMaster = master_key.includes(event.author.id);
-        let msg = getMessageContent(event);
+        }
+        const 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',
-            // 事件类型
             Platform: platform,
             GuildId: event.guild_id,
             ChannelId: event.channel_id,
+            SpaceId: `GUILD:${event.channel_id}`,
             IsMaster: isMaster,
-            // 用户Id
             UserId: event?.author?.id ?? '',
             UserKey,
             UserName: event?.author?.username ?? '',
             UserAvatar: UserAvatar,
             IsBot: event.author?.bot,
-            // message
             MessageId: event.id,
             MessageText: msg?.trim(),
-            OpenId: event.guild_id,
+            OpenId: `DIRECT:${event.guild_id}`,
             CreateAt: Date.now(),
-            //
             tag: 'AT_MESSAGE_CREATE',
             value: event
         };
         cbp.send(e);
     });
-    /**
-     *
-     * @param event
-     * @returns
-     */
     const getMessageContent = event => {
         let msg = event?.content ?? '';
-        // 艾特消息处理
-        const at_users = [];
+        const atUsers = [];
         if (event.mentions) {
-            // 去掉@ 转为纯消息
             for (const item of event.mentions) {
-                at_users.push({
+                atUsers.push({
                     id: item.id
                 });
             }
-            // 循环删除文本中的at信息并去除前后空格
-            at_users.forEach(item => {
+            atUsers.forEach(item => {
                 msg = msg.replace(`<@!${item.id}>`, '').trim();
             });
         }
         return msg;
     };
-    // 私域 -
-    client.on('MESSAGE_CREATE', async (event) => {
-        // 屏蔽其他机器人的消息
-        if (event.author?.bot)
+    client.on('MESSAGE_CREATE', event => {
+        if (event.author?.bot) {
             return;
-        // 撤回消息
-        if (new RegExp(/DELETE$/).test(event.eventType))
+        }
+        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',
-            // 事件类型
             Platform: platform,
-            //
             GuildId: event.guild_id,
             ChannelId: event.channel_id,
+            SpaceId: `GUILD:${event.channel_id}`,
             UserId: event?.author?.id ?? '',
             UserKey,
             UserName: event?.author?.username ?? '',
             UserAvatar: UserAvatar,
             IsMaster: isMaster,
             IsBot: false,
-            // message
             MessageId: event.id,
             MessageText: msg?.trim(),
-            OpenId: event.guild_id,
+            OpenId: `DIRECT:${event.guild_id}`,
             CreateAt: Date.now(),
-            //
             tag: 'MESSAGE_CREATE',
             value: event
         };
         cbp.send(e);
     });
-    client.on('ERROR', console.error);
-    const sendMesage = async (event, val) => {
-        if (val.length < 0)
-            Promise.all([]);
-        // 打  tag
-        const tag = event.tag;
-        // 群at
-        if (tag == 'GROUP_AT_MESSAGE_CREATE') {
-            return await GROUP_AT_MESSAGE_CREATE(client, event, val);
+    client.on('INTERACTION_CREATE', event => {
+        if (event.scene === 'group') {
+            const UserAvatar = createUserAvatarURL(event.group_member_openid);
+            const UserId = event.group_member_openid;
+            const [isMaster, UserKey] = getMaster(UserId);
+            const MessageText = event.data.resolved.button_data?.trim() || '';
+            const e = {
+                name: 'interaction.create',
+                Platform: platform,
+                GuildId: event.group_openid,
+                ChannelId: event.group_openid,
+                SpaceId: `GROUP:${event.group_openid}`,
+                UserId: event.group_member_openid,
+                UserKey,
+                UserAvatar: UserAvatar,
+                IsMaster: isMaster,
+                IsBot: false,
+                MessageId: `INTERACTION_CREATE:${event.id}`,
+                MessageText: MessageText,
+                OpenId: `C2C:${event.group_member_openid}`,
+                tag: 'INTERACTION_CREATE_GROUP',
+                CreateAt: Date.now(),
+                value: event
+            };
+            cbp.send(e);
         }
-        // 私聊
-        if (tag == 'C2C_MESSAGE_CREATE') {
-            return await C2C_MESSAGE_CREATE(client, event, val);
+        else if (event.scene === 'c2c') {
+            const UserAvatar = createUserAvatarURL(event.user_openid);
+            const UserId = event.user_openid;
+            const [isMaster, UserKey] = getMaster(UserId);
+            const MessageText = event.data.resolved.button_data?.trim() || '';
+            const e = {
+                name: 'private.interaction.create',
+                Platform: platform,
+                UserId: event.user_openid,
+                UserKey,
+                UserAvatar: UserAvatar,
+                IsMaster: isMaster,
+                IsBot: false,
+                MessageId: event.id,
+                MessageText: MessageText,
+                OpenId: `C2C:${event.user_openid}`,
+                CreateAt: Date.now(),
+                tag: 'INTERACTION_CREATE_C2C',
+                value: event
+            };
+            cbp.send(e);
         }
-        // 频道私聊
-        if (tag == 'DIRECT_MESSAGE_CREATE') {
-            return await DIRECT_MESSAGE_CREATE(client, event, val);
+        else if (event.scene === 'guild') {
+            const UserAvatar = createUserAvatarURL(event.data.resolved.user_id);
+            const UserId = event.data.resolved.user_id;
+            const [isMaster, UserKey] = getMaster(UserId);
+            const MessageText = event.data.resolved.button_data?.trim() || '';
+            const e = {
+                name: 'interaction.create',
+                Platform: platform,
+                GuildId: event.guild_id,
+                ChannelId: event.channel_id,
+                SpaceId: `GUILD:${event.channel_id}`,
+                UserId: event.data.resolved.user_id,
+                UserKey,
+                UserAvatar: UserAvatar,
+                IsMaster: isMaster,
+                IsBot: false,
+                MessageId: event.data.resolved.message_id,
+                MessageText: MessageText,
+                OpenId: `DIRECT:${event.guild_id}`,
+                CreateAt: Date.now(),
+                tag: 'INTERACTION_CREATE_GUILD',
+                value: event
+            };
+            cbp.send(e);
         }
-        // 频道at
-        if (tag == 'AT_MESSAGE_CREATE') {
-            return await AT_MESSAGE_CREATE(client, event, val);
+        else {
+            logger.warn({
+                code: ResultCode.Fail,
+                message: '暂未更新支持此类型的交互事件',
+                data: event
+            });
         }
-        // 频道消息
-        if (tag == 'MESSAGE_CREATE') {
-            return await MESSAGE_CREATE(client, event, val);
+    });
+    client.on('ERROR', console.error);
+    const api = {
+        active: {
+            send: {
+                channel: async (SpaceId, val) => {
+                    if (/^GUILD:/.test(SpaceId)) {
+                        const id = SpaceId.replace('GUILD:', '');
+                        return await AT_MESSAGE_CREATE(client, {
+                            ChannelId: id
+                        }, val);
+                    }
+                    if (/^GROUP:/.test(SpaceId)) {
+                        const id = SpaceId.replace('GROUP:', '');
+                        return await GROUP_AT_MESSAGE_CREATE(client, {
+                            ChannelId: id
+                        }, val);
+                    }
+                    return [];
+                },
+                user: async (OpenId, val) => {
+                    if (/^C2C:/.test(OpenId)) {
+                        const id = OpenId.replace('C2C:', '');
+                        return await C2C_MESSAGE_CREATE(client, {
+                            UserId: id
+                        }, val);
+                    }
+                    else if (/^DIRECT:/.test(OpenId)) {
+                        const id = OpenId.replace('DIRECT:', '');
+                        return await DIRECT_MESSAGE_CREATE(client, {
+                            UserId: id
+                        }, val);
+                    }
+                    else if (/^GUILD:/.test(OpenId)) {
+                        const id = OpenId.replace('GUILD:', '');
+                        return await AT_MESSAGE_CREATE(client, {
+                            ChannelId: id
+                        }, val);
+                    }
+                    return [];
+                }
+            }
+        },
+        use: {
+            send: async (event, val) => {
+                if (!val || val.length <= 0) {
+                    return [];
+                }
+                const tag = event.tag;
+                if (tag === 'GROUP_AT_MESSAGE_CREATE') {
+                    return await GROUP_AT_MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'C2C_MESSAGE_CREATE') {
+                    return await C2C_MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'DIRECT_MESSAGE_CREATE') {
+                    return await DIRECT_MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'AT_MESSAGE_CREATE') {
+                    return await AT_MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'MESSAGE_CREATE') {
+                    return await MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'INTERACTION_CREATE_GROUP') {
+                    return await GROUP_AT_MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'INTERACTION_CREATE_C2C') {
+                    return await C2C_MESSAGE_CREATE(client, event, val);
+                }
+                if (tag === 'INTERACTION_CREATE_GUILD') {
+                    return await AT_MESSAGE_CREATE(client, event, val);
+                }
+                return [];
+            },
+            mention: event => {
+                const value = event.value || {};
+                const tag = event.tag;
+                const Metions = [];
+                if (tag === 'GROUP_AT_MESSAGE_CREATE' || tag === 'C2C_MESSAGE_CREATE') {
+                    return new Promise(resolve => resolve(Metions));
+                }
+                if (value.mentions) {
+                    const mentions = event.value['mentions'];
+                    const MessageMention = mentions.map(item => {
+                        const UserId = item.id;
+                        const [isMaster, UserKey] = getMaster(UserId);
+                        return {
+                            UserId: item.id,
+                            IsMaster: isMaster,
+                            UserName: item.username,
+                            IsBot: item.bot,
+                            UserKey: UserKey
+                        };
+                    });
+                    return new Promise(resolve => resolve(MessageMention));
+                }
+                else {
+                    return new Promise(resolve => resolve(Metions));
+                }
+            }
         }
-        return Promise.all([]);
-    };
-    const getMesion = async (event) => {
-        event.value;
-        event.tag;
-        // const event = e.value
-        const Metions = [];
-        // group
-        return Metions;
     };
-    const sendMessageChannel = async (channel_id, data) => {
-        if (!channel_id || typeof channel_id !== 'string') {
-            throw new Error('Invalid channel_id: channel_id must be a string');
+    const onactions = async (data, consume) => {
+        try {
+            if (data.action === 'me.info') {
+                const res = await client.usersMe();
+                const UserId = res.id;
+                const [isMaster, UserKey] = getMaster(UserId);
+                const botInfo = {
+                    UserId: res?.id,
+                    UserName: res?.username,
+                    UserAvatar: createUserAvatarURL(res?.id),
+                    IsBot: true,
+                    IsMaster: isMaster,
+                    UserKey: UserKey
+                };
+                consume([createResult(ResultCode.Ok, '请求完成', botInfo)]);
+            }
+            else if (data.action === 'message.send') {
+                const event = data.payload.event;
+                const paramFormat = data.payload.params.format;
+                const res = await api.use.send(event, paramFormat);
+                consume(res);
+            }
+            else if (data.action === 'mention.get') {
+                const event = data.payload.event;
+                const metions = await api.use.mention(event);
+                consume([createResult(ResultCode.Ok, '请求完成', metions)]);
+            }
+            else if (data.action === 'message.send.channel') {
+                const channel_id = data.payload.ChannelId;
+                const paramFormat = data.payload.params.format;
+                const res = await api.active.send.channel(channel_id, paramFormat);
+                consume(res);
+            }
+            else if (data.action === 'message.send.user') {
+                const user_id = data.payload.UserId;
+                const paramFormat = data.payload.params.format;
+                const res = await api.active.send.user(user_id, paramFormat);
+                consume(res);
+            }
+            else {
+                consume([createResult(ResultCode.Fail, '未知请求，请尝试升级版本', null)]);
+            }
         }
-        return await AT_MESSAGE_CREATE(client, {
-            ChannelId: channel_id
-        }, data);
-    };
-    const sendMessageUser = async (user_id, data) => {
-        if (!user_id || typeof user_id !== 'string') {
-            throw new Error('Invalid user_id: user_id must be a string');
+        catch (error) {
+            consume([createResult(ResultCode.Fail, '请求失败', error)]);
         }
-        return await C2C_MESSAGE_CREATE(client, {
-            OpenId: user_id
-        }, data);
     };
-    // 处理行为
-    cbp.onactions(async (data, consume) => {
-        if (data.action === 'message.send') {
-            // 消息发送
-            const event = data.payload.event;
-            const paramFormat = data.payload.params.format;
-            // 消费
-            const res = await sendMesage(event, paramFormat);
-            consume(res);
-        }
-        else if (data.action === 'mention.get') {
-            const event = data.payload.event;
-            // 获取提及
-            const metions = await getMesion(event);
-            // 消费
-            consume([createResult(ResultCode.Ok, '请求完成', metions)]);
-        }
-        else if (data.action === 'message.send.channel') {
-            // 主动发送消息到频道
-            const channel_id = data.payload.ChannelId;
-            const paramFormat = data.payload.params.format;
-            const res = await sendMessageChannel(channel_id, paramFormat);
-            consume(res);
+    cbp.onactions((data, consume) => void onactions(data, consume));
+    const onapis = async (data, consume) => {
+        const key = data.payload?.key;
+        if (client[key]) {
+            const params = data.payload.params;
+            try {
+                const res = await client[key](...params);
+                consume([createResult(ResultCode.Ok, '请求完成', res)]);
+            }
+            catch (error) {
+                consume([createResult(ResultCode.Fail, '请求失败', error)]);
+            }
         }
-        else if (data.action === 'message.send.user') {
-            // 主动发送消息到用户
-            const user_id = data.payload.UserId;
-            const paramFormat = data.payload.params.format;
-            const res = await sendMessageUser(user_id, paramFormat);
-            consume(res);
+        else {
+            consume([createResult(ResultCode.Fail, '未知请求，请尝试升级版本', null)]);
         }
-    });
+    };
+    cbp.onapis((data, consume) => void onapis(data, consume));
 };
 
-export { getQQBotConfig, platform, register };
+export { register };