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

import BaseAction from '../BaseAction';
import { ActionName, BaseCheckResult } from '../types';
import * as fs from 'node:fs';
import { NTQQUserApi } from '@/core/apis/user';
import { checkFileReceived, uri2local } from '@/common/utils/file';

interface Payload {
  file: string
}

export default class SetAvatar extends BaseAction<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,
    };
  }
  protected async _handle(payload: Payload): Promise<null> {
    const { path, isLocal, errMsg, success } = (await uri2local(payload.file));
    if (!success) {
      throw `头像${payload.file}设置失败,file字段可能格式不正确`;
    }
    if (path) {
      await checkFileReceived(path, 5000); // 文件不存在QQ会崩溃，需要提前判断
      const ret = await NTQQUserApi.setQQAvatar(path);
      if (!isLocal) {
        fs.unlink(path, () => { });
      }
      if (!ret) {
        throw `头像${payload.file}设置失败,api无返回`;
      }
      // log(`头像设置返回：${JSON.stringify(ret)}`)
      if (ret['result'] == 1004022) {
        throw `头像${payload.file}设置失败，文件可能不是图片格式`;
      } else if (ret['result'] != 0) {
        throw `头像${payload.file}设置失败,未知的错误,${ret['result']}:${ret['errMsg']}`;
      }
    } else {
      if (!isLocal) {
        fs.unlink(path, () => { });
      }
      throw `头像${payload.file}设置失败,无法获取头像,文件可能不存在`;
    }
    return null;
  }
}