From 1234c56af8148fc3e7610363b7e9ac3cfd5c248b Mon Sep 17 00:00:00 2001
From: simonzhangsz <simonzhangsz@outlook.com>
Date: Fri, 22 Dec 2023 12:22:41 +0800
Subject: [PATCH] fix: omc setup sh

---
 misc/importdb.sh | 39 ---------------------------------
 misc/setupomc.sh | 57 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 57 insertions(+), 39 deletions(-)
 create mode 100644 misc/setupomc.sh

diff --git a/misc/importdb.sh b/misc/importdb.sh
index ef0d55d0..652aaac2 100644
--- a/misc/importdb.sh
+++ b/misc/importdb.sh
@@ -1,26 +1,5 @@
 #!/bin/bash
 
-C_ARG="*"
-M_ARG="*"
-
-check_args() {
-    while getopts "c:m:" option; do
-        case $option in
-            c)  
-                C_ARG=$OPTARG
-                ;;  
-
-            m)  
-                M_ARG=$OPTARG
-                ;;  
-
-            \?) 
-                echo "Invalid option: -$OPTARG" >&2
-                ;;  
-       esac
-   done
-}
-
 USER="root"
 PASSWORD="1000omc@kp!"
 PORT="33066"
@@ -28,14 +7,10 @@ DBNAME="omc_db"
 UpgradeSQLDir=/usr/local/omc/etc/db/upgrade
 Upgvue3SQLDir=/usr/local/omc/etc/db/upgvue3
 InstallSQLDir=/usr/local/omc/etc/db/install
-OMCStaticDir=/usr/local/omc/static
-CustomizedDirBA=${OMCStaticDir}/ba.d
 
 drop_db_sql="drop database IF EXISTS ${DBNAME}"
 create_db_sql="create database IF NOT EXISTS ${DBNAME}"
 
-check_args "$@"
-
 case "$M_ARG" in
         upgrade)
 			echo "Upgrade database ${DBNAME}"
@@ -122,17 +97,3 @@ case "$M_ARG" in
 			done
         ;;
 esac 
-
-case "$C_ARG" in
-    ba)
-		echo "Processing ${C_ARG} customized OMC ..."
-		for SQL in ${CustomizedDirBA}/db/*.sql; do 
-			mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
-		done
-		cp -rf ${CustomizedDirBA}/logo/* ${OMCStaticDir}/logo
-		rm -f ${OMCStaticDir}/helpDoc/*.pdf
-		cp -rf ${CustomizedDirBA}/doc/* ${OMCStaticDir}/helpDoc
-		;;
-	*)
-		;;
-esac
\ No newline at end of file
diff --git a/misc/setupomc.sh b/misc/setupomc.sh
new file mode 100644
index 00000000..94551158
--- /dev/null
+++ b/misc/setupomc.sh
@@ -0,0 +1,57 @@
+#!/bin/bash
+
+C_ARG="*"
+M_ARG="*"
+
+check_args() {
+    while getopts "c:m:" option; do
+        case $option in
+            c)  
+                C_ARG=$OPTARG
+                ;;  
+
+            m)  
+                M_ARG=$OPTARG
+                ;;  
+
+            \?) 
+                echo "Invalid option: -$OPTARG" >&2
+                ;;  
+       esac
+   done
+}
+
+USER="root"
+PASSWORD="1000omc@kp!"
+PORT="33066"
+DBNAME="omc_db"
+UpgradeSQLDir=/usr/local/omc/etc/db/upgrade
+Upgvue3SQLDir=/usr/local/omc/etc/db/upgvue3
+InstallSQLDir=/usr/local/omc/etc/db/install
+OMCStaticDir=/usr/local/omc/static
+CustomizedDirBA=${OMCStaticDir}/ba.d
+
+check_args "$@"
+
+case "${M_ARG}" in
+	install, upgrade, upgvue3)
+		importdb.sh ${M_ARG}
+		;;
+	*)
+		importdb.sh
+		;;
+esac
+
+case "$C_ARG" in
+    ba)
+		echo "Setting ${C_ARG} customized OMC ..."
+		for SQL in ${CustomizedDirBA}/db/*.sql; do 
+			mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
+		done
+		cp -rf ${CustomizedDirBA}/logo/* ${OMCStaticDir}/logo
+		rm -f ${OMCStaticDir}/helpDoc/*.pdf
+		cp -rf ${CustomizedDirBA}/doc/* ${OMCStaticDir}/helpDoc
+		;;
+	*)
+		;;
+esac
\ No newline at end of file