From 6f8617a81e257a7c658378fc34b43748d99e26c6 Mon Sep 17 00:00:00 2001
From: simonzhangsz <simonzhangsz@outlook.com>
Date: Thu, 21 Dec 2023 19:07:34 +0800
Subject: [PATCH] fix: ba customized

---
 misc/importdb.sh | 58 +++++++++++++++++++++++++-----------------------
 1 file changed, 30 insertions(+), 28 deletions(-)

diff --git a/misc/importdb.sh b/misc/importdb.sh
index 7f091701..bdc9e760 100644
--- a/misc/importdb.sh
+++ b/misc/importdb.sh
@@ -1,5 +1,26 @@
 #!/bin/bash
 
+C_ARG="agt"
+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"
@@ -7,35 +28,14 @@ DBNAME="omc_db"
 UpgradeSQLDir=/usr/local/omc/etc/db/upgrade
 Upgvue3SQLDir=/usr/local/omc/etc/db/upgvue3
 InstallSQLDir=/usr/local/omc/etc/db/install
-
-C_ARG="agt"
-M_ARG="upgrade"
+OMCStaticDir=/usr/local/omc/static
+CustomizedDirBA=${OMCStaticDir}/ba.d
+StaticLogoDir=${OMCStaticDir}/logo
 
 drop_db_sql="drop database IF EXISTS ${DBNAME}"
 create_db_sql="create database IF NOT EXISTS ${DBNAME}"
 
-get_args() {
-    getopts "c:m:" option; do
-		case $option in
-			c)
-				C_ARG=$OPTARG
-	        ;;
-
-      		m)
-        		M_ARG=$OPTARG
-        	;;
-
-      		\?)
-        		echo "Invalid option: -$OPTARG" >&2
-        	;;
-    	esac
-  	done
-}
-
-get_args "$@"
-
-echo "C_ARG: $C_ARG"
-echo "M_ARG: $M_ARG"
+check_args "$@"
 
 case "$M_ARG" in
         upgrade)
@@ -126,11 +126,13 @@ esac
 
 case "$C_ARG" in
     ba)
-		echo "C_ARG: $C_ARG"
+		for SQL in ${CustomizedDirBA}/db/*.sql; do 
+			echo "Execute SQL script: ${SQL} ..."
+			mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
+		done
+		cp -rf ${CustomizedDirBA}/logo/* $StaticLogoDir
 		;;
 	agt)
-		echo "C_ARG: $C_ARG"
-		;;
 	*)
 		echo "C_ARG: $C_ARG"
 		;;