#!/bin/bash
payload="/etc/squid/payload.txt"
tput setaf 7 ; tput setab 4 ; tput bold ; printf '%35s%s%-10s\n' "Adicionar Host ao Squid3" ; tput sgr0
if [ ! -f "$payload" ]
then
    tput setaf 7 ; tput setab 4 ; tput bold ; echo "" ; echo "Arquivo $payload não encontrado" ; tput sgr0
    exit 1
else
    tput setaf 2 ; tput bold ; echo ""; echo "Domínios atuais no arquivo $payload:" ; tput sgr0
    tput setaf 3 ; tput bold ; echo "" ; cat $payload ; echo "" ; tput sgr0
    read -p "Digite o domínio que deseja adicionar a lista: " host
    if [[ -z $host ]]
    then
        tput setaf 7 ; tput setab 4 ; tput bold ; echo "" ; echo "Você digitou um domínio vazio ou não existente!" ; echo "" ; tput sgr0
        exit 1
    else
        if [[ `grep -c "^$host" $payload` -eq 1 ]]
        then
            tput setaf 7 ; tput setab 4 ; tput bold ; echo "" ; echo "O domínio $host já existe no arquivo $payload" ; echo "" ; tput sgr0
            exit 1
        else
            if [[ $host != \.* ]]
            then
                tput setaf 7 ; tput setab 4 ; tput bold ; echo "" ; echo "Você deve adicionar um domínio iniciando-o com um ponto!" ; echo "Por exemplo: .phreaker56.xyz" ; echo "Não é necessário adicionar subdomínios para domínios que já estão no arquivo" ; echo "Ou seja, não é necessário adicionar recargawap.claro.com.br" ; echo "se o domínio .claro.com.br já estiver no arquivo." ; echo ""; tput sgr0
                exit 1
            else
                echo "$host" >> $payload && grep -v "^$" $payload > /tmp/a && mv /tmp/a $payload
                tput setaf 7 ; tput setab 1 ; tput bold ; echo "" ; echo "Arquivo $payload atualizado, o domínio foi adicionado com sucesso:" ; tput sgr0
                tput setaf 3 ; tput bold ; echo "" ; cat $payload ; echo "" ; tput sgr0
                if [ ! -f "/etc/init.d/squid" ]
                then
                    service squid reload
                else
                    /etc/init.d/squid reload
                fi
                tput setaf 7 ; tput setab 1 ; tput bold ; echo "" ; echo "O Proxy Squid3 foi recarregado com sucesso!" ; echo "" ; tput sgr0
                exit 1
            fi
        fi
    fi
fi