From e08cfa84bb299386d290c9880dce25c944f05952 Mon Sep 17 00:00:00 2001
From: Alexander Shabarshov <alexander.shabarshov@db.com>
Date: Mon, 3 Nov 2025 14:50:49 +0000
Subject: [PATCH] Minor UI fixes

---
 Rms.Risk.Mango/Pages/Admin/Download.razor | 29 ++++++++++++++++-------
 Rms.Risk.Mango/Pages/User/Browse.razor    |  2 +-
 2 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/Rms.Risk.Mango/Pages/Admin/Download.razor b/Rms.Risk.Mango/Pages/Admin/Download.razor
index 4691d1f..2304ad5 100644
--- a/Rms.Risk.Mango/Pages/Admin/Download.razor
+++ b/Rms.Risk.Mango/Pages/Admin/Download.razor
@@ -382,6 +382,18 @@
         return InvokeAsync(StateHasChanged);
     }
 
+    private static IEnumerable<TreeNode<IndexTreeItem>> GetAllNodes(TreeNode<IndexTreeItem> node)
+    {
+        yield return node;
+        foreach (var child in node.Children)
+        {
+            foreach (var descendant in GetAllNodes(child))
+            {
+                yield return descendant;
+            }
+        }
+    }
+
     private async Task OnDownload()
     {
         var newJob = new MigrationJob
@@ -395,15 +407,14 @@
             Upsert                      = false,
             ClearDestinationBefore      = false,
             BatchSize                   = 1,
-            Status                      = _currentNodes[0].Children
-               .Where(x => (x.Data?.IsSelected  ?? false) && x.Data.Collection != null)
-               .Select(x => new MigrationJob.CollectionJob
-                {
-                    SourceCollection      = x.Data!.Collection!.Name,
-                    DestinationCollection = x.Data!.Collection!.Name,
-                })
-               .ToList()
-
+            Status = GetAllNodes(_currentNodes[0])
+                    .Where(x => (x.Data?.IsSelected ?? false) && x.Data.Collection != null)
+                    .Select(x => new MigrationJob.CollectionJob
+                     {
+                         SourceCollection      = x.Data!.Collection!.Name,
+                         DestinationCollection = x.Data!.Collection!.Name,
+                     })
+                    .ToList()
         };
 
         if (newJob.Status.Count == 0)
diff --git a/Rms.Risk.Mango/Pages/User/Browse.razor b/Rms.Risk.Mango/Pages/User/Browse.razor
index cfb5632..cb5d8f5 100644
--- a/Rms.Risk.Mango/Pages/User/Browse.razor
+++ b/Rms.Risk.Mango/Pages/User/Browse.razor
@@ -104,7 +104,7 @@
                                 {
                                     <div class="m-1">Sharded</div>
                                 }
-                                <div class="m-1 ">@NumbersUtils.ToHumanReadable(stat.TotalSize)</div>
+                                <div class="m-1 ">@NumbersUtils.ToHumanReadable(stat.Size)</div>
                             }
                         </div>
                     </OptionTemplate>