# -*- coding: utf-8 -*-
"""把定稿的隧道配置写进 install.sh：
    --protocol http2 --ha-connections 2

实测数据（同一设备，各跑 2~3 分钟）：
    协议/连接数      可用率    断开    拨号失败
    QUIC  ha=1        48%      16       16
    HTTP2 ha=1      35~65%       0        0
    HTTP2 ha=2      60~75%       0~1      0
"""
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 \\
    --protocol http2 --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 2 --retries 5 \\
    --url http://127.0.0.1:8080'''

BLOCK_OLD_START = "# 连接数：ha-connections=1。"

BLOCK_NEW = '''# 连接数：ha-connections=2。
#   实测（HTTP/2 下）：
#       ha=1 -> 可用 35~65%（连接本身稳，但一条卡住就全卡）
#       ha=2 -> 可用 60~75%（一条卡住另一条顶上）
#   注意：UDP/QUIC 时代 ha=2 是互相拖累（两条抢一个 UDP 通道），
#        换成 TCP 之后才变成真正的冗余。协议换了，连接数结论也得重测。
'''


def main() -> int:
    text = INSTALL.read_text(encoding="utf-8")

    if "--ha-connections 2 --retries 5" in text:
        print("  已经是 ha=2")
    elif EXEC_OLD in text:
        text = text.replace(EXEC_OLD, EXEC_NEW, 1)
        print("  ExecStart：ha-connections 1 -> 2")
    else:
        print("  [失败] ExecStart 锚点没找到")
        for index, line in enumerate(text.splitlines(), 1):
            if "ExecStart" in line and "cloudflared" in line:
                print(f"      实际 {index}: {line}")
        return 1

    # 替换那段过时的 ha=1 说明
    lines = text.splitlines(keepends=True)
    out = []
    skipping = False
    for line in lines:
        if line.startswith(BLOCK_OLD_START):
            skipping = True
            out.append(BLOCK_NEW)
            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 "--protocol http2 --edge-ip-version" in line:
            print(f"    {index}: {line.strip()}")
    return 0


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