mirror of
https://github.com/spf13/cobra
synced 2024-06-18 09:49:09 +00:00
f64bfa1e08
The zsh completion script output by cobra is a stub completion function which replaces itself with the actual completion function. This technique enables cobra to define helper functions without splitting the completion script into multiple files. However, the current implementation forgets to call the actual completion function at the end of the stub function, meaning that completion won't work the first time it's invoked in a shell session. This commit is a fix for this problem.
241 lines
8.1 KiB
Go
241 lines
8.1 KiB
Go
package cobra
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
// GenZshCompletionFile generates zsh completion file including descriptions.
|
|
func (c *Command) GenZshCompletionFile(filename string) error {
|
|
return c.genZshCompletionFile(filename, true)
|
|
}
|
|
|
|
// GenZshCompletion generates zsh completion file including descriptions
|
|
// and writes it to the passed writer.
|
|
func (c *Command) GenZshCompletion(w io.Writer) error {
|
|
return c.genZshCompletion(w, true)
|
|
}
|
|
|
|
// GenZshCompletionFileNoDesc generates zsh completion file without descriptions.
|
|
func (c *Command) GenZshCompletionFileNoDesc(filename string) error {
|
|
return c.genZshCompletionFile(filename, false)
|
|
}
|
|
|
|
// GenZshCompletionNoDesc generates zsh completion file without descriptions
|
|
// and writes it to the passed writer.
|
|
func (c *Command) GenZshCompletionNoDesc(w io.Writer) error {
|
|
return c.genZshCompletion(w, false)
|
|
}
|
|
|
|
// MarkZshCompPositionalArgumentFile only worked for zsh and its behavior was
|
|
// not consistent with Bash completion. It has therefore been disabled.
|
|
// Instead, when no other completion is specified, file completion is done by
|
|
// default for every argument. One can disable file completion on a per-argument
|
|
// basis by using ValidArgsFunction and ShellCompDirectiveNoFileComp.
|
|
// To achieve file extension filtering, one can use ValidArgsFunction and
|
|
// ShellCompDirectiveFilterFileExt.
|
|
//
|
|
// Deprecated
|
|
func (c *Command) MarkZshCompPositionalArgumentFile(argPosition int, patterns ...string) error {
|
|
return nil
|
|
}
|
|
|
|
// MarkZshCompPositionalArgumentWords only worked for zsh. It has therefore
|
|
// been disabled.
|
|
// To achieve the same behavior across all shells, one can use
|
|
// ValidArgs (for the first argument only) or ValidArgsFunction for
|
|
// any argument (can include the first one also).
|
|
//
|
|
// Deprecated
|
|
func (c *Command) MarkZshCompPositionalArgumentWords(argPosition int, words ...string) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *Command) genZshCompletionFile(filename string, includeDesc bool) error {
|
|
outFile, err := os.Create(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer outFile.Close()
|
|
|
|
return c.genZshCompletion(outFile, includeDesc)
|
|
}
|
|
|
|
func (c *Command) genZshCompletion(w io.Writer, includeDesc bool) error {
|
|
buf := new(bytes.Buffer)
|
|
genZshComp(buf, c.Name(), includeDesc)
|
|
_, err := buf.WriteTo(w)
|
|
return err
|
|
}
|
|
|
|
func genZshComp(buf *bytes.Buffer, name string, includeDesc bool) {
|
|
compCmd := ShellCompRequestCmd
|
|
if !includeDesc {
|
|
compCmd = ShellCompNoDescRequestCmd
|
|
}
|
|
buf.WriteString(fmt.Sprintf(`#compdef _%[1]s %[1]s
|
|
|
|
# zsh completion for %-36[1]s -*- shell-script -*-
|
|
|
|
__%[1]s_debug()
|
|
{
|
|
local file="$BASH_COMP_DEBUG_FILE"
|
|
if [[ -n ${file} ]]; then
|
|
echo "$*" >> "${file}"
|
|
fi
|
|
}
|
|
|
|
_%[1]s()
|
|
{
|
|
local shellCompDirectiveError=%[3]d
|
|
local shellCompDirectiveNoSpace=%[4]d
|
|
local shellCompDirectiveNoFileComp=%[5]d
|
|
local shellCompDirectiveFilterFileExt=%[6]d
|
|
local shellCompDirectiveFilterDirs=%[7]d
|
|
|
|
local lastParam lastChar flagPrefix requestComp out directive compCount comp lastComp
|
|
local -a completions
|
|
|
|
__%[1]s_debug "\n========= starting completion logic =========="
|
|
__%[1]s_debug "CURRENT: ${CURRENT}, words[*]: ${words[*]}"
|
|
|
|
# The user could have moved the cursor backwards on the command-line.
|
|
# We need to trigger completion from the $CURRENT location, so we need
|
|
# to truncate the command-line ($words) up to the $CURRENT location.
|
|
# (We cannot use $CURSOR as its value does not work when a command is an alias.)
|
|
words=("${=words[1,CURRENT]}")
|
|
__%[1]s_debug "Truncated words[*]: ${words[*]},"
|
|
|
|
lastParam=${words[-1]}
|
|
lastChar=${lastParam[-1]}
|
|
__%[1]s_debug "lastParam: ${lastParam}, lastChar: ${lastChar}"
|
|
|
|
# For zsh, when completing a flag with an = (e.g., %[1]s -n=<TAB>)
|
|
# completions must be prefixed with the flag
|
|
setopt local_options BASH_REMATCH
|
|
if [[ "${lastParam}" =~ '-.*=' ]]; then
|
|
# We are dealing with a flag with an =
|
|
flagPrefix="-P ${BASH_REMATCH}"
|
|
fi
|
|
|
|
# Prepare the command to obtain completions
|
|
requestComp="${words[1]} %[2]s ${words[2,-1]}"
|
|
if [ "${lastChar}" = "" ]; then
|
|
# If the last parameter is complete (there is a space following it)
|
|
# We add an extra empty parameter so we can indicate this to the go completion code.
|
|
__%[1]s_debug "Adding extra empty parameter"
|
|
requestComp="${requestComp} \"\""
|
|
fi
|
|
|
|
__%[1]s_debug "About to call: eval ${requestComp}"
|
|
|
|
# Use eval to handle any environment variables and such
|
|
out=$(eval ${requestComp} 2>/dev/null)
|
|
__%[1]s_debug "completion output: ${out}"
|
|
|
|
# Extract the directive integer following a : from the last line
|
|
local lastLine
|
|
while IFS='\n' read -r line; do
|
|
lastLine=${line}
|
|
done < <(printf "%%s\n" "${out[@]}")
|
|
__%[1]s_debug "last line: ${lastLine}"
|
|
|
|
if [ "${lastLine[1]}" = : ]; then
|
|
directive=${lastLine[2,-1]}
|
|
# Remove the directive including the : and the newline
|
|
local suffix
|
|
(( suffix=${#lastLine}+2))
|
|
out=${out[1,-$suffix]}
|
|
else
|
|
# There is no directive specified. Leave $out as is.
|
|
__%[1]s_debug "No directive found. Setting do default"
|
|
directive=0
|
|
fi
|
|
|
|
__%[1]s_debug "directive: ${directive}"
|
|
__%[1]s_debug "completions: ${out}"
|
|
__%[1]s_debug "flagPrefix: ${flagPrefix}"
|
|
|
|
if [ $((directive & shellCompDirectiveError)) -ne 0 ]; then
|
|
__%[1]s_debug "Completion received error. Ignoring completions."
|
|
return
|
|
fi
|
|
|
|
compCount=0
|
|
while IFS='\n' read -r comp; do
|
|
if [ -n "$comp" ]; then
|
|
# If requested, completions are returned with a description.
|
|
# The description is preceded by a TAB character.
|
|
# For zsh's _describe, we need to use a : instead of a TAB.
|
|
# We first need to escape any : as part of the completion itself.
|
|
comp=${comp//:/\\:}
|
|
|
|
local tab=$(printf '\t')
|
|
comp=${comp//$tab/:}
|
|
|
|
((compCount++))
|
|
__%[1]s_debug "Adding completion: ${comp}"
|
|
completions+=${comp}
|
|
lastComp=$comp
|
|
fi
|
|
done < <(printf "%%s\n" "${out[@]}")
|
|
|
|
if [ $((directive & shellCompDirectiveFilterFileExt)) -ne 0 ]; then
|
|
# File extension filtering
|
|
local filteringCmd
|
|
filteringCmd='_files'
|
|
for filter in ${completions[@]}; do
|
|
if [ ${filter[1]} != '*' ]; then
|
|
# zsh requires a glob pattern to do file filtering
|
|
filter="\*.$filter"
|
|
fi
|
|
filteringCmd+=" -g $filter"
|
|
done
|
|
filteringCmd+=" ${flagPrefix}"
|
|
|
|
__%[1]s_debug "File filtering command: $filteringCmd"
|
|
_arguments '*:filename:'"$filteringCmd"
|
|
elif [ $((directive & shellCompDirectiveFilterDirs)) -ne 0 ]; then
|
|
# File completion for directories only
|
|
local subDir
|
|
subdir="${completions[1]}"
|
|
if [ -n "$subdir" ]; then
|
|
__%[1]s_debug "Listing directories in $subdir"
|
|
pushd "${subdir}" >/dev/null 2>&1
|
|
else
|
|
__%[1]s_debug "Listing directories in ."
|
|
fi
|
|
|
|
_arguments '*:dirname:_files -/'" ${flagPrefix}"
|
|
if [ -n "$subdir" ]; then
|
|
popd >/dev/null 2>&1
|
|
fi
|
|
elif [ $((directive & shellCompDirectiveNoSpace)) -ne 0 ] && [ ${compCount} -eq 1 ]; then
|
|
__%[1]s_debug "Activating nospace."
|
|
# We can use compadd here as there is no description when
|
|
# there is only one completion.
|
|
compadd -S '' "${lastComp}"
|
|
elif [ ${compCount} -eq 0 ]; then
|
|
if [ $((directive & shellCompDirectiveNoFileComp)) -ne 0 ]; then
|
|
__%[1]s_debug "deactivating file completion"
|
|
else
|
|
# Perform file completion
|
|
__%[1]s_debug "activating file completion"
|
|
_arguments '*:filename:_files'" ${flagPrefix}"
|
|
fi
|
|
else
|
|
_describe "completions" completions $(echo $flagPrefix)
|
|
fi
|
|
}
|
|
|
|
# don't run the completion function when being source-ed or eval-ed
|
|
if [ "$funcstack[1]" = "_%[1]s" ]; then
|
|
_%[1]s
|
|
fi
|
|
`, name, compCmd,
|
|
ShellCompDirectiveError, ShellCompDirectiveNoSpace, ShellCompDirectiveNoFileComp,
|
|
ShellCompDirectiveFilterFileExt, ShellCompDirectiveFilterDirs))
|
|
}
|