"""Apply backed-up Qt bridge fixes; leave real pending queues untouched."""
import ast
from pathlib import Path
import shutil
import os

base = Path(__file__).parent
def replace_method(source, class_name, method_name, replacement):
    tree = ast.parse(source)
    cls = next(n for n in tree.body if isinstance(n, ast.ClassDef) and n.name == class_name)
    node = next(n for n in cls.body if isinstance(n, ast.FunctionDef) and n.name == method_name)
    lines = source.splitlines(keepends=True)
    return ''.join(lines[:node.lineno-1]) + replacement.rstrip() + '\n' + ''.join(lines[node.end_lineno:])

FILTER='''    def _apply_task_filter(self) -> None:
        if not hasattr(self, "task_list"):
            return
        mode = getattr(self, "active_task_filter", "全部")
        for row in range(self.task_list.count()):
            item = self.task_list.item(row)
            display = self._display_task_status(str(item.data(Qt.UserRole + 1) or ""))
            item.setHidden(mode != "全部" and display != mode)
        current = self.task_list.currentRow()
        selected = self.task_list.item(current) if current >= 0 else None
        if selected is None or selected.isHidden():
            visible = next((row for row in range(self.task_list.count()) if not self.task_list.item(row).isHidden()), -1)
            self.task_list.setCurrentRow(visible)
            if visible >= 0:
                self._select_visible_task(visible)
'''
DISPLAY='''    def _display_task_status(state: str) -> str:
        if "审核" in state:
            return "待审核"
        if "人工" in state or "待核对" in state or "处理失败" in state:
            return "需人工"
        if state.startswith("正在"):
            return "处理中"
        if "重试" in state:
            return "待重试"
        return "等待中"
'''
STATE='''    def _state_text(state: dict) -> str:
        review = task_review_state(state)
        send_state = str(state.get("send_state") or "")
        if send_state == "sending":
            return "正在发送"
        if review["deliveryUncertain"]:
            return "发送待核对（需人工）"
        if send_state == "sent_uncommitted":
            return "已发送，待归档"
        if review["needsHuman"]:
            return "需要人工处理"
        if review["awaitingReview"]:
            return "等待人工审核"
        stage = str(state.get("stage") or "")
        if stage in QueuePage.STAGE_LABELS:
            return QueuePage.STAGE_LABELS[stage]
        if state.get("batch_ready"):
            return "正在生成回复"
        if state.get("confirmed_unread"):
            return "等待处理"
        return "排队中"
'''

for root in (Path('C:/kefu/wechat_rpa'),Path('C:/wechat_rpa')):
    path=root/'wechat_gui_qt.py'
    source=path.read_text(encoding='utf-8')
    backup=root/'backups/review-visibility-20260916/qt'
    backup.mkdir(parents=True,exist_ok=True)
    if (backup/path.name).exists():
        assert path.read_bytes() == (backup/path.name).read_bytes(), 'Source changed since backup'
    else:
        shutil.copy2(path,backup/path.name)
    shutil.copy2(base/'review_state.py',root/'review_state.py')
    source=source.replace('import json\n','import json\nfrom review_state import task_review_state\n',1)
    source=source.replace('"manual_review": "等待人工审核",','"manual_review": "等待人工审核",\n        "awaiting_review": "等待人工审核",',1)
    source=replace_method(source,'QueuePage','_display_task_status',DISPLAY)
    source=replace_method(source,'QueuePage','_state_text',STATE)
    source=replace_method(source,'QueuePage','_apply_task_filter',FILTER)
    source=source.replace('("全部", "处理中", "等待中", "待重试")','("全部", "待审核", "需人工", "处理中", "等待中", "待重试")')
    source=source.replace('self.host.queue_page._set_task_filter(name or "全部")\n        self.host._push_console_state()',
        'self.host.queue_page._set_task_filter(name if name in {"全部", "待审核", "需人工", "处理中", "等待中", "待重试"} else "全部")\n        self.host._console_selected_key = ""\n        self.host._push_console_state(force=True)',1)
    # Actual raw state, not a localized label, controls allowed buttons.
    needle='"awaitingReview": bool(state.get("awaiting_review", False)),'
    assert needle in source
    source=source.replace(needle,'**task_review_state(state),\n            "isProtocol": bool(state.get("detected_by") == "protocol_db" or state.get("conv_id")),',1)
    source=source.replace('"reviewReason": str(state.get("review_reason") or ""),','"reviewReason": task_review_state(state)["reviewReason"],',1)
    source=source.replace('state.get("staged_user_text") or table_message or ""','state.get("staged_user_text") or state.get("chat_text") or table_message or ""',1)
    # Counts must include hidden rows, otherwise selecting another filter hides the alert.
    needle='        processing = sum(1 for item in people if item.get("status") == "处理中")'
    assert needle in source
    source=source.replace(needle,'        review_states = [task_review_state(value) for value in (getattr(self.queue_page, "pending_states", {}) or {}).values()]\n'+needle,1)
    source=source.replace('"processing": str(processing),','"processing": str(processing),\n            "review": str(sum(value["awaitingReview"] and not value["needsHuman"] for value in review_states)),\n            "human": str(sum(value["needsHuman"] for value in review_states)),',1)
    # Immediately re-read persisted state after approval rather than showing stale controls.
    start=source.index('    def _approve_queue_tasks(')
    end=source.index('    def _retry_queue_tasks(',start)
    part=source[start:end]
    part=part.replace('        self._schedule_console_push(urgent=True)','        self.queue_page.refresh_data()\n        self._schedule_console_push(urgent=True)')
    part=part.replace('        error = str(result.get("error") or "")','        if result.get("protected"):\n            QMessageBox.information(self, "任务受保护", "当前账号不匹配或任务正在发送/核对回执，不能批准重发。请先核对后转人工处理。")\n        error = str(result.get("error") or "")',1)
    # Protocol startup may be async. Explicit approval is queued until startup succeeds.
    needle='''        if thread is None or not thread.is_alive():
            QMessageBox.warning('''
    if 'if transport == PROTOCOL:' in source:
        part=part.replace(needle,'''        if (thread is None or not thread.is_alive()) and getattr(self, "_protocol_start_pending", False):
            self._deferred_review_approvals = sorted(set(getattr(self, "_deferred_review_approvals", [])) | set(requested))
            self.append_log("已登记审核操作，协议环境检查通过后将核对会话并继续。", "notify")
            return
'''+needle,1)
        source=source.replace('        self._thread.start()\n', '        self._thread.start()\n        deferred = list(getattr(self, "_deferred_review_approvals", []))\n        self._deferred_review_approvals = []\n        if deferred:\n            QTimer.singleShot(0, lambda keys=deferred: self._approve_queue_tasks(keys))\n',1)
        source=source.replace('    def stop_monitoring(self) -> None:\n','    def stop_monitoring(self) -> None:\n        self._deferred_review_approvals = []\n',1)
        # Re-find after the prior source insertion.
        start=source.index('    def _approve_queue_tasks(');end=source.index('    def _retry_queue_tasks(',start)
    source=source[:start]+part+source[end:]
    compile(source,str(path),'exec')
    temporary = path.with_name(path.name + '.review-patch.tmp')
    temporary.write_text(source,encoding='utf-8')
    os.replace(temporary,path)
    print(root)