#!/bin/bash

set -e

IMAGE="/var/lib/incus/virtual-machines/test-incus-os/root.img"

if [ "$(id -u)" -ne 0 ]; then
     echo "This script must be run as root"
     exit 1
fi

rm -rf ./mnt/

## Restore the original GPT table
sgdisk -l gpt.backup "$IMAGE"

## Format the ESP partition and restore its contents
losetup -f -P "$IMAGE"
sleep 1
mkfs.vfat /dev/loop0p1

mkdir ./mnt/
mount /dev/loop0p1 ./mnt/
cp -ar ./esp-orig/* ./mnt/
umount ./mnt/

losetup -d /dev/loop0