LLOneBot/src/common/data.ts

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

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

let globalMsgId = Date.now()

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=null, 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, Friend> = {}  // 一串加密的字符串(uid) -> qq号

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

export const version = "v3.3.0"
export const heartInterval = 15000 // 毫秒