LLOneBot/src/common/data.ts

import {NTQQApi} from '../ntqqapi/ntcall';
import {Friend, Group, GroupMember, GroupNotify, RawMessage, SelfInfo} from "../ntqqapi/types";

export let groups: Group[] = []
export let friends: Friend[] = []
export let msgHistory: Record<string, RawMessage> = {}  // msgId: RawMessage

let globalMsgId = Math.floor(Date.now() / 1000);

export function addHistoryMsg(msg: RawMessage): boolean {
    let existMsg = msgHistory[msg.msgId]
    if (existMsg) {
        Object.assign(existMsg, msg)
        msg.msgShortId = existMsg.msgShortId;
        return false
    }
    msg.msgShortId = ++globalMsgId
    msgHistory[msg.msgId] = msg
    return true
}

export function getHistoryMsgByShortId(shortId: number | string) {
    // log("getHistoryMsgByShortId", shortId, Object.values(msgHistory).map(m=>m.msgShortId))
    return Object.values(msgHistory).find(msg => msg.msgShortId.toString() == shortId.toString())
}


export async function getFriend(qq: string): Promise<Friend | undefined> {
    let friend = friends.find(friend => friend.uin === qq)
    // if (!friend){
    //     friends = (await NTQQApi.getFriends(true))
    //     friend = friends.find(friend => friend.uin === qq)
    // }
    return friend
}

export async function getGroup(qq: string): Promise<Group | undefined> {
    let group = groups.find(group => group.groupCode === qq)
    // if (!group){
    //     groups = await NTQQApi.getGroups(true);
    //     group = groups.find(group => group.groupCode === qq)
    // }
    return group
}

export async function getGroupMember(groupQQ: string, memberQQ: string, memberUid: string = null) {
    const group = await getGroup(groupQQ)
    if (group) {
        let filterFunc: (member: GroupMember) => boolean
        if (memberQQ) {
            filterFunc = member => member.uin === memberQQ
        } else if (memberUid) {
            filterFunc = member => member.uid === memberUid
        }
        let member = group.members?.find(filterFunc)
        if (!member) {
            const _members = await NTQQApi.getGroupMembers(groupQQ)
            if (_members.length) {
                group.members = _members
            }
            member = group.members?.find(filterFunc)
        }
        return member
    }
}

export let selfInfo: SelfInfo = {
    uid: "",
    uin: "",
    nick: "",
}


export function getHistoryMsgBySeq(seq: string) {
    return Object.values(msgHistory).find(msg => msg.msgSeq === seq)
}


export let uidMaps: Record<string, string> = {}  // 一串加密的字符串(uid) -> qq号

export function getUidByUin(uin: string) {
    for (const key in uidMaps) {
        if (uidMaps[key] === uin) {
            return key;
        }
    }
}

export const version = "3.7.0"

export let groupNotifies: Map<string, GroupNotify> = new Map();