#!/bin/bash

# --- 配置 ---
# !!! 重要: 公網(wǎng)服務(wù)器 IP 地址 !!!
SERVER_IP="YOUR_SERVER_PUBLIC_IP" # 請(qǐng)?zhí)鎿Q為您的公網(wǎng) IP

# --- 服務(wù)器的密鑰參數(shù) ---
# 使用 "_" 來(lái)強(qiáng)制啟用密鑰。如果 HBBS_DATA_DIR 中存在密鑰，則使用它們，否則將生成新密鑰。
# 客戶(hù)端需要 id_ed25519.pub 中的公鑰。
KEY_PARAMETER="_" # 推薦：強(qiáng)制使用密鑰

# 用于持久化的數(shù)據(jù)目錄 (在腳本所在目錄下創(chuàng)建)
BASE_DATA_DIR="./rustdesk_server_data"
HBBS_DATA_DIR="${BASE_DATA_DIR}/hbbs" # hbbs 的數(shù)據(jù)和密鑰將存放在這里
HBBR_DATA_DIR="${BASE_DATA_DIR}/hbbr" # hbbr 的數(shù)據(jù)目錄 (密鑰從 HBBS_DATA_DIR 共享)

# Docker 鏡像
RUSTDESK_IMAGE="rustdesk/rustdesk-server"

# --- 腳本開(kāi)始 ---
echo "RustDesk 服務(wù)器部署腳本"
echo "---------------------------------"

# 檢查 SERVER_IP 是否已配置
if [ "${SERVER_IP}" == "YOUR_SERVER_PUBLIC_IP" ] || [ -z "${SERVER_IP}" ]; then
    echo "錯(cuò)誤：請(qǐng)修改腳本，將 SERVER_IP 變量設(shè)置為您的公網(wǎng) IP 地址。"
    exit 1
fi

# 檢查 Docker
if ! command -v docker &> /dev/null
then
    echo "錯(cuò)誤：找不到 Docker 命令。請(qǐng)先安裝 Docker。"
    exit 1
fi
echo "Docker 已找到。"
echo ""

# --- 1. 停止并移除已存在的 RustDesk 容器 ---
echo "正在停止并移除已存在的 RustDesk 容器 (hbbs, hbbr)..."
if sudo docker ps -a --format '{{.Names}}' | grep -qE '^(hbbs|hbbr)$'; then
    sudo docker stop hbbs hbbr > /dev/null 2>&1 || true
    sudo docker rm hbbs hbbr > /dev/null 2>&1 || true
    echo "已存在的 hbbs 和 hbbr 容器已停止并移除。"
else
    echo "未找到名為 hbbs 或 hbbr 的舊容器。"
fi
echo ""

# --- 2. 創(chuàng)建數(shù)據(jù)目錄 ---
echo "服務(wù)器 IP: ${SERVER_IP}"
echo "HBBS 數(shù)據(jù) (及密鑰) 目錄: ${HBBS_DATA_DIR}"
echo "HBBR 數(shù)據(jù)目錄: ${HBBR_DATA_DIR}"
echo ""

echo "正在創(chuàng)建數(shù)據(jù)目錄 (如果不存在)..."
mkdir -p "${HBBS_DATA_DIR}"
mkdir -p "${HBBR_DATA_DIR}"
echo "數(shù)據(jù)目錄已確保存在。"
echo ""

# --- 3. 啟動(dòng) HBBS (信令服務(wù)器) ---
echo "正在啟動(dòng) HBBS (信令服務(wù)器)..."
sudo docker run -d \
  --name hbbs \
  -p 21115:21115 \
  -p 21116:21116 \
  -p 21116:21116/udp \
  -v "${HBBS_DATA_DIR}":/root \
  --restart unless-stopped \
  "${RUSTDESK_IMAGE}" \
  hbbs -k "${KEY_PARAMETER}" -r "${SERVER_IP}"

echo "正在等待 HBBS 初始化 (約 5 秒)..."
sleep 5

# 檢查 hbbs 是否已啟動(dòng)并生成密鑰
if [ ! -f "${HBBS_DATA_DIR}/id_ed25519.pub" ]; then
    echo "錯(cuò)誤：HBBS 啟動(dòng)后未找到公鑰文件 ${HBBS_DATA_DIR}/id_ed25519.pub。"
    echo "請(qǐng)檢查 HBBS 日志: sudo docker logs hbbs"
    sudo docker logs hbbs
    echo "正在清理失敗的 HBBS 容器..."
    sudo docker stop hbbs > /dev/null 2>&1 || true
    sudo docker rm hbbs > /dev/null 2>&1 || true
    exit 1
fi
PUBLIC_KEY=$(cat "${HBBS_DATA_DIR}/id_ed25519.pub")
echo "HBBS 已啟動(dòng)，公鑰已獲取。"
echo ""

# --- 4. 啟動(dòng) HBBR (中繼服務(wù)器) ---
echo "正在啟動(dòng) HBBR (中繼服務(wù)器)..."
# HBBR 掛載 HBBS_DATA_DIR 以共享由 HBBS 生成的密鑰
sudo docker run -d \
  --name hbbr \
  -p 21117:21117 \
  -p 21118:21118 \
  -p 21118:21118/udp \
  -v "${HBBS_DATA_DIR}":/root \
  --restart unless-stopped \
  "${RUSTDESK_IMAGE}" \
  hbbr -k "${KEY_PARAMETER}"

echo "正在等待 HBBR 初始化 (約 3 秒)..."
sleep 3

# 檢查 hbbr 是否啟動(dòng)
if ! sudo docker ps --filter "name=hbbr" --filter "status=running" --format '{{.Names}}' | grep -q '^hbbr$'; then
    echo "錯(cuò)誤: HBBR 容器未能成功啟動(dòng)或未處于運(yùn)行狀態(tài)。"
    echo "請(qǐng)檢查 HBBR 日志: sudo docker logs hbbr"
    sudo docker logs hbbr
    echo "正在清理 HBBS 和 HBBR 容器..."
    sudo docker stop hbbs hbbr > /dev/null 2>&1 || true
    sudo docker rm hbbs hbbr > /dev/null 2>&1 || true
    exit 1
fi
echo "HBBR 已啟動(dòng)。"
echo ""

# --- 部署摘要 ---
echo "--- 部署摘要 ---"
echo "RustDesk 服務(wù)器容器 (hbbs 和 hbbr) 現(xiàn)在應(yīng)該正在運(yùn)行。"
echo "您可以使用以下命令檢查它們的狀態(tài): sudo docker ps -a"
echo "您可以使用以下命令查看日志: sudo docker logs hbbs  或  sudo docker logs hbbr"
echo ""
echo "--- 客戶(hù)端配置 ---"
echo "在您的 RustDesk 客戶(hù)端中，請(qǐng)按如下配置："
echo "  ID 服務(wù)器: ${SERVER_IP}"
echo "  中繼服務(wù)器: ${SERVER_IP}"
echo "  Key (公鑰): ${PUBLIC_KEY}"
echo ""
echo "重要提示: 'Key' 是由 hbbs 生成的公鑰文件的內(nèi)容。"
echo "它已從以下位置讀取: ${HBBS_DATA_DIR}/id_ed25519.pub"
echo ""
echo "如果您以后需要查找公鑰，請(qǐng)運(yùn)行: cat ${HBBS_DATA_DIR}/id_ed25519.pub"
echo "腳本執(zhí)行完畢。"