#!/bin/bash

if (( $EUID == 0 )); then
    echo "Do not run as root."
    exit 1
fi

if [ -z "$DISPLAY" ]; then
    export DISPLAY=:0.0
fi

# help function:
function printhelp {
	echo "usage examples:"
	echo "Download latest available cemu + cemuhook and install to ~/.cemu (default):"
	echo "./cemutil.sh -a"
	echo "Use local zips and install to ~/Documents/cemu:"
	echo "./cemutil.sh -c cemu.zip -h cemuhook.zip -g graphicpacks.zip -i ~/Documents/cemu"
	exit 1
}

function downloadlatest {
	echo "Downloading latest cemu"
	wget -q --show-progress -O cemutemp.zip $(curl -s http://cemu.info |grep .zip |awk -F '"' {'print $2'})
	#wget -q --show-progress -O cemutemp.zip http://cemu.info/releases/cemu_1.15.0.zip
	echo "Downloading latest cemuhook"
	wget -q --show-progress -O cemuhooktemp.zip $(curl -s https://cemuhook.sshnuke.net |grep .zip |awk -F '"' NR==2{'print $2'})
	echo "Downloading cemuhook shared fonts"
	wget -q --show-progress -O sharedFonts.zip https://github.com/CEMULinux/cemutil/raw/master/sharedFonts.zip
	return
}

#Check installed software
declare -a reqsw=("wine" "bsdtar" "unzip" "glxinfo" "curl" "wget" "winetricks")
for i in "${reqsw[@]}"
do
    if ! [ -x "$(command -v $i)" ]; then
        echo "You must install $i"
        exit 1
    fi
done

if $(glxinfo | grep -q -e "NVIDIA"); then
	skipgfxcheck=1
fi

function checkgfxver {
	echo "Checking graphics packages are new enough. To skip this check (on Nvidia for instance), run with -f flag."
	if ! $(glxinfo | grep -q -e 'Mesa 18.2' -e 'Mesa 18.3' -e 'Mesa 18.4' -e 'Mesa 19' -e 'Mesa 20'); then
		echo "You must install at least Mesa 18.2.0"
		exit 1
	fi

	if ! $(glxinfo | grep -q "4.5 (Compat"); then
		if ! $(glxinfo | grep -q "4.6 (Compat"); then
			echo "Your hardware doesn't support the required OpenGL version."
			echo "You may attempt using MESA_GL_VERSION_OVERRIDE=4.4COMPAT in the LaunchCEMU script."
			echo "This isn't officially supported, and may cause heavy glitches or not work. Proceeding as usual..."
		fi
	fi
	return
}

#Check for args
if [[ ! $@ =~ ^\-.+ ]]
then
    printhelp;
fi

#Handle args
while getopts ":c:h:afi:" opt; do
	case ${opt} in
		c )
			cemuzip=$OPTARG
			if [ ! -f "$cemuzip" ]; then
				echo "cemu zip doesn't exist"
				exit 1
			fi
			;;
		h )
			cemuhookzip=$OPTARG
			if [ ! -f "$cemuhookzip" ]; then
				echo "cemuhook zip doesn't exist"
				exit 1
			fi
			;;
		a )
			downloadlatest
			;;
		f )
			skipgfxcheck=1
			;;
		i )
			instdir=$OPTARG
			;;
		\? )
			printhelp
			;;
		: )
			echo "Invalid option: $OPTARG requires an argument" 1>&2
			printhelp
			;;
	esac
done
shift $((OPTIND -1))
skipgfxcheck=1
#check gfx package vers
if [[ "$skipgfxcheck" == "" ]]; then
    checkgfxver
fi

#Set opts if unset
if [[ "$instdir" == "" ]]; then
    instdir=$HOME/.cemu
fi
if [[ "$cemuzip" == "" ]]; then
    cemuzip=cemutemp.zip
fi
if [[ "$cemuhookzip" == "" ]]; then
    cemuhookzip=cemuhooktemp.zip
fi
if [[ "$sharedFonts" == "" ]]; then
    sharedFonts=sharedFonts.zip
fi

#Extract zips
echo "Extracting zips"
mkdir -p $instdir


#Unpack downloaded zips if applicable
if [ -f "$cemuzip" ]; then
    bsdtar -xf "$cemuzip" -s'|[^/]*/||' -C $instdir
fi
if [ -f "$cemuhookzip" ]; then
    unzip -q -o "$cemuhookzip" -d $instdir
fi
if [ -f "$sharedFonts" ]; then
    unzip -q -o "$sharedFonts" -d $instdir
fi

#Delete downloaded zips if applicable

if [ -f "cemutemp.zip" ]; then
    rm -rf cemutemp.zip
fi
if [ -f "cemuhooktemp.zip" ]; then
    rm -rf cemuhooktemp.zip
fi

if [ -f "sharedFonts.zip" ]; then
    rm -rf sharedFonts.zip
fi

#Configure wine prefix
echo "Configuring new wine prefix"
export WINEPREFIX=$(realpath $instdir)/wine
winetricks -q vcrun2017
winetricks settings win7

#Create cemuhook.ini, enabling cemuhook h264 as default
cat > $instdir/cemuhook.ini << EOF1
[Debug]
useH264Decoder = true
EOF1

#Create launch scripts
cat > LaunchCEMU << EOF1
#!/bin/bash
export WINEPREFIX="$(realpath $instdir)/wine"
#for cemuhook
export WINEDLLOVERRIDES="mscoree=;mshtml=;dbghelp.dll=n,b"

cd $(realpath $instdir)
mesa_glthread=true __GL_THREADED_OPTIMIZATIONS=1 vblank_mode=0 WINEESYNC=1 wine Cemu.exe "\$@"
EOF1
chmod +x LaunchCEMU

cat > LaunchCEMUgcn3BOTW << EOF1
#!/bin/bash
export WINEPREFIX="$(realpath $instdir)/wine"
#for cemuhook
export WINEDLLOVERRIDES="mscoree=;mshtml=;dbghelp.dll=n,b"

cd $(realpath $instdir)
R600_DEBUG=nohyperz mesa_glthread=true vblank_mode=0 WINEESYNC=1 wine Cemu.exe "\$@"
EOF1
chmod +x LaunchCEMUgcn3BOTW

echo "Successfully installed to $(realpath $instdir)"
echo "You may now run CEMU with LaunchCEMU written in this directory"
echo "You may place LaunchCEMU anywhere, and even pass arguments to it just like Cemu.exe on Windows"
echo "Note2: gcn3 (radeon 300-500 series) users should use the gcn3BOTW script for launching BOTW"
echo "Note3: CEMU versions >= 1.15.4 utilize a built-in solution for in-game video playback. Turn on Debug -> 'Use CemuHook H264' if you are encountering any green screens."