#!/bin/bash
DEFAULT_VH_ROOT='/var/www/vhosts'
VH_DOC_ROOT=''
APP_NAME=''
DOMAIN=''
WWW_UID=''
WWW_GID=''

help_message(){
    echo 'Command [-app app_name] [-domain domain_name]'
    echo 'Example: download.sh -app wordpress -d example.com'
}

check_input(){
    if [ -z "${1}" ]; then
        help_message
        exit 1
    fi
}

get_owner(){
	WWW_UID=$(stat -c "%u" ${DEFAULT_VH_ROOT}/${1})
	WWW_GID=$(stat -c "%g" ${DEFAULT_VH_ROOT}/${1})
	if [ ${WWW_UID} -eq 0 ] || [ ${WWW_GID} -eq 0 ]; then
		echo "Found ${WWW_UID}:${WWW_GID} has root, will auto fix to 1000"
		WWW_UID=1000
		WWW_GID=1000
	fi
}

set_vh_docroot(){
	if [ -d ${DEFAULT_VH_ROOT}/${1}/html ]; then
        VH_DOC_ROOT="${DEFAULT_VH_ROOT}/${1}/html"
	else
	    echo "${DEFAULT_VH_ROOT}/${1}/html does not exist, please add domain first! Abort!"
		exit 1
	fi	
}

check_sql_native(){
    local COUNTER=0
	local LIMIT_NUM=100
	until [ "$(curl -v mysql:3306 2>&1 | grep native)" ];
	do
	    echo "Counter: ${COUNTER}/${LIMIT_NUM}"
		COUNTER=$((COUNTER+1))
		if [ ${COUNTER} = 10 ]; then
			echo '--- MySQL is starting, please wait... ---'
		elif [ ${COUNTER} = ${LIMIT_NUM} ]; then	
		    echo '--- MySQL is timeout, exit! ---'
			exit 1
		fi
		sleep 1
	done
}

app_wordpress_dl(){
	if [ ! -f "${VH_DOC_ROOT}/wp-config.php" ] && [ ! -f "${VH_DOC_ROOT}/wp-config-sample.php" ]; then
		wp core download \
			--allow-root \
			--force
		chown -R ${WWW_UID}:${WWW_GID} ${DEFAULT_VH_ROOT}/${DOMAIN}
	else
	    echo 'wp-config*.php	already exist, abort!'
		exit 1
	fi
}

main(){
	get_owner
	cd ${VH_DOC_ROOT}
	if [ "${APP_NAME}" = 'wordpress' ] || [ "${APP_NAME}" = 'wp' ]; then
	    check_sql_native
	    app_wordpress_dl
		exit 0
	else
	    echo "APP: ${APP_NAME} not support, exit!"
		exit 1	
	fi
}

while [ ! -z "${1}" ]; do
    case ${1} in
        -[hH] | -help | --help)
            help_message
            ;;
        -app | -a | -A) shift
            check_input "${1}"
            APP_NAME="${1}"
            ;;
        -d | -D | -domain) shift
            check_input "${1}"
            DOMAIN="${1}"
			set_vh_docroot ${DOMAIN}
            ;;          
        *) 
            help_message
            ;;              
    esac
    shift
done
main