From 841eaad972bec031e355c92df3b5359f7d064605 Mon Sep 17 00:00:00 2001
From: Mateus Melchiades <matbme@duck.com>
Date: Mon, 18 Mar 2024 11:44:00 -0300
Subject: [PATCH] fix: Add delay for WaitUntilAvailable

---
 core/disk/partition.go | 17 +++++++----------
 1 file changed, 7 insertions(+), 10 deletions(-)

diff --git a/core/disk/partition.go b/core/disk/partition.go
index 5659d973..ae8a7ad3 100644
--- a/core/disk/partition.go
+++ b/core/disk/partition.go
@@ -5,6 +5,7 @@ import (
 	"os"
 	"slices"
 	"strings"
+	"time"
 
 	luks "github.com/vanilla-os/albius/core/disk/luks"
 	"github.com/vanilla-os/albius/core/util"
@@ -319,9 +320,6 @@ func LUKSSetLabel(part *Partition, name string) error {
 // This is particularly useful to make sure a recently created or modified
 // partition is recognized by the system.
 func (part *Partition) WaitUntilAvailable() {
-	// maxTimeout := 1000
-	// timeout := 0
-
 	for {
 		_, err := os.Stat(part.Path)
 		if !os.IsNotExist(err) {
@@ -333,13 +331,12 @@ func (part *Partition) WaitUntilAvailable() {
 			if uuid, err := part.GetUUID(); err != nil && uuid != "" {
 				return
 			}
+
+			fmt.Println("Partition does not have UUID, retrying...")
+		} else {
+			fmt.Println("Partition not found, retrying...")
 		}
-		// time.Sleep(50 * time.Millisecond)
-		//
-		// timeout += 1
-		// if timeout == maxTimeout {
-		// 	// We can't recover from this, so just panic
-		// 	panic(fmt.Sprintf("Timed out waiting for partition %s", part.Path))
-		// }
+
+		time.Sleep(50 * time.Millisecond)
 	}
 }