Feature: Profile filter

Author: BornToBeRoot

Source/NETworkManager/ViewModels/DNSLookupHostViewModel.cs

@@ -1,5 +1,4 @@
 using Dragablz;
-using MahApps.Metro.Controls.Dialogs;
 using NETworkManager.Controls;
 using NETworkManager.Localization.Resources;
 using NETworkManager.Models;
@@ -22,8 +21,7 @@ namespace NETworkManager.ViewModels;
 public class DNSLookupHostViewModel : ViewModelBase, IProfileManager
 {
     #region Variables
-
-    private readonly IDialogCoordinator _dialogCoordinator;
+    
     private readonly DispatcherTimer _searchDispatcherTimer = new();
 
     public IInterTabClient InterTabClient { get; }
@@ -129,6 +127,70 @@ public bool IsSearching
             OnPropertyChanged();
         }
     }
+    
+    private bool _profileFilterIsOpen;
+
+    public bool ProfileFilterIsOpen
+    {
+        get => _profileFilterIsOpen;
+        set
+        {
+            if (value == _profileFilterIsOpen)
+                return;
+
+            _profileFilterIsOpen = value;
+            OnPropertyChanged();
+        }
+    }
+
+    public ICollectionView ProfileFilterTagsView { get; }
+
+    private ObservableCollection<ProfileFilterTagsInfo> ProfileFilterTags { get; } = [];
+
+    private bool _profileFilterTagsMatchAny = GlobalStaticConfiguration.Profile_TagsMatchAny;
+
+    public bool ProfileFilterTagsMatchAny
+    {
+        get => _profileFilterTagsMatchAny;
+        set
+        {
+            if (value == _profileFilterTagsMatchAny)
+                return;
+
+            _profileFilterTagsMatchAny = value;
+            OnPropertyChanged();
+        }
+    }
+
+    private bool _profileFilterTagsMatchAll;
+
+    public bool ProfileFilterTagsMatchAll
+    {
+        get => _profileFilterTagsMatchAll;
+        set
+        {
+            if (value == _profileFilterTagsMatchAll)
+                return;
+
+            _profileFilterTagsMatchAll = value;
+            OnPropertyChanged();
+        }
+    }
+
+    private bool _isProfileFilterSet;
+
+    public bool IsProfileFilterSet
+    {
+        get => _isProfileFilterSet;
+        set
+        {
+            if (value == _isProfileFilterSet)
+                return;
+
+            _isProfileFilterSet = value;
+            OnPropertyChanged();
+        }
+    }
 
     private bool _canProfileWidthChange = true;
     private double _tempProfileWidth;
@@ -184,14 +246,13 @@ public GridLength ProfileWidth
 
     #region Constructor, load settings
 
-    public DNSLookupHostViewModel(IDialogCoordinator instance)
+    public DNSLookupHostViewModel()
     {
         _isLoading = true;
 
-        _dialogCoordinator = instance;
 
         InterTabClient = new DragablzInterTabClient(ApplicationName.DNSLookup);
-        InterTabPartition = ApplicationName.DNSLookup.ToString();
+        InterTabPartition = nameof(ApplicationName.DNSLookup);
 
         var tabId = Guid.NewGuid();
 
@@ -201,7 +262,12 @@ public DNSLookupHostViewModel(IDialogCoordinator instance)
         ];
 
         // Profiles
-        SetProfilesView();
+        CreateTags();
+
+        ProfileFilterTagsView = CollectionViewSource.GetDefaultView(ProfileFilterTags);
+        ProfileFilterTagsView.SortDescriptions.Add(new SortDescription(nameof(ProfileFilterTagsInfo.Name), ListSortDirection.Ascending));
+
+        SetProfilesView(new ProfileFilterInfo());
 
         ProfileManager.OnProfilesUpdated += ProfileManager_OnProfilesUpdated;
 
@@ -298,6 +364,36 @@ private void ClearSearchAction()
         Search = string.Empty;
     }
 
+    public ICommand OpenProfileFilterCommand => new RelayCommand(_ => OpenProfileFilterAction());
+
+    private void OpenProfileFilterAction()
+    {
+        ProfileFilterIsOpen = true;
+    }
+
+    public ICommand ApplyProfileFilterCommand => new RelayCommand(_ => ApplyProfileFilterAction());
+
+    private void ApplyProfileFilterAction()
+    {
+        RefreshProfiles();
+
+        IsProfileFilterSet = true;
+        ProfileFilterIsOpen = false;
+    }
+
+    public ICommand ClearProfileFilterCommand => new RelayCommand(_ => ClearProfileFilterAction());
+
+    private void ClearProfileFilterAction()
+    {
+        foreach (var tag in ProfileFilterTags)
+            tag.IsSelected = false;
+
+        RefreshProfiles();
+
+        IsProfileFilterSet = false;
+        ProfileFilterIsOpen = false;
+    }
+    
     public ItemActionCallback CloseItemCommand => CloseItemAction;
 
     private static void CloseItemAction(ItemActionCallbackArgs<TabablzControl> args)
@@ -360,38 +456,42 @@ public void OnViewHide()
         _isViewActive = false;
     }
 
-    private void SetProfilesView(ProfileInfo profile = null)
+    private void CreateTags()
     {
-        Profiles = new CollectionViewSource
-        {
-            Source = ProfileManager.Groups.SelectMany(x => x.Profiles).Where(x => x.DNSLookup_Enabled)
-                .OrderBy(x => x.Group).ThenBy(x => x.Name)
-        }.View;
+        var tags = ProfileManager.Groups.SelectMany(x => x.Profiles).Where(x => x.DNSLookup_Enabled).SelectMany(x => x.TagsCollection).Distinct().ToList();
 
-        Profiles.GroupDescriptions.Add(new PropertyGroupDescription(nameof(ProfileInfo.Group)));
+        var tagSet = new HashSet<string>(tags);
 
-        Profiles.Filter = o =>
+        for (var i = ProfileFilterTags.Count - 1; i >= 0; i--)
         {
-            if (string.IsNullOrEmpty(Search))
-                return true;
-
-            if (o is not ProfileInfo info)
-                return false;
-
-
+            if (!tagSet.Contains(ProfileFilterTags[i].Name))
+                ProfileFilterTags.RemoveAt(i);
+        }
 
-            var search = Search.Trim();
+        var existingTagNames = new HashSet<string>(ProfileFilterTags.Select(ft => ft.Name));
 
-            // Search by: Tag=xxx (exact match, ignore case)
-            /*
-            if (search.StartsWith(ProfileManager.TagIdentifier, StringComparison.OrdinalIgnoreCase))
-                return !string.IsNullOrEmpty(info.Tags) && info.PingMonitor_Enabled && info.Tags.Replace(" ", "").Split(';').Any(str => search.Substring(ProfileManager.TagIdentifier.Length, search.Length - ProfileManager.TagIdentifier.Length).Equals(str, StringComparison.OrdinalIgnoreCase));
-            */
+        foreach (var tag in tags.Where(tag => !existingTagNames.Contains(tag)))
+        {
+            ProfileFilterTags.Add(new ProfileFilterTagsInfo(false, tag));
+        }
+    }
+    
+    private void SetProfilesView(ProfileFilterInfo filter, ProfileInfo profile = null)
+    {
+        Profiles = new CollectionViewSource
+        {
+            Source = ProfileManager.Groups.SelectMany(x => x.Profiles).Where(x => x.DNSLookup_Enabled && (
+                    string.IsNullOrEmpty(filter.Search) || x.Name.IndexOf(filter.Search, StringComparison.Ordinal) > -1 || x.DNSLookup_Host.IndexOf(filter.Search, StringComparison.Ordinal) > -1) && (
+                    // If no tags are selected, show all profiles
+                    (!filter.Tags.Any()) ||
+                    // Any tag can match
+                    (filter.TagsFilterMatch == ProfileFilterTagsMatch.Any && filter.Tags.Any(tag => x.TagsCollection.Contains(tag))) ||
+                    // All tags must match
+                    (filter.TagsFilterMatch == ProfileFilterTagsMatch.All && filter.Tags.All(tag => x.TagsCollection.Contains(tag))))
+            ).OrderBy(x => x.Group).ThenBy(x => x.Name)
+        }.View;
 
-            // Search by: Name, DNSLookup_Host
-            return info.Name.IndexOf(search, StringComparison.OrdinalIgnoreCase) > -1 ||
-                   info.DNSLookup_Host.IndexOf(search, StringComparison.OrdinalIgnoreCase) > -1;
-        };
+        Profiles.GroupDescriptions.Add(new PropertyGroupDescription(nameof(ProfileInfo.Group)));
 
         // Set specific profile or first if null
         SelectedProfile = null;
@@ -408,7 +508,12 @@ private void RefreshProfiles()
         if (!_isViewActive)
             return;
 
-        SetProfilesView(SelectedProfile);
+        SetProfilesView(new ProfileFilterInfo
+        {
+            Search = Search,
+            Tags = [.. ProfileFilterTags.Where(x => x.IsSelected).Select(x => x.Name)],
+            TagsFilterMatch = ProfileFilterTagsMatchAny ? ProfileFilterTagsMatch.Any : ProfileFilterTagsMatch.All
+        }, SelectedProfile);
     }
 
     #endregion
@@ -417,6 +522,8 @@ private void RefreshProfiles()
 
     private void ProfileManager_OnProfilesUpdated(object sender, EventArgs e)
     {
+        CreateTags();
+        
         RefreshProfiles();
     }