#!/bin/bash
# Simple install script for 8188eu
# December, 25 2022 v0.0.4 borislavIvanov

MODNAME="8188eu"
DRVNAME="rtl8188eu"
DRVVER="5.13.3"
DRVSTATUS="installed"
KVER="$(uname -r)"
MODDESTDIR="/lib/modules/${KVER}/kernel/drivers/net/wireless/"
BLFILE="/etc/modprobe.d/realtek.conf"
BLDRV="r8188eu"

help () {
	echo ""
	echo "usage: ./install.sh [options]"
	echo ""
	echo "options:"
	echo ""
	echo " -i         : Install Realtek ${DRVNAME} Wireless Lan Driver"
	echo " -u         : Uninstall Realtek ${DRVNAME} Wireless Lan Driver"
	echo " -h, --help : Displays this usage screen"
	echo ""
	exit
}

inst_drv () {
if ! command -v dkms >/dev/null 2>&1;then
	if [ ! -f "${MODDESTDIR}${MODNAME}.ko" ];then
		make clean; Error=$?
		make; Error=$?
		if [ "$Error" != "0" ];then
			echo "Install error: $Error"
			exit
		else
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo make install; Error=$?
			else
				make install; Error=$?
			fi
		fi
	else
		echo "Module ${DRVNAME} already installed"
	fi
else
	if [[ ! "$(echo "`dkms status`" | awk '/'${DRVNAME}'/ {print}')" =~ "${DRVSTATUS}" ]];then
		make clean; Error=$?
		if [ -e /usr/src/${DRVNAME}-${DRVVER} ];then
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo rm -r /usr/src/${DRVNAME}-${DRVVER}
				sudo cp -r "$(pwd)" /usr/src/${DRVNAME}-${DRVVER}
			else
				rm -r /usr/src/${DRVNAME}-${DRVVER}
				cp -r "$(pwd)" /usr/src/${DRVNAME}-${DRVVER}
			fi
		else
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo cp -r "$(pwd)" /usr/src/${DRVNAME}-${DRVVER}
			else
				cp -r "$(pwd)" /usr/src/${DRVNAME}-${DRVVER}
			fi
		fi
		if [ -f "${MODDESTDIR}${MODNAME}.ko" ];then
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo rm -f ${MODDESTDIR}${MODNAME}.ko
			else
				rm -f ${MODDESTDIR}${MODNAME}.ko
			fi
		fi
		if [[ ! -z "$(echo "`dkms status`" | awk '/'${DRVNAME}'/ {print}')" ]];then
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo dkms remove -m ${DRVNAME} -v ${DRVVER} --all; Error=$?
				sudo dkms add -m ${DRVNAME} -v ${DRVVER}; Error=$?
				sudo dkms build -m ${DRVNAME} -v ${DRVVER}; Error=$?
			else
				dkms remove -m ${DRVNAME} -v ${DRVVER} --all; Error=$?
				dkms add -m ${DRVNAME} -v ${DRVVER}; Error=$?
				dkms build -m ${DRVNAME} -v ${DRVVER}; Error=$?
			fi
		else
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo dkms add -m ${DRVNAME} -v ${DRVVER}; Error=$?
				sudo dkms build -m ${DRVNAME} -v ${DRVVER}; Error=$?
			else
				dkms add -m ${DRVNAME} -v ${DRVVER}; Error=$?
				dkms build -m ${DRVNAME} -v ${DRVVER}; Error=$?
			fi
		fi
		if [ "$Error" != "0" ];then
			echo "Install error: $Error"
			exit
		else
			if [ "$EUID" != "0" ];then
				echo "You need root permissions:"
				sudo dkms install -m ${DRVNAME} -v ${DRVVER}; Error=$?
			else
				dkms install -m ${DRVNAME} -v ${DRVVER}; Error=$?
			fi
		fi
	else
		echo "Module ${DRVNAME} already installed"
	fi
fi
}

uinst_drv () {
if ! command -v dkms >/dev/null 2>&1;then
	if [ -f "${MODDESTDIR}${MODNAME}.ko" ];then
		if [ "$EUID" != "0" ];then
			echo "You need root permissions:"
			sudo make uninstall; Error=$?
		else
			make uninstall; Error=$?
		fi
	else
		echo "Module ${DRVNAME} already uninstalled"
	fi
else
	if [[ ! -z "$(echo "`dkms status`" | awk '/'${DRVNAME}'/ {print}')" ]];then
		if [ "$EUID" != "0" ];then
			echo "You need root permissions:"
			sudo dkms remove -m ${DRVNAME} -v ${DRVVER} --all; Error=$?
		else
			dkms remove -m ${DRVNAME} -v ${DRVVER} --all; Error=$?
		fi
	else
		echo "Module ${DRVNAME} already uninstalled"
	fi
fi
}

bl_drv () {
if [ "$1" = "-i" ]; then
	if [ ! -z "${BLDRV}" ] && [ "$Error" = "0" ]; then
		if [ ! -e ${BLFILE} ]; then
			sudo sh -c "echo 'blacklist ${BLDRV}' > ${BLFILE}"
		else
			if [[ ! "$(cat ${BLFILE} | awk '/'${BLDRV}'/ {print}')" =~ "${BLDRV}" ]]; then
				sudo sh -c "echo 'blacklist ${BLDRV}' >> ${BLFILE}"
			else
				echo "Module ${BLDRV} already blacklisted"
			fi
		fi
	fi
elif [ "$1" = "-u" ]; then
	if [ ! -z "${BLDRV}" ]  && [ "$Error" = "0" ]; then
		if [ -e ${BLFILE} ]; then
			if [[ "$(cat ${BLFILE} | awk '/'${BLDRV}'/ {print}')" =~ "${BLDRV}" ]]; then
				sudo sed -i '/'${BLDRV}'/d' ${BLFILE}
				if [ ! -s "${BLFILE}" ];then
					sudo rm -r ${BLFILE}
				fi
			else
				echo "Module ${BLDRV} already unblacklisted"
			fi
		else
			echo "Module ${BLDRV} already unblacklisted"
		fi
	fi
fi
}

if [ -z "$1" ];then
	echo "[options] is not defined"
	help
elif [ "$1" = "-h" ] || [ "$1" = "--help" ];then
	help
elif [ "$1" = "-i" ];then
	echo "Install Realtek ${DRVNAME} Wireless Lan Driver..."
	inst_drv
	bl_drv "$1"
elif [ "$1" = "-u" ];then
	echo "Uninstall Realtek ${DRVNAME} Wireless Lan Driver..."
	uinst_drv
	bl_drv "$1"
else
	echo "Incorrect [options] "$1""
	help
fi