137 lines
4.3 KiB
Bash
137 lines
4.3 KiB
Bash
#!/bin/bash
|
||
# 浏览器 VNC 管理器
|
||
|
||
DAEMON_SCRIPT="$(dirname $0)/start-browser-vnc.sh"
|
||
BASE_VNC_PORT=5900
|
||
BASE_DATA_DIR="/tmp/chrome-profiles" # 用户数据目录
|
||
|
||
start() {
|
||
local instance=${1:-1}
|
||
local vnc_pwd=${2} # 密码必须传入，不再有默认值
|
||
local display_num=$((98 + instance))
|
||
local vnc_port=$((BASE_VNC_PORT + instance - 1))
|
||
local data_dir="${BASE_DATA_DIR}/instance-${instance}"
|
||
|
||
# 检查密码是否提供
|
||
if [ -z "$vnc_pwd" ]; then
|
||
echo "❌ 错误: 启动实例时必须提供密码"
|
||
echo "用法: $0 start <实例编号> <密码>"
|
||
return 1
|
||
fi
|
||
|
||
if [ -f "/tmp/chrome-browser-${display_num}.pid" ] && kill -0 $(cat "/tmp/chrome-browser-${display_num}.pid") 2>/dev/null; then
|
||
echo "实例 ${instance} 已在运行中"
|
||
return 1
|
||
fi
|
||
|
||
echo "正在启动实例 ${instance} (显示编号 :${display_num}, VNC 端口: ${vnc_port})..."
|
||
echo "数据目录: ${data_dir}"
|
||
|
||
# 确保数据目录存在
|
||
mkdir -p "$data_dir"
|
||
|
||
nohup "$DAEMON_SCRIPT" "$vnc_pwd" "$vnc_port" "$display_num" "$data_dir" >> "/tmp/chrome-browser-${display_num}.log" 2>&1 &
|
||
sleep 3
|
||
|
||
local public_ip=$(hostname -I | awk '{print $1}')
|
||
echo "✅ 实例 ${instance} 已启动"
|
||
echo " VNC: ${public_ip}:${vnc_port}"
|
||
echo " 密码: ${vnc_pwd}"
|
||
echo " 数据: ${data_dir} (停止时将删除)"
|
||
}
|
||
|
||
stop() {
|
||
local instance=${1:-1}
|
||
local display_num=$((98 + instance))
|
||
local pid_file="/tmp/chrome-browser-${display_num}.pid"
|
||
local data_dir="${BASE_DATA_DIR}/instance-${instance}"
|
||
|
||
if [ -f "$pid_file" ]; then
|
||
local pid=$(cat "$pid_file")
|
||
echo "正在停止实例 ${instance} (PID: $pid)..."
|
||
|
||
# 先发送 TERM 信号，让浏览器优雅关闭
|
||
kill -TERM $pid 2>/dev/null
|
||
sleep 3
|
||
|
||
# 如果还在运行，强制杀死
|
||
if kill -0 $pid 2>/dev/null; then
|
||
echo "正在强制终止..."
|
||
kill -9 $pid 2>/dev/null
|
||
fi
|
||
|
||
# 删除 PID 文件
|
||
rm -f "$pid_file"
|
||
|
||
# 删除用户数据目录
|
||
if [ -d "$data_dir" ]; then
|
||
echo "正在删除用户数据: ${data_dir}"
|
||
rm -rf "$data_dir"
|
||
fi
|
||
|
||
echo "✅ 实例 ${instance} 已停止"
|
||
else
|
||
echo "实例 ${instance} 未运行，正在清理数据..."
|
||
# 即使进程不在运行，也清理数据目录
|
||
if [ -d "$data_dir" ]; then
|
||
rm -rf "$data_dir"
|
||
echo "✅ 数据已清理: ${data_dir}"
|
||
fi
|
||
fi
|
||
|
||
# 额外清理：杀掉可能残留的 Xvfb 和 x11vnc 进程
|
||
pkill -f "Xvfb :${display_num}" 2>/dev/null
|
||
pkill -f "x11vnc.*${display_num}" 2>/dev/null
|
||
|
||
# 清理临时文件
|
||
rm -f "/tmp/.X${display_num}-lock" 2>/dev/null
|
||
rm -f "/tmp/.X11-unix/X${display_num}" 2>/dev/null
|
||
}
|
||
|
||
cleanup_data() {
|
||
local instance=${1:-1}
|
||
local data_dir="${BASE_DATA_DIR}/instance-${instance}"
|
||
|
||
# 先停止实例
|
||
stop $instance
|
||
|
||
# 如果还有残留数据，强制删除
|
||
if [ -d "$data_dir" ]; then
|
||
echo "正在强制清理数据: ${data_dir}"
|
||
rm -rf "$data_dir"
|
||
fi
|
||
|
||
echo "✅ 实例 ${instance} 的数据已清理"
|
||
}
|
||
|
||
case "${1:-}" in
|
||
start)
|
||
if [ $# -lt 3 ]; then
|
||
echo "❌ 错误: start 命令需要实例编号和密码"
|
||
echo "用法: $0 start <实例编号> <密码>"
|
||
echo "示例: $0 start 1 mypassword"
|
||
exit 1
|
||
fi
|
||
start "$2" "$3"
|
||
;;
|
||
stop)
|
||
stop "${2:-1}"
|
||
;;
|
||
cleanup)
|
||
cleanup_data "${2:-1}"
|
||
;;
|
||
*)
|
||
echo "用法: $0 {start|stop|cleanup} [参数]"
|
||
echo ""
|
||
echo "命令说明:"
|
||
echo " start <编号> <密码> - 启动实例 (必须提供密码)"
|
||
echo " stop [编号] - 停止实例并删除用户数据 (默认: 1)"
|
||
echo " cleanup [编号] - 强制删除指定实例的用户数据 (默认: 1)"
|
||
echo ""
|
||
echo "使用示例:"
|
||
echo " $0 start 1 mypass # 启动实例 1，密码 mypass"
|
||
echo " $0 start 2 123456 # 启动实例 2，密码 123456"
|
||
echo " $0 stop 1 # 停止实例 1 并删除其数据"
|
||
exit 1
|
||
;;
|
||
esac |