From 767f9db2869d470e9ce423624c0e5903ceba1497 Mon Sep 17 00:00:00 2001
From: TheXamlGuy <dan_clark@outlook.com>
Date: Fri, 12 Jul 2024 23:06:45 +0100
Subject: [PATCH] If we are in favourite view, and we unfavourite something, we
 need to remove it

---
 ...temNavigationCollectionViewModelActivatedHandler.cs |  2 +-
 Wallet/ItemNavigationViewModel.cs                      | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/Wallet/ItemNavigationCollectionViewModelActivatedHandler.cs b/Wallet/ItemNavigationCollectionViewModelActivatedHandler.cs
index 8f848c9..6d47e20 100644
--- a/Wallet/ItemNavigationCollectionViewModelActivatedHandler.cs
+++ b/Wallet/ItemNavigationCollectionViewModelActivatedHandler.cs
@@ -31,7 +31,7 @@ public class ItemNavigationCollectionViewModelActivatedHandler(IMediator mediato
                     IDecoratorService<Item<(Guid, string)>> decoratorService = serviceScope.ServiceProvider
                         .GetRequiredService<IDecoratorService<Item<(Guid, string)>>>();
 
-                    if (serviceFactory.Create<ItemNavigationViewModel>(args => args.Initialize(), 
+                    if (serviceFactory.Create<ItemNavigationViewModel>(args => args.Initialize(), configuration.Filter, 
                         Id, Name, "Description", Category, selected, Favourite, Archived) 
                         is ItemNavigationViewModel viewModel)
                     {
diff --git a/Wallet/ItemNavigationViewModel.cs b/Wallet/ItemNavigationViewModel.cs
index ad39c97..4afdb7a 100644
--- a/Wallet/ItemNavigationViewModel.cs
+++ b/Wallet/ItemNavigationViewModel.cs
@@ -12,6 +12,7 @@ public partial class ItemNavigationViewModel(IServiceProvider provider,
     IDisposer disposer,
     IContentTemplate template,
     NamedComponent named,
+    string filter,
     Guid id,
     string name = "",
     string description = "",
@@ -37,6 +38,9 @@ public partial class ItemNavigationViewModel(IServiceProvider provider,
     [ObservableProperty]
     private string? description = description;
 
+    [ObservableProperty]
+    private string filter = filter;
+
     [ObservableProperty]
     private Guid id = id;
 
@@ -60,6 +64,7 @@ public partial class ItemNavigationViewModel(IServiceProvider provider,
 
     [ObservableProperty]
     private string named = $"{named}";
+
     public IContentTemplate Template { get; set; } = template;
 
     public Task Handle(ArchiveEventArgs<Item> args)
@@ -82,6 +87,11 @@ public partial class ItemNavigationViewModel(IServiceProvider provider,
 
     public Task Handle(UnfavouriteEventArgs<Item> args)
     {
+        if (Filter == "Favourites")
+        {
+            Dispose();
+        }
+
         IsFavourite = false;
         return Task.CompletedTask;
     }