#!/bin/sh # START set -e # VARIABLES PROGNAME=$(basename $0) CONFIG=($HOME/.maza/) HOST_FILE=(/etc/hosts) COLOR_RED=`tput setaf 1` COLOR_GREEN=`tput setaf 2` COLOR_RESET=`tput sgr0` LIST="list" START_TAG="## MAZA - List ad blocking" END_TAG="## END MAZA" # FUNCTIONS ## HELP usage() { if [ "$*" != "" ] ; then echo "Error: $*" fi cat << EOF Usage: $PROGNAME [OPTION] Simple and efficient local ad blocking throughout the network. Options: status Check if it's active or not update Update the list of DNS to be blocked start Activate blocking DNS. stop Stop blocking DNS. --help Display this usage message and exit EOF exit 1 } status() { if grep -qF "$START_TAG" "$HOST_FILE";then echo "${COLOR_GREEN}ENABLED${COLOR_RESET}" else echo "${COLOR_RED}DISABLED${COLOR_RESET}" fi } update() { # Make conf folder rm -f $CONFIG/$LIST mkdir -p $CONFIG # Download DNS list curl -L -s "https://pgl.yoyo.org/adservers/serverlist.php?showintro=0;hostformat=hosts" -o "$CONFIG/$LIST" # Clear list ## Get PRE tag sed -i .bak -n "/
/,/<\/pre>/p" "$CONFIG/$LIST"
    ## Remove first line
    sed -i .bak '1d' "$CONFIG/$LIST"
    ## Remove last line
    sed -i .bak '$ d' "$CONFIG/$LIST"
    ## Add start tag DNS list in first line
    sed -i .bak "1i\\
        $START_TAG
        " "$CONFIG/$LIST"
    ## Add end tag DNS list in first line
    echo $END_TAG >> "$CONFIG/$LIST"
    # Notify user
    echo "${COLOR_GREEN}Done!${COLOR_RESET}"
}

start() {
    cat "$CONFIG/$LIST" "$HOST_FILE" >> "$HOST_FILE"
    # Notify user
    echo "${COLOR_GREEN}ENABLED!${COLOR_RESET}"
}

stop() {
    echo "stopdf"
}

# CONTROLE ARGUMENTS
isArg=""

while [ $# -gt 0 ] ; do
    case "$1" in
    --help)
        usage
        ;;
    status)
        isArg="1"
        status
        ;;
    update)
        isArg="1"
        update
        ;;
    start)
        isArg="1"
        start
        ;;
    stop)
        isArg="1"
        stop
        ;;
    *)
    esac
    shift
done

if [ -z $isArg ] ; then
    usage "Not enough arguments"
fi