fix: Sort partitions when locating disk

Vanilla-OS · Feb 3, 2024 · 3fd01a8 · 3fd01a8
1 parent 9e421b3
commit 3fd01a8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/core/disk.go b/core/disk.go
@@ -1,8 +1,10 @@
 package albius
 
 import (
+	"cmp"
 	"encoding/json"
 	"fmt"
+	"slices"
 	"strconv"
 	"strings"
 
@@ -89,6 +91,11 @@ func LocateDisk(diskname string) (*Disk, error) {
 		decoded.Disk.Partitions[i].FillPath(decoded.Disk.Path)
 	}
 
+    // Partitions may be unordered
+    slices.SortFunc(decoded.Disk.Partitions, func(a, b Partition) int {
+        return cmp.Compare(a.Number, b.Number)
+    })
+
 	return &decoded.Disk, nil
 }