#!/usr/bin/env bash

set -euo pipefail

# 永远从脚本所在目录运行（避免在别的目录执行导致 requirements/.env/venv 路径错误）
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

# 函数：加载 .env 文件中的变量
load_env_file() {
    local env_file=".env"
    if [ -f "$env_file" ]; then
        while IFS='=' read -r key value; do
            # 跳过注释和空行
            [[ "$key" =~ ^#.*$ ]] && continue
            [[ -z "$key" ]] && continue
            
            # 移除值两端的引号（如果有）
            value=$(echo "$value" | sed 's/^["'\'']//' | sed 's/["'\'']$//')
            
            # 导出变量
            export "$key=$value"
        done < "$env_file"
    fi
}

echo "🔧 X-Request 框架环境设置"
echo "=========================="

# 检查当前操作系统
# 首先检查是否是Windows环境
if [[ "$OSTYPE" == "msys" || "$OSTYPE" == "cygwin" || "$OSTYPE" == "win32" ]]; then
    OS_TYPE="Windows"
    PYTHON_CMD="python"
elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
    OS_TYPE="Linux"
    PYTHON_CMD="python3"
elif [[ "$OSTYPE" == "darwin"* ]]; then
    OS_TYPE="macOS"
    PYTHON_CMD="python3"
else
    # 使用uname作为后备检测
    OS_TYPE="$(uname -s)"
    if [[ "$OS_TYPE" == *"Windows"* || "$OS_TYPE" == "MSYS"* || "$OS_TYPE" == "MINGW"* ]]; then
        OS_TYPE="Windows"
        PYTHON_CMD="python"
    else
        PYTHON_CMD="python3"
    fi
fi

# 额外检查是否在Windows PowerShell或CMD中运行
if [[ "$SHELL" == *"powershell"* || "$SHELL" == *"cmd.exe"* ]]; then
    OS_TYPE="Windows"
    PYTHON_CMD="python"
fi

# 检查是否存在python.exe文件（Windows特有）
if [[ -f "$(which python.exe 2>/dev/null)" ]]; then
    OS_TYPE="Windows"
    PYTHON_CMD="python.exe"
fi

echo "📋 检测到操作系统: $OS_TYPE"
echo "📋 使用Python命令: $PYTHON_CMD"

# 检查Python版本
echo "📋 检查Python版本..."
python_version=$($PYTHON_CMD --version 2>&1 | grep -Po '(?<=Python )\d+\.\d+')
if [ -z "$python_version" ]; then
    python_version=$($PYTHON_CMD --version 2>&1 | awk '{print $2}' | cut -d'.' -f1,2)
fi

echo "✅ 发现Python: $python_version"

# 简单的版本比较（不需要bc命令）
major_version=$(echo $python_version | cut -d'.' -f1)
minor_version=$(echo $python_version | cut -d'.' -f2)

if [ "$major_version" -lt 3 ] || ([ "$major_version" -eq 3 ] && [ "$minor_version" -lt 8 ]); then
    echo "❌ 需要 Python 3.8 或更高版本，当前版本: $python_version"
    exit 1
fi

echo "✅ Python版本检查通过"

# 检查虚拟环境是否存在且完整
VENV_COMPLETE=false
if [ -d "xrequest" ]; then
    # 检查激活脚本是否存在
    if [ -f "xrequest/Scripts/activate" ] || [ -f "xrequest/bin/activate" ]; then
        echo "✅ 虚拟环境已存在且完整"
        VENV_COMPLETE=true
    else
        echo "⚠️  虚拟环境存在但不完整，删除并重新创建..."
        rm -rf xrequest
    fi
fi

# 如果虚拟环境不存在或不完整，创建新的虚拟环境
if [ "$VENV_COMPLETE" = false ]; then
    echo "📦 创建虚拟环境..."
    
    # 根据操作系统类型选择不同的虚拟环境创建方式
    if [ "$OS_TYPE" = "Windows" ]; then
        # 在Windows上使用python -m venv
        $PYTHON_CMD -m venv xrequest
        
        if [ $? -eq 0 ]; then
            # 检查虚拟环境是否创建成功
            if [ -f "xrequest/Scripts/activate" ]; then
                echo "✅ 虚拟环境创建完成"
            else
                echo "⚠️  虚拟环境创建可能失败，尝试使用pip创建..."
                # 如果venv创建失败，尝试使用pip安装virtualenv并创建虚拟环境
                $PYTHON_CMD -m pip install --user virtualenv
                $PYTHON_CMD -m virtualenv xrequest
                if [ $? -eq 0 ]; then
                    if [ -f "xrequest/Scripts/activate" ]; then
                        echo "✅ 虚拟环境创建完成（使用virtualenv）"
                    else
                        echo "❌ 虚拟环境创建失败，请手动创建虚拟环境"
                        echo "   手动创建方法："
                        echo "   1. 安装virtualenv：pip install virtualenv"
                        echo "   2. 创建虚拟环境：virtualenv xrequest"
                        echo "   3. 激活虚拟环境：xrequest/Scripts/activate"
                        echo "   4. 安装依赖：pip install -r requirements.txt"
                    fi
                else
                    echo "❌ 虚拟环境创建失败，请手动创建虚拟环境"
                    echo "   手动创建方法："
                    echo "   1. 安装virtualenv：pip install virtualenv"
                    echo "   2. 创建虚拟环境：virtualenv xrequest"
                    echo "   3. 激活虚拟环境：xrequest/Scripts/activate"
                    echo "   4. 安装依赖：pip install -r requirements.txt"
                fi
            fi
        else
            echo "❌ 虚拟环境创建失败，请手动创建虚拟环境"
            echo "   手动创建方法："
            echo "   1. 安装virtualenv：pip install virtualenv"
            echo "   2. 创建虚拟环境：virtualenv xrequest"
            echo "   3. 激活虚拟环境：xrequest/Scripts/activate"
            echo "   4. 安装依赖：pip install -r requirements.txt"
        fi
    else
        # 在Linux/Mac上使用python3 -m venv
        $PYTHON_CMD -m venv xrequest
        
        if [ $? -eq 0 ]; then
            # 检查虚拟环境是否创建成功
            if [ -f "xrequest/bin/activate" ]; then
                echo "✅ 虚拟环境创建完成"
            else
                echo "⚠️  虚拟环境创建可能失败，尝试使用pip创建..."
                # 如果venv创建失败，尝试使用pip安装virtualenv并创建虚拟环境
                $PYTHON_CMD -m pip install --user virtualenv
                $PYTHON_CMD -m virtualenv xrequest
                if [ $? -eq 0 ]; then
                    if [ -f "xrequest/bin/activate" ]; then
                        echo "✅ 虚拟环境创建完成（使用virtualenv）"
                    else
                        echo "❌ 虚拟环境创建失败，请手动创建虚拟环境"
                        echo "   手动创建方法："
                        echo "   1. 安装virtualenv：pip install virtualenv"
                        echo "   2. 创建虚拟环境：virtualenv xrequest"
                        echo "   3. 激活虚拟环境：source xrequest/bin/activate"
                        echo "   4. 安装依赖：pip install -r requirements.txt"
                    fi
                else
                    echo "❌ 虚拟环境创建失败，请手动创建虚拟环境"
                    echo "   手动创建方法："
                    echo "   1. 安装virtualenv：pip install virtualenv"
                    echo "   2. 创建虚拟环境：virtualenv xrequest"
                    echo "   3. 激活虚拟环境：source xrequest/bin/activate"
                    echo "   4. 安装依赖：pip install -r requirements.txt"
                fi
            fi
        else
            echo "❌ 虚拟环境创建失败，请手动创建虚拟环境"
            echo "   在Linux上，您可能需要先安装python3-venv包："
            echo "   sudo apt install python3.12-venv"  # 针对Ubuntu/Debian系统
            echo "   或者："
            echo "   1. 安装virtualenv：pip install virtualenv"
            echo "   2. 创建虚拟环境：virtualenv xrequest"
            echo "   3. 激活虚拟环境：source xrequest/bin/activate"
            echo "   4. 安装依赖：pip install -r requirements.txt"
        fi
    fi
fi

# 确定激活脚本路径
# 检查是否存在 Windows 风格的激活脚本
if [ -f "xrequest/Scripts/activate" ]; then
    ACTIVATE_SCRIPT="xrequest/Scripts/activate"
    echo "📋 使用 Windows 风格的激活脚本: $ACTIVATE_SCRIPT"
elif [ -f "xrequest/bin/activate" ]; then
    ACTIVATE_SCRIPT="xrequest/bin/activate"
    echo "📋 使用 Linux/Mac 风格的激活脚本: $ACTIVATE_SCRIPT"
else
    ACTIVATE_SCRIPT=""
    echo "⚠️  未找到激活脚本"
fi

# 安装依赖（强制使用虚拟环境里的 python -m pip；不要因为 curl 检测失败而跳过安装）
echo "📦 安装依赖包..."

VENV_PY=""
if [ -f "xrequest/Scripts/python.exe" ]; then
    VENV_PY="xrequest/Scripts/python.exe"
elif [ -f "xrequest/bin/python" ]; then
    VENV_PY="xrequest/bin/python"
fi

if [ -z "$VENV_PY" ]; then
    echo "❌ 未找到虚拟环境 Python，无法安装依赖。请先确保虚拟环境创建成功。"
    exit 1
fi

echo "📋 使用虚拟环境 Python: $VENV_PY"

echo "  📦 升级 pip/setuptools/wheel..."
$VENV_PY -m pip install --upgrade pip setuptools wheel

echo "  📚 使用 requirements.txt 安装所有依赖..."
$VENV_PY -m pip install -r requirements.txt

echo "  🔎 校验关键依赖..."
$VENV_PY -c "import uvicorn, fastapi; print('OK: uvicorn/fastapi installed')"

echo "✅ 依赖安装完成！"

# 加载 .env 文件中的变量
echo "📄 加载环境配置..."
load_env_file

# 创建日志目录（使用 .env 中的 LOGS_DIR 配置）
echo "📁 创建日志目录..."
mkdir -p "${LOGS_DIR:-logs}"

# 检查是否存在.env文件，如果不存在则创建
if [ ! -f ".env" ]; then
    echo "📄 创建环境配置文件..."
    if [ -f ".env.example" ]; then
        cp .env.example .env
        echo "✅ 已从 .env.example 创建 .env 文件，可根据需要修改配置"
    else
        # 如果没有示例文件，创建一个基本的.env文件
        cat > .env << EOF
# 应用配置
APP_NAME="X-Request API Framework"
APP_VERSION="1.0.0"
DEBUG=false

# 服务器配置
HOST="0.0.0.0"
PORT=1111
WORKERS=1

# 日志配置
LOG_LEVEL="INFO"
LOG_FILE="logs/app.log"
LOG_FORMAT="json"
LOG_TO_CONSOLE=false

# 高级日志配置
ADVANCED_LOGGING=true
LOGS_DIR="logs"
MAX_LOG_DAYS=30
ENABLE_LOG_CLEANUP=true
ROUTE_BASED_LOGGING=true

# 性能配置
MAX_REQUESTS=1000
MAX_CONNECTIONS=1000
REQUEST_TIMEOUT=30

# CORS配置
CORS_ORIGINS=["*"]
CORS_METHODS=["*"]
CORS_HEADERS=["*"]
EOF
        echo "✅ 已创建基本的 .env 文件，可根据需要修改配置"
    fi
fi

echo ""
echo "🎉 环境设置完成！"
echo ""
echo "🚀 启动方法："
echo "   ./start.sh"
echo "   或者:"
echo "   source xrequest/bin/activate && python main.py"
echo ""
echo "📚 API文档地址:"
echo "   http://localhost:${PORT:-3000}/docs"
echo "🏥 健康检查:"
echo "   http://localhost:${PORT:-3000}/health"
echo ""
echo ""