easyai/update.sh

#!/bin/bash

set -e  # 发生错误时终止脚本执行

# 仅支持 -h/--help 快速查看
if [[ "${1:-}" =~ ^(-h|--help)$ ]]; then
    echo "用法: $0"
    echo ""
    echo "默认拉取仓库并更新镜像；也可通过环境变量跳过交互选择。"
    echo ""
    echo "环境变量:"
    echo "  UPDATE_MODE=full|image"
    echo "      full  更新并拉取仓库（git pull）+ 更新镜像并重启"
    echo "      image 仅更新镜像并重启（跳过 git pull）"
    echo "  UPDATE_SKIP_REPO_UPDATE=true|false"
    echo "      true  等同 UPDATE_MODE=image"
    echo "      false 等同 UPDATE_MODE=full"
    exit 0
fi

# 进入脚本所在目录（项目根目录）
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]:-$0}")" && pwd)"
cd "$SCRIPT_DIR"

# 命令行内选择更新方式（默认：更新并拉取仓库）
SKIP_REPO_UPDATE=false

normalize_bool() {
    local value
    value="$(printf '%s' "$1" | tr '[:upper:]' '[:lower:]')"
    case "$value" in
        1|true|yes|y|on) echo "true" ;;
        0|false|no|n|off|"") echo "false" ;;
        *)
            echo "❌ 布尔变量值不合法: $1（支持 true/false/1/0/yes/no/y/n/on/off）" >&2
            return 1
            ;;
    esac
}

resolve_update_mode() {
    local mode
    if [ -n "${UPDATE_SKIP_REPO_UPDATE:-}" ]; then
        mode="$(normalize_bool "$UPDATE_SKIP_REPO_UPDATE")" || exit 1
        SKIP_REPO_UPDATE="$mode"
        echo "使用环境变量: UPDATE_SKIP_REPO_UPDATE=$SKIP_REPO_UPDATE"
        return
    fi

    if [ -n "${UPDATE_MODE:-}" ]; then
        mode="$(printf '%s' "$UPDATE_MODE" | tr '[:upper:]' '[:lower:]')"
        case "$mode" in
            full|repo|pull|git|1)
                SKIP_REPO_UPDATE=false
                ;;
            image|images|docker|skip-git|skip_repo|skip-repo|2)
                SKIP_REPO_UPDATE=true
                ;;
            *)
                echo "❌ UPDATE_MODE 只能是 full 或 image，当前为: $UPDATE_MODE"
                exit 1
                ;;
        esac
        echo "使用环境变量: UPDATE_MODE=$UPDATE_MODE"
        return
    fi

    if [ "${UPDATE_NON_INTERACTIVE:-0}" = "1" ] || [ "${CI:-}" = "true" ] || [ ! -t 0 ]; then
        echo "使用默认更新方式: 更新并拉取仓库（git pull）+ 更新镜像并重启"
        SKIP_REPO_UPDATE=false
        return
    fi

    echo ""
    echo "请选择更新方式："
    echo "   [1] 更新并拉取仓库（git pull）+ 更新镜像并重启（默认）"
    echo "   [2] 仅更新镜像并重启（跳过 git pull）"
    read -r -p "请选择 [1/2，回车默认 1]: " choice
    choice="${choice:-1}"
    case "$choice" in
        2) SKIP_REPO_UPDATE=true ;;
        1) ;;
        *) echo "❌ 无效选择，将使用默认：更新并拉取仓库"; SKIP_REPO_UPDATE=false ;;
    esac
}

resolve_update_mode

# 拉取整个仓库更新
if [ "$SKIP_REPO_UPDATE" = false ]; then
    echo ""
    echo "==========================="
    echo "📥 拉取仓库最新代码"
    echo "==========================="

    if [ ! -d .git ]; then
        echo "❌ 当前目录不是 Git 仓库，无法执行 git pull"
        echo "   请使用 git clone 克隆项目后即可使用 update.sh 更新"
        echo "   克隆命令: git clone https://git.51easyai.com/wangbo/easyai && cd easyai"
        exit 1
    fi

    echo "📥 正在执行 git pull..."
    if git pull; then
        echo "✅ 仓库已更新到最新版本"
    else
        echo "❌ git pull 失败，请检查网络或远程仓库配置"
        exit 1
    fi

    # 确保环境配置文件存在（从 .sample 生成，不覆盖已有文件）
    echo ""
    echo "📝 检查环境配置文件..."
    [ ! -f .env ] && [ -f .env.sample ] && cp .env.sample .env && echo "   ✓ .env"
    [ ! -f .env.tools ] && [ -f .env.tools.sample ] && cp .env.tools.sample .env.tools && echo "   ✓ .env.tools"
    [ ! -f .env.ASG ] && [ -f .env.ASG.sample ] && cp .env.ASG.sample .env.ASG && echo "   ✓ .env.ASG"
    [ ! -f .env.AMS ] && [ -f .env.AMS.sample ] && cp .env.AMS.sample .env.AMS && echo "   ✓ .env.AMS"

    echo ""
else
    echo "⏭️  跳过仓库更新，仅更新镜像并重启"
    echo ""
fi

echo "==========================="
echo "🚀 开始自动安装 Docker 和 Docker Compose"
echo "==========================="

# 检查 Docker 是否已安装
if command -v docker &> /dev/null; then
    echo "✅ Docker 已安装，跳过安装步骤"
else
    # 原有的 Docker 安装步骤
    echo "📦 安装依赖..."
    sudo apt update -y
    sudo apt install -y apt-transport-https ca-certificates curl software-properties-common gnupg lsb-release

    echo "🔑 添加 Docker GPG 密钥..."
    curl -fsSL https://mirrors.nwafu.edu.cn生成注释ker-ce/linux/ubuntu/gpg | sudo apt-key add -

    echo "🌍 添加 Docker 源..."
    sudo add-apt-repository -y "deb [arch=amd64] https://mirrors.nwafu.edu.cn生成注释ker-ce/linux/ubuntu/ $(lsb_release -cs) stable"
    sudo apt-get update -y

    echo "🐳 安装 Docker..."
    sudo apt-get install -y docker.io

    echo "✅ 启动并设置 Docker 开机自启..."
    sudo systemctl enable docker
    sudo systemctl start docker
fi

# 检测并使用正确的 Docker Compose 命令
# 优先使用新版本的 docker compose（作为 docker 的子命令）
if docker compose version &> /dev/null; then
    DOCKER_COMPOSE_CMD="docker compose"
    echo "✅ 检测到 Docker Compose (新版本: docker compose)"
elif command -v docker-compose &> /dev/null; then
    DOCKER_COMPOSE_CMD="docker-compose"
    echo "✅ 检测到 Docker Compose (旧版本: docker-compose)"
else
    echo "⚙️ 安装 Docker Compose..."
    COMPOSE_BIN_URL="https://static.51easyai.com/docker-compose-linux-x86_64"
    TMP_COMPOSE_BIN="/tmp/docker-compose-linux-x86_64"
    curl -fL "$COMPOSE_BIN_URL" -o "$TMP_COMPOSE_BIN"
    sudo install -m 0755 "$TMP_COMPOSE_BIN" /usr/bin/docker-compose
    rm -f "$TMP_COMPOSE_BIN"
    DOCKER_COMPOSE_CMD="docker-compose"
    echo "✅ Docker Compose 安装完成"
fi

echo "📌 Docker 运行状态:"
sudo systemctl status docker --no-pager || true

echo "📌 Docker Compose 版本:"
$DOCKER_COMPOSE_CMD version

echo "🎉 Docker 和 Docker Compose 已就绪！"


echo "🚀 重新启动EasyAI"
sudo $DOCKER_COMPOSE_CMD pull && sudo $DOCKER_COMPOSE_CMD up -d
echo "🎉EasyAI应用更新成功"