Subscribe ComponentStateContainer to ComponentStateHasChanged

BitzArt · May 23, 2024 · a03a75b · a03a75b
1 parent eee849a
commit a03a75b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/BitzArt.Blazor.MVVM/Factory/ViewModelFactory.cs b/src/BitzArt.Blazor.MVVM/Factory/ViewModelFactory.cs
@@ -46,6 +46,12 @@ public ViewModel Create(IServiceProvider serviceProvider, Type viewModelType, Co
         var viewModel = (ViewModel)serviceProvider.GetRequiredKeyedService(typeof(ViewModel), viewModelMap.RegistrationKey);
         viewModel.Signature = signature;
 
+        viewModel.OnComponentStateChanged += (sender) =>
+        {
+            viewModel.ComponentStateContainer?.NotifyStateChanged();
+            return Task.CompletedTask;
+        };
+
         foreach (var injection in viewModelMap.Injections)
         {
             if (injection.IsServiceInjection)
@@ -60,8 +66,10 @@ public ViewModel Create(IServiceProvider serviceProvider, Type viewModelType, Co
                 var injectedViewModel = Create(serviceProvider, injection.DependencyType, nestedSignature, parent: viewModel, affectedViewModels: affectedViewModels);
                 injection.Property.SetValue(viewModel, injectedViewModel);
 
-                viewModel.OnComponentStateContainerWasSet += (container)
-                    => injectedViewModel.ComponentStateContainer = container;
+                viewModel.OnComponentStateContainerWasSet += (container) =>
+                {
+                    injectedViewModel.ComponentStateContainer = container;
+                };
             }
 
             else if (injection.IsParentViewModelInjection)