#!/bin/bash
# Spug 自动发布脚本 - 全栈项目
# 用途：医院绩效考核系统自动化部署脚本
# 执行方式：在 Spug 的"执行任务"中调用此脚本

set -e

# ==================== 配置参数 ====================
# 这些参数可以在 Spug 中通过环境变量传入
# 如果 SPUG_DEPLOY_DIR 为空，则使用默认值
if [ -z "${SPUG_DEPLOY_DIR}" ]; then
    PROJECT_DIR="/var/www/hospital-performance"
else
    PROJECT_DIR="${SPUG_DEPLOY_DIR}"
fi

PROJECT_NAME="${SPUG_APP_NAME:-hospital-performance}"
BACKUP_DIR="${PROJECT_DIR}/backups"
FRONTEND_DIR="${PROJECT_DIR}/frontend"
BACKEND_DIR="${PROJECT_DIR}/backend"

# Git 配置（Spug 会自动设置 SPUG_GIT_URL）
if [ -n "${SPUG_GIT_URL}" ]; then
    GIT_REPO="${SPUG_GIT_URL}"
elif [ -n "${SPUG_GIT_REPO}" ]; then
    GIT_REPO="${SPUG_GIT_REPO}"
else
    GIT_REPO="https://gitea.gentronhealth.com/chenqi/hospital_performance.git"
fi

GIT_BRANCH="${SPUG_GIT_BRANCH:-main}"

# Python 虚拟环境
VENV_DIR="${PROJECT_DIR}/venv"
PYTHON_VERSION="${PYTHON_VERSION:-python3.10}"

# Node.js 配置
NODE_VERSION="${NODE_VERSION:-18}"

# 服务配置
BACKEND_SERVICE="${BACKEND_SERVICE:-hospital-backend}"
BACKEND_PORT="${BACKEND_PORT:-8000}"
FRONTEND_SERVICE="${FRONTEND_SERVICE:-nginx}"

# 日志配置
if [ -z "${LOG_FILE}" ]; then
    LOG_FILE="/var/log/spug/deploy.log"
    # 确保日志目录存在
    mkdir -p /var/log/spug 2>/dev/null || true
fi
DEPLOY_TIME=$(date +"%Y%m%d_%H%M%S")

# ==================== 工具函数 ====================
log() {
    local level=$1
    shift
    local message="$*"
    local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
    echo "[${timestamp}] [${level}] ${message}" | tee -a "${LOG_FILE}"
}

info() {
    log "INFO" "$@"
}

error() {
    log "ERROR" "$@"
}

success() {
    log "SUCCESS" "$@"
}

check_command() {
    if ! command -v $1 &> /dev/null; then
        error "命令 $1 未安装，请先安装"
        exit 1
    fi
}

# ==================== 前置检查 ====================
pre_check() {
    info "========== 开始前置检查 =========="
    
    # 检查必要命令
    check_command git
    check_command python3
    check_command node
    check_command npm
    
    # 检查部署目录
    if [ ! -d "${PROJECT_DIR}" ]; then
        info "创建部署目录：${PROJECT_DIR}"
        mkdir -p "${PROJECT_DIR}"
    fi
    
    # 检查磁盘空间（如果目录存在）
    if [ -d "${PROJECT_DIR}" ]; then
        local available_space=$(df -P "${PROJECT_DIR}" 2>/dev/null | awk 'NR==2 {print $4}')
        if [ -n "${available_space}" ] && [ "${available_space}" -lt 1048576 ]; then
            error "磁盘空间不足 1GB，当前可用：${available_space}KB"
            exit 1
        fi
    fi
    
    info "✓ 前置检查通过"
}

# ==================== 代码更新 ====================
update_code() {
    info "========== 更新代码 =========="
    
    cd "${PROJECT_DIR}"
    
    if [ ! -d ".git" ]; then
        info "首次部署，克隆仓库..."
        # 确保目录为空或不存在
        if [ "$(ls -A ${PROJECT_DIR} 2>/dev/null)" ]; then
            info "目录非空，先备份现有文件..."
            mkdir -p "${PROJECT_DIR}/backup_$(date +%Y%m%d_%H%M%S)_non_git"
            find "${PROJECT_DIR}" -maxdepth 1 -type f -o -type d ! -name '.' -exec mv {} "${PROJECT_DIR}/backup_$(date +%Y%m%d_%H%M%S)_non_git/" \; 2>/dev/null || true
        fi
        git clone "${GIT_REPO}" .
        git checkout "${GIT_BRANCH}"
    else
        info "更新现有代码..."
        git fetch origin "${GIT_BRANCH}"
        git reset --hard "origin/${GIT_BRANCH}"
        git clean -fd
    fi
    
    local commit_hash=$(git rev-parse --short HEAD)
    local commit_msg=$(git log -1 --pretty=format:"%s")
    info "✓ 代码更新完成，当前版本：${commit_hash} - ${commit_msg}"
}

# ==================== 备份旧版本 ====================
backup_old_version() {
    info "========== 备份当前版本 =========="
    
    mkdir -p "${BACKUP_DIR}"
    local backup_name="backup_${DEPLOY_TIME}"
    local backup_path="${BACKUP_DIR}/${backup_name}"
    
    # 备份关键目录
    if [ -d "${BACKEND_DIR}" ]; then
        cp -r "${BACKEND_DIR}" "${backup_path}_backend"
        info "✓ 后端代码已备份到：${backup_path}_backend"
    fi
    
    if [ -d "${FRONTEND_DIR}/dist" ]; then
        cp -r "${FRONTEND_DIR}/dist" "${backup_path}_frontend_dist"
        info "✓ 前端构建已备份到：${backup_path}_frontend_dist"
    fi
    
    # 清理 30 天前的备份
    find "${BACKUP_DIR}" -type d -name "backup_*" -mtime +30 -exec rm -rf {} \; 2>/dev/null || true
    
    info "✓ 备份完成（保留最近 30 天）"
}

# ==================== 后端部署 ====================
deploy_backend() {
    info "========== 部署后端服务 =========="
    
    cd "${BACKEND_DIR}"
    
    # 创建虚拟环境
    if [ ! -d "${VENV_DIR}" ]; then
        info "创建 Python 虚拟环境..."
        ${PYTHON_VERSION} -m venv "${VENV_DIR}"
    fi
    
    # 激活虚拟环境
    source "${VENV_DIR}/bin/activate"
    
    # 升级 pip
    pip install --upgrade pip -q
    
    # 安装依赖
    info "安装 Python 依赖..."
    pip install -r requirements.txt -q
    
    # 数据库迁移
    if [ -f "alembic.ini" ]; then
        info "执行数据库迁移..."
        alembic upgrade head
    fi
    
    # 初始化数据（如果存在）
    if [ -f "init_db.py" ]; then
        info "初始化数据库..."
        python init_db.py || true
    fi
    
    # 重启后端服务
    if systemctl list-units --type=service --all | grep -q "${BACKEND_SERVICE}"; then
        info "重启后端服务..."
        systemctl restart "${BACKEND_SERVICE}"
        sleep 2
        
        # 检查服务状态
        if systemctl is-active --quiet "${BACKEND_SERVICE}"; then
            success "✓ 后端服务重启成功"
        else
            error "✗ 后端服务启动失败"
            exit 1
        fi
    else
        info "未找到 systemd 服务，跳过重启"
    fi
    
    deactivate
    info "✓ 后端部署完成"
}

# ==================== 前端部署 ====================
deploy_frontend() {
    info "========== 部署前端服务 =========="
    
    cd "${FRONTEND_DIR}"
    
    # 安装依赖
    info "安装 Node.js 依赖..."
    npm install --production
    
    # 构建前端
    info "构建前端项目..."
    npm run build
    
    # 检查构建结果
    if [ -d "dist" ] && [ "$(ls -A dist)" ]; then
        success "✓ 前端构建成功"
    else
        error "✗ 前端构建失败，dist 目录为空"
        exit 1
    fi
    
    # 如果使用 Nginx，重新加载配置
    if systemctl list-units --type=service --all | grep -q "${FRONTEND_SERVICE}"; then
        info "重新加载 Nginx..."
        systemctl reload "${FRONTEND_SERVICE}"
    fi
    
    info "✓ 前端部署完成"
}

# ==================== 健康检查 ====================
health_check() {
    info "========== 执行健康检查 =========="
    
    # 等待服务启动
    sleep 5
    
    # 检查后端 API
    local backend_url="http://localhost:${BACKEND_PORT}/api/v1/health"
    if curl -f -s "${backend_url}" > /dev/null; then
        success "✓ 后端 API 健康检查通过"
    else
        error "✗ 后端 API 健康检查失败"
        # 回滚逻辑（可选）
        # rollback
        exit 1
    fi
    
    # 检查前端
    if [ -d "${FRONTEND_DIR}/dist/index.html" ]; then
        success "✓ 前端文件存在"
    else
        error "✗ 前端文件缺失"
        exit 1
    fi
    
    info "✓ 所有健康检查通过"
}

# ==================== 清理工作 ====================
cleanup() {
    info "========== 清理临时文件 =========="
    
    # 清理 npm 缓存
    npm cache clean --force 2>/dev/null || true
    
    # 清理 Python 缓存
    find "${PROJECT_DIR}" -type d -name "__pycache__" -exec rm -rf {} \; 2>/dev/null || true
    find "${PROJECT_DIR}" -type f -name "*.pyc" -delete 2>/dev/null || true
    
    info "✓ 清理完成"
}

# ==================== 主流程 ====================
main() {
    info "========================================"
    info "Spug 自动发布开始"
    info "项目名称：${PROJECT_NAME}"
    info "部署目录：${PROJECT_DIR}"
    info "Git 分支：${GIT_BRANCH}"
    info "========================================"
    
    # 执行部署流程
    pre_check
    update_code
    backup_old_version
    deploy_backend
    deploy_frontend
    health_check
    cleanup
    
    info "========================================"
    success "🎉 部署成功完成！"
    info "========================================"
}

# 执行主流程
main "$@"