#!/bin/bash
# copy from https://gist.github.com/irazasyed/a7b0a079e7727a4315b9

ETC_HOSTS=/etc/hosthosts
IP="$1"
HOSTNAME=$2

echo "removing host";
if [ -n "$(grep $HOSTNAME $ETC_HOSTS)" ]
then
    echo "$HOSTNAME Found in your $ETC_HOSTS, Removing now...";
    sed -i".bak" "/$HOSTNAME/d" $ETC_HOSTS
else
    echo "$HOSTNAME was not found in your $ETC_HOSTS";
fi

echo "adding host";
HOSTS_LINE="$IP     $HOSTNAME"
if [ -n "$(grep $HOSTNAME $ETC_HOSTS)" ]
    then
        echo "$HOSTNAME already exists : $(grep $HOSTNAME $ETC_HOSTS)"
    else
        echo "Adding $HOSTNAME to your $ETC_HOSTS";
        sh -c -e "echo '$HOSTS_LINE' >> $ETC_HOSTS";

        if [ -n "$(grep $HOSTNAME $ETC_HOSTS)" ]
            then
                echo "$HOSTNAME was added succesfully \n $(grep $HOSTNAME $ETC_HOSTS)";
            else
                echo "Failed to Add $HOSTNAME, Try again!";
        fi
fi
#$@