2022-02-05 08:04:10 +02:00
|
|
|
package context
|
|
|
|
|
|
|
|
import (
|
2022-03-19 10:12:58 +02:00
|
|
|
"github.com/jesseduffield/generics/slices"
|
2022-03-27 08:15:17 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/gui/keybindings"
|
2022-03-27 08:22:31 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/gui/style"
|
2022-02-05 08:04:10 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/gui/types"
|
2023-05-21 04:06:22 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
2023-05-21 03:02:36 +02:00
|
|
|
"github.com/samber/lo"
|
2022-02-05 08:04:10 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type MenuContext struct {
|
2023-05-21 03:02:36 +02:00
|
|
|
c *ContextCommon
|
|
|
|
|
2022-02-05 08:04:10 +02:00
|
|
|
*MenuViewModel
|
|
|
|
*ListContextTrait
|
|
|
|
}
|
|
|
|
|
|
|
|
var _ types.IListContext = (*MenuContext)(nil)
|
|
|
|
|
|
|
|
func NewMenuContext(
|
2023-03-23 03:35:07 +02:00
|
|
|
c *ContextCommon,
|
2022-02-05 08:04:10 +02:00
|
|
|
) *MenuContext {
|
2023-05-21 03:02:36 +02:00
|
|
|
viewModel := NewMenuViewModel(c)
|
2022-02-05 08:04:10 +02:00
|
|
|
|
|
|
|
return &MenuContext{
|
2023-05-21 03:02:36 +02:00
|
|
|
c: c,
|
2022-02-05 08:04:10 +02:00
|
|
|
MenuViewModel: viewModel,
|
|
|
|
ListContextTrait: &ListContextTrait{
|
|
|
|
Context: NewSimpleContext(NewBaseContext(NewBaseContextOpts{
|
2023-03-21 11:57:52 +02:00
|
|
|
View: c.Views().Menu,
|
2022-06-13 03:01:26 +02:00
|
|
|
WindowName: "menu",
|
|
|
|
Key: "menu",
|
|
|
|
Kind: types.TEMPORARY_POPUP,
|
|
|
|
Focusable: true,
|
|
|
|
HasUncontrolledBounds: true,
|
2023-03-21 12:01:58 +02:00
|
|
|
})),
|
2023-07-20 13:23:46 +02:00
|
|
|
getDisplayStrings: viewModel.GetDisplayStrings,
|
|
|
|
list: viewModel,
|
|
|
|
c: c,
|
|
|
|
getColumnAlignments: func() []utils.Alignment { return viewModel.columnAlignment },
|
2022-02-05 08:04:10 +02:00
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: remove this thing.
|
|
|
|
func (self *MenuContext) GetSelectedItemId() string {
|
|
|
|
item := self.GetSelected()
|
|
|
|
if item == nil {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2022-05-08 06:23:32 +02:00
|
|
|
return item.Label
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
type MenuViewModel struct {
|
2023-07-20 13:23:46 +02:00
|
|
|
c *ContextCommon
|
|
|
|
menuItems []*types.MenuItem
|
|
|
|
columnAlignment []utils.Alignment
|
2023-05-27 06:14:43 +02:00
|
|
|
*FilteredListViewModel[*types.MenuItem]
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|
|
|
|
|
2023-05-21 03:02:36 +02:00
|
|
|
func NewMenuViewModel(c *ContextCommon) *MenuViewModel {
|
2022-02-05 08:04:10 +02:00
|
|
|
self := &MenuViewModel{
|
|
|
|
menuItems: nil,
|
2023-05-21 03:02:36 +02:00
|
|
|
c: c,
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|
|
|
|
|
2023-05-27 06:14:43 +02:00
|
|
|
self.FilteredListViewModel = NewFilteredListViewModel(
|
|
|
|
func() []*types.MenuItem { return self.menuItems },
|
|
|
|
func(item *types.MenuItem) []string { return item.LabelColumns },
|
|
|
|
)
|
2022-02-05 08:04:10 +02:00
|
|
|
|
|
|
|
return self
|
|
|
|
}
|
|
|
|
|
2023-07-20 13:23:46 +02:00
|
|
|
func (self *MenuViewModel) SetMenuItems(items []*types.MenuItem, columnAlignment []utils.Alignment) {
|
2022-02-05 08:04:10 +02:00
|
|
|
self.menuItems = items
|
2023-07-20 13:23:46 +02:00
|
|
|
self.columnAlignment = columnAlignment
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: move into presentation package
|
2022-03-19 10:12:58 +02:00
|
|
|
func (self *MenuViewModel) GetDisplayStrings(_startIdx int, _length int) [][]string {
|
2023-05-27 06:14:43 +02:00
|
|
|
menuItems := self.FilteredListViewModel.GetItems()
|
|
|
|
showKeys := slices.Some(menuItems, func(item *types.MenuItem) bool {
|
2022-03-27 08:15:17 +02:00
|
|
|
return item.Key != nil
|
|
|
|
})
|
|
|
|
|
2023-05-27 06:14:43 +02:00
|
|
|
return slices.Map(menuItems, func(item *types.MenuItem) []string {
|
2022-07-30 12:27:51 +02:00
|
|
|
displayStrings := item.LabelColumns
|
2023-05-21 03:02:36 +02:00
|
|
|
|
|
|
|
if !showKeys {
|
|
|
|
return displayStrings
|
|
|
|
}
|
|
|
|
|
|
|
|
// These keys are used for general navigation so we'll strike them out to
|
|
|
|
// avoid confusion
|
|
|
|
reservedKeys := []string{
|
|
|
|
self.c.UserConfig.Keybinding.Universal.Confirm,
|
|
|
|
self.c.UserConfig.Keybinding.Universal.Select,
|
|
|
|
self.c.UserConfig.Keybinding.Universal.Return,
|
2023-05-27 06:14:43 +02:00
|
|
|
self.c.UserConfig.Keybinding.Universal.StartSearch,
|
2023-05-21 03:02:36 +02:00
|
|
|
}
|
|
|
|
keyLabel := keybindings.LabelFromKey(item.Key)
|
|
|
|
keyStyle := style.FgCyan
|
|
|
|
if lo.Contains(reservedKeys, keyLabel) {
|
|
|
|
keyStyle = style.FgDefault.SetStrikethrough()
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|
2023-05-21 03:02:36 +02:00
|
|
|
|
|
|
|
displayStrings = slices.Prepend(displayStrings, keyStyle.Sprint(keyLabel))
|
2022-03-27 08:15:17 +02:00
|
|
|
return displayStrings
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *MenuContext) GetKeybindings(opts types.KeybindingsOpts) []*types.Binding {
|
|
|
|
basicBindings := self.ListContextTrait.GetKeybindings(opts)
|
|
|
|
menuItemsWithKeys := slices.Filter(self.menuItems, func(item *types.MenuItem) bool {
|
|
|
|
return item.Key != nil
|
|
|
|
})
|
|
|
|
|
|
|
|
menuItemBindings := slices.Map(menuItemsWithKeys, func(item *types.MenuItem) *types.Binding {
|
|
|
|
return &types.Binding{
|
|
|
|
Key: item.Key,
|
|
|
|
Handler: func() error { return self.OnMenuPress(item) },
|
2022-03-19 10:12:58 +02:00
|
|
|
}
|
|
|
|
})
|
2022-03-27 08:15:17 +02:00
|
|
|
|
|
|
|
// appending because that means the menu item bindings have lower precedence.
|
|
|
|
// So if a basic binding is to escape from the menu, we want that to still be
|
|
|
|
// what happens when you press escape. This matters when we're showing the menu
|
|
|
|
// for all keybindings of say the files context.
|
|
|
|
return append(basicBindings, menuItemBindings...)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *MenuContext) OnMenuPress(selectedItem *types.MenuItem) error {
|
|
|
|
if err := self.c.PopContext(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-07-20 13:05:52 +02:00
|
|
|
if selectedItem == nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2022-03-27 08:15:17 +02:00
|
|
|
if err := selectedItem.OnPress(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2022-02-05 08:04:10 +02:00
|
|
|
}
|