From ea5364111297de87d7fda130c0dd57d69ddde91b Mon Sep 17 00:00:00 2001
From: zhangheng <zhangheng@cowarobot.com>
Date: Sat, 11 Jul 2026 18:24:15 +0800
Subject: [PATCH] fix(build): fix build.sh

---
 build.sh | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/build.sh b/build.sh
index f060193..8d3faeb 100755
--- a/build.sh
+++ b/build.sh
@@ -14,11 +14,29 @@ TAG=stage_latest
 BUILD_FLAG=build:stage
 if [[ $1 == prod ]]; then
     TAG=prod_latest
-    BUILD_FLAG=build:prod
+    BUILD_FLAG=build
 fi
 
 cd $SHELL_FOLDER
 
+
+ENV=stage
+
+# 解析参数
+for arg in "$@"; do
+    case "$arg" in
+        prod)
+            ENV=prod
+            ;;
+        stage)
+            ENV=stage
+            ;;
+        dev)
+            ENV=dev
+            ;;
+    esac
+done
+
 # # 检查 git 状态
 # echo -e "${GREEN}Checking git status...${NC}"
 # git fetch