Як відредагувати завершення команди для ssh на zsh?

Я хотів би встановити завершення команди на zsh для відображення імен хостів після введення

ssh [TAB]

вилучення імен з мого .ssh / config-файлу (а краще з відомих_хостів та / і т.д. / хостів та будь-якого іншого місця, яке має сенс) та подання одного єдиного списку.

Дещо з цього робить і зараз, але

1. він взагалі не використовує .ssh / config
2. спочатку потрібне ім'я користувача, навіть якщо використання .ssh / config робить введення імен користувачів непотрібними
3. він містить декілька списків (можливо, один із відомих_хостів та інший від / etc / hosts, але я цього не підтвердив)

Тому я хочу включати відомі імена користувачів, а також відомі імена хостів у (бажано, єдиний) список після введення ssh [TAB]

(Я заходжу сюди перед Google, тому що 1) це призведе до того, що відповідь зберігається тут, і 2) це, мабуть, більш ефективно. Якщо ніхто більше не відповість, я знайду відповідь.)

Ось відповідна частина з моєї .zshrc. Він не змінювався з 2002 року, тому я можу сьогодні написати його по-іншому, але він все ще працює для заповнення імен хостів ~/.ssh/configі ~/.ssh/known_hosts(якщо HashKnownHostsвимкнено - його не існувало в ті часи).

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

Функція, яка забезпечує sshзавершення, знаходиться в /usr/share/zsh/functions/Completion/Unix/_sshмоїй системі.

Також зверніться man zshcompsysдо документації (особливо зробіть пошук по "хосту", який відображається в декількох місцях, і "ssh", який з’являється в декількох місцях)

Можливо, додавання zstyleкоманди до вашого ~/.zshrcбуде робити те, що ви шукаєте, не змінюючи функції завершення.

Я роблю це, використовуючи список усіх хостів на даному домені за допомогою dig. Ви можете замінити функцію нижче будь-якою системою пошуку, включаючи файл хостів або статичний список:

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

Примітка: наведений вище код може не повністю замінити повну систему для команди ssh у вашій конфігурації. Якщо у вас є проблеми з цим, спробуйте змінити команду "ssh" на якусь іншу випадкову команду, наприклад "mycompletetest", і подивіться, чи завершення працює для цього.

Також зверніть увагу, що це передає зону dns при кожному завершенні! Якщо ви користуєтесь цим багато або на досить статичному домені, було б сенс зробити пошук і зберегти результат, то у своїй функції пошуку просто встановіть reply = zone_result.

Мені подобається зберігати хешований known_hostsфайл і краще не HashKnownHostsвимикатимуться. Я виявив, що насіння того, що має @Gilles, і того, що вже є в моїй історії, було досить ефективним для моїх потреб.

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

Також FWIW, це те, що я використовував для 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