#!/bin/bash
# 生产环境依赖安装脚本

echo "=== 生产环境依赖安装脚本 ==="
echo "当前Python版本: $(python3 --version)"
echo ""

# 方案1：安装编译工具（推荐）
echo "方案1：安装编译工具后使用标准requirements.txt"
echo "执行命令："
echo "yum install -y gcc gcc-c++ python3-devel  # CentOS/RHEL"
echo "# 或"
echo "apt-get update && apt-get install -y build-essential python3-dev  # Ubuntu/Debian"
echo ""
echo "然后执行："
echo "pip3 install --user -r requirements.txt"
echo ""

# 方案2：使用预编译版本
echo "方案2：使用预编译版本（如果方案1失败）"
echo "执行命令："
echo "pip3 install --user -r requirements-prod.txt"
echo ""

# 方案3：手动安装编译工具并重试
echo "方案3：一键安装编译工具并安装依赖"
echo "如果您同意，请输入 'yes' 继续："
read -r answer

if [ "$answer" = "yes" ] || [ "$answer" = "y" ]; then
    echo "检测操作系统类型..."

    if command -v yum &> /dev/null; then
        echo "检测到CentOS/RHEL系统，安装编译工具..."
        yum install -y gcc gcc-c++ python3-devel
    elif command -v apt-get &> /dev/null; then
        echo "检测到Ubuntu/Debian系统，安装编译工具..."
        apt-get update && apt-get install -y build-essential python3-dev
    else
        echo "无法自动检测操作系统类型，请手动安装编译工具"
        exit 1
    fi

    echo "安装Python依赖..."
    pip3 install --user -r requirements.txt

    if [ $? -eq 0 ]; then
        echo "依赖安装成功！"
        python3 test_dependencies.py
    else
        echo "标准版本安装失败，尝试预编译版本..."
        pip3 install --user -r requirements-prod.txt
    fi
else
    echo "请手动选择安装方案"
fi