#!/bin/bash

set -e
set -x

source ./constraint.sh

SRC=${HOME}/src
DST=${HOME}/dst

TEMP=${DST}/temp
ZIPTEMP=${DST}/ziptemp
LOGS=${DST}/logs


clean(){
    echo Cleaning.
    ${SUDO} ${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} -t vfat -o user,noexec,nosuid,nodev,rw,uid=`${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_*

    # mount temp 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} -o noexec,nosuid,nodev -r $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

        #if [ $COPYDIRTYPDF -eq 1 ]; then
        #    pdfCopyDirty $SRC $targetDir
        #else
        #    pdfCopyClean $SRC $targetDir
        #fi

        # copy stuff
        #copySafeFiles $SRC $targetDir
        #convertCopyFiles $SRC $targetDir $TEMP
        #rm -rf ${TEMP}/*

        # unpack and process archives
        #unpackZip $SRC $targetDir $TEMP
    fi
    let PARTCOUNT=$PARTCOUNT+1
done

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