# -*- coding: utf-8 -*-
"""把 DNS 劫持的检测和修复写进 install.sh。

为什么重要：
    这条 4G 线路把 api.cloudflare.com 解析到 141.193.154.210
    （运营商拦截设备，自签名证书）。cloudflared 下载登录证书必须走
    这个域名，于是永远报 Failed to fetch resource ——
    卡了我好几轮才查出来。

    自动处理：
      1. 解析 api.cloudflare.com
      2. 如果拿到的不是 Cloudflare 的真实网段（104.16/104.17/172.67…）
         -> 判定为 DNS 劫持
      3. 在 /etc/hosts 里钉一个真实 IP
      4. 验证修好了
"""
from __future__ import annotations

import sys
from pathlib import Path

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

# 在"检查环境"那一步之后插入 DNS 检查
ANCHOR = '''echo "  内存: $(free -m | awk 'NR==2{print $2" MB 总量 / "$7" MB 可用"}')"'''

NEW = '''echo "  内存: $(free -m | awk 'NR==2{print $2" MB 总量 / "$7" MB 可用"}')"

# ---------- 0b. 检查 Cloudflare API 有没有被 DNS 劫持 ----------
# 踩过的坑：这条 4G 线路把 api.cloudflare.com 解析到运营商设备
# （141.193.154.210，自签名证书）。cloudflared 下载登录证书必须走这个域名，
# 于是永远报 Failed to fetch resource，具名隧道根本建不起来。
echo
echo "[0b] 检查 Cloudflare API"
CF_HOSTS=/etc/hosts
CF_MARK="# LanRemote: 钉住 Cloudflare API（网络会 DNS 劫持）"

resolve_cf() {
    getent ahostsv4 api.cloudflare.com 2>/dev/null | awk 'NR==1{print $1}'
}

is_real_cf() {
    case "$1" in
        104.16.*|104.17.*|104.18.*|104.19.*|104.20.*|104.21.*|172.64.*|172.66.*|172.67.*|162.159.*|188.114.*)
            return 0 ;;
        *)  return 1 ;;
    esac
}

CF_IP=$(resolve_cf)
echo "  api.cloudflare.com 现在解析到: ${CF_IP:-（解析不出来）}"

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)
echo "  实测 HTTP: ${CF_CODE}"

if is_real_cf "$CF_IP" && { [ "$CF_CODE" = "403" ] || [ "$CF_CODE" = "200" ]; }; then
    echo "  ✓ 正常（没被劫持）"
else
    echo "  ⚠ 检测到 Cloudflare API 被 DNS 劫持或不可达"
    echo "    具名隧道需要访问它来下载证书，先修掉。"

    cp -n "$CF_HOSTS" "${CF_HOSTS}.lanremote.bak" 2>/dev/null || true
    sed -i "/$CF_MARK/d" "$CF_HOSTS" 2>/dev/null || true
    sed -i '/api\\.cloudflare\\.com/d' "$CF_HOSTS" 2>/dev/null || true
    {
        echo ""
        echo "$CF_MARK"
        echo "104.16.132.229 api.cloudflare.com"
        echo "104.16.133.229 api.cloudflare.com"
    } >> "$CF_HOSTS"

    sleep 1
    CF_IP2=$(resolve_cf)
    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)
    echo "    修完解析到: ${CF_IP2:-?}   HTTP: ${CF_CODE2}"
    if [ "$CF_CODE2" = "403" ] || [ "$CF_CODE2" = "200" ]; then
        echo "    ✓ 修好了"
    else
        echo "    ✗ 还是不通 —— 具名隧道可能建不起来，"
        echo "      但匿名隧道仍然可用（它走的是另一条链路）"
    fi
fi'''

# 一并写进"具名隧道"那一段的说明
NAMED_ANCHOR = '''  具名隧道不受匿名隧道限流，稳定性完全不是一个级别。'''

NAMED_NEW = '''  具名隧道不受匿名隧道限流，稳定性完全不是一个级别。

  注意：建具名隧道前，install.sh 会检查 api.cloudflare.com 有没有被
  DNS 劫持（有些移动网络会把它解析到运营商设备）。如果被劫持，
  会自动在 /etc/hosts 里钉真实 IP —— 否则 cloudflared 下载证书
  会一直报 Failed to fetch resource。'''


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

    if "CF_MARK" in text:
        print("  install.sh 已经加过 DNS 检查")
    elif ANCHOR not in text:
        print("  [失败] 找不到锚点")
        ok = False
    else:
        text = text.replace(ANCHOR, NEW, 1)
        print("  install.sh: 加了 DNS 劫持检测与修复")

    if NAMED_ANCHOR in text and "DNS 劫持" not in text.split(NAMED_ANCHOR)[1][:200]:
        text = text.replace(NAMED_ANCHOR, NAMED_NEW, 1)
        print("  install.sh: 说明里补了 DNS 劫持提示")

    INSTALL.write_text(text, encoding="utf-8")

    # bash 语法检查（如果在 Windows 上没 bash，就只做个粗略的括号检查）
    import subprocess
    try:
        result = subprocess.run(["bash", "-n", str(INSTALL)],
                                capture_output=True, text=True, timeout=30)
        if result.returncode == 0:
            print("  install.sh: bash 语法 OK")
        else:
            print(f"  install.sh: bash 语法错误\n{result.stderr[:400]}")
            ok = False
    except FileNotFoundError:
        opens = text.count("{")
        closes = text.count("}")
        print(f"  （没有 bash 可检查）花括号 {opens} 开 / {closes} 闭")
    return 0 if ok else 1


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