Come modificare il completamento del comando per ssh su zsh?

Vorrei impostare il completamento del comando su zsh per visualizzare i nomi host dopo aver digitato

ssh [TAB]

prendendo i nomi dal mio file .ssh / config (e preferibilmente da known_hosts e / etc / hosts e da qualsiasi altra parte che abbia senso) e presentando un unico elenco.

Fa un po 'di questo attualmente, ma

1. non usa affatto .ssh / config
2. richiede prima un nome utente, anche se l'uso di .ssh / config rende superflua la digitazione dei nomi utente
3. presenta più elenchi (probabilmente uno da known_hosts e un altro da / etc / hosts, ma non l'ho verificato)

Quindi voglio includere nomi utente noti e nomi host noti nell'elenco (preferibilmente singolo) dopo aver digitato ssh [TAB]

(Vengo qui prima di Google perché 1) comporterà la memorizzazione della risposta qui e 2) probabilmente è più efficiente. Se nessun altro risponde, cercherò la risposta.)

Ecco la parte rilevante della mia .zshrc. Non è cambiato dal 2002, quindi potrei scriverlo diversamente oggi, ma funziona ancora per completare i nomi host da ~/.ssh/confige ~/.ssh/known_hosts(se HashKnownHostsè spento - non esisteva in quei giorni).

h=()
if [[ -r ~/.ssh/config ]]; then
  h=($h ${${${(@M)${(f)"$(cat ~/.ssh/config)"}:#Host *}#Host }:#*[*?]*})
fi
if [[ -r ~/.ssh/known_hosts ]]; then
  h=($h ${${${(f)"$(cat ~/.ssh/known_hosts{,2} || true)"}%%\ *}%%,*}) 2>/dev/null
fi
if [[ $#h -gt 0 ]]; then
  zstyle ':completion:*:ssh:*' hosts $h
  zstyle ':completion:*:slogin:*' hosts $h
fi

La funzione che fornisce il sshcompletamento si trova in /usr/share/zsh/functions/Completion/Unix/_sshsul mio sistema.

Consulta anche la man zshcompsysdocumentazione (in particolare esegui una ricerca su "host" che appare in più punti e "ssh" che appare in un paio di punti).

Potrebbe essere possibile che l'aggiunta di un zstylecomando al tuo ~/.zshrcfaccia quello che stai cercando senza dover modificare la funzione di completamento.

Lo faccio usando un elenco di tutti gli host su un determinato dominio usando dig. È possibile sostituire la funzione seguente con qualsiasi sistema di ricerca desiderato, incluso il file hosts o un elenco statico:

function complete_host_from_zone () {
    reply=(`dig axfr ouraynet.com @ns1.ouraynet.com | grep -e '^[a-z]' | cut -d\. -f1`)
}
compctl -x 'p[1]' -K complete_host_from_zone -- ssh

Nota: il codice sopra potrebbe non sostituire completamente il sistema completo per il comando ssh nella configurazione. Se hai problemi con esso, prova a cambiare il comando "ssh" in qualche altro comando casuale come "mycompletetest" e vedi se il completamento funziona per quello.

Nota anche che questo fa il trasferimento della zona DNS ad ogni completamento! Se lo usi molto o su un dominio abbastanza statico, sarebbe logico fare la ricerca e salvare il risultato, quindi nella tua funzione di ricerca basta impostare reply = zone_result.

Mi piace conservare il mio known_hostsfile con hash e preferirei non HashKnownHostsdisattivarlo. Avevo scoperto che seminare ciò che @Gilles ha con ciò che è già nella mia storia è stato abbastanza efficace per le mie esigenze.

h=($(echo $(history | awk '{print $4 " " $5 "\n"}' | grep 'ssh ' | awk '{print $2}' | sort -u)))
if [[ -r ~/.ssh/config ]]; then
  h=($h ${${${(@M)${(f)"$(cat ~/.ssh/config)"}:#Host *}#Host }:#*[*?]*})
fi
if [[ -r ~/.ssh/known_hosts ]]; then
   h=($h ${${${(f)"$(cat ~/.ssh/known_hosts{,2} || true)"}%%\ *}%%,*}) 2>/dev/null
fi
if [[ $#h -gt 0 ]]; then
  zstyle ':completion:*:ssh:*' hosts $h
  zstyle ':completion:*:slogin:*' hosts $h
fi

Inoltre, FWIW, questo è quello che ho usato per Bash:

# SSH Autocompletion
complete -W "
  $(echo $(grep '^\s*ssh ' ~/.bash_history | sort -u | sed 's/^ssh //' | awk '{print $1}'))
  $(echo $(history | awk '{print $2 " " $3}' | grep 'ssh ' | awk '{print $2}' | sort -u))
  $(sed 's/#.*//;' ~/.ssh/config | awk ' /^Host (.+)$/ {$1 = "";print tolower($0)}')
" ssh