@alemonjs/qq-bot 0.0.18 → 0.0.20

package/lib/register.js

@@ -0,0 +1,405 @@
+import { getConfigValue, useUserHashKey, onProcessor } from 'alemonjs';
+import { AT_MESSAGE_CREATE, GROUP_AT_MESSAGE_CREATE, DIRECT_MESSAGE_CREATE, C2C_MESSAGE_CREATE, MESSAGE_CREATE } from './send/index.js';
+import { isGuild } from './utils.js';
+
+const platform = 'qq-bot';
+const register = (client) => {
+    let value = getConfigValue();
+    if (!value)
+        value = {};
+    const config = value[platform];
+    /**
+     * 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 url = createUserAvatarURL(event.author.id);
+        const UserAvatar = {
+            toBuffer: async () => {
+                const arrayBuffer = await fetch(url).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer);
+            },
+            toBase64: async () => {
+                const arrayBuffer = await fetch(url).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer).toString('base64');
+            },
+            toURL: async () => {
+                return url;
+            }
+        };
+        const UserId = event.author.id;
+        const UserKey = useUserHashKey({
+            Platform: platform,
+            UserId: UserId
+        });
+        // 定义消
+        const e = {
+            name: 'message.create',
+            // 事件类型
+            Platform: platform,
+            // guild
+            GuildId: event.group_id,
+            ChannelId: event.group_id,
+            // 用户Id
+            UserId: event.author.id,
+            UserKey,
+            UserAvatar: UserAvatar,
+            IsMaster: isMaster,
+            IsBot: false,
+            // 格式化数据
+            MessageId: event.id,
+            MessageText: event.content?.trim(),
+            OpenId: event.author.member_openid,
+            CreateAt: Date.now(),
+            tag: 'GROUP_AT_MESSAGE_CREATE',
+            value: null
+        };
+        // 处理消息
+        onProcessor('message.create', e, event);
+    });
+    client.on('C2C_MESSAGE_CREATE', async (event) => {
+        const master_key = config?.master_key ?? [];
+        const isMaster = master_key.includes(event.author.id);
+        const url = createUserAvatarURL(event.author.id);
+        const UserAvatar = {
+            toBuffer: async () => {
+                const arrayBuffer = await fetch(url).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer);
+            },
+            toBase64: async () => {
+                const arrayBuffer = await fetch(url).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer).toString('base64');
+            },
+            toURL: async () => {
+                return url;
+            }
+        };
+        const UserId = event.author.id;
+        const UserKey = useUserHashKey({
+            Platform: platform,
+            UserId: UserId
+        });
+        // 定义消
+        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,
+            //
+            tag: 'C2C_MESSAGE_CREATE',
+            value: null
+        };
+        // 处理消息
+        onProcessor('private.message.create', e, event);
+    });
+    /**
+     * guild
+     */
+    client.on('DIRECT_MESSAGE_CREATE', async (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 UserAvatar = {
+            toBuffer: async () => {
+                const arrayBuffer = await fetch(event.author.avatar).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer);
+            },
+            toBase64: async () => {
+                const arrayBuffer = await fetch(event?.author?.avatar).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer).toString('base64');
+            },
+            toURL: async () => {
+                return event?.author?.avatar;
+            }
+        };
+        const UserId = event.author.id;
+        const UserKey = useUserHashKey({
+            Platform: platform,
+            UserId: 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,
+            CreateAt: Date.now(),
+            //
+            tag: 'DIRECT_MESSAGE_CREATE',
+            //
+            value: null
+        };
+        // 处理消息
+        onProcessor('private.message.create', e, event);
+    });
+    // 监听消息
+    client.on('AT_MESSAGE_CREATE', async (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 UserAvatar = {
+            toBuffer: async () => {
+                const arrayBuffer = await fetch(event.author.avatar).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer);
+            },
+            toBase64: async () => {
+                const arrayBuffer = await fetch(event?.author?.avatar).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer).toString('base64');
+            },
+            toURL: async () => {
+                return event?.author?.avatar;
+            }
+        };
+        const UserId = event.author.id;
+        const UserKey = useUserHashKey({
+            Platform: platform,
+            UserId: UserId
+        });
+        // 定义消
+        const e = {
+            name: 'message.create',
+            // 事件类型
+            Platform: platform,
+            GuildId: event.guild_id,
+            ChannelId: 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,
+            CreateAt: Date.now(),
+            //
+            tag: 'AT_MESSAGE_CREATE',
+            //
+            value: null
+        };
+        // 处理消息
+        onProcessor('message.create', e, event);
+    });
+    /**
+     *
+     * @param event
+     * @returns
+     */
+    const getMessageContent = event => {
+        let msg = event?.content ?? '';
+        // 艾特消息处理
+        const at_users = [];
+        if (event.mentions) {
+            // 去掉@ 转为纯消息
+            for (const item of event.mentions) {
+                at_users.push({
+                    id: item.id
+                });
+            }
+            // 循环删除文本中的at信息并去除前后空格
+            at_users.forEach(item => {
+                msg = msg.replace(`<@!${item.id}>`, '').trim();
+            });
+        }
+        return msg;
+    };
+    // 私域 -
+    client.on('MESSAGE_CREATE', async (event) => {
+        // 屏蔽其他机器人的消息
+        if (event.author?.bot)
+            return;
+        // 撤回消息
+        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 = {
+            toBuffer: async () => {
+                const arrayBuffer = await fetch(event.author.avatar).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer);
+            },
+            toBase64: async () => {
+                const arrayBuffer = await fetch(event?.author?.avatar).then(res => res.arrayBuffer());
+                return Buffer.from(arrayBuffer).toString('base64');
+            },
+            toURL: async () => {
+                return event?.author?.avatar;
+            }
+        };
+        const UserKey = useUserHashKey({
+            Platform: platform,
+            UserId: UserId
+        });
+        // 定义消
+        const e = {
+            name: 'message.create',
+            // 事件类型
+            Platform: platform,
+            //
+            GuildId: event.guild_id,
+            ChannelId: 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,
+            CreateAt: Date.now(),
+            //
+            tag: 'MESSAGE_CREATE',
+            value: null
+        };
+        // 处理消息
+        onProcessor('message.create', e, event);
+    });
+    client.on('ERROR', console.error);
+};
+const createClientAPI = (client) => {
+    return {
+        platform,
+        api: {
+            // 主动消息
+            active: {
+                send: {
+                    channel: async (channel_id, data) => {
+                        if (isGuild(channel_id)) {
+                            return await AT_MESSAGE_CREATE(client, {
+                                ChannelId: channel_id
+                            }, data);
+                        }
+                        else {
+                            // 需要message_id 。如果没有，则是主动消息，在group中，只能发送4条。
+                            return await GROUP_AT_MESSAGE_CREATE(client, {
+                                GuildId: channel_id
+                            }, data);
+                        }
+                    },
+                    user: async (user_id, data) => {
+                        if (isGuild(user_id)) {
+                            return await DIRECT_MESSAGE_CREATE(client, {
+                                OpenId: user_id
+                            }, data);
+                        }
+                        else {
+                            return await C2C_MESSAGE_CREATE(client, {
+                                OpenId: user_id
+                            }, data);
+                        }
+                    }
+                }
+            },
+            // 被动消息
+            use: {
+                send: async (event, val) => {
+                    if (val.length < 0)
+                        ;
+                    // 打  tag
+                    const tag = event.tag;
+                    try {
+                        // 群at
+                        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);
+                        }
+                        // 频道at
+                        if (tag == 'AT_MESSAGE_CREATE') {
+                            return await AT_MESSAGE_CREATE(client, event, val);
+                        }
+                        // 频道消息
+                        if (tag == 'MESSAGE_CREATE') {
+                            return await MESSAGE_CREATE(client, event, val);
+                        }
+                    }
+                    catch (error) {
+                        return [error];
+                    }
+                    return [];
+                },
+                mention: async (e) => {
+                    const event = e.value;
+                    const tag = e.tag;
+                    // const event = e.value
+                    const Metions = [];
+                    // group
+                    if (tag == 'GROUP_AT_MESSAGE_CREATE' || 'C2C_MESSAGE_CREATE')
+                        return Metions;
+                    // guild
+                    if (event.mentions) {
+                        const mentions = e.value['mentions'];
+                        // 艾特消息处理
+                        const MessageMention = mentions.map(item => {
+                            return {
+                                UserId: item.id,
+                                IsMaster: false,
+                                UserName: item.username,
+                                IsBot: item.bot,
+                                UserKey: useUserHashKey({
+                                    Platform: 'qq-guild-bot',
+                                    UserId: item.id
+                                })
+                            };
+                        }) ?? [];
+                        return MessageMention;
+                    }
+                    else {
+                        return Metions;
+                    }
+                }
+            }
+        }
+    };
+};
+
+export { createClientAPI, platform, register };

package/lib/sdk/api.d.ts

@@ -19,9 +19,8 @@ declare class QQBotAPI {
     API_URL: string;
     /**
      * 得到鉴权
-     * @param appId
+     * @param app_id
      * @param clientSecret
-     * @param url
      * @returns
      */
     getAuthentication(app_id: string, clientSecret: string): Promise<axios.AxiosResponse<any, any>>;
@@ -63,8 +62,7 @@ declare class QQBotAPI {
     /**
      * 发送群聊消息
      * @param group_openid
-     * @param content
-     * @param msg_id
+     * @param data
      * @returns
      */
     groupOpenMessages(group_openid: string, data: ApiRequestData): Promise<{
@@ -74,8 +72,7 @@ declare class QQBotAPI {
     /**
      * 发送私聊富媒体文件
      * @param openid
-     * @param content
-     * @param file_type
+     * @param data
      * @returns
      *  1 图文 2 视频 3 语言 4 文件
      * 图片：png/jpg，视频：mp4，语音：silk
@@ -93,8 +90,7 @@ declare class QQBotAPI {
     /**
      * 发送私聊富媒体文件
      * @param openid
-     * @param content
-     * @param file_type
+     * @param data
      * @returns
      *  1 图文 2 视频 3 语言 4 文件
      * 图片：png/jpg，视频：mp4，语音：silk
@@ -112,8 +108,7 @@ declare class QQBotAPI {
     /**
      * 发送群里文件
      * @param openid
-     * @param content
-     * @param file_type
+     * @param data
      * @returns
      *  1 图文 2 视频 3 语言 4 文件
      * 图片：png/jpg，视频：mp4，语音：silk

package/lib/sdk/api.js

@@ -18,9 +18,8 @@ class QQBotAPI {
     API_URL = 'https://api.sgroup.qq.com';
     /**
      * 得到鉴权
-     * @param appId
+     * @param app_id
      * @param clientSecret
-     * @param url
      * @returns
      */
     getAuthentication(app_id, clientSecret) {
@@ -126,8 +125,7 @@ class QQBotAPI {
     /**
      * 发送群聊消息
      * @param group_openid
-     * @param content
-     * @param msg_id
+     * @param data
      * @returns
      */
     async groupOpenMessages(group_openid, data) {
@@ -140,8 +138,7 @@ class QQBotAPI {
     /**
      * 发送私聊富媒体文件
      * @param openid
-     * @param content
-     * @param file_type
+     * @param data
      * @returns
      *  1 图文 2 视频 3 语言 4 文件
      * 图片：png/jpg，视频：mp4，语音：silk
@@ -156,8 +153,7 @@ class QQBotAPI {
     /**
      * 发送私聊富媒体文件
      * @param openid
-     * @param content
-     * @param file_type
+     * @param data
      * @returns
      *  1 图文 2 视频 3 语言 4 文件
      * 图片：png/jpg，视频：mp4，语音：silk
@@ -172,8 +168,7 @@ class QQBotAPI {
     /**
      * 发送群里文件
      * @param openid
-     * @param content
-     * @param file_type
+     * @param data
      * @returns
      *  1 图文 2 视频 3 语言 4 文件
      * 图片：png/jpg，视频：mp4，语音：silk

package/lib/sdk/{websoket.group.js → client.websoket.group.js}

@@ -93,7 +93,7 @@ class QQBotGroupClient extends QQBotAPI {
         await this.#setTimeoutBotConfig();
         // 请求url
         if (!this.#gatewayUrl) {
-            this.#gatewayUrl = gatewayURL ?? await this.gateway().then(res => res?.url);
+            this.#gatewayUrl = gatewayURL ?? (await this.gateway().then(res => res?.url));
         }
         if (!this.#gatewayUrl)
             return;

package/lib/sdk/{websoket.guild.js → client.websoket.guild.js}

@@ -70,15 +70,17 @@ class QQBotGuildClient extends QQBotAPI {
      */
     async connect(gatewayURL) {
         //
-        this.#gatewayUrl = gatewayURL ?? await this.gateway()
-            .then(res => res.url)
-            .catch(err => {
-            if (this.#events['ERROR']) {
-                for (const item of this.#events['ERROR']) {
-                    item(err);
-                }
-            }
-        });
+        this.#gatewayUrl =
+            gatewayURL ??
+                (await this.gateway()
+                    .then(res => res.url)
+                    .catch(err => {
+                    if (this.#events['ERROR']) {
+                        for (const item of this.#events['ERROR']) {
+                            item(err);
+                        }
+                    }
+                }));
         // 请求url
         if (!this.#gatewayUrl)
             return this;