import BaseAction from '../BaseAction'; import { ActionName, BaseCheckResult } from '../types'; import * as fs from 'node:fs'; import { checkFileReceived, uri2local } from '@/common/file'; interface Payload { file: string; } export default class SetAvatar extends BaseAction { actionName = ActionName.SetQQAvatar; // 用不着复杂检测 protected async check(payload: Payload): Promise { if (!payload.file || typeof payload.file != 'string') { return { valid: false, message: 'file字段不能为空或者类型错误', }; } return { valid: true, }; } async _handle(payload: Payload): Promise { const { path, success } = (await uri2local(this.core.NapCatTempPath, payload.file)); if (!success) { throw `头像${payload.file}设置失败,file字段可能格式不正确`; } if (path) { await checkFileReceived(path, 5000); // 文件不存在QQ会崩溃,需要提前判断 const ret = await this.core.apis.UserApi.setQQAvatar(path); fs.unlink(path, () => { }); if (!ret) { throw `头像${payload.file}设置失败,api无返回`; } // log(`头像设置返回:${JSON.stringify(ret)}`) if (ret.result as number == 1004022) { throw `头像${payload.file}设置失败,文件可能不是图片格式`; } else if (ret.result != 0) { throw `头像${payload.file}设置失败,未知的错误,${ret.result}:${ret.errMsg}`; } } else { fs.unlink(path, () => { }); throw `头像${payload.file}设置失败,无法获取头像,文件可能不存在`; } return null; } }