Added OnItemAdded and OnItemRemoved for collections

YohDeadfall · Jun 29, 2024 · da4575a · da4575a
1 parent 2e77b2d
commit da4575a
Show file tree

Hide file tree

Showing 4 changed files with 190 additions and 0 deletions.
diff --git a/src/Kinetic/Linq/ObservableView.OnItemAdded.cs b/src/Kinetic/Linq/ObservableView.OnItemAdded.cs
@@ -0,0 +1,17 @@
+using System;
+using Kinetic.Linq.StateMachines;
+
+namespace Kinetic.Linq;
+
+public static partial class ObservableView
+{
+    public static ObserverBuilder<ListChange<TSource>> OnItemAdded<TSource>(this ObserverBuilder<ListChange<TSource>> source, Action<TSource> action) =>
+        source.Do(change =>
+        {
+            if (change.Action is ListChangeAction.Insert or ListChangeAction.Replace)
+                action(change.NewItem);
+        });
+
+    public static ObserverBuilder<ListChange<TSource>> OnItemAdded<TSource>(this ReadOnlyObservableList<TSource> source, Action<TSource> action) =>
+        source.Changed.ToBuilder().OnItemAdded(action);
+}
diff --git a/src/Kinetic/Linq/ObservableView.OnItemRemoved.cs b/src/Kinetic/Linq/ObservableView.OnItemRemoved.cs
@@ -0,0 +1,110 @@
+using System;
+using System.Collections.Generic;
+using Kinetic.Linq.StateMachines;
+
+namespace Kinetic.Linq;
+
+public static partial class ObservableView
+{
+    public static ObserverBuilder<ListChange<TSource>> OnItemRemoved<TSource>(this ObserverBuilder<ListChange<TSource>> source, Action<TSource> action) =>
+        source.ContinueWith<OnItemRemovedStateMachineFactory<TSource>, ListChange<TSource>>(new() { Action = action });
+
+    public static ObserverBuilder<ListChange<TSource>> OnItemRemoved<TSource>(this ReadOnlyObservableList<TSource> source, Action<TSource> action) =>
+        source.Changed.ToBuilder().OnItemRemoved(action);
+
+    private struct OnItemRemovedStateMachineFactory<TSource> : IStateMachineFactory<ListChange<TSource>, ListChange<TSource>>
+    {
+        public required Action<TSource> Action { get; init; }
+
+        public void Create<TContinuation>(in TContinuation continuation, ObserverStateMachine<ListChange<TSource>> source)
+            where TContinuation : struct, IStateMachine<ListChange<TSource>>
+        {
+            source.ContinueWith<OnItemRemovedStateMachine<TSource, TContinuation>>(new(continuation, Action));
+        }
+    }
+
+    private struct OnItemRemovedStateMachine<TSource, TContinuation> : IStateMachine<ListChange<TSource>>
+        where TContinuation : struct, IStateMachine<ListChange<TSource>>
+    {
+        private TContinuation _continuation;
+        private readonly Action<TSource> _action;
+        private readonly List<TSource> _items = new();
+
+        public OnItemRemovedStateMachine(in TContinuation continuation, Action<TSource> action)
+        {
+            _continuation = continuation;
+            _action = action;
+        }
+
+        public StateMachineBox Box =>
+            _continuation.Box;
+
+        public void Initialize(StateMachineBox box) =>
+            _continuation.Initialize(box);
+
+        public void Dispose() =>
+            _continuation.Dispose();
+
+        public void OnCompleted() =>
+            _continuation.OnCompleted();
+
+        public void OnError(Exception error) =>
+            _continuation.OnError(error);
+
+        public void OnNext(ListChange<TSource> value)
+        {
+            _continuation.OnNext(value);
+
+            switch (value.Action)
+            {
+                case ListChangeAction.RemoveAll:
+                    {
+                        foreach (var item in _items)
+                            _action(item);
+
+                        _items.Clear();
+
+                        break;
+                    }
+                case ListChangeAction.Remove:
+                    {
+                        var index = value.OldIndex;
+                        var item = _items[value.OldIndex];
+
+                        _action(item);
+                        _items.RemoveAt(index);
+
+                        break;
+                    }
+                case ListChangeAction.Insert:
+                    {
+                        _items.Insert(
+                            value.NewIndex,
+                            value.NewItem);
+
+                        break;
+                    }
+                case ListChangeAction.Replace:
+                    {
+                        var index = value.OldIndex;
+                        var item = _items[index];
+
+                        _action(item);
+                        _items[index] = value.NewItem;
+
+                        break;
+                    }
+                case ListChangeAction.Move:
+                    {
+                        var index = value.OldIndex;
+                        var item = _items[index];
+
+                        _items.RemoveAt(index);
+                        _items.Insert(index, item);
+
+                        break;
+                    }
+            }
+        }
+    }
+}
diff --git a/src/Kinetic/Linq/ObservableView.OnItemRemovedDispose.cs b/src/Kinetic/Linq/ObservableView.OnItemRemovedDispose.cs
@@ -0,0 +1,15 @@
+using System;
+using Kinetic.Linq.StateMachines;
+
+namespace Kinetic.Linq;
+
+public static partial class ObservableView
+{
+    public static ObserverBuilder<ListChange<TSource>> OnItemRemovedDispose<TSource>(this ObserverBuilder<ListChange<TSource>> source)
+        where TSource : IDisposable =>
+        source.OnItemRemoved(item => item?.Dispose());
+
+    public static ObserverBuilder<ListChange<TSource>> OnItemRemovedDispose<TSource>(this ReadOnlyObservableList<TSource> source)
+        where TSource : IDisposable =>
+        source.Changed.ToBuilder().OnItemRemovedDispose();
+}
diff --git a/test/Kinetic.Tests/ObservableViewTests.cs b/test/Kinetic.Tests/ObservableViewTests.cs
@@ -336,6 +336,54 @@ public void GroupByDynamicWithComparer()
         Assert.Empty(groupingsOrdered);
     }
 
+    [Fact]
+    public void OnItemAdded()
+    {
+        var list = new ObservableList<int>();
+        var handledBefore = false;
+        var handledAfter = false;
+
+        using var changes = list
+            .Changed
+            .Do(_ => handledBefore = true)
+            .OnItemAdded(item =>
+            {
+                Assert.True(handledBefore);
+                Assert.False(handledAfter);
+            })
+            .Do(_ => handledAfter = true)
+            .Subscribe();
+
+        list.Add(0);
+
+        Assert.True(handledBefore);
+        Assert.True(handledAfter);
+    }
+
+    [Fact]
+    public void OnItemRemoved()
+    {
+        var list = new ObservableList<int>();
+        var handledBefore = false;
+        var handledAfter = false;
+
+        using var changes = list
+            .Changed
+            .Do(_ => handledBefore = true)
+            .OnItemRemoved(item =>
+            {
+                Assert.True(handledBefore);
+                Assert.True(handledAfter);
+            })
+            .Do(_ => handledAfter = true)
+            .Subscribe();
+
+        list.Add(0);
+
+        Assert.True(handledBefore);
+        Assert.True(handledAfter);
+    }
+
     [Fact]
     public void OrderByStatic()
     {