#autoload

local -a config_hosts
local config
integer ind

# If users-hosts matches, we shouldn't complete anything else.
if [[ "$IPREFIX" == *@ ]]; then
  _combination -s '[:@]' my-accounts users-hosts "users=${IPREFIX/@}" hosts "$@" && return
else
  _combination -s '[:@]' my-accounts users-hosts \
    ${opt_args[-l]:+"users=${opt_args[-l]:q}"} hosts "$@" && return
fi
if (( ind = ${words[(I)-F]} )); then
  config=${~words[ind+1]} 2>/dev/null
else
  config="$HOME/.ssh/config"
fi
if [[ -r $config ]]; then
  local key line host
  local -a lines=("${(@f)$(<"$config")}") 2>/dev/null
  local -a match_args
  local idx=1
  while (( idx <= $#lines )); do
    IFS=$'=\t ' read -r key line <<<"${lines[idx]}"
    if [[ "$key" == ((#i)match) ]]; then
      match_args=( ${(Z.C.)line} )
      while [[ $#match_args -ge 2 ]]; do
	if [[ "${match_args[1]}" == (#i)(canonical|final|(|original)host) ]]; then
	  key="Host"
	  line="${match_args[2]//,/ }"
	  break
	fi
	shift 2 match_args
      done
    fi
    case "$key" in
    ((#i)include)
      lines[idx]=( "${(@f)$(cd $HOME/.ssh; cat ${(Z.C.)~line} 2>/dev/null)}" ) ;;
    ((#i)host(|name))
      for host in ${(Z.C.)line}; do
        [[ $host != *[*?%]* ]] && config_hosts+=( $host )
      done ;&
    (*) (( ++idx ));;
    esac
  done
  if (( ${#config_hosts} )); then
    _wanted hosts expl 'remote host name' \
      compadd -M 'm:{a-zA-Z}={A-Za-z} r:|.=* r:|=*' "$@" $config_hosts
  fi
fi
