1 files changed, 23 insertions, 7 deletions

diff --git a/initrd.sh b/initrd.sh
index 5adb405..1eca9bb 100755
--- a/initrd.sh
+++ b/initrd.sh
@@ -6,14 +6,30 @@ vmlinuz=isolinux/live/vmlinuz
 version=$(uname -r)
 conf_dir=initramfs-tools
 
-if [ ! -e "$initrd" -o ! -e "$vmlinuz" ]; then
-    need_update=1
-elif [ "$(find "$conf_dir" -newer "$initrd" -print -quit)" ]; then
-    need_update=1
-fi
+apt_dependencies=initrd-dependencies.txt
+
+find_source_dirs() {
+    set -- find "$conf_dir" ./old-school "$@"
+    "$@"
+}
+
+force_rebuild()
+{
+    touch "$conf_dir"
+    return 1
+}
+
+rebuild()
+{
+    [ ! -f "$apt_dependencies" ] || sudo apt-get install -q=10 --no-upgrade -y $(cat "$apt_dependencies")
 
-if [ "$need_update" ]; then
     set -ex
     cp -f /boot/vmlinuz-${version} "$vmlinuz"
-    /usr/sbin/mkinitramfs -d "$conf_dir" -o "$initrd" ${version}
+    /usr/sbin/mkinitramfs -d "$conf_dir" -o "$initrd" ${version} || force_rebuild
+}
+
+if [ ! -e "$initrd" -o ! -e "$vmlinuz" ]; then
+    rebuild
+elif [ "$(find_source_dirs -newer "$initrd" -print -quit)" ]; then
+    rebuild
 fi