#!/bin/bash
IMG=$(find . -regextype posix-extended -regex "\./[0-9]+\.[0-9]+\.[0-9]+\.[0-9]{4,5}\.img" | sort -t. -k1,1n -k2,2n -k3,3n -k4,4n | tail -n 1)
TARGET_IMG_DISK="./root.img"
TARGET_DATA_DISK="./data.img"
TARGET_DATA_DISK_SIZE="8G"

if [ ! -z $1 ]; then
    TARGET_DATA_DISK_SIZE=$1
fi

qemu-img create -f raw $TARGET_DATA_DISK $TARGET_DATA_DISK_SIZE
fdisk $TARGET_DATA_DISK <<EOF
g
n



p
w
EOF

LOOP=$(sudo losetup -f)
sudo losetup -fP $TARGET_DATA_DISK
sudo mkfs.ext4 "${LOOP}p1"
sudo losetup -d ${LOOP}

qemu-img convert -c -f raw -O qcow2 $TARGET_DATA_DISK ${TARGET_DATA_DISK}.qcow2
rm -rf ${TARGET_DATA_DISK}
mv ${TARGET_DATA_DISK}.qcow2 ${TARGET_DATA_DISK}
cp $IMG $TARGET_IMG_DISK
U=$SUDO_USER
if [ -z $U ]; then
    U=$(whoami)
fi
sudo chown $U:$U $TARGET_IMG_DISK
sudo chown $U:$U $TARGET_DATA_DISK