change the patch logic to entrypoint of SMUS

Author: Hua Cao

template/v2/dirs/etc/patches/apply_patches.sh

@@ -2,9 +2,35 @@
 
 set -eux
 
+# Check if parameter is provided
+if [ $# -ne 1 ]; then
+    echo "Usage: $0 [smus|studio-ai]"
+    exit 1
+fi
+
+# Validate parameter
+case "$1" in
+    "smus")
+        PATCH_DIR="/etc/patches/smus"
+        ;;
+    "studio-ai")
+        PATCH_DIR="/etc/patches/studio-ai"
+        ;;
+    *)
+        echo "Error: Parameter must be either 'smus' or 'studio-ai'"
+        exit 1
+        ;;
+esac
+
+# Check if patch directory exists
+if [ ! -d "$PATCH_DIR" ]; then
+    echo "Error: Patch directory $PATCH_DIR does not exist"
+    exit 1
+fi
+
 # Patch files can be generated via "diff -u /path/to/original_file /path/to/new_file > XXX_bad_package.patch"
 # See https://www.thegeekstuff.com/2014/12/patch-command-examples/
-for PATCHFILE in /etc/patches/*.patch; do
+for PATCHFILE in "$PATCH_DIR"/*.patch; do
     [ -f "$PATCHFILE" ] || continue
     echo "Applying $PATCHFILE"
     (cd "/opt/conda" && patch --strip=3 < "$PATCHFILE")

template/v2/dirs/etc/patches/0001-sparkmagic.patch renamed to template/v2/dirs/etc/patches/smus/0001-sparkmagic.patch

@@ -6,6 +6,9 @@ set -e
 # micromamba commands (e.g. using `micromamba activate` to activate environments)
 eval "$(micromamba shell hook --shell=bash)"
 
+# apply patches for SMUS
+/etc/patches/apply_patches.sh smus && sudo rm -rf /etc/patches
+
 # Activate conda environment 'base', where supervisord is installed
 micromamba activate base