#!/bin/bash

# Define color codes
INFO='\033[0;36m'  # Cyan
BANNER='\033[0;35m' # Magenta
WARNING='\033[0;33m'
ERROR='\033[0;31m'
SUCCESS='\033[0;32m'
NC='\033[0m' # No Color


if ! command -v docker &> /dev/null
then
    echo "Docker could not be found. Installing Docker..."
    sudo apt update
    sudo apt install -y docker.io
    sudo systemctl start docker
    sudo systemctl enable docker
    echo "Docker installed successfully."
fi

# Check if the directory exists
if [ -d "network3-docker" ]; then
  echo -e "${INFO}Directory network3-docker already exists.${NC}"
else
  # Create the directory
  mkdir network3-docker
  echo -e "${SUCCESS}Directory network3-docker created.${NC}"
fi

# Navigate into the directory
cd network3-docker

# Retrieve public IP
public_ip=$(curl -s ifconfig.me)
if [ -z "$public_ip" ]; then
  echo -e "${ERROR}Failed to retrieve public IP address. Exiting.${NC}"
  exit 1
fi

# Create or replace the Dockerfile with the specified content
cat <<EOL > Dockerfile
# Use an official Ubuntu as a parent image
FROM ubuntu:latest
# Install wget, ufw, tar, nano, sudo, net-tools, iproute2, and procps
RUN apt-get update && apt-get install -y \\
    wget \\
    ufw \\
    tar \\
    nano \\
    sudo \\
    net-tools \\
    iproute2 \\
    procps
# Download and extract Network3
RUN wget https://network3.io/ubuntu-node-v2.1.1.tar && \\
    tar -xf ubuntu-node-v2.1.1.tar && \\
    rm ubuntu-node-v2.1.1.tar
# Change directory
WORKDIR /ubuntu-node
# Allow port 8080
RUN ufw allow 8080
# Start the node and provide a shell
CMD ["bash", "-c", "bash manager.sh up; bash manager.sh key; exec bash"]
EOL

# Detect existing network3-docker instances and find the highest instance number
existing_instances=$(docker ps -a --filter "name=network3-docker-" --format "{{.Names}}" | grep -Eo 'network3-docker-[0-9]+' | grep -Eo '[0-9]+'$ | sort -n | tail -1)

# Set the instance number
if [ -z "$existing_instances" ]; then
  instance_number=1
else
  instance_number=$((existing_instances + 1))
fi

# Set the container name
container_name="network3-docker-$instance_number"

# Calculate the port number
port_number=$((8080 + instance_number - 1))

# Build the Docker image with the specified name
sudo docker build -t $container_name .

# Check if ufw is installed and add rule for the port number
if command -v ufw > /dev/null; then
  echo -e "${INFO}Configuring UFW to allow traffic on port $port_number...${NC}"
  sudo ufw allow $port_number
  echo -e "${SUCCESS}UFW configured successfully.${NC}"
fi

# Display the completion message and command to view logs
echo -e "${SUCCESS}The Docker container will be built and will run on port $port_number.${NC}"
echo -e "${INFO}To consult the dashboard, visit:${NC}"
echo -e "${BANNER}https://account.network3.ai/main?o=$public_ip:$port_number${NC}"
echo -e "${INFO}Use the key that will be displayed to link node with your email${NC}"

# Run the Docker container with the necessary privileges and an interactive shell
sudo docker run -it --cap-add=NET_ADMIN --device /dev/net/tun --name $container_name -p $port_number:8080 $container_name