-#
-# Ceph - scalable distributed file system
-#
-# Copyright (C) 2011 Wido den Hollander <wido@widodh.nl>
-#
-# This is free software; you can redistribute it and/or
-# modify it under the terms of the GNU Lesser General Public
-# License version 2.1, as published by the Free Software
-# Foundation. See file COPYING.
-#
-_rbd_commands="" # lazy init
-_rbd_global_options="" # lazy init
-
-_rbd()
-{
- if [ "x${_rbd_commands}" == "x" ]; then
- local rbc="$(rbd bash-completion 2>/dev/null)"
- _rbd_commands="$(echo ${rbc} | sed -e 's/|-[^|]*//g' -e 's/||*/|/g')"
- _rbd_global_options="$(echo ${rbc} | sed -e 's/|[^-][^\|]*//g' -e 's/||*/|/g')"
- fi
-
- COMPREPLY=()
-
- local arg_count=${#COMP_WORDS[@]}
- local args=()
- local help_request="false"
- for (( i=1; i<arg_count; i++ )); do
- word="${COMP_WORDS[i]}"
- if [[ "x${word}" == "xhelp" && ${#args} == 0 ]]; then
- # treat help request as a special case
- help_request="true"
- continue
- elif [[ $(echo ${_rbd_global_options} | grep "|${word}|") ]]; then
- # skip flag option
- continue
- elif [[ "x${word:0:1}" == "x-" ]]; then
- # skip option with argument
- let i=$i+1
- continue
- elif [[ "x${word}" == "x" ]]; then
- # skip blank arguments
- continue
- fi
-
- args+=("${word}")
- done
-
- local cur="${COMP_WORDS[COMP_CWORD]}"
- local prev="${COMP_WORDS[COMP_CWORD-1]}"
-
- local options_exp=${_rbd_global_options}
- local command="${args[@]}"
- local valid_command="false"
- if [[ ${#args} != 0 && "${args[-1]}" != "${cur}" &&
- $(echo "${_rbd_commands}" | grep -c "|${command}|") == 1 ]]; then
- # combine global and command-specific options
- local rbd_command_options="$(rbd bash-completion ${args[@]} 2>/dev/null)"
- options_exp="${options_exp} ${rbd_command_options}"
- valid_command="true"
- fi
-
- if [[ "x${cur}" == "xhelp" ]]; then
- COMPREPLY=()
- elif [[ "${options_exp}}" =~ "|${prev} path|" ]]; then
- # perform path completion for path argument
- COMPREPLY=($(compgen -f ${cur}))
- elif [[ "${options_exp}}" =~ "|${prev} host|" ]]; then
- # perform host completion for host argument
- COMPREPLY=($(compgen -A hostname ${cur}))
- elif [[ "${help_request}" == "false" && ( "x${cur:0:1}" == "x-" ||
- ( "x${cur}" == "x" && "${valid_command}" == "true" ) ) ]]; then
- # all valid options for current command
- options="$(echo ${options_exp} | sed -e 's/||*/ /g' -r -e 's/ (arg|path|host)//g')"
- COMPREPLY=($(compgen -W "${options}" -- ${cur}))
- elif [[ "${valid_command}" == "false" ]]; then
- # search for valid command
- [[ "x${command}" != "x" && "x${cur}" == "x" ]] && command="${command} "
- COMPREPLY=($(echo ${_rbd_commands} | grep -o "|${command}[^ |]*" | \
- uniq | sed -e 's/|//g' | awk -F' ' '{print $(NF)}'))
- fi
-}
-
-complete -F _rbd rbd