Handle tilde expansion in cros_workon tab completion.

bash_completion

 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 # Add programmable completion to some Chromium OS build scripts
 
 
 # Echo a list of -- flags that the current command accepts. The
 # function assumes that the command supports shflags' --help flag.
 _flags() {
@@ skipping 116 matching lines @@
 
 # Complete the cros_workon <command> argument.
 #
 # TODO(petkov): We should probably extract the list of commands from
 # cros_workon --help, just like we do for flags (see _flag_complete).
 #
 # TODO(petkov): Currently, this assumes that the command is the first
 # argument. In practice, the command is the first non-flag
 # argument. I.e., this should be fixed to support something like
 # "./cros_workon --all list".
-_cros_workon_command_complete() {
+_complete_cros_workon_command() {
   [ ${COMP_CWORD} -eq 1 ] || return 1
   local command="${COMP_WORDS[1]}"
   COMPREPLY=($(compgen -W "start stop list iterate" -- "$command"))
   return 0
 }
 
+# Prints the full path to the cros_workon executable, handling tilde
+# expansion for the current user.
+_cros_workon_executable() {
+  local cros_workon="${COMP_WORDS[0]}"
+  if [[ "$cros_workon" == '~/'* ]]; then
+    cros_workon="$HOME/${cros_workon#'~/'}"
+  fi
+  echo "$cros_workon"
+}
+
 # Lists the workon (or live, if --all is passed in) ebuilds. Lists
 # both the full names (e.g., chromeos-base/metrics) as well as just
 # the ebuild names (e.g., metrics).
 _cros_workon_list() {
-  local cros_workon="${COMP_WORDS[0]}"
+  local cros_workon=$(_cros_workon_executable)
   ${cros_workon} list $1 | sed 's,\(.\+\)/\(.\+\),\1/\2 \2,'
 }
 
 # Completes the current cros_workon argument assuming it's a
 # package/ebuild name.
-_cros_workon_package_complete() {
+_complete_cros_workon_package() {
   [ ${COMP_CWORD} -gt 1 ] || return 1
   local package="${COMP_WORDS[COMP_CWORD]}"
   local command="${COMP_WORDS[1]}"
   # If "start", complete based on all workon packages.
   if [[ ${command} == "start" ]]; then
     COMPREPLY=($(compgen -W "$(_cros_workon_list --all)" -- "$package"))
     return 0
   fi
   # If "stop" or "iterate", complete based on all live packages.
   if [[ ${command} == "stop" ]] || [[ ${command} == "iterate" ]]; then
     COMPREPLY=($(compgen -W "$(_cros_workon_list)" -- "$package"))
     return 0
   fi
   return 1
 }
 
 # Complete the cros_workon arguments.
 _cros_workon() {
   COMPREPLY=()
   _flag_complete && return 0
   _complete_board_sysroot_flag && return 0
-  _cros_workon_command_complete && return 0
-  _cros_workon_package_complete && return 0
+  _complete_cros_workon_command && return 0
+  _complete_cros_workon_package && return 0
   return 0
 }
 
 complete -o bashdefault -o default -F _board_sysroot \
   build_autotest.sh \
   build_image \
   build_packages \
   image_to_usb.sh \
   mod_image_for_test.sh
 complete -o bashdefault -o default -F _board_overlay setup_board
 complete -o bashdefault -o default -o nospace -F _autotest_complete autotest
-complete -o bashdefault -o default -F _cros_workon cros_workon
+complete -F _cros_workon cros_workon
 
 ###  Local Variables:
 ###  mode: shell-script
 ###  End: