v-add-dns-domain

#!/bin/bash
# info: add dns domain
# options: USER DOMAIN IP [NS1] [NS2] [NS3] [NS4] [NS5] [NS6] [NS7] [NS8] [RESTART]
# labels: dns
#
# example: v-add-dns-domain admin example.com ns1.example.com ns2.example.com '' '' '' '' '' '' yes
#
# The function adds DNS zone with records defined in the template. If the exp
# argument isn't stated, the expiration date value will be set to next year.
# The soa argument is responsible for the relevant record. By default the first
# user's NS server is used. TTL is set as common for the zone and for all of
# its records with a default value of 14400 seconds.


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

# Argument definition
user=$1
domain=$2
ip=$3
ns1=$4
ns2=$5
ns3=$6
ns4=$7
ns5=$8
ns6=$9
ns7=${10}
ns8=${11}
restart=${12}

# Includes
# shellcheck source=/usr/local/hestia/func/main.sh
source $HESTIA/func/main.sh
# shellcheck source=/usr/local/hestia/func/domain.sh
source $HESTIA/func/domain.sh
# shellcheck source=/usr/local/hestia/conf/hestia.conf
source $HESTIA/conf/hestia.conf

# Additional argument formatting
format_domain
format_domain_idn
domain_utf=$(idn -t --quiet -u "$domain_idn")

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

check_args '3' "$#" 'USER DOMAIN IP [NS1] [NS2] [NS3] [..] [NS8] [RESTART]'
is_format_valid 'user' 'domain' 'ip'
is_system_enabled "$DNS_SYSTEM" 'DNS_SYSTEM'
is_object_valid 'user' 'USER' "$user"
is_object_unsuspended 'user' 'USER' "$user"

if [ "$($BIN/v-list-dns-domain $user $domain_utf plain |cut -f 1) " != "$domain" ]; then
    is_domain_new 'dns' "$domain_utf"
fi
if [ "$($BIN/v-list-dns-domain $user $domain_idn plain |cut -f 1) " != "$domain" ]; then
    is_domain_new 'dns' "$domain_idn"
else
    is_domain_new 'dns' "$domain"
fi
if [ -z "$(is_ip_format_valid $domain)" ]; then
    echo "Error: Invalid domain format. IP address detected as input." 
    exit 1
fi

is_package_full 'DNS_DOMAINS'
template=$(get_user_value '$DNS_TEMPLATE')
is_dns_template_valid $template

is_base_domain_owner "$domain"

if [ ! -z "$ns1" ]; then
    ns1=$(echo $4 |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns1'
fi
if [ ! -z "$ns2" ]; then
    ns2=$(echo $5 |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns2'
fi
if [ ! -z "$ns3" ]; then
    ns3=$(echo $6 |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns3'
fi
if [ ! -z "$ns4" ]; then
    ns4=$(echo $7 |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns4'
fi
if [ ! -z "$ns5" ]; then
    ns5=$(echo $8 |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns5'
fi
if [ ! -z "$ns6" ]; then
    ns6=$(echo $9 |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns6'
fi
if [ ! -z "$ns7" ]; then
    ns7=$(echo ${10} |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns7'
fi
if [ ! -z "$ns8" ]; then
    ns8=$(echo ${11} |sed -e 's/\.*$//g' -e 's/^\.*//g')
    is_format_valid 'ns8'
fi

# Perform verification if read-only mode is enabled
check_hestia_demo_mode


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

# Defining NS variables
if [ -z $ns2 ]; then
    i=1
    ns=$(get_user_value '$NS')
    for nameserver in ${ns//,/ };do
        eval ns$i=$nameserver
        (( ++i))
    done
fi
soa="$ns1"
exp=$(date +%F -d "+ 1 year")
serial=$(date +'%Y%m%d01')
ttl=14400

# Reading template
template_data=$(cat $DNSTPL/$template.tpl)

# Deleting unused nameservers
if [ -z "$ns3" ]; then
    template_data=$(echo "$template_data" |grep -v %ns3%)
fi
if [ -z "$ns4" ]; then
    template_data=$(echo "$template_data" |grep -v %ns4%)
fi
if [ -z "$ns5" ]; then
    template_data=$(echo "$template_data" |grep -v %ns5%)
fi
if [ -z "$ns6" ]; then
    template_data=$(echo "$template_data" |grep -v %ns6%)
fi
if [ -z "$ns7" ]; then
    template_data=$(echo "$template_data" |grep -v %ns7%)
fi
if [ -z "$ns8" ]; then
    template_data=$(echo "$template_data" |grep -v %ns8%)
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 dns zone to the user config
echo "$template_data" |\
    sed -e "s/%ip%/$ip/g" \
        -e "s/%domain_idn%/$domain_idn/g" \
        -e "s/%domain%/$domain/g" \
        -e "s/%ns1%/$ns1/g" \
        -e "s/%ns2%/$ns2/g" \
        -e "s/%ns3%/$ns3/g" \
        -e "s/%ns4%/$ns4/g" \
        -e "s/%ns5%/$ns5/g" \
        -e "s/%ns6%/$ns6/g" \
        -e "s/%ns7%/$ns7/g" \
        -e "s/%ns8%/$ns8/g" \
        -e "s/%time%/$time/g" \
        -e "s/%date%/$date/g" > $USER_DATA/dns/$domain.conf

chmod 660 $USER_DATA/dns/$domain.conf
records="$(wc -l $USER_DATA/dns/$domain.conf |cut -f 1 -d ' ')"

# Adding dns.conf record
dns_rec="DOMAIN='$domain' IP='$ip' TPL='$template' TTL='$ttl' EXP='$exp'"
dns_rec="$dns_rec SOA='$soa' SERIAL='$serial' SRC='' RECORDS='$records'"
dns_rec="$dns_rec SUSPENDED='no' TIME='$time' DATE='$date'"

echo "$dns_rec" >> $USER_DATA/dns.conf
chmod 660 $USER_DATA/dns.conf

# Creating system configs
if [[ "$DNS_SYSTEM" =~ named|bind ]]; then
    if [ -e '/etc/named.conf' ]; then
        dns_conf='/etc/named.conf'
        dns_group='named'
    else
        dns_conf='/etc/bind/named.conf'
        dns_group='bind'
    fi

    # Adding zone in named.conf
    named="zone \"$domain_idn\" {type master; file"
    named="$named \"$HOMEDIR/$user/conf/dns/$domain.db\";};"
    echo "$named" >> $dns_conf

    # Updating domain dns zone
    update_domain_zone

    # Changing permissions
    chmod 640 $HOMEDIR/$user/conf/dns/$domain.db
    chown root:$dns_group $HOMEDIR/$user/conf/dns/$domain.db
fi

# Updating dns-cluster queue
if [ ! -z "$DNS_CLUSTER" ]; then
    cmd="$BIN/v-add-remote-dns-domain $user $domain yes"
    echo "$cmd" >> $HESTIA/data/queue/dns-cluster.pipe
fi


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

# Increasing domain value
increase_user_value "$user" '$U_DNS_DOMAINS'
increase_user_value "$user" '$U_DNS_RECORDS' "$records"

# Restart named
$BIN/v-restart-dns $restart
check_result $? "DNS restart failed"

# Logging
$BIN/v-log-action "$user" "Info" "DNS" "Added new DNS domain (Name: $domain)."
log_event "$OK" "$ARGUMENTS"

exit