chore: retry on fail for initial installs

Author: samrose

ebssurrogate/scripts/surrogate-bootstrap-nix.sh

@@ -29,18 +29,33 @@ function waitfor_boot_finished {
 }
 
 function install_packages {
-	# Setup Ansible on host VM
-	apt-get update && sudo apt-get install software-properties-common -y
-	add-apt-repository --yes --update ppa:ansible/ansible && sudo apt-get install ansible -y
-	ansible-galaxy collection install community.general
-
-	# Update apt and install required packages
-	apt-get update
-	apt-get install -y \
-		gdisk \
-		e2fsprogs \
-		debootstrap \
-		nvme-cli
+    trap 'echo "Error on line $LINENO"' ERR
+    
+    retry() {
+        local retries=3
+        local count=1
+        until "$@"; do
+            [[ $count -eq $retries ]] && return 1
+            echo "Command failed, attempt $count/$retries"
+            ((count++))
+            sleep 10
+        done
+        return 0
+    }
+
+    retry apt-get update
+    retry apt-get install -y software-properties-common
+    retry add-apt-repository --yes --update ppa:ansible/ansible
+    
+    retry apt-get install -y ansible
+    if ! command -v ansible &>/dev/null; then
+        echo "Ansible installation failed"
+        return 1
+    fi
+
+    retry ansible-galaxy collection install community.general
+    retry apt-get update
+    retry apt-get install -y gdisk e2fsprogs debootstrap nvme-cli
 }
 
 # Partition the new root EBS volume