manage-etc-hosts.sh

#!/bin/sh

# PATH TO YOUR HOSTS FILE
ETC_HOSTS=/etc/hosts

# DEFAULT IP FOR HOSTNAME
IP="127.0.0.1"

# Hostname to add/remove.
function removehost() {
    HOSTNAME=$1
    LOCAL_IP=$2
    NEED_IP="$(getNeedIp $LOCAL_IP $IP)"
    if [ -n "$(grep $HOSTNAME /etc/hosts)" ]; then
        echo "$HOSTNAME Found in your $ETC_HOSTS, Removing now...";
        sudo sed -i".bak" "/$HOSTNAME/d" $ETC_HOSTS
    else
        echo "$HOSTNAME was not found in your $ETC_HOSTS";
    fi
}

function addhost() {
    HOSTNAME=$1
    LOCAL_IP=$2
    NEED_IP="$(getNeedIp $LOCAL_IP $IP)"

    HOSTS_LINE="$NEED_IP\t$HOSTNAME"
    if [ -n "$(grep $HOSTNAME /etc/hosts)" ]; then
            echo "$HOSTNAME already exists : $(grep $HOSTNAME $ETC_HOSTS)"
        else
            echo "Adding $HOSTNAME to your $ETC_HOSTS";
            sudo -- 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
}

function getNeedIp(){
    if [[ -n $1 ]]; then 
        echo $1 
    else 
        echo $2 
    fi
}