From e39d2ed44bc8cc187abe3cd59349c40ca126b7a6 Mon Sep 17 00:00:00 2001
From: Glenn Vriesman <glenn.vriesman@gmail.com>
Date: Sun, 2 Dec 2018 17:34:15 +0100
Subject: [PATCH] Added check to invoke continue/refresh

---
 pkg/gui/branches_panel.go | 5 ++---
 pkg/gui/merge_panel.go    | 4 ++++
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/pkg/gui/branches_panel.go b/pkg/gui/branches_panel.go
index ce1649f64..7252665f9 100644
--- a/pkg/gui/branches_panel.go
+++ b/pkg/gui/branches_panel.go
@@ -108,17 +108,16 @@ func (gui *Gui) handleRebase(g *gocui.Gui, v *gocui.View) error {
 			}
 
 			if err := gui.GitCommand.RebaseBranch(selectedBranch); err != nil {
+				gui.Log.Errorln(err)
 				if err := gui.createConfirmationPanel(g, v, "Rebase failed", "Rebasing failed, would you like to resolve it?",
 					func(g *gocui.Gui, v *gocui.View) error {
 						return nil
-					},
-					func(g *gocui.Gui, v *gocui.View) error {
+					}, func(g *gocui.Gui, v *gocui.View) error {
 						return gui.GitCommand.AbortRebaseBranch()
 					}); err != nil {
 				}
 			}
 
-			gui.Log.Println("Reached refresh")
 			return gui.refreshSidePanels(g)
 		}, nil)
 }
diff --git a/pkg/gui/merge_panel.go b/pkg/gui/merge_panel.go
index 2592275bf..1a4a25816 100644
--- a/pkg/gui/merge_panel.go
+++ b/pkg/gui/merge_panel.go
@@ -256,5 +256,9 @@ func (gui *Gui) handleCompleteMerge(g *gocui.Gui) error {
 	}
 	gui.stageSelectedFile(g)
 	gui.refreshFiles(g)
+	if rebase, err := gui.GitCommand.IsInRebaseState(); rebase && err == nil {
+		gui.GitCommand.ContinueRebaseBranch()
+		gui.refreshSidePanels(g)
+	}
 	return gui.switchFocus(g, nil, filesView)
 }