From ad8f4ddfbf58bf89c48956acaebdd5daebda518b Mon Sep 17 00:00:00 2001
From: Pat Hartl <pat@pathar.tl>
Date: Fri, 12 Jun 2026 23:48:38 -0500
Subject: [PATCH] Fix error thrown on first time setup when no storage
 locations exist

---
 .../Components/StorageLocationSelector.razor  | 15 +++++++-----
 .../UI/Pages/FirstTimeSetup/Paths.razor       | 24 +++++++------------
 2 files changed, 17 insertions(+), 22 deletions(-)

diff --git a/LANCommander.Server/UI/Components/StorageLocationSelector.razor b/LANCommander.Server/UI/Components/StorageLocationSelector.razor
index a63ffdc2..337adbb6 100644
--- a/LANCommander.Server/UI/Components/StorageLocationSelector.razor
+++ b/LANCommander.Server/UI/Components/StorageLocationSelector.razor
@@ -1,12 +1,15 @@
 @using LANCommander.SDK.Enums
 @inject StorageLocationService StorageLocationService
 
-<Select
-    DataSource="StorageLocations"
-    ItemValue="l => l"
-    ItemLabel="l => l.Path"
-    @bind-Value="Value"
-    OnSelectedItemChanged="ValueChanged" />
+@if (StorageLocations.Any())
+{
+    <Select
+        DataSource="StorageLocations"
+        ItemValue="l => l"
+        ItemLabel="l => l.Path"
+        @bind-Value="Value"
+        OnSelectedItemChanged="ValueChanged" />
+}
 
 @code {
     [Parameter] public StorageLocation Value { get; set; } = new();
diff --git a/LANCommander.Server/UI/Pages/FirstTimeSetup/Paths.razor b/LANCommander.Server/UI/Pages/FirstTimeSetup/Paths.razor
index 0af28929..7f3bcf68 100644
--- a/LANCommander.Server/UI/Pages/FirstTimeSetup/Paths.razor
+++ b/LANCommander.Server/UI/Pages/FirstTimeSetup/Paths.razor
@@ -146,38 +146,30 @@
 
         try
         {
-            SavesStorageLocation = await StorageLocationService
+            var existingSaves = await StorageLocationService
                 .AsNoTracking()
                 .FirstOrDefaultAsync(l => l.Type == StorageLocationType.Save && l.Default);
+
+            if (existingSaves != null)
+                SavesStorageLocation = existingSaves;
         }
         catch (Exception ex)
         {
             Logger?.LogError(ex, "Couldn't retrieve existing store locations");
-
-            SavesStorageLocation = new StorageLocation
-            {
-                Path = Path.Join(AppPaths.GetConfigDirectory(), "Saves"),
-                Type = StorageLocationType.Save,
-                Default = true,
-            };
         }
 
         try
         {
-            MediaStorageLocation = await StorageLocationService
+            var existingMedia = await StorageLocationService
                 .AsNoTracking()
                 .FirstOrDefaultAsync(l => l.Type == SDK.Enums.StorageLocationType.Media && l.Default);
+
+            if (existingMedia != null)
+                MediaStorageLocation = existingMedia;
         }
         catch (Exception ex)
         {
             Logger?.LogError(ex, "Couldn't retrieve existing store locations");
-
-            MediaStorageLocation = new StorageLocation
-            {
-                Path = Path.Join(AppPaths.GetConfigDirectory(), "Media"),
-                Type = StorageLocationType.Media,
-                Default = true,
-            };
         }
     }