@alemonjs/qq-bot 0.0.1

package/lib/client.js

@@ -0,0 +1,176 @@
+import { QQBotAPI } from './api.js';
+import bodyParser from 'koa-bodyparser';
+import Router from 'koa-router';
+import { WebhookAPI } from './webhook.js';
+import Koa from 'koa';
+import { config } from './config.js';
+import { v4 } from 'uuid';
+import { WebSocketServer, WebSocket } from 'ws';
+
+class QQBotClient extends QQBotAPI {
+    #events = {};
+    #app = null;
+    #count = 0;
+    #client = [];
+    #ws = null;
+    /**
+     * 设置配置
+     * @param opstion
+     */
+    constructor(opstion) {
+        super();
+        config.set('secret', opstion.secret);
+        config.set('appId', opstion.appId);
+        config.set('token', opstion.token);
+        config.set('port', opstion.port);
+        config.set('ws', opstion.ws);
+    }
+    /**
+     * 注册事件处理程序
+     * @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() {
+        this.#app = new Koa();
+        this.#app.use(bodyParser());
+        const router = new Router();
+        const port = config.get('port');
+        const cfg = {
+            secret: config.get('secret'),
+            port: port ? Number(port) : 17157
+        };
+        const ntqqWebhook = new WebhookAPI({
+            secret: 'YI2mWG0lWH2nYJ4qcOAwiUH4reRE1pdR'
+        });
+        this.#app.use(async (ctx, next) => {
+            let rawData = '';
+            ctx.req.on('data', chunk => (rawData += chunk));
+            ctx.req.on('end', () => (ctx.request.rawBody = rawData));
+            await next();
+        });
+        // 启动服务
+        router.post('/webhook', async (ctx) => {
+            const sign = ctx.req.headers['x-signature-ed25519'];
+            const timestamp = ctx.req.headers['x-signature-timestamp'];
+            const rawBody = ctx.request.rawBody;
+            const isValid = ntqqWebhook.validSign(timestamp, rawBody, String(sign));
+            if (!isValid) {
+                ctx.status = 400;
+                ctx.body = { msg: 'invalid signature' };
+                return;
+            }
+            const body = ctx.request.body;
+            if (body.op == 13) {
+                ctx.status = 200;
+                ctx.body = {
+                    // 返回明文 token
+                    plain_token: body.d.plain_token,
+                    // 生成签名
+                    signature: ntqqWebhook.getSign(body.d.event_ts, body.d.plain_token)
+                };
+            }
+            else if (body.op == 0) {
+                ctx.status = 204;
+                console.log('body', body.d);
+                // 根据事件类型，处理事件
+                for (const event of this.#events[body.t] || []) {
+                    event(body.d);
+                }
+                // 也可以分法到客户端。 发送失败需要处理 或清理调
+                for (const client of this.#client) {
+                    try {
+                        client.ws.send(JSON.stringify(body.d));
+                    }
+                    catch (error) {
+                        console.error('send error', error);
+                    }
+                }
+            }
+        });
+        this.#app.use(router.routes());
+        this.#app.use(router.allowedMethods());
+        // 启动服务
+        const server = this.#app.listen(cfg.port, () => {
+            console.log('Server running at http://localhost:' + cfg.port + '/webhook');
+        });
+        // 创建 WebSocketServer 并监听同一个端口
+        const wss = new WebSocketServer({ server: server });
+        // 处理客户端连接
+        wss.on('connection', ws => {
+            const clientId = v4();
+            ws['clientId'] = clientId;
+            console.log(clientId, 'connection');
+            this.#client.push({ id: clientId, ws });
+            // 处理消息事件
+            ws.on('message', (message) => {
+                // 拿到消息
+                try {
+                    const body = JSON.parse(message.toString());
+                    for (const event of this.#events[body.t] || []) {
+                        event(body);
+                    }
+                }
+                catch (e) {
+                    console.error('parse error', e);
+                }
+            });
+            // 处理关闭事件
+            ws.on('close', () => {
+                console.log(`Client ${clientId} disconnected`);
+                this.#client = this.#client.filter(client => client.id !== clientId);
+            });
+        });
+        const reconnect = () => {
+            const ws = config.get('ws');
+            if (!ws)
+                return;
+            // 使用了ws服务器
+            this.#ws = new WebSocket(ws);
+            this.#ws.on('open', () => {
+                this.#count = 0;
+                console.log('ws connected');
+            });
+            this.#ws.on('message', data => {
+                try {
+                    // 拿到消息
+                    const body = JSON.parse(data.toString());
+                    for (const event of this.#events[body.t] || []) {
+                        event(body);
+                    }
+                }
+                catch (e) {
+                    console.error('parse error', e);
+                }
+            });
+            this.#ws.on('close', () => {
+                console.log('ws closed');
+                // 重连5次，超过5次不再重连
+                if (this.#count > 5) {
+                    return;
+                }
+                // 23s 后重连
+                setTimeout(() => {
+                    reconnect();
+                }, 23000);
+            });
+            this.#ws.on('error', () => {
+                console.log('ws error ');
+            });
+        };
+        reconnect();
+    }
+}
+
+export { QQBotClient };

package/lib/config.js

@@ -0,0 +1,3 @@
+const config = new Map();
+
+export { config };

package/lib/from.js

@@ -0,0 +1,44 @@
+import { existsSync, createReadStream } from 'fs';
+import { Readable, isReadable } from 'stream';
+import { basename } from 'path';
+import { fileTypeFromBuffer, fileTypeFromStream } from 'file-type';
+
+/**
+ * 创建form
+ * @param image
+ * @param name
+ * @returns
+ */
+async function createPicFrom(image, name = 'image.jpg') {
+    let picData;
+    // 是 string
+    if (typeof image === 'string') {
+        if (!existsSync(image))
+            return false;
+        if (!name)
+            name = basename(image);
+        picData = createReadStream(image);
+        // 是 buffer
+    }
+    else if (Buffer.isBuffer(image)) {
+        const { ext } = await fileTypeFromBuffer(image);
+        if (!name)
+            name = 'file.' + ext;
+        picData = new Readable();
+        picData.push(image);
+        picData.push(null);
+        // 是 文件流
+    }
+    else if (isReadable(image)) {
+        const { ext } = await fileTypeFromStream(image);
+        if (!name)
+            name = 'file.' + ext;
+        picData = image;
+    }
+    else {
+        return false;
+    }
+    return { picData, image, name };
+}
+
+export { createPicFrom };

package/lib/index.js

@@ -0,0 +1,396 @@
+import { defineBot, getConfig, useUserHashKey, OnProcessor } from 'alemonjs';
+import { QQBotClient } from './client.js';
+
+const platform = 'qq-bot';
+var index = defineBot(() => {
+    const cfg = getConfig();
+    const config = cfg.value['qq-bot'];
+    if (!config)
+        return;
+    const client = new QQBotClient({
+        secret: config.secret,
+        appId: config.appId,
+        token: config.token,
+        port: config.port,
+        ws: config?.ws
+    });
+    // 连接
+    client.connect();
+    // 监听消息
+    client.on('GROUP_AT_MESSAGE_CREATE', async (event) => {
+        const master_key = config?.master_key ?? [];
+        const isMaster = master_key.includes(event.author.id);
+        const url = `https://q.qlogo.cn/qqapp/${config.app_id}/${event.author.id}/640`;
+        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 = {
+            // 事件类型
+            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
+        };
+        // 当访问的时候获取
+        Object.defineProperty(e, 'value', {
+            get() {
+                return event;
+            }
+        });
+        // 处理消息
+        OnProcessor(e, 'message.create');
+    });
+    client.on('C2C_MESSAGE_CREATE', async (event) => {
+        const master_key = config?.master_key ?? [];
+        const isMaster = master_key.includes(event.author.id);
+        const url = `https://q.qlogo.cn/qqapp/${config.app_id}/${event.author.id}/640`;
+        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 = {
+            // 事件类型
+            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: '',
+            //
+            tag: 'GROUP_AT_MESSAGE_CREATE',
+            value: null
+        };
+        // 当访问的时候获取
+        Object.defineProperty(e, 'value', {
+            get() {
+                return event;
+            }
+        });
+        // 处理消息
+        OnProcessor(e, 'private.message.create');
+    });
+    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 = {
+            // 事件类型
+            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,
+            OpenId: event.guild_id,
+            CreateAt: Date.now(),
+            //
+            tag: 'AT_MESSAGE_CREATE',
+            //
+            value: null
+        };
+        // 当访问的时候获取
+        Object.defineProperty(e, 'value', {
+            get() {
+                return event;
+            }
+        });
+        // 处理消息
+        OnProcessor(e, 'private.message.create');
+    });
+    // 监听消息
+    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 = {
+            // 事件类型
+            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,
+            OpenId: event.guild_id,
+            CreateAt: Date.now(),
+            //
+            tag: 'AT_MESSAGE_CREATE',
+            //
+            value: null
+        };
+        // 当访问的时候获取
+        Object.defineProperty(e, 'value', {
+            get() {
+                return event;
+            }
+        });
+        // 处理消息
+        OnProcessor(e, 'message.create');
+    });
+    /**
+     *
+     * @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 = {
+            // 事件类型
+            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,
+            OpenId: event.guild_id,
+            CreateAt: Date.now(),
+            //
+            tag: 'AT_MESSAGE_CREATE',
+            value: null
+        };
+        // 当访问的时候获取
+        Object.defineProperty(e, 'value', {
+            get() {
+                return event;
+            }
+        });
+        // 处理消息
+        OnProcessor(e, 'message.create');
+    });
+    // FRIEND_ADD
+    global.client = client;
+    return {
+        api: {
+            use: {
+                send: (event, val) => {
+                    if (val.length < 0)
+                        return Promise.all([]);
+                    const content = val
+                        .filter(item => item.type == 'Link' || item.type == 'Mention' || item.type == 'Text')
+                        .map(item => {
+                        if (item.type == 'Link') {
+                            return `[${item.options?.title ?? item.value}](${item.value})`;
+                        }
+                        else if (item.type == 'Mention') {
+                            if (item.value == 'everyone' ||
+                                item.value == 'all' ||
+                                item.value == '' ||
+                                typeof item.value != 'string') {
+                                return ``;
+                            }
+                            if (item.options?.belong == 'user') {
+                                return `<@${item.value}>`;
+                            }
+                            return '';
+                            // return `<qqbot-at-everyone />`
+                        }
+                        else if (item.type == 'Text') {
+                            return item.value;
+                        }
+                    })
+                        .join('');
+                    if (content) {
+                        return Promise.all([content].map(item => client.groupOpenMessages(event.GuildId, {
+                            content: item,
+                            msg_id: event.MessageId,
+                            msg_type: 0,
+                            msg_seq: client.getMessageSeq(event.MessageId)
+                        })));
+                    }
+                    const images = val.filter(item => item.type == 'Image').map(item => item.value);
+                    if (images) {
+                        return Promise.all(images.map(async (msg) => {
+                            const file_info = await client
+                                .postRichMediaByGroup(event.GuildId, {
+                                file_type: 1,
+                                file_data: msg.toString('base64')
+                            })
+                                .then(res => res?.file_info);
+                            if (!file_info)
+                                return Promise.resolve(null);
+                            return client.groupOpenMessages(event.GuildId, {
+                                content: '',
+                                media: {
+                                    file_info
+                                },
+                                msg_id: event.MessageId,
+                                msg_type: 7,
+                                msg_seq: client.getMessageSeq(event.MessageId)
+                            });
+                        }));
+                    }
+                    return Promise.all([]);
+                },
+                mention: async () => {
+                    // const event = e.value
+                    const Metions = [];
+                    return Metions;
+                }
+            }
+        }
+    };
+});
+
+export { index as default, platform };

package/lib/webhook.js

@@ -0,0 +1,53 @@
+import { ed25519 } from '@noble/curves/ed25519';
+
+class WebhookAPI {
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     *  验证签名
+     * @param ts
+     * @param body
+     * @param sign
+     * @returns
+     */
+    validSign(ts, body, sign) {
+        const { publicKey } = this.getKey();
+        const sig = Buffer.isBuffer(sign) ? sign : Buffer.from(sign, 'hex');
+        const httpBody = Buffer.from(body);
+        const msg = Buffer.from(ts + httpBody);
+        return ed25519.verify(sig, msg, publicKey);
+    }
+    /**
+     *   生成签名
+     * @param eventTs
+     * @param plainToken
+     * @returns
+     */
+    getSign(eventTs, plainToken) {
+        const { privateKey } = this.getKey();
+        const msg = Buffer.from(eventTs + plainToken);
+        const signature = Buffer.from(ed25519.sign(msg, privateKey)).toString('hex');
+        return signature;
+    }
+    /**
+     * 获取 key
+     * @returns
+     */
+    getKey() {
+        let seed = this.config.secret;
+        if (!seed)
+            throw new Error("secret not set, can't calc ed25519 key");
+        while (seed.length < 32)
+            seed = seed.repeat(2); // Ed25519 的种子大小是 32 字节
+        seed = seed.slice(0, 32); // 修剪到 32 字节
+        const privateKey = Buffer.from(seed);
+        return {
+            privateKey,
+            publicKey: ed25519.getPublicKey(privateKey)
+        };
+    }
+}
+
+export { WebhookAPI };