添加一个相对快速的方法找到对应的头

Author: heartacker

src/Commands/QueryCommits.cs

@@ -62,6 +62,7 @@ public QueryCommits(string repo, string filter, Models.CommitSearchMethod method
             if (!rs.IsSuccess)
                 return _commits;
 
+            var commitIdx = 0;
             var nextPartIdx = 0;
             var start = 0;
             var end = rs.StdOut.IndexOf('\n', start);
@@ -71,7 +72,7 @@ public QueryCommits(string repo, string filter, Models.CommitSearchMethod method
                 switch (nextPartIdx)
                 {
                     case 0:
-                        _current = new Models.Commit() { SHA = line };
+                        _current = new Models.Commit() { SHA = line, Index = commitIdx++ };
                         _current.IsCommitFilterHead = _patterns.Count > 0 && _patterns.Any(f => line.StartsWith(f));
                         _commits.Add(_current);
                         break;

src/Models/Commit.cs

@@ -55,6 +55,11 @@ public static double OpacityForNotMerged
         public bool IsCommitterVisible => !Author.Equals(Committer) || AuthorTime != CommitterTime;
         public bool IsCurrentHead => Decorators.Find(x => x.Type is DecoratorType.CurrentBranchHead or DecoratorType.CurrentCommitHead) != null;
 
+        /// <summary>
+        /// the index of commit in current history's commits
+        /// </summary>
+        public int Index { get; set; } = 0;
+
         public int Color { get; set; } = 0;
         public double Opacity => IsMerged ? 1 : OpacityForNotMerged;
         public FontWeight FontWeight => IsCurrentHead ? FontWeight.Bold : FontWeight.Regular;

src/ViewModels/Histories.cs

@@ -446,6 +446,47 @@ private void NavigateTo(Models.Commit commit)
             }
         }
 
+        public List<Models.Commit> GetChildrensInCurHistory(Models.Commit commit)
+        {
+            if (commit == null || commit.Index == 0) return new List<Models.Commit>();
+            return _commits[..(commit.Index)].Where(x => x.Parents.Contains(commit.SHA)).ToList();
+        }
+
+        public List<Models.Commit> GetTopHeadsInCurHistory(Models.Commit commit)
+        {
+            // if (commit == null || commit.Index <= 0) return new List<Models.Commit>();
+            var heads = new List<Models.Commit>();
+            if (commit == null || commit?.Index <= 0)
+                return heads;
+            var queue = new Queue<Models.Commit>();
+            var visited = new HashSet<string> { commit.SHA };
+
+            queue.Enqueue(commit);
+
+            while (queue.Any())
+            {
+                var currentCommit = queue.Dequeue();
+                var children = GetChildrensInCurHistory(currentCommit);
+                // var children = _commits.Where(x => x.Parents.Contains(currentCommit.SHA)).ToList();
+                if (children.Any())
+                {
+                    heads.Remove(currentCommit);
+
+                    foreach (var child in children)
+                    {
+                        // 如果这个子提交还没有被访问过，则加入队列并标记为已访问
+                        if (visited.Add(child.SHA))
+                        {
+                            queue.Enqueue(child);
+                            heads.Add(child);
+                        }
+                    }
+                }
+            }
+
+            return heads;
+        }
+
         private Repository _repo = null;
         private bool _isLoading = true;
         private List<Models.Commit> _commits = new List<Models.Commit>();