# shellcheck shell=bash
# Bash completion support for ssh.

# Remove : and @ from COMP_WORDBREAKS to support user@host completion
export COMP_WORDBREAKS=${COMP_WORDBREAKS//[:@]/}

_sshcomplete() {
	local line CURRENT_PROMPT="${COMP_WORDS[COMP_CWORD]}"
	if [[ ${CURRENT_PROMPT} == *@* ]]; then
		local OPTIONS="-P ${CURRENT_PROMPT/@*/}@ -- ${CURRENT_PROMPT/*@/}"
	else
		local OPTIONS=" -- ${CURRENT_PROMPT}"
	fi

	# parse all defined hosts from .ssh/config and files included there
	for fl in "$HOME/.ssh/config" \
		$(grep "^\s*Include" "$HOME/.ssh/config" \
			| awk '{for (i=2; i<=NF; i++) print $i}' \
			| sed -Ee "s|^([^/~])|$HOME/.ssh/\1|" -e "s|^~/|$HOME/|"); do
		if [ -r "$fl" ]; then
			#shellcheck disable=SC2086
			while IFS='' read -r line; do COMPREPLY+=("$line"); done < <(compgen -W "$(grep -i ^Host "$fl" | grep -v '[*!]' | awk '{for (i=2; i<=NF; i++) print $i}')" ${OPTIONS})
		fi
	done

	# parse all hosts found in .ssh/known_hosts
	if [ -r "$HOME/.ssh/known_hosts" ] && grep -v -q -e '^ ssh-rsa' "$HOME/.ssh/known_hosts"; then
		#shellcheck disable=SC2086
		while IFS='' read -r line; do COMPREPLY+=("$line"); done < <(compgen -W "$(awk '{print $1}' "$HOME/.ssh/known_hosts" | grep -v ^\| | cut -d, -f 1 | sed -e 's/\[//g' | sed -e 's/\]//g' | cut -d: -f1 | grep -v ssh-rsa)" ${OPTIONS})
	fi

	# parse hosts defined in /etc/hosts
	if [ -r /etc/hosts ]; then
		#shellcheck disable=SC2086
		while IFS='' read -r line; do COMPREPLY+=("$line"); done < <(compgen -W "$(grep -v '^[[:space:]]*$' /etc/hosts | grep -v '^#' | awk '{for (i=2; i<=NF; i++) print $i}')" ${OPTIONS})
	fi

	return 0
}

complete -o default -o nospace -F _sshcomplete ssh scp slogin sftp
