LQConstPlanWriterAgent/deploy_agent.sh

#!/bin/bash

# ============ 强制要求 Bash 执行 ============
if [ -z "$BASH_VERSION" ]; then
    echo "****************************************************************"
    echo "* 错误：此脚本必须使用 bash 执行，不支持 sh/dash！"
    echo "* 请使用以下任一方式运行："
    echo "*   1. 赋予执行权限后： ./deploy_agent.sh"
    echo "*   2. 显式指定 bash：  bash deploy_agent.sh"
    echo "****************************************************************"
    exit 1
fi
# ==========================================

# ================= 配置区域 =================
SOURCE_DIR="/home/lq/lq_workspace/LQAgentServer/source/LQAgentWritePlatform"
DOCKER_APP_DIR="/home/lq/lq_workspace/LQAgentServer/app/LqAgentWritePlatform/docker"
COMPOSE_FILE="docker-compose.yml"
IMAGE_NAME="lq_agent_write_platform_server_dev"
GIT_USER="WangXuMing"
GIT_PASS="123456"
PROXY_HOST="127.0.0.1"
PROXY_PORT="7890"

# ================= 辅助函数 =================
log_info() {
    echo -e "\033[32m[INFO] $(date '+%Y-%m-%d %H:%M:%S') - $1\033[0m"
}

log_error() {
    echo -e "\033[31m[ERROR] $(date '+%Y-%m-%d %H:%M:%S') - $1\033[0m"
}

log_warn() {
    echo -e "\033[33m[WARN] $(date '+%Y-%m-%d %H:%M:%S') - $1\033[0m"
}

check_status() {
    if [ $? -ne 0 ]; then
        log_error "$1 执行失败，脚本终止。"
        exit 1
    fi
}

ask_use_proxy() {
    USE_PROXY=false
    PROXY_URL=""

    if [ ! -t 0 ]; then
        log_info "非交互终端，默认不使用代理"
        return
    fi

    echo ""
    echo "============================================"
    echo "  是否使用代理加速依赖下载？"
    echo "  默认: 不使用代理（直接走清华镜像源）"
    echo "============================================"
    read -p "使用代理 [y/N]: " proxy_choice

    case "$proxy_choice" in
        [yY][eE][sS]|[yY])
            USE_PROXY=true
            if [ -z "$PROXY_HOST" ]; then
                PROXY_HOST=$(hostname -I | awk '{print $1}')
            fi
            PROXY_URL="http://${PROXY_HOST}:${PROXY_PORT}"
            log_info "已启用代理: ${PROXY_URL}"

            if ! curl -s --connect-timeout 2 "http://${PROXY_HOST}:${PROXY_PORT}" > /dev/null 2>&1; then
                echo ""
                echo "[WARN] 代理端口 ${PROXY_HOST}:${PROXY_PORT} 未响应，可能尚未启动"
                echo "       请在另一个终端执行: clashctl on"
                echo ""
                read -p "确认代理已开启后按回车继续，或 Ctrl+C 退出: "
            fi
            ;;
        *)
            log_info "不使用代理，走默认镜像源"
            ;;
    esac
}

compare_versions() {
    v1=$(echo "$1" | sed 's/v//')
    v2=$(echo "$2" | sed 's/v//')

    if [ "$(echo "$v1 == $v2" | bc)" -eq 1 ]; then
        return 0
    elif [ "$(echo "$v1 > $v2" | bc)" -eq 1 ]; then
        return 1
    else
        return 2
    fi
}

# ================= 步骤 1: Git 拉取代码 =================
log_info "步骤 1: 进入源码目录并拉取最新代码..."

if [ ! -d "$SOURCE_DIR" ]; then
    log_error "源码目录不存在: $SOURCE_DIR"
    exit 1
fi

if [ ! -x "$SOURCE_DIR" ]; then
    log_error "源码目录无进入权限！正在修复..."
    sudo chmod +x "$SOURCE_DIR"
    sudo chown -R lq:lq "$SOURCE_DIR"
fi

cd "$SOURCE_DIR" || {
    log_error "进入源码目录失败！路径：$SOURCE_DIR"
    exit 1
}
check_status "进入源码目录"

if [ ! -d ".git" ]; then
    log_error "当前目录不是 Git 仓库！路径：$SOURCE_DIR"
    exit 1
fi

log_info "检查本地是否存在可能与远程冲突的已修改文件..."

HAS_CONFLICT_FILES=$(git status --porcelain | grep -v "^??")

if [ -n "$HAS_CONFLICT_FILES" ]; then
    log_info "发现以下文件存在本地修改（将被远程最新代码覆盖）："
    echo "$HAS_CONFLICT_FILES" | awk '{print "  - " $2}'
    log_info "正在强制丢弃本地修改，确保同步远程最新代码..."

    git checkout -- .

    log_info "本地冲突文件修改已丢弃，准备拉取远程最新代码..."
else
    log_info "本地无可能冲突的已修改文件，直接拉取远程最新代码..."
fi

ORIGIN_URL=$(git remote get-url origin 2>/dev/null)
if [ $? -ne 0 ]; then
    log_error "获取 Git 远程地址失败！请检查 remote 配置"
    exit 1
fi

CLEAN_URL=${ORIGIN_URL#*://}
AUTH_URL="http://${GIT_USER}:${GIT_PASS}@${CLEAN_URL}"

UPSTREAM_URL=$(git remote get-url upstream 2>/dev/null)
if [ $? -ne 0 ]; then
    log_warn "未配置 upstream 远程，503 时无法切换备用源"
    UPSTREAM_AVAILABLE=0
else
    UPSTREAM_CLEAN_URL=${UPSTREAM_URL#*://}
    UPSTREAM_AUTH_URL="http://${GIT_USER}:${GIT_PASS}@${UPSTREAM_CLEAN_URL}"
    UPSTREAM_AVAILABLE=1
fi

MAX_RETRIES=3
COUNT=0
GIT_SUCCESS=0
CURRENT_AUTH_URL="$AUTH_URL"

while [ $COUNT -lt $MAX_RETRIES ]; do
    log_info "正在执行 Git Pull (第 $((COUNT+1)) 次尝试) - 强制拉取 dev 分支最新代码..."
    log_info "当前使用远程地址：${CURRENT_AUTH_URL}"

    PULL_OUTPUT=$(git pull "$CURRENT_AUTH_URL" dev --force --allow-unrelated-histories 2>&1)
    PULL_EXIT_CODE=$?

    if [ $PULL_EXIT_CODE -eq 0 ]; then
        GIT_SUCCESS=1
        LATEST_COMMIT=$(git log -1 --format="%h - %s ")
        log_info "Git Pull 成功！当前部署提交版本：$LATEST_COMMIT"
        break
    else
        if echo "$PULL_OUTPUT" | grep -qiE "503 Service Unavailable|503 Unavailable|returned error: 503" && [ $UPSTREAM_AVAILABLE -eq 1 ]; then
            log_error "Git Pull 失败：当前远程（origin）返回 503 不可达，切换到备用远程（upstream）重试..."
            log_error "错误详情：$PULL_OUTPUT"
            CURRENT_AUTH_URL="$UPSTREAM_AUTH_URL"
            COUNT=$((COUNT+1))
            sleep 3
        elif echo "$PULL_OUTPUT" | grep -qiE "503 Service Unavailable|503 Unavailable|returned error: 503" && [ $UPSTREAM_AVAILABLE -eq 0 ]; then
            log_error "Git Pull 失败：远程返回 503 不可达，但未配置 upstream 备用源，无法切换..."
            log_error "错误详情：$PULL_OUTPUT"
            COUNT=$((COUNT+1))
            sleep 3
        else
            log_error "Git Pull 失败（非 503 错误），准备重试..."
            log_error "错误详情：$PULL_OUTPUT"
            COUNT=$((COUNT+1))
            sleep 3
        fi
    fi
done

if [ $GIT_SUCCESS -eq 0 ]; then
    log_error "Git Pull 已重试 $MAX_RETRIES 次，全部失败！"
    exit 1
fi

# ================= 步骤 2: 关闭当前容器 =================
log_info "步骤 2: 关闭正在运行的容器..."

if [ ! -d "$DOCKER_APP_DIR" ]; then
    log_error "Docker 运行目录不存在: $DOCKER_APP_DIR"
    exit 1
fi

cd "$DOCKER_APP_DIR"
check_status "进入 Docker 运行目录"

docker compose down
if [ $? -ne 0 ]; then
    log_error "警告: Docker Compose Down 返回非零状态，尝试继续..."
fi

# ================= 步骤 3: 获取当前版本并计算新版本号 =================
log_info "步骤 3: 查找当前运行版本并计算新版本号..."

CURRENT_CONFIG_TAG=$(grep -m 1 "image: ${IMAGE_NAME}:" "$DOCKER_APP_DIR/$COMPOSE_FILE" | sed "s|.*image: ${IMAGE_NAME}:||")
if [ -z "$CURRENT_CONFIG_TAG" ]; then
    CURRENT_CONFIG_TAG="v0.01"
    log_warn "未在配置文件中找到版本号，使用默认版本: $CURRENT_CONFIG_TAG"
else
    log_info "当前配置文件中的版本: $CURRENT_CONFIG_TAG"
fi

VERSION_NUM=$(echo "$CURRENT_CONFIG_TAG" | sed 's/v//')
NEW_VERSION_NUM=$(echo "$VERSION_NUM" | awk '{printf "%.2f", $1 + 0.01}')
NEW_TAG="v$NEW_VERSION_NUM"

log_info "计算出的新版本号为: $NEW_TAG"

# ================= 步骤 4: 清理旧镜像 =================
log_info "步骤 4: 清理旧镜像（保留最新版本，删除上上次版本）..."

HISTORY_IMAGES=$(docker images --filter "reference=${IMAGE_NAME}:*" --format "{{.Tag}} {{.ID}} {{.CreatedAt}}" | sort -r)

if [ -n "$HISTORY_IMAGES" ]; then
    mapfile -t IMAGE_ARRAY <<< "$HISTORY_IMAGES"

    log_info "发现 ${#IMAGE_ARRAY[@]} 个历史镜像版本:"
    for ((i=0; i<${#IMAGE_ARRAY[@]}; i++)); do
        TAG=$(echo "${IMAGE_ARRAY[$i]}" | awk '{print $1}')
        IMAGE_ID=$(echo "${IMAGE_ARRAY[$i]}" | awk '{print $2}')
        CREATED=$(echo "${IMAGE_ARRAY[$i]}" | awk '{$1=$2=""; print $0}' | sed 's/^  //')

        if [ "$TAG" = "$CURRENT_CONFIG_TAG" ]; then
            log_info "  [$i] $TAG - $IMAGE_ID (当前运行版本) - $CREATED"
            CURRENT_INDEX=$i
        else
            log_info "  [$i] $TAG - $IMAGE_ID - $CREATED"
        fi
    done

    if [ ${#IMAGE_ARRAY[@]} -gt 1 ]; then
        SECOND_TAG=$(echo "${IMAGE_ARRAY[1]}" | awk '{print $1}')
        SECOND_ID=$(echo "${IMAGE_ARRAY[1]}" | awk '{print $2}')

        if [ "$SECOND_TAG" != "$NEW_TAG" ] && [ "$SECOND_TAG" != "$CURRENT_CONFIG_TAG" ]; then
            log_info "正在删除上上次版本镜像: $SECOND_TAG ($SECOND_ID)"
            docker rmi -f "$SECOND_ID" 2>/dev/null
            if [ $? -eq 0 ]; then
                log_info "成功删除上上次版本镜像: $SECOND_TAG"
            else
                log_warn "删除镜像 $SECOND_TAG 失败（可能已被删除或正在使用），跳过..."
            fi
        else
            log_info "跳过删除 $SECOND_TAG，因为它是当前运行版本或将要构建的新版本"
        fi

        for ((i=2; i<${#IMAGE_ARRAY[@]}; i++)); do
            OLD_TAG=$(echo "${IMAGE_ARRAY[$i]}" | awk '{print $1}')
            OLD_ID=$(echo "${IMAGE_ARRAY[$i]}" | awk '{print $2}')

            if [ "$OLD_TAG" != "$NEW_TAG" ] && [ "$OLD_TAG" != "$CURRENT_CONFIG_TAG" ]; then
                log_info "删除更旧的镜像: $OLD_TAG ($OLD_ID)"
                docker rmi -f "$OLD_ID" 2>/dev/null
            fi
        done
    else
        log_info "只有1个历史镜像，无需清理"
    fi
else
    log_info "未找到历史镜像"
fi

# ================= 步骤 5: 构建新镜像 =================
BASE_IMAGE_NAME="lq-agent-base"
BASE_IMAGE_TAG="latest"
REQUIREMENTS_FILE="$SOURCE_DIR/requirements.txt"
BASE_HASH_FILE="$DOCKER_APP_DIR/.base_image_req_hash"

NEED_REBUILD_BASE=false

cd "$SOURCE_DIR"
check_status "返回源码目录"

ask_use_proxy

if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}$"; then
    log_info "步骤 5a: base 镜像不存在，首次构建 ${BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}（包含所有依赖，仅此一次）..."
    NEED_REBUILD_BASE=true
else
    CURRENT_HASH=$(md5sum "$REQUIREMENTS_FILE" | awk '{print $1}')
    if [ -f "$BASE_HASH_FILE" ]; then
        STORED_HASH=$(cat "$BASE_HASH_FILE")
        if [ "$CURRENT_HASH" != "$STORED_HASH" ]; then
            log_info "步骤 5a: requirements.txt 已变化，重建 base 镜像..."
            NEED_REBUILD_BASE=true
        else
            log_info "步骤 5a: base 镜像已存在且依赖未变化，跳过重建"
        fi
    else
        log_info "步骤 5a: 未找到依赖哈希记录，首次记录并确保 base 镜像一致..."
        NEED_REBUILD_BASE=true
    fi
fi

if [ "$NEED_REBUILD_BASE" = true ]; then
    if [ "$USE_PROXY" = true ]; then
        log_info "步骤 5a: 使用代理构建 base 镜像 (${PROXY_URL})..."
        docker build -f Dockerfile.base \
            --network host \
            --build-arg HTTP_PROXY="${PROXY_URL}" \
            --build-arg HTTPS_PROXY="${PROXY_URL}" \
            --build-arg http_proxy="${PROXY_URL}" \
            --build-arg https_proxy="${PROXY_URL}" \
            -t "${BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}" .
    else
        log_info "步骤 5a: 不使用代理，使用默认镜像源构建 base 镜像..."
        docker build -f Dockerfile.base -t "${BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}" .
    fi
    check_status "base 镜像构建"
    md5sum "$REQUIREMENTS_FILE" | awk '{print $1}' > "$BASE_HASH_FILE"
    log_info "base 镜像构建成功: ${BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}"
fi

log_info "步骤 5b: 构建应用镜像 $IMAGE_NAME:$NEW_TAG ..."

docker build -t "${IMAGE_NAME}:${NEW_TAG}" .
check_status "应用镜像构建"
log_info "应用镜像构建成功: ${IMAGE_NAME}:${NEW_TAG}"

# ================= 步骤 6: 修改 docker-compose.yml 版本号 =================
log_info "步骤 6: 更新 docker-compose.yml 中的版本号..."

cd "$DOCKER_APP_DIR"
check_status "进入 Docker 运行目录"

if [ ! -f "$COMPOSE_FILE" ]; then
    log_error "找不到配置文件: $COMPOSE_FILE"
    exit 1
fi

sed -i "s|image: ${IMAGE_NAME}:.*|image: ${IMAGE_NAME}:${NEW_TAG}|" "$COMPOSE_FILE"
check_status "修改 docker-compose.yml"

MATCH_LINE=$(grep "image: ${IMAGE_NAME}:" "$COMPOSE_FILE" | tr '\n' '; ')
log_info "配置文件已更新: $MATCH_LINE"

# ================= 步骤 7: 启动容器 =================
log_info "步骤 7: 启动 Docker Compose..."

docker compose up -d
check_status "启动容器"

# ================= 步骤 8: 显示当前保留的镜像 =================
log_info "步骤 8: 当前保留的镜像版本列表:"
docker images --filter "reference=${IMAGE_NAME}:*" --format "table {{.Tag}}\t{{.ID}}\t{{.Size}}\t{{.CreatedAt}}"

log_info "===================================================="
log_info " 开发版部署成功！"
log_info " 当前运行端口: 8003"
log_info " 部署版本: $NEW_TAG"
log_info " 保留镜像: 最新版本 + 前一个版本"
log_info "===================================================="