From 9f379546cbb155c1986d3b59fc9be77946c66574 Mon Sep 17 00:00:00 2001
From: lijiahangmax <ljh1553488six@139.com>
Date: Wed, 9 Jul 2025 16:31:05 +0800
Subject: [PATCH] =?UTF-8?q?:zap:=20=E5=8D=87=E7=BA=A7=E7=89=88=E6=9C=AC.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-upgrade.sh | 49 ++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 42 insertions(+), 7 deletions(-)

diff --git a/docker-upgrade.sh b/docker-upgrade.sh
index 590118a3..9996bf96 100644
--- a/docker-upgrade.sh
+++ b/docker-upgrade.sh
@@ -1,16 +1,51 @@
 #!/bin/bash
 
-# 停止并移除现有容器
-docker compose down --remove-orphans
+# 初始化标志变量
+PULL_IMAGES=false
+DEMO_MODE=false
 
-if [ "$1" == "demo" ]; then
-    # 设置 DEMO_MODE 环境变量为 true
+# 解析命令行参数
+for arg in "$@"
+do
+    case $arg in
+        --pull)
+        PULL_IMAGES=true
+        shift
+        ;;
+        --demo)
+        DEMO_MODE=true
+        shift
+        ;;
+        *)
+        echo "Unknown argument: $arg"
+        exit 1
+        ;;
+    esac
+done
+
+# 停止并移除现有容器
+echo "Stopping all services..."
+docker compose down --remove-orphans
+echo "Stopped all services..."
+
+# 拉取镜像
+if [ "$PULL_IMAGES" = true ]; then
+    echo "Pulling latest images..."
+    docker compose pull
+    echo "Pulled latest images..."
+fi
+
+if [ "$DEMO_MODE" = true ]; then
+    # 启用 demo 模式
     export DEMO_MODE=true
     echo "Starting services for demo mode..."
+
     # 启动指定的服务
-    docker compose up -d --pull --remove-orphans mysql redis ui service guacd adminer
+    docker compose up -d --remove-orphans mysql redis ui service guacd adminer
+    echo "Started services for demo mode..."
 else
+    # 启动所有服务
     echo "Starting all services..."
-    # 正常启动所有服务
-    docker compose up -d --pull --remove-orphans
+    docker compose up -d --remove-orphans
+    echo "Started all services..."
 fi