From f4914fd9e5e73f1ae64e67cc911208728fffd771 Mon Sep 17 00:00:00 2001
From: Vasiliy Stelmachenok <92667539+ventureoo@users.noreply.github.com>
Date: Thu, 4 May 2023 21:58:17 +0300
Subject: [PATCH] Add support for dracut and booster for the hook (#163)

---
 system/nvidia-tkg.hook | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/system/nvidia-tkg.hook b/system/nvidia-tkg.hook
index a852812..75f556e 100644
--- a/system/nvidia-tkg.hook
+++ b/system/nvidia-tkg.hook
@@ -14,4 +14,4 @@ Target=nvidia-open-dkms-tkg
 Description=Update Nvidia modules in initcpio
 When=PostTransaction
 NeedsTargets
-Exec=/bin/sh -c 'if command -v mkinitcpio >/dev/null 2>&1; then mkinitcpio -P; else printf "\\033[31m mkinitcpio not found, please update initramfs manually\\033[0m\\n"; fi'
+Exec=/bin/sh -c 'if command -v mkinitcpio >/dev/null 2>&1; then mkinitcpio -P; elif command -v /usr/lib/booster/regenerate_images >/dev/null 2>&1; then /usr/lib/booster/regenerate_images; elif command -v dracut >/dev/null 2>&1; then dracut --regenerate-all --force; else printf "\\033[31m The initramfs generator was not found, please update initramfs manually\\033[0m\\n"; fi'