Changes to stateful ViewModel lifecycle

src/BitzArt.Blazor.MVVM/Interfaces/IStatefulViewModel.cs

@@ -7,5 +7,10 @@ internal interface IStatefulViewModel
 
     public void InitializeState();
     public Task InitializeStateAsync();
+
     public void OnStateRestored();
+    public Task OnStateRestoredAsync();
+
+    public void OnStateChanged();
+    public Task OnStateChangedAsync();
 }

src/BitzArt.Blazor.MVVM/Models/PageBase.cs

@@ -39,6 +39,7 @@ private static async Task RestoreComponentStateAsync(ViewModel viewModel, string
             var buffer = Convert.FromBase64String(state);
             var json = Encoding.UTF8.GetString(buffer);
             statefulViewModel.State = JsonSerializer.Deserialize(json, statefulViewModel.StateType, StateJsonOptionsProvider.Options)!;
+            await statefulViewModel.OnStateRestoredAsync();
             statefulViewModel.OnStateRestored();
         }
         else
@@ -48,6 +49,9 @@ private static async Task RestoreComponentStateAsync(ViewModel viewModel, string
             statefulViewModel.InitializeState();
             await statefulViewModel.InitializeStateAsync();
         }
+
+        statefulViewModel.OnStateChanged();
+        await statefulViewModel.OnStateChangedAsync();
     }
 
     /// <summary>

src/BitzArt.Blazor.MVVM/Models/ViewModel.cs

@@ -33,11 +33,7 @@ public abstract class ViewModel<TState> : ViewModel, IStatefulViewModel
     public TState State
     {
         get => _state;
-        set
-        {
-            _state = value;
-            OnStateChanged(State);
-        }
+        set => _state = value;
     }
 
     object IStatefulViewModel.State
@@ -69,8 +65,18 @@ public virtual Task InitializeStateAsync()
     /// </summary>
     public virtual void OnStateRestored() { }
 
+    public virtual Task OnStateRestoredAsync()
+    {
+        return Task.CompletedTask;
+    }
+
     /// <summary>
     /// Called when the state has changed.
     /// </summary>
-    public virtual void OnStateChanged(TState state) { }
+    public virtual void OnStateChanged() { }
+
+    public virtual Task OnStateChangedAsync()
+    {
+        return Task.CompletedTask;
+    }
 }