#!/bin/bash
version="0.6"
versionDate="26.11.2011"

# Hilfetext
function usage() {
	cat <<EOF

Beschreibung:
Kopiert Dateien aus einem Recovery-Verzeichnis nach Datei-Erweiterung sortiert
in entpsrechende Unterverzeichnisse des Ziel-Verzeichnises. Die Sortierung kann
auf bestimmte Erweiterungen und bestimmte Dateigrößen eingeschränkt werden.


Befehls-Syntax:
$scriptname <Recovery-Verzeichnis> <Ziel-Verzeichnis> [-type <Dateierweiterung>
-size <Dateigröße> -query-pdf -logformat <Format>]

Pflichtangaben:

<Recovery-Verzeichnis>	Pfad zum Verzeichnis, in dem sich die Verzeichnisse mit
			wiederhergestellten bzw. gefundenen Dateien befinden.

<Ziel-Verzeichnis>	Pfad zum Verzeichnis, in das die gefundenen Dateien nach
			Erweiterung einsortiert werden sollen.
Optionen:
-type	Dateierweiterung(en) der Dateien, nach den in den wiederhergestellten
	Dateien gesucht und sortiert werden soll.
			
	Es können mehrere Erweiterungen angegeben werden, wobei	eine Liste mit
	Erweiterungen in Anführungszeichen ("") eingeschlossen werden muss.

	Wird "auto" angegeben, so werden die Erweiterungen der Dateien automa-
	tisch ermittelt und entsprechend in das Ziel-Verzeichnis einsortiert.
	"auto" ist auch	die Standardeinstellung, falls type nicht angeben wird.

	Beispiele:
	-type jpg
	-type "jpg mp3 ogg"
	-type auto

-size	n[cbwkMG] Größenangabe oder ein Größenbereich der für die Sortierung
	berücksichtigt werden soll. Bei einem Bereich müssen die Angaben in
	Anführungszeichen ("") eingefasst werden.
	
	Folgende Einheiten können angegeben werden:

	c 	für Byte
	b 	für 512 Byte-Blöcke
        w       für 2 Byte Wörter
	k 	für Kilobyte (Einheit von 1024 Byte)
	M 	für Megabyte (Einheit von 1048576 Byte)
	G 	für Gigabyte (Einheit von 1073741824 Byte)

	Beispiele:
	-size 550k	nur Dateien, die 550 k-Einheiten groß sind.
	-size -2G	nur Dateien, kleiner als 2 G-Einheiten sind.
	-size +500M	nur Dateien, größer 500 M-Einheiten.
	-size "+5M -9M"	Dateien größer 5 M-Einheiten und kleiner 9 M-Einheiten.


-query-pdf	Wird diese Option gesetzt so werden mittels "pdfinfo" Titel
		aus den Meta-Daten des PDF-Dokuments ermittelt und sofern
		der Titel nicht leer ist, als Ergebnis in eine Text-Datei
		geschrieben.

-logformat	<Format> kann "unix" oder "windows" sein. Standardwert ist 
		"unix". Bei "windows" wird am Ende jeder Zeile einer Ergeb-
		nisdatei ein <CR> eingefügt, wodurch sich die Textdateien
		dann auch von Windows aus übersichtlich lesen lassen.

-v --version	Zeigt die Version dieses Skriptes an.

-h --help	Zeigt diesen Hilfe-Text an.
EOF
}

#
# Historie
#
# Version 0.6 vom 26.11.2011
# - Bugfix: "find: unbekannte Option `-size" beseitigt.
# - Bugfix: "Zeile 142: processcopy_Files_ext: Kommando nicht gefunden." beseitigt.
# - Bugfix: Eventuell auftretende Probleme mit Leerzeichen in Datei-Namen beseitigt.
#
#
# Version 0.5 vom 07.04.2011
# - erste veröffentlichte Version
#

# -size Option auswerten. Die Werte sind an die von "find" (siehe "man find")
# angekoppelt.
function check_size() {
	string=$1
	Message="-size Option: $string...ok"
	case ${string:(-1)} in
		[0-9] | b )
			sizeSuffix="512 Byte-Blöcke"
			divisor=512;;
		c )
			sizeSuffix="Byte"
			divisor=1;;
		G )
			sizeSuffix="G"
			divisor=1073741824;;
		k )
			sizeSuffix="k"
			divisor=1024;;
		M )
			sizeSuffix="M"
			divisor=1048576;;
		w )
			sizeSuffix="2 Byte-Wort"
			divisor=2;;
		* )
			Message="FEHLER: -size Angabe entspricht nicht den syntaktischen Vorgaben."
			exit 1;;
	esac
	echo_toScrnLog "$Message"
}

# Ermittelt die Dateierweiterung, der an $1 übergebenen Datei
# Speichert eine Liste der bisher verarbeiteten Erweiterungen
# in der Feld-Variablen $processedExtensions
function get_extension () {
	file="$1"
	extension=$(echo "$file"| awk -F . '{if (NF>1) {print $NF}}')
	if [ -z "$extension" ]; then
		extension="NONE"
	fi
	case "${processedExtensions[@]}" in
		*"$extension"*);;
		*)
			processedExtensions=("${processedExtensions[@]}" "$extension");;
	esac
}

# Ermittel abhängig von den Optionen -type und -size, die Art des Kopiervorgangs.
# Die Optionen werden über die Variable $parameter aus dem Hauptlauf an $1 zuge-
# wiesen.
function copy_files() {
	IFSbak=$IFS
	IFS=$'\n'
	case $1 in
		auto )
			cp "$path/$file" "$fulltargetDir/$file" && cpfileCount=$((cpfileCount+1)) && totalSize=$((totalSize+$(stat -c%s "$file")))
			;;	
		auto_size )
			case $sizeSet in
				1 )
					copyFiles=$(find -maxdepth 1 -name "$file" -size $size1 -type f);;
				2)
					copyFiles=$(find -maxdepth 1 -name "$file" -size $size1 -size $size2 -type f);;
			esac
			process_copyFiles_auto
			;;
		ext )
			copyFiles=$(find -maxdepth 1 -name "*.$extension" -type f)
			process_copyFiles_ext
			;;
		ext_size )
			case $sizeSet in
				1 )
					copyFiles=$(find -maxdepth 1 -name "*.$extension" -size $size1 -type f);;
				2 )
					copyFiles=$(find -maxdepth 1 -name "*.$extension" -size $size1 -size $size2 -type f);;
			esac
			process_copyFiles_ext
			;;
	esac
	IFS=$IFSbak
}

function process_copyFiles_ext() {
	for file in ${copyFiles[@]};
	do
		cp "$file" "$fulltargetDir/$file" && cpfileCount=$((cpfileCount+1)) && totalSize=$((totalSize+$(stat -c%s "$file")))
		eval $MSG_pcpStatus
	done	
}

function process_copyFiles_auto() {
	for file in ${copyFiles[@]};
	do
		cp "$file" "$fulltargetDir/$file" && cpfileCount=$((cpfileCount+1)) && totalSize=$((totalSize+$(stat -c%s "$file")))
	done	
}

# Ergebnisse des Umkopierens zusammentragen
function query_Results() {
	print_toScr "Trage Ergebnisse zusammen..."
	if [ "$type" == "auto" ];
	then
		processedExtensions=$(echo "${processedExtensions[@]}" |tr " " "\n"|sort|tr "\n" " ")
		for extension in $processedExtensions;
		do
			newstring=$extension
			if [ ${#extension} -gt 8 ];
			then
				shorten_Extension
			fi
			if [ "$newstring" != "$extension" ];
			then
				if [ -d "$targetDir/$newstring" ];
				then
					rm -R "$targetDir/$newstring"
				fi
				mv "$targetDir/$extension$sizeLable" "$targetDir/$newstring$sizeLable"
				extension=$newstring
			fi
			fulltargetDir="$targetDir/$extension$sizeLable"
			fileCount=$(ls -1 "$fulltargetDir" | wc -l)
			remove_emptytargetDir
		done
	else
			fileCount=$(ls -1 "$fulltargetDir" | wc -l)
			remove_emptytargetDir
	fi		
}

# Dateierweiterungen, in denen sich das selbe Zeichen vier mal oder mehr wieder-
# holt, kürzen. Das betreffende Zeichen wird nur einmal ausgegeben und dahinter 
# der Wert der Wiederholung. Beispiel:
# "exe____________" wird zu "exe_(12)"
function shorten_Extension() {
	string=$extension
	for character in $(echo $extension | sed 's/\(.\)/\1\n/g');
	do
		case "$processedCharacters" in
			*"$character"*);;
			*)
				processedCharacters="$processedCharacters $character"
				strangestring=$(echo "$extension" | grep -o "\($character\{4,\}\)")
				charcount="${strangestring//[^$character]/}"
				newstring=$(echo $string | sed "s/$character\{4,\}/$character\(${#charcount}\)/")
				string=$newstring
				;;
		esac
	done
}

#Im Zielverzeichnis angelegte Verzeichnisse, die keine Dateien enthalten löschen
function remove_emptytargetDir() {
	case $fileCount in
		0 )				
			rm -R "$fulltargetDir";;
		* )
			get_dirSize;;
	esac
}

# Die Größe eines Verzeichnisses ermitteln
function get_dirSize() {
	dirSize=0
	cd "$fulltargetDir"
	for file in *;
	do
		dirSize=$((dirSize+$(stat -c%s "$file")))
	done
	formatnsave_resultVars
}

# Die Resultate für die Tabellen-Auflistung formatieren
function formatnsave_resultVars() {
	dirSize=$((dirSize/$divisor))
	dirSize=$(echo -e "$dirSize" | sed  -e :a -e 's/^.\{1,12\}$/ & /;ta')
	fileCount=$(echo -e "$fileCount" | sed  -e :a -e 's/^.\{1,19\}$/ & /;ta')
	IFSbak=$IFS
	IFS=$'\n'
	cpResults[${#cpResults[*]}]=$(echo -e "$extension\t\t$fileCount\t\t$dirSize")
	IFS=$IFSbak

}

# Vom Skript erzeugte Text-Dateien mit <CR> am Zeilen-Ende ergänzen
function change_LOGS_toWin() {
	cd "$workDir"
	for log in *;
	do
		if [ -f $log ];
		then
			awk '{printf("%s\r\n",$0)}' $log > "win-$log"
		fi
	done
}

# Sofern PDF-Dateien vorhanden sind, diese nach ausgefüllten Titel-Einträgen
# mittels pdfinfo durchsuchen und die gefundenen Titel in die Datei
# "FoundPDFTitle.txt" zusammen mit dem Titel schreiben. PDFs ohne Titel werden
# in der Datei "FoundPDFnoTitle.txt" gelistet. PDFs, die von pdfinfo nicht ver-
# arbeitet werden konnten werden in "PDFqueryError.txt" aufgeführt.
function query_PDF_Title() {
	print_toScrnLog "Suche nach PDF-Titeln..."
	PDFTitles=$workDir/FoundPDFTitle.txt
	PDFnoTitles=$workDir/FoundPDFnoTitle.txt
	queryERROR=$workDir/PDFqueryError.txt
	tmpqueryERROR=$workDir/tmpError.txt
	errorCount=0
	PDFTitleCount=0
	PDFnoTitleCount=0
	processedPDFCount=0
	if [ -d "$targetDir/pdf" ];
	then
		cd "$targetDir/pdf"
		for file in *;
		do
			Title=""
			pdfinfoResult=$($pdfinfo "$file" 2>$tmpqueryERROR) 
			pdfError=$?
			Title=$(echo "$pdfinfoResult" | grep ^Title: | sed 's/^Title://' | tr -s " " | sed 's/^.//')
			case $pdfError in
				0 )
					if [ "$Title" == "" ];
					then
						echo -e "$file\t$Title" >> $PDFnoTitles
						PDFnoTitleCount=$((PDFnoTitleCount+1))
					else
						echo -e "$file\t$Title" >> $PDFTitles
						PDFTitleCount=$((PDFTitleCount+1))
					fi;;
				* )
					tmpError=$(tail $tmpqueryERROR)					
					errorCount=$((errorCount+1))
					echo -e "$file\t$tmpError" >> $queryERROR
					echo >> $queryERROR
					pdfErrorStatus=1;;
			esac
			processedPDFCount=$((PDFCount+1))
			echo -ne "PDFs: $PDFCount\tTitel:$PDFTitleCount\tohne Titel: $PDFnoTitleCount\tFehler: $errorCount"\\r
		done
	else
		echo_toScrnLog "Es wurden keine PDF-Dateien im Ziel-Verzeichnis gefunden."
	fi
	case $pdfErrorStatus in
		0 )
			echo_toScrnLog "ohne Fehler abgeschlossen.\n";;
		1 )
			echo_toScrnLog "mit $errorCount Fehlern abgeschlossen.\n\nSiehe $queryError";;
	esac
	if [ -e "$tmpqueryERROR" ];
	then
		rm -f "$tmpqueryERROR"
	fi
}

# Alle Unterverzeichnisse des Recovery-Verzeichnisses ermitteln und in der Feld-
# variable $DIRS speichern
function get_DIRS() {
	cd "$recoverDir"
	IFSbak=$IFS
	IFS=$'\n'
	DIRS=($(find "." -maxdepth 1 -mindepth 1 -type d))
	dirCount=${#DIRS[@]}
	IFS=$IFSbak
	for (( index=0; index<${dirCount}; index++ ));
	do
		IFSbak=$IFS
		IFS=$'\n'
		files=($(find "${DIRS[$index]}" -maxdepth 1 -mindepth 1 -type f))
		dirfileCount=${#files[@]}
		IFS=$IFSbak
		totalfileCount=$((totalfileCount+$dirfileCount))
	done
	print_toScrnLog "abgeschlossen.\n"
	echo_toScrnLog "Es wurden $dirCount Verzeichnisse mit insgesamt $totalfileCount Dateien gezählt."
}

function print_toScrnLog() {
	printf "$1" | tee -a $LOG
}

function print_toScr() {
	printf "$1"
}

function echo_toScr() {
	echo "$1"
}

function echo_toScrnLog() {
	echo "$1" | tee -a $LOG	
}

function echo_toLog() {
	echo $1 >> $LOG
}

# Schreibrechte des Benutzers im Ziel-Verzeichnis prüfen
function check_permission() {
	error=$(mkdir "$targetDir" 2>&1)
	if [[ "$error" == "" ]]; then
		rm -R "$targetDir"
		return 0
	elif [[ "$error" =~ "$permError" ]]; then
		echo_toScrnLog "FEHLER: Keine Schreib-Berechtigung für Ziel-Verzeichnis."
		exit 1
	else
		echo_toScrnLog "FEHLER: $error"
		exit 1
	fi
}

# Den Dateinamen dieses Skriptes ermitteln
function get_scriptname() {
	scriptname=$(echo $0 | awk -F / '{print $NF}')
}

# Den Namen der Portokoll-Dateien festlegen
function set_LOG() {
	LogFile="${workDir}/sort_recovered_log"
	rotationCounter=0
	while ( [ -e "${LogFile}${rotationCounter}.txt" ] )
	do 
		rotationCounter=$((rotationCounter+1))
		wait
	done
	LOG="${LogFile}${rotationCounter}.txt"
	cpRESULT="${workDir}/cpRESULT.txt"
}

# Arbeitsverzeichnis unter $HOME anlegen
function set_workdir() {
	workDir="$HOME/Sort_Recovered"
	[ ! -d "$workDir" ] && mkdir "$workDir"
}


################################## Hauptlauf ##################################

recoverDir=""
targetDir=""
echo

# Name des Skriptes ermitteln
get_scriptname

#Versionsinformationen ausgeben.
if [ "$1" == "-v" -o "$1" == "--version" ];
then
	echo_toScr "$scriptname\nVersion: $version\nDatum:\t $versionDate\n"
	exit 0
fi

# Arbeitsverzeichnis und Logdatei setzen
set_workdir
set_LOG


#Skriptaufruf und Parameter prüfen:

# Falls gar nichts mit dem Skriptaufruf übergeben wurde
if [ "$1" == "" ];
then
	echo_toScrnLog "FEHLER: Verzeichnis-Angaben fehlen!"
	usage
	exit 1
fi

# Prüfen, dass erste Angabe ein vorhandenes Verzeichnis ist
print_toScrnLog "Prüfe Recovery-Verzeichnis..."
if [ -d "$1" ];
then
	recoverDir="$1"
	print_toScrnLog "Gefunden!\n"
else

	echo_toScrnLog "FEHLER: Das angegebene Recovery-Verzeichnis "$1" existiert nicht!"
	usage
	exit 1
fi

# Prüfen, dass zweite Angabe übergeben wurde
shift
if [ "$1" == "" ];
then
	echo_toScrnLog "FEHLER: Es wurde kein Ziel-Verzeichnis angegeben!"
	usage
	exit 1
fi

# Prüfen, dass zweite Angabe ein bestehendes Verzeichnis ist und Verzeichnis
# erstellen, falls es nicht existiert
print_toScrnLog "Prüfe Ziel-Verzeichnis..."
if [ -d "$1" ];
then
	targetDir="$1"
	if [ -w "$targetDir" ]; then
		print_toScrnLog "gefunden! Schreibberechtigung besteht.\n"
	else
		echo_toScrnLog "FEHLER: Keine Schreib-Berechtigung für Ziel-Verzeichnis."
		exit 1
	fi
else
	eval $MSG_targetDirnotexist
	targetDir="$1"
	check_permission
	mkdir "$targetDir" && print_toScrnLog "erfolgreich.\n"
fi

# Standardwerte für den Fall setzen, dass Optionen nicht angegeben sind:
type="auto"
sizeSet=0
sizeLable=""
sizeSuffix="k"
divisor=1024	
logFormat="unix"
queryPDF="NO"

# Optionen prüfen
shift
echo_toScrnLog "Prüfe Skript-Parameter..."
echo >> $LOG
	while [ "$1" != "" ];
	do
		Option="$1"
		case $1 in
			-h | --help )
				usage
				exit 0;;
			-type )
				shift
				type=$1
				echo_toLog "Überprüfe -type-Parameter...\ntype-Parameter= $type"
				if [ "$type" == "" ];
				then
					echo_toScrnLog "FEHLER: Angabe für Parameter -type fehlt."
					exit 1
				fi
				[ "$type" != "auto" ] && type=$(echo "$type" |tr " " "\n"|sort|tr "\n" " ")
				echo "\$type=$type" >> $LOG
				echo_toLog "...-type-Parameter ok."
				echo >> $LOG;;
			-size )
				shift
				tempsize=$1
				echo_toLog "-size-Parameter wird gepürft..."
				for item in ${tempsize[*]};
				do
					check_size $item					
					if [ $sizeSet == 0 ];
					then
						size1=$item
						sizeLable="$item"
						sizeSet=1
					else
						size2=$item
						sizeLable=$sizeLable"$item"
						sizeSet=2				
					fi
				done
				echo "\$size=-size $size1 -size $size2" >> $LOG
				echo "\$sizeLable=$sizeLable" >> $LOG
				echo_toLog "-size-Parameter ok!"
				echo >> $LOG
				;;
			-logformat )
				shift
				logFormat="$1"
				;;
			-query-pdf )
				if [ ! -e /usr/bin/pdfinfo ];
				then
					echo_toScrnLog "FEHLER: Das für die Suche nach PDF-Titeln notwendige Programm $pdfinfo existiert nicht."
				else
					pdfinfo="/usr/bin/pdfinfo"
					queryPDF=YES
				fi
				;;
			-* )
				echo_toScrnLog "FEHLER: Ungültige Option: $Option"
				usage
				exit 1
				;;
			* )
				usage
				exit 1;;
		esac
		shift	
	done
echo_toScrnLog "...Skript-Parameter ok!"
echo | tee -a $LOG

# Unterverzeichnisse im angegeben Recovery-Verzeichnis ermitteln
print_toScrnLog "Ermittle Unterverzeichnisse in $recoverDir..."
get_DIRS
echo >> $LOG

# Sortiertes Kopieren starten
echo_toScrnLog "Starte Kopiervorgang..."
Starttime=$(date)
echo_toScrnLog "Start: $Starttime"
echo | tee -a $LOG
echo_toScrnLog "Recovery-Verzeichnis: $recoverDir"
echo_toScrnLog "Ziel-Verzeichnis: $targetDir"
IFSbak=$IFS
IFS=$'\n'
cpResults[0]=$(echo -e "Dateityp\t\tAnzahl Dateien\t\tGröße ($sizeSuffix)")
IFS=$IFSbak
totalSize=0
dirCountlocal=0
cpfileCount=0
if [ "$type" == "auto" ];
then
	for (( index=0; index<${dirCount}; index++ ));
	do
		cd "$recoverDir"
		cd "${DIRS[$index]}"
		path=$(pwd)
		dirCountlocal=$((dirCountlocal+1))
		for file in *;
		do
			get_extension "$file"
			if [ $sizeSet == 0 ];
			then
				fulltargetDir="$targetDir/$extension"
				parameter="auto"
			else
				fulltargetDir="$targetDir/$extension$sizeLable"
				parameter="auto_size"
			fi
			if [ ! -d "$fulltargetDir" ]; then
				mkdir "$fulltargetDir"
			fi
			if [ -f "$file" ]; then
				copy_files $parameter
			fi
			echo -ne "Verzeichnisse: $dirCountlocal\tDateien gesamt: $cpfileCount\tDateitypen: ${#processedExtensions[*]}"\\r
		done
	done
#	echo | tee -a $LOG
#	echo_toScrnLog "...Kopiervorgang abgeschlossen."
#	Endtime=$(date)
#	echo_toScrnLog "Ende: $Endtime"
#	echo | tee -a $LOG
#	query_Results

else
	for extension in $type;
	do
		if [ "$sizeLable" == "" ];
		then
			fulltargetDir="$targetDir"/$extension
		else
			fulltargetDir="$targetDir"/$extension$sizeLable
		fi
		if [ ! -d "$fulltargetDir" ];
		then
			mkdir "$fulltargetDir"
		fi
		index=0
		while [ $index -ne $dirCount ];
		do
			cd "$recoverDir"
			cd "${DIRS[$index]}"
			IFSbak=$IFS
			IFS=$'\n'
			if [ $sizeSet == 0 ]; then
				parameter="ext"
			else
				parameter="ext_size"
			fi
			copy_files $parameter
			index=$((index+1))
			IFS=$IFSbak
		done
#>
	done
fi
		echo | tee -a $LOG
		echo_toScrnLog "...Kopiervorgang abgeschlossen."
		Endtime=$(date)
		echo_toScrnLog "Ende: $Endtime"
		echo | tee -a $LOG
		query_Results
#Ergebnisse auswerten und für Tabellenausgabe aufbereiten
totalSize=$((totalSize/$divisor))
totalSize=$(echo -e "$totalSize" | sed  -e :a -e 's/^.\{1,12\}$/ & /;ta')
cpfileCount=$(echo -e "$cpfileCount" | sed  -e :a -e 's/^.\{1,19\}$/ & /;ta')
IFSbak=$IFS
IFS=$'\n'
cpResults[${#cpResults[*]}]=$(echo -e "Gesamt:\t\t$cpfileCount\t\t$totalSize")
echo
OUTPUT=$(
	for line in ${cpResults[@]};
	do
		echo -e "$line"
	done
	)
IFS=$IFSbak
print_toScr "abgeschlossen:\n\n"
echo -e "$OUTPUT" | column -t -s"$(echo -e "\t")" | tee -a $LOG
echo | tee -a $LOG 

# Nach Titeln in PDF-Meta-Daten suchen, falls "-query-pdf" angegeben wurde.
if [ "$queryPDF" == "YES" ];
then
	echo_toScrnLog "Suche nach PDF-Titeln..."
	Starttime=$(date)
	echo_toScrnLog "Start: $Starttime"
	query_PDF_Title
	echo_toScrnLog "...Suche abgeschlossen."
	Endtime=$(date)
	echo_toScrnLog "Ende: $Endtime"
fi

# LOGS in Windows-txt-Format umwandeln, falls "-logformat windows" angegeben wurde.
if [ "$logFormat" == "windows" ];
then
	change_LOGS_toWin
fi

echo
chown $(basename ~) "$LOG"
chown $(basename ~) "$workDir"
if [ "$queryPDF" == "YES" ];
then
	chown $(basename ~) "$PDFTitles"
	chown $(basename ~) "$PDFnoTitles"
	chown $(basename ~) "$queryERROR"
fi