From b7a878130828a5a1cef6c1d752b014a8d096cacd Mon Sep 17 00:00:00 2001
From: "Wesley F. Young" <willamwei@outlook.com>
Date: Sun, 11 Aug 2024 00:16:26 +0800
Subject: [PATCH] feat: onInputStatusPush & onRecvMsg

---
 src/onebot/main.ts | 94 ++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 83 insertions(+), 11 deletions(-)

diff --git a/src/onebot/main.ts b/src/onebot/main.ts
index 3aa1fc35..1156c0fe 100644
--- a/src/onebot/main.ts
+++ b/src/onebot/main.ts
@@ -1,10 +1,14 @@
-import { InstanceContext, MsgListener, NapCatCore } from '@/core';
+import { InstanceContext, MsgListener, NapCatCore, RawMessage } from '@/core';
 import { OB11Config } from './helper/config';
 import { NapCatPathWrapper } from '@/common/framework/napcat';
 import { OneBotApiContextType } from './types/adapter';
 import { OneBotFriendApi, OneBotGroupApi, OneBotUserApi } from './api';
 import { OB11NetworkManager } from '@/onebot/network';
 import { OB11InputStatusEvent } from '@/onebot/event/notice/OB11InputStatusEvent';
+import { MessageUnique } from '@/common/utils/MessageUnique';
+import { OB11Constructor } from '@/onebot/helper/data';
+import { logOB11Message } from '@/onebot/helper/log';
+import { proxiedListenerOf } from '@/common/utils/proxy-handler';
 
 //OneBot实现类
 export class NapCatOneBot11Adapter {
@@ -15,6 +19,8 @@ export class NapCatOneBot11Adapter {
     apiContext: OneBotApiContextType;
     networkManager: OB11NetworkManager;
 
+    private bootTime = Date.now() / 1000;
+
     constructor(core: NapCatCore, context: InstanceContext, pathWrapper: NapCatPathWrapper) {
         this.core = core;
         this.context = context;
@@ -27,16 +33,18 @@ export class NapCatOneBot11Adapter {
         this.networkManager = new OB11NetworkManager();
         this.InitOneBot();
     }
+
     async InitOneBot() {
-        let NTQQUserApi = this.core.getApiContext().UserApi;
-        let selfInfo = this.core.selfInfo;
-        let ob11Config = this.config.configData;
+        const NTQQUserApi = this.core.getApiContext().UserApi;
+        const selfInfo = this.core.selfInfo;
+        const ob11Config = this.config.configData;
+
         const serviceInfo = `
     HTTP服务 ${ob11Config.http.enable ? '已启动' : '未启动'}, ${ob11Config.http.host}:${ob11Config.http.port}
     HTTP上报服务 ${ob11Config.http.enablePost ? '已启动' : '未启动'}, 上报地址: ${ob11Config.http.postUrls}
     WebSocket服务 ${ob11Config.ws.enable ? '已启动' : '未启动'}, ${ob11Config.ws.host}:${ob11Config.ws.port}
-    WebSocket反向服务 ${ob11Config.reverseWs.enable ? '已启动' : '未启动'}, 反向地址: ${ob11Config.reverseWs.urls}
-    `;
+    WebSocket反向服务 ${ob11Config.reverseWs.enable ? '已启动' : '未启动'}, 反向地址: ${ob11Config.reverseWs.urls}`;
+
         NTQQUserApi.getUserDetailInfo(selfInfo.uid).then(user => {
             selfInfo.nick = user.nick;
             this.context.logger.setLogSelfInfo(selfInfo);
@@ -46,17 +54,81 @@ export class NapCatOneBot11Adapter {
         await this.initMsgListener();
 
     }
+
     async initMsgListener() {
         const msgListener = new MsgListener();
 
         msgListener.onInputStatusPush = async data => {
             const uin = await this.core.ApiContext.UserApi.getUinByUidV2(data.fromUin);
             this.context.logger.log(`[Notice] [输入状态] ${uin} ${data.statusText}`);
-            // this.networkManager.emitEvent(new OB11InputStatusEvent(
-            //     parseInt(uin),
-            //     data.eventType,
-            //     data.statusText
-            // ));
+            await this.networkManager.emitEvent(new OB11InputStatusEvent(
+                this.core,
+                parseInt(uin),
+                data.eventType,
+                data.statusText
+            ));
         };
+
+        msgListener.onRecvMsg = async msg => {
+            for (const m of msg) {
+                if (this.bootTime > parseInt(m.msgTime)) {
+                    this.context.logger.logDebug(`消息时间${m.msgTime}早于启动时间${this.bootTime}，忽略上报`);
+                    continue;
+                }
+                m.id = MessageUnique.createMsg(
+                    {
+                        chatType: m.chatType,
+                        peerUid: m.peerUid,
+                        guildId: ''
+                    },
+                    m.msgId
+                );
+                await this.postReceiveMsg(m)
+                    .catch(e => this.context.logger.logError('处理消息失败', e));
+            }
+        };
+
+        this.context.session.getMsgService().addKernelMsgListener(
+            new this.context.wrapper.NodeIKernelMsgListener(proxiedListenerOf(msgListener, this.context.logger)),
+        );
+    }
+
+    async postReceiveMsg(message: RawMessage) {
+        const { debug, reportSelfMessage } = this.config;
+        this.context.logger.logDebug('收到新消息', message);
+        OB11Constructor.message(this.core, message, this.config.messagePostFormat).then((ob11Msg) => {
+            this.context.logger.logDebug('收到消息: ', ob11Msg);
+            if (debug) {
+                ob11Msg.raw = message;
+            } else {
+                if (ob11Msg.message.length === 0) {
+                    return;
+                }
+            }
+            logOB11Message(this.core, ob11Msg)
+                .catch(e => this.context.logger.logError('logMessage error: ', e));
+            const isSelfMsg = ob11Msg.user_id.toString() == this.core.selfInfo.uin;
+            if (isSelfMsg && !reportSelfMessage) {
+                return;
+            }
+            if (isSelfMsg) {
+                ob11Msg.target_id = parseInt(message.peerUin);
+            }
+            this.networkManager.emitEvent(ob11Msg);
+        }).catch(e => this.context.logger.logError('constructMessage error: ', e));
+
+        OB11Constructor.GroupEvent(this.core, message).then(groupEvent => {
+            if (groupEvent) {
+                // log("post group event", groupEvent);
+                this.networkManager.emitEvent(groupEvent);
+            }
+        }).catch(e => this.context.logger.logError('constructGroupEvent error: ', e));
+
+        OB11Constructor.PrivateEvent(this.core, message).then(privateEvent => {
+            if (privateEvent) {
+                // log("post private event", privateEvent);
+                this.networkManager.emitEvent(privateEvent);
+            }
+        });
     }
 }