95-refind-copy-icon.install: also copy icon in efistub layout

Signed-off-by: Andrew Ammerlaan <[email protected]>

Author: Nowa-Ammerlaan

hooks/95-refind-copy-icon.install

@@ -24,9 +24,7 @@ main() {
 
 	[[ ${EUID} -eq 0 ]] || die "Please run this script as root"
 
-	if [[ ${INSTALLKERNEL_LAYOUT} == compat || ${INSTALLKERNEL_LAYOUT} == uki ]]; then
-		cp "${REFIND_ICON}" "${img%.efi}.png" || die "Failed to copy icon file"
-	fi
+	cp "${REFIND_ICON}" "${img%.efi}.png" || die "Failed to copy icon file"
 }
 
 main

hooks/systemd/95-refind-copy-icon.install

@@ -29,4 +29,11 @@ elif [[ ${KERNEL_INSTALL_LAYOUT} == "uki" ]]; then
 	elif [[ ${COMMAND} == remove ]]; then
 		rm -f "${UKI_DIR}/${KERNEL_INSTALL_ENTRY_TOKEN}-${KERNEL_VERSION}.png" || exit 1
 	fi
+elif [[ ${KERNEL_INSTALL_LAYOUT} == "efistub" ]]; then
+	EFISTUB_DIR="${KERNEL_INSTALL_BOOT_ROOT}/EFI/Gentoo"
+	if [[ ${COMMAND} == add ]]; then
+		cp "${REFIND_ICON}" "${EFISTUB_DIR}/kernel-${KERNEL_VERSION}.png" || exit 1
+	elif [[ ${COMMAND} == remove ]]; then
+		rm -f "${EFISTUB_DIR}/kernel-${KERNEL_VERSION}.png" || exit 1
+	fi
 fi