#!/bin/bash

# PolyHermes 一体化部署脚本
# 将前后端一起部署到一个 Docker 容器中

set -e

# 颜色输出
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color
COMPOSE_FILES=(-f docker-compose.yml)

# 打印信息
info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

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

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

# 从环境变量或 .env 文件读取配置值，避免 source .env 时被特殊字符影响
get_config_value() {
    local key="$1"
    local value="${!key}"
    
    if [ -n "$value" ]; then
        echo "$value"
        return
    fi
    
    if [ -f ".env" ]; then
        grep "^${key}=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^["'\'']//;s/["'\'']$//' | tr -d '\r' || true
    fi
}

# 根据数据库配置选择 compose 文件
configure_compose_files() {
    local db_url
    db_url=$(get_config_value "DB_URL")
    
    COMPOSE_FILES=(-f docker-compose.yml)
    
    if [[ "$db_url" == *"host.docker.internal"* ]]; then
        COMPOSE_FILES=(-f docker-compose.host-mysql.yml)
        info "检测到 DB_URL 指向宿主机 MySQL，将使用 docker-compose.host-mysql.yml"
    else
        info "使用默认 docker-compose.yml（包含内置 MySQL 服务）"
    fi
}

compose() {
    docker-compose "${COMPOSE_FILES[@]}" "$@"
}

# 检查 Docker 环境
check_docker() {
    if ! command -v docker &> /dev/null; then
        error "Docker 未安装，请先安装 Docker"
        exit 1
    fi
    
    if ! command -v docker-compose &> /dev/null; then
        error "Docker Compose 未安装，请先安装 Docker Compose"
        exit 1
    fi
    
    info "Docker 环境检查通过"
}

# 生成随机字符串
generate_random_string() {
    local length=${1:-32}
    openssl rand -hex $length 2>/dev/null || \
    cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w $length | head -n 1
}

# 创建 .env 文件（如果不存在）
create_env_file() {
    if [ ! -f ".env" ]; then
        warn ".env 文件不存在，创建示例文件..."
        
        # 生成随机值
        DB_PASSWORD=$(generate_random_string 32)
        JWT_SECRET=$(generate_random_string 64)
        ADMIN_RESET_KEY=$(generate_random_string 32)
        
        cat > .env <<EOF
# 数据库配置
DB_URL=jdbc:mysql://mysql:3306/polyhermes?useSSL=false&serverTimezone=UTC&characterEncoding=utf8&allowPublicKeyRetrieval=true
DB_USERNAME=root
DB_PASSWORD=${DB_PASSWORD}

# Spring Profile
SPRING_PROFILES_ACTIVE=prod

# 服务器端口（对外暴露的端口）
SERVER_PORT=80

# MySQL 端口（可选，用于外部连接，默认 3307 避免与本地 MySQL 冲突）
MYSQL_PORT=3307

# JWT 密钥（已自动生成随机值，生产环境建议修改）
JWT_SECRET=${JWT_SECRET}

# 管理员密码重置密钥（已自动生成随机值，生产环境建议修改）
ADMIN_RESET_PASSWORD_KEY=${ADMIN_RESET_KEY}

# 日志级别配置（可选，默认值：root=WARN, app=INFO）
# 可选值：TRACE, DEBUG, INFO, WARN, ERROR, OFF
# LOG_LEVEL_ROOT=WARN
# LOG_LEVEL_APP=INFO
EOF
        info ".env 文件已创建，已自动生成随机密码和密钥"
        warn "生产环境建议修改以下参数："
        warn "  - DB_PASSWORD: 数据库密码（当前: ${DB_PASSWORD:0:8}...）"
        warn "  - JWT_SECRET: JWT 密钥（当前: ${JWT_SECRET:0:8}...）"
        warn "  - ADMIN_RESET_PASSWORD_KEY: 管理员密码重置密钥（当前: ${ADMIN_RESET_KEY:0:8}...）"
        exit 1
    fi
}

# 检查安全配置
check_security_config() {
    # 默认值常量
    DEFAULT_JWT_SECRET="change-me-in-production"
    DEFAULT_ADMIN_RESET_KEY="change-me-in-production"
    
    # 从 .env 文件读取配置（如果存在）
    local jwt_secret=""
    local admin_reset_key=""
    
    if [ -f ".env" ]; then
        # 从 .env 文件读取（使用 grep 和 sed 避免 source 可能的问题）
        jwt_secret=$(grep "^JWT_SECRET=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^"//;s/"$//' || echo "")
        admin_reset_key=$(grep "^ADMIN_RESET_PASSWORD_KEY=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^"//;s/"$//' || echo "")
    fi
    
    # 如果环境变量已设置，优先使用环境变量
    if [ -n "$JWT_SECRET" ]; then
        jwt_secret="$JWT_SECRET"
    fi
    if [ -n "$ADMIN_RESET_PASSWORD_KEY" ]; then
        admin_reset_key="$ADMIN_RESET_PASSWORD_KEY"
    fi
    
    local errors=0
    
    # 检查 JWT_SECRET
    if [ -z "$jwt_secret" ] || [ "$jwt_secret" = "$DEFAULT_JWT_SECRET" ]; then
        error "JWT_SECRET 不能使用默认值 '${DEFAULT_JWT_SECRET}'"
        error "请在 .env 文件中设置 JWT_SECRET 为安全的随机字符串"
        errors=$((errors + 1))
    fi
    
    # 检查 ADMIN_RESET_PASSWORD_KEY
    if [ -z "$admin_reset_key" ] || [ "$admin_reset_key" = "$DEFAULT_ADMIN_RESET_KEY" ]; then
        error "ADMIN_RESET_PASSWORD_KEY 不能使用默认值 '${DEFAULT_ADMIN_RESET_KEY}'"
        error "请在 .env 文件中设置 ADMIN_RESET_PASSWORD_KEY 为安全的随机字符串"
        errors=$((errors + 1))
    fi
    
    if [ $errors -gt 0 ]; then
        echo ""
        error "安全配置检查失败，部署已取消"
        echo ""
        info "提示：可以使用以下命令生成随机密钥："
        info "  openssl rand -hex 32  # 生成 32 字节的随机字符串（用于 ADMIN_RESET_PASSWORD_KEY）"
        info "  openssl rand -hex 64  # 生成 64 字节的随机字符串（用于 JWT_SECRET）"
        exit 1
    fi
    
    info "安全配置检查通过"
}

# 构建并启动
deploy() {
    # 检查安全配置
    check_security_config
    configure_compose_files
    
    # 检查是否使用 Docker Hub 镜像
    USE_DOCKER_HUB="${USE_DOCKER_HUB:-false}"
    
    if [ "$USE_DOCKER_HUB" = "true" ]; then
        info "使用 Docker Hub 镜像（推荐生产环境）..."
        info "拉取最新镜像..."
        docker pull wrbug/polyhermes:latest || warn "拉取镜像失败，将使用本地构建"
        
        # 修改 docker-compose.yml 使用镜像而不是构建
        # 注意：这里需要手动修改 docker-compose.yml，或者使用环境变量
        warn "请确保 docker-compose.yml 中已配置使用 image: wrbug/polyhermes:latest"
    else
        # 版本号：优先环境变量 DOCKER_VERSION，其次 .env 中的 DOCKER_VERSION，否则用当前分支名
        if [ -z "${DOCKER_VERSION}" ] && [ -f ".env" ]; then
            DOCKER_VERSION=$(grep "^DOCKER_VERSION=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^["'\'']//;s/["'\'']$//' | tr -d '\r')
        fi
        if [ -z "${DOCKER_VERSION}" ]; then
            CURRENT_BRANCH=$(git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "dev")
            DOCKER_VERSION=$(echo "$CURRENT_BRANCH" | tr '/' '-')
        fi
        export DOCKER_VERSION
        
        info "构建 Docker 镜像（本地构建，版本号: ${DOCKER_VERSION}）..."
        
        # 创建占位符目录（如果不存在），避免 Dockerfile COPY 失败
        # 当 BUILD_IN_DOCKER=true 时，backend/build 可能不存在
        mkdir -p backend/build/libs
        
        # 设置构建参数（通过环境变量传递给 docker-compose.yml）
        export VERSION=${DOCKER_VERSION}
        export GIT_TAG=${DOCKER_VERSION}
        export GITHUB_REPO_URL=https://github.com/WrBug/PolyHermes
        
        compose build
    fi
    
    info "启动服务..."
    compose up -d
    
    info "等待服务启动..."
    sleep 5
    
    info "检查服务状态..."
    compose ps
    
    info "查看日志: docker-compose ${COMPOSE_FILES[*]} logs -f"
    info "停止服务: docker-compose ${COMPOSE_FILES[*]} down"
}

# 主函数
main() {
    echo "=========================================="
    echo "  PolyHermes 一体化部署脚本"
    echo "=========================================="
    echo ""
    
    # 解析参数
    if [ "$1" = "--use-docker-hub" ] || [ "$1" = "-d" ]; then
        export USE_DOCKER_HUB=true
        info "将使用 Docker Hub 镜像（生产环境推荐）"
        echo ""
    fi
    
    check_docker
    create_env_file
    deploy
    
    echo ""
    info "部署完成！"
    info "访问地址: http://localhost:${SERVER_PORT:-80}"
    echo ""
    if [ "$USE_DOCKER_HUB" != "true" ]; then
        if [ -z "${DOCKER_VERSION}" ] && [ -f ".env" ]; then
            DOCKER_VERSION=$(grep "^DOCKER_VERSION=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^["'\'']//;s/["'\'']$//' | tr -d '\r')
        fi
        if [ -z "${DOCKER_VERSION}" ]; then
            CURRENT_BRANCH=$(git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "dev")
            DOCKER_VERSION=$(echo "$CURRENT_BRANCH" | tr '/' '-')
        fi
        info "提示：本地构建的版本号: ${DOCKER_VERSION}（可在 .env 或环境变量中设置 DOCKER_VERSION）"
        info "生产环境推荐使用 Docker Hub 镜像："
        info "  ./deploy.sh --use-docker-hub"
        info "  或修改 docker-compose.yml 使用 image: wrbug/polyhermes:latest"
    fi
}

main "$@"