feat: support custom keybinds in custom command prompt menus

Author: HerrNaN

docs/Custom_Command_Keybindings.md

@@ -192,6 +192,7 @@ The permitted option fields are:
 | name | The first part of the label | no |
 | description | The second part of the label | no |
 | value | the value that will be used in the command | yes |
+| key | Keybinding to invoke this menu option without needing to navigate to it. Can be a single letter or one of the values from [here](https://github.com/jesseduffield/lazygit/blob/master/docs/keybindings/Custom_Keybindings.md) | no |
 
 If an option has no name the value will be displayed to the user in place of the name, so you're allowed to only include the value like so:
 
@@ -233,6 +234,34 @@ customCommands:
             description: 'branch for a release'
 ```
 
+Here's an example of supplying keybindings for menu options:
+
+```yml
+customCommands:
+  - key: 'a'
+    command: 'echo {{.Form.BranchType | quote}}'
+    context: 'commits'
+    prompts:
+      - type: 'menu'
+        title: 'What kind of branch is it?'
+        key: 'BranchType'
+        options:
+          - value: 'feature'
+            name: 'feature branch'
+            description: 'branch based off develop'
+            key: 'f'
+          - value: 'hotfix'
+            name: 'hotfix branch'
+            description: 'branch based off main for fast bug fixes'
+            key: 'h'
+          - value: 'release'
+            name: 'release branch'
+            description: 'branch for a release'
+            key: 'r'
+```
+
+In this example, pressing 'f', 'h', or 'r' will directly select the corresponding option without needing to navigate to it first.
+
 ### Menu-from-command
 
 | _field_           | _description_                                                                                  | _required_ |

pkg/config/user_config.go

@@ -733,6 +733,8 @@ type CustomCommandMenuOption struct {
 	Description string `yaml:"description"`
 	// The value that will be used in the command
 	Value string `yaml:"value" jsonschema:"example=feature,minLength=1"`
+	// Keybinding to invoke this menu option without needing to navigate to it
+	Key string `yaml:"key"`
 }
 
 type CustomIconsConfig struct {

pkg/gui/context/menu_context.go

@@ -193,9 +193,8 @@ func (self *MenuViewModel) GetNonModelItems() []*NonModelItem {
 }
 
 func (self *MenuContext) GetKeybindings(opts types.KeybindingsOpts) []*types.Binding {
-	allBindings := self.ListContextTrait.GetKeybindings(opts)
+	basicBindings := self.ListContextTrait.GetKeybindings(opts)
 
-	// Define essential keys that should always work (select, confirm, escape, and prev/next item navigation)
 	essentialKeys := []types.Key{
 		opts.GetKey(opts.Config.Universal.Select),
 		opts.GetKey(opts.Config.Universal.ConfirmMenu),
@@ -206,15 +205,13 @@ func (self *MenuContext) GetKeybindings(opts types.KeybindingsOpts) []*types.Bin
 		opts.GetKey(opts.Config.Universal.NextItemAlt),
 	}
 
-	// Group bindings by whether they're essential or not
-	bindingsByPriority := lo.GroupBy(allBindings, func(binding *types.Binding) bool {
+	bindingsByPriority := lo.GroupBy(basicBindings, func(binding *types.Binding) bool {
 		return lo.Contains(essentialKeys, binding.Key)
 	})
 
 	essentialBindings := bindingsByPriority[true]
 	navigationBindings := bindingsByPriority[false]
 
-	// Create bindings for menu items
 	menuItemBindings := lo.FilterMap(self.menuItems, func(item *types.MenuItem, _ int) (*types.Binding, bool) {
 		if item.Key == nil {
 			return nil, false

pkg/gui/services/custom_commands/handler_creator.go

@@ -9,6 +9,7 @@ import (
 	"github.com/jesseduffield/gocui"
 	"github.com/jesseduffield/lazygit/pkg/config"
 	"github.com/jesseduffield/lazygit/pkg/gui/controllers/helpers"
+	"github.com/jesseduffield/lazygit/pkg/gui/keybindings"
 	"github.com/jesseduffield/lazygit/pkg/gui/style"
 	"github.com/jesseduffield/lazygit/pkg/gui/types"
 	"github.com/jesseduffield/lazygit/pkg/utils"
@@ -203,6 +204,7 @@ func (self *HandlerCreator) menuPrompt(prompt *config.CustomCommandPrompt, wrapp
 			OnPress: func() error {
 				return wrappedF(option.Value)
 			},
+			Key: keybindings.GetKey(option.Key),
 		}
 	})
 

pkg/gui/services/custom_commands/resolver.go

@@ -108,6 +108,7 @@ func (self *Resolver) resolveMenuOption(option *config.CustomCommandMenuOption,
 		Name:        name,
 		Description: description,
 		Value:       value,
+		Key:         option.Key,
 	}, nil
 }
 

pkg/integration/tests/custom_commands/menu_prompt_with_keys.go

@@ -0,0 +1,68 @@
+package custom_commands
+
+import (
+	"github.com/jesseduffield/lazygit/pkg/config"
+	. "github.com/jesseduffield/lazygit/pkg/integration/components"
+)
+
+var MenuPromptWithKeys = NewIntegrationTest(NewIntegrationTestArgs{
+	Description:  "Using a custom command with menu options that have keybindings",
+	ExtraCmdArgs: []string{},
+	Skip:         false,
+	SetupRepo: func(shell *Shell) {
+		shell.EmptyCommit("initial commit")
+	},
+	SetupConfig: func(cfg *config.AppConfig) {
+		cfg.GetUserConfig().CustomCommands = []config.CustomCommand{
+			{
+				Key:     "a",
+				Context: "files",
+				Command: `echo {{.Form.Choice | quote}} > result.txt`,
+				Prompts: []config.CustomCommandPrompt{
+					{
+						Key:   "Choice",
+						Type:  "menu",
+						Title: "Choose an option",
+						Options: []config.CustomCommandMenuOption{
+							{
+								Name:        "first",
+								Description: "First option",
+								Value:       "FIRST",
+								Key:         "1",
+							},
+							{
+								Name:        "second",
+								Description: "Second option",
+								Value:       "SECOND",
+								Key:         "H",
+							},
+							{
+								Name:        "third",
+								Description: "Third option",
+								Value:       "THIRD",
+								Key:         "3",
+							},
+						},
+					},
+				},
+			},
+		}
+	},
+	Run: func(t *TestDriver, keys config.KeybindingConfig) {
+		t.Views().Files().
+			IsFocused().
+			Press("a")
+
+		// Verify the menu appears
+		t.ExpectPopup().Menu().
+			Title(Equals("Choose an option"))
+
+		// Press 'H' to directly select the second option via its keybinding
+		// 'H' is normally a navigation key (ScrollLeft), so this tests that menu item
+		// keybindings have proper precedence over non-essential navigation keys
+		t.Views().Menu().Press("H")
+
+		// After pressing 'H', the menu should close and the command should execute
+		t.FileSystem().FileContent("result.txt", Equals("SECOND\n"))
+	},
+})

pkg/integration/tests/test_list.go

@@ -174,6 +174,7 @@ var tests = []*components.IntegrationTest{
 	custom_commands.GlobalContext,
 	custom_commands.MenuFromCommand,
 	custom_commands.MenuFromCommandsOutput,
+	custom_commands.MenuPromptWithKeys,
 	custom_commands.MultipleContexts,
 	custom_commands.MultiplePrompts,
 	custom_commands.RunCommand,

schema-master/config.json

@@ -163,6 +163,10 @@
           "examples": [
             "feature"
           ]
+        },
+        "key": {
+          "type": "string",
+          "description": "Keybinding to invoke this menu option without needing to navigate to it. Can be a single letter or one of the values from https://github.com/jesseduffield/lazygit/blob/master/docs/keybindings/Custom_Keybindings.md"
         }
       },
       "additionalProperties": false,