# maa_utils.py

class MaaWorker:
    def __init__(self, queue: SimpleQueue, interface):
        # ... 原有初始化代码 ...
        self.send_log("MAA初始化成功")
        self.agent_process: subprocess.Popen | None = None
        self.load_agent()
        self.send_log("Agent加载完成")
        self.http_client = httpx.Client(timeout=30)

    # ========== 在此添加您的自定义方法 ==========

    def my_custom_task(self, param: str):
        """自定义任务示例"""
        self.send_log(f"开始执行自定义任务: {param}")

        # 获取当前屏幕截图
        image = self.controller.post_screencap().wait().get()

        # 执行 pipeline 任务
        result = self.tasker.post_task("MyCustomEntry").wait().get()

        # 处理结果
        if result.status.succeeded:
            self.send_log("任务执行成功")
        else:
            self.send_log("任务执行失败")

        return result.status.succeeded

    def detect_custom_objects(self) -> tuple[list, list]:
        """自定义识别示例 - 使用 YOLO 检测"""
        result = self.tasker.post_task("yolo_detect").wait().get()
        if result.status.failed:
            return [], []

        details = result.nodes[0].recognition.raw_detail["all"]
        boxes, labels = [], []
        for detail in details:
            boxes.append(detail["box"])
            labels.append(detail["label"])
        return boxes, labels

    # ========== 自定义方法结束 ==========

# maa_utils.py

from maa.custom_action import CustomAction
from maa.custom_recognition import CustomRecognition
from maa.define import TaskDetail
import numpy as np
from PIL import Image


# ========== 自定义 Recognition ==========
@resource.custom_recognition("MyCustomReco")
class MyCustomRecognition(CustomRecognition):
    def analyze(self, context, argv: CustomRecognition.AnalyzeArg):
        """自定义识别逻辑"""
        # 获取当前屏幕图像
        image = context.tasker.controller.post_screencap().wait().get()

        # 您的识别逻辑...
        # 返回识别结果
        return CustomRecognition.AnalyzeResult(
            box=(x, y, w, h),  # 识别框坐标
            detail="识别详情"
        )


# ========== 自定义 Action ==========
@resource.custom_action("MyCustomAction")
class MyCustomAction(CustomAction):
    def run(self, context, argv: CustomAction.RunArg):
        """自定义操作逻辑"""
        # 获取点击坐标
        box = argv.rec_box
        x, y = box[0] + box[2] // 2, box[1] + box[3] // 2

        # 执行点击
        context.tasker.controller.post_click(x, y).wait()

        return CustomAction.RunResult(success=True)

# maa_utils.py

def _run_process(self, task_list):
    """任务执行主流程 - 可在此扩展"""
    self.send_log("任务开始")
    try:
        for task in task_list:
            if self.stop_flag:
                self.tasker.post_stop().wait()
                self.send_log("任务已终止")
                return

            # 原有任务执行
            if task == "启动游戏":
                self._startup_game()
            elif task == "我的自定义任务":
                # 调用您的自定义方法
                self.my_custom_task("参数")
            else:
                # 默认 pipeline 任务
                t = self.tasker.post_task(task)
                self.send_log("正在运行任务: " + task)
                while not t.done:
                    time.sleep(0.5)
                    if self.stop_flag:
                        self.tasker.post_stop().wait()
                        self.send_log("任务已终止")
                        return

    except Exception:
        traceback.print_exc()
        self.send_notification("错误", "任务出现异常")
        self.send_log("任务出现异常，请检查终端日志")
    finally:
        self.running = False
        self._task_thread = None
        self.send_log("所有任务完成")


def _startup_game(self):
    """封装启动游戏逻辑"""
    self.send_log("正在启动游戏...")
    self.tasker.post_task("StartUp").wait()