summaryrefslogtreecommitdiffstats
path: root/pkg/gui/controllers/switch_to_diff_files_controller.go
blob: 5207aeaf5bc9d019ce584dc75bb63bc0a7775bdf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package controllers

import (
	"github.com/jesseduffield/lazygit/pkg/gui/context"
	"github.com/jesseduffield/lazygit/pkg/gui/types"
)

// This controller is for all contexts that contain commit files.

var _ types.IController = &SwitchToDiffFilesController{}

type CanSwitchToDiffFiles interface {
	types.IListContext
	CanRebase() bool
	GetSelectedRef() types.Ref
}

// Not using our ListControllerTrait because our 'selected' item is not a list item
// but an attribute on it i.e. the ref of an item.
type SwitchToDiffFilesController struct {
	baseController
	*ListControllerTrait[types.Ref]
	c                *ControllerCommon
	context          CanSwitchToDiffFiles
	diffFilesContext *context.CommitFilesContext
}

func NewSwitchToDiffFilesController(
	c *ControllerCommon,
	context CanSwitchToDiffFiles,
	diffFilesContext *context.CommitFilesContext,
) *SwitchToDiffFilesController {
	return &SwitchToDiffFilesController{
		baseController: baseController{},
		ListControllerTrait: NewListControllerTrait[types.Ref](
			c,
			context,
			context.GetSelectedRef,
			func() ([]types.Ref, int, int) {
				panic("Not implemented")
			},
		),
		c:                c,
		context:          context,
		diffFilesContext: diffFilesContext,
	}
}

func (self *SwitchToDiffFilesController) GetKeybindings(opts types.KeybindingsOpts) []*types.Binding {
	bindings := []*types.Binding{
		{
			Key:               opts.GetKey(opts.Config.Universal.GoInto),
			Handler:           self.withItem(self.enter),
			GetDisabledReason: self.require(self.singleItemSelected()),
			Description:       self.c.Tr.ViewItemFiles,
		},
	}

	return bindings
}

func (self *SwitchToDiffFilesController) GetOnClick() func() error {
	return self.withItemGraceful(self.enter)
}

func (self *SwitchToDiffFilesController) enter(ref types.Ref) error {
	return self.viewFiles(SwitchToCommitFilesContextOpts{
		Ref:       ref,
		CanRebase: self.context.CanRebase(),
		Context:   self.context,
	})
}

func (self *SwitchToDiffFilesController) viewFiles(opts SwitchToCommitFilesContextOpts) error {
	diffFilesContext := self.diffFilesContext

	diffFilesContext.SetSelection(0)
	diffFilesContext.SetRef(opts.Ref)
	diffFilesContext.SetTitleRef(opts.Ref.Description())
	diffFilesContext.SetCanRebase(opts.CanRebase)
	diffFilesContext.SetParentContext(opts.Context)
	diffFilesContext.SetWindowName(opts.Context.GetWindowName())
	diffFilesContext.ClearSearchString()
	diffFilesContext.GetView().TitlePrefix = opts.Context.GetView().TitlePrefix

	if err := self.c.Refresh(types.RefreshOptions{
		Scope: []types.RefreshableView{types.COMMIT_FILES},
	}); err != nil {
		return err
	}

	return self.c.PushContext(diffFilesContext)
}