fix群成员接口

This commit is contained in:
linyuchen
2023-11-01 21:55:05 +08:00
parent e2759417f5
commit 18ffad8109
6 changed files with 166 additions and 91 deletions

View File

@@ -6,7 +6,9 @@
"scripts": { "scripts": {
"test": "echo \"Error: no test specified\" && exit 1", "test": "echo \"Error: no test specified\" && exit 1",
"build-mac": "tsc && npm run deploy-mac", "build-mac": "tsc && npm run deploy-mac",
"deploy-mac": "cp dist/* ~/Library/Containers/com.tencent.qq/Data/Documents/LiteLoaderQQNT/plugins/LLOnebot/" "deploy-mac": "cp dist/* ~/Library/Containers/com.tencent.qq/Data/Documents/LiteLoaderQQNT/plugins/LLOnebot/",
"build-win": "tsc && npm run deploy-win",
"deploy-win": "cmd /c \"copy dist\\* %USERPROFILE%\\documents\\LiteLoaderQQNT\\plugins\\LLOnebot\\\""
}, },
"author": "", "author": "",
"license": "ISC", "license": "ISC",

5
src/global.d.ts vendored
View File

@@ -4,6 +4,7 @@ import {Group, GroupMemberInfo, MessageElement, Peer, PostDataSendMsg, SendMessa
declare var LLAPI: { declare var LLAPI: {
on(event: "new-messages", callback: (data: MessageElement[]) => void): void; on(event: "new-messages", callback: (data: MessageElement[]) => void): void;
on(event: "context-msg-menu", callback: (event: any, target: any, msgIds:any) => void): void;
getAccountInfo(): Promise<{ getAccountInfo(): Promise<{
uid: string // qq uid: string // qq
uin: string // 一串加密的字符串 uin: string // 一串加密的字符串
@@ -14,7 +15,8 @@ declare var LLAPI: {
sendMessage(peer: Peer, message: SendMessage[]): Promise<void>; sendMessage(peer: Peer, message: SendMessage[]): Promise<void>;
getGroupsList(forced: boolean): Promise<Group[]> getGroupsList(forced: boolean): Promise<Group[]>
getFriendsList(forced: boolean): Promise<User[]> getFriendsList(forced: boolean): Promise<User[]>
getGroupMemberList(group_id: string, num: number): Promise<{result: { infos: Record<string, GroupMemberInfo> }}> getGroupMemberList(group_id: string, num: number): Promise<{result: { infos: Map<string, GroupMemberInfo> }}>
getPeer(): Promise<Peer>
}; };
@@ -28,6 +30,7 @@ declare var llonebot: {
updateFriends: (friends: User[]) => void updateFriends: (friends: User[]) => void
updateGroupMembers: (data: { groupMembers: User[], group_id: string }) => void updateGroupMembers: (data: { groupMembers: User[], group_id: string }) => void
startExpress: () => void startExpress: () => void
log(data: any): void
}; };
declare global { declare global {

View File

@@ -10,7 +10,6 @@ const CHANNEL_SEND_MSG = "llonebot_sendMsg"
let groups: Group[] = [] let groups: Group[] = []
let friends: User[] = [] let friends: User[] = []
let groupMembers: {group_id: string, groupMembers: User[]}[] = []
function sendIPCMsg(channel: string, data: any){ function sendIPCMsg(channel: string, data: any){
let contents = webContents.getAllWebContents(); let contents = webContents.getAllWebContents();
@@ -68,16 +67,16 @@ function startExpress(event: any) {
}) })
} }
else if (jsonData.action == "get_group_member_list"){ else if (jsonData.action == "get_group_member_list"){
let group = groupMembers.find(group => group.group_id == jsonData.params.group_id) let group = groups.find(group => group.uid == jsonData.params.group_id)
if (group){ if (group){
resData["data"] = group.groupMembers.map(member => { resData["data"] = group?.members?.map(member => {
return { return {
user_id: member.uin, user_id: member.uin,
user_name: member.nickName, user_name: member.cardName || member.nick,
user_display_name: "" user_display_name: member.cardName || member.nick
} }
}) }) || []
} }
else{ else{
resData["data"] = [] resData["data"] = []
@@ -113,16 +112,6 @@ function onLoad(plugin: any) {
friends = arg friends = arg
}) })
ipcMain.on("updateGroupMembers", (event: any, arg: {groupMembers: User[], group_id: string}) => {
let existMembers = groupMembers.find(group => group.group_id == arg.group_id)
if (existMembers){
existMembers.groupMembers = arg.groupMembers
}
else{
groupMembers.push(arg);
}
})
ipcMain.on("postOnebotData", (event: any, arg: any) => { ipcMain.on("postOnebotData", (event: any, arg: any) => {
// try { // try {
// // const fetch2 = require("./electron-fetch"); // // const fetch2 = require("./electron-fetch");

View File

@@ -16,9 +16,6 @@ contextBridge.exposeInMainWorld("llonebot", {
updateFriends: (friends: User[]) => { updateFriends: (friends: User[]) => {
ipcRenderer.send("updateFriends", friends); ipcRenderer.send("updateFriends", friends);
}, },
updateGroupMembers: (data: {groupMembers: User[], group_id: string}) => {
ipcRenderer.send("updateGroupMembers", data);
},
listenSendMessage: (handle: (jsonData: PostDataSendMsg) => void) => { listenSendMessage: (handle: (jsonData: PostDataSendMsg) => void) => {
ipcRenderer.on("llonebot_sendMsg", (event: any, args: PostDataSendMsg) => { ipcRenderer.on("llonebot_sendMsg", (event: any, args: PostDataSendMsg) => {
handle(args) handle(args)

View File

@@ -2,7 +2,7 @@
// import express from "express"; // import express from "express";
// const { ipcRenderer } = require('electron'); // const { ipcRenderer } = require('electron');
import {AtType, Group, MessageElement, Peer, PostDataSendMsg, User} from "./types"; import {Group, MessageElement, Peer, PostDataSendMsg, User} from "./types";
const host = "http://localhost:5000" const host = "http://localhost:5000"
@@ -10,9 +10,9 @@ let self_qq: string = ""
let groups: Group[] = [] let groups: Group[] = []
let friends: User[] = [] let friends: User[] = []
let uid_maps: Record<string, User> = {} // 一串加密的字符串 -> qq号 let uid_maps: Record<string, User> = {} // 一串加密的字符串 -> qq号
async function getUserInfo(uid: string): Promise<User>{ async function getUserInfo(uid: string): Promise<User> {
let user = uid_maps[uid] let user = uid_maps[uid]
if (!user){ if (!user) {
// 从服务器获取用户信息 // 从服务器获取用户信息
user = await window.LLAPI.getUserInfo(uid) user = await window.LLAPI.getUserInfo(uid)
uid_maps[uid] = user uid_maps[uid] = user
@@ -24,13 +24,65 @@ function getFriend(qq: string) {
return friends.find(friend => friend.uid == qq) return friends.find(friend => friend.uid == qq)
} }
function getGroup(qq: string) { async function getGroup(qq: string) {
return groups.find(group => group.uid == qq) let group = groups.find(group => group.uid == qq)
if (!group) {
await getGroups();
group = groups.find(group => group.uid == qq)
}
return group
}
async function getGroups() {
let __groups = await window.LLAPI.getGroupsList(false)
for (let group of __groups) {
group.members = [];
let existGroup = groups.find(g => g.uid == group.uid)
if (!existGroup) {
console.log("更新群列表", groups)
groups.push(group)
window.llonebot.updateGroups(groups)
}
}
return groups
}
async function getGroupMembers(group_qq: string, forced: boolean = false) {
let group = await getGroup(group_qq)
if (!group?.members || group!.members!.length == 0 || forced) {
let res = (await window.LLAPI.getGroupMemberList(group_qq + "_groupMemberList_MainWindow", 5000))
// console.log(`更新群${group}成员列表 await`, _res)
// window.LLAPI.getGroupMemberList(group_qq + "_groupMemberList_MainWindow", 5000).then(res =>{
let members = res.result.infos.values();
if (members && forced) {
group!.members = []
}
for (const member of members) {
if (!group!.members!.find(m => m.uid == member.uid)) {
group!.members!.push(member)
}
}
window.llonebot.updateGroups(groups)
console.log(`更新群${group}成员列表`, group)
// })
}
return group?.members
}
async function getGroupMember(group_qq: string, member_uid: string) {
let members = await getGroupMembers(group_qq)
if (members) {
let member = members.find(member => member.uid == member_uid)
if (!member) {
members = await getGroupMembers(group_qq, true)
member = members?.find(member => member.uid == member_uid)
}
return member
}
} }
async function forwardMessage(message: MessageElement) {
function forwardMessage(message: MessageElement) {
try { try {
let onebot_message_data: any = { let onebot_message_data: any = {
self: { self: {
@@ -43,20 +95,13 @@ function forwardMessage(message: MessageElement) {
sub_type: "", sub_type: "",
message: [] message: []
} }
let group: Group // let group: Group
if (message.peer.chatType == "group") { if (message.peer.chatType == "group") {
let group_id = message.peer.uid let group_id = message.peer.uid
group = groups.find(group => group.uid == group_id)! let group = (await getGroup(group_id))!
onebot_message_data["group_id"] = message.peer.uid onebot_message_data["group_id"] = message.peer.uid
let groupMember = group.members!.find(member => member.uid == message.sender.uid) let groupMember = await getGroupMember(group_id, message.sender.uid)
if (groupMember) { onebot_message_data["user_id"] = groupMember!.uin
console.log("群成员信息存在,使用群成员信息")
onebot_message_data["user_id"] = groupMember.uin
}
else{
console.log("群成员信息不存在使用原始uid")
onebot_message_data["user_id"] = message.sender.uid
}
console.log("收到群消息", onebot_message_data) console.log("收到群消息", onebot_message_data)
} else if (message.peer.chatType == "private") { } else if (message.peer.chatType == "private") {
onebot_message_data["user_id"] = message.peer.uid onebot_message_data["user_id"] = message.peer.uid
@@ -66,13 +111,13 @@ function forwardMessage(message: MessageElement) {
data: {} data: {}
} }
if (element.textElement?.atType == AtType.atUser) { if (element.textElement?.atType == 2) {
message_data["type"] = "at" message_data["type"] = "at"
if (element.textElement.atUid != "0") { if (element.textElement.atUid != "0") {
message_data["data"]["mention"] = element.textElement.atUid message_data["data"]["mention"] = element.textElement.atUid
} else { } else {
let uid = element.textElement.atNtUid let uid = element.textElement.atNtUid
let atMember = group!.members!.find(member => member.uid == uid) let atMember = await getGroupMember(message.peer.uid, uid)
message_data["data"]["mention"] = atMember!.uin message_data["data"]["mention"] = atMember!.uin
} }
} else if (element.textElement) { } else if (element.textElement) {
@@ -98,77 +143,115 @@ function forwardMessage(message: MessageElement) {
async function handleNewMessage(messages: MessageElement[]) { async function handleNewMessage(messages: MessageElement[]) {
for (let message of messages) { for (let message of messages) {
if (message.peer.chatType == "group") { console.log("new message raw", message)
let group = groups.find(group => group.uid == message.peer.uid) forwardMessage(message).then();
if (!group) {
let members = (await window.LLAPI.getGroupMemberList(message.peer.uid + "_groupMemberList_MainWindow", 5000)).result.infos
let membersList = Object.values(members)
group = {
name: message.peer.name,
uid: message.peer.uid,
members: membersList
}
groups.push(group)
window.llonebot.updateGroups(groups);
}
}
forwardMessage(message);
} }
} }
async function getGroups(){ async function listenSendMessage(postData: PostDataSendMsg) {
groups = await window.LLAPI.getGroupsList(false) if (postData.action == "send_private_msg" || postData.action == "send_group_msg") {
for (let group of groups) { let peer: Peer | null = null;
group.members = []; if (postData.action == "send_private_msg") {
let friend = getFriend(postData.params.user_id)
if (friend) {
peer = {
chatType: "private",
name: friend.nickName,
uid: friend.uin
}
}
} else if (postData.action == "send_group_msg") {
let group = await getGroup(postData.params.group_id)
if (group) {
peer = {
chatType: "group",
name: group.name,
uid: group.uid
}
} else {
console.log("未找到群, 发送群消息失败", postData)
}
}
if (peer) {
window.LLAPI.sendMessage(peer, postData.params.message).then(res => console.log("消息发送成功:", res),
err => console.log("消息发送失败", postData, err))
}
} }
window.llonebot.updateGroups(groups)
} }
let chatListEle: HTMLCollectionOf<Element>
function onLoad() { function onLoad() {
window.llonebot.startExpress(); window.llonebot.startExpress();
window.llonebot.listenSendMessage((postData: PostDataSendMsg) => { window.llonebot.listenSendMessage((postData: PostDataSendMsg) => {
if (postData.action == "send_private_msg" || postData.action == "send_group_msg") { listenSendMessage(postData).then()
let peer: Peer | null = null;
if (postData.action == "send_private_msg") {
let friend = getFriend(postData.params.user_id)
if (friend) {
peer = {
chatType: "private",
name: friend.nickName,
uid: friend.uin
}
}
} else if (postData.action == "send_group_msg") {
let group = getGroup(postData.params.group_id)
if (group) {
peer = {
chatType: "group",
name: group.name,
uid: group.uid
}
}
}
if (peer) {
window.LLAPI.sendMessage(peer, postData.params.message).then(res => console.log("消息发送成功:", res),
err => console.log("消息发送失败", postData, err))
}
}
}); });
getGroups().then() getGroups().then()
function onNewMessages(messages: MessageElement[]){
async function func(messages: MessageElement[]){ function onNewMessages(messages: MessageElement[]) {
async function func(messages: MessageElement[]) {
console.log("收到新消息", messages) console.log("收到新消息", messages)
if (!self_qq) { if (!self_qq) {
self_qq = (await window.LLAPI.getAccountInfo()).uin self_qq = (await window.LLAPI.getAccountInfo()).uin
} }
await handleNewMessage(messages); await handleNewMessage(messages);
} }
func(messages).then(() => {})
func(messages).then(() => {
})
console.log("chatListEle", chatListEle)
} }
window.LLAPI.on("new-messages", onNewMessages); window.LLAPI.on("new-messages", onNewMessages);
window.LLAPI.on("context-msg-menu", (event, target, msgIds) => {
console.log(event);
window.LLAPI.getPeer().then(peer => {
// console.log("current peer", peer)
if (peer && peer.chatType == "group") {
getGroupMembers(peer.uid, false).then()
}
})
})
// console.log("getAccountInfo", LLAPI.getAccountInfo()); // console.log("getAccountInfo", LLAPI.getAccountInfo());
function getChatListEle() {
chatListEle = document.getElementsByClassName("viewport-list__inner")
console.log("chatListEle", chatListEle)
if (chatListEle.length == 0) {
setTimeout(getChatListEle, 500)
} else {
// try {
// // 选择要观察的目标节点
// const targetNode = chatListEle[0];
//
// // 创建一个观察器实例并传入回调函数
// const observer = new MutationObserver(function (mutations) {
// mutations.forEach(function (mutation) {
// // console.log("chat list changed", mutation.type); // 输出 mutation 的类型
// // 获得当前聊天窗口
// window.LLAPI.getPeer().then(peer => {
// // console.log("current peer", peer)
// if (peer && peer.chatType == "group"){
// getGroupMembers(peer.uid, false).then()
// }
// })
// });
// });
//
// // 配置观察选项
// const config = {attributes: true, childList: true, subtree: true};
//
// // 传入目标节点和观察选项
// observer.observe(targetNode, config);
//
// }catch (e) {
// window.llonebot.log(e)
// }
}
}
getChatListEle();
} }
// 打开设置界面时触发 // 打开设置界面时触发

View File

@@ -6,7 +6,8 @@
"strict": true, "strict": true,
"esModuleInterop": true, "esModuleInterop": true,
"allowSyntheticDefaultImports": true, "allowSyntheticDefaultImports": true,
"moduleResolution": "node" "moduleResolution": "node",
// "declaration": false
}, },
"include": [ "include": [
"src" "src"