#!/bin/bash

set -e
#set -x

source ./constraint.sh
if ! [ "${ID}" -ge "1000" ]; then
    echo "This script cannot run as root."
    exit
fi

source ./functions.sh

clean(){
    echo Cleaning.
    ${SYNC}

    # Cleanup source
    ${SUDO} ${UMOUNT} ${SRC} || true
    rm -rf ${SRC}

    # Cleanup destination
    rm -rf ${TEMP}
    rm -rf ${ZIPTEMP}
    ${SUDO} ${UMOUNT} ${DST} || true
    rm -rf ${DST}

    exit
}

trap clean EXIT TERM INT

# De we have a source device
if [ ! -b ${DEV_SRC} ]; then
    echo "Source device (${DEV_SRC}) does not exists."
    exit
fi
# Find the partition names on the source device
DEV_PARTITIONS=`ls "${DEV_SRC}"* | grep "${DEV_SRC}[1-9][0-6]*" || true`
if [ -z ${DEV_PARTITIONS} ]; then
    echo "${DEV_SRC} does not have any partitions."
    exit
fi

# Do we have a destination device
if [ ! -b ${DEV_DST} ]; then
    echo "Destination device (${DEV_DST}) does not exists."
    exit
fi

# Prepare mount points
if [ ! -d ${SRC} ]; then
    mkdir ${SRC}
fi
if [ ! -d ${DST} ]; then
    mkdir ${DST}
fi

# mount and prepare destination device
if ${MOUNT}|grep ${DST}; then
    ${SUDO} ${UMOUNT} ${DST} || true
fi
# uid= only works on a vfat FS. What should wedo if we get an ext* FS ?
${SUDO} ${MOUNT_DST} ${ID} ${DEV_DST} ${DST}
if [ ${?} -ne 0 ]; then
    echo "Unable to mount ${DEV_DST} on ${DST}"
    exit
else
    echo "Target USB device (${DEV_DST}) mounted at ${DST}"
    rm -rf "${DST}/FROM_PARTITION_"*

    # prepare temp dirs and make sure it's empty
    mkdir -p "${TEMP}"
    mkdir -p "${ZIPTEMP}"
    mkdir -p "${LOGS}"

    rm -rf "${TEMP}/"*
    rm -rf "${ZIPTEMP}/"*
    rm -rf "${LOGS}/"*
fi

# Groom da kitteh!

COPYDIRTYPDF=0
PARTCOUNT=1
for partition in ${DEV_PARTITIONS}
do
    # Processing a partition
    echo "Processing partition: ${partition}"
    if ${MOUNT}|grep ${SRC}; then
        ${SUDO} ${UMOUNT} ${SRC}
    fi

    ${SUDO} ${MOUNT_SRC} ${partition} ${SRC}
    if [ ${?} -ne 0 ]; then
        echo "Unable to mount ${partition} on ${SRC}"
    else
        echo "${partition} mounted at ${SRC}"

        # Print the filenames on the current partition in a logfile
        find "${SRC}" -fls "${LOGS}/${PARTCOUNT}"

        # create a directory on ${DST} named PARTION_$PARTCOUNT
        target_dir="${DST}/FROM_PARTITION_${PARTCOUNT}"
        echo "copying to: ${target_dir}"
        mkdir -p "${target_dir}"

        main ${target_dir}

        ls -lR "${target_dir}"
    fi
    let PARTCOUNT=`expr $PARTCOUNT + 1`
done

# The cleanup is automatically done in the function clean called when
# the program quits