#!/usr/bin/env bash
SYSTEM=$(uname -s)
size="1024M"
nvme_disk="/var/lib/libvirt/images/nvme_disk.img"
type="nvme"

function usage() {
	echo "Usage: ${0##*/} [-s <disk_size>] [-n <backing file name>]"
	echo "-s <disk_size> with postfix e.g. 2G        default: 1024M"
	echo "                                    for OCSSD default: 9G"
	echo "-n <backing file name>        backing file path with name"
	echo "           default: /var/lib/libvirt/images/nvme_disk.img"
	echo "-t <type>                  default: nvme available: ocssd"
}

while getopts "s:n:t:h-:" opt; do
	case "${opt}" in
		-)
			echo "  Invalid argument: $OPTARG"
			usage
			exit 1
			;;
		s)
			size=$OPTARG
			;;
		n)
			nvme_disk=$OPTARG
			;;
		t)
			type=$OPTARG
			;;
		h)
			usage
			exit 0
			;;
		*)
			echo "  Invalid argument: $OPTARG"
			usage
			exit 1
			;;
	esac
done

if [ "${SYSTEM}" != "Linux" ]; then
	echo "This script supports only Linux OS" >&2
	exit 2
fi

WHICH_OS=$(lsb_release -i | awk '{print $3}')
case $type in
	"nvme")
		qemu-img create -f raw "$nvme_disk" $size
		;;
	"ocssd")
		if [ $size == "1024M" ]; then
			size="9G"
		fi
		fallocate -l $size "$nvme_disk"
		touch "${nvme_disk}_ocssd_md"
		;;
	*)
		echo "We support only nvme and ocssd disks types"
		exit 1
		;;
esac

case $WHICH_OS in
	"Fedora")
		qemu_user_group="qemu:qemu"

		# Change SE Policy
		sudo chcon -t svirt_image_t "$nvme_disk"
		;;
	"Ubuntu")
		qemu_user_group="libvirt-qemu:kvm"
		;;
	*)
		# That's just a wild guess for now
		# TODO: needs improvement for other distros
		qemu_user_group="libvirt-qemu:kvm"
		;;
esac

chmod 777 "$nvme_disk"
chown $qemu_user_group "$nvme_disk"
if [ "$type" == "ocssd" ]; then
	chown $qemu_user_group "${nvme_disk}_ocssd_md"
fi