enhance: auto stash and re-apply local changes before squashing (#1141)

Signed-off-by: leo <[email protected]>

Author: love-linger

src/ViewModels/Histories.cs

@@ -419,12 +419,6 @@ public ContextMenu MakeContextMenu(ListBox list)
                         squash.Icon = App.CreateMenuIcon("Icons.SquashIntoParent");
                         squash.Click += (_, e) =>
                         {
-                            if (_repo.LocalChangesCount > 0)
-                            {
-                                App.RaiseException(_repo.FullPath, "You have local changes. Please run stash or discard first.");
-                                return;
-                            }
-
                             if (_repo.CanCreatePopup())
                                 _repo.ShowPopup(new Squash(_repo, commit, commit.SHA));
 
@@ -458,12 +452,6 @@ public ContextMenu MakeContextMenu(ListBox list)
                     squash.IsEnabled = commit.Parents.Count == 1;
                     squash.Click += (_, e) =>
                     {
-                        if (_repo.LocalChangesCount > 0)
-                        {
-                            App.RaiseException(_repo.FullPath, "You have local changes. Please run stash or discard first.");
-                            return;
-                        }
-
                         if (commit.Parents.Count == 1)
                         {
                             var parent = _commits.Find(x => x.SHA == commit.Parents[0]);

src/ViewModels/Squash.cs

@@ -33,9 +33,28 @@ public override Task<bool> Sure()
 
             return Task.Run(() =>
             {
-                var succ = new Commands.Reset(_repo.FullPath, Target.SHA, "--soft").Exec();
+                var autoStashed = false;
+                var succ = false;
+
+                if (_repo.LocalChangesCount > 0)
+                {
+                    succ = new Commands.Stash(_repo.FullPath).Push("SQUASH_AUTO_STASH");
+                    if (!succ)
+                    {
+                        CallUIThread(() => _repo.SetWatcherEnabled(true));
+                        return false;
+                    }
+
+                    autoStashed = true;
+                }
+
+                succ = new Commands.Reset(_repo.FullPath, Target.SHA, "--soft").Exec();
                 if (succ)
                     succ = new Commands.Commit(_repo.FullPath, _message, true, _repo.Settings.EnableSignOffForCommit).Run();
+
+                if (succ && autoStashed)
+                    new Commands.Stash(_repo.FullPath).Pop("stash@{0}");
+
                 CallUIThread(() => _repo.SetWatcherEnabled(true));
                 return succ;
             });