mirror of
https://github.com/NapNeko/NapCatQQ.git
synced 2026-02-12 07:50:25 +00:00
NapCatQQ
This commit is contained in:
linyuchen
335
src/onebot11/main.ts
Normal file
335
src/onebot11/main.ts
Normal file
@@ -0,0 +1,335 @@
|
||||
import { napCatCore } from '@/core';
|
||||
import { MsgListener } from '@/core/qqnt/listeners';
|
||||
import { OB11Constructor } from '@/onebot11/constructor';
|
||||
import { postOB11Event } from '@/onebot11/server/postOB11Event';
|
||||
import {
|
||||
ChatType,
|
||||
FriendRequest,
|
||||
Group,
|
||||
GroupMemberRole,
|
||||
GroupNotify,
|
||||
GroupNotifyTypes,
|
||||
RawMessage
|
||||
} from '@/core/qqnt/entities';
|
||||
import { ob11Config } from '@/onebot11/config';
|
||||
import { ob11HTTPServer } from '@/onebot11/server/http';
|
||||
import { ob11WebsocketServer } from '@/onebot11/server/ws/WebsocketServer';
|
||||
import { ob11ReverseWebsockets } from '@/onebot11/server/ws/ReverseWebsocket';
|
||||
import { friendRequests, getFriend, getGroup, getGroupMember, groupNotifies, selfInfo } from '@/common/data';
|
||||
import { dbUtil } from '@/common/utils/db';
|
||||
import { BuddyListener, GroupListener, NodeIKernelBuddyListener } from '@/core/qqnt/listeners';
|
||||
import { OB11FriendRequestEvent } from '@/onebot11/event/request/OB11FriendRequest';
|
||||
import { NTQQGroupApi, NTQQUserApi } from '@/core/qqnt/apis';
|
||||
import { log } from '@/common/utils/log';
|
||||
import { OB11GroupRequestEvent } from '@/onebot11/event/request/OB11GroupRequest';
|
||||
import { OB11GroupAdminNoticeEvent } from '@/onebot11/event/notice/OB11GroupAdminNoticeEvent';
|
||||
import { GroupDecreaseSubType, OB11GroupDecreaseEvent } from '@/onebot11/event/notice/OB11GroupDecreaseEvent';
|
||||
import { OB11FriendRecallNoticeEvent } from '@/onebot11/event/notice/OB11FriendRecallNoticeEvent';
|
||||
import { OB11GroupRecallNoticeEvent } from '@/onebot11/event/notice/OB11GroupRecallNoticeEvent';
|
||||
|
||||
|
||||
export class NapCatOnebot11 {
|
||||
private bootTime: number = Date.now() / 1000;
|
||||
|
||||
|
||||
constructor() {
|
||||
// console.log('ob11 init');
|
||||
napCatCore.addLoginSuccessCallback(this.onReady.bind(this));
|
||||
}
|
||||
|
||||
public onReady() {
|
||||
console.log('ob11 ready');
|
||||
ob11Config.read();
|
||||
if (ob11Config.enableHttp) {
|
||||
ob11HTTPServer.start(ob11Config.httpPort);
|
||||
}
|
||||
if (ob11Config.enableWs) {
|
||||
ob11WebsocketServer.start(ob11Config.wsPort);
|
||||
}
|
||||
if (ob11Config.enableWsReverse) {
|
||||
ob11ReverseWebsockets.start();
|
||||
}
|
||||
// MsgListener
|
||||
const msgListener = new MsgListener();
|
||||
msgListener.onRecvSysMsg = (protobuf: number[]) => {
|
||||
// todo: 解码protobuf，这里可以拿到戳一戳，但是群戳一戳只有群号
|
||||
const buffer = Buffer.from(protobuf);
|
||||
// 转换为十六进制字符串
|
||||
const hexString = protobuf.map(byte => {
|
||||
// 将负数转换为补码表示的正数
|
||||
byte = byte < 0 ? 256 + byte : byte;
|
||||
// 转换为十六进制，确保结果为两位数
|
||||
return ('0' + byte.toString(16)).slice(-2);
|
||||
}).join('');
|
||||
// console.log('ob11 onRecvSysMsg', hexString, Date.now() / 1000);
|
||||
// console.log(buffer.toString());
|
||||
// console.log('ob11 onRecvSysMsg', JSON.stringify(msg, null, 2));
|
||||
};
|
||||
msgListener.onRecvMsg = (msg) => {
|
||||
// console.log('ob11 onRecvMsg', JSON.stringify(msg, null, 2));
|
||||
for (const m of msg) {
|
||||
if (this.bootTime > parseInt(m.msgTime)) {
|
||||
continue;
|
||||
}
|
||||
new Promise((resolve) => {
|
||||
dbUtil.addMsg(m).then(msgShortId => {
|
||||
m.id = msgShortId;
|
||||
this.postReceiveMsg([m]).then().catch(log);
|
||||
}).catch(log);
|
||||
}).then();
|
||||
}
|
||||
};
|
||||
msgListener.onMsgInfoListUpdate = (msgList) => {
|
||||
this.postRecallMsg(msgList).then().catch(log);
|
||||
};
|
||||
msgListener.onAddSendMsg = (msg) => {
|
||||
if (ob11Config.reportSelfMessage) {
|
||||
dbUtil.addMsg(msg).then(id => {
|
||||
msg.id = id;
|
||||
this.postReceiveMsg([msg]).then().catch(log);
|
||||
});
|
||||
}
|
||||
};
|
||||
napCatCore.service.msg.addMsgListener(msgListener);
|
||||
console.log('ob11 msg listener added');
|
||||
|
||||
// BuddyListener
|
||||
const buddyListener = new BuddyListener();
|
||||
buddyListener.onBuddyReqChange = ((req) => {
|
||||
this.postFriendRequest(req.buddyReqs).then().catch(log);
|
||||
});
|
||||
napCatCore.service.buddy.addBuddyListener(buddyListener);
|
||||
console.log('ob11 buddy listener added');
|
||||
|
||||
// GroupListener
|
||||
const groupListener = new GroupListener();
|
||||
groupListener.onGroupNotifiesUpdated = (doubt, notifies) => {
|
||||
// console.log('ob11 onGroupNotifiesUpdated', notifies);
|
||||
this.postGroupNotifies(notifies).then().catch(e => log('postGroupNotifies error: ', e));
|
||||
};
|
||||
groupListener.onJoinGroupNotify = (...notify) => {
|
||||
// console.log('ob11 onJoinGroupNotify', notify);
|
||||
};
|
||||
groupListener.onGroupListUpdate = (updateType, groupList) => {
|
||||
// console.log('ob11 onGroupListUpdate', updateType, groupList);
|
||||
// this.postGroupMemberChange(groupList).then();
|
||||
};
|
||||
|
||||
napCatCore.service.group.addGroupListener(groupListener);
|
||||
console.log('ob11 group listener added');
|
||||
}
|
||||
|
||||
async postReceiveMsg(msgList: RawMessage[]) {
|
||||
const {debug, reportSelfMessage} = ob11Config;
|
||||
for (const message of msgList) {
|
||||
// console.log("ob11 收到新消息", message)
|
||||
// if (message.senderUin !== selfInfo.uin){
|
||||
// message.msgShortId = await dbUtil.addMsg(message);
|
||||
// }
|
||||
|
||||
OB11Constructor.message(message).then((msg) => {
|
||||
if (debug) {
|
||||
msg.raw = message;
|
||||
} else {
|
||||
if (msg.message.length === 0) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
const isSelfMsg = msg.user_id.toString() == selfInfo.uin;
|
||||
if (isSelfMsg && !reportSelfMessage) {
|
||||
return;
|
||||
}
|
||||
if (isSelfMsg) {
|
||||
msg.target_id = parseInt(message.peerUin);
|
||||
}
|
||||
postOB11Event(msg);
|
||||
// log("post msg", msg)
|
||||
}).catch(e => log('constructMessage error: ', e));
|
||||
OB11Constructor.GroupEvent(message).then(groupEvent => {
|
||||
if (groupEvent) {
|
||||
// log("post group event", groupEvent);
|
||||
postOB11Event(groupEvent);
|
||||
}
|
||||
}).catch(e => log('constructGroupEvent error: ', e));
|
||||
}
|
||||
}
|
||||
|
||||
async postGroupNotifies(notifies: GroupNotify[]) {
|
||||
for (const notify of notifies) {
|
||||
try {
|
||||
notify.time = Date.now();
|
||||
const notifyTime = parseInt(notify.seq) / 1000 / 1000;
|
||||
// log(`群通知时间${notifyTime}`, `LLOneBot启动时间${this.bootTime}`);
|
||||
if (notifyTime < this.bootTime) {
|
||||
continue;
|
||||
}
|
||||
const flag = notify.group.groupCode + '|' + notify.seq;
|
||||
const existNotify = groupNotifies[flag];
|
||||
if (existNotify) {
|
||||
continue;
|
||||
}
|
||||
log('收到群通知', notify);
|
||||
groupNotifies[flag] = notify;
|
||||
// let member2: GroupMember;
|
||||
// if (notify.user2.uid) {
|
||||
// member2 = await getGroupMember(notify.group.groupCode, null, notify.user2.uid);
|
||||
// }
|
||||
if ([GroupNotifyTypes.ADMIN_SET, GroupNotifyTypes.ADMIN_UNSET].includes(notify.type)) {
|
||||
const member1 = await getGroupMember(notify.group.groupCode, notify.user1.uid);
|
||||
log('有管理员变动通知');
|
||||
// refreshGroupMembers(notify.group.groupCode).then();
|
||||
const groupAdminNoticeEvent = new OB11GroupAdminNoticeEvent();
|
||||
groupAdminNoticeEvent.group_id = parseInt(notify.group.groupCode);
|
||||
log('开始获取变动的管理员');
|
||||
if (member1) {
|
||||
log('变动管理员获取成功');
|
||||
groupAdminNoticeEvent.user_id = parseInt(member1.uin);
|
||||
groupAdminNoticeEvent.sub_type = notify.type == GroupNotifyTypes.ADMIN_UNSET ? 'unset' : 'set';
|
||||
// member1.role = notify.type == GroupNotifyTypes.ADMIN_SET ? GroupMemberRole.admin : GroupMemberRole.normal;
|
||||
postOB11Event(groupAdminNoticeEvent, true);
|
||||
} else {
|
||||
log('获取群通知的成员信息失败', notify, getGroup(notify.group.groupCode));
|
||||
}
|
||||
} else if (notify.type == GroupNotifyTypes.MEMBER_EXIT || notify.type == GroupNotifyTypes.KICK_MEMBER) {
|
||||
log('有成员退出通知', notify);
|
||||
try {
|
||||
const member1 = await NTQQUserApi.getUserDetailInfo(notify.user1.uid);
|
||||
let operatorId = member1.uin;
|
||||
let subType: GroupDecreaseSubType = 'leave';
|
||||
if (notify.user2.uid) {
|
||||
// 是被踢的
|
||||
const member2 = await getGroupMember(notify.group.groupCode, notify.user2.uid);
|
||||
if (member2) {
|
||||
operatorId = member2.uin;
|
||||
}
|
||||
subType = 'kick';
|
||||
}
|
||||
const groupDecreaseEvent = new OB11GroupDecreaseEvent(parseInt(notify.group.groupCode), parseInt(member1.uin), parseInt(operatorId), subType);
|
||||
postOB11Event(groupDecreaseEvent, true);
|
||||
} catch (e: any) {
|
||||
log('获取群通知的成员信息失败', notify, e.stack.toString());
|
||||
}
|
||||
} else if ([GroupNotifyTypes.JOIN_REQUEST].includes(notify.type)) {
|
||||
log('有加群请求');
|
||||
const groupRequestEvent = new OB11GroupRequestEvent();
|
||||
groupRequestEvent.group_id = parseInt(notify.group.groupCode);
|
||||
let requestQQ = '';
|
||||
try {
|
||||
requestQQ = (await NTQQUserApi.getUserDetailInfo(notify.user1.uid)).uin;
|
||||
} catch (e) {
|
||||
log('获取加群人QQ号失败', e);
|
||||
}
|
||||
groupRequestEvent.user_id = parseInt(requestQQ) || 0;
|
||||
groupRequestEvent.sub_type = 'add';
|
||||
groupRequestEvent.comment = notify.postscript;
|
||||
groupRequestEvent.flag = flag;
|
||||
postOB11Event(groupRequestEvent);
|
||||
} else if (notify.type == GroupNotifyTypes.INVITE_ME) {
|
||||
log('收到邀请我加群通知');
|
||||
const groupInviteEvent = new OB11GroupRequestEvent();
|
||||
groupInviteEvent.group_id = parseInt(notify.group.groupCode);
|
||||
let user_id = (await getFriend(notify.user2.uid))?.uin;
|
||||
if (!user_id) {
|
||||
user_id = (await NTQQUserApi.getUserDetailInfo(notify.user2.uid))?.uin;
|
||||
}
|
||||
groupInviteEvent.user_id = parseInt(user_id);
|
||||
groupInviteEvent.sub_type = 'invite';
|
||||
groupInviteEvent.flag = flag;
|
||||
postOB11Event(groupInviteEvent);
|
||||
}
|
||||
} catch (e: any) {
|
||||
log('解析群通知失败', e.stack.toString());
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
async postRecallMsg(msgList: RawMessage[]) {
|
||||
for (const message of msgList) {
|
||||
// log("message update", message.sendStatus, message.msgId, message.msgSeq)
|
||||
if (message.recallTime != '0') { //todo: 这个判断方法不太好，应该使用灰色消息元素来判断?
|
||||
// 撤回消息上报
|
||||
const oriMessage = await dbUtil.getMsgByLongId(message.msgId);
|
||||
if (!oriMessage) {
|
||||
continue;
|
||||
}
|
||||
if (message.chatType == ChatType.friend) {
|
||||
const friendRecallEvent = new OB11FriendRecallNoticeEvent(parseInt(message!.senderUin), oriMessage!.id!);
|
||||
postOB11Event(friendRecallEvent);
|
||||
} else if (message.chatType == ChatType.group) {
|
||||
let operatorId = message.senderUin;
|
||||
for (const element of message.elements) {
|
||||
const operatorUid = element.grayTipElement?.revokeElement.operatorUid;
|
||||
const operator = await getGroupMember(message.peerUin, operatorUid);
|
||||
operatorId = operator?.uin || message.senderUin;
|
||||
}
|
||||
const groupRecallEvent = new OB11GroupRecallNoticeEvent(
|
||||
parseInt(message.peerUin),
|
||||
parseInt(message.senderUin),
|
||||
parseInt(operatorId),
|
||||
oriMessage.id!
|
||||
);
|
||||
postOB11Event(groupRecallEvent);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async postFriendRequest(reqs: FriendRequest[]) {
|
||||
for (const req of reqs) {
|
||||
if (parseInt(req.reqTime) < this.bootTime) {
|
||||
continue;
|
||||
}
|
||||
const flag = req.friendUid + '|' + req.reqTime;
|
||||
if (friendRequests[flag]) {
|
||||
continue;
|
||||
}
|
||||
friendRequests[flag] = req;
|
||||
const friendRequestEvent = new OB11FriendRequestEvent();
|
||||
try {
|
||||
const requester = await NTQQUserApi.getUserDetailInfo(req.friendUid);
|
||||
friendRequestEvent.user_id = parseInt(requester.uin);
|
||||
} catch (e) {
|
||||
log('获取加好友者QQ号失败', e);
|
||||
}
|
||||
friendRequestEvent.flag = flag;
|
||||
friendRequestEvent.comment = req.extWords;
|
||||
postOB11Event(friendRequestEvent);
|
||||
}
|
||||
}
|
||||
|
||||
async postGroupMemberChange(groupList: Group[]) {
|
||||
// todo: 有无更好的方法判断群成员变动
|
||||
const newGroupList = groupList;
|
||||
for (const group of newGroupList) {
|
||||
const existGroup = await getGroup(group.groupCode);
|
||||
if (existGroup) {
|
||||
if (existGroup.memberCount > group.memberCount) {
|
||||
log(`群(${group.groupCode})成员数量减少${existGroup.memberCount} -> ${group.memberCount}`);
|
||||
const oldMembers = existGroup.members;
|
||||
const newMembers = await NTQQGroupApi.getGroupMembers(group.groupCode);
|
||||
group.members = newMembers;
|
||||
const newMembersSet = new Set<string>(); // 建立索引降低时间复杂度
|
||||
|
||||
for (const member of newMembers) {
|
||||
newMembersSet.add(member.uin);
|
||||
}
|
||||
|
||||
// 判断bot是否是管理员，如果是管理员不需要从这里得知有人退群，这里的退群无法得知是主动退群还是被踢
|
||||
const bot = await getGroupMember(group.groupCode, selfInfo.uin);
|
||||
if (bot!.role == GroupMemberRole.admin || bot!.role == GroupMemberRole.owner) {
|
||||
continue;
|
||||
}
|
||||
for (const member of oldMembers) {
|
||||
if (!newMembersSet.has(member.uin) && member.uin != selfInfo.uin) {
|
||||
postOB11Event(new OB11GroupDecreaseEvent(parseInt(group.groupCode), parseInt(member.uin), parseInt(member.uin), 'leave'));
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// export const napCatOneBot11 = new NapCatOnebot11();
|
||||
Reference in New Issue
Block a user