@alemonjs/qq-bot 0.0.19 → 0.0.20

package/lib/sdk/client.websoket.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 QQBotClients 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('access_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(gatewayURL) {
+        // 定时模式
+        await this.#setTimeoutBotConfig();
+        // 请求url
+        if (!this.#gatewayUrl) {
+            this.#gatewayUrl = 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 { QQBotClients };

package/lib/sdk/intents.js

@@ -71,29 +71,18 @@ PUBLIC_GUILD_MESSAGES (1 << 30) // 消息事件，此为公域的消息事件
  * 订阅事件集合
  */
 const intentsMap = {
-    GUILDS: 1 << 0,
-    GUILD_MEMBERS: 1 << 1,
-    GUILD_MESSAGES: 1 << 9,
-    GUILD_MESSAGE_REACTIONS: 1 << 10,
-    DIRECT_MESSAGE: 1 << 12,
-    INTERACTION: 1 << 26,
-    MESSAGE_AUDIT: 1 << 27,
-    FORUMS_EVENT: 1 << 28,
-    AUDIO_ACTION: 1 << 29,
-    PUBLIC_GUILD_MESSAGES: 1 << 30,
-    INTERACTION_CREATE: 1 << 26,
-    // group
-    GROUP_AND_C2C_EVENT: 1 << 25,
-    C2C_MESSAGE_CREATE: 1 << 25,
-    FRIEND_ADD: 1 << 25,
-    FRIEND_DEL: 1 << 25,
-    C2C_MSG_REJECT: 1 << 25,
-    C2C_MSG_RECEIVE: 1 << 25,
-    GROUP_AT_MESSAGE_CREATE: 1 << 25,
-    GROUP_ADD_ROBOT: 1 << 25,
-    GROUP_DEL_ROBOT: 1 << 25,
-    GROUP_MSG_REJECT: 1 << 25,
-    GROUP_MSG_RECEIVE: 1 << 25
+    GUILDS: 1 << 0, // 频道事件
+    GUILD_MEMBERS: 1 << 1, // 成员
+    GUILD_MESSAGES: 1 << 9, // 消息事件，仅 *私域* 
+    GUILD_MESSAGE_REACTIONS: 1 << 10, // 消息表情表态
+    DIRECT_MESSAGE: 1 << 12, // 私信消息
+    INTERACTION: 1 << 26, // 互动事件
+    // INTERACTION_CREATE: 1 << 26,
+    MESSAGE_AUDIT: 1 << 27, // 消息审核
+    FORUMS_EVENT: 1 << 28, // 论坛事件，仅 *私域*
+    AUDIO_ACTION: 1 << 29, // 音频 
+    PUBLIC_GUILD_MESSAGES: 1 << 30, //  消息事件，此为公域的消息事件
+    GROUP_AND_C2C_EVENT: 1 << 25, // group all
 };
 /**
  *

package/package.json

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