@alemonjs/onebot 0.2.1 → 0.2.3

package/lib/sdk/wss.js

@@ -0,0 +1,292 @@
+import WebSocket, { WebSocketServer } from 'ws';
+import { randomUUID } from 'crypto';
+
+/**
+ * 连接
+ */
+class OneBotClient {
+    #options = {
+        url: '',
+        access_token: '',
+        reverse_enable: false,
+        reverse_port: 17158
+    };
+    /**
+     * 设置配置
+     * @param opstion
+     */
+    constructor(opstion) {
+        for (const key in opstion) {
+            if (Object.prototype.hasOwnProperty.call(opstion, key)) {
+                this.#options[key] = opstion[key];
+            }
+        }
+    }
+    #ws = null;
+    #events = {};
+    #echo = {};
+    timeout = 30000;
+    /**
+     * 注册事件处理程序
+     * @param key 事件名称
+     * @param val 事件处理函数
+     */
+    on(key, val) {
+        this.#events[key] = val;
+        return this;
+    }
+    /**
+     *
+     * @param cfg
+     * @param conversation
+     */
+    async connect() {
+        const { url, access_token: token, reverse_enable, reverse_port } = this.#options;
+        const c = token == '' || !token
+            ? {}
+            : {
+                headers: {
+                    ['Authorization']: `Bearer ${token}`
+                }
+            };
+        const onMessage = async (data) => {
+            try {
+                const event = JSON.parse(data.toString());
+                if (!event) {
+                    if (this.#events['ERROR'])
+                        this.#events['ERROR'](event);
+                    return;
+                }
+                else if (event?.post_type == 'meta_event') {
+                    if (this.#events['META'])
+                        this.#events['META'](event);
+                    return;
+                }
+                else if (event?.post_type == 'message') {
+                    if (event?.message_type == 'group') {
+                        if (this.#events['MESSAGES'])
+                            this.#events['MESSAGES'](event);
+                    }
+                    else if (event?.message_type == 'private') {
+                        if (this.#events['DIRECT_MESSAGE'])
+                            this.#events['DIRECT_MESSAGE'](event);
+                    }
+                    else {
+                        // console.info('未知消息类型', event)
+                    }
+                    return;
+                }
+                else if (event?.post_type == 'notice') {
+                    if (event?.notice_type == 'group_increase') {
+                        // 群成员增加
+                        if (this.#events['NOTICE_GROUP_MEMBER_INCREASE'])
+                            this.#events['NOTICE_GROUP_MEMBER_INCREASE'](event);
+                    }
+                    else if (event?.notice_type == 'group_decrease') {
+                        // 群成员减少
+                        if (this.#events['NOTICE_GROUP_MEMBER_REDUCE'])
+                            this.#events['NOTICE_GROUP_MEMBER_REDUCE'](event);
+                    }
+                    return;
+                }
+                else if (event?.post_type == 'request') {
+                    // 收到加群 或 加好友的请求。
+                    if (event?.request_type == 'friend') {
+                        if (this.#events['REQUEST_ADD_FRIEND'])
+                            this.#events['REQUEST_ADD_FRIEND'](event);
+                    }
+                    else if (event?.request_type == 'group') {
+                        if (this.#events['REQUEST_ADD_GROUP'])
+                            this.#events['REQUEST_ADD_GROUP'](event);
+                    }
+                    return;
+                }
+                else if (event?.echo === 'get_friend_list' || event?.echo === 'get_group_list' || event?.echo === 'get_group_member_list') {
+                    // 处理获取好友列表和群列表的响应
+                    // if (this.#events['META']) this.#events['META'](event)
+                    // console.debug('响应', event)
+                    return;
+                }
+                if (!event?.post_type && event?.echo && this.#echo[event?.echo]) {
+                    if (![0, 1].includes(event?.retcode))
+                        this.#echo[event?.echo].reject(Object.assign(this.#echo[event?.echo].request, { error: event }));
+                    else
+                        this.#echo[event?.echo].resolve(event?.data
+                            ? new Proxy(event, {
+                                get: (target, prop) => target.event[prop] ?? target[prop]
+                            })
+                            : event);
+                    clearTimeout(this.#echo[event?.echo].timeout);
+                    delete this.#echo[event?.echo];
+                }
+            }
+            catch (err) {
+                if (this.#events['ERROR'])
+                    this.#events['ERROR'](err);
+            }
+        };
+        const onClose = (code, reason) => {
+            if (this.#events['ERROR'])
+                this.#events['ERROR']({
+                    de: code,
+                    reason: reason.toString('utf8')
+                });
+        };
+        if (!this.#ws) {
+            if (reverse_enable) {
+                // reverse_open
+                const server = new WebSocketServer({ port: reverse_port ?? 17158 });
+                server.on('connection', ws => {
+                    this.#ws = ws;
+                    // message
+                    this.#ws.on('message', onMessage);
+                    // close
+                    this.#ws.on('close', onClose);
+                    console.info('connected: ws://127.0.0.1:' + reverse_port);
+                });
+            }
+            else {
+                // forward_open
+                this.#ws = new WebSocket(url, c);
+                this.#ws.on('open', () => {
+                    console.debug(`open:${url}`);
+                });
+                // message
+                this.#ws.on('message', onMessage);
+                // close
+                this.#ws.on('close', onClose);
+            }
+        }
+    }
+    /**
+     * 发送私聊消息
+     * @param options
+     * @returns
+     */
+    sendPrivateMessage(options) {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'send_private_msg',
+            params: options,
+            echo: randomUUID()
+        }));
+    }
+    /**
+     * 发送群消息
+     * @param options
+     * @returns
+     */
+    sendGroupMessage(options) {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'send_group_msg',
+            params: options,
+            echo: randomUUID()
+        }));
+    }
+    /**
+     * 发送消息
+     * @param options
+     * @returns
+     */
+    sendMessage(options) {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'send_msg',
+            params: options,
+            echo: randomUUID()
+        }));
+    }
+    /**
+     * 好友列表
+     */
+    getFriendList() {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'get_friend_list',
+            params: {},
+            echo: 'get_friend_list'
+        }));
+    }
+    /**
+     * 群列表
+     */
+    getGroupList() {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'get_group_list',
+            params: {},
+            echo: 'get_group_list'
+        }));
+    }
+    /**
+     * 群成员列表
+     * @param options
+     * @returns
+     */
+    getGroupMemberList(options) {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'get_group_member_list',
+            params: options,
+            echo: 'get_group_member_list'
+        }));
+    }
+    /**
+     * 处理好友请求
+     * @param options
+     * @returns
+     */
+    setFriendAddRequest(options) {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'set_friend_add_request',
+            params: options,
+            echo: randomUUID()
+        }));
+    }
+    /**
+     * 处理群请求
+     * @param options
+     * @returns
+     */
+    setGroupAddRequest(options) {
+        if (!this.#ws)
+            return;
+        return this.#ws.send(JSON.stringify({
+            action: 'set_group_add_request',
+            params: options,
+            echo: randomUUID()
+        }));
+    }
+    /**
+     * @param options
+     * @returns
+     */
+    sendApi(options) {
+        if (!this.#ws)
+            return;
+        if (!options.echo)
+            options.echo = randomUUID();
+        this.#ws.send(JSON.stringify(options));
+        return new Promise((resolve, reject) => (this.#echo[options.echo] = {
+            request: options,
+            resolve,
+            reject,
+            timeout: setTimeout(() => {
+                reject(Object.assign(options, { timeout: this.timeout }));
+                delete this.#echo[options.echo];
+                console.error('请求超时:', options);
+            }, this.timeout)
+        }));
+    }
+}
+
+export { OneBotClient };

package/lib/server/index.js

@@ -0,0 +1,41 @@
+import Koa from 'koa';
+import Router from 'koa-router';
+
+// 使用koa + router 写 hello world
+const createServer = (client) => {
+    const app = new Koa();
+    const router = new Router();
+    const port = 8080; // 端口号
+    // 检测
+    router.get('/', (ctx) => {
+        ctx.body = 'Hello, World!';
+    });
+    // 获取群列表
+    router.get('/get_group_list', async (ctx) => {
+        client.getGroupList();
+        ctx.body = 'Hello, World!';
+    });
+    // 获取好友列表
+    router.get('/get_friend_list', async (ctx) => {
+        client.getFriendList();
+        ctx.body = 'Hello, World!';
+    });
+    // 获取群成员列表
+    router.get('/get_group_member_list', async (ctx) => {
+        const group_id = ctx.query.group_id;
+        if (!group_id) {
+            ctx.status = 400;
+            ctx.body = 'group_id is required';
+            return;
+        }
+        client.getGroupMemberList({ group_id: Number(group_id) });
+        ctx.body = 'Hello, World!';
+    });
+    app.use(router.routes());
+    app.use(router.allowedMethods());
+    app.listen(port, () => {
+        console.log(`Server is running at http://localhost:${port}`);
+    });
+};
+
+export { createServer };

package/package.json

@@ -1,24 +1,38 @@
 {
   "name": "@alemonjs/onebot",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "onebot",
   "author": "lemonade",
   "license": "MIT",
   "type": "module",
   "main": "lib/index.js",
   "types": "lib",
+  "scripts": {
+    "build": "node bundle.js"
+  },
   "exports": {
     ".": {
       "import": "./lib/index.js",
       "types": "./lib/index.d.ts"
-    }
+    },
+    "./package": "./package.json"
   },
   "keywords": [
-    "alemonjs",
-    "onebot",
-    "bot",
-    "chat-bot"
+    "alemonjs"
   ],
+  "dependencies": {
+    "ws": "^8.18.1"
+  },
+  "alemonjs": {
+    "desktop": {
+      "platform": [
+        {
+          "name": "onebot"
+        }
+      ],
+      "logo": "antd.RobotOutlined"
+    }
+  },
   "publishConfig": {
     "registry": "https://registry.npmjs.org",
     "access": "public"

package/lib/index-11.js

@@ -1,160 +0,0 @@
-import { defineBot, getConfig, useUserHashKey, OnProcessor } from 'alemonjs';
-import { OneBotClient } from './sdk-v11/wss.js';
-
-const platform = 'onebot';
-var INDEXV11 = defineBot(() => {
-    const cfg = getConfig();
-    const config = cfg.value?.onebot;
-    if (!config)
-        return;
-    //
-    const client = new OneBotClient({
-        // url
-        url: config?.url ?? '',
-        // token
-        access_token: config?.token ?? ''
-    });
-    //
-    client.connect();
-    //
-    client.on('META', event => {
-        if (event?.self_id) {
-            String(event.self_id);
-        }
-    });
-    //
-    client.on('MESSAGES', event => {
-        const uis = config.master_uids ?? [];
-        let msg = '';
-        const arr = [];
-        // let at_users = []
-        for (const item of event.message) {
-            if (item.type == 'text') {
-                msg = item.data.text;
-            }
-        }
-        for (const item of arr) {
-            msg = msg.replace(item.text, '').trim();
-        }
-        const url = `https://q1.qlogo.cn/g?b=qq&s=0&nk=${event.user_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 = String(event.user_id);
-        const UserKey = useUserHashKey({
-            Platform: platform,
-            UserId
-        });
-        // 定义消
-        const e = {
-            // 平台类型
-            Platform: platform,
-            // 频道
-            GuildId: String(event.group_id),
-            // guild_name: event.group_name,
-            // 子频道
-            ChannelId: String(event.group_id),
-            IsMaster: uis.includes(String(event.user_id)),
-            IsBot: false,
-            UserId: UserId,
-            UserName: event.sender.nickname,
-            UserKey,
-            UserAvatar: UserAvatar,
-            // message
-            MessageId: String(event.message_id),
-            MessageText: msg,
-            // 用户openId
-            OpenId: String(event.user_id),
-            // 创建时间
-            CreateAt: Date.now(),
-            // 标签
-            tag: 'MESSAGES',
-            //
-            value: null
-        };
-        // 当访问的时候获取
-        Object.defineProperty(e, 'value', {
-            get() {
-                return event;
-            }
-        });
-        // 处理消息
-        OnProcessor(e, 'message.create');
-    });
-    client.on('DIRECT_MESSAGE', () => {
-        // const e = {
-        //   isMaster: event.user_id == masterId,
-        //   msg_txt: event.raw_message,
-        //   msg: event.raw_message.trim(),
-        //   msg_id: event.message_id,
-        //   open_id: event.user_id,
-        //   user_id: event.user_id,
-        //   user_avatar:
-        //     event.platform == 'qq'
-        //       ? `https://q1.qlogo.cn/g?b=qq&s=0&nk=${event.user_id}`
-        //       : '',
-        //   user_name: event.sender.nickname,
-    });
-    // 错误处理
-    client.on('ERROR', event => {
-        console.error('ERROR', event);
-    });
-    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 => item.value)
-                        .join('');
-                    if (content) {
-                        return Promise.all([content].map(item => client.sendGroupMessage({
-                            group_id: event.ChannelId,
-                            message: [
-                                {
-                                    type: 'text',
-                                    data: {
-                                        text: item
-                                    }
-                                }
-                            ]
-                        })));
-                    }
-                    const images = val.filter(item => item.type == 'Image').map(item => item.value);
-                    if (images) {
-                        return Promise.all(images.map(item => client.sendGroupMessage({
-                            group_id: event.ChannelId,
-                            message: [
-                                {
-                                    type: 'image',
-                                    data: {
-                                        file: `base64://${item.toString('base64')}`
-                                    }
-                                }
-                            ]
-                        })));
-                    }
-                    return Promise.all([]);
-                },
-                mention: async () => {
-                    return [];
-                }
-            }
-        }
-    };
-});
-
-export { INDEXV11 as default };