From a52444177430f6552333a0e1e3aa497341143213 Mon Sep 17 00:00:00 2001
From: Alfonso <vivancos@outlook.es>
Date: Fri, 2 Jun 2023 12:47:20 +0200
Subject: [PATCH] Added colors

---
 blocklist.sh | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/blocklist.sh b/blocklist.sh
index 4a0e472..3640413 100755
--- a/blocklist.sh
+++ b/blocklist.sh
@@ -1,4 +1,9 @@
 #!/bin/bash
+RED='\033[1;31m'
+GREEN='\033[1;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[1;35m'
+NC='\033[0m'
 if [ "$(whoami)" != "root" ]; then
     SUDO=sudo
 fi
@@ -17,8 +22,10 @@ abuse_key="INSERT_YOUR_API_KEY_HERE" #https://www.abuseipdb.com/account/api
 installed() {
     # $1 should be the command to look for
     if ! [ -x "$(command -v $1)" ]; then
-        echo "$1 is not available. Please install it and run again."
+        echo -e "${RED}$1 is not available. Please install it and run again.${NC}"
         exit 1
+        else
+        echo -e "${GREEN}$1 installed${NC}"
     fi
 }
 
@@ -28,16 +35,16 @@ installed sort
 installed jq
 installed grep
 
-echo "Downloading the most recent IP list from $BLOCKLISTDE ... and adding them to ipset blocklistde"
+echo -e "${YELLOW}Downloading the most recent IP list from $BLOCKLISTDE ... and adding them to ipset blocklistde${NC}"
 ${SUDO} $(whereis ipset | cut -d" " -f 2) create blocklistde hash:ip
 curl -s https://lists.blocklist.de/lists/all.txt | grep -E -o "(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)" | xargs -L1 ${SUDO} $IPSET_PATH add blocklistde 2>&1
-echo "Downloading the most recent IP list from $CRWALERS ... and adding them to ipset crawler_bots"
+echo -e "${YELLOW}Downloading the most recent IP list from $CRWALERS ... and adding them to ipset crawler_bots${NC}"
 ${SUDO} $(whereis ipset | cut -d" " -f 2) create crawler_bots hash:ip
 curl -s https://isc.sans.edu/api/threatcategory/research?json | jq '.[] | {ipv4}' | grep ':' | awk '{ print $2 }' | tr -d '"' | xargs -L1 ${SUDO} $IPSET_PATH add crawler_bots 2>&1
-echo "Downloading the most recent IP list from $ABUSE and adding them to abuseipdb"
+echo -e "${YELLOW}Downloading the most recent IP list from $ABUSE and adding them to abuseipdb${NC}"
 ${SUDO} $(whereis ipset | cut -d" " -f 2) create abuseipdb hash:ip
 curl -G -H "key: $abuse_key" -H "Accept: text/plain" -d confidenceMinimum=90 https://api.abuseipdb.com/api/v2/blacklist | grep -v : | xargs -L1 ${SUDO} $IPSET_PATH add abuseipdb 2>&1
-echo "Adding the iptables rules..."
+echo -e "${YELLOW}Adding the iptables rules...${NC}"
 ${SUDO} $IPTABLES_PATH -I INPUT -m set --match-set crawler_bots src -j DROP
 ${SUDO} $IPTABLES_PATH -I INPUT -m set --match-set blocklistde src -j DROP
 ${SUDO} $IPTABLES_PATH -I INPUT -m set --match-set abuseipdb src -j DROP