hestiacp/bin/v-add-database-host

#!/bin/bash
# info: add new database server
# options: TYPE HOST DBUSER DBPASS [MAX_DB] [CHARSETS] [TEMPLATE] [PORT]
# labels: 
#
# example: v-add-database-host mysql localhost alice p@$$wOrd
#
# The function add new database server to the server pool. It supports local
# and remote database servers, which is useful for clusters. By adding a host
# you can set limit for number of databases on a host. Template parameter is
# used only for PostgreSQL and has an default value "template1". You can read
# more about templates in official PostgreSQL documentation.


#----------------------------------------------------------#
#                    Variable&Function                     #
#----------------------------------------------------------#

# Argument definition
type=$1
host=$2
dbuser=$3
password=$4; HIDE=4
max_db=${5-500}
charsets=${6}
template=${7}
port=${8}

# Includes
source $HESTIA/func/main.sh
source $HESTIA/func/db.sh
source $HESTIA/conf/hestia.conf

is_mysql_host_alive() {
    mycnf=$(mktemp)
    echo "[client]">$mycnf
    echo "host='$host'" >> $mycnf
    echo "user='$user'" >> $mycnf
    echo "password='$password'" >> $mycnf
    echo "port='$port'" >> $mycnf
    
    chmod 600 $mycnf
    mysql --defaults-file=$mycnf -e 'SELECT VERSION()' >/dev/null 2>&1
    rm $mycnf
    
    if [ "$?" -ne '0' ]; then
        echo "Error: MySQL connection to $host failed"
        log_event "$E_CONNECT" "$ARGUMENTS"
        exit $E_CONNECT
    fi
}

is_pgsql_host_alive() {
    export PGPASSWORD="$dbpass"
    psql -h $host -U $dbuser -p $port -c "SELECT VERSION()" > /dev/null 2>&1
    if [ "$?" -ne '0' ]; then
        echo "Error: PostgreSQL connection to $host failed"
        log_event "$E_CONNECT" "$ARGUMENTS"
        exit $E_CONNECT
    fi
}


#----------------------------------------------------------#
#                    Verifications                         #
#----------------------------------------------------------#

args_usage='TYPE HOST DBUSER DBPASS [MAX_DB] [CHARSETS] [TPL] [PORT]'
check_args '4' "$#" "$args_usage"

if [ -z $charsets ]; then charsets="UTF8,LATIN1,WIN1250,WIN1251,WIN1252,WIN1256,WIN1258,KOI8"; fi
if [ -z $template ]; then template="template1"; fi

database_set_default_ports

is_format_valid 'host' 'dbuser' 'max_db' 'charsets' 'template' 'port'
is_system_enabled "$DB_SYSTEM" 'DB_SYSTEM'
is_type_valid "$DB_SYSTEM" "$type"
is_dbhost_new
is_password_valid
dbpass="$password"

case $type in
    mysql) is_mysql_host_alive ;;
    pgsql) is_pgsql_host_alive ;;
esac

# Perform verification if read-only mode is enabled
check_hestia_demo_mode

#----------------------------------------------------------#
#                       Action                             #
#----------------------------------------------------------#

# Generating timestamp
time_n_date=$(date +'%T %F')
time=$(echo "$time_n_date" |cut -f 1 -d \ )
date=$(echo "$time_n_date" |cut -f 2 -d \ )

# Concatenating db host string
case $type in
    mysql) str="HOST='$host' USER='$dbuser' PASSWORD='$dbpass'";
           str="$str CHARSETS='$charsets' MAX_DB='$max_db' U_SYS_USERS=''";
           str="$str U_DB_BASES='0' SUSPENDED='no' TIME='$time' DATE='$date' PORT='$port'";;
    pgsql) str="HOST='$host' USER='$dbuser' PASSWORD='$dbpass'";
           str="$str CHARSETS='$charsets' TPL='$template' MAX_DB='$max_db'";
           str="$str U_SYS_USERS='' U_DB_BASES='0' SUSPENDED='no'";
           str="$str TIME='$time' DATE='$date' PORT='$port'";;
esac


#----------------------------------------------------------#
#                       Hestia                             #
#----------------------------------------------------------#

# Adding host to conf
echo "$str" >> $HESTIA/conf/$type.conf
chmod 660 $HESTIA/conf/$type.conf

# Updating hestia.conf
if [ -z "$(grep DB_SYSTEM $HESTIA/conf/hestia.conf)" ]; then
    echo "DB_SYSTEM='$type'" >> $HESTIA/conf/hestia.conf
else
    db=$(echo "$DB_SYSTEM,$type" |\
        sed "s/,/\n/g"|\
        sort -r -u |\
        sed "/^$/d"|\
        sed ':a;N;$!ba;s/\n/,/g')
    sed -i "s/DB_SYSTEM=.*/DB_SYSTEM='$db'/g" $HESTIA/conf/hestia.conf
fi

# Logging
log_event "$OK" "$ARGUMENTS"

exit