"""统一收尾能力不足、资源缺失和动作修正失败"""

from app.processor.base import BaseNode
from app.processor.state import QueryState


class FallbackNode(BaseNode):
    name = "fallback_node"

    def process(self, state: QueryState) -> QueryState:
        # 浅拷贝状态
        updated_state: QueryState = state.copy()
        # 读取上游记录的具体失败原因
        reason = state.get("failure_reason")

        # 如果上游没有提供有效原因，则根据校验错误补充失败说明
        if not reason:
            # 如果仍有校验错误，使用达到自动修正上限的说明，并带上已重试次数
            if state.get("validation_errors"):
                reason = f"动作校验未通过，已达到自动修正上限，共重试 {state.get('retry_count', 0)} 次"
            else:
                # 如果既没有具体原因也没有校验错误，使用通用说明，避免返回空原因
                reason = "当前需求未能完成，缺少具体失败原因"

        # 将任务标记为最终失败，结束本轮处理
        updated_state["status"] = "failed"
        # 保存上游原因或本节点补充的说明，供后续输出给用户
        updated_state["failure_reason"] = reason
        # 显式标记校验未通过，防止残留动作被当作可执行结果
        updated_state["validation_passed"] = False
        # 清空旧模拟快照，避免失败结果中仍携带之前的成功快照
        updated_state["simulated_screen"] = None

        # 返回统一的失败状态，保留原动作和错误信息供排查
        return updated_state


if __name__ == "__main__":
    from app.schemas.action_validation import ActionValidationError

    state = {
        "request": {},
        "retry_count": 2,
        "actions": None,
        "validation_errors": [
            ActionValidationError(code="generation_error", message="动作生成失败")
        ],
    }
    node = FallbackNode()
    final_state = node.process(state)
    print(final_state)