Docker轻松自建RustDesk服务器的实现步骤

#!/bin/bash

# --- 配置 ---
# !!! 重要: 公网服务器 IP 地址 !!!
SERVER_IP="YOUR_SERVER_PUBLIC_IP" # 请替换为您的公网 IP

# --- 服务器的密钥参数 ---
# 使用 "_" 来强制启用密钥。如果 HBBS_DATA_DIR 中存在密钥，则使用它们，否则将生成新密钥。
# 客户端需要 id_ed25519.pub 中的公钥。
KEY_PARAMETER="_" # 推荐：强制使用密钥

# 用于持久化的数据目录 (在脚本所在目录下创建)
BASE_DATA_DIR="./rustdesk_server_data"
HBBS_DATA_DIR="${BASE_DATA_DIR}/hbbs" # hbbs 的数据和密钥将存放在这里
HBBR_DATA_DIR="${BASE_DATA_DIR}/hbbr" # hbbr 的数据目录 (密钥从 HBBS_DATA_DIR 共享)

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

# --- 脚本开始 ---
echo "RustDesk 服务器部署脚本"
echo "---------------------------------"

# 检查 SERVER_IP 是否已配置
if [ "${SERVER_IP}" == "YOUR_SERVER_PUBLIC_IP" ] || [ -z "${SERVER_IP}" ]; then
    echo "错误：请修改脚本，将 SERVER_IP 变量设置为您的公网 IP 地址。"
    exit 1
fi

# 检查 Docker
if ! command -v docker &> /dev/null
then
    echo "错误：找不到 Docker 命令。请先安装 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. 创建数据目录 ---
echo "服务器 IP: ${SERVER_IP}"
echo "HBBS 数据 (及密钥) 目录: ${HBBS_DATA_DIR}"
echo "HBBR 数据目录: ${HBBR_DATA_DIR}"
echo ""

echo "正在创建数据目录 (如果不存在)..."
mkdir -p "${HBBS_DATA_DIR}"
mkdir -p "${HBBR_DATA_DIR}"
echo "数据目录已确保存在。"
echo ""

# --- 3. 启动 HBBS (信令服务器) ---
echo "正在启动 HBBS (信令服务器)..."
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 是否已启动并生成密钥
if [ ! -f "${HBBS_DATA_DIR}/id_ed25519.pub" ]; then
    echo "错误：HBBS 启动后未找到公钥文件 ${HBBS_DATA_DIR}/id_ed25519.pub。"
    echo "请检查 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 已启动，公钥已获取。"
echo ""

# --- 4. 启动 HBBR (中继服务器) ---
echo "正在启动 HBBR (中继服务器)..."
# 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 是否启动
if ! sudo docker ps --filter "name=hbbr" --filter "status=running" --format '{{.Names}}' | grep -q '^hbbr$'; then
    echo "错误: HBBR 容器未能成功启动或未处于运行状态。"
    echo "请检查 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 已启动。"
echo ""

# --- 部署摘要 ---
echo "--- 部署摘要 ---"
echo "RustDesk 服务器容器 (hbbs 和 hbbr) 现在应该正在运行。"
echo "您可以使用以下命令检查它们的状态: sudo docker ps -a"
echo "您可以使用以下命令查看日志: sudo docker logs hbbs  或  sudo docker logs hbbr"
echo ""
echo "--- 客户端配置 ---"
echo "在您的 RustDesk 客户端中，请按如下配置："
echo "  ID 服务器: ${SERVER_IP}"
echo "  中继服务器: ${SERVER_IP}"
echo "  Key (公钥): ${PUBLIC_KEY}"
echo ""
echo "重要提示: 'Key' 是由 hbbs 生成的公钥文件的内容。"
echo "它已从以下位置读取: ${HBBS_DATA_DIR}/id_ed25519.pub"
echo ""
echo "如果您以后需要查找公钥，请运行: cat ${HBBS_DATA_DIR}/id_ed25519.pub"
echo "脚本执行完毕。"