#!/usr/bin/env python3 """环节 1→5 编排器 —— 把整条自动回复链串起来 消费 watcher 的 incoming.jsonl（cursor 增量）→ 对候选事件（私聊/群@我/群引用我）： env2 查上下文（发送者身份 + 最近历史）→ env3 hermes -z 生成草稿 → env4 policy.decide()（白名单读 wechat-viewer 的 circle.db）→ env5 落地：auto 直接 send_text_via_hook + 入队(auto_sent) / review 入队(pending，在 /approvals 等人工) 作为 root 在 Termux 跑（同 watcher 上下文）。由 watcher 每次读到新消息后触发 --drain。 安全：policy 默认 auto_send_enabled=False → 一切进 review（不自动发）。生成超时/失败 → 无草稿 → policy skip → 不发（失败即静默，绝不误发）。 ponytail: 每次 drain 拷一次整库(~96MB)供该批所有事件查上下文；事件稀疏可接受。 升级：与 watcher 共享一次拷贝；incoming.jsonl 增长需按 offset 游标+轮转。 """ import json import os import sqlite3 import subprocess import sys import time _HERE = os.path.dirname(os.path.abspath(__file__)) sys.path.insert(0, _HERE) import wechat_reply_watcher as w # 复用解密/查询/账号推导 import wechat_reply_policy as policy WORK_DIR = w.WORK_DIR QUEUE_FILE = f"{WORK_DIR}/incoming.jsonl" CURSOR_FILE = f"{WORK_DIR}/orch_state.json" ORCH_LOG = f"{WORK_DIR}/orchestrator.log" CIRCLE_DB = "/data/data/com.termux/files/home/wechat-viewer/circle.db" EVENT_BUS = "/data/data/com.termux/files/home/.hermes/event_bus" HERMES = "/data/data/com.termux/files/usr/bin/hermes" CANDIDATE_KINDS = {"private", "group_at_me", "group_quote_me"} HISTORY_N = 12 GEN_TIMEOUT = 160 def log(msg): line = f"{time.strftime('%Y-%m-%d %H:%M:%S')} {msg}" print(line, flush=True) try: with open(ORCH_LOG, "a") as f: f.write(line + "\n") except OSError: pass # ── env2：查上下文 ───────────────────────────────────── def _display_name(stage, key, wxid): if not wxid: return "" rows = w.sqlcipher_query( stage, key, f"SELECT conRemark, nickname, alias FROM rcontact WHERE username='{wxid}';") if rows and rows[0]: r = rows[0] + ["", "", ""] return r[0] or r[1] or r[2] or wxid return wxid def _placeholder(mtype, text): if mtype == 1: return text return {3: "[图片]", 34: "[语音]", 43: "[视频]", 47: "[表情]", 42: "[名片]", 49: "[链接/文件]", 48: "[位置]"}.get(mtype, text[:40] if text else "[消息]") def build_context(stage, key, event, self_wxid): talker = event.get("talker", "") is_group = event.get("isGroup") or talker.endswith("@chatroom") sender_wxid = event.get("sender") or talker sender_disp = _display_name(stage, key, sender_wxid) lines = [f"对方：{sender_disp}"] if is_group: gname = _display_name(stage, key, talker) lines.insert(0, f"[群] {gname}") # 最近历史 rows = w.sqlcipher_query( stage, key, "SELECT isSend, type, createTime, hex(content) FROM message " f"WHERE talker='{talker}' ORDER BY msgId DESC LIMIT {HISTORY_N};") hist = [] for r in reversed(rows): if len(r) < 4: continue is_send, mtype, _, chex = r try: raw = bytes.fromhex(chex).decode("utf-8", "ignore") except ValueError: raw = "" if is_send == "1": who, body = "我", raw else: s, body = w.parse_group_content(raw) if is_group else (sender_wxid, raw) who = _display_name(stage, key, s) if is_group and s else sender_disp txt = _placeholder(int(mtype) if mtype.isdigit() else 0, body).replace("\n", " ").strip() if txt: hist.append(f" {who}: {txt}") ctx = "\n".join(lines) if hist: ctx += "\n最近对话：\n" + "\n".join(hist[-HISTORY_N:]) ctx += f"\n本条：{sender_disp}: {(event.get('body') or '').strip()[:200]}" return ctx, sender_disp # ── env3：hermes 生成 ───────────────────────────────── def generate(context_text): prompt = ( "【任务】下面是别人发到微信来的一条消息，请以你本人（火星人）的口吻，写一条你会真的发出去的" "微信回复。只输出这条回复本身，不要解释、不要引号、不要用任何工具、不要真的发送。\n\n" "【上下文】\n" + context_text) env = dict(os.environ) env.update({ "HOME": "/data/data/com.termux/files/home", "PREFIX": "/data/data/com.termux/files/usr", "PATH": "/data/data/com.termux/files/usr/bin:/system/bin", "LD_LIBRARY_PATH": "/data/data/com.termux/files/usr/lib", }) try: p = subprocess.run([HERMES, "-z", prompt], capture_output=True, text=True, timeout=GEN_TIMEOUT, env=env) out = (p.stdout or "").strip() # 取最后一段非空文本（躲开可能的日志行） return out.split("\n")[-1].strip() if out else "" except subprocess.TimeoutExpired: log("生成超时") return "" except Exception as e: # noqa: BLE001 log(f"生成异常: {e}") return "" # ── env4→5：落地 ───────────────────────────────────── def _load_whitelist(): contacts, groups = [], [] try: con = sqlite3.connect(CIRCLE_DB) for kind, wxid in con.execute("SELECT kind, wxid FROM reply_whitelist"): (groups if kind == "group" else contacts).append(wxid) con.close() except Exception: pass return contacts, groups def _queue_insert(event, draft, sender_disp, action, reason, status): con = sqlite3.connect(CIRCLE_DB) con.execute( """INSERT INTO reply_queue (msg_id, talker, sender, sender_name, kind, incoming, draft, action, reason, status, sent_at) VALUES (?,?,?,?,?,?,?,?,?,?, CASE WHEN ?='auto_sent' THEN datetime('now','localtime') ELSE NULL END)""", (event.get("msgId"), event.get("talker"), event.get("sender"), sender_disp, event.get("kind"), event.get("body"), draft, action, reason, status, status)) con.commit() con.close() VERIFY_TIMEOUT = 25 # 轮询确认送达的总超时(秒) VERIFY_POLL = 3 # 轮询间隔 def _cleanup_verify(): for p in __import__("glob").glob(f"{WORK_DIR}/_stage_verify/en.db*"): try: os.remove(p) except OSError: pass def _sent_status(db, key, talker, base_msgid, text): """查我方(isSend=1)在 base 之后、内容==text 的那条消息的 status。 微信 status: 1=发送中(转圈) / 2=已发送 / 5=失败。查不到返回 None。""" st = w.copy_db(db, "_stage_verify") try: rows = w.sqlcipher_query( st, key, f"SELECT status, hex(content) FROM message WHERE talker='{talker}' " f"AND isSend=1 AND msgId>{base_msgid} ORDER BY msgId DESC LIMIT 8;") finally: _cleanup_verify() for r in rows: if len(r) < 2: continue try: body = bytes.fromhex(r[1]).decode("utf-8", "ignore") except ValueError: body = "" if body == text: return int(r[0]) if str(r[0]).lstrip("-").isdigit() else None return None def _send_text_verified(to_wxid, text, key): """可靠发送：切前台唤醒微信 → 广播发送 → 轮询 status 直到 =2(确认送达)。 超时仍卡在发送中/失败 → 返回 ok=False,由调用方标 failed 进审核(不假装成功)。 ponytail: 校验靠反复拷库(~96MB/次)读 status；发送稀疏可接受,首轮多半即达。 上限=拷库 I/O;升级=让 hook 直接回报送达 / 常驻前台。""" _, db, _ = w.account_paths() st = w.copy_db(db, "_stage_verify") try: r = w.sqlcipher_query( st, key, f"SELECT COALESCE(MAX(msgId),0) FROM message WHERE talker='{to_wxid}';") base = int(r[0][0]) if r and r[0] else 0 finally: _cleanup_verify() # 微信在后台被 Android 冻结→NetScene 队列不提交,消息卡"发送中"。先切前台唤醒。 try: subprocess.run(["am", "start", "-n", "com.tencent.mm/.ui.LauncherUI"], capture_output=True, timeout=8) time.sleep(3) except Exception: # noqa: BLE001 pass sys.path.insert(0, EVENT_BUS) from wechat_hook_send import send_text_via_hook res = send_text_via_hook(to_wxid, text) if res.get("status") != "sent": return {"ok": False, "reason": "broadcast失败:" + str(res.get("error", ""))[:40]} # 关键机制:z2.b 只把消息入队(status=1 转圈),WeChat 发送队列要一次 onResume # (后台→前台切换)才真正派发。真机验证:发送后 HOME 切后台再拉回前台，卡住的消息 # 全部 flush 到 status=2。（等同"重启微信才发出去",但轻得多。） time.sleep(2) # 让消息先入库 try: subprocess.run(["input", "keyevent", "KEYCODE_HOME"], capture_output=True, timeout=6) time.sleep(1.5) subprocess.run(["am", "start", "-n", "com.tencent.mm/.ui.LauncherUI"], capture_output=True, timeout=8) time.sleep(2) except Exception: # noqa: BLE001 pass deadline = time.time() + VERIFY_TIMEOUT last = None while time.time() < deadline: time.sleep(VERIFY_POLL) last = _sent_status(db, key, to_wxid, base, text) if last == 2: return {"ok": True, "reason": "verified(status=2)"} if last == 5: return {"ok": False, "reason": "status=5(微信报失败)"} # 超时判定：真机验证 status=1 是"已入队发送中"，微信连接慢时会延迟但最终送达 # (曾观察到十几条 status=1 在微信重连后全部 flush 到 2)→ 视为已发,不误报失败。 # 仅 status=None(消息压根没创建,hook 没处理/微信没就绪)才算真失败 → 进审核。 if last == 1: return {"ok": True, "reason": "queued(status=1,已入队将送达)"} return {"ok": False, "reason": f"消息未创建(status={last})"} def process_event(event, stage, key, self_wxid, cfg): ctx, sender_disp = build_context(stage, key, event, self_wxid) draft = generate(ctx) d = policy.decide(event, draft, self_wxid=self_wxid, cfg=cfg) action, reason = d["action"], d["reason"] if action == "skip": log(f"skip #{event.get('msgId')} {sender_disp}: {reason}") return if action == "auto": res = _send_text_verified(event.get("talker"), draft, key) ok = res["ok"] _queue_insert(event, draft, sender_disp, "auto", reason, "auto_sent" if ok else "failed") log(f"auto {'OK ' + res['reason'] if ok else 'FAILED ' + res['reason']} " f"#{event.get('msgId')} {sender_disp} → {draft!r}") else: # review _queue_insert(event, draft, sender_disp, "review", reason, "pending") log(f"review #{event.get('msgId')} {sender_disp}: {reason} → 草稿 {draft!r}") # ── drain：消费 incoming.jsonl ───────────────────────── def _read_events(): evs = [] try: with open(QUEUE_FILE) as f: for line in f: line = line.strip() if line: try: evs.append(json.loads(line)) except ValueError: pass except OSError: pass return evs def drain(): """非阻塞单实例：已有 drain 在跑就跳过（后续消息的 inotify 会再触发）。""" lock = f"{WORK_DIR}/orch.lock" try: fd = os.open(lock, os.O_CREAT | os.O_EXCL | os.O_WRONLY) os.close(fd) except FileExistsError: try: # 5min 以上视为残留锁，抢占 if time.time() - os.path.getmtime(lock) > 300: os.remove(lock) os.close(os.open(lock, os.O_CREAT | os.O_EXCL | os.O_WRONLY)) else: return except OSError: return try: _drain_inner() finally: try: os.remove(lock) except OSError: pass def _drain_inner(): evs = _read_events() if not evs: return try: with open(CURSOR_FILE) as f: cursor = json.load(f).get("last_msgId", 0) except (OSError, ValueError): cursor = None max_id = max((int(e.get("msgId", 0)) for e in evs), default=0) if cursor is None: # 首跑：基线，不回放 with open(CURSOR_FILE, "w") as f: json.dump({"last_msgId": max_id}, f) w.chown_to_hermes(CURSOR_FILE) log(f"编排器首跑基线 last_msgId={max_id}") return todo = [e for e in evs if int(e.get("msgId", 0)) > cursor and e.get("kind") in CANDIDATE_KINDS] if todo: _, db, key = w.account_paths() stage = w.copy_db(db, subdir="_stage_orch") # 独立暂存,避开与 watcher 的 _stage 竞态 try: self_wxid, _ = w.detect_self(stage, key) cfg = policy.load_config() cfg["whitelist_contacts"], cfg["whitelist_groups"] = _load_whitelist() log(f"drain：{len(todo)} 条候选（auto_send={cfg['auto_send_enabled']}）") for e in todo: try: process_event(e, stage, key, self_wxid, cfg) except Exception as ex: # noqa: BLE001 log(f"处理 #{e.get('msgId')} 失败: {ex}") finally: for p in __import__("glob").glob(f"{WORK_DIR}/_stage_orch/en.db*"): try: os.remove(p) except OSError: pass with open(CURSOR_FILE, "w") as f: json.dump({"last_msgId": max_id}, f) w.chown_to_hermes(CURSOR_FILE) def main(): args = sys.argv[1:] if "--drain" in args: drain() else: print("用法: wechat_reply_orchestrator.py --drain") if __name__ == "__main__": main()