allow toggling tree view for commit files panel

1 files changed, 24 insertions, 0 deletions

diff --git a/pkg/gui/commit_files_panel.go b/pkg/gui/commit_files_panel.go
index d0c0228ec..aa95b0862 100644
--- a/pkg/gui/commit_files_panel.go
+++ b/pkg/gui/commit_files_panel.go
@@ -275,3 +275,27 @@ func (gui *Gui) switchToCommitFilesContext(refName string, canRebase bool, conte
 
 	return gui.pushContext(gui.Contexts.CommitFiles.Context)
 }
+
+// NOTE: this is very similar to handleToggleFileTreeView, could be DRY'd with generics
+func (gui *Gui) handleToggleCommitFileTreeView() error {
+	path := gui.getSelectedCommitFilePath()
+
+	gui.State.CommitFileChangeManager.ToggleShowTree()
+
+	// find that same node in the new format and move the cursor to it
+	if path != "" {
+		gui.State.CommitFileChangeManager.ExpandToPath(path)
+		index, found := gui.State.CommitFileChangeManager.GetIndexForPath(path)
+		if found {
+			gui.commitFilesListContext().GetPanelState().SetSelectedLineIdx(index)
+		}
+	}
+
+	if gui.getCommitFilesView().Context == COMMIT_FILES_CONTEXT_KEY {
+		if err := gui.Contexts.CommitFiles.Context.HandleRender(); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}