diff options
Diffstat (limited to 'pkg/gui/context/remotes_context.go')
-rw-r--r-- | pkg/gui/context/remotes_context.go | 86 |
1 files changed, 86 insertions, 0 deletions
diff --git a/pkg/gui/context/remotes_context.go b/pkg/gui/context/remotes_context.go new file mode 100644 index 000000000..28d0db20a --- /dev/null +++ b/pkg/gui/context/remotes_context.go @@ -0,0 +1,86 @@ +package context + +import ( + "github.com/jesseduffield/gocui" + "github.com/jesseduffield/lazygit/pkg/commands/models" + "github.com/jesseduffield/lazygit/pkg/gui/context/traits" + "github.com/jesseduffield/lazygit/pkg/gui/types" +) + +type RemotesContext struct { + *RemotesViewModel + *ListContextTrait +} + +var _ types.IListContext = (*RemotesContext)(nil) + +func NewRemotesContext( + getModel func() []*models.Remote, + view *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, +) *RemotesContext { + viewModel := NewRemotesViewModel(getModel) + + return &RemotesContext{ + RemotesViewModel: viewModel, + ListContextTrait: &ListContextTrait{ + Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{ + ViewName: "branches", + WindowName: "branches", + Key: REMOTES_CONTEXT_KEY, + Kind: types.SIDE_CONTEXT, + Focusable: true, + }), ContextCallbackOpts{ + OnFocus: onFocus, + OnFocusLost: onFocusLost, + OnRenderToMain: onRenderToMain, + }), + list: viewModel, + viewTrait: NewViewTrait(view), + getDisplayStrings: getDisplayStrings, + c: c, + }, + } +} + +func (self *RemotesContext) GetSelectedItemId() string { + item := self.GetSelected() + if item == nil { + return "" + } + + return item.ID() +} + +type RemotesViewModel struct { + *traits.ListCursor + getModel func() []*models.Remote +} + +func NewRemotesViewModel(getModel func() []*models.Remote) *RemotesViewModel { + self := &RemotesViewModel{ + getModel: getModel, + } + + self.ListCursor = traits.NewListCursor(self) + + return self +} + +func (self *RemotesViewModel) GetItemsLength() int { + return len(self.getModel()) +} + +func (self *RemotesViewModel) GetSelected() *models.Remote { + if self.GetItemsLength() == 0 { + return nil + } + + return self.getModel()[self.GetSelectedLineIdx()] +} |