from pathlib import Path

path=Path('C:/wechat_rpa/protocol_engine.py')
text=path.read_text(encoding='utf-8')
def replace(old,new,count=1):
    global text
    assert text.count(old)==count,(str(path),old[:90],text.count(old),count)
    text=text.replace(old,new)
replace("self._since=time.time();self._pending={};self._seen={};self._contact_blocked={}", "self._since=time.time();self._pending={};self._seen={};self._contact_blocked={};self._manual_handoffs={}")
replace("            self._pending={k:v for k,v in raw.get('pending',{}).items() if isinstance(v,dict)}", """            self._pending={k:v for k,v in raw.get('pending',{}).items() if isinstance(v,dict)}
            self._manual_handoffs=raw.get('manual_handoffs',{})
            if not isinstance(self._manual_handoffs,dict):raise NativeUnavailable('人工接管记录格式错误，已停止启动')""")
replace("'contactBlocked':self._contact_blocked},f", "'contactBlocked':self._contact_blocked,'manual_handoffs':self._manual_handoffs},f")
replace("        removed=[]\n        with self._lock:\n            for key in dict.fromkeys", """        removed=[]
        with self._lock:
            previous_pending=dict(self._pending);previous_seen=dict(self._seen)
            previous_handoffs=dict(self._manual_handoffs);previous_attention=self.needs_attention
            for key in dict.fromkeys""")
replace("                if state.get('dedup_key'):self._seen[key]=state['dedup_key']\n                self._pending.pop(key,None)", """                snapshot=json.loads(json.dumps(state,ensure_ascii=False))
                self._manual_handoffs[key]={'state':snapshot,'handed_off_at':time.time(),
                                            'receipt_checked_by_user':bool(confirm_uncertain),
                                            'delivery_confirmed':False}
                if state.get('dedup_key'):self._seen[key]=state['dedup_key']
                self._pending.pop(key,None)""")
replace("                                     for s in self._pending.values())\n            self._save()\n            self._reply_wakeup.set()", """                                     for s in self._pending.values())
            try:self._save()
            except Exception:
                self._pending=previous_pending;self._seen=previous_seen
                self._manual_handoffs=previous_handoffs;self.needs_attention=previous_attention
                raise
            self._reply_wakeup.set()""")
compile(text,str(path),'exec')
path.write_text(text,encoding='utf-8')

for root in (Path('C:/wechat_rpa'),Path('C:/kefu/wechat_rpa')):
    path=root/'gui_runtime.py'
    text=path.read_text(encoding='utf-8')
    replace('    pending = raw["pending"] if isinstance(raw.get("pending"), dict) else raw\n    removed = []', '''    pending = raw["pending"] if isinstance(raw.get("pending"), dict) else raw
    if handoff:
        history = raw.get("manual_handoffs", {})
        if not isinstance(history, dict):
            result["error"] = "人工接管记录格式不正确，无法安全转交人工"
            return result
        if pending is raw:
            pending = {key: state for key, state in raw.items() if key != "manual_handoffs"}
            raw = {"pending": pending, "manual_handoffs": history}
        else:
            raw["manual_handoffs"] = history
    removed = []''')
    replace('        if handoff and protocol and state.get("dedup_key"):\n            raw.setdefault("seen", {})[key] = state["dedup_key"]\n        pending.pop(key, None)', '''        if handoff and protocol and state.get("dedup_key"):
            raw.setdefault("seen", {})[key] = state["dedup_key"]
        if handoff:
            raw["manual_handoffs"][key] = {
                "state": dict(state), "handed_off_at": time.time(),
                "receipt_checked_by_user": bool(confirm_uncertain), "delivery_confirmed": False,
            }
        pending.pop(key, None)''')
    compile(text,str(path),'exec')
    path.write_text(text,encoding='utf-8')
print('Preserving full handoff evidence in the same atomically replaced queue file.')