gpumanager

#!/bin/bash

function GetAlternative() {
	if [ "$(ls /var/lib/pacman/local/mesa-libgl* 2> /dev/null)" ]; then
		echo "intel"
	elif [ "$(ls /var/lib/pacman/local/nvidia-libgl* 2> /dev/null)" ]; then
		echo "nvidia"
	fi
}

#Install the desired LibGL
function SetAlternative() {
	desired=$1
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Getting current Alternative"
	currentalternative=$(GetAlternative)
	if [ "$desired" = "$currentalternative" ]; then
		echo "$(date +%Y-%m-%d:%H:%M:%S) - LibGL is already $currentalternative nothing to do"
	else
		if [ "$desired" = "nvidia" ]; then
			echo "$(date +%Y-%m-%d:%H:%M:%S) - Setting up desired Alternative"
			#pacman --noconfirm -Rdd mesa lib32-mesa
			pacman --noconfirm -S nvidia-utils lib32-nvidia-utils xf86-video-intel
		elif [ "$desired" = "intel" ]; then
			echo "$(date +%Y-%m-%d:%H:%M:%S) - Setting up desired Alternative"
			pacman --noconfirm -Rdd nvidia-utils lib32-nvidia-utils
			pacman --noconfirm -S mesa lib32-mesa
			pacman --noconfirm -S xf86-video-intel
		fi
	fi
}

#Read the desired discrete card state
function ReadAlternative() {
	file="/etc/prime/state"
	cat $file
}

#Write discrete card desired state, to be used in startup
function WriteDesiredConfig() {
	discretestate=$1
	file="/etc/prime/state"
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Setting up desired state"
	if [ "$discretestate" = "nvidia" ]; then
		echo "ON" > "$file"
	elif [ "$discretestate" = "intel" ]; then
		echo "OFF" > "$file"
	fi
}

#Clean the config files required by the discrete card
function CleanConfigs() {
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Cleaning configs"
	rm -f /etc/X11/xorg.conf
	rm -f /etc/X11/xinit/xinitrc.d/nvidia-optimus.sh
#	rm -f /usr/share/gdm/greeter/autostart/display_setup.desktop
	rm -f /usr/share/sddm/scripts/Xsetup
    cp /etc/prime/20-intel.conf /etc/X11/xorg.conf.d/
}

#Setup the config files required by the discrete card
function SetupConfigs() {
	xorg_conf="/etc/prime/xorg.conf"
	offload="/etc/prime/nvidia-optimus.sh"
#	offloadgdm="/etc/prime/display_setup.desktop"
	offloadsddm="/etc/prime/Xsetup"
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Setting up configs"
	cat $xorg_conf > /etc/X11/xorg.conf
#	cat $offloadgdm > /usr/share/gdm/greeter/autostart/display_setup.desktop
	cat $offloadsddm > /usr/share/sddm/scripts/Xsetup
	cat $offload > /etc/X11/xinit/xinitrc.d/nvidia-optimus.sh
	chmod +x /etc/X11/xinit/xinitrc.d/nvidia-optimus.sh
	chmod +x /usr/share/sddm/scripts/Xsetup
}


#Load the nvidia module
function LoadModule() {
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Loading NVIDIA module"
	modprobe nvidia
	modprobe nvidia_uvm
  modprobe nvidia_drm
  modprobe nvidia_modeset
}

#Unload the nvidia module
function UnloadModule() {
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Checking if NVIDIA modules are loaded"
	if lsmod | grep "nvidia" &> /dev/null ; then
		echo "$(date +%Y-%m-%d:%H:%M:%S) - Unloading NVIDIA modules"
		modprobe -r nvidia_uvm
		modprobe -r nvidia_drm
		modprobe -r nvidia_modeset
		modprobe -r nvidia
	fi
}

#Power the discrete card on or off
function DesiredCardState() {
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Checking if bbswitch module is loaded"
	if ! lsmod | grep "bbswitch" &> /dev/null ; then
		echo "$(date +%Y-%m-%d:%H:%M:%S) - Loading bbswitch module"
		modprobe bbswitch
	fi
	cardstate=$1
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Changing bbswitch state to $cardstate"
	echo "$cardstate" >> /proc/acpi/bbswitch
}

function Main() {
	#Always assume that the content of the "state" file is the desired, if the OpenGL settings do not match with that they will be updated. Then load the correct configs
	#turn on/off the discrete card and load/unload the nvidia driver.
	echo "$(date +%Y-%m-%d:%H:%M:%S) - Reading desired alternative state"
	echo "MAin"
	desiredstate=$(ReadAlternative)
	echo "$desiredstate"

	if [ "$desiredstate" = "OFF" ]; then
		SetAlternative "intel"
		CleanConfigs
		UnloadModule
		DesiredCardState "$desiredstate"
elif [ "$desiredstate" = "ON" ]; then
		SetAlternative "nvidia"
		SetupConfigs
		DesiredCardState "$desiredstate"
		LoadModule
	fi
}

#Log Stuff (Close stdout file descriptor, close stderr file descriptor, opens stdout as logfile and redirects stderr to stdout)
LogFile="/var/log/prime"
exec 1<&-
exec 2<&-
exec 1<>$LogFile
exec 2>&1

#Switch to intel on startup (called by systemd service file), if no parameter specified execute the main script function
if [ "$1" != "startup" ]; then
	echo "main no startup"
	Main
else
	echo "intel again!!"
	WriteDesiredConfig "intel"
	SetAlternative "intel"
	CleanConfigs
	UnloadModule
	DesiredCardState "OFF"
fi