#!/bin/bash

# Docker Compose Windows VM Launcher for Omarchy
# Starts Windows VM via Docker Compose and connects with RDP
# When RDP closes, docker-compose will stop
#
# First tiem run: Please run `docker-compose up` and wait until windows is downloaded (the img file is 64 GB) and installed (takes up to 15 minutes)
# Please connect to 127.0.0.1:8006 via browser to check progress (in case you have a wrong product-key, or someting else)
# After that, you can use the launcher "Windows VM" that automatically start VM and connects RDesktop when docker is up and running

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &> /dev/null && pwd)"
COMPOSE_DIR="$SCRIPT_DIR/../config/windows-vm"
RDP_COMMAND="rdesktop -g 1920x1200 -P -z -x l -r sound:off -u docker 127.0.0.1:3389 -p admin"

echo "Starting Windows VM..."

# Create Windows storage directory if it doesn't exist
mkdir -p "$HOME/.windows"

# Change to compose directory where docker-compose.yml is located
cd "$COMPOSE_DIR" || {
    echo "Error: Could not change to $COMPOSE_DIR"
    exit 1
}


# Function to handle cleanup
cleanup() {
    echo "Shutting down Windows VM..."
    docker-compose down
    exit 0
}

# Set up signal handlers
trap cleanup SIGTERM SIGINT

# Start docker-compose in background
docker-compose up &
COMPOSE_PID=$!

# Wait for container to be running
echo "Waiting for Windows container to start..."
while true; do
    CONTAINER_STATUS=$(docker inspect --format='{{.State.Status}}' windows 2>/dev/null)
    if [ "$CONTAINER_STATUS" = "running" ]; then
        echo "Container is running!"
        break
    fi
    echo "  Container status: $CONTAINER_STATUS"
    sleep 2
    
    # Check if docker-compose process is still running
    if ! kill -0 $COMPOSE_PID 2>/dev/null; then
        echo "Docker Compose process died unexpectedly"
        exit 1
    fi
done

# Check if web interface is responding (faster than waiting for RDP)
echo "Checking if Windows VM web interface is responding..."
while true; do
    if curl -s --connect-timeout 5 http://127.0.0.1:8006 > /dev/null 2>&1; then
        echo "Web interface is responding at http://127.0.0.1:8006"
        break
    fi
    echo "  Web interface not ready yet..."
    sleep 3
done

# Now wait for RDP port specifically
echo "Waiting for RDP service to be ready..."
while ! nc -z 127.0.0.1 3389; do
    echo "  RDP port not ready yet..."
    sleep 5
done

echo "RDP port is open, testing connection..."
# Quick RDP test with shorter timeout
timeout 5 rdesktop -g 320x240 -u docker 127.0.0.1:3389 -p admin 2>/dev/null
if [ $? -ne 0 ]; then
    echo "RDP not fully ready yet, waiting 20 more seconds..."
    sleep 20
fi

echo "Starting RDP connection..."
echo "When you close RDP, the Windows VM will automatically shut down."

# Start RDP in foreground - when it exits, cleanup will run
$RDP_COMMAND

# If RDP exits, cleanup
cleanup