#!/bin/sh
yum -y install kmod-nvidia xorg-x11-drv-nvidia

perl -pi -w -e 's|GRUB_GFXPAYLOAD_LINUX=text||g;' /etc/default/grub
gfx=`grep -o "GRUB_GFXPAYLOAD_LINUX=auto" /etc/default/grub`
if [ "$gfx" != "GRUB_GFXPAYLOAD_LINUX=auto" ]
then
echo 'GRUB_GFXPAYLOAD_LINUX=auto' >> /etc/default/grub
fi

ch=`grep -o "rdblacklist=nouveau rd.driver.blacklist=nouveau nouveau.modeset=0 rhgb" /etc/default/grub`
if [ "$ch" != "rdblacklist=nouveau rd.driver.blacklist=nouveau nouveau.modeset=0 rhgb" ]
then
perl -pi -w -e 's/rhgb/rdblacklist=nouveau rd.driver.blacklist=nouveau nouveau.modeset=0 rhgb/g;' /etc/default/grub
fi

if [ -f "/boot/initramfs-$(uname -r)-nouveau.img" ]
then
rm -rf /boot/initramfs-$(uname -r).img
else
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img
fi

dracut -v --force /boot/initramfs-$(uname -r).img $(uname -r)

# check if destro use efi or not
OsName="`fgrep -n 'ID' /etc/os-release | grep 3 | cut -c6-20`"
if [ -d "/sys/firmware/efi" ]
then
	grub2-mkconfig -o /boot/efi/EFI/$OsName/grub.cfg
else
	grub2-mkconfig -o /boot/grub2/grub.cfg
fi

#echo '
##Video drivers
#blacklist nouveau
#' >> /etc/modprobe.d/blacklist.conf
exit 0
