Update start.sh

start.sh

@@ -1,18 +1,39 @@
 #!/bin/bash
 
-# --- Git and SSH Setup (Corrected for non-root) ---
-echo "Setting up SSH for Git in a temporary directory..."
-SSH_DIR="/tmp/.ssh"
-mkdir -p "$SSH_DIR"
-
-# This assumes the key is copied to /opt/app/id_rsa_huggingface in the Dockerfile
-if [ -f "/opt/app/id_rsa_huggingface" ]; then
-  cp /opt/app/id_rsa_huggingface "$SSH_DIR/id_rsa"
-  chmod 600 "$SSH_DIR/id_rsa"
-  echo "SSH key prepared."
+# --- Git and SSH Setup (Using Secrets) ---
+echo "Setting up SSH for Git..."
+
+# Check if the SSH_PRIVATE_KEY secret is provided from Hugging Face Space secrets
+if [ -z "$SSH_PRIVATE_KEY" ]; then
+  echo "Warning: SSH_PRIVATE_KEY secret is not set. Skipping git synchronization."
 else
-  echo "Warning: SSH key /opt/app/id_rsa_huggingface not found."
+  echo "SSH_PRIVATE_KEY secret found. Setting up for git synchronization..."
+  SSH_DIR="/tmp/.ssh"
+  mkdir -p "$SSH_DIR"
+  
+  # Write the key from the environment variable to a file. Use -e to handle newlines.
+  echo -e "$SSH_PRIVATE_KEY" > "$SSH_DIR/id_rsa"
+  chmod 600 "$SSH_DIR/id_rsa"
+  echo "SSH key prepared from secret."
+
+  # Use GIT_SSH_COMMAND to specify key and options
+  export GIT_SSH_COMMAND="ssh -i $SSH_DIR/id_rsa -o StrictHostKeyChecking=no"
+
+  echo "Cloning/updating repository..."
+  # Clone the repository into the persistent /data directory if it doesn't exist
+  if [ ! -d "/data/.git" ]; then
+    # Clone to a temporary directory and move contents to /data
+    git clone git@hf.co:spaces/rachkovii68/x-ui /tmp/repo
+    mv /tmp/repo/.git /data/
+  else
+    echo "Repo already exists in /data. Pulling latest changes."
+    cd /data
+    git pull
+    cd /
+  fi
 fi
+# --- End Git and SSH Setup ---
+
 
 echo "Cloning/updating repository..."
 # Use GIT_SSH_COMMAND to specify key and options without writing to system files