# -*- mode: shell-script; sh-basic-offset: 8; indent-tabs-mode: t -*-
# ex: ts=8 sw=8 noet filetype=sh
#
# ssh(1) completion
#
have ssh && {

_ssh_options() {
	COMPREPLY=( $( compgen -W 'AddressFamily BatchMode BindAddress \
		ChallengeResponseAuthentication CheckHostIP Cipher Ciphers \
		ClearAllForwardings Compression CompressionLevel \
		ConnectionAttempts ConnectTimeout ControlMaster ControlPath \
		DynamicForward EscapeChar ExitOnForwardFailure ForwardAgent \
		ForwardX11 ForwardX11Trusted GatewayPorts GlobalKnownHostsFile \
		GSSAPIAuthentication GSSAPIDelegateCredentials HashKnownHosts \
		Host HostbasedAuthentication HostKeyAlgorithms HostKeyAlias \
		HostName IdentityFile IdentitiesOnly KbdInteractiveDevices \
		LocalCommand LocalForward LogLevel MACs \
		NoHostAuthenticationForLocalhost NumberOfPasswordPrompts \
		PasswordAuthentication PermitLocalCommand Port \
		PreferredAuthentications Protocol ProxyCommand \
		PubkeyAuthentication RekeyLimit RemoteForward \
		RhostsRSAAuthentication RSAAuthentication SendEnv \
		ServerAliveInterval ServerAliveCountMax SmartcardDevice \
		StrictHostKeyChecking TCPKeepAlive Tunnel TunnelDevice \
		UsePrivilegedPort User UserKnownHostsFile VerifyHostKeyDNS \
		VisualHostKey XAuthLocation' -- $cur ) )
}

_ssh()
{
	local cur prev
	local optconfigfile
	local -a config

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

	case "$prev" in
		-@(F|i|S))
			_filedir
			return 0
			;;
		-c)
			COMPREPLY=( $( compgen -W '3des-cbc aes128-cbc \
				aes192-cbc aes256-cbc aes128-ctr aes192-ctr \
				aes256-ctr arcfour128 arcfour256 arcfour \
				blowfish-cbc cast128-cbc' -- $cur ) )
			return 0
			;;
		-c)
			COMPREPLY=( $( compgen -W 'hmac-md5 hmac-sha1 \
				umac-64@openssh.com hmac-ripemd160 \
				hmac-sha1-96 hmac-md5-96' -- $cur ) )
			return 0
			;;
		-l)
			COMPREPLY=( $( compgen -u -- $cur ) )
			return 0
			;;
		-o)
			_ssh_options
			return 0
			;;
		-w)
			_available_interfaces
			return 0
			;;
		-b)
			COMPREPLY=( $( compgen -W "$(/sbin/ifconfig | \
				awk '/adr:/ {print $2}' | \
				awk -F: '{print $2}' )" -- $cur ) )
			return 0
			;;
	esac

	if [[ "$cur" == -* ]]; then
		COMPREPLY=( $( compgen -W '-1 -2 -4 -6 -A -a -C -f -g -K -k -M \
			-N -n -q -s -T -t -V -v -X -v -Y -y -b -b -c -D -e -F \
			-i -L -l -m -O -o -p -R -S -w' -- $cur ) )
	else
		# Search COMP_WORDS for '-F configfile' argument
		set -- "${COMP_WORDS[@]}"
		while [ $# -gt 0 ]; do
			if [ "${1:0:2}" = -F ]; then
				if [ ${#1} -gt 2 ]; then
					optconfigfile="$(dequote "$1")"
				else
					shift
					[ "$1" ] && optconfigfile="$(dequote "-F$1")"
				fi
				break
			fi
			shift
		done
		_known_hosts_real -a $optconfigfile "$cur"
		if [ $COMP_CWORD -ne 1 ]; then
			COMPREPLY=( "${COMPREPLY[@]}" $( compgen -c -- $cur ) )
		fi
	fi

	return 0
}
shopt -u hostcomplete && complete -F _ssh ssh slogin autossh

# sftp(1) completion
#
_sftp()
{
	local cur prev
	local optconfigfile

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

	case "$prev" in
		-@(b|F|P))
			_filedirs
			return 0
			;;
		-o)
			_ssh_options
			return 0
			;;
	esac

	if [[ "$cur" == -* ]]; then
		COMPREPLY=( $( compgen -W '-1 -C -v -B -b -F -o -P -R -S -s' \
			-- $cur ) )
	else
		# Search COMP_WORDS for '-F configfile' argument
		set -- "${COMP_WORDS[@]}"
		while [ $# -gt 0 ]; do
			if [ "${1:0:2}" = -F ]; then
				if [ ${#1} -gt 2 ]; then
					optconfigfile="$(dequote "$1")"
				else
					shift
					[ "$1" ] && optconfigfile="$(dequote "-F$1")"
				fi
				break
			fi
			shift
		done
		_known_hosts_real -a $optconfigfile "$cur"
	fi

	return 0
}
shopt -u hostcomplete && complete -F _sftp sftp


# scp(1) completion
#
_scp()
{
	local cur userhost path
	local optconfigfile

	COMPREPLY=()
	cur=`_get_cword ":"`

	_expand || return 0

	if [[ "$cur" == *:* ]]; then
		local IFS=$'\t\n'
		# remove backslash escape from :
		cur=${cur/\\:/:}
		userhost=${cur%%?(\\):*}
		path=${cur#*:}
		# unescape spaces
		path=${path//\\\\\\\\ / }
		if [ -z "$path" ]; then
			# default to home dir of specified user on remote host
			path=$(ssh -o 'Batchmode yes' $userhost pwd 2>/dev/null)
		fi
		# escape spaces; remove executables, aliases, pipes and sockets;
		# add space at end of file names
		COMPREPLY=( $( ssh -o 'Batchmode yes' $userhost \
			       command ls -aF1d "$path*" 2>/dev/null | \
			       sed -e "s/[][(){}<>\",:;^&\!$=?\`|\\ ']/\\\\\\\\\\\\&/g" \
				   -e 's/[*@|=]$//g' -e 's/[^\/]$/& /g' ) )
		return 0
	fi

	# Search COMP_WORDS for '-F configfile' argument
	set -- "${COMP_WORDS[@]}"
	while [ $# -gt 0 ]; do
		if [ "${1:0:2}" = -F ]; then
			if [ ${#1} -gt 2 ]; then
				optconfigfile="$(dequote "$1")"
			else
				shift
				[ "$1" ] && optconfigfile="$(dequote "-F$1")"
			fi
			break
		fi
		shift
	done

	[[ "$cur" == */* ]] || _known_hosts_real -c -a $optconfigfile "$cur"

	# This approach is used instead of _filedir to get a space appended
	# after local file/dir completions, and $nospace retained for others.
	local IFS=$'\t\n'
	COMPREPLY=( "${COMPREPLY[@]}" $( command ls -aF1d $cur* \
		    2>/dev/null | sed \
		    -e "s/[][(){}<>\",:;^&\!$=?\`|\\ ']/\\\\&/g" \
		    -e 's/[*@|=]$//g' -e 's/[^\/]$/& /g' ) )

	return 0
}
complete -F _scp $nospace scp

# ssh-copy-id(1) completion
#
_ssh_copy_id() {
	local cur prev

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

	case "$prev" in
		-i)
			_filedir
			return 0
			;;
	esac

	if [[ "$cur" == -* ]]; then
		COMPREPLY=( $( compgen -W '-i' -- $cur ) )
	else
		_known_hosts_real -a "$cur"
	fi

	return 0
}
complete -F _ssh_copy_id $filenames ssh-copy-id
}
