#!/bin/bash
# 部署中转服务器，并重新开隧道。
set -u

RELAY_DIR=/root/lanremote-relay
mkdir -p "$RELAY_DIR"

echo "===== 1. 停掉之前的东西 ====="
pkill -f 'http.server 8080' 2>/dev/null && echo "  停了测试 HTTP 服务"
pkill -f 'relay.py' 2>/dev/null && echo "  停了旧的中转服务"
pkill -f 'cloudflared tunnel' 2>/dev/null && echo "  停了旧隧道"
sleep 2

echo
echo "===== 2. 语法检查 ====="
python3 -m py_compile "$RELAY_DIR/relay.py" && echo "  relay.py 语法 OK" || {
    echo "  ✗ 语法错误"; exit 1; }

echo
echo "===== 3. 启动中转服务器 ====="
cd "$RELAY_DIR"
RELAY_PORT=8080 \
RELAY_BIND=127.0.0.1 \
RELAY_DATA="$RELAY_DIR/data" \
nohup python3 "$RELAY_DIR/relay.py" > "$RELAY_DIR/relay.log" 2>&1 &
sleep 3

if curl -s --max-time 5 http://127.0.0.1:8080/v1/health | grep -q '"ok"'; then
    echo "  本机健康检查通过："
    curl -s --max-time 5 http://127.0.0.1:8080/v1/health
    echo
else
    echo "  ✗ 起不来，日志："
    cat "$RELAY_DIR/relay.log"
    exit 1
fi

echo
echo "===== 4. 开隧道 ====="
rm -f "$RELAY_DIR/cf.log"
nohup cloudflared tunnel --no-autoupdate --url http://127.0.0.1:8080 \
    > "$RELAY_DIR/cf.log" 2>&1 &

URL=""
for i in $(seq 1 40); do
    sleep 2
    URL=$(grep -oE 'https://[a-z0-9-]+\.trycloudflare\.com' "$RELAY_DIR/cf.log" | head -1)
    [ -n "$URL" ] && break
done

if [ -z "$URL" ]; then
    echo "  ✗ 拿不到隧道地址："; tail -20 "$RELAY_DIR/cf.log"; exit 1
fi
echo "  隧道地址: $URL"
echo "$URL" > "$RELAY_DIR/public_url.txt"

echo
echo "===== 5. 从设备自己验证公网可达 ====="
for i in 1 2 3 4 5; do
    BODY=$(curl -s --max-time 20 "$URL/v1/health")
    if echo "$BODY" | grep -q '"ok"'; then
        echo "  第 $i 次: 通了"
        break
    fi
    echo "  第 $i 次: 隧道建立中…"
    sleep 4
done

echo
echo "===== 6. 资源占用 ====="
free -m | head -2
echo
ps -eo rss,comm --sort=-rss | head -5 | awk '{printf "  %-8s %s\n", $1, $2}'
echo
echo "PUBLIC_URL=$URL"