From 80fb41ea51f4a3a377bffd5041b221d1bd6def9b Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 18 Apr 2025 19:59:44 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=A0=B9=E6=8D=AE=E5=AE=89=E8=A3=85?=
 =?UTF-8?q?=E6=A8=A1=E5=BC=8F=E5=8E=BB=E9=99=A4=E5=85=B6=E4=BD=99=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 linux/usr/local/etc/omc/script/setup.sh | 67 +++++++++++++++++--------
 1 file changed, 45 insertions(+), 22 deletions(-)

diff --git a/linux/usr/local/etc/omc/script/setup.sh b/linux/usr/local/etc/omc/script/setup.sh
index eedc1b0..27abe15 100644
--- a/linux/usr/local/etc/omc/script/setup.sh
+++ b/linux/usr/local/etc/omc/script/setup.sh
@@ -93,32 +93,55 @@ if [ -n "$C_PARAM" ]; then
 fi
 # echo "Customize parameter: $VENDORS"
 
-# =========================
-if [[ "$T_PARAM" == "install" && "$MODE" == "std" ]]; then
-  if ! command -v mysql &>/dev/null && ! command -v mariadb &>/dev/null; then
-      echo "MySQL or MariaDB not installed"
-      exit 1
-  fi
-  rm -rf $OMCEtcDir/database/lite
-  rm -rf $OMCEtcDir/default/omc_db.sqlite
-fi
-if [[ "$T_PARAM" == "install" && "$MODE" == "lite" ]]; then
-  rm -rf $OMCEtcDir/database/std
-  cp -rf $OMCEtcDir/default/omc_db.sqlite $OMCEtcDir/database/omc_db.sqlite
-fi
+# ========================= Reject non-specified pattern data
+case "$T_PARAM" in
+  "install")
+    case "$T_PARAM" in
+      "std")
+        if ! command -v mysql &>/dev/null && ! command -v mariadb &>/dev/null; then
+            echo "MySQL or MariaDB not installed"
+            exit 1
+        fi
+        rm -rf $OMCEtcDir/database/lite
+        rm -rf $OMCEtcDir/default/omc_db.sqlite
+        ;;
+      "lite")
+        rm -rf $OMCEtcDir/database/std
+        cp -rf $OMCEtcDir/default/omc_db.sqlite $OMCEtcDir/database/omc_db.sqlite
+        ;;
+    esac
+    ;;
+  "upgrade")
+    case "$T_PARAM" in
+      "std")
+        rm -rf $OMCEtcDir/database/lite
+        rm -rf $OMCEtcDir/default/omc_db.sqlite
+        ;;
+      "lite")
+        rm -rf $OMCEtcDir/database/std
+        ;;
+    esac
+    ;;
+esac
 
-# =========================
+# ========================= Execute the catalog sql file according to the schema
 $OMCBinFile -c $confFile --sqlPath $OMCEtcDir/database/$MODE/$T_PARAM --sqlSource $MODE
 
 
 # ========================= Customize
-if [[ "$T_PARAM" == "install" && -d $OMCEtcDir/vendor ]]; then
-  cp -rf $OMCEtcDir/vendor/$VENDORS/web/* $OMCEtcDir/web/background
-  cp -rf $OMCEtcDir/vendor/$VENDORS/static/* /usr/local/omc/static
-  $OMCBinFile -c $confFile --sqlPath $OMCEtcDir/vendor/$VENDORS/database/$MODE/$T_PARAM --sqlSource $MODE
-fi
-if [[ "$T_PARAM" == "upgrade" && -d $OMCEtcDir/vendor ]]; then
-  $OMCBinFile -c $confFile --sqlPath $OMCEtcDir/vendor/$VENDORS/database/$MODE/$T_PARAM --sqlSource $MODE
-fi
+case "$T_PARAM" in
+  "install")
+    if [[ -d $OMCEtcDir/vendor ]]; then
+      cp -rf $OMCEtcDir/vendor/$VENDORS/web/* $OMCEtcDir/web/background
+      cp -rf $OMCEtcDir/vendor/$VENDORS/static/* /usr/local/omc/static
+      $OMCBinFile -c $confFile --sqlPath $OMCEtcDir/vendor/$VENDORS/database/$MODE/$T_PARAM --sqlSource $MODE
+    fi
+    ;;
+  "upgrade")
+    if [[ -d $OMCEtcDir/vendor ]]; then
+      $OMCBinFile -c $confFile --sqlPath $OMCEtcDir/vendor/$VENDORS/database/$MODE/$T_PARAM --sqlSource $MODE
+    fi
+    ;;
+esac
 
 # bash setup.sh -i -m std -c omc