Make SelectColumn respond to programmatic changes in grid data

src/Core/Components/DataGrid/Columns/SelectColumn.cs

@@ -5,14 +5,15 @@
 using Microsoft.AspNetCore.Components;
 using Microsoft.AspNetCore.Components.Rendering;
 using Microsoft.AspNetCore.Components.Web;
+using Microsoft.FluentUI.AspNetCore.Components.Infrastructure;
 
 namespace Microsoft.FluentUI.AspNetCore.Components;
 
 /// <summary>
 /// Represents a <see cref="FluentDataGrid{TGridItem}"/> column whose cells render a selected checkbox updated when the user click on a row.
 /// </summary>
 /// <typeparam name="TGridItem">The type of data represented by each row in the grid.</typeparam>
-public class SelectColumn<TGridItem> : ColumnBase<TGridItem>
+public class SelectColumn<TGridItem> : ColumnBase<TGridItem>, IDisposable
 {
     /// <summary>
     /// List of keys to press, to select/unselect a row.
@@ -27,13 +28,22 @@ public class SelectColumn<TGridItem> : ColumnBase<TGridItem>
     private DataGridSelectMode _selectMode = DataGridSelectMode.Single;
     private readonly List<TGridItem> _selectedItems = [];
 
+    private readonly EventCallbackSubscriber<object?> _itemsChanged;
+
     /// <summary>
     /// Initializes a new instance of <see cref="SelectColumn{TGridItem}"/>.
     /// </summary>
     public SelectColumn()
     {
         Width = "50px";
         ChildContent = GetDefaultChildContent();
+
+        _itemsChanged = new(EventCallback.Factory.Create<object?>(this, UpdateSelectedItemsAsync));
+    }
+
+    protected override void OnInitialized()
+    {
+        _itemsChanged.SubscribeOrMove(InternalGridContext.ItemsChanged);
     }
 
     /// <summary>
@@ -326,6 +336,21 @@ Task CallOnSelectAsync(TGridItem item, bool isSelected)
         }
     }
 
+    private async Task UpdateSelectedItemsAsync()
+    {
+
+        if (!SelectedItems.Any() || InternalGridContext == null || InternalGridContext.Items == null)
+        {
+            return;
+        }
+
+        var itemsToRemove = _selectedItems.Where(item => !InternalGridContext.Items.Contains(item)).ToList();
+        foreach (var item in itemsToRemove)
+        {
+            await AddOrRemoveSelectedItemAsync(item);
+        }
+    }
+
     private Icon GetIcon(bool? selected)
     {
         if (selected == true)
@@ -555,6 +580,13 @@ internal async Task OnKeyAllAsync(KeyboardEventArgs e)
             await OnClickAllAsync(new MouseEventArgs());
         }
     }
+
+    /// <inheritdoc />
+    public void Dispose()
+    {
+        _itemsChanged.Dispose();
+
+    }
 }
 
 public record SelectAllTemplateArgs(bool? AllSelected) { }

src/Core/Components/DataGrid/FluentDataGrid.razor.cs

@@ -789,6 +789,8 @@ private async Task RefreshDataCoreAsync()
         if (!thisLoadCts.IsCancellationRequested)
         {
             _internalGridContext.Items = result.Items;
+            await _internalGridContext.ItemsChanged.InvokeCallbacksAsync(null);
+
             _internalGridContext.TotalItemCount = result.TotalItemCount;
             if (RefreshItems is null)
             {
@@ -831,7 +833,7 @@ private async Task RefreshDataCoreAsync()
 
         if (!request.CancellationToken.IsCancellationRequested)
         {
-            // ARIA's rowcount is part of the UI, so it should reflect what the human user regards as the number of rows in the table,
+            // ARIA's row count is part of the UI, so it should reflect what the human user regards as the number of rows in the table,
             // not the number of physical <tr> elements. For virtualization this means what's in the entire scrollable range, not just
             // the current viewport. In the case where you're also paginating then it means what's conceptually on the current page.
             // TODO: This currently assumes we always want to expand the last page to have ItemsPerPage rows, but the experience might

src/Core/Components/DataGrid/Infrastructure/InternalGridContext.cs

@@ -1,3 +1,7 @@
+// ------------------------------------------------------------------------
+// This file is licensed to you under the MIT License.
+// ------------------------------------------------------------------------
+
 using Microsoft.FluentUI.AspNetCore.Components.Infrastructure;
 
 namespace Microsoft.FluentUI.AspNetCore.Components.DataGrid.Infrastructure;
@@ -21,6 +25,7 @@ internal sealed class InternalGridContext<TGridItem>
 
     public FluentDataGrid<TGridItem> Grid { get; }
     public EventCallbackSubscribable<object?> ColumnsFirstCollected { get; } = new();
+    public EventCallbackSubscribable<object?> ItemsChanged { get; } = new();
 
     public InternalGridContext(FluentDataGrid<TGridItem> grid)
     {