wcp/bin/v-add-database-host at a12a3bfaace47d535b6fdd307f91aa7e6769ecd4 · ghzhost/wcp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#!/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
# shellcheck source=/usr/local/hestia/func/main.sh
source $HESTIA/func/main.sh
# shellcheck source=/usr/local/hestia/func/db.sh
source $HESTIA/func/db.sh
# shellcheck source=/usr/local/hestia/conf/hestia.conf
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
$BIN/v-log-action "system" "Info" "Database" "Added external $type database server ($host) to the system."
log_event "$OK" "$ARGUMENTS"

exit