# -*- coding: utf-8 -*-
"""临时调试：在 RelayTransport 的收发路径上打点。

被控端日志显示 auth 成功了，但客户端收不到回复 ——
所以问题在"回复怎么回到客户端"这一段。先把收发都打出来看。
用完会撤掉。
"""
from __future__ import annotations

import sys
from pathlib import Path

RELAY = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote\android\src"
             r"\com\lanremote\watch\RelayTransport.java")

OLD_POLL = '''                online = true;
                lastError = "";
                backoff = RETRY_MIN_MS;
                if (result.body.length > 0) {
                    inbox.offer(result.body);
                }'''

NEW_POLL = '''                online = true;
                lastError = "";
                backoff = RETRY_MIN_MS;
                System.err.println("[dbg] poll -> " + result.code
                        + " bytes=" + result.body.length
                        + " session=" + session.substring(0, Math.min(8, session.length())));
                if (result.body.length > 0) {
                    inbox.offer(result.body);
                }'''

OLD_SEND = '''                HttpResult result = request("/v1/client/send", chunk, true,
                        SEND_TIMEOUT_MS);
                if (result.code != 200) {
                    lastError = "发送返回 " + result.code;
                }'''

NEW_SEND = '''                HttpResult result = request("/v1/client/send", chunk, true,
                        SEND_TIMEOUT_MS);
                System.err.println("[dbg] send " + chunk.length + " bytes -> "
                        + result.code);
                if (result.code != 200) {
                    lastError = "发送返回 " + result.code;
                }'''

OLD_ATTACH = '''            session = Json.str(reply, "session", "");
            deviceName = Json.str(reply, "deviceName", "");'''

NEW_ATTACH = '''            session = Json.str(reply, "session", "");
            deviceName = Json.str(reply, "deviceName", "");
            System.err.println("[dbg] attach ok session="
                    + session.substring(0, Math.min(8, session.length()))
                    + " device=" + deviceName);'''


def main() -> int:
    text = RELAY.read_text(encoding="utf-8")
    ok = True
    for old, new, tag in ((OLD_ATTACH, NEW_ATTACH, "attach 打点"),
                          (OLD_POLL, NEW_POLL, "poll 打点"),
                          (OLD_SEND, NEW_SEND, "send 打点")):
        if old not in text:
            print(f"  [失败] {tag} —— 锚点没找到")
            ok = False
            continue
        text = text.replace(old, new, 1)
        print(f"  {tag}")
    RELAY.write_text(text, encoding="utf-8")
    return 0 if ok else 1


if __name__ == "__main__":
    sys.exit(main())