#!/bin/bash
# burnit.sh - Interactive ISO-to-USB writer with menu

set -euo pipefail

ISO_DIR="/gigapool/vms/isos"  # Change this to wherever you keep ISOs

echo "🔥 Welcome to burnit.sh – ISO burner for Unix junkies."
echo

# Step 1: List ISOs in ISO_DIR
echo "📁 Scanning for ISOs in: $ISO_DIR"
mapfile -t isos < <(find "$ISO_DIR" -maxdepth 1 -type f \( -iname "*.iso" -o -iname "*.img" \) | sort)

if [[ ${#isos[@]} -eq 0 ]]; then
  echo "❌ No ISO or IMG files found in $ISO_DIR."
  exit 1
fi

echo "📦 Available ISO/IMG files:"
for i in "${!isos[@]}"; do
  printf "  [%d] %s\n" "$i" "$(basename "${isos[$i]}")"
done

echo
read -rp "🎯 Pick an image number: " index

if ! [[ "$index" =~ ^[0-9]+$ ]] || (( index < 0 || index >= ${#isos[@]} )); then
  echo "❌ Invalid selection."
  exit 1
fi

iso="${isos[$index]}"
echo "✅ Selected: $(basename "$iso")"
echo

# Step 2: List available removable devices
echo "💽 Scanning removable block devices..."
lsblk -o NAME,SIZE,MODEL,MOUNTPOINT,TRAN,VENDOR,HOTPLUG | grep -E '1$' || true
echo

read -rp "🧨 Enter the target device (e.g., /dev/sdX): " target
if [[ ! -b "$target" ]]; then
  echo "❌ Device not found: $target"
  exit 1
fi

echo
read -rp "⚠️ ARE YOU SURE you want to overwrite $target with $iso? [y/N]: " confirm
if [[ "$confirm" != "y" && "$confirm" != "Y" ]]; then
  echo "🛑 Aborted."
  exit 1
fi

echo
echo "🚀 Flashing image..."
sleep 1

pv "$iso" | sudo dd of="$target" bs=4M status=progress oflag=sync

echo
echo "✅ Done writing to $target."

# Optional: Eject
if mount | grep "$target" > /dev/null; then
  sudo umount "${target}"* || true
fi
if command -v eject &> /dev/null; then
  sudo eject "$target" || true
fi

echo "💿 Bootable media is ready to rock."