Update install.sh

install.sh

@@ -1,67 +1,96 @@
 #!/bin/bash
+set -o pipefail
+shopt -s expand_aliases
+alias die='EXIT=$? LINE=$LINENO error_exit'
+trap die ERR
+function error_exit() {
+    REASON=$1
+    MSG="\e[91mERROR: \e[93m$EXIT@"
+    if [ -z "$REASON" ]; then
+        MSG="$MSG$LINE:"
+        REASON="Unknown failure occured."
+    else
+        MSG="$MSG`echo $(( $LINE - 1 ))`:"
+    fi
+    echo -e "$MSG \e[97m$REASON\e[39m\e[49m"
+    exit $EXIT
+}
+function cleanup() {
+    popd >/dev/null
+    rm -rf $TMP
+}
+trap cleanup EXIT
+TMP=`mktemp -d`
+pushd $TMP >/dev/null
 
-VM_STORAGE=${1:-local-lvm} && \
+STORAGE=${1:-local-lvm}
-pvesm list $VM_STORAGE >& /dev/null || { \
+pvesm list $STORAGE >& /dev/null ||
-echo -e "\n\n\nERROR: '$VM_STORAGE' is not a valid storage ID.\n\n\n" && \
+    die "'$STORAGE' is not a valid storage ID."
-exit 1
+pvesm status -content images -storage $STORAGE >&/dev/null ||
-} && \
+    die "'$STORAGE' does not allow 'Disk image' to be stored."
-STORAGE_TYPE=`pvesm status -storage local | awk 'NR>1 {print $2}'` && \
+STORAGE_TYPE=`pvesm status -storage $STORAGE | awk 'NR>1 {print $2}'`
 VMID=$(cat<<EOF | python3
 import json
 with open('/etc/pve/.vmlist') as vmlist:
-        vmids = json.load(vmlist)
+    vmids = json.load(vmlist)
 if 'ids' not in vmids:
-        print(100)
+    print(100)
 else:
-        last_vm = sorted(vmids['ids'].keys())[-1:][0]
+    last_vm = sorted(vmids['ids'].keys())[-1:][0]
-        print(int(last_vm)+1)
+    print(int(last_vm)+1)
 EOF
-) && \
+)
 echo -e "\n\n\n" \
-        "********************************\n" \
+    "********************************\n" \
-        "*  Getting latest HassOS Info  *\n" \
+    "*  Getting latest HassOS Info  *\n" \
-        "********************************\n" && \
+    "********************************\n"
 URL=$(cat<<EOF | python3
 import requests
 url = 'https://api.github.com/repos/home-assistant/hassos/releases/latest'
 r = requests.get(url).json()
 if 'message' in r:
-        exit()
+    exit()
 for asset in r['assets']:
-        if asset['name'].endswith('vdi.gz'):
+    if asset['name'].endswith('vdi.gz'):
-                print(asset['browser_download_url'])
+        print(asset['browser_download_url'])
 EOF
-) && \
+)
 if [ -z "$URL" ]; then
-        echo "Github has returned an error. A rate limit may have been applied to your connection. Please wait a while, then try again."
+    die "Github has returned an error. A rate limit may have been applied to your connection."
-        exit 1
+fi
-fi && \
 echo -e "\n\n\n" \
-        "********************************\n" \
+    "********************************\n" \
-        "*      Downloading HassOS      *\n" \
+    "*      Downloading HassOS      *\n" \
-        "********************************\n" && \
+    "********************************\n"
-wget -q $URL && \
+wget -q --show-progress $URL
-FILE=$(basename $URL) && \
+FILE=$(basename $URL)
 echo -e "\n\n\n" \
-        "********************************\n" \
+    "********************************\n" \
-        "*      Extracting HassOS       *\n" \
+    "*      Extracting HassOS       *\n" \
-        "********************************\n" && \
+    "********************************\n"
-gunzip -f $FILE && \
+gunzip -f $FILE
 echo -e "\n\n\n" \
-        "********************************\n" \
+    "********************************\n" \
-        "*       Creating new VM        *\n" \
+    "*       Creating new VM        *\n" \
-        "********************************\n" && \
+    "********************************\n"
-VMDISK=vm-${VMID}-disk-
+if [ "$STORAGE_TYPE" = "dir" ]; then
-VMDISK_EXT=`if [ "$STORAGE_TYPE" = "dir" ];then echo ".qcow2";fi`
+    DISK_EXT=".qcow2"
-VMDISK_REF=`if [ "$STORAGE_TYPE" = "dir" ];then echo "${VMID}/";fi`${VMDISK}
+    DISK_REF="$VMID/"
+    IMPORT_OPT="-format qcow2"
+fi
+for i in {0,1}; do
+    disk="DISK$i"
+    eval DISK${i}=vm-${VMID}-disk-${i}${DISK_EXT}
+    eval DISK${i}_REF=${STORAGE}:$DISK_REF${!disk}
+done
 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 && \
+    -net0 virtio,bridge=vmbr0 -onboot 1 -ostype l26 -scsihw virtio-scsi-pci
-pvesm alloc $VM_STORAGE $VMID ${VMDISK}0${VMDISK_EXT} 128 1>&/dev/null && \
+pvesm alloc $STORAGE $VMID $DISK0 128 1>&/dev/null
-qm importdisk $VMID ${FILE%".gz"} $VM_STORAGE `if [ ! -z $VMDISK_EXT ];then echo "-format qcow2";fi` 1>&/dev/null && \
+qm importdisk $VMID ${FILE%".gz"} $STORAGE $IMPORT_OPT 1>&/dev/null
-qm set $VMID -bootdisk sata0 -efidisk0 ${VM_STORAGE}:${VMDISK_REF}0${VMDISK_EXT},size=128K \
+qm set $VMID -bootdisk sata0 -efidisk0 ${DISK0_REF},size=128K \
-        -sata0 ${VM_STORAGE}:${VMDISK_REF}1${VMDISK_EXT},size=6G > /dev/null && \
+    -sata0 ${DISK1_REF},size=6G > /dev/null
 echo -e "\n\n\n" \
-        "********************************\n" \
+    "********************************\n" \
-        "*    Completed Successfully    *\n" \
+    "*    Completed Successfully    *\n" \
-        "*       New VM ID is $VMID       *\n" \
+    "*       New VM ID is \e[1m$VMID\e[0m       *\n" \
-        "********************************\n"
+    "********************************\n"