#!/bin/bash
#
# DataOps Platform 重启脚本
# 使用 supervisorctl 重启 gunicorn 服务
#

set -e

# 配置变量
APP_NAME="dataops-platform"
APP_DIR="/opt/dataops-platform"
VENV_DIR="${APP_DIR}/venv"
LOG_DIR="${APP_DIR}/logs"

# 颜色输出
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

echo_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

echo_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# 检查虚拟环境是否存在
check_venv() {
    if [ ! -d "${VENV_DIR}" ]; then
        echo_error "虚拟环境不存在: ${VENV_DIR}"
        echo_info "请先运行部署脚本创建虚拟环境"
        exit 1
    fi
}

# 检查 supervisor 是否运行
check_supervisor() {
    if ! pgrep -x "supervisord" > /dev/null; then
        echo_warn "supervisord 未运行，正在启动..."
        sudo supervisord -c /etc/supervisor/supervisord.conf
        sleep 2
    fi
}

# 重启应用
restart_app() {
    echo_info "正在重启 ${APP_NAME}..."
    
    sudo supervisorctl restart ${APP_NAME}
    
    # 等待重启
    sleep 3
    
    # 检查状态
    status=$(sudo supervisorctl status ${APP_NAME} | awk '{print $2}')
    if [ "$status" = "RUNNING" ]; then
        echo_info "${APP_NAME} 重启成功!"
        sudo supervisorctl status ${APP_NAME}
    else
        echo_error "${APP_NAME} 重启失败!"
        echo_info "查看日志: tail -f ${LOG_DIR}/gunicorn_error.log"
        exit 1
    fi
}

# 健康检查
health_check() {
    echo_info "正在进行健康检查..."
    sleep 2
    
    response=$(curl -s -o /dev/null -w "%{http_code}" http://127.0.0.1:80/api/system/health 2>/dev/null || echo "000")
    
    if [ "$response" = "200" ]; then
        echo_info "健康检查通过! HTTP 状态码: ${response}"
    else
        echo_warn "健康检查返回状态码: ${response}"
        echo_info "服务可能需要更多时间启动，请稍后手动检查"
    fi
}

# 主函数
main() {
    echo "=========================================="
    echo "  DataOps Platform 重启脚本"
    echo "=========================================="
    
    check_venv
    check_supervisor
    restart_app
    health_check
    
    echo ""
    echo_info "重启完成!"
    echo_info "访问地址: http://localhost:80"
    echo_info "查看日志: tail -f ${LOG_DIR}/gunicorn_error.log"
}

main "$@"