fix(build): fix build.sh

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