v-add-sys-ip

#!/bin/bash
# info: add system ip address
# options: IP NETMASK [INTERFACE] [USER] [IP_STATUS] [IP_NAME] [NAT_IP] [HELO]
# labels:
#
# example: v-add-sys-ip 216.239.32.21 255.255.255.0
#
# The function adds ip address into a system. It also creates rc scripts. You
# can specify ip name which will be used as root domain for temporary aliases.
# For example, if you set a1.myhosting.com as name, each new domain created on
# this ip will automatically receive alias $domain.a1.myhosting.com. Of course
# you must have wildcard record *.a1.myhosting.com pointed to ip. This feature 
# is very handy when customer wants to test domain before dns migration.


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

# Get interface name
iface=$(/bin/ip token | awk -F 'dev ' '{print $2}')

# Argument definition
ip=${1// /}
netmask=$2
iface="${3-$iface}"
user="${4-admin}"
ip_status="${5-shared}"
ip_name=$6
nat_ip=$7
helo=$8

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


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

check_args '2' "$#" 'IP NETMASK [INTERFACE] [USER] [STATUS] [NAME] [NATED_IP] [HELO]'
is_format_valid 'ip' 'netmask' 'interface' 'user' 'ip_status'
is_ip_free
is_object_valid 'user' 'USER' "$user"
is_object_unsuspended 'user' 'USER' "$user"
if [ ! -z "$ip_name" ] ; then
    is_format_valid 'ip_name'
fi
if [ ! -z "$nat_ip" ] ; then
    is_format_valid 'nat_ip'
fi
if [ "$user" != "admin" ]; then
    ip_status="dedicated"
fi

# Perform verification if read-only mode is enabled
check_hestia_demo_mode


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

cidr=$(convert_netmask $netmask)
broadcast=$(get_broadcast $ip $netmask)

sys_ip_check=$(/sbin/ip addr | grep "$ip")
if [ -z "$sys_ip_check" ]; then
    # Adding sys ip
    /sbin/ip addr add $ip/$cidr dev $iface \
        broadcast $broadcast label $iface

    if dpkg-query -W -f'${Status}' "netplan*" 2>/dev/null | grep -q "ok installed"; then
        if [ -f "/etc/netplan/60-hestia.yaml" ]; then
            sys_ip="        - $ip/$cidr"
        else
            sys_ip="# Added by hestia, please do not edit the file manually!"
            sys_ip="$sys_ip\nnetwork:"
            sys_ip="$sys_ip\n  version: 2"
            sys_ip="$sys_ip\n  renderer: networkd"
            sys_ip="$sys_ip\n  ethernets:"
            sys_ip="$sys_ip\n    $iface:"
            sys_ip="$sys_ip\n      addresses:"
            sys_ip="$sys_ip\n        - $ip/$cidr"
        fi
        IFS='%'
        echo -e $sys_ip >> /etc/netplan/60-hestia.yaml
        unset IFS
    else
        sys_ip="\n# Added by hestia"
        sys_ip="$sys_ip\nauto $iface"
        sys_ip="$sys_ip\niface $iface inet static"
        sys_ip="$sys_ip\naddress $ip"
        sys_ip="$sys_ip\nnetmask $netmask"
        echo -e $sys_ip >> /etc/network/interfaces
    fi
fi

# 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 \ )

# Adding hestia ip
echo "OWNER='$user'
STATUS='$ip_status'
NAME='$ip_name'
U_SYS_USERS=''
U_WEB_DOMAINS='0'
INTERFACE='$iface'
NETMASK='$netmask'
NAT='$nat_ip'
HELO='$helo'
TIME='$time'
DATE='$date'" > $HESTIA/data/ips/$ip
chmod 660 $HESTIA/data/ips/$ip

# WEB support
if [ ! -z "$WEB_SYSTEM" ]; then
    web_conf="/etc/$WEB_SYSTEM/conf.d/$ip.conf"
    rm -f $web_conf

    if [ "$WEB_SYSTEM" = 'httpd' ] || [ "$WEB_SYSTEM" = 'apache2' ]; then
        if [ -z "$(/usr/sbin/apachectl -v | grep Apache/2.4)" ]; then
            echo "NameVirtualHost $ip:$WEB_PORT" >  $web_conf
        fi
        echo "Listen $ip:$WEB_PORT" >> $web_conf
        cat $HESTIA_INSTALL_DIR/apache2/unassigned.conf >> $web_conf
        sed -i 's/directIP/'$ip'/g' $web_conf
        sed -i 's/directPORT/'$WEB_PORT'/g' $web_conf

    elif [ "$WEB_SYSTEM" = 'nginx' ]; then
        cp -f $HESTIA_INSTALL_DIR/nginx/unassigned.inc $web_conf
        sed -i 's/directIP/'$ip'/g' $web_conf
    fi

    if [ "$WEB_SSL" = 'mod_ssl' ]; then
        if [ -z "$(/usr/sbin/apachectl -v | grep Apache/2.4)" ]; then
            sed -i "1s/^/NameVirtualHost $ip:$WEB_SSL_PORT\n/" $web_conf
        fi
        sed -i "1s/^/Listen $ip:$WEB_SSL_PORT\n/" $web_conf
        sed -i 's/directSSLPORT/'$WEB_SSL_PORT'/g' $web_conf
    fi
fi

# Proxy support
if [ ! -z "$PROXY_SYSTEM" ]; then
    cat $WEBTPL/$PROXY_SYSTEM/proxy_ip.tpl |\
        sed -e "s/%ip%/$ip/g" \
            -e "s/%web_port%/$WEB_PORT/g" \
            -e "s/%proxy_port%/$PROXY_PORT/g" \
            -e "s/%proxy_ssl_port%/$PROXY_SSL_PORT/g" \
        > /etc/$PROXY_SYSTEM/conf.d/$ip.conf

    # mod_extract_forwarded
    fw_conf="/etc/$WEB_SYSTEM/conf.d/mod_extract_forwarded.conf"
    if [ -e "$fw_conf" ]; then
        ips=$(grep 'MEFaccept ' $fw_conf | grep -v '#' | head -n1)
        sed -i "s/$ips/$ips $ip/g" $fw_conf
    fi

    # mod_rpaf
    rpaf_conf="/etc/$WEB_SYSTEM/mods-enabled/rpaf.conf"
    if [ -e "$rpaf_conf" ]; then
        rpaf_str=$(grep RPAFproxy_ips $rpaf_conf)
        [ -z "$rpaf_str" ] && sed -i 's|</IfModule>|RPAFproxy_ips\n</IfModule>|' $rpaf_conf && rpaf_str='RPAFproxy_ips'
        rpaf_str="$rpaf_str $ip"
        sed -i "s/.*RPAFproxy_ips.*/$rpaf_str/" $rpaf_conf
    fi

    #mod_remoteip
    remoteip_conf="/etc/$WEB_SYSTEM/mods-enabled/remoteip.conf"
    if [ -e "$remoteip_conf" ]; then
        if [ $( grep -ic "$ip" $remoteip_conf ) -eq 0 ]; then
            sed -i "s/<\/IfModule>/RemoteIPInternalProxy $ip\n<\/IfModule>/g" $remoteip_conf
        fi
    fi
fi


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

# Updating user counters
increase_user_value "$user" '$IP_OWNED'
if [ "$user" = 'admin' ]; then
    if [ "$ip_status" = 'shared' ]; then
        for hestia_user in $($HESTIA/bin/v-list-sys-users plain); do
            increase_user_value "$hestia_user" '$IP_AVAIL'
        done
    else
        increase_user_value 'admin' '$IP_AVAIL'
    fi
else
    increase_user_value "$user" '$IP_AVAIL'
    increase_user_value 'admin' '$IP_AVAIL'
fi

# Restarting web server
$BIN/v-restart-web
check_result $? "Web restart failed" >/dev/null

# Restarting proxy server
if [ ! -z "$PROXY_SYSTEM" ]; then
    $BIN/v-restart-proxy
    check_result $? "Proxy restart failed" >/dev/null
fi

# Restarting firewall
if [ ! -z "$FIREWALL_SYSTEM" ]; then
    $BIN/v-update-firewall
fi

# Update ip helo for exim
if [ ! -z "$MAIL_SYSTEM" ] && [ ! -z "$helo"]; then
    $BIN/v-change-sys-ip-helo $ip $helo
fi

# Logging
log_history "added system ip address $ip" '' 'admin'
log_event "$OK" "$ARGUMENTS"

exit