From ef49f04b6a35cee45ba613883e61c1081ca9854d Mon Sep 17 00:00:00 2001
From: msgupta <mgupta@onemedical.com>
Date: Fri, 15 May 2026 12:05:27 -0700
Subject: [PATCH] Removing the stale symlinks before starting the app

---
 .../local/bin/start-sagemaker-ui-code-editor   | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/template/v3/dirs/usr/local/bin/start-sagemaker-ui-code-editor b/template/v3/dirs/usr/local/bin/start-sagemaker-ui-code-editor
index 3fee4bd91..9ecae7c60 100755
--- a/template/v3/dirs/usr/local/bin/start-sagemaker-ui-code-editor
+++ b/template/v3/dirs/usr/local/bin/start-sagemaker-ui-code-editor
@@ -57,6 +57,24 @@ install_prepackaged_extensions() {
         ln -s "$dir" "${persistent_extensions_dir}/${dir_name}"
         echo "Added extension: $dir"
     done
+  else
+    # Remove stale symlinks pointing to non-existent targets (old image versions)
+    # and re-symlink extensions from the current image
+    find "$persistent_extensions_dir" -maxdepth 1 -type l | while read -r link; do
+      if [ ! -e "$link" ]; then
+        echo "Removing stale extension symlink: $link"
+        rm -f "$link"
+      fi
+    done
+
+    # Add symlinks for any new image extensions not already in the persistent dir
+    find "$prepackaged_extensions_dir" -mindepth 1 -maxdepth 1 -type d | while read -r dir; do
+      dir_name=$(basename "$dir")
+      if [ ! -e "${persistent_extensions_dir}/${dir_name}" ]; then
+        ln -s "$dir" "${persistent_extensions_dir}/${dir_name}"
+        echo "Added new extension: $dir"
+      fi
+    done
   fi
 }