#!/bin/bash

# Defaults
TIMEOUT=600
ELAPSED=0
INITIAL_SLEEP=60
SLEEP_DURATION=30
SSH_AVAILABLE=0
DEBUG_ID="[homelab-ci]"

# Run Vagrant Up in the background
PLAYBOOK=dockerbox vagrant up --debug &
VAGRANT_UP_PID=$!

# Initial delay
echo "$DEBUG_ID Waiting for VM to start..."
sleep $INITIAL_SLEEP

# Loop until timeout or breaks
while [[ $ELAPSED -lt $TIMEOUT ]]; do
	VAGRANT_SSH_CONFIG=$(mktemp)
	vagrant ssh-config > "$VAGRANT_SSH_CONFIG"
	set -x
	cat "$VAGRANT_SSH_CONFIG"
	set +x
	vagrant status

	#echo "$DEBUG_ID Checking SSH connection availability..."
	#echo "$DEBUG_ID Private Key: $PRIVATE_KEY"
	#echo "$DEBUG_ID Host IP: $HOST_IP"
	#echo "$DEBUG_ID Running nmap to check open ports..."

	#if [[ $SSH_AVAILABLE -eq 1 ]]; then
	echo "$DEBUG_ID SSH port is open, attempting connection..."
	set -x
	ssh -vvv -F "$VAGRANT_SSH_CONFIG" cat /etc/os-release && break || echo "$DEBUG_ID SSH connection failed, retrying..."
	set +x
	echo "$DEBUG_ID SSH port not open, retrying in $SLEEP_DURATION seconds..."
	# Sleep and start again
	sleep $SLEEP_DURATION
	((ELAPSED+=SLEEP_DURATION))
done

# Success?
if [[ $SSH_AVAILABLE -ne 1 ]]; then
	echo "$DEBUG_ID Timeout reached without successful SSH connection."
fi

# Ensure the Vagrant up process completes
wait $VAGRANT_UP_PID