package gui import ( "log" "strings" "github.com/jesseduffield/generics/slices" "github.com/jesseduffield/lazygit/pkg/gui/keybindings" "github.com/jesseduffield/lazygit/pkg/gui/presentation" "github.com/jesseduffield/lazygit/pkg/gui/types" "github.com/samber/lo" ) func (gui *Gui) getBindings(context types.Context) []*types.Binding { var bindingsGlobal, bindingsPanel, bindingsNavigation []*types.Binding bindings, _ := gui.GetInitialKeybindings() customBindings, err := gui.CustomCommandsClient.GetCustomCommandKeybindings() if err != nil { log.Fatal(err) } bindings = append(customBindings, bindings...) for _, binding := range bindings { if keybindings.GetKeyDisplay(binding.Key) != "" && binding.Description != "" { if len(binding.Contexts) == 0 && binding.ViewName == "" { bindingsGlobal = append(bindingsGlobal, binding) } else if binding.Tag == "navigation" { bindingsNavigation = append(bindingsNavigation, binding) } else if lo.Contains(binding.Contexts, string(context.GetKey())) { bindingsPanel = append(bindingsPanel, binding) } } } resultBindings := []*types.Binding{} resultBindings = append(resultBindings, uniqueBindings(bindingsPanel)...) // adding a separator between the panel-specific bindings and the other bindings resultBindings = append(resultBindings, &types.Binding{}) resultBindings = append(resultBindings, uniqueBindings(bindingsGlobal)...) resultBindings = append(resultBindings, uniqueBindings(bindingsNavigation)...) return resultBindings } // We shouldn't really need to do this. We should define alternative keys for the same // handler in the keybinding struct. func uniqueBindings(bindings []*types.Binding) []*types.Binding { return lo.UniqBy(bindings, func(binding *types.Binding) string { return binding.Description }) } func (gui *Gui) displayDescription(binding *types.Binding) string { if binding.OpensMenu { return presentation.OpensMenuStyle(binding.Description) } return binding.Description } func (gui *Gui) handleCreateOptionsMenu() error { context := gui.currentContext() bindings := gui.getBindings(context) menuItems := slices.Map(bindings, func(binding *types.Binding) *types.MenuItem { return &types.MenuItem{ Label: gui.displayDescription(binding), OnPress: func() error { if binding.Key == nil { return nil } return binding.Handler() }, Key: binding.Key, Tooltip: binding.Tooltip, } }) return gui.c.Menu(types.CreateMenuOptions{ Title: strings.Title(gui.c.Tr.LcMenu), Items: menuItems, HideCancel: true, }) }