571 lines
14 KiB
Bash
571 lines
14 KiB
Bash
#!/bin/bash
|
|
|
|
# 确保以 bash 运行
|
|
if [ -z "$BASH_VERSION" ]; then
|
|
exec bash "$0" "$@"
|
|
fi
|
|
|
|
# VoiceHub 日常维护管理脚本
|
|
# 支持上下键选择菜单
|
|
|
|
set -e
|
|
|
|
# 颜色定义
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
BLUE='\033[0;34m'
|
|
CYAN='\033[0;36m'
|
|
NC='\033[0m' # No Color
|
|
|
|
# 默认安装目录
|
|
PROJECT_DIR="/opt/voicehub"
|
|
DIRECT_LOG_DIR="$PROJECT_DIR/logs"
|
|
DIRECT_PID_FILE="$DIRECT_LOG_DIR/voicehub.pid"
|
|
DIRECT_LOG_FILE="$DIRECT_LOG_DIR/voicehub-out.log"
|
|
|
|
ensure_pnpm() {
|
|
export PNPM_HOME="${PNPM_HOME:-$HOME/.local/share/pnpm}"
|
|
export COREPACK_ENABLE_DOWNLOAD_PROMPT=0
|
|
mkdir -p "$PNPM_HOME"
|
|
export PATH="$PNPM_HOME:$PATH"
|
|
|
|
if ! command -v corepack &> /dev/null; then
|
|
sudo npm install -g corepack
|
|
hash -r
|
|
fi
|
|
|
|
corepack enable
|
|
|
|
local package_manager=""
|
|
if [[ -f "$PROJECT_DIR/package.json" ]]; then
|
|
package_manager=$(node -p "try { JSON.parse(require('fs').readFileSync('$PROJECT_DIR/package.json', 'utf8')).packageManager || '' } catch { '' }")
|
|
fi
|
|
|
|
if [[ -n "$package_manager" ]]; then
|
|
corepack prepare "$package_manager" --activate
|
|
fi
|
|
|
|
hash -r
|
|
|
|
if ! command -v pnpm &> /dev/null; then
|
|
echo -e "${RED}错误: pnpm 不可用${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
has_working_pm2() {
|
|
local pm2_root
|
|
local pm2_bin
|
|
|
|
pm2_root=$(pnpm root -g 2>/dev/null || true)
|
|
pm2_bin="$pm2_root/pm2/bin/pm2"
|
|
|
|
if [[ -f "$pm2_bin" ]]; then
|
|
node "$pm2_bin" -v > /dev/null 2>&1
|
|
return $?
|
|
fi
|
|
|
|
if command -v pm2 &> /dev/null; then
|
|
pm2 -v > /dev/null 2>&1
|
|
return $?
|
|
fi
|
|
|
|
return 1
|
|
}
|
|
|
|
run_pm2() {
|
|
local pm2_root
|
|
local pm2_bin
|
|
|
|
pm2_root=$(pnpm root -g 2>/dev/null || true)
|
|
pm2_bin="$pm2_root/pm2/bin/pm2"
|
|
|
|
if [[ -f "$pm2_bin" ]]; then
|
|
node "$pm2_bin" "$@"
|
|
return $?
|
|
fi
|
|
|
|
if command -v pm2 &> /dev/null; then
|
|
pm2 "$@"
|
|
return $?
|
|
fi
|
|
|
|
echo -e "${RED}错误: PM2 不可用，请重新运行部署脚本修复 PM2${NC}"
|
|
return 1
|
|
}
|
|
|
|
is_direct_service_running() {
|
|
if [[ ! -f "$DIRECT_PID_FILE" ]]; then
|
|
return 1
|
|
fi
|
|
|
|
local pid
|
|
pid=$(cat "$DIRECT_PID_FILE" 2>/dev/null)
|
|
|
|
if [[ -z "$pid" ]]; then
|
|
rm -f "$DIRECT_PID_FILE"
|
|
return 1
|
|
fi
|
|
|
|
if kill -0 "$pid" 2>/dev/null; then
|
|
return 0
|
|
fi
|
|
|
|
rm -f "$DIRECT_PID_FILE"
|
|
return 1
|
|
}
|
|
|
|
start_direct_service() {
|
|
mkdir -p "$DIRECT_LOG_DIR"
|
|
|
|
if is_direct_service_running; then
|
|
echo -e "${YELLOW}VoiceHub 已在直接运行模式下启动${NC}"
|
|
return 0
|
|
fi
|
|
|
|
if [[ ! -f "$PROJECT_DIR/.output/server/index.mjs" ]]; then
|
|
echo -e "${RED}错误: 未找到构建产物，请先执行 voicehub build${NC}"
|
|
exit 1
|
|
fi
|
|
|
|
nohup node "$PROJECT_DIR/.output/server/index.mjs" >> "$DIRECT_LOG_FILE" 2>&1 &
|
|
local pid=$!
|
|
echo "$pid" > "$DIRECT_PID_FILE"
|
|
sleep 2
|
|
|
|
if is_direct_service_running; then
|
|
echo -e "${GREEN}✓ 直接运行模式已启动 (PID: $pid)${NC}"
|
|
else
|
|
echo -e "${RED}错误: 直接运行模式启动失败，请查看日志: $DIRECT_LOG_FILE${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
stop_direct_service() {
|
|
if ! is_direct_service_running; then
|
|
echo -e "${YELLOW}直接运行模式未启动${NC}"
|
|
return 0
|
|
fi
|
|
|
|
local pid
|
|
pid=$(cat "$DIRECT_PID_FILE")
|
|
kill "$pid" 2>/dev/null || true
|
|
|
|
for _ in {1..10}; do
|
|
if ! kill -0 "$pid" 2>/dev/null; then
|
|
rm -f "$DIRECT_PID_FILE"
|
|
echo -e "${GREEN}✓ 直接运行模式已停止${NC}"
|
|
return 0
|
|
fi
|
|
sleep 1
|
|
done
|
|
|
|
kill -9 "$pid" 2>/dev/null || true
|
|
rm -f "$DIRECT_PID_FILE"
|
|
echo -e "${GREEN}✓ 直接运行模式已强制停止${NC}"
|
|
}
|
|
|
|
# 检查项目目录
|
|
check_project() {
|
|
if [[ ! -d "$PROJECT_DIR" ]]; then
|
|
echo -e "${RED}错误: 项目目录不存在: $PROJECT_DIR${NC}"
|
|
echo -e "${RED}请先运行部署脚本安装 VoiceHub${NC}"
|
|
exit 1
|
|
fi
|
|
cd "$PROJECT_DIR"
|
|
ensure_pnpm
|
|
}
|
|
|
|
# 检测服务管理器类型
|
|
detect_service_manager() {
|
|
if [[ -f "$PROJECT_DIR/ecosystem.config.cjs" || -f "$PROJECT_DIR/ecosystem.config.js" ]]; then
|
|
echo "pm2"
|
|
elif [[ -f "/etc/systemd/system/voicehub.service" ]]; then
|
|
echo "systemd"
|
|
elif [[ -f "$PROJECT_DIR/.output/server/index.mjs" ]]; then
|
|
echo "direct"
|
|
else
|
|
echo "none"
|
|
fi
|
|
}
|
|
|
|
# 查看状态
|
|
cmd_status() {
|
|
check_project
|
|
echo -e "${BLUE}检查服务状态...${NC}"
|
|
|
|
local service_type=$(detect_service_manager)
|
|
|
|
if [[ "$service_type" == "pm2" ]]; then
|
|
if has_working_pm2 && run_pm2 list | grep -q "voicehub"; then
|
|
echo -e "${GREEN}PM2 服务状态:${NC}"
|
|
run_pm2 list | grep voicehub
|
|
else
|
|
echo -e "${YELLOW}PM2 服务未运行或 PM2 命令异常${NC}"
|
|
fi
|
|
elif [[ "$service_type" == "systemd" ]]; then
|
|
echo -e "${GREEN}systemctl 服务状态:${NC}"
|
|
sudo systemctl status voicehub
|
|
elif [[ "$service_type" == "direct" ]]; then
|
|
if is_direct_service_running; then
|
|
echo -e "${GREEN}直接运行模式状态: 运行中${NC}"
|
|
echo -e "${BLUE}PID: $(cat "$DIRECT_PID_FILE")${NC}"
|
|
echo -e "${BLUE}日志: $DIRECT_LOG_FILE${NC}"
|
|
else
|
|
echo -e "${YELLOW}直接运行模式未启动${NC}"
|
|
echo -e "${BLUE}可使用 voicehub start 启动${NC}"
|
|
fi
|
|
else
|
|
echo -e "${YELLOW}未检测到服务配置${NC}"
|
|
fi
|
|
}
|
|
|
|
# 启动服务
|
|
cmd_start() {
|
|
check_project
|
|
echo -e "${BLUE}启动服务...${NC}"
|
|
|
|
local service_type=$(detect_service_manager)
|
|
|
|
if [[ "$service_type" == "pm2" ]]; then
|
|
if [[ -f "ecosystem.config.cjs" ]]; then
|
|
run_pm2 start ecosystem.config.cjs
|
|
else
|
|
run_pm2 start ecosystem.config.js
|
|
fi
|
|
echo -e "${GREEN}✓ PM2 服务已启动${NC}"
|
|
elif [[ "$service_type" == "systemd" ]]; then
|
|
sudo systemctl start voicehub
|
|
echo -e "${GREEN}✓ systemctl 服务已启动${NC}"
|
|
elif [[ "$service_type" == "direct" ]]; then
|
|
start_direct_service
|
|
else
|
|
echo -e "${YELLOW}未找到服务配置，请先运行部署脚本配置服务${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 停止服务
|
|
cmd_stop() {
|
|
check_project
|
|
echo -e "${BLUE}停止服务...${NC}"
|
|
|
|
local service_type=$(detect_service_manager)
|
|
|
|
if [[ "$service_type" == "pm2" ]]; then
|
|
run_pm2 stop voicehub
|
|
echo -e "${GREEN}✓ PM2 服务已停止${NC}"
|
|
elif [[ "$service_type" == "systemd" ]]; then
|
|
sudo systemctl stop voicehub
|
|
echo -e "${GREEN}✓ systemctl 服务已停止${NC}"
|
|
elif [[ "$service_type" == "direct" ]]; then
|
|
stop_direct_service
|
|
else
|
|
echo -e "${YELLOW}未找到服务配置${NC}"
|
|
fi
|
|
}
|
|
|
|
# 重启服务
|
|
cmd_restart() {
|
|
check_project
|
|
echo -e "${BLUE}重启服务...${NC}"
|
|
|
|
local service_type=$(detect_service_manager)
|
|
|
|
if [[ "$service_type" == "pm2" ]]; then
|
|
run_pm2 restart voicehub
|
|
echo -e "${GREEN}✓ PM2 服务已重启${NC}"
|
|
elif [[ "$service_type" == "systemd" ]]; then
|
|
sudo systemctl restart voicehub
|
|
echo -e "${GREEN}✓ systemctl 服务已重启${NC}"
|
|
elif [[ "$service_type" == "direct" ]]; then
|
|
stop_direct_service
|
|
start_direct_service
|
|
else
|
|
echo -e "${YELLOW}未找到服务配置，请先运行部署脚本配置服务${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 更新
|
|
cmd_update() {
|
|
check_project
|
|
if [[ -f "$PROJECT_DIR/sh/update.sh" ]]; then
|
|
bash "$PROJECT_DIR/sh/update.sh"
|
|
else
|
|
echo -e "${RED}错误: 找不到更新脚本 $PROJECT_DIR/sh/update.sh${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 重装依赖
|
|
cmd_deps() {
|
|
check_project
|
|
echo -e "${BLUE}重新安装依赖...${NC}"
|
|
|
|
rm -rf node_modules
|
|
pnpm install --frozen-lockfile || pnpm install
|
|
|
|
echo -e "${GREEN}✓ 依赖重装完成${NC}"
|
|
}
|
|
|
|
# 重新编译
|
|
cmd_build() {
|
|
check_project
|
|
echo -e "${BLUE}重新编译项目...${NC}"
|
|
|
|
pnpm run build
|
|
|
|
echo -e "${GREEN}✓ 编译完成${NC}"
|
|
}
|
|
|
|
# 重新安装
|
|
cmd_reinstall() {
|
|
check_project
|
|
echo -e "${BLUE}========================================${NC}"
|
|
echo -e "${BLUE} VoiceHub 重新安装${NC}"
|
|
echo -e "${BLUE}========================================${NC}"
|
|
echo ""
|
|
|
|
cmd_stop
|
|
|
|
echo -e "${YELLOW}[1/4] 更新代码...${NC}"
|
|
git fetch origin main
|
|
git stash
|
|
git reset --hard origin/main
|
|
echo -e "${GREEN}✓ 代码更新完成${NC}"
|
|
|
|
echo -e "${YELLOW}[2/4] 重装依赖...${NC}"
|
|
rm -rf node_modules
|
|
pnpm install --frozen-lockfile || pnpm install
|
|
echo -e "${GREEN}✓ 依赖重装完成${NC}"
|
|
|
|
echo -e "${YELLOW}[3/4] 重新编译...${NC}"
|
|
pnpm run build
|
|
echo -e "${GREEN}✓ 编译完成${NC}"
|
|
|
|
echo -e "${YELLOW}[4/4] 启动服务...${NC}"
|
|
cmd_start
|
|
|
|
echo ""
|
|
echo -e "${GREEN}========================================${NC}"
|
|
echo -e "${GREEN} 重新安装完成！${NC}"
|
|
echo -e "${GREEN}========================================${NC}"
|
|
}
|
|
|
|
# 查看日志
|
|
cmd_logs() {
|
|
check_project
|
|
|
|
local service_type=$(detect_service_manager)
|
|
|
|
if [[ "$service_type" == "pm2" ]]; then
|
|
run_pm2 logs voicehub
|
|
elif [[ "$service_type" == "systemd" ]]; then
|
|
sudo journalctl -u voicehub -f
|
|
elif [[ "$service_type" == "direct" ]]; then
|
|
mkdir -p "$DIRECT_LOG_DIR"
|
|
touch "$DIRECT_LOG_FILE"
|
|
tail -f "$DIRECT_LOG_FILE"
|
|
else
|
|
echo -e "${YELLOW}未找到服务配置，无法查看日志${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 交互式菜单
|
|
show_menu() {
|
|
# 菜单选项
|
|
local options=(
|
|
"查看服务状态"
|
|
"启动服务"
|
|
"停止服务"
|
|
"重启服务"
|
|
"更新"
|
|
"重新安装"
|
|
"重装依赖"
|
|
"重新编译"
|
|
"查看日志"
|
|
"退出"
|
|
)
|
|
|
|
local selected=0
|
|
local total=${#options[@]}
|
|
|
|
# 打印菜单
|
|
print_menu() {
|
|
clear
|
|
echo -e "${BLUE}========================================${NC}"
|
|
echo -e "${BLUE} VoiceHub 管理面板${NC}"
|
|
echo -e "${BLUE}========================================${NC}"
|
|
echo ""
|
|
|
|
for i in "${!options[@]}"; do
|
|
if [[ $i -eq $selected ]]; then
|
|
echo -e "${GREEN}► ${options[$i]}${NC}"
|
|
else
|
|
echo -e " ${options[$i]}"
|
|
fi
|
|
done
|
|
|
|
echo ""
|
|
echo -e "${YELLOW}使用 ↑↓ 选择，Enter 确认${NC}"
|
|
}
|
|
|
|
# 临时关闭 set -e，防止 read 超时或非零返回导致脚本退出
|
|
set +e
|
|
|
|
while true; do
|
|
print_menu
|
|
|
|
# 读取 1 个字符
|
|
read -rsn1 key
|
|
|
|
# 检查是否是转义序列 (方向键)
|
|
if [[ "$key" == $'\x1b' ]]; then
|
|
# 使用 || true 防止超时导致 set -e 触发（虽然已关闭，但保持习惯）
|
|
read -rsn2 -t 0.01 chars || true
|
|
key="$key$chars"
|
|
fi
|
|
|
|
case "$key" in
|
|
$'\x1b[A'|$'\x1bOA'|k|K) # 上
|
|
((selected--))
|
|
if [[ $selected -lt 0 ]]; then
|
|
selected=$((total - 1))
|
|
fi
|
|
;;
|
|
$'\x1b[B'|$'\x1bOB'|j|J) # 下
|
|
((selected++))
|
|
if [[ $selected -ge $total ]]; then
|
|
selected=0
|
|
fi
|
|
;;
|
|
"") # Enter
|
|
break
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# 恢复 set -e
|
|
set -e
|
|
|
|
echo ""
|
|
|
|
# 执行选择的操作
|
|
case $selected in
|
|
0)
|
|
cmd_status
|
|
;;
|
|
1)
|
|
cmd_start
|
|
;;
|
|
2)
|
|
cmd_stop
|
|
;;
|
|
3)
|
|
cmd_restart
|
|
;;
|
|
4)
|
|
cmd_update
|
|
;;
|
|
5)
|
|
cmd_reinstall
|
|
;;
|
|
6)
|
|
cmd_deps
|
|
;;
|
|
7)
|
|
cmd_build
|
|
;;
|
|
8)
|
|
cmd_logs
|
|
;;
|
|
9)
|
|
echo -e "${CYAN}退出${NC}"
|
|
exit 0
|
|
;;
|
|
esac
|
|
}
|
|
|
|
# 检查是否有交互式终端
|
|
if [[ ! -t 0 ]]; then
|
|
# 非交互模式，显示帮助
|
|
echo -e "${BLUE}VoiceHub 管理脚本${NC}"
|
|
echo ""
|
|
echo "用法: voicehub"
|
|
echo ""
|
|
echo "运行脚本后使用上下键选择操作"
|
|
echo ""
|
|
echo "或在非交互环境中使用以下命令："
|
|
echo " bash $0 status # 查看状态"
|
|
echo " bash $0 start # 启动服务"
|
|
echo " bash $0 stop # 停止服务"
|
|
echo " bash $0 restart # 重启服务"
|
|
echo " bash $0 update # 更新代码"
|
|
echo " bash $0 reinstall # 重新安装"
|
|
echo " bash $0 deps # 重装依赖"
|
|
echo " bash $0 build # 重新编译"
|
|
echo " bash $0 logs # 查看日志"
|
|
exit 0
|
|
fi
|
|
|
|
# 如果有参数，直接执行对应命令
|
|
if [[ -n "$1" ]]; then
|
|
case "$1" in
|
|
status)
|
|
cmd_status
|
|
;;
|
|
start)
|
|
cmd_start
|
|
;;
|
|
stop)
|
|
cmd_stop
|
|
;;
|
|
restart)
|
|
cmd_restart
|
|
;;
|
|
update)
|
|
cmd_update
|
|
;;
|
|
reinstall)
|
|
cmd_reinstall
|
|
;;
|
|
deps)
|
|
cmd_deps
|
|
;;
|
|
build)
|
|
cmd_build
|
|
;;
|
|
logs)
|
|
cmd_logs
|
|
;;
|
|
help|--help|-h)
|
|
echo -e "${BLUE}VoiceHub 管理脚本${NC}"
|
|
echo ""
|
|
echo "用法: voicehub"
|
|
echo ""
|
|
echo "直接运行脚本可使用交互式菜单（上下键选择）"
|
|
echo ""
|
|
echo "或使用以下命令："
|
|
echo " voicehub status # 查看状态"
|
|
echo " voicehub start # 启动服务"
|
|
echo " voicehub stop # 停止服务"
|
|
echo " voicehub restart # 重启服务"
|
|
echo " voicehub update # 更新代码"
|
|
echo " voicehub reinstall # 重新安装"
|
|
echo " voicehub deps # 重装依赖"
|
|
echo " voicehub build # 重新编译"
|
|
echo " voicehub logs # 查看日志"
|
|
;;
|
|
*)
|
|
echo -e "${RED}未知命令: $1${NC}"
|
|
exit 1
|
|
;;
|
|
esac
|
|
else
|
|
# 无参数，显示交互式菜单
|
|
show_menu
|
|
fi
|