#!/bin/bash

GREEN='\033[0;32m'
RED='\033[0;31m'
NC='\033[0m'


# Checks if loclat is installed
if ! command -v lolcat >/dev/null 2>&1; then
	echo "Dependencies not found. Installing...."
	sudo apt update
	sudo apt install lolcat -y
fi

echo "Scanning For Updates...."

if sudo apt update | grep -q "All Packages are up to date."; then
	echo "No updates available"
else
	echo "Updates are available (Warning): Packages may be kept back to be upgraded later."
fi

# Prompt user to initiate updates
read -p "Do you want to proceed? (y/n) " yn

case "$yn" in
	[yY])
		echo "Starting Update"

			sudo apt update | lolcat
			sudo apt upgrade -y | lolcat
			sudo snap refresh | lolcat
			sudo apt autoremove --purge # Note I do not choose to include a -y here so I don't automatically purge old packages the user may want to keep.


			echo "DONE!✅"
			sleep 2

			echo "REBOOTING!"
			sleep 3
			
			sudo reboot
			;;
	[nN])
		echo "Update cancelled."
		exit 0
		;;
	*)
		echo "Invalid response. Please run the script again."
		exit 1
		;;
esac