#!/bin/bash

# 设置全局变量
WORK_DIR="$(dirname "$0")"
# /home/fastapi_vue3_admin

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

# 检查权限
check_permissions() {
    log "🚀 第一步: 检查权限..."
    # 检查脚本文件是否有执行权限
    if [ ! -x "$0" ]; then
        log "⚠️ 当前脚本没有执行权限，尝试添加执行权限..."
        chmod +x "$0"
        if [ $? -eq 0 ]; then
            log "✅ 已成功为脚本添加执行权限"
        else
            log "❌ 为脚本添加执行权限失败"
            exit 1
        fi
    else
        log "✅ 脚本已有执行权限"
    fi
}

# 检查依赖
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; }
    git pull --force || { log "❌ 拉取更新失败，请检查网络或仓库权限"; exit 1; }
    git log -1 || { log "❌ 获取提交信息失败"; exit 1; }
    log "✅ 代码更新成功"
}

# 停止并删除容器
stop_and_remove_containers() {
    log "🚀 第四步: 终止容器..."
    cd "${WORK_DIR}" || { log "❌ 无法进入工作目录：${WORK_DIR}"; exit 1; }
    [ -f "docker-compose.yaml" ] || { log "❌ docker-compose.yaml 文件未找到"; exit 1; }
    docker compose down
    log "✅ 容器已停止并删除"
}

# 构建前端
build_frontend() {
    log "🚀 第五步: 构建前端..."
    cd "${WORK_DIR}/frontend" || { log "❌ 无法进入前端目录"; exit 1; }
    if [ ! -f "package.json" ]; then
        log "❌ 前端项目缺少 package.json 文件"
        exit 1
    fi
    if [ ! -d "dist" ] || [ ! -d "node_modules" ] || [ "$(git diff --name-only HEAD~1 HEAD -- frontend/)" ]; then
        npm install || { log "❌ 前端依赖安装失败"; exit 1; }
        npm run build || { log "❌ 前端工程打包失败"; exit 1; }
        log "✅ 前端工程打包成功"
    else
        log "⚠️ 未检测到前端代码变更且非首次克隆，跳过前端构建"
    fi
}

# 构建镜像
build_image() {
    log "🚀 第六步: 构建镜像..."
    cd "${WORK_DIR}" || { log "❌ 无法进入工作目录：${WORK_DIR}"; exit 1; }
    docker compose build || { log "❌ 镜像构建失败"; exit 1; }
    log "✅ Docker镜像构建成功"
}

# 启动容器
start_containers() {
    log "🚀 第七步: 启动容器..."
    cd "${WORK_DIR}" || { log "❌ 无法进入工作目录：${WORK_DIR}"; exit 1; }
    docker compose up -d --force-recreate || { 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
    cleanup_old_images
    log "==========🎉 部署完成=========="
}

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