.bash_completion

# This function checks whether we have a given program on the system.
# No need for bulky functions in memory if we don't.
#
have()
{
	unset -v have
	PATH=$PATH:/sbin:/usr/sbin:/usr/local/sbin type $1 &>/dev/null &&
		have="yes"
}

# AWS CLI completion
have aws && complete -C aws_completer aws

# Sudo completion for -u flag for users
have sudo && have _root_command && {
_sudo()
{
	local prev cur opts
	prev="${COMP_WORDS[COMP_CWORD-1]}"
	cur="${COMP_WORDS[COMP_CWORD]}"
	if [ $prev == "-u" ] ; then
		opts=$(getent passwd | cut -d: -f1)
		COMPREPLY=($(compgen -W "${opts}" "${cur}"))
	elif [[ "${COMP_WORDS[1]}" == "-u" ]] && [[ $COMP_CWORD -ge 3 ]] ; then
		for (( i=1 ; i<=($COMP_CWORD-2) ; i++)) ; do
			COMP_WORDS[i]=${COMP_WORDS[i+2]}
		done
		COMP_CWORD=$(($COMP_CWORD-2))
		_root_command $1 $2 $3
	else
		_root_command $1 $2 $3
	fi
}
complete -F _sudo $filenames sudo
}

# Console completion
have console && {
_console_hosts()
{
	local prev cur opts
	prev="${COMP_WORDS[COMP_CWORD-1]}"
	cur="${COMP_WORDS[COMP_CWORD]}"
	[ -r /etc/conserver.d ] && opts=$(awk '$1 ~ /^console/ {print $2}' /etc/conserver.d/conserver_*)
	COMPREPLY=($(compgen -W "${opts}" "${cur}"))
}
complete -F _console_hosts console
}

# Perforce completion
have p4 && {
_p4()
{
	local cur prev prev2 p4commands p4filetypes

	COMPREPLY=()
	cur=${COMP_WORDS[COMP_CWORD]}
	prev=${COMP_WORDS[COMP_CWORD-1]}

	# rename isn't really a command
	p4commands="$( p4 help commands | awk 'NF>3 {print $1}' )"
	p4filetypes="ctext cxtext ktext kxtext ltext tempobj ubinary \
		uresource uxbinary xbinary xltext xtempobj xtext \
		text binary resource"

	if [ $COMP_CWORD -eq 1 ]; then
		COMPREPLY=( $( compgen -W "$p4commands" -- $cur ) )
	elif [ $COMP_CWORD -eq 2 ]; then
		case "$prev" in
		help)
			COMPREPLY=( $( compgen -W "simple commands \
				environment filetypes jobview revisions \
				usage views $p4commands" -- $cur ) )
			;;
		admin)
			COMPREPLY=( $( compgen -W "checkpoint stop" -- $cur ) )
			;;
		*)
			;;
		esac
	elif [ $COMP_CWORD -gt 2 ]; then
		prev2=${COMP_WORDS[COMP_CWORD-2]}
		case "$prev" in
		-t)
			case "$prev2" in
			add|edit|reopen)
				COMPREPLY=( $( compgen -W "$p4filetypes" \
					       -- $cur) )
				;;
			*)
				;;
			esac
			;;
		*)
			;;
		esac
	fi

	return 0
}
complete -F _p4 -o default p4 g4
}