kalibox

#!/bin/bash

# LICENSE: https://github.com/sakibulalikhan/pentestbox/blob/main/LICENSE

set -e 

print_banner() {
    printf " ▄▄▄   ▄ ▄▄▄▄▄▄ ▄▄▄     ▄▄▄ ▄▄▄▄▄▄▄ ▄▄▄▄▄▄▄ ▄▄   ▄▄  \n"
    printf "█   █ █ █      █   █   █   █  ▄    █       █  █▄█  █ \n"
    printf "█   █▄█ █  ▄   █   █   █   █ █▄█   █   ▄   █       █ \n"
    printf "█      ▄█ █▄█  █   █   █   █       █  █ █  █       █ \n"
    printf "█     █▄█      █   █▄▄▄█   █  ▄   ██  █▄█  ██     █  \n"
    printf "█    ▄  █  ▄   █       █   █ █▄█   █       █   ▄   █ \n"
    printf "█▄▄▄█ █▄█▄█ █▄▄█▄▄▄▄▄▄▄█▄▄▄█▄▄▄▄▄▄▄█▄▄▄▄▄▄▄█▄▄█ █▄▄█ \n"
    printf "+---------------------------------------------------+\n"
    printf "|            Author : @sakibulalikhan               |\n"
    printf "+---------------------------------------------------+\n\n"
}

command_exists() {
    command -v "$1" >/dev/null 2>&1
}

container_exists() {
    local container_name="$1"
    docker ps -a --format '{{.Names}}' | grep -w "$container_name" >/dev/null 2>&1
}

create_container() {
    echo "Cannot find container kalibox."
    read -p "Create it now, out of image kalilinux/kali-rolling? [Y/n]: " create_container
    while [[ ! "$create_container" =~ ^[YyNn]$ ]]; do
        read -p "Invalid input. Please enter Y or N: " create_container
    done

    if [[ "$create_container" =~ ^[Yy]$ ]]; then
        read -p "Do you want to use NVIDIA support? [Y/n]: " use_nvidia
        while [[ ! "$use_nvidia" =~ ^[YyNn]$ ]]; do
            read -p "Invalid input. Please enter Y or N: " use_nvidia
        done

        if [[ "$use_nvidia" =~ ^[Yy]$ ]]; then
            distrobox create --nvidia -n kalibox -i kalilinux/kali-rolling || {
                echo "Failed to create container with NVIDIA support."
                exit 1
            }
        else
            distrobox create -n kalibox -i kalilinux/kali-rolling || {
                echo "Failed to create container."
                exit 1
            }
        fi
    else
        echo "For creating it, run this command:"
        echo "  distrobox create -n <name-of-container> -i <image>"
    fi
}

print_banner

if ! command_exists "distrobox"; then
    echo "distrobox command not found. Installing..."
    wget -qO- https://raw.githubusercontent.com/89luca89/distrobox/main/install | sudo sh
fi

sudo systemctl start docker || { echo "Failed to start Docker service."; exit 1; }
sudo chmod 666 /var/run/docker.sock || { echo "Failed to change permissions on Docker socket."; exit 1; }

case "$1" in
    ""|start)
        echo "Starting services..."
        if ! container_exists "kalibox"; then
            create_container
        fi
        distrobox enter kalibox
        ;;
    stop)
        echo "Stopping services..."
        distrobox stop kalibox || echo "Failed to stop kalibox."
        ;;
    remove)
        echo "Removing container kalibox..."
        if container_exists "kalibox"; then
            distrobox rm kalibox || echo "Failed to remove kalibox."
        else
            echo "Container kalibox does not exist."
        fi
        ;;
    help)
        echo "Usage: kalibox [start|stop|remove|help]"
        echo "       kalibox start  : Start the kalibox services"
        echo "       kalibox stop   : Stop the kalibox services"
        echo "       kalibox remove : Remove the kalibox container"
        echo "       kalibox help   : Show the help message"
        ;;
    *)
        echo "Invalid command. Use 'kalibox help' for usage information."
        ;;
esac