#!/bin/bash

# 设置全局变量
PROJECT_NAME="fastapi_vue3_admin"
WORK_DIR="/home"
GIT_REPO="https://gitee.com/tao__tao/fastapi_vue3_admin.git"
WEB_URL="http://8.137.99.5:80"

# 打印带时间戳的日志
log() {
    echo "[$(date '+%Y-%m-%d %H:%M:%S')] $1"
}

# 检查权限
check_permissions() {
    log "==========🔍 第一步：检查权限...=========="
    [ "$(id -u)" = "0" ] || { log "❌ 需要 root 权限"; exit 1; }
    log "✅ 权限检查通过"
}

# 检查依赖
check_dependencies() {
    log "==========🔍 第二步：检查系统依赖...=========="
    for cmd in git docker node npm; do
        command -v $cmd &> /dev/null || { log "❌ $cmd 未安装"; exit 1; }
        log "🎉 $cmd 已安装 - $($cmd -v)"
    done
    log "✅ 所有依赖检查通过"
}

# 更新代码
update_code() {
    log "==========🔍 第三步：检查项目代码...=========="
    cd "${WORK_DIR}" || { log "❌ 无法进入工作目录：${WORK_DIR}"; exit 1; }
    if [ -d "${PROJECT_NAME}/" ]; then
        log "🔄 项目已存在，开始更新代码"
        cd "${PROJECT_NAME}" || { log "❌ 无法进入项目目录：${PROJECT_NAME}"; exit 1; }
        git pull origin master || { log "❌ 拉取更新失败"; exit 1; }
        git log -1 || { log "❌ 获取提交信息失败"; exit 1; }
        log "✅ 代码更新成功"
    else
        log "📥 项目不存在，开始克隆代码"
        git clone "${GIT_REPO}" || { log "❌ 项目克隆失败：${GIT_REPO}"; exit 1; }
        cd "${PROJECT_NAME}" || { log "❌ 无法进入项目目录：${PROJECT_NAME}"; exit 1; }
        log "✅ 代码克隆成功"
    fi
}

# 停止并删除容器
stop_and_remove_containers() {
    log "==========🗑️ 第四五步：停止并删除现有容器...=========="
    [ -f "docker-compose.yaml" ] || { log "❌ docker-compose.yaml 文件未找到"; exit 1; }
    docker compose down
    log "✅ 容器已停止并删除"
}

# 构建前端
build_frontend() {
    log "==========🔍 第五步：开始构建前端代码...=========="
    # 如果是首次克隆项目，或者检测到前端代码变更，则构建前端
    if [ ! -d "frontend/dist" ] || [ "$(git diff --name-only HEAD~1 HEAD -- frontend/)" ]; then
        log "🚀 检测到前端代码变更或首次克隆，开始构建前端..."
        cd frontend || { log "❌ 无法进入前端目录"; exit 1; }
        npm install || { log "❌ 前端依赖安装失败"; exit 1; }
        npm run build || { log "❌ 前端工程打包失败"; exit 1; }
        log "✅ 前端工程打包成功"
        cd .. || { log "❌ 无法返回项目根目录"; exit 1; }
    else
        log "⚠️ 未检测到前端代码变更且非首次克隆，跳过前端构建"
    fi
}

# 构建镜像
build_image() {
    log "==========🚀 第六步：开始构建Docker镜像...==========🗑️ "
    docker compose build || { log "❌ 镜像构建失败"; exit 1; }
    log "✅  Docker镜像构建成功"
}

# 启动容器
start_containers() {
    log "==========🚀 第七步：启动容器...==========🗑️ "
    docker compose up -d || { log "❌ 容器启动失败"; exit 1; }
    log "✅  容器启动成功"
}

# 健康检查
health_check() {
    log "==========🔍 第八步：进行健康检查...==========🗑️ "
    sleep 10  # 等待容器启动
    curl --output /dev/null --silent --head --fail "${WEB_URL}" || { log "❌ 前端服务健康检查失败"; exit 1; }
    log "✅ 服务健康检查通过"
}

# 清理旧镜像
cleanup_old_images() {
    log "==========🗑️ 第九步：清理24小时前的旧镜像...==========🗑️ "
    docker image prune -f --filter "until=24h"
    log "✅ 旧镜像清理完成"
}

# 主函数
main() {
    log "==========🚀 开始部署流程=========="
    check_permissions
    check_dependencies
    update_code
    stop_and_remove_containers
    build_frontend
    build_image
    start_containers
    health_check
    cleanup_old_images
    log "==========🎉 【部署完成】访问地址: ${WEB_URL}=========="
}

main
trap 'log "==========⚠️ 脚本中断=========="; exit 1' INT TERM