From 25a171180abc71dce9632d98cb8985d84c3e0e0e Mon Sep 17 00:00:00 2001
From: Grégoire Duchêne <gduchene@awhk.org>
Date: Fri, 19 Mar 2021 18:54:35 +0000
Subject: zsh/op-ssh-add: Do not drop arguments

---
 zsh/functions/op-ssh-add | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'zsh/functions')

diff --git a/zsh/functions/op-ssh-add b/zsh/functions/op-ssh-add
index 90cff9a..2e7fc68 100644
--- a/zsh/functions/op-ssh-add
+++ b/zsh/functions/op-ssh-add
@@ -16,11 +16,14 @@ if [[ ${@[1]} =~ '-[DdKLlTXx]' ]]; then
   return $?
 fi
 
-local fp prv pub
+local args=$@ fp prv pub
 for prv (${@[-1]} ~/.ssh/id_{ecdsa,ed25519}{,_sk}); do
   if [[ -f ${prv}.pub ]]; then
     pub=${prv}.pub
     fp=$(cut -d' ' -f2 <<< $(ssh-keygen -f ${pub} -l))
+    if [[ ${prv} != ${@[-1]} ]]; then
+      args+=(${prv})
+    fi
     break
   fi
 done
@@ -36,7 +39,7 @@ if [[ -z ${session} ]]; then
 fi
 
 zmodload zsh/zpty
-zpty ssh $(which -p ssh-add) ${prv}
+zpty ssh $(which -p ssh-add) ${args}
 
 _cleanup() {
   zpty -d ssh
-- 
cgit v1.2.3-70-g09d2