วิธีแก้ไขคำสั่งเสร็จสิ้นสำหรับ ssh บน zsh?

ฉันต้องการตั้งค่าความสมบูรณ์ของคำสั่งใน zsh เพื่อแสดงชื่อโฮสต์หลังจากที่ฉันพิมพ์

ssh [TAB]

นำชื่อออกจากไฟล์. ssh / config ของฉัน (และควรเป็นที่รู้จักจาก known_hosts และ / etc / hosts และอื่น ๆ ที่เหมาะสม) และนำเสนอรายการเดียว

มันทำอะไรบางอย่างในขณะนี้ แต่

1. มันไม่ได้ใช้. ssh / config เลย
2. ต้องการชื่อผู้ใช้ก่อนแม้ว่าการใช้. ssh / config จะทำให้การพิมพ์ชื่อผู้ใช้ไม่จำเป็น
3. มันมีหลายรายการ (อาจเป็นหนึ่งใน known_hosts และอีกรายการจาก / 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