setup.sh

#!bin/bash

#run as root checker
if [ "$EUID" -ne 0 ]
  then echo "Please run as root"
  exit
fi

#Requirement checker
command -v docker >/dev/null 2>&1 || { echo >&2 "This service requires Docker, but your computer doesn't have it. Install Docker then try again. Aborting."; exit 1; }
command -v docker-compose >/dev/null 2>&1 || { echo >&2 "This service requires Docker-Compose, but your computer doesn't have it. Install Docker-Compose then try again. Aborting."; exit 1;}
command -v mosquitto >/dev/null 2>&1 || { echo >&2 "This service requires Mosquitto Broker, but your computer doesn't have it. Install Mosquitto Broker then try again. Aborting."; exit 1; }

#Opening
printf '
 __  __    _  _____  _       _____ _        _    _   _  ____    _        _    ____
|  \/  |  / \|_   _|/ \     | ____| |      / \  | \ | |/ ___|  | |      / \  | __ )
| |\/| | / _ \ | | / _ \    |  _| | |     / _ \ |  \| | |  _   | |     / _ \ |  _ \
| |  | |/ ___ \| |/ ___ \   | |___| |___ / ___ \| |\  | |_| |  | |___ / ___ \| |_) |
|_|  |_/_/   \_|_/_/   \_\  |_____|_____/_/   \_|_| \_|\____|  |_____/_/   \_|____/'

echo -e "\n"
echo ----------------------------------------------------------
echo -e "\tBigData Platform for Network Research"
echo ----------------------------------------------------------
echo -e "\n"

YELLOW='\033[1;33m'
NC='\033[0m' # No Color

#Note
echo -----------
printf "${YELLOW}Warning: ${NC}If Using Virtual Machine, Please use NAT Mode\n"
echo -e "-----------\n"

#All input
#===========================================================

printf "If You have Your own ${YELLOW}DATASET${NC}, please put Your dataset in a folder ${YELLOW}${PWD}/resource${NC}.\n"
printf "So you can access it via ${YELLOW}'/resource'${NC} folder on zeppelin-notebook\n"
read -p "Press any key to resume ..."

echo -e "\nAvailable Network Interface : `ls -C /sys/class/net`"
echo "Network Interface Card for Tapping (ex: eth0)"
read -p "Your Choice: " NETINT

echo -e "\nInput Mongo Chart UserAdmin Account"
read -p "Username: " USERNAME
read -s -p "Password: " PASSWORD

echo -e "\n\nWhat kind Mode do you want to use?\n\t1. Standalone\n\t2. Spark-Cluster"
read -p "Your choice : " MELMODE

RULE_CHOICE=1

#Choose commpose mode
if [[ ! $MELMODE -eq 1 && ! $MELMODE -eq 2 ]]; then
  echo -e "Choose a valid choice.\nExited."
  exit 1
fi
#============================================================

#Export Folder Project Path
export MEL_PATH=$PWD

#Get NIC IP then override env file
ip4=$(/sbin/ip -o -4 addr list $NETINT | awk '{print $4}' | cut -d/ -f1)
subnet=$(/sbin/ifconfig $NETINT | awk '/Mask:/{ print $4;} ')

##Netflowmeter
sed -i 's/^NETINT=.*/NETINT='$NETINT'/' envfile/netflowmeter.env
sed -i 's/^MQTT_HOST=.*/MQTT_HOST='$ip4'/' envfile/netflowmeter.env
##Snort
sed -i 's/^PROTECTED_SUBNET=.*/PROTECTED_SUBNET='$subnet'\/24/' envfile/snort.env 
sed -i 's/^ALERT_MQTT_SERVER=.*/ALERT_MQTT_SERVER='$ip4'/' envfile/snort.env
sed -i 's/^NETINT=.*/NETINT='$NETINT'/' envfile/snort.env

#Running MQTT
systemctl start mosquitto

#Allow mqqt in firewall
echo -e "\nAdding rule for port-forwarding"
ufw allow 1883 
ufw allow 7077 

#Remove Exist Volume
rm -rf ./volume/mongochart ./spark ./notebooks

#Starting Big Data
echo -e "\nStarting Compose BigData..."
echo "-----------------------------------\ "

if [[ $MELMODE -eq 1 ]]; then
  echo "Using Standalone MataElangLab.."
  compose_file="docker-compose.yml"
fi

if [[ $MELMODE -eq 2 ]]; then
  echo "Using Single-Cluster MataElangLab.."
  compose_file="docker-compose-spark.yml"
fi

#Set Config for Mode
mkdir -p /etc/mataelanglab
cat > /etc/mataelanglab/mode <<EOL
${compose_file}
EOL

docker-compose --file $compose_file up -d

echo -----------------------------------/
chars="/-\|"
container_name="connect"

#Set MQTT IP in confluent-config file for connection between Kafka-MQTT
sed -i 's|tcp://.*|tcp://'$ip4':1883",|' connector-config/nfm-connector.json
sed -i 's|tcp://.*|tcp://'$ip4':1883",|' connector-config/snort-connector.json

while [ "$( docker container inspect -f '{{.State.Running}}' $container_name )" == "true" ];
do
  for (( i=0; i<${#chars}; i++ )); do
    sleep 0.1
    echo -en "Configuring Kafka Connector...[${chars:$i:1}]" "\r"
  done

  if [ "$( docker container inspect -f '{{.State.Health.Status}}' $container_name )" == "healthy" ]
  then
    curl -s -X POST -H 'Content-Type: application/json' --data @connector-config/nfm-connector.json http://localhost:8083/connectors
    curl -s -X POST -H 'Content-Type: application/json' --data @connector-config/snort-connector.json http://localhost:8083/connectors
    curl -s -X POST -H 'Content-Type: application/json' --data @connector-config/nfm-mongo-sink.json http://localhost:8083/connectors
    curl -s -X POST -H 'Content-Type: application/json' --data @connector-config/snort-mongo-sink.json http://localhost:8083/connectors
    break;
  elif [ "$( docker container inspect -f '{{.State.ExitCode}}' $container_name )" == 1 ]
  then
    docker-compose --file $compose_file down -v
    echo "Some process has failed to Instance"
    exit
  fi
done

#Display Available Link
echo -e "\n\n-----------------------------------"
echo -e "${YELLOW}Available WebApp${NC}"
echo -e "(For easier Please Bookmark or Copy link)"
echo -e "-----------------------------------"

#Create Charts users
if [ "$( docker container inspect -f '{{.State.Running}}' mongo-charts )" == "true" ];
then
  echo -en "MongoDB Chart UserAdmin... \n"
  sleep 10
  docker exec -it mongo-charts bash -c \
  "charts-cli add-user \
  --first-name "$USERNAME" \
  --last-name \"lab\" \
  --email \"$USERNAME@mail.com\" \
  --password "$PASSWORD" \
  --role \"UserAdmin\""
fi

cat scripts/web-info.txt

echo -e "For easier debugging process, You can install portainer"

echo -e "\n\n-----------------------------------"
echo -e "${YELLOW}Setup completed.${NC}"
echo -e "-----------------------------------"