From e114b5b5e8dab86e081f0fe789e1bd902687b727 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Luka=20Marku=C5=A1i=C4=87?= <luka.markusic@microblink.com>
Date: Wed, 13 Apr 2022 15:46:12 +0200
Subject: [PATCH] Refresh BRANCHES and COMMITS after (un)staging upstream

---
 pkg/gui/controllers/branches_controller.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/pkg/gui/controllers/branches_controller.go b/pkg/gui/controllers/branches_controller.go
index 66187e36a..096082ac4 100644
--- a/pkg/gui/controllers/branches_controller.go
+++ b/pkg/gui/controllers/branches_controller.go
@@ -116,6 +116,15 @@ func (self *BranchesController) setUpstream(selectedBranch *models.Branch) error
 					if err := self.git.Branch.UnsetUpstream(selectedBranch.Name); err != nil {
 						return self.c.Error(err)
 					}
+					if err := self.c.Refresh(types.RefreshOptions{
+						Mode: types.SYNC,
+						Scope: []types.RefreshableView{
+							types.BRANCHES,
+							types.COMMITS,
+						},
+					}); err != nil {
+						return self.c.Error(err)
+					}
 					return nil
 				},
 				Key: 'u',
@@ -132,6 +141,15 @@ func (self *BranchesController) setUpstream(selectedBranch *models.Branch) error
 						if err := self.git.Branch.SetUpstream(upstreamRemote, upstreamBranch, selectedBranch.Name); err != nil {
 							return self.c.Error(err)
 						}
+						if err := self.c.Refresh(types.RefreshOptions{
+							Mode: types.SYNC,
+							Scope: []types.RefreshableView{
+								types.BRANCHES,
+								types.COMMITS,
+							},
+						}); err != nil {
+							return self.c.Error(err)
+						}
 						return nil
 					})
 				},