#!/bin/bash

[ "$1" != "--run" ] && echo "Install and configure Waydroid" && exit

. $(dirname $0)/common.sh


[ "$(epm print info -s)" = "alt" ] || fatal "Only ALTLinux are supported"

epm assure lspci pciutils || exit

display_help()
{
    echo "
Use: epm prescription waydroid [option]
--gpu
	Select GPU for Waydroid

--init
    Initialize Waydroid

--integrate
	Enable desktop window integration for Waydroid

--software-render
	Use software render in Waydroid (maybe fix work in Nvidia)

--help
	Display this page
"
    exit
}

# TODO: fix used_kflavour export in epm update-kernel --check-run-kernel
used_kflavour () {
    if [ $(uname -r | grep "def") ] ; then
		USED_KFLAVOUR=$(uname -r | awk -F'-' '{print $2 "-" $3}')
    else
		USED_KFLAVOUR=$(uname -r | awk -F'-' '{print $2}')
    fi
}

waydroid_install () {
	epm update-kernel --add-kernel-options psi1

	epm update || fatal
	epm update-kernel || fatal

	if ! epm update-kernel --check-run-kernel ; then
		fatal
	fi

	used_kflavour
	epm install --skip-installed kernel-modules-anbox-$USED_KFLAVOUR libgbinder1 waydroid || fatal

	a= update-grub

	echo "Waydroid has been installed"
}

waydroid_select_gpu () {
	echo "Select GPU for Waydroid"
	gpu_info=$(lspci -nn | grep '\[03')
	gpu_number=$(echo "$gpu_info" | awk '{print $1}')
	gpu_name=$(echo "$gpu_info" | grep -oP '(?<=: ).*')

	if [ -z "$gpu_info" ]; then
	echo "No GPU devices found."
	exit 1
	fi

	echo "Available GPU device"
	for i in "${!gpu_number[@]}"; do
	echo "$((i)). ${gpu_number[$i]} ${gpu_name}"
	done

	# Prompt the user to select a device ID
	while true; do
	read -p "Enter the number of the GPU device to use: " selected_gpu_num
	if ! [[ "$selected_gpu_num" =~ ^[0-9]+$ ]]; then
		echo "Invalid entry: $selected_gpu_num is not a number."
		continue
	fi

	if (( selected_gpu_num < 0 || selected_gpu_num >= ${#gpu_number[@]} )); then
		echo "Invalid entry: $selected_gpu_num is not a valid option."
		continue
	fi
	selected_gpu=${gpu_number[$selected_gpu_num]}
	break
	done

	card=$(ls -l /dev/dri/by-path/ | grep -i $selected_gpu | grep -o "card[0-9]")
	rendernode=$(ls -l /dev/dri/by-path/ | grep -i $selected_gpu | grep -o "renderD[1-9][1-9][1-9]")

	cp /var/lib/waydroid/lxc/waydroid/config_nodes /var/lib/waydroid/lxc/waydroid/config_nodes.bak
	sed -i '/dri/d' /var/lib/waydroid/lxc/waydroid/config_nodes
	echo "lxc.mount.entry = /dev/dri/$card dev/dri/card0 none bind,create=file,optional 0 0" >> /var/lib/waydroid/lxc/waydroid/config_nodes
	echo "lxc.mount.entry = /dev/dri/$rendernode dev/dri/renderD128 none bind,create=file,optional 0 0" >> /var/lib/waydroid/lxc/waydroid/config_nodes
}

waydroid_init () {
	serv on waydroid-container
	waydroid init -c 'https://ota.waydro.id/system' -v 'https://ota.waydro.id/vendor'
}

waydroid_integrate () {
	waydroid prop set persist.waydroid.multi_windows true
}

waydroid_software_rendering () {
	sed -i "s/ro.hardware.gralloc=.*/ro.hardware.gralloc=default/g" /var/lib/waydroid/waydroid_base.prop
	sed -i "s/ro.hardware.egl=.*/ro.hardware.egl=swiftshader/g" /var/lib/waydroid/waydroid_base.prop
}

case "${3}" in
    '--gpu' )
        assure_root
        waydroid_select_gpu ;;

    '--init')
        assure_root
        waydroid_init ;;

    '--integrate')
		assure_root
        waydroid_integrate ;;

    '--software-render')
		assure_root
        waydroid_software_rendering ;;

    '--help')
        display_help;;

        *)
		assure_root
		waydroid_install ;;
esac