refactor to get view tab context map into gui state

pkg/gui/context.go

@@ -694,7 +694,7 @@ func (gui *Gui) changeMainViewsContext(contextKey string) {
 }
 
 func (gui *Gui) viewTabNames(viewName string) []string {
-	tabContexts := gui.ViewTabContextMap[viewName]
+	tabContexts := gui.State.ViewTabContextMap[viewName]
 
 	if len(tabContexts) == 0 {
 		return nil
@@ -710,7 +710,7 @@ func (gui *Gui) viewTabNames(viewName string) []string {
 
 func (gui *Gui) setViewTabForContext(c Context) {
 	// search for the context in our map and if we find it, set the tab for the corresponding view
-	tabContexts, ok := gui.ViewTabContextMap[c.GetViewName()]
+	tabContexts, ok := gui.State.ViewTabContextMap[c.GetViewName()]
 	if !ok {
 		return
 	}

pkg/gui/gui.go

@@ -60,6 +60,8 @@ func NewContextManager(contexts ContextTree) ContextManager {
 	}
 }
 
+type Repo string
+
 // Gui wraps the gocui Gui object which handles rendering and events
 type Gui struct {
 	g                    *gocui.Gui
@@ -68,6 +70,7 @@ type Gui struct {
 	OSCommand            *oscommands.OSCommand
 	SubProcess           *exec.Cmd
 	State                *guiState
+	RepoStateMap         map[Repo]*guiState
 	Config               config.AppConfigurer
 	Tr                   *i18n.TranslationSet
 	Errors               SentinelErrors
@@ -81,9 +84,8 @@ type Gui struct {
 
 	// when lazygit is opened outside a git directory we want to open to the most
 	// recent repo with the recent repos popup showing
-	showRecentRepos   bool
+	showRecentRepos bool
-	Contexts          ContextTree
+	Contexts        ContextTree
-	ViewTabContextMap map[string][]tabContext
 
 	// this array either includes the events that we're recording in this session
 	// or the events we've recorded in a prior session
@@ -314,8 +316,9 @@ type guiState struct {
 
 	Modes Modes
 
-	ContextManager ContextManager
+	ContextManager    ContextManager
-	ViewContextMap map[string]Context
+	ViewContextMap    map[string]Context
+	ViewTabContextMap map[string][]tabContext
 
 	// WindowViewNameMap is a mapping of windows to the current view of that window.
 	// Some views move between windows for example the commitFiles view and when cycling through
@@ -327,6 +330,16 @@ type guiState struct {
 }
 
 func (gui *Gui) resetState(filterPath string) {
+	currentDir, err := os.Getwd()
+	if err == nil {
+		if state := gui.RepoStateMap[Repo(currentDir)]; state != nil {
+			gui.State = state
+			return
+		}
+	} else {
+		gui.Log.Error(err)
+	}
+
 	showTree := gui.Config.GetUserConfig().Gui.ShowFileTree
 
 	screenMode := SCREEN_NORMAL
@@ -374,12 +387,13 @@ func (gui *Gui) resetState(filterPath string) {
 			},
 			Diffing: Diffing{},
 		},
-		ViewContextMap: gui.initialViewContextMap(),
+		ViewContextMap:    gui.initialViewContextMap(),
-		ScreenMode:     screenMode,
+		ViewTabContextMap: gui.initialViewTabContextMap(),
-		ContextManager: NewContextManager(gui.Contexts),
+		ScreenMode:        screenMode,
+		ContextManager:    NewContextManager(gui.Contexts),
 	}
 
-	gui.ViewTabContextMap = gui.initialViewTabContextMap()
+	gui.RepoStateMap[Repo(gui.GitCommand.DotGitDir)] = gui.State
 }
 
 // for now the split view will always be on
@@ -397,6 +411,7 @@ func NewGui(log *logrus.Entry, gitCommand *commands.GitCommand, oSCommand *oscom
 		showRecentRepos:      showRecentRepos,
 		RecordedEvents:       []RecordedEvent{},
 		RepoPathStack:        []string{},
+		RepoStateMap:         map[Repo]*guiState{},
 	}
 
 	gui.Contexts = gui.contextTree()

pkg/gui/view_helpers.go

@@ -431,7 +431,7 @@ func (gui *Gui) clearEditorView(v *gocui.View) {
 }
 
 func (gui *Gui) onViewTabClick(viewName string, tabIndex int) error {
-	context := gui.ViewTabContextMap[viewName][tabIndex].contexts[0]
+	context := gui.State.ViewTabContextMap[viewName][tabIndex].contexts[0]
 
 	return gui.pushContext(context)
 }