#!/bin/bash

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

drop_db_sql="drop database IF EXISTS ${DBNAME}"
create_db_sql="create database IF NOT EXISTS ${DBNAME}"

case "$1" in
        upgrade)
			echo "Upgrade database ${DBNAME}"
			for SQL in ${UpgradeSQLDir}/*.sql; do 
				echo "Execute SQL script: ${SQL} ..."
				mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
			done

        ;;

        upgvue3)
			echo "Upgrade to vue3 database ${DBNAME}"
			for SQL in ${Upgvue3SQLDir}/*.sql; do 
				echo "Execute SQL script: ${SQL} ..."
				mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
			done
        ;;

        install)
			echo "Drop database ${DBNAME} ...!!!"
		    mysql -u${USER} -p${PASSWORD} -P $PORT --protocol tcp -e "${drop_db_sql}"
			echo "Create database ${DBNAME} if not exist"
			mysql -u${USER} -p${PASSWORD} -P $PORT --protocol tcp -e "${create_db_sql}"
			for SQL in ${InstallSQLDir}/*.sql; do 
				echo "Execute SQL script: ${SQL} ..."
				mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
			done
        ;;
		
        *)
			while true
			do
				read -r -p "Do you upgrade or upgrade from layui to vue3 or fresh install database ${DBNAME}? [Upgrade/upgVue3/Install/Quit] " input

				case $input in
				    [uU][pP][gG][rR][aA][dD][eE]|[uU])
						echo "Skip to drop database ${DBNAME}"
						echo "Upgrade database ${DBNAME}"
							
						for SQL in ${UpgradeSQLDir}/*.sql; do 
							echo "Execute SQL script: ${SQL} ..."
							mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
						done

			            break
						;;

				    [uU][pP][gG][vV][uU][eE][3]|[vV]])
						echo "Skip to drop database ${DBNAME}"
						echo "Upgrade to vue3 database ${DBNAME}"

						for SQL in ${Upgvue3SQLDir}/*.sql; do 
							echo "Execute SQL script: ${SQL} ..."
							mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
						done

			            break
						;;

				    [iI][nN][sS][tT][aA][lL][lL]|[iI])
			            echo "Drop database ${DBNAME} ...!!!"

			            mysql -u${USER} -p${PASSWORD} -P $PORT --protocol tcp -e "${drop_db_sql}"
						echo "Create database ${DBNAME} if not exist"
						mysql -u${USER} -p${PASSWORD} -P $PORT --protocol tcp -e "${create_db_sql}"

						for SQL in ${InstallSQLDir}/*.sql; do 
							echo "Execute SQL script: ${SQL} ..."
							mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
						done

			            break           
						;;

				    [qQ][uU][iI][tT]|[qQ])
						echo "Nothing to be done! GOOD BYE"
			            exit 1;
						;;

				    *)
						echo "Invalid input..."
						;;
				esac
			done
        ;;
esac