From 5e966203aece09f7e87af973ef54d552d651bc52 Mon Sep 17 00:00:00 2001
From: Pandu POLUAN <pepoluan@gmail.com>
Date: Fri, 28 Feb 2025 11:30:03 +0700
Subject: [PATCH 1/3] Add ssh_fingerprint (and ssh_fp alias)

Also update README.md
---
 plugins/ssh/README.md      |  7 +++++++
 plugins/ssh/ssh.plugin.zsh | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+)

diff --git a/plugins/ssh/README.md b/plugins/ssh/README.md
index 3dd32ec76..cf3318326 100644
--- a/plugins/ssh/README.md
+++ b/plugins/ssh/README.md
@@ -14,3 +14,10 @@ plugins=(... ssh)
 - `ssh_rmhkey`: remove host key from known hosts based on a host section name from `.ssh/config`.
 - `ssh_load_key`: load SSH key into agent.
 - `ssh_unload_key`: remove SSH key from agent.
+- `ssh_fingerprint` : calculate fingerprint of specifed key files. It has some options:
+  - `-md5` : Use MD5 fingerprint instead of SHA256
+  - `-n` : Don't colorize using ANSI
+  - `-q` : Don't print filename(s)
+  - If key files are not specified, defaults to `~/.ssh/authorized_keys`
+  - `ssh_fp` is an alias to this function
+
diff --git a/plugins/ssh/ssh.plugin.zsh b/plugins/ssh/ssh.plugin.zsh
index b5b050536..f837c8e94 100644
--- a/plugins/ssh/ssh.plugin.zsh
+++ b/plugins/ssh/ssh.plugin.zsh
@@ -51,3 +51,35 @@ function ssh_unload_key {
     ssh-add -d "$keyfile"
   fi
 }
+
+############################################################
+# Calculate SSH key fingerprint
+function ssh_fingerprint {
+  local fptype
+  local quiet
+  local ansi
+
+  zmodload zsh/zutil
+  zparseopts -D -- md5=fptype q=quiet n=ansi
+  fptype="${fptype[*]}"
+  if [[ -z $fptype ]]; then
+    fptype="sha256"
+  else
+    fptype="${fptype:1}"
+  fi
+  if [[ -z ${ansi[*]} ]]; then
+    ansi=("\e[1;30m" "\e[0m")
+  else
+    ansi=("" "")
+  fi
+
+  local keyfiles=( "$@" )
+  if [[ ${#keyfiles[@]} == 0 ]]; then keyfiles=( "$HOME/.ssh/authorized_keys" ); fi
+  local f
+  for f in "${keyfiles[@]}"; do
+    if [[ $quiet != "-q" ]]; then printf "${ansi[1]}$f:${ansi[2]}\n"; fi
+    ssh-keygen -l -E "$fptype" -f "$f"
+  done
+}
+alias ssh_fp='ssh_fingerprint'
+

From 03e568ad81614dac4aa9acfb347cce9861e7fdfe Mon Sep 17 00:00:00 2001
From: Pandu POLUAN <pepoluan@gmail.com>
Date: Fri, 28 Feb 2025 11:39:07 +0700
Subject: [PATCH 2/3] STYLE: Add parens to func names

In accordance to code style guide
---
 plugins/ssh/ssh.plugin.zsh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/plugins/ssh/ssh.plugin.zsh b/plugins/ssh/ssh.plugin.zsh
index f837c8e94..48cc85021 100644
--- a/plugins/ssh/ssh.plugin.zsh
+++ b/plugins/ssh/ssh.plugin.zsh
@@ -20,7 +20,7 @@ unset _ssh_configfile
 ############################################################
 # Remove host key from known hosts based on a host section
 # name from .ssh/config
-function ssh_rmhkey {
+function ssh_rmhkey() {
   local ssh_configfile="$HOME/.ssh/config"
   local ssh_host="$1"
   if [[ -z "$ssh_host" ]]; then return; fi
@@ -42,7 +42,7 @@ function ssh_load_key() {
 
 ############################################################
 # Remove SSH key from agent
-function ssh_unload_key {
+function ssh_unload_key() {
   local key="$1"
   if [[ -z "$key" ]]; then return; fi
   local keyfile="$HOME/.ssh/$key"
@@ -54,7 +54,7 @@ function ssh_unload_key {
 
 ############################################################
 # Calculate SSH key fingerprint
-function ssh_fingerprint {
+function ssh_fingerprint() {
   local fptype
   local quiet
   local ansi

From 25ed84afb60fdde77ce1ea5df573b034519d9f05 Mon Sep 17 00:00:00 2001
From: Pandu POLUAN <pepoluan@gmail.com>
Date: Fri, 28 Feb 2025 11:42:38 +0700
Subject: [PATCH 3/3] Provide return values

---
 plugins/ssh/ssh.plugin.zsh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/plugins/ssh/ssh.plugin.zsh b/plugins/ssh/ssh.plugin.zsh
index 48cc85021..6527b42b5 100644
--- a/plugins/ssh/ssh.plugin.zsh
+++ b/plugins/ssh/ssh.plugin.zsh
@@ -78,8 +78,9 @@ function ssh_fingerprint() {
   local f
   for f in "${keyfiles[@]}"; do
     if [[ $quiet != "-q" ]]; then printf "${ansi[1]}$f:${ansi[2]}\n"; fi
-    ssh-keygen -l -E "$fptype" -f "$f"
+    if ! ssh-keygen -l -E "$fptype" -f "$f"; then return $?; fi
   done
+  return 0
 }
 alias ssh_fp='ssh_fingerprint'