# -*- coding: utf-8 -*-
"""install.sh 的两处小瑕疵：

1. HTTP 码显示成 "000000"
   curl 失败时 -w '%{http_code}' 已经输出 "000"，我又 || echo 000
   补了一个，于是变成 "000000"。
   改成先把输出收进变量，空才填默认值。

2. 残留的旧注释
   我之前手工修过一次，留了句文案不同的标记行，没被清掉。
   把清理规则放宽，凡是提到 LanRemote 且带 api.cloudflare.com 的注释都清。
"""
from __future__ import annotations

import sys
from pathlib import Path

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

REPLACEMENTS = [
    ('''CF_CODE=$(timeout 12 curl -4 -s -o /dev/null -w '%{http_code}' \\
          https://api.cloudflare.com/client/v4/user 2>/dev/null || echo 000)''',
     '''CF_CODE=$(timeout 12 curl -4 -s -o /dev/null -w '%{http_code}' \\
          https://api.cloudflare.com/client/v4/user 2>/dev/null)
CF_CODE=${CF_CODE:-000}'''),

    ('''    CF_CODE2=$(timeout 12 curl -4 -s -o /dev/null -w '%{http_code}' \\
               https://api.cloudflare.com/client/v4/user 2>/dev/null || echo 000)''',
     '''    CF_CODE2=$(timeout 12 curl -4 -s -o /dev/null -w '%{http_code}' \\
               https://api.cloudflare.com/client/v4/user 2>/dev/null)
    CF_CODE2=${CF_CODE2:-000}'''),

    ('''    sed -i "/$CF_MARK/d" "$CF_HOSTS" 2>/dev/null || true
    sed -i '/api\\.cloudflare\\.com/d' "$CF_HOSTS" 2>/dev/null || true''',
     '''    # 清掉所有和本程序相关的 hosts 记录（含早期版本留下的不同文案）
    sed -i '/LanRemote.*Cloudflare API/d' "$CF_HOSTS" 2>/dev/null || true
    sed -i '/api\\.cloudflare\\.com/d' "$CF_HOSTS" 2>/dev/null || true
    # 顺手删掉清理后可能剩下的空行
    sed -i '/^[[:space:]]*$/d' "$CF_HOSTS" 2>/dev/null || true
    echo "" >> "$CF_HOSTS"'''),
]


def main() -> int:
    text = INSTALL.read_text(encoding="utf-8")
    ok = True
    for old, new in REPLACEMENTS:
        if old not in text:
            print(f"  [跳过] 锚点没找到: {old.strip().splitlines()[0][:50]}")
            continue
        text = text.replace(old, new, 1)
        print(f"  改了一处")
    INSTALL.write_text(text, encoding="utf-8", newline="\n")

    # 确保还是 LF
    raw = INSTALL.read_bytes()
    if b"\r\n" in raw:
        INSTALL.write_bytes(raw.replace(b"\r\n", b"\n"))
        print("  顺手又修了 CRLF")
    print("  install.sh 已更新（LF）")
    return 0 if ok else 1


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