export COMP_WORDBREAKS=${COMP_WORDBREAKS/\:/}
_sshcomplete() {
- if [ -f $HOME/.ssh/config ]; then
- COMPREPLY=($(compgen -W "`ruby -e"puts open('${HOME}/.ssh/config', 'r') { |f| f.readlines }.find_all { |l| l =~ /^Host/ }.inject([]) { |hosts, line| hosts << line[5..-1].split }.flatten.sort.uniq"`" -- ${COMP_WORDS[COMP_CWORD]}))
- return 0
+
+ # parse all defined hosts from .ssh/config
+ if [ -r $HOME/.ssh/config ]; then
+ COMPREPLY=($(compgen -W "$(grep ^Host $HOME/.ssh/config | awk '{print $2}' )" -- ${COMP_WORDS[COMP_CWORD]}))
fi
+
+ # parse all hosts found in .ssh/known_hosts
+ if [ -r $HOME/.ssh/known_hosts ]; then
+ if grep -v -q -e '^ ssh-rsa' $HOME/.ssh/known_hosts ; then
+ COMPREPLY=( $COMPREPLY $(compgen -W "$( awk '{print $1}' $HOME/.ssh/known_hosts | cut -d, -f 1 | sed -e 's/\[//g' | sed -e 's/\]//g' | cut -d: -f1 | grep -v ssh-rsa)" -- ${COMP_WORDS[COMP_CWORD]} ))
+ fi
+ fi
+
+ return 0
}
complete -o default -o nospace -F _sshcomplete ssh