From d9de394d9645f65712c7d712ebfbc322fd5c8d48 Mon Sep 17 00:00:00 2001
From: Hugues Hiegel <hugues@hiegel.fr>
Date: Mon, 2 Feb 2009 12:59:10 +0100
Subject: [KEYCHAIN] meilleure gestion au login et à chaque prompt
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 12_Prompts.zsh | 40 +++++++++++++++++++++++++++-------------
 1 file changed, 27 insertions(+), 13 deletions(-)

(limited to '12_Prompts.zsh')

diff --git a/12_Prompts.zsh b/12_Prompts.zsh
index 364c22c..9927b88 100644
--- a/12_Prompts.zsh
+++ b/12_Prompts.zsh
@@ -101,29 +101,43 @@ update_prompt()
 	set_prompt_date
 
 	# GPG/SSH agents
-
-	KEYCHAIN=~/.keychain/$(hostname)-sh
-	for file in $(find $KEYCHAIN:h -name "$(hostname)-sh" -o -name "$(hostname)-sh-*")
-	do
-		source $file
-	done
-
 	AGENTS=""
-	if [ "$SSH_AGENT_PID" -gt 0 -a -e /proc/$SSH_AGENT_PID/cmdline ]
+	[ -f ${KEYCHAIN}     ] && source ${KEYCHAIN}
+	[ -f ${KEYCHAIN}-gpg ] && source ${KEYCHAIN}-gpg
+
+	# Check ssh-agent only if the env socket has been set and is accessible
+	if [ -S "$SSH_AUTH_SOCK" ]
 	then
-		if [ "`strings /proc/$SSH_AGENT_PID/cmdline | head -n1`" = "ssh-agent" ]
+		# Get keylist
+		SSH_AGENT_KEYLIST="$( ssh-add -l | grep "^[[:digit:]]\+ \([[:digit:]a-f]\{2\}:\)\{15\}[[:digit:]a-f]\{2\} .* (.*)$" )"
+
+		# Check if it is a forwarded agent
+		if [ "$SSH_AGENT_PID" -gt 0 -a -e /proc/$SSH_AGENT_PID/cmdline ]
 		then
-			## We have a running ssh agent
-			# get list of keys
-			if [ "$( ssh-add -l | grep "^[[:digit:]]\+ \([[:digit:]a-f]\{2\}:\)\{15\}[[:digit:]a-f]\{2\} .* (.*)$" )" != "" ]
+			# That's a local agent
+			if [ "$SSH_AGENT_KEYLIST" != "" ]
+			then
+				AGENTCOLOR="has_keys"
+				AGENTCHAR="★"
+			else
+				AGENTCOLOR="empty"
+				AGENTCHAR="☆"
+			fi
+		else
+			# That's a forwarded agent
+			if [ "$SSH_AGENT_KEYLIST" != "" ]
 			then
 				AGENTCOLOR="has_keys"
+				AGENTCHAR="●"
 			else
 				AGENTCOLOR="empty"
+				AGENTCHAR="○"
 			fi
-			AGENTS=$C_$agent_colors[$AGENTCOLOR]$_C"★"
 		fi
+
+		AGENTS=$C_$agent_colors[$AGENTCOLOR]$_C"$AGENTCHAR"
 	fi
+
 	GPG_AGENT_PID="$(echo $GPG_AGENT_INFO | cut -d: -f2)"
 	if [ "$GPG_AGENT_PID" != "" -a -e /proc/$GPG_AGENT_PID/cmdline ]
 	then
-- 
cgit v1.2.3