Begin refactoring gui

This begins a big refactor of moving more code out of the Gui struct into contexts, controllers, and helpers. We also move some code into structs in the
gui package purely for the sake of better encapsulation

1 files changed, 198 insertions, 0 deletions

diff --git a/pkg/gui/controllers/status_controller.go b/pkg/gui/controllers/status_controller.go
new file mode 100644
index 000000000..f89f7e16f
--- /dev/null
+++ b/pkg/gui/controllers/status_controller.go
@@ -0,0 +1,198 @@
+package controllers
+
+import (
+	"errors"
+	"fmt"
+	"strings"
+
+	"github.com/jesseduffield/generics/slices"
+	"github.com/jesseduffield/lazygit/pkg/commands/types/enums"
+	"github.com/jesseduffield/lazygit/pkg/constants"
+	"github.com/jesseduffield/lazygit/pkg/gui/presentation"
+	"github.com/jesseduffield/lazygit/pkg/gui/style"
+	"github.com/jesseduffield/lazygit/pkg/gui/types"
+	"github.com/jesseduffield/lazygit/pkg/utils"
+)
+
+type StatusController struct {
+	baseController
+	*controllerCommon
+}
+
+var _ types.IController = &StatusController{}
+
+func NewStatusController(
+	common *controllerCommon,
+) *StatusController {
+	return &StatusController{
+		baseController:   baseController{},
+		controllerCommon: common,
+	}
+}
+
+func (self *StatusController) GetKeybindings(opts types.KeybindingsOpts) []*types.Binding {
+	bindings := []*types.Binding{
+		{
+			Key:         opts.GetKey(opts.Config.Universal.OpenFile),
+			Handler:     self.openConfig,
+			Description: self.c.Tr.OpenConfig,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Universal.Edit),
+			Handler:     self.editConfig,
+			Description: self.c.Tr.EditConfig,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Status.CheckForUpdate),
+			Handler:     self.handleCheckForUpdate,
+			Description: self.c.Tr.LcCheckForUpdate,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Status.RecentRepos),
+			Handler:     self.helpers.Repos.CreateRecentReposMenu,
+			Description: self.c.Tr.SwitchRepo,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Status.AllBranchesLogGraph),
+			Handler:     self.showAllBranchLogs,
+			Description: self.c.Tr.LcAllBranchesLogGraph,
+		},
+	}
+
+	return bindings
+}
+
+func (self *StatusController) GetOnRenderToMain() func() error {
+	return func() error {
+		dashboardString := strings.Join(
+			[]string{
+				lazygitTitle(),
+				"Copyright 2022 Jesse Duffield",
+				fmt.Sprintf("Keybindings: %s", constants.Links.Docs.Keybindings),
+				fmt.Sprintf("Config Options: %s", constants.Links.Docs.Config),
+				fmt.Sprintf("Tutorial: %s", constants.Links.Docs.Tutorial),
+				fmt.Sprintf("Raise an Issue: %s", constants.Links.Issues),
+				fmt.Sprintf("Release Notes: %s", constants.Links.Releases),
+				style.FgMagenta.Sprintf("Become a sponsor: %s", constants.Links.Donate), // caffeine ain't free
+			}, "\n\n")
+
+		return self.c.RenderToMainViews(types.RefreshMainOpts{
+			Pair: self.c.MainViewPairs().Normal,
+			Main: &types.ViewUpdateOpts{
+				Title: self.c.Tr.StatusTitle,
+				Task:  types.NewRenderStringTask(dashboardString),
+			},
+		})
+	}
+}
+
+func (self *StatusController) GetOnClick() func() error {
+	return self.onClick
+}
+
+func (self *StatusController) Context() types.Context {
+	return self.contexts.Status
+}
+
+func (self *StatusController) onClick() error {
+	// TODO: move into some abstraction (status is currently not a listViewContext where a lot of this code lives)
+	currentBranch := self.helpers.Refs.GetCheckedOutRef()
+	if currentBranch == nil {
+		// need to wait for branches to refresh
+		return nil
+	}
+
+	if err := self.c.PushContext(self.Context()); err != nil {
+		return err
+	}
+
+	cx, _ := self.c.Views().Status.Cursor()
+	upstreamStatus := presentation.BranchStatus(currentBranch, self.c.Tr)
+	repoName := utils.GetCurrentRepoName()
+	workingTreeState := self.git.Status.WorkingTreeState()
+	switch workingTreeState {
+	case enums.REBASE_MODE_REBASING, enums.REBASE_MODE_MERGING:
+		workingTreeStatus := fmt.Sprintf("(%s)", presentation.FormatWorkingTreeState(workingTreeState))
+		if cursorInSubstring(cx, upstreamStatus+" ", workingTreeStatus) {
+			return self.helpers.MergeAndRebase.CreateRebaseOptionsMenu()
+		}
+		if cursorInSubstring(cx, upstreamStatus+" "+workingTreeStatus+" ", repoName) {
+			return self.helpers.Repos.CreateRecentReposMenu()
+		}
+	default:
+		if cursorInSubstring(cx, upstreamStatus+" ", repoName) {
+			return self.helpers.Repos.CreateRecentReposMenu()
+		}
+	}
+
+	return nil
+}
+
+func runeCount(str string) int {
+	return len([]rune(str))
+}
+
+func cursorInSubstring(cx int, prefix string, substring string) bool {
+	return cx >= runeCount(prefix) && cx < runeCount(prefix+substring)
+}
+
+func lazygitTitle() string {
+	return `
+   _                       _ _
+  | |                     (_) |
+  | | __ _ _____   _  __ _ _| |_
+  | |/ _` + "`" + ` |_  / | | |/ _` + "`" + ` | | __|
+  | | (_| |/ /| |_| | (_| | | |_
+  |_|\__,_/___|\__, |\__, |_|\__|
+                __/ | __/ |
+               |___/ |___/       `
+}
+
+func (self *StatusController) askForConfigFile(action func(file string) error) error {
+	confPaths := self.c.GetConfig().GetUserConfigPaths()
+	switch len(confPaths) {
+	case 0:
+		return errors.New(self.c.Tr.NoConfigFileFoundErr)
+	case 1:
+		return action(confPaths[0])
+	default:
+		menuItems := slices.Map(confPaths, func(path string) *types.MenuItem {
+			return &types.MenuItem{
+				Label: path,
+				OnPress: func() error {
+					return action(path)
+				},
+			}
+		})
+
+		return self.c.Menu(types.CreateMenuOptions{
+			Title: self.c.Tr.SelectConfigFile,
+			Items: menuItems,
+		})
+	}
+}
+
+func (self *StatusController) openConfig() error {
+	return self.askForConfigFile(self.helpers.Files.OpenFile)
+}
+
+func (self *StatusController) editConfig() error {
+	return self.askForConfigFile(self.helpers.Files.EditFile)
+}
+
+func (self *StatusController) showAllBranchLogs() error {
+	cmdObj := self.git.Branch.AllBranchesLogCmdObj()
+	task := types.NewRunPtyTask(cmdObj.GetCmd())
+
+	return self.c.RenderToMainViews(types.RefreshMainOpts{
+		Pair: self.c.MainViewPairs().Normal,
+		Main: &types.ViewUpdateOpts{
+			Title: self.c.Tr.LogTitle,
+			Task:  task,
+		},
+	})
+}
+
+func (self *StatusController) handleCheckForUpdate() error {
+	return self.helpers.Update.CheckForUpdateInForeground()
+}