#!/bin/bash

HEIGHT=30
WIDTH=80
CHOICE_HEIGHT=4
BACKTITLE="Do you want to block China and/or Korea?"
TITLE="Country Block"
MENU="A lot of spam, as well as malicious traffic originates from Korea and China. If you don't plan on having to ever have those countries connect to your server you may block them.
	This will add a cron that will update weekly, and block all IP blocks to those countries you choose
	Choose one of the following options:"

OPTIONS=(1 "China"
         2 "Korea"
         3 "China and Korea"
		 4 "Do nothing")

CHOICE=$(dialog --clear \
                --backtitle "$BACKTITLE" \
                --title "$TITLE" \
                --menu "$MENU" \
                $HEIGHT $WIDTH $CHOICE_HEIGHT \
                "${OPTIONS[@]}" \
                2>&1 >/dev/tty)

clear
case $CHOICE in
        1)
            echo "Are you Donald Trump?"
			cp conf/blacklist/china /etc/cron.weekly/china
			chmod +x /etc/cron.weekly/china
			time /etc/cron.weekly/china
            ;;
        2)
            echo "Starting the Korean war again"
			cp conf/blacklist/korea /etc/cron.weekly/korea
			chmod +x /etc/cron.weekly/korea
			time /etc/cron.weekly/korea
            ;;
        3)
            echo "Blocking almost 1/3 of the world"
			cp conf/blacklist/sinokorea /etc/cron.weekly/sinokorea
			chmod +x /etc/cron.weekly/sinokorea
			time /etc/cron.weekly/sinokorea
            ;;
		4)  echo "Doing nothing"
		    ;;
esac