#!/bin/bash

C_ARG_LOWER="agt"
C_ARG_UPPER="AGT"
M_ARG_LOWER="*"

check_args() {
    while getopts "c:m:" option; do
        case $option in
            c)  
                C_ARG_LOWER=$(echo $OPTARG | tr '[:upper:]' '[:lower:]')
                C_ARG_UPPER=$(echo $OPTARG | tr '[:lower:]' '[:upper:]')
                ;;  
            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"
OMCRootDir=/usr/local/omc
OMCBinDir=${OMCRootDir}/bin
UpgradeSQLDir=${OMCRootDir}/etc/db/upgrade
Upgvue3SQLDir=${OMCRootDir}/etc/db/upgvue3
InstallSQLDir=${OMCRootDir}/etc/db/install
OMCStaticDir=${OMCRootDir}/static

check_args "$@"

case "${M_ARG}" in
    install)
        ${OMCBinDir}/importdb.sh ${M_ARG}

        if [ "${C_ARG_LOWER}" != "" ]; then
           CustomizedDir=${OMCStaticDir}/${C_ARG_LOWER}.d
           if [ ! -d "${CustomizedDir}" ]; then
               echo "Not found ${C_ARG_UPPER} customized directory, nothing to be done"
               exit 1
           fi  
           echo "Setting ${C_ARG_UPPER} customized OMC ..."
           for SQL in ${CustomizedDir}/db/*.sql; do  
                   mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
           done
           cp -rf ${CustomizedDir}/logo/* ${OMCStaticDir}/logo
           cp -rf ${CustomizedDir}/doc/* ${OMCStaticDir}/helpDoc
           perl -0777 -i -pe 's/agtuser/bluearcus/g' ${OMCRootDir}/etc/default/restconf.yaml
           perl -0777 -i -pe 's/agtuser/bluearcus/g' ${OMCBinDir}/nehosts
        fi       
        ;;
    upgrade | upgvue3)
        ${OMCBinDir}/importdb.sh ${M_ARG}
        ;;
    skip)
        if [ "${C_ARG_LOWER}" != "" ]; then
           CustomizedDir=${OMCStaticDir}/${C_ARG_LOWER}.d
           if [ ! -d "${CustomizedDir}" ]; then
               echo "Not found ${C_ARG_UPPER} customized directory, nothing to be done"
               exit 1
           fi  
           echo "Setting ${C_ARG_UPPER} customized OMC ..."
           for SQL in ${CustomizedDir}/db/*.sql; do  
                   mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
           done
           cp -rf ${CustomizedDir}/logo/* ${OMCStaticDir}/logo
           cp -rf ${CustomizedDir}/doc/* ${OMCStaticDir}/helpDoc
        fi
        ;;    
    *)
        ${OMCBinDir}/importdb.sh
        ;;
esac

exit 0