From 8cb10f76e46ef2c9ed413b7c6260b6678a72a662 Mon Sep 17 00:00:00 2001
From: Jesse Duffield <jessedduffield@gmail.com>
Date: Sat, 3 Apr 2021 14:34:04 +1100
Subject: [PATCH] refresh main panel when switching between tree and flat mode

---
 pkg/gui/commit_files_panel.go | 3 +++
 pkg/gui/files_panel.go        | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/pkg/gui/commit_files_panel.go b/pkg/gui/commit_files_panel.go
index 626fe173c..287410ba9 100644
--- a/pkg/gui/commit_files_panel.go
+++ b/pkg/gui/commit_files_panel.go
@@ -291,6 +291,9 @@ func (gui *Gui) handleToggleCommitFileTreeView() error {
 		if err := gui.Contexts.CommitFiles.HandleRender(); err != nil {
 			return err
 		}
+		if err := gui.Contexts.CommitFiles.HandleFocus(); err != nil {
+			return err
+		}
 	}
 
 	return nil
diff --git a/pkg/gui/files_panel.go b/pkg/gui/files_panel.go
index d95875d5e..5a7ee6c66 100644
--- a/pkg/gui/files_panel.go
+++ b/pkg/gui/files_panel.go
@@ -868,6 +868,9 @@ func (gui *Gui) handleToggleFileTreeView() error {
 		if err := gui.Contexts.Files.HandleRender(); err != nil {
 			return err
 		}
+		if err := gui.Contexts.Files.HandleFocus(); err != nil {
+			return err
+		}
 	}
 
 	return nil