Add IS_GIT_PROJECT variable and write to smus-storage-metadata.json (#730)

Author: ruijiang-rjian

template/v2/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh

@@ -158,9 +158,11 @@ is_s3_storage
 is_s3_storage_flag=$?  # 0 if S3 storage, 1 if Git
 
 if [ "$is_s3_storage_flag" -eq 0 ]; then
+    export IS_GIT_PROJECT=false
     export SMUS_PROJECT_DIR="$HOME/shared"
     echo "Project is using S3 storage, project directory set to: $SMUS_PROJECT_DIR"
 else
+    export IS_GIT_PROJECT=true
     export SMUS_PROJECT_DIR="$HOME/src"
     echo "Project is using Git storage, project directory set to: $SMUS_PROJECT_DIR"
 fi
@@ -173,9 +175,14 @@ else
 fi
 
 # Write SMUS_PROJECT_DIR to a JSON file to be accessed by JupyterLab Extensions
+mkdir -p "$HOME/.config"  # Create config directory if it doesn't exist
 jq -n \
   --arg smusProjectDirectory "$SMUS_PROJECT_DIR" \
-  '{ smusProjectDirectory: $smusProjectDirectory }' > $HOME/.config/smus-storage-metadata.json
+  --arg isGitProject "$IS_GIT_PROJECT" \
+  '{ 
+    smusProjectDirectory: $smusProjectDirectory,
+    isGitProject: ($isGitProject == "true")
+  }' > "$HOME/.config/smus-storage-metadata.json"
 
 if [ $is_s3_storage_flag -ne 0 ]; then
   # Creating a directory where the repository will be cloned

template/v3/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh

@@ -158,9 +158,11 @@ is_s3_storage
 is_s3_storage_flag=$?  # 0 if S3 storage, 1 if Git
 
 if [ "$is_s3_storage_flag" -eq 0 ]; then
+    export IS_GIT_PROJECT=false
     export SMUS_PROJECT_DIR="$HOME/shared"
     echo "Project is using S3 storage, project directory set to: $SMUS_PROJECT_DIR"
 else
+    export IS_GIT_PROJECT=true
     export SMUS_PROJECT_DIR="$HOME/src"
     echo "Project is using Git storage, project directory set to: $SMUS_PROJECT_DIR"
 fi
@@ -173,9 +175,14 @@ else
 fi
 
 # Write SMUS_PROJECT_DIR to a JSON file to be accessed by JupyterLab Extensions
+mkdir -p "$HOME/.config"  # Create config directory if it doesn't exist
 jq -n \
   --arg smusProjectDirectory "$SMUS_PROJECT_DIR" \
-  '{ smusProjectDirectory: $smusProjectDirectory }' > $HOME/.config/smus-storage-metadata.json
+  --arg isGitProject "$IS_GIT_PROJECT" \
+  '{ 
+    smusProjectDirectory: $smusProjectDirectory,
+    isGitProject: ($isGitProject == "true")
+  }' > "$HOME/.config/smus-storage-metadata.json"
 
 if [ $is_s3_storage_flag -ne 0 ]; then
   # Creating a directory where the repository will be cloned