enhance: manually update local branch tree after checking out a exiting local branch (#2169)

Since the remote branch tree did not changed after checkout, and the changes in the local branch tree are completely predictable, update it manually.

Signed-off-by: leo <longshuang@msn.cn>

Author: love-linger

src/ViewModels/Checkout.cs

@@ -80,18 +80,15 @@ public override async Task<bool> Sure()
                     await new Commands.Stash(_repo.FullPath)
                         .Use(log)
                         .PopAsync("stash@{0}");
+
+                _repo.FastRefreshBranchesAfterCheckout(Branch);
+            }
+            else
+            {
+                _repo.MarkWorkingCopyDirtyManually();
             }
 
             log.Complete();
-
-            var b = _repo.Branches.Find(x => x.IsLocal && x.Name == Branch);
-            if (b != null && _repo.HistoryFilterMode == Models.FilterMode.Included)
-                _repo.SetBranchFilterMode(b, Models.FilterMode.Included, false, false);
-
-            _repo.MarkBranchesDirtyManually();
-
-            ProgressDescription = "Waiting for branch updated...";
-            await Task.Delay(400);
             return succ;
         }
 

src/ViewModels/Repository.cs

@@ -788,6 +788,50 @@ public IDisposable LockWatcher()
             return _watcher?.Lock();
         }
 
+        public void FastRefreshBranchesAfterCheckout(string checkouted)
+        {
+            _watcher?.MarkBranchUpdated();
+            _watcher?.MarkWorkingCopyUpdated();
+
+            var oldWorktree = _currentBranch.WorktreePath;
+            _currentBranch.WorktreePath = null;
+
+            List<Models.Branch> locals = [];
+            Models.Branch detached = null;
+            foreach (var b in _branches)
+            {
+                if (b.IsLocal)
+                {
+                    if (b.IsDetachedHead)
+                    {
+                        detached = b;
+                        continue;
+                    }
+
+                    b.IsCurrent = b.Name.Equals(checkouted, StringComparison.Ordinal);
+                    if (b.IsCurrent)
+                    {
+                        CurrentBranch = b;
+                        b.WorktreePath = oldWorktree;
+                        if (_historyFilterMode == Models.FilterMode.Included)
+                            SetBranchFilterMode(b, Models.FilterMode.Included, false, false);
+                    }
+
+                    locals.Add(b);
+                }
+            }
+
+            if (detached != null)
+                _branches.Remove(detached);
+
+            var builder = BuildBranchTree(locals, []);
+            LocalBranchTrees = builder.Locals;
+
+            RefreshCommits();
+            RefreshWorkingCopyChanges();
+            RefreshWorktrees();
+        }
+
         public void MarkBranchesDirtyManually()
         {
             _watcher?.MarkBranchUpdated();