const { Keyboard } = require('leanpro.common');
const { QtAuto } = require("leanpro.qt");
const { SmartMonkey } = require("leanpro.monkey");

async function smartTest() {
    Keyboard.disableIme(); // 禁用输入法防止干扰

    const monkey = new SmartMonkey({
        launchApp: () => { return QtAuto.launchQtProcessAsync("C:/path/to/your/app.exe"); },
        appName: "YourAppName",
        maxActions: 50,            // 操作次数
        maxDuration: 3 * 60,       // 最长持续时间，单位为秒
        code: true, // 生成代码
        tech: 'qt', // qt 或 win
        excludes: ["Bold", "Italic", "Underline", {
            "type": "MenuBar",
            "className": "QMenuBar"
        }], // 被排除控件列表。设置后，该控件及其下的所有子控件都会被排除
        verbose: true
    });

    // 可以在此处自定义控件筛选逻辑
    monkey.onControl(async (control, interested) => {
        if (!interested) return;
        // console.log('正在探索控件:', await control.modelProperties());
    });

    // 定义压力测试的起始视图/容器
    let view = QtAuto.getGeneric([...]); 
    
    // 启动测试，仅探索该容器内的控件树
    await monkey.start(view, {onlySubtree: true});

    // 生成报告
    await monkey.generateReport();
}

smartTest();

export interface SmartMonkeyOptions {
    launchApp?: () => Promise<void>; // 启动应用的回调函数
    appName: string;    // 应用名称
    verbose?: boolean;  // 详细日志输出
    slowMo?: number;    // 动作执行减速（毫秒）
    maxActions?: number;    // 最大动作数量
    maxDuration?: number;    // 最大执行时长（毫秒）
    code?: boolean;    // 是否生成代码
    overwrite?: boolean;    // 是否覆盖已有文件
    excludes?: any[];    // 排除的元素列表（支持 text 字符串或过滤属性对象，设置后其下的子控件也会被排除）
    tech?: 'qt' | 'win'; // 应用技术类型，支持 "qt" (默认) 或 "win"
}

// 进程状态枚举
export enum ProcessState {
    skip = 'skip',  // 跳过
    close = 'close',// 关闭窗口，仅对 Window 类型的控件有效
    stop = 'stop',  // 停止
    act = 'act'     // 执行动作
}

// 动作执行结果接口
export interface IActionResult {
    name: string    // 动作名称
    params: any[],  // 动作参数列表
    error?: string  // 动作错误信息
}

// SmartMonkey 事件枚举
export enum MonkeyEvent {
    start = 'start',    // 开始事件
    end = 'end',        // 结束事件
    startIteration = 'startIteration',  // 开始迭代
    endIteration = 'endIteration',      // 结束迭代
    startDialog = 'startDialog',        // 开始对话框
    endDialog = 'endDialog',            // 结束对话框
    launchApp = 'launchApp',            // 启动应用
}

// SmartMonkey 主类
export class SmartMonkey {
    constructor(options: SmartMonkeyOptions);
    // 开始执行自动化测试。支持 options.onlySubtree 参数，设置为 true 后，仅探索该控件的子树区域。
    start(startControl?: any, options?: { onlySubtree?: boolean }): Promise<void>;
    // 获取感兴趣的控件类型
    get interestedControls(): QtControlType;
    // 为每个控件触发回调，ProcessState 可以控制如何处理该控件
    onControl: (listener: (control: IQtControl, interested: boolean) => Promise<void | ProcessState>) => () => void;
    // 事件回调，可以是 start, end, startDialog, endDialog 等事件
    onEvent: (listener: (eventName: MonkeyEvent, ...params: any[]) => Promise<void>) => () => void;
    // 在执行每个动作后触发回调
    onAction: (listener: (control: IQtControl, action: IActionResult) => void) => () => void;
    // 生成测试报告
    generateReport(): Promise<void>;
}

const monkey = new SmartMonkey({
    appName: "diagramscene",
    maxActions: 50
});

let view = await QtAuto.getApplication({ "appName": "diagramscene" })
    .getWindow({ "windowTitle": "Diagramscene" })
    .getGeneric({ "className": "QToolBox" });

// 仅在 QToolBox 控件内部进行压力探索
await monkey.start(view, { onlySubtree: true });

const controls = monkey.interestedControls;
console.log("感兴趣的控件类型有:", controls);

// 排除所有带有特定文字（如“删除”）的按钮
const unsubscribe = monkey.onControl(async (control, interested) => {
    if (interested) {
        let name = await control.name();
        if (name.includes("删除")) {
            return "skip"; // 动态跳过
        }
    }
});

monkey.onEvent(async (eventName, ...params) => {
    if (eventName === 'start') {
        console.log("压力测试已经正式启动。");
    }
});

monkey.onAction((control, action) => {
    if (action.error) {
        console.warn(`动作 ${action.name} 执行失败，原因: ${action.error}`);
    }
});

await monkey.generateReport();