#!/bin/sh
print_title()
{
  printf "\e[34;47m%s\e[39;49m\n" "$1"
}
print_success()
{
  printf "\e[32;107msuccess\e[39;49m\n"
}
print_failure()
{
  printf "\e[31;107m%s\e[39;49m\n" "$1"
}

if [ "--release" = "$1" ]; then
  BUILD_CMD="cargo build --release"
  RELEASE_EFI_FILE=target/x86_64-unknown-uefi/release/untitled_os.efi
else
  print_title "no --release arg : assuming debug mode"
  BUILD_CMD="cargo build"
  RELEASE_EFI_FILE=target/x86_64-unknown-uefi/debug/untitled_os.efi
fi

if [ ! -f $RELEASE_EFI_FILE ]; then
  print_failure "missing file: ${RELEASE_EFI_FILE}"
  print_failure "Please build first with '$BUILD_CMD'"
  exit 1
fi

mkdir qemu
mkdir qemu/esp
mkdir qemu/esp/efi
mkdir qemu/esp/efi/boot

print_title "copying OVMF files" &&
cp /usr/share/OVMF/OVMF_CODE.fd qemu &&
cp /usr/share/OVMF/OVMF_VARS.fd qemu &&

print_title "copying EFI file" &&
cp $RELEASE_EFI_FILE qemu/esp/efi/boot/bootx64.efi &&

print_title "launching QEMU" &&
qemu-system-x86_64 -enable-kvm \
-drive if=pflash,format=raw,readonly=on,file=qemu/OVMF_CODE.fd \
-drive if=pflash,format=raw,readonly=on,file=qemu/OVMF_VARS.fd  \
-drive format=raw,file=fat:rw:qemu/esp &&

print_success &&
rm -f qemu/esp/efi/boot/bootx64.efi &&
rmdir qemu/esp/efi/boot &&
rmdir qemu/esp/efi &&
rm -f qemu/esp/NvVars &&
rmdir qemu/esp &&
rm -f qemu/OVMF_CODE.fd &&
rm -f qemu/OVMF_VARS.fd &&
rmdir qemu &&
exit 0

print_failure "something went wrong"
exit 1