feat: action check data

src/onebot11/action/extends/SetOnlineStatus.ts

@@ -2,17 +2,27 @@ import { OB11User } from '../../types';
 import { OB11Constructor } from '../../constructor';
 import { friends } from '@/core/data';
 import BaseAction from '../BaseAction';
-import { ActionName } from '../types';
+import { ActionName, BaseCheckResult } from '../types';
 import { NTQQUserApi } from '@/core/apis';
+import { FromSchema, JSONSchema } from 'json-schema-to-ts';
+import Ajv from "ajv"
 // 设置在线状态
-interface Payload {
-    status: number;
-    extStatus: number;
-    batteryStatus: number;
-}
+
+const SchemaData = {
+  type: 'object',
+  properties: {
+    status: { type: 'number' },
+    extStatus: { type: 'number' },
+    batteryStatus: { type: 'number' }
+  },
+  required: ['status', 'extStatus', 'batteryStatus'],
+} as const satisfies JSONSchema;
+
+type Payload = FromSchema<typeof SchemaData>;
+
 export class SetOnlineStatus extends BaseAction<Payload, null> {
   actionName = ActionName.SetOnlineStatus;
-
+  PayloadSchema = SchemaData;
   protected async _handle(payload: Payload) {
     // 可设置状态
     // { status: 10, extStatus: 1027, batteryStatus: 0 }

src/onebot11/action/extends/SetQQAvatar.ts

@@ -1,27 +1,38 @@
 import BaseAction from '../BaseAction';
-import { ActionName } from '../types';
+import { ActionName, BaseCheckResult } from '../types';
 import * as fs from 'node:fs';
 import { NTQQUserApi } from '@/core/apis/user';
 import { checkFileReceived, uri2local } from '@/common/utils/file';
 // import { log } from "../../../common/utils";
 
 interface Payload {
-    file: string
+  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 } = (await uri2local(payload.file));
-    if (errMsg){
+    if (errMsg) {
       throw `头像${payload.file}设置失败,file字段可能格式不正确`;
     }
     if (path) {
       await checkFileReceived(path, 5000); // 文件不存在QQ会崩溃，需要提前判断
       const ret = await NTQQUserApi.setQQAvatar(path);
-      if (!isLocal){
-        fs.unlink(path, () => {});
+      if (!isLocal) {
+        fs.unlink(path, () => { });
       }
       if (!ret) {
         throw `头像${payload.file}设置失败,api无返回`;
@@ -29,15 +40,15 @@ export default class SetAvatar extends BaseAction<Payload, null> {
       // log(`头像设置返回：${JSON.stringify(ret)}`)
       if (ret['result'] == 1004022) {
         throw `头像${payload.file}设置失败，文件可能不是图片格式`;
-      } else if(ret['result'] != 0) {
+      } else if (ret['result'] != 0) {
         throw `头像${payload.file}设置失败,未知的错误,${ret['result']}:${ret['errMsg']}`;
       }
     } else {
-      if (!isLocal){
-        fs.unlink(path, () => {});
+      if (!isLocal) {
+        fs.unlink(path, () => { });
       }
       throw `头像${payload.file}设置失败,无法获取头像,文件可能不存在`;
-    } 
+    }
     return null;
   }
 }