#!/bin/bash
# info: add mail domain
# options: USER DOMAIN [ANTISPAM] [ANTIVIRUS] [DKIM] [DKIM_SIZE]
#
# The function adds MAIL domain.


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

# Argument defenition
user=$1
domain=$(idn -t --quiet -u "$2" )
domain=$(echo $domain | sed -e 's/\.*$//g' -e 's/^\.*//g')
domain=$(echo $domain | tr '[:upper:]' '[:lower:]')
domain_idn=$(idn -t --quiet -a "$domain")
antispam=${3-yes}
antivirus=${4-yes}
dkim=${5-yes}
dkim_size=${6-512}

# Includes
source $VESTA/conf/vesta.conf
source $VESTA/func/main.sh
source $VESTA/func/domain.sh


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

check_args '2' "$#" 'USER DOMAIN [ANTISPAM] [ANTIVIRUS] [DKIM] [DKIM_SIZE]'
validate_format 'user' 'domain' 'antispam' 'antivirus' 'dkim' 'dkim_size'
is_system_enabled "$MAIL_SYSTEM" 'MAIL_SYSTEM'
is_object_valid 'user' 'USER' "$user"
is_object_unsuspended 'user' 'USER' "$user"
is_domain_new 'mail'
is_package_full 'MAIL_DOMAINS'


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

# Adding domain directory
mkdir $HOMEDIR/$user/conf/mail/$domain
touch $HOMEDIR/$user/conf/mail/$domain/aliases
touch $HOMEDIR/$user/conf/mail/$domain/protection
touch $HOMEDIR/$user/conf/mail/$domain/passwd
chown -R dovecot:mail $HOMEDIR/$user/conf/mail/$domain
chmod 770 $HOMEDIR/$user/conf/mail/$domain
chmod 660 $HOMEDIR/$user/conf/mail/$domain/*
mkdir $HOMEDIR/$user/mail/$domain_idn
chown $user:mail $HOMEDIR/$user/mail/$domain_idn
chmod 770 $HOMEDIR/$user/mail/$domain_idn

# Adding symlink
ln -s $HOMEDIR/$user/conf/mail/$domain /etc/$MAIL_SYSTEM/domains/$domain_idn

# Adding antispam protection
if [ "$antispam" = 'yes' ]; then
    echo 'antispam' >> $HOMEDIR/$user/conf/mail/$domain/protection
fi

# Adding antivirus protection
if [ "$antivirus" = 'yes' ]; then
    echo 'antivirus' >> $HOMEDIR/$user/conf/mail/$domain/protection
fi

# Adding dkim
if [ "$dkim" = 'yes' ]; then
    openssl genrsa -out $USER_DATA/mail/$domain.pem $dkim_size &>/dev/null
    openssl rsa -pubout -in $USER_DATA/mail/$domain.pem \
        -out $USER_DATA/mail/$domain.pub &>/dev/null
    chmod 660 $USER_DATA/mail/$domain.*

    cp $USER_DATA/mail/$domain.pem $HOMEDIR/$user/conf/mail/$domain/dkim.pem
    chown root:mail $HOMEDIR/$user/conf/mail/$domain/dkim.pem
    chmod 660 $HOMEDIR/$user/conf/mail/$domain/dkim.pem

    # Adding dkim dns records
    check_dns_domain=$(is_object_valid 'dns' 'DOMAIN' "$domain")
    if [ "$?" -eq 0 ]; then
        p=$(cat $USER_DATA/mail/$domain.pub|grep -v ' KEY---'|tr -d '\n')
        record='_domainkey'
        policy="\"t=y; o=~;\""
        $BIN/v-add-dns-record $user $domain $record TXT "$policy"

        record='mail._domainkey'
        selector="\"k=rsa\; p=$p\""
        $BIN/v-add-dns-record $user $domain $record TXT "$selector"
    fi
fi

# Adding domain to vesta db
s="DOMAIN='$domain' ANTIVIRUS='$antivirus' ANTISPAM='$antispam' DKIM='$dkim'"
s="$s ACCOUNTS='0' U_DISK='0' CATCHALL='' SUSPENDED='no' TIME='$TIME'"
s="$s DATE='$DATE'"

echo $s >> $USER_DATA/mail.conf
touch $USER_DATA/mail/$domain.conf
chmod 660 $USER_DATA/mail.conf
chmod 660 $USER_DATA/mail/$domain.conf


#----------------------------------------------------------#
#                       Vesta                              #
#----------------------------------------------------------#

# Increasing domain value
increase_user_value "$user" '$U_MAIL_DOMAINS'
if [ "$dkim" = 'yes' ]; then
    increase_user_value "$user" '$U_MAIL_DKMI'
fi

# Logging
log_history "added mail domain $domain"
log_event "$OK" "$EVENT"

exit