NapCatQQ/src/onebot/action/extends/SetQQAvatar.ts

import { OneBotAction } from '@/onebot/action/OneBotAction';
import { ActionName, BaseCheckResult } from '@/onebot/action/router';
import * as fs from 'node:fs';
import { checkFileExist, uri2local } from '@/common/file';

interface Payload {
    file: string;
}

export default class SetAvatar extends OneBotAction<Payload, null> {
    actionName = ActionName.SetQQAvatar;

    // 用不着复杂检测
    protected async check(payload: Payload): Promise<BaseCheckResult> {
        if (!payload.file || typeof payload.file != 'string') {
            return {
                valid: false,
                message: 'file字段不能为空或者类型错误',
            };
        }
        return {
            valid: true,
        };
    }

    async _handle(payload: Payload): Promise<null> {
        const { path, success } = (await uri2local(this.core.NapCatTempPath, payload.file));
        if (!success) {
            throw new Error(`头像${payload.file}设置失败,file字段可能格式不正确`);
        }
        if (path) {
            await checkFileExist(path, 5000);// 避免崩溃
            const ret = await this.core.apis.UserApi.setQQAvatar(path);
            fs.unlink(path, () => {
            });

            if (!ret) {
                throw new Error(`头像${payload.file}设置失败,api无返回`);
            }
            // log(`头像设置返回：${JSON.stringify(ret)}`)
            if (ret.result as number == 1004022) {
                throw new Error(`头像${payload.file}设置失败，文件可能不是图片格式`);
            } else if (ret.result != 0) {
                throw new Error(`头像${payload.file}设置失败,未知的错误,${ret.result}:${ret.errMsg}`);
            }
        } else {
            fs.unlink(path, () => { });

            throw new Error(`头像${payload.file}设置失败,无法获取头像,文件可能不存在`);
        }
        return null;
    }
}