glenux-archive/spf13--cobra
1
0
Fork 0
mirror of https://github.com/spf13/cobra synced 2025-05-05 12:57:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

Allow overriding Run/RunE of comp cmds

Browse source 
Signed-off-by: Marc Khouzam <marc.khouzam@montreal.ca>
This commit is contained in:
Marc Khouzam 2021-11-03 08:38:11 -04:00
parent 3ae090bcba
commit 9fa0b83ed0
1 changed files with 20 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
completionCmd.go
View file

@ -112,9 +112,11 @@ $ %[1]s %[2]s %[3]s > /usr/local/etc/bash_completion.d/%[1]s
You will need to start a new shell for this setup to take effect.`, You will need to start a new shell for this setup to take effect.`,
c.Root().Name(), CompletionCmd.Name(), BashCompletionCmd.Name()) c.Root().Name(), CompletionCmd.Name(), BashCompletionCmd.Name())
} }
if bash.RunE == nil && bash.Run == nil {
bash.RunE = func(cmd *Command, args []string) error { bash.RunE = func(cmd *Command, args []string) error {
return cmd.Root().GenBashCompletionV2(out, !noDesc) return cmd.Root().GenBashCompletionV2(out, !noDesc)
} }
}
bash.ResetFlags() // Tests can call this function multiple times in a row, so we must reset bash.ResetFlags() // Tests can call this function multiple times in a row, so we must reset
if haveNoDescFlag { if haveNoDescFlag {
@ -140,12 +142,14 @@ $ %[1]s %[2]s %[3]s > /usr/local/share/zsh/site-functions/_%[1]s
You will need to start a new shell for this setup to take effect.`, You will need to start a new shell for this setup to take effect.`,
c.Root().Name(), CompletionCmd.Name(), ZshCompletionCmd.Name()) c.Root().Name(), CompletionCmd.Name(), ZshCompletionCmd.Name())
} }
if zsh.RunE == nil && zsh.Run == nil {
zsh.RunE = func(cmd *Command, args []string) error { zsh.RunE = func(cmd *Command, args []string) error {
if noDesc { if noDesc {
return cmd.Root().GenZshCompletionNoDesc(out) return cmd.Root().GenZshCompletionNoDesc(out)
} }
return cmd.Root().GenZshCompletion(out) return cmd.Root().GenZshCompletion(out)
} }
}
zsh.ResetFlags() // Tests can call this function multiple times in a row, so we must reset zsh.ResetFlags() // Tests can call this function multiple times in a row, so we must reset
if haveNoDescFlag { if haveNoDescFlag {
@ -166,9 +170,11 @@ $ %[1]s %[2]s %[3]s > ~/.config/fish/completions/%[1]s.fish
You will need to start a new shell for this setup to take effect.`, You will need to start a new shell for this setup to take effect.`,
c.Root().Name(), CompletionCmd.Name(), FishCompletionCmd.Name()) c.Root().Name(), CompletionCmd.Name(), FishCompletionCmd.Name())
} }
if fish.RunE == nil && fish.Run == nil {
fish.RunE = func(cmd *Command, args []string) error { fish.RunE = func(cmd *Command, args []string) error {
return cmd.Root().GenFishCompletion(out, !noDesc) return cmd.Root().GenFishCompletion(out, !noDesc)
} }
}
fish.ResetFlags() // Tests can call this function multiple times in a row, so we must reset fish.ResetFlags() // Tests can call this function multiple times in a row, so we must reset
if haveNoDescFlag { if haveNoDescFlag {
@ -187,12 +193,14 @@ To load completions for every new session, add the output of the above command
to your powershell profile.`, to your powershell profile.`,
c.Root().Name(), CompletionCmd.Name(), PwshCompletionCmd.Name()) c.Root().Name(), CompletionCmd.Name(), PwshCompletionCmd.Name())
} }
if pwsh.RunE == nil && pwsh.Run == nil {
pwsh.RunE = func(cmd *Command, args []string) error { pwsh.RunE = func(cmd *Command, args []string) error {
if noDesc { if noDesc {
return cmd.Root().GenPowerShellCompletion(out) return cmd.Root().GenPowerShellCompletion(out)
} }
return cmd.Root().GenPowerShellCompletionWithDesc(out) return cmd.Root().GenPowerShellCompletionWithDesc(out)
} }
}
pwsh.ResetFlags() // Tests can call this function multiple times in a row, so we must reset pwsh.ResetFlags() // Tests can call this function multiple times in a row, so we must reset
if haveNoDescFlag { if haveNoDescFlag {