Add IsSelected

Nexus-Mods · Sep 12, 2024 · 8fecafc · 8fecafc
1 parent 1bc55cd
commit 8fecafc
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/NexusMods.App.UI/Controls/TreeDataGrid/TreeDataGridAdapter.cs b/src/NexusMods.App.UI/Controls/TreeDataGrid/TreeDataGridAdapter.cs
@@ -72,7 +72,18 @@ protected TreeDataGridAdapter()
                     self._selectionModelsSerialDisposable.Disposable = selectionObservable.Subscribe(self, static (eventArgs, self) =>
                     {
                         self.SelectedModels.RemoveRange(eventArgs.DeselectedItems.NotNull());
+                        foreach (var item in eventArgs.DeselectedItems)
+                        {
+                            if (item is null) continue;
+                            item.IsSelected.Value = false;
+                        }
+
                         self.SelectedModels.AddRange(eventArgs.SelectedItems.NotNull());
+                        foreach (var item in eventArgs.SelectedItems)
+                        {
+                            if (item is null) continue;
+                            item.IsSelected.Value = true;
+                        }
                     });
 
                     self.Source.Value = source;

diff --git a/src/NexusMods.App.UI/Controls/TreeDataGrid/TreeDataGridItemModel.cs b/src/NexusMods.App.UI/Controls/TreeDataGrid/TreeDataGridItemModel.cs
@@ -22,6 +22,8 @@ public class TreeDataGridItemModel<TModel, TKey> : TreeDataGridItemModel
     where TModel : TreeDataGridItemModel<TModel, TKey>
     where TKey : notnull
 {
+    public ReactiveProperty<bool> IsSelected { get; } = new(value: false);
+
     public IObservable<bool> HasChildrenObservable { get; init; } = Observable.Return(false);
     public BindableReactiveProperty<bool> HasChildren { get; } = new();
 
@@ -130,7 +132,8 @@ protected override void Dispose(bool disposing)
                     _modelActivationDisposable,
                     _childrenObservableSerialDisposable,
                     _childrenCollectionInitializationSerialDisposable,
-                    HasChildren
+                    HasChildren,
+                    IsSelected
                 );
             }