#!/bin/bash

VERSION=1.0

DATE=`date +%Y%m%d`
COUNT=1
TITLE=""

RSYNC=rsync
#RSYNCFLAGS="-arptgoA --info=progress2,flist --human-readable"
RSYNCFLAGS="-arpt --info=progress2,flist --human-readable"

CP=cp
CPFLAGS=-Rpfv

# override default...
COPY=$RSYNC
COPYFLAGS=$RSYNCFLAGS
VERIFY=$RSYNC
VERIFYFLAGS=-n $RSYNCFLAGS


COMPRESSOR=./compress-archive.sh
COMPRESS=1


# Config file to contain all the default settings...
# XXX not sure if this is a good idea...
#	...should we also check file sec?
##CONFIG=.sync-flash.rc
##if ! [ -z $CONFIG ] && [ -e ~/$CONFIG ] ; then
##	# XXX executing an external file...
##	source ~/$CONFIG
##fi


# base mount dir...
# systems with /mnt
if [ -d /mnt ] ; then
	BASE=/mnt

# raw Cygwin
elif [ -d /cygdrive ] ; then
	BASE=/cygdrive

# OSX
elif [ -d /Volumes ] ; then
	BASE=/Volumes
fi


while true ; do
	case "$1" in
		-h|-help|--help)
			echo "usage: `basename $0` FLAGS DRIVE [TITLE]"
			echo
			echo "	-h|-help	print this message and exit."
			echo "	-m|-multi	single base, multiple sub dirs"
			echo "			for multiple flash cards in a"
			echo "			single shoot."
			echo "	-l|-last	last flash card in set, run"
			echo "			process-archive.sh after copying."
			echo "	-b|-base BASE	the base dir to look for drives in"
			echo "			default: $BASE"
			echo "	--rsync		use rsync (default)"
			echo "	--cp		use cp"
			if ! [ -z $COMPRESSOR ] ; then
				echo "	--compress	toggle archive compression"
				echo "			default: `[[ $COMPRESS ]] && echo "on" || echo "off"`"
			fi
			# notes...
			echo
			if ! [ -z $COMPRESSOR ] ; then
				echo "NOTE: the index is fully usable during the compression stage"
			fi
			echo "NOTE: cp under Cygwin may messup permissions, use rsync."
			echo
			exit
			;;

		-i|--interactive)
			INTERACTIVE=1
			shift
			;;
		-m|-multi|--multi)
			MULTI=1
			shift
			;;
		-l|-last|--last)
			LAST=1
			shift
			;;
		-b|-base|--base)
			BASE=$2
			shift 2
			;;
		-cp|--cp)
			COPY=cp
			COPYFLAGS=-Rpfv
			shift
			break
			;;
		-rsync|--rsync)
			COPY=$RSYNC
			COPYFLAGS=$RSYNCFLAGS
			shift
			break
			;;
		-compress|--compress)
			COMPRESS=`[[ $COMPRESS ]] && echo "" || echo 1`
			shift
			break
			;;
		*)
			break
			;;
	esac
done

if ! [ -z "$2" ] ; then
	TITLE=" - $2"
fi

BASE=${BASE%/}/
DRIVE=${1}

__BASE=$BASE
while true ; do
	BASE=$__BASE
	if [[ $INTERACTIVE || ! $DRIVE ]] ; then
		INTERACTIVE=1
		echo "Select/toggle an option:"
		if [ -z $MULTI_STARTED ] ; then
			if [[ $MULTI ]] ; then
				echo "0) Multiple flash cards"
			else
				echo "0) Single flash card"
			fi
		else
			echo "0) Build after this flash card: `[[ $LAST ]] && echo "yes" || echo "no"`"
		fi
		echo "1) Directoy description is: \"$TITLE\""
		if [[ ! $DRIVE ]] ; then
			echo "a-z|name) Type a drive letter, mount name in $BASE or path and start"
			echo "          (paths must start with \"/\", \"./\" or \"[A-Z]:\")"
		else
			echo "a-z|name) Type a drive letter, mount name in $BASE or path and start"
			echo "          (paths must start with \"/\", \"./\" or \"[A-Z]:\")"
			echo "Enter) Copy drive ${DRIVE}"
		fi
		echo "2) Build"
		if ! [ -z $COMPRESSOR ] ; then
			echo "3) Compresion is `[[ $COMPRESS ]] && echo "on" || echo "off"`"
			echo "4) Quit"
		else
			echo "3) Quit"
		fi
		read -ep ": " RES
	
		# NOTE: we can't use letters here as they will shadow 
		# 	with drive letters...
		case $RES in
			# toggle multi mode...
			0)
				if [ -z $MULTI_STARTED ] ; then
					MULTI=`[[ ! $MULTI ]] && echo 1 || echo ""`
				else
					LAST=`[[ ! $LAST ]] && echo 1 || echo ""`
				fi
				continue
				;;
			1)
				read -i "${TITLE# - }" -ep "new description: " TITLE
				TITLE=" - $TITLE"
				continue
				;;
			# continue with same drive or ask again...
			"")
				if [[ ! $DRIVE ]] ; then
					echo "ERR: need a drive to copy from, no defaults."
					echo
					continue
				fi
				DRIVE=$DRIVE
				;;
			2)
				LAST=1
				break
				;;
			3)
				if ! [ -z $COMPRESSOR ] ; then
					COMPRESS=`[[ ! $COMPRESS ]] && echo 1 || echo ""`
				else
					exit
				fi
				continue
				;;
			4)
				exit
				;;

			# new drive letter...
			*)
				DRIVE=$RES
				;;
		esac
	fi

	# explicit path given...
	if [[ "${DRIVE::1}" == "/" ]] \
			|| [[ "${DRIVE::2}" == "./" ]] \
			|| [[ "${DRIVE::2}" =~ [a-zA-Z]: ]] \
			&& [ -e "$DRIVE" ] ; then
		BASE=
	fi

	# sanity check...
	if ! [ -e "${BASE}${DRIVE}" ] ; then
		echo
		echo "ERR: ${BASE}${DRIVE}: does not exist, nothing to copy."
		echo
		if [[ $INTERACTIVE || ! $DRIVE ]] ; then
			continue
		fi
		exit
	fi

	# XXX do a real three digit count...
	# single flash card...
	SCOUNT=`printf "%03d" $COUNT`
	if [ -z $MULTI ] ; then
		DIR="${DATE}.${SCOUNT}${TITLE}"
		while [ -e *"$DIR"* ] ; do
			COUNT=$((COUNT+1))
			SCOUNT=`printf "%03d" $COUNT`
			DIR="${DATE}.${SCOUNT}${TITLE}"
		done
		BASE_DIR=$DIR

	# multiple flash cards shoot...
	else
		BASE_DIR="${DATE}${TITLE}/"
		DIR="${BASE_DIR}/${DATE}.${SCOUNT}"
		# get next dir index...
		while [ -e *"$DIR"* ] ; do
			COUNT=$((COUNT+1))
			SCOUNT=`printf "%03d" $COUNT`
			DIR="${BASE_DIR}/${DATE}.${SCOUNT}"
		done
	fi

	MULTI_STARTED=1

	# normalize paths...
	BASE_DIR="./- ${BASE_DIR}/"
	DIR="./- $DIR/"

	mkdir -vp "$DIR"

	while true ; do
		echo "Copying files from ${BASE}${DRIVE} (~`du -hs "${BASE}${DRIVE}" | cut -f 1`)..."
		$COPY $COPYFLAGS ${BASE}${DRIVE}/* "$DIR" \
			2> >(tee "${DIR}"/copy-err.log)
		# no errors -> remove log...
		if ! [ -s "${DIR}/copy-err.log" ] ; then
			rm -f "${DIR}"/copy-err.log
		fi
		echo "Copying files: done."

		# verify copy...
		# XXX make this more generic...
		if ! [ -z $VERIFY ] ; then
			echo "Verifying copied files..."
			$VERIFY $VERIFYFLAGS ${BASE}${DRIVE}/* "$DIR" \
				> >(tee "${DIR}"/verification-err.log)
			if ! [ -s "${DIR}/verification-err.log" ] ; then
				rm -f "${DIR}"/verification-err.log
			else
				echo
				echo "WARNING: found mismatching files"
				echo "	(see: "${DIR}"/verification-err.log)"
				echo
				while true; do
					read -ep "[R]etry, [c]ontinue, or Ctrl-C to cancel: " ACTION
					ACTION=`echo ${ACTION,,} | xargs`
					if [[ $ACTION =~ [rc] ]] \
							|| [ -z  $ACTION ] ; then
						break
					fi
					echo "Unknown input: \"$ACTION\""
				done
				if [[ $ACTION == "c" ]] ; then
					break
				else
					continue
				fi
			fi
			echo "Verifification: done."
			break
		# no verification defined...
		else
			break
		fi
	done

	# exit interactive mode...
	if [[ ! $MULTI || ! $INTERACTIVE || $LAST ]] ; then
		break
	fi
done

if [[ $LAST ]] ; then
	COMMON_FLAG=-c
fi

if [[ ! $MULTI || $LAST ]] ; then
	echo "Building archive..."
	./process-archive.sh $COMMON_FLAG "$BASE_DIR"
	echo "Building archive: done."
fi

if [[ $COMPRESS ]] ; then
	echo "Compressing archive..."
	${COMPRESSOR} "$BASE_DIR"
	echo "Compressing archive: done."
fi


echo "`basename "$0"`: done."

# vim:set nowrap :