# -*- coding: utf-8 -*-
"""真正把 install.sh 的 ExecStart 改成 --protocol http2。

上一版判断条件写的是 "--protocol http2" in text —— 结果被
一段**旧注释**（"千万不要写 --protocol http2"）骗过去了，实际没改。

这次只认 ExecStart 那两行。
"""
from __future__ import annotations

import sys
from pathlib import Path

ROOT = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote")
INSTALL = ROOT / "relay" / "install.sh"

EXEC_OLD = '''ExecStart=/usr/local/bin/cloudflared tunnel --no-autoupdate \\
    --edge-ip-version 4 --ha-connections 1 --retries 5 \\
    --url http://127.0.0.1:8080'''

EXEC_NEW = '''ExecStart=/usr/local/bin/cloudflared tunnel --no-autoupdate \\
    --protocol http2 --edge-ip-version 4 --ha-connections 1 --retries 5 \\
    --url http://127.0.0.1:8080'''

COMMENT_OLD_START = "# 协议：**用默认的 QUIC，千万不要写 --protocol http2**。"


def main() -> int:
    text = INSTALL.read_text(encoding="utf-8")
    if EXEC_NEW in text:
        print("  ExecStart 已经是 http2")
        return 0

    if EXEC_OLD not in text:
        print("  [失败] 找不到 ExecStart 段")
        for index, line in enumerate(text.splitlines(), 1):
            if "ExecStart" in line and "cloudflared" in line:
                print(f"      实际 {index}: {line}")
        return 1

    text = text.replace(EXEC_OLD, EXEC_NEW, 1)
    print("  ExecStart 加了 --protocol http2")

    # 把那段过时的注释也改掉（它现在说的是反话）
    lines = text.splitlines(keepends=True)
    out = []
    skipping = False
    for line in lines:
        if line.startswith(COMMENT_OLD_START):
            skipping = True
            out.append(
                "# 协议：用 HTTP/2（走 TCP 7844）。\n"
                "#   实测对比（同一设备、各跑 90 秒）：\n"
                "#       QUIC(UDP)  可用 48%，断开 16 次，拨号失败 16 次\n"
                "#       HTTP/2     可用 80%，断开  2 次，拨号失败  0 次\n"
                "#   这条线路在丢 UDP 包 —— 日志里全是\n"
                "#     'Failed to dial a quic connection: no recent network activity'\n"
                "#     'Connection terminated: Failed to accept QUIC stream'\n"
                "#   TCP 有重传，慢一点但稳得多。\n"
                "#   注意：早先在 192.168.2.x 的出口上 TCP 7844 是被挡的，\n"
                "#   换了出口就通了 —— 换网络必须重测，别照搬旧结论。\n")
            continue
        if skipping:
            if line.strip().startswith("#"):
                continue          # 跳过旧注释块剩余的注释行
            skipping = False
        out.append(line)
    text = "".join(out)

    INSTALL.write_text(text, encoding="utf-8", newline="\n")
    raw = INSTALL.read_bytes()
    if b"\r\n" in raw:
        INSTALL.write_bytes(raw.replace(b"\r\n", b"\n"))
        print("  顺手修了 CRLF")

    print("  复核：")
    for index, line in enumerate(
            INSTALL.read_text(encoding="utf-8").splitlines(), 1):
        if "ExecStart" in line and "cloudflared" in line:
            print(f"    {index}: {line}")
        if "--protocol http2 --edge-ip-version" in line:
            print(f"    {index}: {line.strip()}")
    return 0


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