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
94
95
96
97
98
99
100
101
|
package context
import (
"github.com/jesseduffield/gocui"
"github.com/jesseduffield/lazygit/pkg/commands/models"
"github.com/jesseduffield/lazygit/pkg/gui/filetree"
"github.com/jesseduffield/lazygit/pkg/gui/types"
"github.com/sirupsen/logrus"
)
type WorkingTreeContext struct {
*WorkingTreeViewModal
*BaseContext
*ListContextTrait
}
var _ types.IListContext = (*WorkingTreeContext)(nil)
func NewWorkingTreeContext(
getModel func() []*models.File,
getView func() *gocui.View,
getDisplayStrings func(startIdx int, length int) [][]string,
onFocus func(...types.OnFocusOpts) error,
onRenderToMain func(...types.OnFocusOpts) error,
onFocusLost func() error,
c *types.ControllerCommon,
) *WorkingTreeContext {
baseContext := NewBaseContext(NewBaseContextOpts{
ViewName: "files",
WindowName: "files",
Key: FILES_CONTEXT_KEY,
Kind: types.SIDE_CONTEXT,
})
self := &WorkingTreeContext{}
takeFocus := func() error { return c.PushContext(self) }
list := NewWorkingTreeViewModal(getModel, c.Log, c.UserConfig.Gui.ShowFileTree)
viewTrait := NewViewTrait(getView)
listContextTrait := &ListContextTrait{
base: baseContext,
listTrait: list.ListTrait,
viewTrait: viewTrait,
GetDisplayStrings: getDisplayStrings,
OnFocus: onFocus,
OnRenderToMain: onRenderToMain,
OnFocusLost: onFocusLost,
takeFocus: takeFocus,
// TODO: handle this in a trait
RenderSelection: false,
c: c,
}
self.BaseContext = baseContext
self.ListContextTrait = listContextTrait
self.WorkingTreeViewModal = list
return self
}
type WorkingTreeViewModal struct {
*ListTrait
*filetree.FileTreeViewModel
getModel func() []*models.File
}
func (self *WorkingTreeViewModal) GetItemsLength() int {
return self.FileTreeViewModel.GetItemsLength()
}
func (self *WorkingTreeViewModal) GetSelectedFileNode() *filetree.FileNode {
if self.GetItemsLength() == 0 {
return nil
}
return self.FileTreeViewModel.GetItemAtIndex(self.selectedIdx)
}
func (self *WorkingTreeViewModal) GetSelectedItem() (types.ListItem, bool) {
item := self.GetSelectedFileNode()
return item, item != nil
}
func NewWorkingTreeViewModal(getModel func() []*models.File, log *logrus.Entry, showTree bool) *WorkingTreeViewModal {
self := &WorkingTreeViewModal{
getModel: getModel,
FileTreeViewModel: filetree.NewFileTreeViewModel(getModel, log, showTree),
}
self.ListTrait = &ListTrait{
selectedIdx: 0,
HasLength: self,
}
return self
}
|