NapCatQQ/src/core/apis/webapi.ts

import { RequestUtil } from '@/common/request';
import {
    GroupEssenceMsgRet,
    InstanceContext,
    WebApiGroupMember,
    WebApiGroupMemberRet,
    WebApiGroupNoticeRet,
    WebHonorType,
} from '@/core';
import { NapCatCore } from '..';
import { createReadStream, readFileSync, statSync } from 'node:fs';
import { createHash } from 'node:crypto';
import { basename } from 'node:path';

export class NTQQWebApi {
    context: InstanceContext;
    core: NapCatCore;

    constructor(context: InstanceContext, core: NapCatCore) {
        this.context = context;
        this.core = core;
    }

    async shareDigest(groupCode: string, msgSeq: string, msgRandom: string, targetGroupCode: string) {
        const cookieObject = await this.core.apis.UserApi.getCookies('qun.qq.com');
        const url = `https://qun.qq.com/cgi-bin/group_digest/share_digest?${new URLSearchParams({
            bkn: this.getBknFromCookie(cookieObject),
            group_code: groupCode,
            msg_seq: msgSeq,
            msg_random: msgRandom,
            target_group_code: targetGroupCode,
        }).toString()}`;
        try {
            return RequestUtil.HttpGetText(url, 'GET', '', { 'Cookie': this.cookieToString(cookieObject) });
        } catch (e) {
            return undefined;
        }
    }
    async getGroupEssenceMsgAll(GroupCode: string) {
        const ret: GroupEssenceMsgRet[] = [];
        for (let i = 0; i < 20; i++) {
            const data = await this.getGroupEssenceMsg(GroupCode, i, 50);
            if (!data) break;
            ret.push(data);
            if (data.data.is_end) break;
        }
        return ret;
    }
    async getGroupEssenceMsg(GroupCode: string, page_start: number = 0, page_limit: number = 50) {
        const cookieObject = await this.core.apis.UserApi.getCookies('qun.qq.com');
        const url = `https://qun.qq.com/cgi-bin/group_digest/digest_list?${new URLSearchParams({
            bkn: this.getBknFromCookie(cookieObject),
            page_start: page_start.toString(),
            page_limit: page_limit.toString(),
            group_code: GroupCode,
        }).toString()}`;
        try {
            const ret = await RequestUtil.HttpGetJson<GroupEssenceMsgRet>(
                url,
                'GET',
                '',
                { 'Cookie': this.cookieToString(cookieObject) }
            );
            return ret.retcode === 0 ? ret : undefined;
        } catch {
            return undefined;
        }
    }

    async getGroupMembers(GroupCode: string, cached: boolean = true): Promise<WebApiGroupMember[]> {
        //logDebug('webapi 获取群成员', GroupCode);
        const memberData: Array<WebApiGroupMember> = new Array<WebApiGroupMember>();
        const cookieObject = await this.core.apis.UserApi.getCookies('qun.qq.com');
        const retList: Promise<WebApiGroupMemberRet>[] = [];
        const fastRet = await RequestUtil.HttpGetJson<WebApiGroupMemberRet>(
            `https://qun.qq.com/cgi-bin/qun_mgr/search_group_members?${new URLSearchParams({
                st: '0',
                end: '40',
                sort: '1',
                gc: GroupCode,
                bkn: this.getBknFromCookie(cookieObject),
            }).toString()}`,
            'POST',
            '',
            { 'Cookie': this.cookieToString(cookieObject) }
        );
        if (!fastRet?.count || fastRet?.errcode !== 0 || !fastRet?.mems) {
            return [];
        } else {
            for (const key in fastRet.mems) {
                memberData.push(fastRet.mems[key]);
            }
        }
        //初始化获取PageNum
        const PageNum = Math.ceil(fastRet.count / 40);
        //遍历批量请求
        for (let i = 2; i <= PageNum; i++) {
            const ret = RequestUtil.HttpGetJson<WebApiGroupMemberRet>(
                `https://qun.qq.com/cgi-bin/qun_mgr/search_group_members?${new URLSearchParams({
                    st: ((i - 1) * 40).toString(),
                    end: (i * 40).toString(),
                    sort: '1',
                    gc: GroupCode,
                    bkn: this.getBknFromCookie(cookieObject),
                }).toString()}`,
                'POST',
                '',
                { 'Cookie': this.cookieToString(cookieObject) }
            );
            retList.push(ret);
        }
        //批量等待
        for (let i = 1; i <= PageNum; i++) {
            const ret = await (retList[i]);
            if (!ret?.count || ret?.errcode !== 0 || !ret?.mems) {
                continue;
            }
            for (const key in ret.mems) {
                memberData.push(ret.mems[key]);
            }
        }
        return memberData;
    }

    // public  async addGroupDigest(groupCode: string, msgSeq: string) {
    //   const url = `https://qun.qq.com/cgi-bin/group_digest/cancel_digest?random=665&X-CROSS-ORIGIN=fetch&group_code=${groupCode}&msg_seq=${msgSeq}&msg_random=444021292`;
    //   const res = await this.request(url);
    //   return await res.json();
    // }

    // public async getGroupDigest(groupCode: string) {
    //   const url = `https://qun.qq.com/cgi-bin/group_digest/digest_list?random=665&X-CROSS-ORIGIN=fetch&group_code=${groupCode}&page_start=0&page_limit=20`;
    //   const res = await this.request(url);
    //   return await res.json();
    // }

    async setGroupNotice(
        GroupCode: string,
        Content: string,
        pinned: number = 0,
        type: number = 1,
        is_show_edit_card: number = 1,
        tip_window_type: number = 1,
        confirm_required: number = 1,
        picId: string = '',
        imgWidth: number = 540,
        imgHeight: number = 300,
    ) {
        interface SetNoticeRetSuccess {
            ec: number;
            em: string;
            id: number;
            ltsm: number;
            new_fid: string;
            read_only: number;
            role: number;
            srv_code: number;
        }

        const cookieObject = await this.core.apis.UserApi.getCookies('qun.qq.com');

        try {
            const settings = JSON.stringify({
                is_show_edit_card: is_show_edit_card,
                tip_window_type: tip_window_type,
                confirm_required: confirm_required
            });
            const externalParam = {
                pic: picId,
                imgWidth: imgWidth.toString(),
                imgHeight: imgHeight.toString(),
            };
            const ret: SetNoticeRetSuccess = await RequestUtil.HttpGetJson<SetNoticeRetSuccess>(
                `https://web.qun.qq.com/cgi-bin/announce/add_qun_notice?${new URLSearchParams({
                    bkn: this.getBknFromCookie(cookieObject),
                    qid: GroupCode,
                    text: Content,
                    pinned: pinned.toString(),
                    type: type.toString(),
                    settings: settings,
                    ...(picId === '' ? {} : externalParam)
                }).toString()}`,
                'POST',
                '',
                { 'Cookie': this.cookieToString(cookieObject) }
            );
            return ret;
        } catch (e) {
            return undefined;
        }
    }

    async getGroupNotice(GroupCode: string): Promise<undefined | WebApiGroupNoticeRet> {
        const cookieObject = await this.core.apis.UserApi.getCookies('qun.qq.com');
        try {
            const ret = await RequestUtil.HttpGetJson<WebApiGroupNoticeRet>(
                `https://web.qun.qq.com/cgi-bin/announce/get_t_list?${new URLSearchParams({
                    bkn: this.getBknFromCookie(cookieObject),
                    qid: GroupCode,
                    ft: '23',
                    ni: '1',
                    n: '1',
                    i: '1',
                    log_read: '1',
                    platform: '1',
                    s: '-1',
                }).toString()}&n=20`,
                'GET',
                '',
                { 'Cookie': this.cookieToString(cookieObject) }
            );
            return ret?.ec === 0 ? ret : undefined;
        } catch (e) {
            return undefined;
        }
    }

    private async getDataInternal(cookieObject: any, groupCode: string, type: number) {
        let resJson;
        try {
            const res = await RequestUtil.HttpGetText(
                `https://qun.qq.com/interactive/honorlist?${new URLSearchParams({
                    gc: groupCode,
                    type: type.toString(),
                }).toString()}`,
                'GET',
                '',
                { 'Cookie': this.cookieToString(cookieObject) }
            );
            const match = /window\.__INITIAL_STATE__=(.*?);/.exec(res);
            if (match) {
                resJson = JSON.parse(match[1].trim());
            }
            return type === 1 ? resJson?.talkativeList : resJson?.actorList;
        } catch (e) {
            this.context.logger.logDebug('获取当前群荣耀失败', e);
            return undefined;
        }
    }

    private async getHonorList(cookieObject: any, groupCode: string, type: number) {
        const data = await this.getDataInternal(cookieObject, groupCode, type);
        if (!data) {
            this.context.logger.logError(`获取类型 ${type} 的荣誉信息失败`);
            return [];
        }
        return data.map((item: any) => ({
            user_id: item?.uin,
            nickname: item?.name,
            avatar: item?.avatar,
            description: item?.desc,
        }));
    }

    async getGroupHonorInfo(groupCode: string, getType: WebHonorType) {
        const cookieObject = await this.core.apis.UserApi.getCookies('qun.qq.com');
        const HonorInfo: any = { group_id: groupCode };

        if (getType === WebHonorType.TALKATIVE || getType === WebHonorType.ALL) {
            const talkativeList = await this.getHonorList(cookieObject, groupCode, 1);
            if (talkativeList.length > 0) {
                HonorInfo.current_talkative = talkativeList[0];
                HonorInfo.talkative_list = talkativeList;
            }
        }

        if (getType === WebHonorType.PERFORMER || getType === WebHonorType.ALL) {
            HonorInfo.performer_list = await this.getHonorList(cookieObject, groupCode, 2);
        }

        if (getType === WebHonorType.LEGEND || getType === WebHonorType.ALL) {
            HonorInfo.legend_list = await this.getHonorList(cookieObject, groupCode, 3);
        }

        if (getType === WebHonorType.EMOTION || getType === WebHonorType.ALL) {
            HonorInfo.emotion_list = await this.getHonorList(cookieObject, groupCode, 6);
        }

        // 冒尖小春笋好像已经被tx扬了 R.I.P.
        if (getType === WebHonorType.EMOTION || getType === WebHonorType.ALL) {
            HonorInfo.strong_newbie_list = [];
        }

        return HonorInfo;
    }

    private cookieToString(cookieObject: any) {
        return Object.entries(cookieObject).map(([key, value]) => `${key}=${value}`).join('; ');
    }

    public getBknFromCookie(cookieObject: any) {
        const sKey = cookieObject.skey as string;

        let hash = 5381;
        for (let i = 0; i < sKey.length; i++) {
            const code = sKey.charCodeAt(i);
            hash = hash + (hash << 5) + code;
        }
        return (hash & 0x7FFFFFFF).toString();
    }
    public getBknFromSKey(sKey: string) {
        let hash = 5381;
        for (let i = 0; i < sKey.length; i++) {
            const code = sKey.charCodeAt(i);
            hash = hash + (hash << 5) + code;
        }
        return (hash & 0x7FFFFFFF).toString();
    }
    async createQunAlbumSession(gc: string, sAlbumID: string, sAlbumName: string, path: string, skey: string, pskey: string, uin: string) {
        const img = readFileSync(path);
        const img_md5 = createHash('md5').update(img).digest('hex');
        const img_size = img.length;
        const img_name = basename(path);
        const time = Math.floor(Date.now() / 1000);
        const GTK = this.getBknFromSKey(pskey);
        const cookie = `p_uin=${uin}; p_skey=${pskey}; skey=${skey}; uin=${uin}`;
        const body = {
            control_req: [{
                uin: uin,
                token: {
                    type: 4,
                    data: pskey,
                    appid: 5
                },
                appid: "qun",
                checksum: img_md5,
                check_type: 0,
                file_len: img_size,
                env: {
                    refer: "qzone",
                    deviceInfo: "h5"
                },
                model: 0,
                biz_req: {
                    sPicTitle: img_name,
                    sPicDesc: "",
                    sAlbumName: sAlbumName,
                    sAlbumID: sAlbumID,
                    iAlbumTypeID: 0,
                    iBitmap: 0,
                    iUploadType: 0,
                    iUpPicType: 0,
                    iBatchID: time,
                    sPicPath: "",
                    iPicWidth: 0,
                    iPicHight: 0,
                    iWaterType: 0,
                    iDistinctUse: 0,
                    iNeedFeeds: 1,
                    iUploadTime: time,
                    mapExt: {
                        appid: "qun",
                        userid: gc
                    }
                },
                session: "",
                asy_upload: 0,
                cmd: "FileUpload"
            }]
        };
        const api = `https://h5.qzone.qq.com/webapp/json/sliceUpload/FileBatchControl/${img_md5}?g_tk=${GTK}`;
        const post = await RequestUtil.HttpGetJson(api, 'POST', body, {
            "Cookie": cookie,
            "Content-Type": "application/json"
        });

        return post;
    }

    async uploadQunAlbumSlice(path: string, session: string, skey: string, pskey: string, uin: string, slice_size: number) {
        const img_size = statSync(path).size;
        const img_name = basename(path);
        let seq = 0;
        let offset = 0;
        const GTK = this.getBknFromSKey(pskey);
        const cookie = `p_uin=${uin}; p_skey=${pskey}; skey=${skey}; uin=${uin}`;

        const stream = createReadStream(path, { highWaterMark: slice_size });

        for await (const chunk of stream) {
            const end = Math.min(offset + chunk.length, img_size);
            const boundary = `----WebKitFormBoundary${Math.random().toString(36).substring(2)}`;
            const formData = await RequestUtil.createFormData(boundary, path);

            const api = `https://h5.qzone.qq.com/webapp/json/sliceUpload/FileUpload?seq=${seq}&retry=0&offset=${offset}&end=${end}&total=${img_size}&type=form&g_tk=${GTK}`;
            const body = {
                uin: uin,
                appid: "qun",
                session: session,
                offset: offset,
                data: formData,
                checksum: "",
                check_type: 0,
                retry: 0,
                seq: seq,
                end: end,
                cmd: "FileUpload",
                slice_size: slice_size,
                "biz_req.iUploadType": 0
            };

            const post = await RequestUtil.HttpGetJson(api, 'POST', body, {
                "Cookie": cookie,
                "Content-Type": `multipart/form-data; boundary=${boundary}`
            });

            offset += chunk.length;
            seq++;
        }
    }
    async uploadQunAlbum(path: string, albumId: string, group: string, skey: string, pskey: string, uin: string) {
        const session = (await this.createQunAlbumSession(group, albumId, group, path, skey, pskey, uin) as { data: { session: string } }).data.session;
        return await this.uploadQunAlbumSlice(path, session, skey, pskey, uin, 1024 * 1024);
    }
}