@alemonjs/qq-bot 0.0.12 → 0.0.14

package/lib/sdk/websoket.group.js

@@ -0,0 +1,221 @@
+import WebSocket from 'ws';
+import { QQBotAPI } from './api.js';
+import { config } from './config.js';
+import { getIntentsMask } from './intents.js';
+import { Counter } from './counter.js';
+
+/**
+ * 连接
+ */
+class QQBotGroupClient extends QQBotAPI {
+    //
+    #counter = new Counter(1); // 初始值为1
+    // 标记是否已连接
+    #isConnected = false;
+    // 存储最新的消息序号
+    #heartbeat_interval = 30000;
+    // 鉴权
+    #IntervalId = null;
+    // url
+    #gatewayUrl = null;
+    /**
+     * 设置配置
+     * @param opstion
+     */
+    constructor(opstion) {
+        super();
+        for (const key in opstion) {
+            config.set(key, opstion[key]);
+        }
+    }
+    /**
+     * 定时鉴权
+     * @param cfg
+     * @returns
+     */
+    async #setTimeoutBotConfig() {
+        const accessToken = async () => {
+            const app_id = config.get('app_id');
+            const secret = config.get('secret');
+            if (!app_id || !secret)
+                return;
+            // 发送请求
+            const data = await this.getAuthentication(app_id, secret).then(res => res.data);
+            config.set('access_token', data.access_token);
+            console.info('refresh', data.expires_in, 's');
+            setTimeout(accessToken, data.expires_in * 1000);
+        };
+        await accessToken();
+        return;
+    }
+    /**
+     * 鉴权数据
+     * @returns
+     */
+    #aut() {
+        const token = config.get('token');
+        const intents = config.get('intents');
+        const shard = config.get('shard');
+        return {
+            op: 2, // op = 2
+            d: {
+                token: `QQBot ${token}`,
+                intents: getIntentsMask(intents),
+                shard: shard,
+                properties: {
+                    $os: process.platform,
+                    $browser: 'alemonjs',
+                    $device: 'alemonjs'
+                }
+            }
+        };
+    }
+    #ws = null;
+    #events = {};
+    /**
+     * 注册事件处理程序
+     * @param key 事件名称
+     * @param val 事件处理函数
+     */
+    on(key, val) {
+        if (!this.#events[key])
+            this.#events[key] = [];
+        this.#events[key].push(val);
+        return this;
+    }
+    /**
+     *
+     * @param cfg
+     * @param conversation
+     */
+    async connect() {
+        // 定时模式
+        await this.#setTimeoutBotConfig();
+        // 请求url
+        if (!this.#gatewayUrl) {
+            this.#gatewayUrl = await this.gateway().then(res => res?.url);
+        }
+        if (!this.#gatewayUrl)
+            return;
+        // 重新连接的逻辑
+        const reconnect = async () => {
+            if (this.#counter.get() >= 5) {
+                console.info('The maximum number of reconnections has been reached, cancel reconnection');
+                return;
+            }
+            setTimeout(() => {
+                console.info('[ws] reconnecting...');
+                // 重新starrt
+                start();
+                // 记录
+                this.#counter.getNextId();
+            }, 5000);
+        };
+        const start = () => {
+            if (this.#gatewayUrl) {
+                const map = {
+                    0: async ({ t, d }) => {
+                        if (this.#events[t]) {
+                            try {
+                                for (const event of this.#events[t]) {
+                                    // 是否是函数
+                                    if (typeof event != 'function')
+                                        continue;
+                                    event(d);
+                                }
+                            }
+                            catch (err) {
+                                if (this.#events['ERROR']) {
+                                    for (const event of this.#events['ERROR']) {
+                                        // 是否是函数
+                                        if (typeof event != 'function')
+                                            continue;
+                                        event(err);
+                                    }
+                                }
+                            }
+                        }
+                        // Ready Event，鉴权成功
+                        if (t === 'READY') {
+                            this.#IntervalId = setInterval(() => {
+                                if (this.#isConnected) {
+                                    this.#ws &&
+                                        this.#ws.send(JSON.stringify({
+                                            op: 1, //  op = 1
+                                            d: null // 如果是第一次连接，传null
+                                        }));
+                                }
+                            }, this.#heartbeat_interval);
+                        }
+                        // Resumed Event，恢复连接成功
+                        if (t === 'RESUMED') {
+                            console.info('[ws] restore connection');
+                            // 重制次数
+                            this.#counter.reStart();
+                        }
+                        return;
+                    },
+                    6: ({ d }) => {
+                        console.info('[ws] connection attempt', d);
+                        return;
+                    },
+                    7: async ({ d }) => {
+                        // 执行重新连接
+                        console.info('[ws] reconnect', d);
+                        // 取消鉴权发送
+                        if (this.#IntervalId)
+                            clearInterval(this.#IntervalId);
+                        return;
+                    },
+                    9: ({ d }) => {
+                        console.info('[ws] parameter error', d);
+                        return;
+                    },
+                    10: ({ d }) => {
+                        // 重制次数
+                        this.#isConnected = true;
+                        // 记录新循环
+                        this.#heartbeat_interval = d.heartbeat_interval;
+                        // 发送鉴权
+                        this.#ws && this.#ws.send(JSON.stringify(this.#aut()));
+                        return;
+                    },
+                    11: () => {
+                        // OpCode 11 Heartbeat ACK 消息，心跳发送成功
+                        console.info('[ws] heartbeat transmission');
+                        // 重制次数
+                        this.#counter.reStart();
+                        return;
+                    },
+                    12: ({ d }) => {
+                        console.info('[ws] platform data', d);
+                        return;
+                    }
+                };
+                // 连接
+                this.#ws = new WebSocket(this.#gatewayUrl);
+                this.#ws.on('open', () => {
+                    console.info('[ws] open');
+                });
+                // 监听消息
+                this.#ws.on('message', async (msg) => {
+                    const message = JSON.parse(msg.toString('utf8'));
+                    if (process.env.NTQQ_WS == 'dev')
+                        console.info('message', message);
+                    // 根据 opcode 进行处理
+                    if (map[message.op]) {
+                        map[message.op](message);
+                    }
+                });
+                // 关闭
+                this.#ws.on('close', async (err) => {
+                    await reconnect();
+                    console.info('[ws] close', err);
+                });
+            }
+        };
+        start();
+    }
+}
+
+export { QQBotGroupClient };

package/lib/sdk/websoket.guild.js

@@ -0,0 +1,203 @@
+import WebSocket from 'ws';
+import { config } from './config.js';
+import { getIntentsMask } from './intents.js';
+import { QQBotAPI } from './api.js';
+import { Counter } from './counter.js';
+
+/**
+ * 连接
+ */
+class QQBotGuildClient extends QQBotAPI {
+    //
+    #counter = new Counter(1); // 初始值为1
+    // 标记是否已连接
+    #isConnected = false;
+    // 存储最新的消息序号
+    #heartbeat_interval = 30000;
+    // 鉴权
+    #IntervalId = null;
+    // url
+    #gatewayUrl = null;
+    #ws;
+    /**
+     *
+     * @param opstion
+     */
+    constructor(opstion) {
+        super();
+        for (const key in opstion) {
+            config.set(key, opstion[key]);
+        }
+    }
+    /**
+     * 鉴权数据
+     * @returns
+     */
+    #aut() {
+        const app_id = config.get('app_id');
+        const token = config.get('token');
+        const intents = config.get('intents');
+        const shard = config.get('shard');
+        return {
+            op: 2, // op = 2
+            d: {
+                token: `Bot ${app_id}.${token}`,
+                intents: getIntentsMask(intents),
+                shard,
+                properties: {
+                    $os: process.platform,
+                    $browser: 'alemonjs',
+                    $device: 'alemonjs'
+                }
+            }
+        };
+    }
+    #events = {};
+    /**
+     * 注册事件处理程序
+     * @param key 事件名称
+     * @param val 事件处理函数
+     */
+    on(key, val) {
+        if (!this.#events[key])
+            this.#events[key] = [];
+        this.#events[key].push(val);
+        return this;
+    }
+    /**
+     *
+     * @param cfg
+     */
+    async connect() {
+        //
+        this.#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;
+        // 重新连接的逻辑
+        const reconnect = async () => {
+            if (this.#counter.get() >= 5) {
+                console.info('The maximum number of reconnections has been reached, cancel reconnection');
+                return;
+            }
+            setTimeout(() => {
+                console.info('[ws] reconnecting...');
+                // 重新starrt
+                start();
+                // 记录
+                this.#counter.getNextId();
+            }, 5000);
+        };
+        const start = () => {
+            if (this.#gatewayUrl) {
+                const map = {
+                    0: async ({ t, d }) => {
+                        // 执行
+                        if (this.#events[t]) {
+                            try {
+                                for (const event of this.#events[t]) {
+                                    // 是否是函数
+                                    if (typeof event != 'function')
+                                        continue;
+                                    event(d);
+                                }
+                            }
+                            catch (err) {
+                                if (this.#events['ERROR']) {
+                                    for (const item of this.#events['ERROR']) {
+                                        item(err);
+                                    }
+                                }
+                            }
+                        }
+                        // Ready Event，鉴权成功
+                        if (t === 'READY') {
+                            this.#IntervalId = setInterval(() => {
+                                if (this.#isConnected) {
+                                    this.#ws.send(JSON.stringify({
+                                        op: 1, //  op = 1
+                                        d: null // 如果是第一次连接，传null
+                                    }));
+                                }
+                            }, this.#heartbeat_interval);
+                        }
+                        // Resumed Event，恢复连接成功
+                        if (t === 'RESUMED') {
+                            console.info('[ws] restore connection');
+                            // 重制次数
+                            this.#counter.reStart();
+                        }
+                        return;
+                    },
+                    6: ({ d }) => {
+                        console.info('[ws] connection attempt', d);
+                        return;
+                    },
+                    7: async ({ d }) => {
+                        // 执行重新连接
+                        console.info('[ws] reconnect', d);
+                        // 取消鉴权发送
+                        if (this.#IntervalId)
+                            clearInterval(this.#IntervalId);
+                        return;
+                    },
+                    9: ({ d }) => {
+                        console.info('[ws] parameter error', d);
+                        return;
+                    },
+                    10: ({ d }) => {
+                        // 重制次数
+                        this.#isConnected = true;
+                        // 记录新循环
+                        this.#heartbeat_interval = d.heartbeat_interval;
+                        // 发送鉴权
+                        this.#ws.send(JSON.stringify(this.#aut()));
+                        return;
+                    },
+                    11: () => {
+                        // OpCode 11 Heartbeat ACK 消息，心跳发送成功
+                        console.info('[ws] heartbeat transmission');
+                        // 重制次数
+                        this.#counter.reStart();
+                        return;
+                    },
+                    12: ({ d }) => {
+                        console.info('[ws] platform data', d);
+                        return;
+                    }
+                };
+                // 连接
+                this.#ws = new WebSocket(this.#gatewayUrl);
+                this.#ws.on('open', () => {
+                    console.info('[ws] open');
+                });
+                // 监听消息
+                this.#ws.on('message', async (msg) => {
+                    const message = JSON.parse(msg.toString('utf8'));
+                    if (process.env.QQ_WS == 'dev')
+                        console.info('message', message);
+                    if (map[message.op]) {
+                        map[message.op](message);
+                    }
+                });
+                // 关闭
+                this.#ws.on('close', async (err) => {
+                    await reconnect();
+                    console.info('[ws] close', err);
+                });
+            }
+        };
+        start();
+        return this;
+    }
+}
+
+export { QQBotGuildClient };

package/lib/send/index.js

@@ -1,3 +1,6 @@
+import { readFileSync } from 'fs';
+import axios from 'axios';
+
 const GROUP_AT_MESSAGE_CREATE = (client, event, val) => {
     const content = val
         .filter(item => item.type == 'Link' || item.type == 'Mention' || item.type == 'Text')
@@ -30,13 +33,27 @@ const GROUP_AT_MESSAGE_CREATE = (client, event, val) => {
             msg_seq: client.getMessageSeq(event.MessageId)
         })));
     }
-    const images = val.filter(item => item.type == 'Image').map(item => item.value);
+    const images = val.filter(item => item.type == 'Image' || item.type == 'ImageFile' || item.type == 'ImageURL');
     if (images) {
-        return Promise.all(images.map(async (msg) => {
+        return Promise.all(images.map(async (item) => {
+            if (item.type == 'ImageURL') {
+                return client.groupOpenMessages(event.GuildId, {
+                    content: '',
+                    media: {
+                        file_info: item.value
+                    },
+                    msg_id: event.MessageId,
+                    msg_type: 7,
+                    msg_seq: client.getMessageSeq(event.MessageId)
+                });
+            }
+            const file_data = item.type == 'ImageFile'
+                ? readFileSync(item.value, 'base64')
+                : item.value.toString('base64');
             const file_info = await client
                 .postRichMediaByGroup(event.GuildId, {
                 file_type: 1,
-                file_data: msg.toString('base64')
+                file_data: file_data
             })
                 .then(res => res?.file_info);
             if (!file_info)
@@ -72,13 +89,27 @@ const C2C_MESSAGE_CREATE = (client, event, val) => {
             msg_seq: client.getMessageSeq(event.MessageId)
         })));
     }
-    const images = val.filter(item => item.type == 'Image').map(item => item.value);
+    const images = val.filter(item => item.type == 'Image' || item.type == 'ImageFile' || item.type == 'ImageURL');
     if (images) {
-        return Promise.all(images.map(async (msg) => {
+        return Promise.all(images.map(async (item) => {
+            if (item.type == 'ImageURL') {
+                return client.usersOpenMessages(event.OpenId, {
+                    content: '',
+                    media: {
+                        file_info: item.value
+                    },
+                    msg_id: event.MessageId,
+                    msg_type: 7,
+                    msg_seq: client.getMessageSeq(event.MessageId)
+                });
+            }
+            const file_data = item.type == 'ImageFile'
+                ? readFileSync(item.value, 'base64')
+                : item.value.toString('base64');
             const file_info = await client
                 .postRichMediaByUsers(event.OpenId, {
                 file_type: 1,
-                file_data: msg.toString('base64')
+                file_data: file_data
             })
                 .then(res => res?.file_info);
             if (!file_info)
@@ -119,12 +150,27 @@ const DIRECT_MESSAGE_CREATE = (client, event, val) => {
             msg_id: event.MessageId
         })));
     }
-    const images = val.filter(item => item.type == 'Image').map(item => item.value);
+    const images = val.filter(item => item.type == 'Image' || item.type == 'ImageFile' || item.type == 'ImageURL');
     if (images) {
-        return Promise.all(images.map(item => client.postDirectImage(event.OpenId, {
-            msg_id: event.MessageId,
-            image: item
-        })));
+        return Promise.all(images.map(async (item) => {
+            if (item.value == 'ImageURL') {
+                // 请求得到buffer
+                const data = await axios
+                    .get(item.value, {
+                    responseType: 'arraybuffer'
+                })
+                    .then(res => res.data);
+                return client.postDirectImage(event.OpenId, {
+                    msg_id: event.MessageId,
+                    image: data
+                });
+            }
+            const file_data = item.type == 'ImageFile' ? readFileSync(item.value) : item.value;
+            return client.postDirectImage(event.OpenId, {
+                msg_id: event.MessageId,
+                image: Buffer.isBuffer(file_data) ? file_data : Buffer.from(file_data)
+            });
+        }));
     }
     return [];
 };
@@ -161,12 +207,27 @@ const AT_MESSAGE_CREATE = (client, event, val) => {
             msg_id: event.MessageId
         })));
     }
-    const images = val.filter(item => item.type == 'Image').map(item => item.value);
+    const images = val.filter(item => item.type == 'Image' || item.type == 'ImageFile' || item.type == 'ImageURL');
     if (images) {
-        return Promise.all(images.map(item => client.postImage(event.ChannelId, {
-            msg_id: event.MessageId,
-            image: item
-        })));
+        return Promise.all(images.map(async (item) => {
+            if (item.value == 'ImageURL') {
+                // 请求得到buffer
+                const data = await axios
+                    .get(item.value, {
+                    responseType: 'arraybuffer'
+                })
+                    .then(res => res.data);
+                return client.postImage(event.ChannelId, {
+                    msg_id: event.MessageId,
+                    image: data
+                });
+            }
+            const file_data = item.type == 'ImageFile' ? readFileSync(item.value) : item.value;
+            return client.postImage(event.ChannelId, {
+                msg_id: event.MessageId,
+                image: Buffer.isBuffer(file_data) ? file_data : Buffer.from(file_data)
+            });
+        }));
     }
     return [];
 };
@@ -209,12 +270,27 @@ const MESSAGE_CREATE = (client, event, val) => {
             msg_id: event.MessageId
         })));
     }
-    const images = val.filter(item => item.type == 'Image').map(item => item.value);
+    const images = val.filter(item => item.type == 'Image' || item.type == 'ImageFile' || item.type == 'ImageURL');
     if (images) {
-        return Promise.all(images.map(item => client.postImage(event.ChannelId, {
-            msg_id: event.MessageId,
-            image: item
-        })));
+        return Promise.all(images.map(async (item) => {
+            if (item.value == 'ImageURL') {
+                // 请求得到buffer
+                const data = await axios
+                    .get(item.value, {
+                    responseType: 'arraybuffer'
+                })
+                    .then(res => res.data);
+                return client.postImage(event.ChannelId, {
+                    msg_id: event.MessageId,
+                    image: data
+                });
+            }
+            const file_data = item.type == 'ImageFile' ? readFileSync(item.value) : item.value;
+            return client.postImage(event.ChannelId, {
+                msg_id: event.MessageId,
+                image: Buffer.isBuffer(file_data) ? file_data : Buffer.from(file_data)
+            });
+        }));
     }
     return [];
 };