controller test2

deployment/02_deploy_to_controller/scripts/initial_ssh_config.sh

@@ -26,19 +26,22 @@ if [ -z "$TAILSCALE_KEY" ]; then
   usage
 fi
 
+# SSH options to avoid known hosts issues and prompts
+SSH_OPTIONS="-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o LogLevel=ERROR"
+
 # Function to test SSH connection
 test_ssh_connection() {
-  ssh -o BatchMode=yes -o ConnectTimeout=5 -J turing.wpi.edu app echo "SSH connection successful" >/dev/null 2>&1
+  ssh $SSH_OPTIONS -o BatchMode=yes -o ConnectTimeout=5 -J turing.wpi.edu student-admin@app echo "SSH connection successful" >/dev/null 2>&1
 }
 
 # Backup the existing authorized_keys file
-ssh -i /opt/CS_553/keys/student-admin-key -J turing.wpi.edu app <<EOF || handle_error "Failed to backup authorized_keys"
+ssh $SSH_OPTIONS -i /opt/CS_553/keys/student-admin-key -J turing.wpi.edu student-admin@app <<EOF || handle_error "Failed to backup authorized_keys"
     cp ~/.ssh/authorized_keys ~/.ssh/authorized_keys.bak || handle_error "Failed to create backup of authorized_keys"
     echo "Backup of authorized_keys created"
 EOF
 
 # Update authorized_keys file with the new key while keeping existing keys
-ssh -i /opt/CS_553/keys/student-admin-key -J turing.wpi.edu app <<EOF || handle_error "Failed to update authorized_keys"
+ssh $SSH_OPTIONS -i /opt/CS_553/keys/student-admin-key -J turing.wpi.edu student-admin@app <<EOF || handle_error "Failed to update authorized_keys"
     NEW_KEY="ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIARTYgwoPW+VpBofWGYuHIldh18EUo42PHF/e08Dzcyp admin key CS553"
     if ! grep -q "\$NEW_KEY" ~/.ssh/authorized_keys; then
         echo "\$NEW_KEY" >> ~/.ssh/authorized_keys || handle_error "Failed to append new key to authorized_keys"
@@ -52,7 +55,7 @@ if test_ssh_connection; then
   echo "SSH connection with new key successful"
 else
   echo "SSH connection with new key failed. Restoring backup..."
-  ssh -i /opt/CS_553/keys/student-admin-key -J turing.wpi.edu app <<EOF || handle_error "Failed to restore authorized_keys backup"
+  ssh $SSH_OPTIONS -i /opt/CS_553/keys/student-admin-key -J turing.wpi.edu student-admin@app <<EOF || handle_error "Failed to restore authorized_keys backup"
         cp ~/.ssh/authorized_keys.bak ~/.ssh/authorized_keys || handle_error "Failed to restore backup of authorized_keys"
         chmod 600 ~/.ssh/authorized_keys || handle_error "Failed to set permissions on restored authorized_keys"
         echo "Backup of authorized_keys restored"
@@ -61,9 +64,14 @@ EOF
 fi
 
 # Install Tailscale and set up with provided key
-ssh -J turing.wpi.edu app <<EOF || handle_error "Failed to set up Tailscale"
-    # Install Tailscale
-    curl -fsSL https://tailscale.com/install.sh | sh || handle_error "Failed to install Tailscale"
+ssh $SSH_OPTIONS -J turing.wpi.edu student-admin@app <<EOF || handle_error "Failed to set up Tailscale"
+    # Check if Tailscale is already installed
+    if ! command -v tailscale &> /dev/null; then
+        echo "Tailscale not found. Installing..."
+        curl -fsSL https://tailscale.com/install.sh | sudo sh || handle_error "Failed to install Tailscale"
+    else
+        echo "Tailscale is already installed."
+    fi
 
     # Run Tailscale with the provided auth key
     sudo tailscale up --authkey "$TAILSCALE_KEY" || handle_error "Failed to run Tailscale"