NapCatQQ/src/core/apis/friend.ts

import { Friend, FriendV2, User } from '@/core/entities';
import { BuddyListReqType, InstanceContext, NapCatCore, NodeIKernelBuddyListener, NodeIKernelBuddyService, NodeIKernelProfileService, OnBuddyChangeParams } from '@/core';
import { LimitedHashTable } from '@/common/utils/MessageUnique';

export class NTQQFriendApi {
    context: InstanceContext;
    core: NapCatCore;
    //friends: Map<string, Friend> = new Map<string, FriendV2>();

    constructor(context: InstanceContext, core: NapCatCore) {
        this.context = context;
        this.core = core;
        // if (!this.context.basicInfoWrapper.requireMinNTQQBuild('26702')) {
        //     this.getFriends(true);
        // }
    }

    async getBuddyV2(refresh = false): Promise<FriendV2[]> {
        const uids: string[] = [];
        const buddyService = this.context.session.getBuddyService();
        const buddyListV2 = refresh ? await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL) : await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL);
        uids.push(...buddyListV2.data.flatMap(item => item.buddyUids));
        const data = await this.core.eventWrapper.callNoListenerEvent<NodeIKernelProfileService['getCoreAndBaseInfo']>(
            'NodeIKernelProfileService/getCoreAndBaseInfo', 'nodeStore', uids,
        );
        return Array.from(data.values());
    }

    async getBuddyIdMap(refresh = false): Promise<LimitedHashTable<string, string>> {
        const uids: string[] = [];
        const retMap: LimitedHashTable<string, string> = new LimitedHashTable<string, string>(5000);
        const buddyService = this.context.session.getBuddyService();
        const buddyListV2 = refresh ? await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL) : await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL);
        uids.push(...buddyListV2.data.flatMap(item => item.buddyUids));
        const data = await this.core.eventWrapper.callNoListenerEvent<NodeIKernelProfileService['getCoreAndBaseInfo']>(
            'NodeIKernelProfileService/getCoreAndBaseInfo', 'nodeStore', uids,
        );
        data.forEach((value, key) => {
            retMap.set(value.uin!, value.uid!);
        });
        //console.log('getBuddyIdMap', retMap.getValue);
        return retMap;
    }

    async getBuddyV2ExWithCate(refresh = false) {
        const uids: string[] = [];
        const categoryMap: Map<string, any> = new Map();
        const buddyService = this.context.session.getBuddyService();
        const buddyListV2 = refresh ? (await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL)).data : (await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL)).data;
        uids.push(
            ...buddyListV2.flatMap(item => {
                item.buddyUids.forEach(uid => {
                    categoryMap.set(uid, { categoryId: item.categoryId, categoryName: item.categroyName });
                });
                return item.buddyUids;
            }));
        const data = await this.core.eventWrapper.callNoListenerEvent<NodeIKernelProfileService['getCoreAndBaseInfo']>(
            'NodeIKernelProfileService/getCoreAndBaseInfo', 'nodeStore', uids,
        );
        return buddyListV2.map(category => ({
            categoryId: category.categoryId,
            categorySortId: category.categorySortId,
            categoryName: category.categroyName,
            categoryMbCount: category.categroyMbCount,
            onlineCount: category.onlineCount,
            buddyList: category.buddyUids.map(uid => data.get(uid)!).filter(value => value),
        }));
    }

    async isBuddy(uid: string) {
        return this.context.session.getBuddyService().isBuddy(uid);
    }
    async clearBuddyReqUnreadCnt() {
        return this.context.session.getBuddyService().clearBuddyReqUnreadCnt();
    }
    async getBuddyReq() {
        const [, ret] = await this.core.eventWrapper.CallNormalEventV2<
            NodeIKernelBuddyService['getBuddyReq'],
            NodeIKernelBuddyListener['onBuddyReqChange']
        >(
            'NodeIKernelBuddyService/getBuddyReq',
            'NodeIKernelBuddyListener/onBuddyReqChange',
            1,
            5000);
        return ret;
    }

    async handleFriendRequest(flag: string, accept: boolean) {
        const data = flag.split('|');
        if (data.length < 2) {
            return;
        }
        const friendUid = data[0];
        const reqTime = data[1];
        this.context.session.getBuddyService()?.approvalFriendRequest({
            friendUid: friendUid,
            reqTime: reqTime,
            accept,
        });
    }
}