#!/bin/bash
# shellcheck source=kzcommon.sh
#-------------------------------------------------------------------------------
# USB-sticks maken.
#
# Geschreven door Karel Zimmer <info@karelzimmer.nl>.
#
# Auteursrecht © 2014-2020 Karel Zimmer.
# GNU Algemene Publieke Licentie <https://www.gnu.org/licenses/gpl.html>.
#
# RelNum=17.02.08
# RelDat=2020-10-15
# RelMsg="Ontkenning in 'if' verplaatst"
#-------------------------------------------------------------------------------

#-------------------------------------------------------------------------------
# Settings
#-------------------------------------------------------------------------------
source "$(dirname "$0")"/kzcommon.sh

#-------------------------------------------------------------------------------
# Global constants
#-------------------------------------------------------------------------------
readonly DEVICE_DEFAULT=/dev/sdb
readonly SOURCE_DEFAULT=/home/${SUDO_USER:-$USER}/Downloads
readonly RUN_AS_SUPERUSER=true

# Bij aanpassingen ook .completion aanpassen!
readonly OPTIONS_SHORT=$OPTIONS_SHORT_COMMON's:t:'
readonly OPTIONS_LONG=$OPTIONS_LONG_COMMON'source:,target:'
readonly USAGE="Gebruik: $PROGNAME [-s|--source=BRON] [-t|--target=DOEL]
                 $OPTIONS_USAGE_COMMON MAP

$OPTIONS_LONG_SHORT

$ARGUMENT_LONG_SHORT"
readonly HELP="Gebruik: $PROGNAME [OPTIE...] [--] MAP

USB-sticks maken.

Opties:
  $OPTIONS_LONG_SHORT

  $ARGUMENT_LONG_SHORT

  -s --source=BRONMAP Gebruik beeldbestanden in de BRONMAP
  -t --target=APPARAAT
                      Gebruik de opgegeven USB-stick-device
$OPTIONS_HELP_COMMON

Argument:
  MAP                 Gebruik beelbestanden (.iso) in opgegeven map (verplicht)"

#-------------------------------------------------------------------------------
# Global variables
#-------------------------------------------------------------------------------
declare ARGUMENT_SOURCE=false
declare DEVICE=''
declare NO_FILES_FOUND=true
declare OPTION_SOURCE=false
declare OPTION_TARGET=false
declare SOURCEDIR=''
declare SOURCE_ARGUMENT=''
declare TARGET_ARGUMENT=''

#-------------------------------------------------------------------------------
# Functions
#-------------------------------------------------------------------------------
check_input() {
    local -i getopt_rc=0
    local parsed=''

    set +o errexit
    parsed=$(
        getopt  --alternative                       \
                --options       "$OPTIONS_SHORT"    \
                --longoptions   "$OPTIONS_LONG"     \
                --name          "$PROGNAME"         \
                -- "$@"
        )
    getopt_rc=$?
    set -o errexit
    if [[ $getopt_rc -ne 0 ]]; then
        printf '%s\n' "$USAGELINE" >&2
        QUIET=true exit $ERROR
    fi
    eval set -- "$parsed"
    process_general_options "$@"

    while true; do
        case $1 in
            -s|--source)
                if $OPTION_SOURCE; then
                    printf  "$PROGNAME: %s\n%s\n"   \
                            "optie '$1' éénmaal opgeven" "$USAGELINE" >&2
                    QUIET=true exit $ERROR
                else
                    OPTION_SOURCE=true
                    SOURCE_ARGUMENT=$2
                fi
                shift 2
                ;;
            -t|--target)
                if $OPTION_TARGET; then
                    printf  "$PROGNAME: %s\n%s\n"   \
                            "optie '$1' éénmaal opgeven" "$USAGELINE" >&2
                    QUIET=true exit $ERROR
                else
                    OPTION_TARGET=true
                    TARGET_ARGUMENT=$2
                fi
                shift 2
                ;;
            --)
                shift
                break
                ;;
            *)
                shift
                ;;
        esac
    done

    while [[ "$*" ]]; do
        if $ARGUMENT_SOURCE; then
            printf  "$PROGNAME: %s\n%s\n"   \
                    "argument 'MAP' éénmaal opgeven" "$USAGELINE" >&2
            QUIET=true exit $ERROR
        else
            ARGUMENT_SOURCE=true
            SOURCE_ARGUMENT=$1
        fi
        shift
    done

    if $OPTION_SOURCE; then
        SOURCEDIR=$SOURCE_ARGUMENT
    else
        SOURCEDIR=$SOURCE_DEFAULT
    fi
    if ! [[ -d $SOURCEDIR ]]; then
        printf  "$PROGNAME: %s\n%s\n"   \
                "map $SOURCEDIR bestaat niet" "$USAGELINE" >&2
        QUIET=true exit $ERROR
    fi

    if $OPTION_TARGET; then
        DEVICE=$TARGET_ARGUMENT
    else
        DEVICE=$DEVICE_DEFAULT
    fi

    # Start in grafische modus; Starter: Terminal=true, CLI: optie gui.
    if $OPTION_GUI; then
        OPTION_GUI=false
        TERMINAL=true
    fi

    check_user
    request_input
}

request_input() {
    local msg='Gebruik optie --target om een ander device op te geven.'

    warning 'De volgende /dev/sd schijven en partities zijn aanwezig:'
    if  findmnt --list                  \
                --output=SOURCE,TARGET  |
        grep    --regexp='/dev/sd'; then
        warning "
---------------------------------------------------
'$DEVICE' wordt gebruikt als doel en OVERSCHREVEN!
---------------------------------------------------"
        while true; do
            read -rp 'Is dit OK? [j/N]: ' < /dev/tty
            case $REPLY in
                j*|J*)
                    break
                    ;;
                n*|N*|'')
                    printf '%s\n%s\n%s\n' "$msg" "$USAGELINE" 'Gestopt.'
                    QUIET=true exit $SUCCESS
                    ;;
                *)
                    printf '%b\n' "${UP_ONE_LINE}${ERASE_LINE}"
                    continue
                    ;;
            esac
        done
    else
        warning "
--------------------------------------------------
Geen '$DEVICE' gevonden om te gebruiken als doel!
--------------------------------------------------"
        printf '%s\n%s\n%s\n' "$msg" "$USAGELINE" 'Gestopt.'
        QUIET=true exit $WARNING
    fi
}

process_input() {
    local -i isosize=0
    local usbname=''
    local file=''

    NO_FILES_FOUND=true

    for file in "$SOURCEDIR"/*.iso; do

        if ! [[ -f "$file" ]]; then
            continue
        fi

        NO_FILES_FOUND=false
        usbname=$(basename "$file" .iso | tr '-' ' ')

        printf "\n${BOLD}%s${NORMAL}\n" "$usbname"

        while true; do
            read -rp 'maken/Overslaan/stoppen? [m/O/s]: ' < /dev/tty
            case $REPLY in
                m*|M*)
                    isosize=$(
                        du  --apparent-size \
                            --block-size=1  \
                            "$file"         |
                        awk '{print $1}'
                        )
                    create_usb_stick "$file"
                    while true; do
                        read    -rp                                           \
                                "Nog een USB-stick '$usbname' maken? [j/N]: " \
                                < /dev/tty
                        case $REPLY in
                            j*|J*)
                                create_usb_stick "$file"
                                continue
                                ;;
                            n*|N*|'')
                                printf '%s\n' 'Klaar.'
                                break
                                ;;
                            *)
                                printf '%b\n' "${UP_ONE_LINE}${ERASE_LINE}"
                                continue
                                ;;
                        esac
                    done
                    break
                    ;;
                o*|O*|'')
                    printf '%s\n' 'Overgeslagen.'
                    break
                    ;;
                s*|S*)
                    printf '%s\n' 'Gestopt.'
                    QUIET=true exit $SUCCESS
                    ;;
                *)
                    printf '%b\n' "${UP_ONE_LINE}${ERASE_LINE}"
                    continue
                    ;;
            esac
        done
    done

    if $NO_FILES_FOUND; then
        warning "Geen beeldbestanden (.iso) gevonden in map '$SOURCEDIR'."
        QUIET=true exit $WARNING
    fi
}

create_usb_stick() {
    local dd_from=${1:-dd_from?}

    read    -rp                                                         \
            "Plaats USB-stick '$usbname', gevolgd door Enter [Enter]: " \
            < /dev/tty
    if [[ -e $DEVICE ]]; then
        info "USB-stick $usbname wordt gemaakt ..."

        dd  if="$dd_from"       \
            status=none         |
        pv  --size="$isosize"   |
        dd  of=$DEVICE          \
            bs=4M               \
            status=none

        info "Gegevens schrijven naar $usbname (dit kan even duren) ..."
        sync

        info "USB-stick '$usbname' is gemaakt."

        info "Ter controle is er een Kernel-based Virtuele Machine (KVM) \
gestart."
        if ! qemu-system-x86_64 -enable-kvm     \
                                -m 2048         \
                                -hda $DEVICE    |& $LOGCMD; then
            true
        fi

        if ! umount "$DEVICE" 2> /dev/null; then
            read -rp \
            "Ontkoppel de USB-stick '$usbname', gevolgd door Enter [Enter]: "  \
            < /dev/tty
        fi
        read -rp \
             "Verwijder de USB-stick '$usbname', gevolgd door Enter [Enter]: " \
             < /dev/tty
    else
        warning "USB-stick '$usbname' is niet aanwezig op $DEVICE."
        QUIET=true exit $WARNING
    fi
}

term_script() {
    QUIET=true exit $SUCCESS
}

process_script() {
    check_input "$@"
    process_input
}

#-------------------------------------------------------------------------------
# Main line
#-------------------------------------------------------------------------------
main() {
    init_script
    process_script "$@"
    term_script
}


main "$@"

# EOF