From 364f149725a915c2e2f209f0ee6d7bb48322d89b Mon Sep 17 00:00:00 2001
From: Antoine Viallon <antoine@lesviallon.fr>
Date: Tue, 26 Dec 2023 17:54:28 +0100
Subject: [PATCH] [Boot] successful boot marking: handle specialization

---
 boot.nix | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/boot.nix b/boot.nix
index 1d8bb27..92fbe1e 100644
--- a/boot.nix
+++ b/boot.nix
@@ -320,10 +320,23 @@ in {
         in ''
           rpath=
           generation=
+          specialization=
           boot_generation_path=$(realpath /run/booted-system)  
           for path in /nix/var/nix/profiles/system-*-link; do
             rpath=$(realpath "$path")
+            ok=false
             if [ "$rpath" = "$boot_generation_path" ]; then
+              echo "Good path: $path"
+              ok=true
+            fi
+            for specialization in "$path"/specialisation/*; do
+              if [ "$(realpath $specialization)" = "$boot_generation_path" ]; then
+                ok=true
+                echo "Good specialization: $specialization"
+                break
+              fi
+            done
+            if $ok; then
               generation="''${path##*/system-}"
               generation="''${generation%%-link}"
               break
@@ -335,13 +348,19 @@ in {
           fi
 
           loader_entry="${efiDir}/loader/entries/nixos-generation-''${generation}.conf"
+          if ! [ -z "$specialization" ]; then
+            specialization_name=$(basename -- "$specialization")
+            echo "Specialization is: $specialization_name"
+            loader_entry="${efiDir}/loader/entries/nixos-generation-''${generation}-specialisation-''${specialization_name}.conf"
+          fi
+          
           if ! [ -f "$loader_entry" ]; then
             echo "Failed to find corresponding loader generation entry:" ''${loader_entry} "not found"
             echo -e "\e[33mWARNING:\e[0m This may mean that your aviallon.boot.configurationLimit is set too low!"
             exit 1
           fi
 
-          sed -i 's/version /version <LAST> /' "${efiDir}/loader/entries/nixos-generation-''${generation}.conf" &&
+          sed -i 's/version /version <LAST> /' "$loader_entry" &&
           echo "Marked generation $generation as last sucessfully booted"
         '';
       };