Properly handling spaces and quotes in bash completion
_foo ()
{
words="bar one"$'\n'"bar two"
COMPREPLY=()
cur=${COMP_WORDS[COMP_CWORD]}
prev=${COMP_WORDS[COMP_CWORD-1]}
cur=${cur//\./\\\.}
local IFS=$'\n'
COMPREPLY=( $( grep -i "^$cur" <( echo "$words" ) | sed -e 's/ /\\ /g' ) )
return 0
}
complete -o bashdefault -o default -o nospace -F _foo words
The question is rather loaded but this answer attempts to explain each aspect:
- How to handle spaces with
COMPREPLY
. - How does
ls
do it.
There're also people reaching this question wanting to know how to implement the completion function in general. So:
- How how do I implement the completion function and correctly set
COMPREPLY
?
How does ls
do it
Moreover, why does it behave differently to when I set COMPREPLY
?
Back in '12 (before I updated this answer), I was in a similar situation and searched high and low for the answer to this discrepancy myself. Here's the answer I came up with.
ls
, or rather, the default completion routine does it using the -o filenames
functionality. This option performs: filename-specific processing (like adding a slash to directory names or suppressing trailing spaces.
To demonstrate:
$ foo () { COMPREPLY=("bar one" "bar two"); }
$ complete -o filenames -F foo words
$ words ░
Tab
$ words bar\ ░ # Ex.1: notice the space is completed escaped
TabTab
bar one bar two # Ex.2: notice the spaces are displayed unescaped
$ words bar\ ░
Immediately there are two points I want to make clear to avoid any confusion:
First of all, your completion function cannot be implemented simply by setting
COMPREPLY
to an array of your word list! The example above is hard-coded to return candidates starting with b-a-r just to show what happens when TabTab is pressed. (Don't worry, we'll get to a more general implementation shortly.)Second, the above format for
COMPREPLY
only works because-o filenames
is specified. For an explanation of how to setCOMPREPLY
when not using-o filenames
, look no further than the next heading.
Also note, there's a downside of using -o filenames
: If there's a directory lying about with the same name as the matching word, the completed word automatically gets an arbitrary slash attached to the end. (e.g. bar\ one/
)
How to handle spaces with COMPREPLY
without using -o filenames
Long story short, it needs to be escaped.
In contrast to the above -o filenames
demo:
$ foo () { COMPREPLY=("bar\ one" "bar\ two"); } # Notice the blackslashes I've added
$ complete -F foo words # Notice the lack of -o filenames
$ words ░
Tab
$ words bar\ ░ # Same as -o filenames, space is completed escaped
TabTab
bar\ one bar\ two # Unlike -o filenames, notice the spaces are displayed escaped
$ words bar\ ░
How do I actually implement a completion function?
Implementing a completion functions involves:
- Representing your word list.
- Filtering your word list to just candidates for the current word.
- Setting
COMPREPLY
correctly.
I'm not going to assume to know all the complex requirements there can be for 1 and 2 and the following is only a very basic implementation. I'm providing an explanation for each part so one can mix-and-match to fit their own requirements.
foo() {
# Get the currently completing word
local CWORD=${COMP_WORDS[COMP_CWORD]}
# This is our word list (in a bash array for convenience)
local WORD_LIST=(foo 'bar one' 'bar two')
# Commands below depend on this IFS
local IFS=$'\n'
# Filter our candidates
CANDIDATES=($(compgen -W "${WORD_LIST[*]}" -- "$CWORD"))
# Correctly set our candidates to COMPREPLY
if [ ${#CANDIDATES[*]} -eq 0 ]; then
COMPREPLY=()
else
COMPREPLY=($(printf '%q\n' "${CANDIDATES[@]}"))
fi
}
complete -F foo words
In this example, we use compgen
to filter our words. (It's provided by bash for this exact purpose.) One could use any solution they like but I'd advise against using grep
-like programs simply because of the complexities of escaping regex.
compgen
takes the word list with the -W
argument and returns the filtered result with one word per line. Since our words can contain spaces, we set IFS=$'\n'
beforehand in order to only count newlines as element delimiters when putting the result into our array with the CANDIDATES=(...)
syntax.
Another point of note is what we're passing for the -W
argument. This argument takes an IFS
delimited word list. Again, our words contain spaces so this too requires IFS=$'\n'
to prevent our words being broken up.
Incidentally, "${WORD_LIST[*]}"
expands with elements also delimited with what we've set for IFS
and is exactly what we need.
In the example above I chose to define WORD_LIST
literally in code.
One could also initialize the array from an external source such as a file. Just make sure to move IFS=$'\n'
beforehand if words are going to be line-delimited such as in the original question:
local IFS=$'\n'
local WORD_LIST=($(cat /path/to/words.dat))`
Finally, we set COMPREPLY
making sure to escape the likes of spaces. Escaping is quite complicated but thankfully printf
's %q
format performs all the necessary escaping we need and that's what we use to expand CANDIDATES
. (Note we're telling printf
to put \n
after each element because that's what we've set IFS
to.)
Those observant may spot this form for COMPREPLY
only applies if -o filenames
is not used. No escaping is necessary if it is and COMPREPLY
may be set to the same contents as CANDIDATES
with COMPREPLY=("$CANDIDATES[@]")
.
Extra care should be taken when expansions may be performed on empty arrays as this can lead to unexpected results. The example above handles this by branching when the length of CANDIDATES
is zero.
This not too elegant postprocessing solution seems to work for me (GNU bash, version 3.1.17(6)-release (i686-pc-cygwin)). (Unless I didn't test some border case as usual :))
Don't need to eval things, there are only 2 kinds of quotes.
Since compgen doesn't want to escape spaces for us, we will escape them ourselves (only if word didn't start with a quote). This has a side effect of full list (on double tab) having escaped values as well. Not sure if that's good or not, since ls doesn't do it...
EDIT: Fixed to handle single and double qoutes inside the words. Essentially we have to pass 3 unescapings :). First for grep, second for compgen, and last for words command itself when autocompletion is done.
_find_words()
{
search=$(eval echo "$cur" 2>/dev/null || eval echo "$cur'" 2>/dev/null || eval echo "$cur\"" 2>/dev/null || "")
grep -- "^$search" words.dat | sed -e "{" -e 's#\\#\\\\#g' -e "s#'#\\\'#g" -e 's#"#\\\"#g' -e "}"
}
_words_complete()
{
local IFS=$'\n'
COMPREPLY=()
local cur="${COMP_WORDS[COMP_CWORD]}"
COMPREPLY=( $( compgen -W "$(_find_words)" -- "$cur" ) )
local escaped_single_qoute="'\''"
local i=0
for entry in ${COMPREPLY[*]}
do
if [[ "${cur:0:1}" == "'" ]]
then
# started with single quote, escaping only other single quotes
# [']bla'bla"bla\bla bla --> [']bla'\''bla"bla\bla bla
COMPREPLY[$i]="${entry//\'/${escaped_single_qoute}}"
elif [[ "${cur:0:1}" == "\"" ]]
then
# started with double quote, escaping all double quotes and all backslashes
# ["]bla'bla"bla\bla bla --> ["]bla'bla\"bla\\bla bla
entry="${entry//\\/\\\\}"
COMPREPLY[$i]="${entry//\"/\\\"}"
else
# no quotes in front, escaping _everything_
# [ ]bla'bla"bla\bla bla --> [ ]bla\'bla\"bla\\bla\ bla
entry="${entry//\\/\\\\}"
entry="${entry//\'/\'}"
entry="${entry//\"/\\\"}"
COMPREPLY[$i]="${entry// /\\ }"
fi
(( i++ ))
done
}