#!/bin/bash

C_ARG=""
M_ARG="*"

check_args() {
    while getopts "c:m:" option; do
        case $option in
            c)  
                C_ARG=$(echo $OPTARG | tr '[:upper:]' '[:lower:]')
                ;;  
            m)  
                M_ARG=$(echo $OPTARG | tr '[:upper:]' '[:lower:]')
                ;;  
            \?) 
                echo "Invalid option: -$OPTARG" >&2
                ;;  
       esac
   done
}

USER="root"
PASSWORD="1000omc@kp!"
PORT="33066"
DBNAME="omc_db"
OMCBinDir=/usr/local/omc/bin
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
CustomizedDirAGT=${OMCStaticDir}/agt.d
CustomizedDirGeneric=${OMCStaticDir}/generic.d

check_args "$@"

case "${M_ARG}" in
	install | upgrade | upgvue3)
		${OMCBinDir}/importdb.sh ${M_ARG}
		;;
	skip)
		;;	
	*)
		${OMCBinDir}/importdb.sh
		;;
esac

case "$C_ARG" in
    ba)
		echo "Setting BA customized OMC ..."
		for SQL in ${CustomizedDirBA}/db/*.sql; do 
			mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
		done
		rm -f ${OMCStaticDir}/logo/*.png
		cp -rf ${CustomizedDirBA}/logo/* ${OMCStaticDir}/logo
		rm -f ${OMCStaticDir}/helpDoc/*.pdf
		cp -rf ${CustomizedDirBA}/doc/* ${OMCStaticDir}/helpDoc
		;;
	agt)
		echo "Setting AGT customized OMC ..."
		for SQL in ${CustomizedDirBA}/db/*.sql; do 
			mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
		done
		rm -f ${OMCStaticDir}/logo/*.png
		cp -rf ${CustomizedDirBA}/logo/* ${OMCStaticDir}/logo
		rm -f ${OMCStaticDir}/helpDoc/*.pdf
		cp -rf ${CustomizedDirBA}/doc/* ${OMCStaticDir}/helpDoc
		;;

	*)
		echo "Setting Generic OMC ..."
		if [ ${M_ARG} == "install" ]; then
			cp -rf ${CustomizedDirGeneric}/logo/* ${OMCStaticDir}/logo
			cp -rf ${CustomizedDirGeneric}/doc/* ${OMCStaticDir}/helpDoc
		fi
		;;
	"")
		;;
esac