LLOneBot/src/ntqqapi/api/friend.ts

import { Friend, FriendV2, SimpleInfo, CategoryFriend } from '../types'
import { ReceiveCmdS } from '../hook'
import { invoke, NTMethod, NTClass } from '../ntcall'
import { getSession } from '@/ntqqapi/wrapper'
import { BuddyListReqType } from '../services'
import { Dict, pick } from 'cosmokit'
import { Service, Context } from 'cordis'

declare module 'cordis' {
  interface Context {
    ntFriendApi: NTQQFriendApi
  }
}

export class NTQQFriendApi extends Service {
  constructor(protected ctx: Context) {
    super(ctx, 'ntFriendApi', true)
  }

  /** 大于或等于 26702 应使用 getBuddyV2 */
  async getFriends() {
    const data = await invoke<{
      data: {
        categoryId: number
        categroyName: string
        categroyMbCount: number
        buddyList: Friend[]
      }[]
    }>(
      'getBuddyList',
      [],
      {
        className: NTClass.NODE_STORE_API,
        cbCmd: ReceiveCmdS.FRIENDS,
        afterFirstCmd: false,
      }
    )
    const _friends: Friend[] = []
    for (const item of data.data) {
      _friends.push(...item.buddyList)
    }
    return _friends
  }

  async handleFriendRequest(flag: string, accept: boolean) {
    const data = flag.split('|')
    if (data.length < 2) {
      return
    }
    const friendUid = data[0]
    const reqTime = data[1]
    const session = getSession()
    if (session) {
      return session.getBuddyService().approvalFriendRequest({
        friendUid,
        reqTime,
        accept
      })
    } else {
      return await invoke(NTMethod.HANDLE_FRIEND_REQUEST, [{
        approvalInfo: {
          friendUid,
          reqTime,
          accept,
        },
      }])
    }
  }

  async getBuddyV2(refresh = false): Promise<FriendV2[]> {
    const session = getSession()
    if (session) {
      const uids: string[] = []
      const buddyService = session.getBuddyService()
      const buddyListV2 = await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL)
      uids.push(...buddyListV2.data.flatMap(item => item.buddyUids))
      const data = await session.getProfileService().getCoreAndBaseInfo('nodeStore', uids)
      return Array.from(data.values())
    } else {
      const data = await invoke<{
        buddyCategory: CategoryFriend[]
        userSimpleInfos: Record<string, SimpleInfo>
      }>(
        'getBuddyList',
        [refresh],
        {
          className: NTClass.NODE_STORE_API,
          cbCmd: ReceiveCmdS.FRIENDS,
          afterFirstCmd: false,
        }
      )
      const categoryUids: Map<number, string[]> = new Map()
      for (const item of data.buddyCategory) {
        categoryUids.set(item.categoryId, item.buddyUids)
      }
      return Object.values(data.userSimpleInfos).filter(v => v.baseInfo && categoryUids.get(v.baseInfo.categoryId)?.includes(v.uid!))
    }
  }

  /** uid => uin */
  async getBuddyIdMap(refresh = false): Promise<Map<string, string>> {
    const retMap: Map<string, string> = new Map()
    const session = getSession()
    if (session) {
      const uids: string[] = []
      const buddyService = session.getBuddyService()
      const buddyListV2 = await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL)
      uids.push(...buddyListV2.data.flatMap(item => item.buddyUids))
      const data = await session.getProfileService().getCoreAndBaseInfo('nodeStore', uids)
      for (const [, item] of data) {
        if (retMap.size > 5000) {
          break
        }
        retMap.set(item.uid!, item.uin!)
      }
    } else {
      const data = await invoke<{
        buddyCategory: CategoryFriend[]
        userSimpleInfos: Record<string, SimpleInfo>
      }>(
        'getBuddyList',
        [refresh],
        {
          className: NTClass.NODE_STORE_API,
          cbCmd: ReceiveCmdS.FRIENDS,
          afterFirstCmd: false,
        }
      )
      for (const item of Object.values(data.userSimpleInfos)) {
        if (retMap.size > 5000) {
          break
        }
        retMap.set(item.uid!, item.uin!)
      }
    }
    return retMap
  }

  async getBuddyV2ExWithCate(refresh = false) {
    const session = getSession()
    if (session) {
      const uids: string[] = []
      const categoryMap: Map<string, Dict> = new Map()
      const buddyService = session.getBuddyService()
      const buddyListV2 = (await buddyService.getBuddyListV2('0', BuddyListReqType.KNOMAL)).data
      uids.push(
        ...buddyListV2.flatMap(item => {
          item.buddyUids.forEach(uid => {
            categoryMap.set(uid, { categoryId: item.categoryId, categroyName: item.categroyName })
          })
          return item.buddyUids
        }))
      const data = await session.getProfileService().getCoreAndBaseInfo('nodeStore', uids)
      return Array.from(data).map(([key, value]) => {
        const category = categoryMap.get(key)
        return category ? { ...value, categoryId: category.categoryId, categroyName: category.categroyName } : value
      })
    } else {
      const data = await invoke<{
        buddyCategory: CategoryFriend[]
        userSimpleInfos: Record<string, SimpleInfo>
      }>(
        'getBuddyList',
        [refresh],
        {
          className: NTClass.NODE_STORE_API,
          cbCmd: ReceiveCmdS.FRIENDS,
          afterFirstCmd: false,
        }
      )
      const category: Map<number, Pick<CategoryFriend, 'buddyUids' | 'categroyName'>> = new Map()
      for (const item of data.buddyCategory) {
        category.set(item.categoryId, pick(item, ['buddyUids', 'categroyName']))
      }
      return Object.values(data.userSimpleInfos)
        .filter(v => v.baseInfo && category.get(v.baseInfo.categoryId)?.buddyUids.includes(v.uid!))
        .map(value => {
          return {
            ...value,
            categoryId: value.baseInfo.categoryId,
            categroyName: category.get(value.baseInfo.categoryId)?.categroyName
          }
        })
    }
  }

  async isBuddy(uid: string): Promise<boolean> {
    const session = getSession()
    if (session) {
      return session.getBuddyService().isBuddy(uid)
    } else {
      return await invoke('nodeIKernelBuddyService/isBuddy', [{ uid }, null])
    }
  }
}