#!/usr/bin/env bash
# SPDX-License-Identifier: GPL-3.0-or-later
# archboot-restore-usbstick.sh - restore usbstick to FAT32
# by Tobias Powalowski <tpowa@archlinux.org>
. /usr/lib/archboot/common.sh
_usage()
{
    echo -e "\e[1m\e[36mARCHBOOT\e[m \e[1m- Restore USB Stick\e[m"
    echo -e "\e[1m----------------------------\e[m"
    echo -e "This script restores an USB device to a \e[1mFAT32\e[m device."
    echo -e "\e[1m\e[91mWARNING: ALL DATA WILL BE LOST ON THE DEVICE! \e[m"
    echo ""
    echo -e "Usage: \e[1m${_BASENAME} <device>\e[m"
    exit 0
}
##################################################
[[ -z "${1}" ]] && _usage
_root_check
if [[ -b "${1}" ]]; then
    echo -e "\e[91mWARNING: 10 seconds for hitting CTRL+C to stop the process on ${1} now! \e[m"
    sleep 10
    # clean partitiontable
    echo -e "\e[1mRestoring USB STICK...\e[m"
    echo -e "\e[1mSTEP 1/3:\e[m Cleaning partition table..."
    dd if=/dev/zero of="${1}" bs=512 count=2048
    wipefs -a "${1}"
    # create new MBR and partition on <DEVICE>
    echo -e "\e[1mSTEP 2/3:\e[m Create new MBR and partitiontable..."
    fdisk "${1}" <<EOF
n
p
1


t
b
w
EOF
    # wait for partitiontable to be resynced
    sleep 5
    # create FAT32 filesystem on <device-partition>
    echo -e "\e[1mSTEP 3/3:\e[m Create FAT32 filesystem..."
    mkfs.vfat -F32 "${1}"1
    echo -e "\e[1mFinished.\e[m"
else
    echo -e "\e[1m\e[91mError: ${1} not a valid blockdevice! \e[m"
fi