#!/bin/bash
# Script to copy data on usb keys

# dependencies : aptitude install ncurses-bin parted dosfstools
# mlabel is now useless

# text color in bash
normal=$(tput sgr0)
red=$(tput setaf 1)
green=$(tput setaf 2)

# dependencies check
i=0
for arg in "tput" "parted" "mkdosfs"; do
	hash "${arg}"
	if [[ $? -gt 0  ]]; then
		printf "%40s\n" "${red}Error: Could not find \"$arg\" application.${normal}";
		i=$((i+1))
	fi
done
[[ $i -eq 0 ]] || exit $i

# this part depends of the event
keylabel="CHOCOLAS"                      # label of the keys
wherethefilesare="/ipr/home.ipr/jegardai/chocolas"         # where are the files you need to copy
badkeyid="abcd"                                # depends of key models (lsusb)
goodkeyid="058f"

# all our usb keys have the same ID, we decided to write only on these keys
keysearchbadid=$(lsusb | grep $badkeyid | wc -l)
keysearchgoodid=$(lsusb | grep $goodkeyid | wc -l)
keysearchid=$(($keysearchbadid+$keysearchgoodid))
mountpath="/mnt/usbkey"

# MD5SUM check
sourcemd5="/tmp/source_directory.md5"
destmd5="/tmp/destination_directory.md5"

case $keysearchid in
	0) printf "%40s\n" "${red}Alert: No usb key detected${normal}" ;;
	1|2) # we only have 1 or 2 usb ports
		printf "%40s\n" "${green}At least one key detected: Here we go${normal}"
		# si lsusb et ls /dev/sd* ne donnent pas le même nombre de clés → erreur
		nbsd=$(ls /dev/sd* | grep -v sda | grep -v 1 | grep -v 2 | grep -v 3 | wc -l)
		if [[ $keysearchid != $nbsd ]]; then
			printf "%40s\n" "${red}Alert: Error 1${normal} Keysearchid: ${keysearchid}"
		else # if everything is ok
			# Generate the md5sum for source directory
			pushd "${wherethefilesare}"
			find . -type f -exec md5sum {} \; | sort > "${sourcemd5}"
			popd

			for i in $(seq 1 $keysearchid) # from 1 to $keysearchid by step of 1
				# if $keysearchid=1: for i in 1
				# if $keysearchid=2: for i in 1 2
			do
				[[ $i == 1 ]] && device="sdb" || device="sdc" # if 3 or more keys this line has to be adapt
				parted -s /dev/"${device}" mklabel msdos
				parted /dev/"${device}" mkpart -a optimal primary fat32 0% 100%
				mkdosfs -F 32 -I -n $keylabel /dev/"${device}"1
				# fatlabel /dev/${device}1 $keylabel # Cette commande est remplacée par l’option -n de la commande précédente
				mkdir "${mountpath}""${device}"
				mount /dev/"${device}"1 "${mountpath}""${device}"
				cp -r $wherethefilesare/* "${mountpath}""${device}"/
				sync
				# Generate the md5sum for the mountpoint
				pushd "${mountpath}""${device}"
				find . -type f -exec md5sum {} \; | sort > "${destmd5}"
				popd
				# Umount the key
				umount "${mountpath}""${device}"
				rmdir "${mountpath}""${device}"
				# Compare the md5sum
				diff -q "${sourcemd5}" "${destmd5}"
				if [[ $? -gt 0  ]]; then
					printf "%40s\n" "${red}Error: MD5 diff detected.${normal}";
					exit 1
				else
					printf "%40s\n" "${green}MD5 check OK for ${device}.${normal}"
				fi
			done
			printf "%40s\n" "${green}DONE${normal}"
		fi ;;
	*) printf "%40s\n" "${red}Alert: Error 2${normal}" ;;
esac

exit 0