!/bin/bash

Set default parameters

action=$1
domain=$2
owner=$(who am i | awk '{print $1}')
email='webmaster@localhost.com'
sitesEnable='/etc/apache2/sites-enabled/'
sitesAvailable='/etc/apache2/sites-available/'
userDir=$(eval echo ~${SUDO_USER})'/www/'

if [ "$(whoami)" != 'root' ]; then
echo "You have no permission to run $0 as non-root user. Use sudo !!!"
exit 1;
fi

if [ "$action" != 'create' ] && [ "$action" != 'delete' ]
then
echo "You need to prompt for action (create or delete) -- Lower-case only !!!!!!"
exit 1;
fi

while [ "$domain" == "" ]
do
echo -e "Please provide domain. e.g.dev,staging"
read domain
done

rootdir=${domain//./}

if [ "$action" == 'create' ]
then

check if domain already exists

    if [ -e $sitesAvailable$domain ]; then
        echo -e 'This domain already exists.\nPlease Try Another one'
        exit;
    fi

    ### check if directory exists or not
    if [ -d $userDir$rootdir ]; then
        echo -e 'Directory already exists !'
        exit;
    fi

    ### create the directory
    mkdir $userDir$rootdir

    ### create virtual host rules file
    if ! echo "<VirtualHost *:80>
        ServerAdmin $email
        ServerName $domain
        ServerAlias $domain www.$domain
        DocumentRoot $userDir$rootdir
        <Directory />
            AllowOverride All
        </Directory>
        <Directory $rootdir>
            AllowOverride All
        </Directory>
        ErrorLog /var/log/apache2/$domain
         LogLevel error
        CustomLog /var/log/apache2/$domain custom
    </VirtualHost>" > $sitesAvailable$domain
    then
        echo -e 'There is an ERROR create $domain file'
        exit;
    else
        echo -e '\nNew Virtual Host Created\n'
    fi

    ### Add domain in /etc/hosts
    if ! echo "127.0.0.1       $domain" >> /etc/hosts
    then
        echo "ERROR: Not able write in /etc/hosts"
        exit;
    else
        echo -e "Host added to /etc/hosts file \n"
    fi

    ### enable website
    a2ensite $domain

    ### restart Apache
    /etc/init.d/apache2 reload

    ### give permission to root dir
    chmod 755 $userDir$rootdir

    ### write test file in the new domain dir
    if ! echo "<?php echo phpinfo(); ?>" > $userDir$rootdir/phpinfo.php
    then
        echo "ERROR: Not able to write in file "$userDir"/"$rootdir"/phpinfo.php. Please check permissions."
        exit;
    else
        echo "Added content to "$userDir$rootdir"/phpinfo.php."
    fi

    if [ "$owner" == ""  ]; then
        chown -R $(whoami):$(whoami) $userDir$rootdir
    else
        chown -R $owner:$owner $userDir$rootdir
    fi

    ### show the finished message
    echo -e "Complete!
    You now have a new Virtual Host
    Your new host is: http://"$domain"
    And its located at "$userDir$rootdir
    exit;
else
    ### check whether domain already exists
    if ! [ -e $sitesAvailable$domain ]; then
        echo -e 'This domain dont exists.\nPlease Try Another one'
        exit;
    fi

    ### check if directory exists or not
    if ! [ -d $userDir$rootdir ]; then
        echo -e 'Directory not exists !'
        exit;
    fi

    ### Delete the directory
    rm -rf $userDir$rootdir

    ### Delete virtual host rules file
    rm $sitesAvailable$domain

    ### Delete domain in /etc/hosts
    newhost=${domain//./\\.}
    sed -i "/$newhost/d" /etc/hosts

    ### enable website
    rm $sitesEnable$domain

    ### restart Apache
    /etc/init.d/apache2 reload

    ### show the finished message
    echo -e "Complete!
    You just removed Virtual Host "$domain
    exit 0;

fi