#!/usr/bin/env bash
# OrionOS ARM64 ISO builder — for Apple Silicon Macs (runs natively, no emulation).
# First run downloads Arch Linux ARM rootfs (~300MB); cached by Docker afterwards.
#
# Usage:  ./build-iso-arm64.sh
# Output: dist/orionos-arm64-<date>.iso
set -euo pipefail

REPO="$(cd "$(dirname "$0")" && pwd)"
DIST="$REPO/dist"
IMAGE="orionos-builder-arm64"
ISO_TAG="orionos-arm64-$(date +%Y%m%d)"

# ── preflight ─────────────────────────────────────────────────────────
if ! command -v docker &>/dev/null; then
    echo "  Docker is not installed."
    echo "  Install Docker Desktop: https://www.docker.com/products/docker-desktop/"
    exit 1
fi

if ! docker info &>/dev/null; then
    echo "  Docker is not running. Open Docker Desktop and re-run."
    exit 1
fi

echo ""
echo "┌─────────────────────────────────────────────┐"
echo "│       OrionOS ARM64 ISO Builder              │"
echo "│  Native on Apple Silicon — no emulation     │"
echo "└─────────────────────────────────────────────┘"
echo ""

# ── build the builder image ───────────────────────────────────────────
echo "Step 1/3: Building Docker image"
echo "          First run downloads ALARM rootfs (~300MB) — cached afterwards."
echo ""
docker buildx build \
    --platform linux/arm64 \
    --load \
    -t "$IMAGE" \
    -f "$REPO/iso/docker/Dockerfile.arm64" \
    "$REPO"

# ── run the ISO build inside the container ────────────────────────────
echo ""
echo "Step 2/3: Building ISO inside ARM64 Arch Linux container..."
echo "          (5-15 min depending on internet speed)"
echo ""
mkdir -p "$DIST"
docker run \
    --rm \
    --privileged \
    --platform linux/arm64 \
    --security-opt seccomp=unconfined \
    -v "$DIST:/dist" \
    -e DIST=/dist \
    "$IMAGE"

# ── rename with date tag ──────────────────────────────────────────────
echo ""
echo "Step 3/3: Finalizing..."
ISO="$(find "$DIST" -name "orionos-arm64*.iso" | head -1)"
if [[ -z "$ISO" ]]; then
    echo "ERROR: ISO not found in $DIST — check the build output above."
    exit 1
fi

FINAL="$DIST/$ISO_TAG.iso"
mv "$ISO" "$FINAL"
SIZE="$(du -sh "$FINAL" | cut -f1)"

echo ""
echo "┌─────────────────────────────────────────────────────┐"
echo "│  Done!                                               │"
echo "│                                                      │"
echo "│  ISO: dist/$ISO_TAG.iso"
echo "│  Size: $SIZE"
echo "│                                                      │"
echo "│  Load in UTM:                                        │"
echo "│    1. Open UTM → New VM → Virtualize                │"
echo "│    2. Choose Linux → enable Apple Virtualization     │"
echo "│    3. Boot ISO: select the file above               │"
echo "│    4. Display: VirtIO GPU (recommended)             │"
echo "│    5. Boot — OrionOS auto-logs in as 'orion'        │"
echo "│                                                      │"
echo "│  Copy-paste between macOS and VM:                    │"
echo "│    spice-vdagent runs automatically on login        │"
echo "└─────────────────────────────────────────────────────┘"
echo ""