From 1ca249fe5ecb1cd0504416f56c59ee5d34e6bf8b Mon Sep 17 00:00:00 2001
From: Jusong Yu <jusong.yeu@gmail.com>
Date: Fri, 13 Oct 2023 16:31:52 +0200
Subject: [PATCH] f-dd

---
 stack/base/before-notebook.d/10_prepare-home-config.sh | 6 +++++-
 stack/base/load-singlesshagent.sh                      | 4 ++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/stack/base/before-notebook.d/10_prepare-home-config.sh b/stack/base/before-notebook.d/10_prepare-home-config.sh
index 567d6659..a4d0ab70 100644
--- a/stack/base/before-notebook.d/10_prepare-home-config.sh
+++ b/stack/base/before-notebook.d/10_prepare-home-config.sh
@@ -42,7 +42,11 @@ if [[ -f /opt/bin/load-singlesshagent.sh ]] && ! grep -q "${header}" /home/${NB_
 ${header}
 if [ -f /opt/bin/load-singlesshagent.sh ]; then
     source /opt/bin/load-singlesshagent.sh
-    ssh-add /home/${NB_USER}/.ssh/id_rsa 2> /dev/null
 fi
 EOF
 fi
+
+# load the ssh-agent and add the default key generated
+# the return code can be non-zero if the ssh-agent is not running
+# which will cause the notebook to fail to start so we need to ignore the return code
+source /opt/bin/load-singlesshagent.sh || true
diff --git a/stack/base/load-singlesshagent.sh b/stack/base/load-singlesshagent.sh
index c2e8f9f0..d4545fc8 100644
--- a/stack/base/load-singlesshagent.sh
+++ b/stack/base/load-singlesshagent.sh
@@ -44,8 +44,8 @@ load_singlesshagent() {
     elif [ "$SSHADD_RETVAL" == "1" ]
     then
         [ "$VERBOSE" == "true" ] && echo "  - ssh-agent found (${SSH_AGENT_PID}), no keys (I might want to add keys here)" >&2
-        # Maybe the user wants to do it by himself?
-        #ssh-add
+        # run ssh-add to add the default generate key `id_rsa` to the agent
+        ssh-add ~/.ssh/id_rsa
     elif [ "$SSHADD_RETVAL" == "0" ]
     then
         NUMKEYS=`echo "$SSH_ADD_OUTPUT" | wc -l`