#!/bin/bash


cowsay -p 'Good morning sir. May I ask you to hand me over a yummy Pi image to digest?'
echo -e "\n"


lsblk -d | grep "^sd*"

read -r -p "Enter Device to copy from (sdx):" SD

read -r -p "Enter Image name (no space): " IMGNAME
while [[ -e $HOME/Images/${IMGNAME}.img ]];do
    read -r -p  "Image already exists please enter new name: " IMGNAME

done
echo  -e "\n\nSD Card IMG from:\n"
lsblk -d | grep "${SD}"

echo -e "\n will be written to:\n\n$HOME/Images/${IMGNAME}.img\n\n"

read -r -p "Are you shure?[y/N]" response
case "$response" in
    [yY][eE][sS]|[yY])
        echo "WRITING IMAGE...."
        sudo mkdir -p $HOME/Images 
        sudo dd if=/dev/${SD} of=$HOME/Images/${IMGNAME}.img bs=4M status=progress && sudo sync

        
        echo "Shrinking IMAGE....."

        sudo bash pishrink $HOME/Images/${IMGNAME}.img

        echo "Packing IMAGE....."

        sudo gzip $HOME/Images/${IMGNAME}.img

            

        ;;

    *)
        echo "No image was written"
        ;;
esac


$SHELL
