reflog controller

2025-07-09 01:17:06 +02:00 · 2022-02-13 17:13:09 +11:00
parent 371b8d638b
commit eab00de273
4 changed files with 123 additions and 109 deletions
--- a/pkg/gui/controllers/reflog_controller.go
+++ b/pkg/gui/controllers/reflog_controller.go
@ -0,0 +1,120 @@
+package controllers
+
+import (
+	"github.com/jesseduffield/lazygit/pkg/commands/models"
+	"github.com/jesseduffield/lazygit/pkg/gui/context"
+	"github.com/jesseduffield/lazygit/pkg/gui/types"
+)
+
+type ReflogController struct {
+	baseController
+	*controllerCommon
+
+	switchToCommitFilesContext SwitchToCommitFilesContextFn
+}
+
+var _ types.IController = &ReflogController{}
+
+func NewReflogController(
+	common *controllerCommon,
+	switchToCommitFilesContext SwitchToCommitFilesContextFn,
+) *ReflogController {
+	return &ReflogController{
+		baseController:             baseController{},
+		controllerCommon:           common,
+		switchToCommitFilesContext: switchToCommitFilesContext,
+	}
+}
+
+func (self *ReflogController) GetKeybindings(opts types.KeybindingsOpts) []*types.Binding {
+	bindings := []*types.Binding{
+		{
+			Key:         opts.GetKey(opts.Config.Universal.GoInto),
+			Handler:     self.checkSelected(self.handleViewReflogCommitFiles),
+			Description: self.c.Tr.LcViewCommitFiles,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Universal.Select),
+			Handler:     self.checkSelected(self.CheckoutReflogCommit),
+			Description: self.c.Tr.LcCheckoutCommit,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Commits.ViewResetOptions),
+			Handler:     self.checkSelected(self.handleCreateReflogResetMenu),
+			Description: self.c.Tr.LcViewResetOptions,
+			OpensMenu:   true,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Commits.CherryPickCopy),
+			Handler:     opts.Guards.OutsideFilterMode(self.checkSelected(self.handleCopyReflogCommit)),
+			Description: self.c.Tr.LcCherryPickCopy,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Commits.CherryPickCopyRange),
+			Handler:     opts.Guards.OutsideFilterMode(self.checkSelected(self.handleCopyReflogCommitRange)),
+			Description: self.c.Tr.LcCherryPickCopyRange,
+		},
+		{
+			Key:         opts.GetKey(opts.Config.Commits.ResetCherryPick),
+			Handler:     self.helpers.CherryPick.Reset,
+			Description: self.c.Tr.LcResetCherryPick,
+		},
+	}
+
+	return bindings
+}
+
+func (self *ReflogController) checkSelected(callback func(*models.Commit) error) func() error {
+	return func() error {
+		commit := self.context().GetSelected()
+		if commit == nil {
+			return nil
+		}
+
+		return callback(commit)
+	}
+}
+
+func (self *ReflogController) Context() types.Context {
+	return self.context()
+}
+
+func (self *ReflogController) context() *context.ReflogCommitsContext {
+	return self.contexts.ReflogCommits
+}
+
+func (self *ReflogController) CheckoutReflogCommit(commit *models.Commit) error {
+	err := self.c.Ask(types.AskOpts{
+		Title:  self.c.Tr.LcCheckoutCommit,
+		Prompt: self.c.Tr.SureCheckoutThisCommit,
+		HandleConfirm: func() error {
+			self.c.LogAction(self.c.Tr.Actions.CheckoutReflogCommit)
+			return self.helpers.Refs.CheckoutRef(commit.Sha, types.CheckoutRefOptions{})
+		},
+	})
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (self *ReflogController) handleCreateReflogResetMenu(commit *models.Commit) error {
+	return self.helpers.Refs.CreateGitResetMenu(commit.Sha)
+}
+
+func (self *ReflogController) handleViewReflogCommitFiles(commit *models.Commit) error {
+	return self.switchToCommitFilesContext(SwitchToCommitFilesContextOpts{
+		RefName:   commit.Sha,
+		CanRebase: false,
+		Context:   self.context(),
+	})
+}
+
+func (self *ReflogController) handleCopyReflogCommit(commit *models.Commit) error {
+	return self.helpers.CherryPick.Copy(commit, self.model.FilteredReflogCommits, self.context())
+}
+
+func (self *ReflogController) handleCopyReflogCommitRange(commit *models.Commit) error {
+	return self.helpers.CherryPick.CopyRange(self.context().GetSelectedLineIdx(), self.model.FilteredReflogCommits, self.context())
+}