layout/blspec.py: consider bls named after ID from os-release

OpenRC machines are not guaranteed to have the machine-id Closes: #55 Signed-off-by: Andrew Ammerlaan <[email protected]>
projg2 · Oct 5, 2024 · 4b5ef20 · 4b5ef20
1 parent 55aa5e8
commit 4b5ef20
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/ecleankernel/layout/blspec.py b/ecleankernel/layout/blspec.py
@@ -41,8 +41,9 @@ def __init__(self,
                  root: Path
                  ) -> None:
         super().__init__(root)
-        # TODO: according to bootctl(1), we should fall back to IMAGE_ID=
-        # and then ID= from os-release
+
+        self.distro_id = distro.id() or "linux"
+
         for path in ("etc/kernel/entry-token", "etc/machine-id"):
             try:
                 with open(root / path) as f:
@@ -51,7 +52,7 @@ def __init__(self,
             except FileNotFoundError:
                 pass
         else:
-            raise LayoutNotFound("/etc/machine-id not found")
+            self.kernel_id = self.distro_id
 
         for d in self.potential_dirs:
             # Present if type 1
@@ -137,10 +138,9 @@ def find_kernels(self,
                     # Not an UKI
                     continue
 
-                distro_id = distro.id() or "linux"
-
                 ver = basename.removeprefix(f"{self.kernel_id}-"
-                                            ).removeprefix(f"{distro_id}-")
+                                            ).removeprefix(f"{self.distro_id}-"
+                                                           )
                 if basename == ver:
                     # Not our UKI
                     continue