fud

#!/bin/bash

CONFIGFILE="$HOME/.fud/fud.conf"
LOGFILE="$HOME/.fud/fud.log"
ICON="/usr/share/app-install/icons/fud.png"
MOUNTPOINT="$HOME/.fud/remotedir"

#Colors
RED="\e[1;31m" #Red
GREEN="\e[1;32m" #Green
YELLOW="\e[1;33m" #Yellow
RESET="\e[0m" #Reset!

# NOTA DA RIMUOVERE: questa è una semplice funzione di stampa. Per ora non fa nulla di particolarmente straordinario, ma potrebbe essere utilizzata più in là per sopprimere agevolmente l'output nel terminale, se passata un'opzione come -silent.
# Print on screen
printScreen() { echo -e "$@"; }

# Test if software installation and server connection are ok
testStuff () {
	# Check if $MOUNTPOINT is avaible
	if [[ ! -e $MOUNTPOINT ]]; then
		printScreen $RED"Error: "$RESET"mount point $MOUNTPOINT is not available. Type \`mkdir -p $MOUNTPOINT\` first, then restart this script.";
		exit 1
	fi

	# Check if $MOUNTPOINT is writable
	if [[ ! -w $MOUNTPOINT ]]; then
		printScreen $RED"Error: "$RESET"mount point $MOUNTPOINT is not writable. Type \`chmod -R 755 $MOUNTPOINT\` first, then restart this script.";
		exit 1
	fi
	# Check if inotify is installed
	if ! which inotifywait &> /dev/null; then
		printScreen $RED"Error: "$RESET"install inotify-tools first, then restart this script."
		exit 1
	fi

	# Check if unison is installed
	if ! which unison &> /dev/null; then
		printScreen $RED"Error: "$RESET"install unison first, then restart this script.";
		exit 1
	fi

	# Check if notify-send is installed
	if ! which notify-send &> /dev/null; then
		printScreen $RED"Error: "$RESET"install libnotify-bin first, then restart this script.";
		exit 1
	fi

	# Check if user is in the fuse group
	if ! groups | grep "fuse" &> /dev/null; then
		printScreen $RED"Error: "$RESET"your user is not in the fuse group. Type \`addgroup $USER fuse\` as root first, then logout, login and restart this script.";
		exit 1
	fi	
}

# Mount remote directory in $MOUNTPOINT
mountRemote() {
	while :; do
		# If $MOUNTPOINT is already mounted, skip
		if cat /proc/mounts | grep $MOUNTPOINT &> /dev/null; then
			break;
		# Else, if $MOUNTPOINT is not already mounted
		else		
			# Check for ssh connection
			if [ "$METHOD" == "ssh" ]; then
				# Check if sshfs is installed
				if ! which sshfs &> /dev/null; then
					printScreen $RED"Error:"$RESET" install sshfs first, then restart this script.";
					exit 1
				fi
				# Try to mount the remote directory with sshfs and verify 
				if ! sshfs $LOGIN@$SERVER:$RPATH $MOUNTPOINT -p $PORT -C &> /dev/null; then
					printScreen $RED"Error:"$RESET" impossible to login to ssh server. Check your server configuration and your internet connection first, then restart the script."
					printScreen "Fud retry the connection in 30 seconds."
					sleep 30
				fi
			# Check for ftp connection
			elif [ "$METHOD" == "ftp" ]; then
				# Check if curlftpfs is installed
				if ! which curlftpfs &> /dev/null; then
					printScreen $RED"Error:"$RESET" install curlftpfs first, then restart this script.";
					exit 1
				fi
				# Try to mount the remote directory with curlftpfs and verify
				if ! curlftpfs -v $SERVER:$PORT/$RPATH $MOUNTPOINT -o user=$LOGIN:$PASSWORD -o allow_other -o nonempty; then
					printScreen $RED"Error:"$RESET" impossible to login to ftp server. Check your server configuration and your internet connection first, then restart the script."
					printScreen "Fud retry the connection in 30 seconds."
					sleep 30
				fi
			fi
			# Continue while (and re-check for a connection)
			continue
		fi
	done
}

# Notify how many files have been synchronized
notifySinc() {
	# Check if at least one file has been synchronized
	if [[ "$1" -ge "1" ]]; then
		# If a file (and only one) has been synchronized
		if [[ "$1" -eq "1" ]]; then
			MESSAGE="$1 file have been synchronized"
		# If more files has been synchronized
		elif [[ "$1" -gt "1" ]]; then
			MESSAGE="$1 files have been synchronized"
		fi

		# Notify how many files have been synchronized
		notify-send --icon=$ICON "$MESSAGE"
		printScreen $MESSAGE
	fi
}

# Sync 
syncDir() {
	OUTPUT=$(unison $LPATH $MOUNTPOINT -logfile $LOGFILE $EXTRAOPTIONS -batch -auto -terse -confirmbigdel="false" -retry 3 2>&1)
	# SYNCFILES will contain the number of files synchronized
	SYNCFILES=$(echo `expr match "$OUTPUT" '.*\(([0-9]* item[s]* transferred\)'` | sed -n "s/(//g;s/item[s]* transferred//g;/[0-9]*/p")
	# Notify how many files have been synchronized
	notifySinc $SYNCFILES
}

# SCRIPT STARTS HERE
# Check if config file exists
if [[ -e $CONFIGFILE ]]; then
	# Include config file
	source $CONFIGFILE
	# Perform test
	testStuff
	
	# Mount remote directory and sync
	while :; do
		# Mount remote directory
		mountRemote
		# Finally, sync dirs
		syncDir
		while inotifywait -r -e modify -e create -e move -e delete -e open $LPATH $MOUNTPOINT &> /dev/null; do
			sleep 10
			syncDir
		done
		# Sleep if remote folder is not mounted or if the synchronization failed
		sleep 30
	done
	exit 0
# If config file doesn't exists
else
	printScreen $RED"Error: "$RESET"create and set "$CONFIGFILE" first, then restart the script.";	
	exit 1
fi