From fd935259aafa01e584801cbf84903f8767349eb3 Mon Sep 17 00:00:00 2001
From: leo <longshuang@msn.cn>
Date: Fri, 16 May 2025 12:22:37 +0800
Subject: [PATCH] refactor: build tags view data in viewmodels instead of views

Signed-off-by: leo <longshuang@msn.cn>
---
 src/ViewModels/Preferences.cs         |   7 +-
 src/ViewModels/Repository.cs          |  28 ++++-
 src/ViewModels/SubmoduleCollection.cs |  10 +-
 src/ViewModels/TagCollection.cs       |  72 ++++++++++++-
 src/Views/Repository.axaml            |   5 +-
 src/Views/TagsView.axaml              |  53 +++++----
 src/Views/TagsView.axaml.cs           | 148 +++++---------------------
 7 files changed, 151 insertions(+), 172 deletions(-)

diff --git a/src/ViewModels/Preferences.cs b/src/ViewModels/Preferences.cs
index 5ce6f769..2698067e 100644
--- a/src/ViewModels/Preferences.cs
+++ b/src/ViewModels/Preferences.cs
@@ -178,9 +178,9 @@ namespace SourceGit.ViewModels
 
         public bool ShowTagsAsTree
         {
-            get => _showTagsAsTree;
-            set => SetProperty(ref _showTagsAsTree, value);
-        }
+            get;
+            set;
+        } = false;
 
         public bool ShowTagsInGraph
         {
@@ -677,7 +677,6 @@ namespace SourceGit.ViewModels
         private double _lastCheckUpdateTime = 0;
         private string _ignoreUpdateTag = string.Empty;
 
-        private bool _showTagsAsTree = false;
         private bool _showTagsInGraph = true;
         private bool _useTwoColumnsLayoutInHistories = false;
         private bool _displayTimeAsPeriodInHistories = false;
diff --git a/src/ViewModels/Repository.cs b/src/ViewModels/Repository.cs
index c3e7c478..d24f6fbf 100644
--- a/src/ViewModels/Repository.cs
+++ b/src/ViewModels/Repository.cs
@@ -198,7 +198,21 @@ namespace SourceGit.ViewModels
             private set => SetProperty(ref _tags, value);
         }
 
-        public List<Models.Tag> VisibleTags
+        public bool ShowTagsAsTree
+        {
+            get => Preferences.Instance.ShowTagsAsTree;
+            set
+            {
+                if (value != Preferences.Instance.ShowTagsAsTree)
+                {
+                    Preferences.Instance.ShowTagsAsTree = value;
+                    VisibleTags = BuildVisibleTags();
+                    OnPropertyChanged();
+                }
+            }
+        }
+
+        public object VisibleTags
         {
             get => _visibleTags;
             private set => SetProperty(ref _visibleTags, value);
@@ -548,7 +562,7 @@ namespace SourceGit.ViewModels
             _localBranchTrees.Clear();
             _remoteBranchTrees.Clear();
             _tags.Clear();
-            _visibleTags.Clear();
+            _visibleTags = null;
             _submodules.Clear();
             _visibleSubmodules = null;
             _searchedCommits.Clear();
@@ -2492,7 +2506,7 @@ namespace SourceGit.ViewModels
             return builder;
         }
 
-        private List<Models.Tag> BuildVisibleTags()
+        private object BuildVisibleTags()
         {
             switch (_settings.TagSortMode)
             {
@@ -2523,7 +2537,11 @@ namespace SourceGit.ViewModels
 
             var historiesFilters = _settings.CollectHistoriesFilters();
             UpdateTagFilterMode(historiesFilters);
-            return visible;
+
+            if (Preferences.Instance.ShowTagsAsTree)
+                return TagCollectionAsTree.Build(visible, _visibleTags as TagCollectionAsTree);
+            else
+                return new TagCollectionAsList() { Tags = visible };
         }
 
         private object BuildVisibleSubmodules()
@@ -2775,7 +2793,7 @@ namespace SourceGit.ViewModels
         private List<BranchTreeNode> _remoteBranchTrees = new List<BranchTreeNode>();
         private List<Models.Worktree> _worktrees = new List<Models.Worktree>();
         private List<Models.Tag> _tags = new List<Models.Tag>();
-        private List<Models.Tag> _visibleTags = new List<Models.Tag>();
+        private object _visibleTags = null;
         private List<Models.Submodule> _submodules = new List<Models.Submodule>();
         private object _visibleSubmodules = null;
 
diff --git a/src/ViewModels/SubmoduleCollection.cs b/src/ViewModels/SubmoduleCollection.cs
index e2ebf634..4600496e 100644
--- a/src/ViewModels/SubmoduleCollection.cs
+++ b/src/ViewModels/SubmoduleCollection.cs
@@ -150,18 +150,12 @@ namespace SourceGit.ViewModels
             collection.Tree = SubmoduleTreeNode.Build(submodules, oldExpanded);
 
             var rows = new List<SubmoduleTreeNode>();
-            collection.MakeTreeRows(rows, collection.Tree);
+            MakeTreeRows(rows, collection.Tree);
             collection.Rows.AddRange(rows);
 
             return collection;
         }
 
-        public void Clear()
-        {
-            Tree.Clear();
-            Rows.Clear();
-        }
-
         public void ToggleExpand(SubmoduleTreeNode node)
         {
             node.IsExpanded = !node.IsExpanded;
@@ -193,7 +187,7 @@ namespace SourceGit.ViewModels
             }
         }
 
-        private void MakeTreeRows(List<SubmoduleTreeNode> rows, List<SubmoduleTreeNode> nodes)
+        private static void MakeTreeRows(List<SubmoduleTreeNode> rows, List<SubmoduleTreeNode> nodes)
         {
             foreach (var node in nodes)
             {
diff --git a/src/ViewModels/TagCollection.cs b/src/ViewModels/TagCollection.cs
index f05a727e..ce9c9508 100644
--- a/src/ViewModels/TagCollection.cs
+++ b/src/ViewModels/TagCollection.cs
@@ -1,6 +1,8 @@
 using System;
 using System.Collections.Generic;
+
 using Avalonia.Collections;
+
 using CommunityToolkit.Mvvm.ComponentModel;
 
 namespace SourceGit.ViewModels
@@ -61,7 +63,7 @@ namespace SourceGit.ViewModels
             Counter = 1;
         }
 
-        public static List<TagTreeNode> Build(IList<Models.Tag> tags, HashSet<string> expaneded)
+        public static List<TagTreeNode> Build(List<Models.Tag> tags, HashSet<string> expaneded)
         {
             var nodes = new List<TagTreeNode>();
             var folders = new Dictionary<string, TagTreeNode>();
@@ -131,7 +133,7 @@ namespace SourceGit.ViewModels
 
     public class TagCollectionAsList
     {
-        public AvaloniaList<Models.Tag> Tags
+        public List<Models.Tag> Tags
         {
             get;
             set;
@@ -151,5 +153,71 @@ namespace SourceGit.ViewModels
             get;
             set;
         } = [];
+
+        public static TagCollectionAsTree Build(List<Models.Tag> tags, TagCollectionAsTree old)
+        {
+            var oldExpanded = new HashSet<string>();
+            if (old != null)
+            {
+                foreach (var row in old.Rows)
+                {
+                    if (row.IsFolder && row.IsExpanded)
+                        oldExpanded.Add(row.FullPath);
+                }
+            }
+
+            var collection = new TagCollectionAsTree();
+            collection.Tree = TagTreeNode.Build(tags, oldExpanded);
+
+            var rows = new List<TagTreeNode>();
+            MakeTreeRows(rows, collection.Tree);
+            collection.Rows.AddRange(rows);
+
+            return collection;
+        }
+
+        public void ToggleExpand(TagTreeNode node)
+        {
+            node.IsExpanded = !node.IsExpanded;
+
+            var rows = Rows;
+            var depth = node.Depth;
+            var idx = rows.IndexOf(node);
+            if (idx == -1)
+                return;
+
+            if (node.IsExpanded)
+            {
+                var subrows = new List<TagTreeNode>();
+                MakeTreeRows(subrows, node.Children);
+                rows.InsertRange(idx + 1, subrows);
+            }
+            else
+            {
+                var removeCount = 0;
+                for (int i = idx + 1; i < rows.Count; i++)
+                {
+                    var row = rows[i];
+                    if (row.Depth <= depth)
+                        break;
+
+                    removeCount++;
+                }
+                rows.RemoveRange(idx + 1, removeCount);
+            }
+        }
+
+        private static void MakeTreeRows(List<TagTreeNode> rows, List<TagTreeNode> nodes)
+        {
+            foreach (var node in nodes)
+            {
+                rows.Add(node);
+
+                if (!node.IsExpanded || !node.IsFolder)
+                    continue;
+
+                MakeTreeRows(rows, node.Children);
+            }
+        }
     }
 }
diff --git a/src/Views/Repository.axaml b/src/Views/Repository.axaml
index 99bdd6ab..4eb303fe 100644
--- a/src/Views/Repository.axaml
+++ b/src/Views/Repository.axaml
@@ -271,7 +271,7 @@
               <ToggleButton Grid.Column="2"
                             Classes="show_as_tree"
                             Width="14"
-                            IsChecked="{Binding Source={x:Static vm:Preferences.Instance}, Path=ShowTagsAsTree, Mode=TwoWay}"
+                            IsChecked="{Binding ShowTagsAsTree, Mode=TwoWay}"
                             ToolTip.Tip="{DynamicResource Text.Repository.ShowTagsAsTree}"/>
               <Button Grid.Column="3"
                       Classes="icon_button"
@@ -296,8 +296,7 @@
                       Height="0"
                       Margin="8,0,4,0"
                       Background="Transparent"
-                      ShowTagsAsTree="{Binding Source={x:Static vm:Preferences.Instance}, Path=ShowTagsAsTree, Mode=OneWay}"
-                      Tags="{Binding VisibleTags}"
+                      Content="{Binding VisibleTags}"
                       Focusable="False"
                       IsVisible="{Binding IsTagGroupExpanded, Mode=OneWay}"
                       SelectionChanged="OnTagsSelectionChanged"
diff --git a/src/Views/TagsView.axaml b/src/Views/TagsView.axaml
index 2795c403..423d2425 100644
--- a/src/Views/TagsView.axaml
+++ b/src/Views/TagsView.axaml
@@ -23,32 +23,33 @@
       <ListBox Classes="repo_left_content_list"
                ItemsSource="{Binding Rows}"
                SelectionMode="Single"
-               SelectionChanged="OnRowSelectionChanged">
+               SelectionChanged="OnSelectionChanged">
+
+        <ListBox.DataTemplates>
+          <DataTemplate DataType="vm:TagTreeNodeToolTip">
+            <StackPanel Orientation="Vertical" Spacing="6">
+              <StackPanel Orientation="Horizontal">
+                <Path Width="10" Height="10" Data="{StaticResource Icons.Tag}"/>
+                <TextBlock FontWeight="Bold" Margin="4,0,0,0" Text="{Binding Name}"/>
+                <Border Background="Green" Margin="4,0,0,0" CornerRadius="4" IsVisible="{Binding IsAnnotated}">
+                  <TextBlock Text="{DynamicResource Text.CreateTag.Type.Annotated}" Classes="primary" Margin="4,0" Foreground="White"/>
+                </Border>
+              </StackPanel>
+
+              <TextBlock Text="{Binding Message}" IsVisible="{Binding Message, Converter={x:Static StringConverters.IsNotNullOrEmpty}}"/>
+            </StackPanel>
+          </DataTemplate>
+        </ListBox.DataTemplates>
+        
         <ListBox.ItemTemplate>
           <DataTemplate DataType="vm:TagTreeNode">
             <Border Height="24"
                     Background="Transparent"
-                    PointerPressed="OnRowPointerPressed"
-                    DoubleTapped="OnDoubleTappedNode"
-                    ContextRequested="OnRowContextRequested"
+                    PointerPressed="OnItemPointerPressed"
+                    DoubleTapped="OnItemDoubleTapped"
+                    ContextRequested="OnItemContextRequested"
                     ToolTip.Tip="{Binding ToolTip}"
-                    ToolTip.Placement="Right">
-              <Border.DataTemplates>
-                <DataTemplate DataType="vm:TagTreeNodeToolTip">
-                  <StackPanel Orientation="Vertical" Spacing="6">
-                    <StackPanel Orientation="Horizontal">
-                      <Path Width="10" Height="10" Data="{StaticResource Icons.Tag}"/>
-                      <TextBlock FontWeight="Bold" Margin="4,0,0,0" Text="{Binding Name}"/>
-                      <Border Background="Green" Margin="4,0,0,0" CornerRadius="4" IsVisible="{Binding IsAnnotated}">
-                        <TextBlock Text="{DynamicResource Text.CreateTag.Type.Annotated}" Classes="primary" Margin="4,0" Foreground="White"/>
-                      </Border>
-                    </StackPanel>
-
-                    <TextBlock Text="{Binding Message}" IsVisible="{Binding Message, Converter={x:Static StringConverters.IsNotNullOrEmpty}}"/>
-                  </StackPanel>
-                </DataTemplate>
-              </Border.DataTemplates>
-              
+                    ToolTip.Placement="Right">              
               <Grid ColumnDefinitions="16,Auto,*,Auto"
                     Margin="{Binding Depth, Converter={x:Static c:IntConverters.ToTreeMargin}}"
                     VerticalAlignment="Center">
@@ -59,9 +60,7 @@
                                            IsChecked="{Binding IsExpanded, Mode=OneWay}"
                                            IsVisible="{Binding IsFolder}"/>
 
-                <v:TagTreeNodeIcon Grid.Column="1"
-                                   Node="{Binding .}"
-                                   IsExpanded="{Binding IsExpanded, Mode=OneWay}"/>
+                <v:TagTreeNodeIcon Grid.Column="1" IsExpanded="{Binding IsExpanded, Mode=OneWay}"/>
 
                 <TextBlock Grid.Column="2"
                            Classes="primary"
@@ -89,13 +88,13 @@
                Margin="8,0,0,0"
                ItemsSource="{Binding Tags}"
                SelectionMode="Single"
-               SelectionChanged="OnRowSelectionChanged">
+               SelectionChanged="OnSelectionChanged">
         <ListBox.ItemTemplate>
           <DataTemplate DataType="m:Tag">
             <Border Height="24"
                     Background="Transparent"
-                    PointerPressed="OnRowPointerPressed"
-                    ContextRequested="OnRowContextRequested"
+                    PointerPressed="OnItemPointerPressed"
+                    ContextRequested="OnItemContextRequested"
                     ToolTip.Placement="Right">
               <ToolTip.Tip>
                 <StackPanel Orientation="Vertical" Spacing="6">
diff --git a/src/Views/TagsView.axaml.cs b/src/Views/TagsView.axaml.cs
index ba6740c0..6764c7ea 100644
--- a/src/Views/TagsView.axaml.cs
+++ b/src/Views/TagsView.axaml.cs
@@ -1,5 +1,4 @@
 using System;
-using System.Collections.Generic;
 
 using Avalonia;
 using Avalonia.Controls;
@@ -31,15 +30,6 @@ namespace SourceGit.Views
 
     public class TagTreeNodeIcon : UserControl
     {
-        public static readonly StyledProperty<ViewModels.TagTreeNode> NodeProperty =
-            AvaloniaProperty.Register<TagTreeNodeIcon, ViewModels.TagTreeNode>(nameof(Node));
-
-        public ViewModels.TagTreeNode Node
-        {
-            get => GetValue(NodeProperty);
-            set => SetValue(NodeProperty, value);
-        }
-
         public static readonly StyledProperty<bool> IsExpandedProperty =
             AvaloniaProperty.Register<TagTreeNodeIcon, bool>(nameof(IsExpanded));
 
@@ -49,16 +39,23 @@ namespace SourceGit.Views
             set => SetValue(IsExpandedProperty, value);
         }
 
-        static TagTreeNodeIcon()
+        protected override void OnDataContextChanged(EventArgs e)
         {
-            NodeProperty.Changed.AddClassHandler<TagTreeNodeIcon>((icon, _) => icon.UpdateContent());
-            IsExpandedProperty.Changed.AddClassHandler<TagTreeNodeIcon>((icon, _) => icon.UpdateContent());
+            base.OnDataContextChanged(e);
+            UpdateContent();
+        }
+
+        protected override void OnPropertyChanged(AvaloniaPropertyChangedEventArgs change)
+        {
+            base.OnPropertyChanged(change);
+
+            if (change.Property == IsExpandedProperty)
+                UpdateContent();
         }
 
         private void UpdateContent()
         {
-            var node = Node;
-            if (node == null)
+            if (DataContext is not ViewModels.TagTreeNode node)
             {
                 Content = null;
                 return;
@@ -92,24 +89,6 @@ namespace SourceGit.Views
 
     public partial class TagsView : UserControl
     {
-        public static readonly StyledProperty<bool> ShowTagsAsTreeProperty =
-            AvaloniaProperty.Register<TagsView, bool>(nameof(ShowTagsAsTree));
-
-        public bool ShowTagsAsTree
-        {
-            get => GetValue(ShowTagsAsTreeProperty);
-            set => SetValue(ShowTagsAsTreeProperty, value);
-        }
-
-        public static readonly StyledProperty<List<Models.Tag>> TagsProperty =
-            AvaloniaProperty.Register<TagsView, List<Models.Tag>>(nameof(Tags));
-
-        public List<Models.Tag> Tags
-        {
-            get => GetValue(TagsProperty);
-            set => SetValue(TagsProperty, value);
-        }
-
         public static readonly RoutedEvent<RoutedEventArgs> SelectionChangedEvent =
             RoutedEvent.Register<TagsView, RoutedEventArgs>(nameof(SelectionChanged), RoutingStrategies.Tunnel | RoutingStrategies.Bubble);
 
@@ -150,33 +129,7 @@ namespace SourceGit.Views
         {
             if (Content is ViewModels.TagCollectionAsTree tree)
             {
-                node.IsExpanded = !node.IsExpanded;
-
-                var depth = node.Depth;
-                var idx = tree.Rows.IndexOf(node);
-                if (idx == -1)
-                    return;
-
-                if (node.IsExpanded)
-                {
-                    var subrows = new List<ViewModels.TagTreeNode>();
-                    MakeTreeRows(subrows, node.Children);
-                    tree.Rows.InsertRange(idx + 1, subrows);
-                }
-                else
-                {
-                    var removeCount = 0;
-                    for (int i = idx + 1; i < tree.Rows.Count; i++)
-                    {
-                        var row = tree.Rows[i];
-                        if (row.Depth <= depth)
-                            break;
-
-                        removeCount++;
-                    }
-                    tree.Rows.RemoveRange(idx + 1, removeCount);
-                }
-
+                tree.ToggleExpand(node);
                 Rows = tree.Rows.Count;
                 RaiseEvent(new RoutedEventArgs(RowsChangedEvent));
             }
@@ -186,9 +139,15 @@ namespace SourceGit.Views
         {
             base.OnPropertyChanged(change);
 
-            if (change.Property == ShowTagsAsTreeProperty || change.Property == TagsProperty)
+            if (change.Property == ContentProperty)
             {
-                UpdateDataSource();
+                if (Content is ViewModels.TagCollectionAsTree tree)
+                    Rows = tree.Rows.Count;
+                else if (Content is ViewModels.TagCollectionAsList list)
+                    Rows = list.Tags.Count;
+                else
+                    Rows = 0;
+
                 RaiseEvent(new RoutedEventArgs(RowsChangedEvent));
             }
             else if (change.Property == IsVisibleProperty)
@@ -197,7 +156,7 @@ namespace SourceGit.Views
             }
         }
 
-        private void OnDoubleTappedNode(object sender, TappedEventArgs e)
+        private void OnItemDoubleTapped(object sender, TappedEventArgs e)
         {
             if (sender is Control { DataContext: ViewModels.TagTreeNode { IsFolder: true } node })
                 ToggleNodeIsExpanded(node);
@@ -205,7 +164,7 @@ namespace SourceGit.Views
             e.Handled = true;
         }
 
-        private void OnRowPointerPressed(object sender, PointerPressedEventArgs e)
+        private void OnItemPointerPressed(object sender, PointerPressedEventArgs e)
         {
             var p = e.GetCurrentPoint(this);
             if (!p.Properties.IsLeftButtonPressed)
@@ -220,7 +179,7 @@ namespace SourceGit.Views
                 repo.NavigateToCommit(nodeTag.SHA);
         }
 
-        private void OnRowContextRequested(object sender, ContextRequestedEventArgs e)
+        private void OnItemContextRequested(object sender, ContextRequestedEventArgs e)
         {
             var control = sender as Control;
             if (control == null)
@@ -243,7 +202,7 @@ namespace SourceGit.Views
             e.Handled = true;
         }
 
-        private void OnRowSelectionChanged(object sender, SelectionChangedEventArgs _)
+        private void OnSelectionChanged(object sender, SelectionChangedEventArgs _)
         {
             var selected = (sender as ListBox)?.SelectedItem;
             var selectedTag = null as Models.Tag;
@@ -255,63 +214,6 @@ namespace SourceGit.Views
             if (selectedTag != null)
                 RaiseEvent(new RoutedEventArgs(SelectionChangedEvent));
         }
-
-        private void MakeTreeRows(List<ViewModels.TagTreeNode> rows, List<ViewModels.TagTreeNode> nodes)
-        {
-            foreach (var node in nodes)
-            {
-                rows.Add(node);
-
-                if (!node.IsExpanded || !node.IsFolder)
-                    continue;
-
-                MakeTreeRows(rows, node.Children);
-            }
-        }
-
-        private void UpdateDataSource()
-        {
-            var tags = Tags;
-            if (tags == null || tags.Count == 0)
-            {
-                Rows = 0;
-                Content = null;
-                return;
-            }
-
-            if (ShowTagsAsTree)
-            {
-                var oldExpanded = new HashSet<string>();
-                if (Content is ViewModels.TagCollectionAsTree oldTree)
-                {
-                    foreach (var row in oldTree.Rows)
-                    {
-                        if (row.IsFolder && row.IsExpanded)
-                            oldExpanded.Add(row.FullPath);
-                    }
-                }
-
-                var tree = new ViewModels.TagCollectionAsTree();
-                tree.Tree = ViewModels.TagTreeNode.Build(tags, oldExpanded);
-
-                var rows = new List<ViewModels.TagTreeNode>();
-                MakeTreeRows(rows, tree.Tree);
-                tree.Rows.AddRange(rows);
-
-                Content = tree;
-                Rows = rows.Count;
-            }
-            else
-            {
-                var list = new ViewModels.TagCollectionAsList();
-                list.Tags.AddRange(tags);
-
-                Content = list;
-                Rows = tags.Count;
-            }
-
-            RaiseEvent(new RoutedEventArgs(RowsChangedEvent));
-        }
     }
 }