#!/bin/bash
# Setup testing environment for archsetup
# Author: Craig Jennings <craigmartinjennings@gmail.com>
# License: GNU GPLv3
#
# This script performs one-time setup of the testing infrastructure:
# - Installs QEMU/KVM, sshpass, OVMF firmware, and socat
# - Verifies KVM support
# - Creates directories for test artifacts

set -e

# Get script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"

# Source utilities
source "$SCRIPT_DIR/lib/logging.sh"

# Initialize logging
mkdir -p "$PROJECT_ROOT/test-results"
LOGFILE="$PROJECT_ROOT/test-results/setup-$(date +'%Y%m%d-%H%M%S').log"
init_logging "$LOGFILE"

section "ArchSetup Testing Environment Setup"

# Check if running on Arch Linux
if [ ! -f /etc/arch-release ]; then
    fatal "This script is designed for Arch Linux"
fi

# Check if user has sudo
if ! sudo -n true 2>/dev/null; then
    warn "This script requires sudo access"
    info "You may be prompted for your password"
fi

# Install required packages
section "Installing Required Packages"

PACKAGES=(
    qemu-full
    sshpass
    edk2-ovmf
    socat
)

for pkg in "${PACKAGES[@]}"; do
    if pacman -Qi "$pkg" &>/dev/null; then
        info "$pkg is already installed"
    else
        step "Installing $pkg"
        if sudo pacman -S --noconfirm "$pkg" >> "$LOGFILE" 2>&1; then
            success "$pkg installed"
        else
            error "Failed to install $pkg"
            fatal "Package installation failed"
        fi
    fi
done

# Verify KVM support
section "Verifying KVM Support"

if [ -e /dev/kvm ]; then
    success "KVM is available"
else
    error "KVM is not available"
    info "Check if virtualization is enabled in BIOS"
    info "Load kvm module: sudo modprobe kvm-intel (or kvm-amd)"
    fatal "KVM not available"
fi

# Check which KVM module is loaded
if lsmod | grep -q kvm_intel; then
    info "Using Intel KVM"
elif lsmod | grep -q kvm_amd; then
    info "Using AMD KVM"
else
    warn "No KVM module detected"
    info "Load with: sudo modprobe kvm-intel (or kvm-amd)"
fi

# Verify OVMF firmware
section "Verifying OVMF Firmware"

OVMF_CODE="/usr/share/edk2/x64/OVMF_CODE.4m.fd"
OVMF_VARS="/usr/share/edk2/x64/OVMF_VARS.4m.fd"

if [ -f "$OVMF_CODE" ] && [ -f "$OVMF_VARS" ]; then
    success "OVMF firmware files present"
else
    fatal "OVMF firmware files not found (expected at $OVMF_CODE)"
fi

# Create directory structure
section "Creating Directory Structure"

DIRS=(
    "$PROJECT_ROOT/vm-images"
    "$PROJECT_ROOT/test-results"
)

for dir in "${DIRS[@]}"; do
    if [ -d "$dir" ]; then
        info "Directory exists: $dir"
    else
        step "Creating directory: $dir"
        if mkdir -p "$dir" 2>> "$LOGFILE"; then
            success "Directory created: $dir"
        else
            error "Failed to create directory: $dir"
        fi
    fi
done

# Summary
section "Setup Complete"

success "Testing environment is ready"
info ""
info "Next steps:"
info "  1. Copy an archangel ISO to: $PROJECT_ROOT/vm-images/"
info "  2. Create base VM: ./scripts/testing/create-base-vm.sh"
info "  3. Run a test: ./scripts/testing/run-test.sh"
info ""
info "Log file: $LOGFILE"