diff options
Diffstat (limited to 'pkg/gui/context/suggestions_context.go')
-rw-r--r-- | pkg/gui/context/suggestions_context.go | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/pkg/gui/context/suggestions_context.go b/pkg/gui/context/suggestions_context.go new file mode 100644 index 000000000..5320e40c6 --- /dev/null +++ b/pkg/gui/context/suggestions_context.go @@ -0,0 +1,85 @@ +package context + +import ( + "github.com/jesseduffield/gocui" + "github.com/jesseduffield/lazygit/pkg/gui/context/traits" + "github.com/jesseduffield/lazygit/pkg/gui/types" +) + +type SuggestionsContext struct { + *SuggestionsViewModel + *ListContextTrait +} + +var _ types.IListContext = (*SuggestionsContext)(nil) + +func NewSuggestionsContext( + getModel func() []*types.Suggestion, + 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, +) *SuggestionsContext { + viewModel := NewSuggestionsViewModel(getModel) + + return &SuggestionsContext{ + SuggestionsViewModel: viewModel, + ListContextTrait: &ListContextTrait{ + Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{ + ViewName: "suggestions", + WindowName: "suggestions", + Key: SUGGESTIONS_CONTEXT_KEY, + Kind: types.PERSISTENT_POPUP, + Focusable: true, + }), ContextCallbackOpts{ + OnFocus: onFocus, + OnFocusLost: onFocusLost, + OnRenderToMain: onRenderToMain, + }), + list: viewModel, + viewTrait: NewViewTrait(view), + getDisplayStrings: getDisplayStrings, + c: c, + }, + } +} + +func (self *SuggestionsContext) GetSelectedItemId() string { + item := self.GetSelected() + if item == nil { + return "" + } + + return item.Value +} + +type SuggestionsViewModel struct { + *traits.ListCursor + getModel func() []*types.Suggestion +} + +func NewSuggestionsViewModel(getModel func() []*types.Suggestion) *SuggestionsViewModel { + self := &SuggestionsViewModel{ + getModel: getModel, + } + + self.ListCursor = traits.NewListCursor(self) + + return self +} + +func (self *SuggestionsViewModel) GetItemsLength() int { + return len(self.getModel()) +} + +func (self *SuggestionsViewModel) GetSelected() *types.Suggestion { + if self.GetItemsLength() == 0 { + return nil + } + + return self.getModel()[self.GetSelectedLineIdx()] +} |