#
# SPDX-License-Identifier: GPL-2.0
# Armbian build framework extension
#
# Enables 3D and multimedia, 4K VPU with Chromium, acceleration for Ubuntu. Debian only 3D
#

function extension_prepare_config__3d() {
	# Silently deny old releases which are not supported but are still in the system
	[[ "${RELEASE}" =~ ^(bullseye|buster|focal)$ ]] && return 0

	# Deny on minimal CLI images
	if [[ "${BUILD_MINIMAL}" == "yes" ]]; then
		display_alert "Extension: ${EXTENSION}" "skip installation in minimal images" "warn"
		return 0
	fi

	# some desktops doesn't support wayland
	[[ "${DESKTOP_ENVIRONMENT}" == "xfce" || "${DESKTOP_ENVIRONMENT}" == "i3-wm" ]] && return 0

	# Define image suffix
	if [[ "${LINUXFAMILY}" =~ ^(rockchip-rk3588|rk35xx)$ && "$BRANCH" =~ ^(legacy)$ && "${RELEASE}" =~ ^(jammy|noble)$ ]]; then
		EXTRA_IMAGE_SUFFIXES+=("-panfork")
	elif [[ "${DISTRIBUTION}" == "Ubuntu" && "${RELEASE}" =~ ^(jammy)$ ]]; then
		EXTRA_IMAGE_SUFFIXES+=("-kisak")
	elif [[ "${DISTRIBUTION}" == "Debian" && "${RELEASE}" == "bookworm" ]]; then
		EXTRA_IMAGE_SUFFIXES+=("-backported-mesa")
	fi

	# This should be enabled on all for rk3588 distributions where mesa and vendor kernel is present
	if [[ "${LINUXFAMILY}" =~ ^(rockchip-rk3588|rk35xx)$ && "$BRANCH" == vendor ]]; then
		if [[ -n $DEFAULT_OVERLAYS ]]; then
			DEFAULT_OVERLAYS+=" panthor-gpu"
		else
			declare -g DEFAULT_OVERLAYS="panthor-gpu"
		fi
	fi

}

function post_install_kernel_debs__3d() {
	# Silently deny old releases which are not supported but are still in the system
	[[ "${RELEASE}" =~ ^(bullseye|buster|focal)$ ]] && return 0

	# Deny on minimal CLI images
	if [[ "${BUILD_MINIMAL}" == "yes" ]]; then
		display_alert "Extension: ${EXTENSION}" "skip installation in minimal images" "warn"
		return 0
	fi

	# some desktops doesn't support wayland
	[[ "${DESKTOP_ENVIRONMENT}" == "xfce" || "${DESKTOP_ENVIRONMENT}" == "i3-wm" ]] && return 0

	# Packages that are going to be installed, always, both for cli and desktop
	declare -a pkgs=("libgl1-mesa-dri")

	if [[ "${BUILD_DESKTOP}" == "yes" ]]; then
		pkgs+=("libglx-mesa0") # x11 stuff all the way
		pkgs+=("mesa-utils" "mesa-utils-extra")
		pkgs+=("glmark2" "glmark2-wayland" "glmark2-es2-wayland" "glmark2-es2")
		[[ "${RELEASE}" != jammy ]] && pkgs+=("glmark2-x11" "glmark2-es2-x11") # Some packages, x11gl benchmark, came late into Ubuntu
	fi

	# Rockchip RK3588 will use panfork only with legacy kernel
	if [[ "${LINUXFAMILY}" =~ ^(rockchip-rk3588|rk35xx)$ && "$BRANCH" =~ ^(legacy)$ && "${RELEASE}" =~ ^(jammy|noble)$ ]]; then

		display_alert "Adding amazingfated's rk3588 PPAs" "${EXTENSION}" "info"
		do_with_retries 3 chroot_sdcard add-apt-repository ppa:liujianfeng1994/panfork-mesa --yes --no-update

		display_alert "Pinning amazingfated's rk3588 PPAs" "${EXTENSION}" "info"
		cat <<- EOF > "${SDCARD}"/etc/apt/preferences.d/amazingfated-rk3588-panfork-pin
			Package: *
			Pin: release o=LP-PPA-liujianfeng1994-panfork-mesa
			Pin-Priority: 1001
		EOF

		sed -i "s/noble/jammy/g" "${SDCARD}"/etc/apt/sources.list.d/liujianfeng1994-ubuntu-panfork-mesa-"${RELEASE}".*

	elif [[ "${DISTRIBUTION}" == "Ubuntu" && "${RELEASE}" =~ ^(jammy)$ ]]; then

		display_alert "Adding kisak PPAs" "${EXTENSION}" "info"
		do_with_retries 3 chroot_sdcard add-apt-repository ppa:kisak/kisak-mesa --yes --no-update

		display_alert "Pinning kisak PPAs" "${EXTENSION}" "info"
		cat <<- EOF > "${SDCARD}"/etc/apt/preferences.d/mesa-kisak-kisak-mesa-pin
			Package: *
			Pin: release o=LP-PPA-kisak-kisak-mesa
			Pin-Priority: 1001
		EOF
	fi

	# Add chromium if building a desktop
	if [[ "${BUILD_DESKTOP}" == "yes" ]]; then
		if [[ "${DISTRIBUTION}" == "Debian" ]]; then
			pkgs+=("chromium")
		elif [[ "${DISTRIBUTION}" == "Ubuntu" && "${RELEASE}" =~ ^(jammy|noble)$ ]]; then
			pkgs+=("chromium")
		fi
	fi

	if [[ "${BUILD_DESKTOP}" == "yes" ]]; then # if desktop, add amazingfated's multimedia PPAs and rockchip-multimedia-config utility, chromium, gstreamer, etc
		if [[ "${LINUXFAMILY}" =~ ^(rockchip-rk3588|rk35xx)$ && "${RELEASE}" =~ ^(jammy|noble)$ && "${BRANCH}" =~ ^(legacy|vendor)$ ]]; then

			pkgs+=("rockchip-multimedia-config" "chromium" "libv4l-rkmpp" "gstreamer1.0-rockchip")
			if [[ "${RELEASE}" == "jammy" ]]; then
				pkgs+=(libwidevinecdm)
			else
				pkgs+=(libwidevinecdm0)
			fi

			display_alert "Adding amazingfated's multimedia PPAs" "${EXTENSION}" "info"
			do_with_retries 3 chroot_sdcard add-apt-repository ppa:liujianfeng1994/rockchip-multimedia --yes --no-update

			display_alert "Pinning amazingfated's multimedia PPAs" "${EXTENSION}" "info"
			cat <<- EOF > "${SDCARD}"/etc/apt/preferences.d/amazingfated-rk3588-rockchip-multimedia-pin
				Package: *
				Pin: release o=LP-PPA-liujianfeng1994-rockchip-multimedia
				Pin-Priority: 1001
			EOF
		fi
	fi
	display_alert "Updating sources list, after adding all PPAs" "${EXTENSION}" "info"
	do_with_retries 3 chroot_sdcard_apt_get_update

	# KDE neon downgrades base-files for some reason. This prevents tacking it
	do_with_retries 3 chroot_sdcard apt-mark hold base-files

	if [[ "${BUILD_DESKTOP}" == "yes" ]]; then # This library must be installed before rockchip-multimedia; only for desktops
		do_with_retries 3 chroot_sdcard_apt_get_install libv4l-0
	fi

	display_alert "Installing 3D extension packages" "${EXTENSION}" "info"
	if [[ "${DISTRIBUTION}" == "Debian" && "${RELEASE}" == "bookworm" ]]; then
		do_with_retries 3 chroot_sdcard_apt_get_install -t bookworm-backports "${pkgs[@]}"
	else
		do_with_retries 3 chroot_sdcard_apt_get_install "${pkgs[@]}"
	fi

	# This library gets downgraded
	if [[ "${BUILD_DESKTOP}" == "yes" ]]; then
		if [[ "${RELEASE}" =~ ^(oracular|noble|jammy)$ && "${ARCH}" == arm* ]]; then
			do_with_retries 3 chroot_sdcard apt-mark hold libdav1d7
		fi
	fi

	display_alert "Upgrading all packages, including hopefully all mesa packages" "${EXTENSION}" "info"
	do_with_retries 3 chroot_sdcard_apt_get -o Dpkg::Options::="--force-confold" --allow-downgrades dist-upgrade

	# KDE neon downgrade hack undo
	do_with_retries 3 chroot_sdcard apt-mark unhold base-files

	if [[ "${BUILD_DESKTOP}" == "yes" ]]; then
		if [[ "${RELEASE}" =~ ^(oracular|noble|jammy)$ && "${ARCH}" == arm* ]]; then
			do_with_retries 3 chroot_sdcard apt-mark unhold libdav1d7
		fi
	fi

	# Disable wayland flag for XFCE
	#if [[ "${DESKTOP_ENVIRONMENT}" == "xfce" ]]; then
	#	sed -e '/wayland/ s/^#*/#/' -i "${SDCARD}"/etc/chromium.d/default-flags
	#fi

	return 0
}