From 261dbdc23c94ea608d674bb4793f60d15dd0b8fe Mon Sep 17 00:00:00 2001
From: WrBug <iwrbug@gmail.com>
Date: Sat, 14 Feb 2026 06:26:38 +0800
Subject: [PATCH] =?UTF-8?q?feat(deploy):=20DOCKER=5FVERSION=20=E6=94=AF?=
 =?UTF-8?q?=E6=8C=81=E4=BB=8E=20.env=20=E8=AF=BB=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

版本号优先级：环境变量 > .env 中的 DOCKER_VERSION > 当前分支名
提示文案更新为可在 .env 或环境变量中设置

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 deploy.sh | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/deploy.sh b/deploy.sh
index e20ced4..6f6f659 100755
--- a/deploy.sh
+++ b/deploy.sh
@@ -161,7 +161,10 @@ deploy() {
         # 注意：这里需要手动修改 docker-compose.yml，或者使用环境变量
         warn "请确保 docker-compose.yml 中已配置使用 image: wrbug/polyhermes:latest"
     else
-        # 版本号：优先使用环境变量 DOCKER_VERSION，未设置时使用当前分支名
+        # 版本号：优先环境变量 DOCKER_VERSION，其次 .env 中的 DOCKER_VERSION，否则用当前分支名
+        if [ -z "${DOCKER_VERSION}" ] && [ -f ".env" ]; then
+            DOCKER_VERSION=$(grep "^DOCKER_VERSION=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^["'\'']//;s/["'\'']$//' | tr -d '\r')
+        fi
         if [ -z "${DOCKER_VERSION}" ]; then
             CURRENT_BRANCH=$(git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "dev")
             DOCKER_VERSION=$(echo "$CURRENT_BRANCH" | tr '/' '-')
@@ -218,11 +221,14 @@ main() {
     info "访问地址: http://localhost:${SERVER_PORT:-80}"
     echo ""
     if [ "$USE_DOCKER_HUB" != "true" ]; then
+        if [ -z "${DOCKER_VERSION}" ] && [ -f ".env" ]; then
+            DOCKER_VERSION=$(grep "^DOCKER_VERSION=" .env 2>/dev/null | cut -d'=' -f2- | sed 's/^["'\'']//;s/["'\'']$//' | tr -d '\r')
+        fi
         if [ -z "${DOCKER_VERSION}" ]; then
             CURRENT_BRANCH=$(git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "dev")
             DOCKER_VERSION=$(echo "$CURRENT_BRANCH" | tr '/' '-')
         fi
-        info "提示：本地构建的版本号: ${DOCKER_VERSION}（可通过环境变量 DOCKER_VERSION 指定）"
+        info "提示：本地构建的版本号: ${DOCKER_VERSION}（可在 .env 或环境变量中设置 DOCKER_VERSION）"
         info "生产环境推荐使用 Docker Hub 镜像："
         info "  ./deploy.sh --use-docker-hub"
         info "  或修改 docker-compose.yml 使用 image: wrbug/polyhermes:latest"