LLOneBot/src/onebot11/action/BaseAction.ts

import {ActionName, BaseCheckResult} from "./types"
import {OB11Response} from "./utils"
import {OB11Return} from "../types";
import {log} from "../../common/utils";

class BaseAction<PayloadType, ReturnDataType> {
    actionName: ActionName

    protected async check(payload: PayloadType): Promise<BaseCheckResult> {
        return {
            valid: true,
        }
    }

    public async handle(payload: PayloadType): Promise<OB11Return<ReturnDataType | null>> {
        const result = await this.check(payload);
        if (!result.valid) {
            return OB11Response.error(result.message, 400);
        }
        try {
            const resData = await this._handle(payload);
            return OB11Response.ok(resData);
        } catch (e) {
            log("发生错误", e)
            return OB11Response.error(e?.toString() || e?.stack?.toString() || "未知错误，可能操作超时", 200);
        }
    }

    public async websocketHandle(payload: PayloadType, echo: any): Promise<OB11Return<ReturnDataType | null>> {
        const result = await this.check(payload)
        if (!result.valid) {
            return OB11Response.error(result.message, 1400)
        }
        try {
            const resData = await this._handle(payload)
            return OB11Response.ok(resData, echo);
        } catch (e) {
            log("发生错误", e)
            return OB11Response.error(e.stack?.toString() || e.toString(), 1200, echo)
        }
    }

    protected async _handle(payload: PayloadType): Promise<ReturnDataType> {
        throw `pleas override ${this.actionName} _handle`;
    }
}

export default BaseAction