#!/bin/bash

#Colors
RED="\e[1;31m"
GREEN="\e[1;32m"
YELLOW="\e[1;33m"
BLUE="\e[1;34m"
DEFAULT="\e[0m"

# Check if the script is run with root privileges
if [ "$EUID" -ne 0 ]; then
  echo "Please run this script as root or using sudo."
  exit 1
fi

echo -e "${YELLOW}Do you want to Update the system?: [Y/n]${DEFAULT}"
read -r confirmation
if [[ ! "$confirmation" =~ ^[Nn]|[Nn][Oo]$ ]]; then
	echo "Updating..."
	sudo apt-get update -y
	sudo apt-get upgrade -y
else
	echo "You choose no"
fi

echo -e "${YELLOW}Do you want to Install VirtualBox-7.2?: [Y/n]${DEFAULT}"
read -r confirmation
if [[ ! "$confirmation" =~ ^[Nn]|[Nn][Oo]$ ]]; then
	echo "Installing..."
	if [ $(dpkg -l | grep -c "virtualbox" ) -eq 0 ]; then
		sudo apt-get install -y \
			ca-certificates \
			wget \
			gnupg \
			lsb-release \

		echo -e "${BLUE}Adding Docker's official GPG key${DEFAULT}"
			wget -qO - https://www.virtualbox.org/download/oracle_vbox_2016.asc | gpg --dearmor | sudo tee /etc/apt/keyrings/oracle-virtualbox-2016.gpg -

		echo -e "${BLUE}Add the repository to Apt sources${DEFAULT}"
			sudo tee /etc/apt/sources.list.d/virtualbox.sources <<EOF
Types: deb
URIs: https://download.virtualbox.org/virtualbox/debian
Suites: $(. /etc/os-release && echo "${UBUNTU_CODENAME:-$VERSION_CODENAME}")
Components: contrib
Architectures: $(dpkg --print-architecture)
Signed-By: /etc/apt/keyrings/oracle-virtualbox-2016.gpg
EOF

			sudo apt-get update -y
			sudo apt-get install -y \
				linux-headers-$(uname -r) \
				virtualbox-7.2 \
				dkms

		echo -e "${BLUE}Make VirtualBox run with Extension Pack${DEFAULT}"
			wget https://download.virtualbox.org/virtualbox/7.2.8/Oracle_VirtualBox_Extension_Pack-7.2.8.vbox-extpack
			sudo vboxmanage extpack install --replace --accept-license=$(sha256sum Oracle_VirtualBox_Extension_Pack-7.2.8.vbox-extpack | awk '{print $1}') Oracle_VirtualBox_Extension_Pack-7.2.8.vbox-extpack
			rm -f Oracle_VirtualBox_Extension_Pack-7.2.8.vbox-extpack

		echo -e "${BLUE}Make VirtualBox run with admin privileges${DEFAULT}"
			sudo groupadd vboxusers
			sudo usermod -aG vboxusers $USER
			newgrp vboxusers
	else
		echo -e "${GREEN}  \"Some packet like 'virtualbox' installed, seems VirtualBox is already Installed\"${DEFAULT}"
	fi
else
	echo "You choose no"
fi

echo -e "${YELLOW}Do you want Vagrant too?: [y/N]${DEFAULT}"
read -r confirmation
if [[ "$confirmation" =~ ^[Yy]|[Yy][Ee][Ss]$ ]]; then
	echo "Installing..."
	if [ $(dpkg -l | grep -c "vagrant " ) -eq 0 ]; then
		echo -e "${BLUE}Adding Vagrant's official GPG key${DEFAULT}"
			wget -O- https://apt.releases.hashicorp.com/gpg | sudo gpg --dearmor -o /usr/share/keyrings/hashicorp-archive-keyring.gpg

			echo \
			"deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/hashicorp-archive-keyring.gpg] https://apt.releases.hashicorp.com \
			$(. /etc/os-release && echo $VERSION_CODENAME) main" | \
			sudo tee /etc/apt/sources.list.d/hashicorp.list

			sudo apt-get update -y
			sudo apt-get install -y \
				vagrant
	else
		echo -e "${GREEN}  \"Some packet like 'vagrant ' installed, seems VirtualBox is already Installed\"${DEFAULT}"
	fi
else
	echo "You choose no"
fi
echo -e "${YELLOW}Do you want to Remove VirtualBox?: [y/N]${DEFAULT}"
read -r confirmation
if [[ "$confirmation" =~ ^[Yy]|[Yy][Ee][Ss]$ ]]; then
	echo -e "${RED}Are you sure to Remove VirtualBox?: [y/N]${DEFAULT}"
	read -r confirmationagain
	if [[ "$confirmationagain" =~ ^[Yy]|[Yy][Ee][Ss]$ ]]; then
		echo "Removing..."
		echo "TODO"
		# TODO
	else
		echo "Not Removing"
	fi
else
	echo "Not Removing"
fi

exit 0