From ca78c0509f069287ed031108154248d3408b615e Mon Sep 17 00:00:00 2001
From: whiskerz007 <2713522+whiskerz007@users.noreply.github.com>
Date: Wed, 13 Mar 2019 16:56:53 +0100
Subject: [PATCH] Update install.sh

---
 install.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/install.sh b/install.sh
index 02f12b2..7e0ce09 100644
--- a/install.sh
+++ b/install.sh
@@ -5,6 +5,7 @@ pvesm list $VM_STORAGE >& /dev/null || { \
 echo -e "\n\n\nERROR: '$VM_STORAGE' is not a valid storage ID.\n\n\n" && \
 exit 1
 } && \
+STORAGE_TYPE=`pvesm status -storage local | awk 'NR>1 {print $2}'` && \
 VMID=$(cat<<EOF | python3
 import json
 with open('/etc/pve/.vmlist') as vmlist:
@@ -53,7 +54,7 @@ echo -e "\n\n\n" \
 VMDISK=vm-${VMID}-disk-
 qm create $VMID -bios ovmf -name $(sed -e "s/\_//g" -e "s/.vdi.gz//" <<< $FILE) \
         -net0 virtio,bridge=vmbr0 -onboot 1 -ostype l26 -scsihw virtio-scsi-pci && \
-pvesm alloc $VM_STORAGE $VMID ${VMDISK}0 128 1>&/dev/null && \
+pvesm alloc $VM_STORAGE $VMID ${VMDISK}0$(if [ "$STORAGE_TYPE" = "dir"]; then echo "qcow2";fi) 128 1>&/dev/null && \
 qm importdisk $VMID ${FILE%".gz"} $VM_STORAGE 1>&/dev/null && \
 qm set $VMID -bootdisk sata0 -efidisk0 ${VM_STORAGE}:${VMDISK}0,size=128K \
         -sata0 ${VM_STORAGE}:${VMDISK}1,size=6G > /dev/null && \