#!/bin/bash

if [ -e "serverconf.sh" ]
then
	source "serverconf.sh"
else
	echo No configuration found in PWD. Exiting.
	exit 1
fi

source "backends/tar.sh"
source "backends/bup.sh"
source "backends/borg.sh"

function echo_debug() {
	if [ $VERBOSE -ne 0 ]; then
		echo "$1"
	fi
}

function backup_hook_example {
	bup -d $CUR_BACK_DIR ls -l $BACKUP_NAME/latest/var/minecraft
}

function send_cmd () {
	tmux -S $TMUX_SOCKET send -t $TMUX_WINDOW "$1" enter
}

function assert_running() {
	if server_running; then
		echo "It seems a server is already running. If this is not the case,\
			manually attach to the running screen and close it."
		exit 1
	fi
}

function assert_not_running() {
	if ! server_running; then
		echo "Server not running"
		exit 1
	fi
}

function server_start() {
	assert_running

	if [ ! -f "eula.txt" ]
	then
		echo "eula.txt not found. Creating and accepting EULA."
		echo "eula=true" > "eula.txt"
	fi

	tmux -S $TMUX_SOCKET new-session -s $TMUX_WINDOW -d \
		$JRE_JAVA $JVM_ARGS -jar $JAR $JAR_ARGS
	pid=`tmux -S $TMUX_SOCKET list-panes -t $TMUX_WINDOW -F "#{pane_pid}"`
	echo $pid > $PIDFILE
	echo Started with PID $pid
	exit
}

function server_stop() {
	# Allow success even if server is not running
	#trap "exit 0" EXIT

	assert_not_running
	send_cmd "stop"

	local RET=1
	while [ ! $RET -eq 0 ]
	do
		sleep 1
		ps -p $(cat $PIDFILE) > /dev/null
		RET=$?
	done

	echo "stopped the server"

	rm -f $PIDFILE

	exit
}

function server_attach() {
	assert_not_running
	tmux -S $TMUX_SOCKET attach -t $TMUX_WINDOW
	exit
}

function server_running() {
	if [ -f $PIDFILE ] && [ "$(cat $PIDFILE)" != "" ]; then
		ps -p $(cat $PIDFILE) > /dev/null
		return
	fi

	false
}

function server_status() {
	if server_running
	then
		echo "Server is running"
	else
		echo "Server is not running"
	fi
	exit
}

function players_online() {
	send_cmd "list"
	sleep 1
	while [ $(tail -n 3 "$LOGFILE" | grep -c "There are") -lt 1 ]
	do
		sleep 1
	done

	[ `tail -n 3 "$LOGFILE" | grep -c "There are 0"` -lt 1 ]
}

function init_backup() {
	# even though bup and borg are smart, they will not create a path for a repo
	for backup_dir in ${BACKUP_DIRS[*]}
	do
		if [[ $backup_dir == *:* ]]; then
			local remote="$(echo "$backup_dir" | cut -d: -f1)"
			local remote_dir="$(echo "$backup_dir" | cut -d: -f2)"
			ssh "$remote" "mkdir -p \"$remote_dir\""
		else
			mkdir -p "$backup_dir"
		fi
	done

	if [ $BACKUP_BACKEND = "bup" ]; then
		bup_init
	elif [ $BACKUP_BACKEND = "borg" ]; then
		borg_init
	else
		tar_init
	fi
}

function same_world() {
	delta=$(diff -r "$1" "$2")
	if [ $? -ne 0 ]; then
		return 1
	fi
	if [ -z "$delta" ] ; then
		return 0
	fi
	return 1
}

# checking if latest snapshots are the same as the current world
function test_backup_integrity() {
	local retcode=0
	for backup_dir in ${BACKUP_DIRS[*]}
	do
		local tmpdir=$(mktemp -d);

		# restore most recent backup to a temporary dir
		if ! server_restore "$backup_dir" 0 "$tmpdir" ; then
			echo "Failed to get latest snapshot from \"$backup_dir\""
			retcode=1
		elif ! same_world "$WORLD_NAME" "$tmpdir/$WORLD_NAME" ; then
			echo "Latest backup from \"$backup_dir\" differs from current world!"
			retcode=1
		else
			echo "Backup at \"$backup_dir\" is OK"
		fi

		rm -r "$tmpdir"
	done

	if [ $retcode -ne 0 ] ; then
		echo "Backup integrity check: FAILED"
	else
		echo "Backup integrity check: OK"
	fi
	return $retcode
}

function create_backup() {
	init_backup

	if [ $BACKUP_BACKEND = "bup" ]; then
		bup_create_backup
	elif [ $BACKUP_BACKEND = "borg" ]; then
		borg_create_backup
	else
		tar_create_backup
	fi

	test_backup_integrity
}

function server_backup_safe() {
	local force=$1
	echo "Detected running server. Checking if players online..."
	if [ "$force" != "true" ] && ! players_online; then
		echo "Players are not online. Not backing up."
		return
	fi

	echo "Disabling autosave"
	send_cmd "save-off"
	send_cmd "save-all flush"
	echo "Waiting for save... If froze, run /save-on to re-enable autosave!!"

	sleep 1
	while [ $(tail -n 3 "$LOGFILE" | grep -c "Saved the game") -lt 1 ]
	do
		sleep 1
	done
	sleep 2
	echo "Done! starting backup..."

	create_backup

	local RET=$?

	echo "Re-enabling auto-save"
	send_cmd "save-on"

	if [ $RET -eq 0 ]; then
		echo Running backup hook
		$BACKUP_HOOK
	fi
}

function server_backup_unsafe() {
	echo "No running server detected. Running Backup"

	create_backup
	local status=$?

	if [ $status -eq 0 ]; then
		echo Running backup hook
		$BACKUP_HOOK
	fi
}

function backup_running() {
	systemctl is-active --quiet mc-backup.service
}

function fbackup_running() {
	systemctl is-active --quiet mc-fbackup.service
}

function server_backup() {
	local force=$1

	if [ "$force" = "true" ]; then
		if backup_running; then
			echo "A backup is running. Aborting..."
			return
		fi
	else
		if fbackup_running; then
			echo "A force backup is running. Aborting..."
			return
		fi
	fi

	if server_running; then
		server_backup_safe "$force"
	else
		server_backup_unsafe
	fi
}

function ls_backups() {
	if [ $BACKUP_BACKEND = "bup" ]; then
		bup_ls_all
	elif [ $BACKUP_BACKEND = "borg" ]; then
		borg_ls_all
	else
		tar_ls_all
	fi
}

# creates a selection dialog
function choose_from() {
	local items=("$@")
	select item in "${items[@]}"; do
		echo "$item"
		return
	done
	echo ""
}

# checks if an item is in the array
function is_in() {
	local item="$1"
	shift
	local array=("$@")

	# these :space: things allow checking that *exactly* this item is in array
	if [[ ${array[*]} =~ (^|[[:space:]])"$item"($|[[:space:]]) ]]; then
		return
	fi
	false
}

function server_restore() {
	local backup_dir
	local snapshot_index
	local dest="$PWD"

	# parameters are only used for testing backups, so thorough checks are not needed
	if [ $# -ge 2 ]; then
		backup_dir="$1"
		snapshot_index=$2
	fi
	if [ $# -eq 3 ]; then
		dest="$3"
	fi

	if [ ${#BACKUP_DIRS[@]} -eq 0 ]; then
		echo "No backup directories found, abort"
		return 1
	fi


	if [ -z $backup_dir ]; then
		echo "From where get the snapshot?"
		backup_dir="$(choose_from "${BACKUP_DIRS[@]}")"
	fi
	if ! is_in "$backup_dir" "${BACKUP_DIRS[@]}" ; then
		echo "No valid backup directory selected, abort"
		return 1
	fi


	local snapshots=$(
		if [ $BACKUP_BACKEND = "bup" ]; then
			bup_ls_dir "$backup_dir"
		elif [ $BACKUP_BACKEND = "borg" ]; then
			borg_ls_dir "$backup_dir"
		else
			tar_ls_dir "$backup_dir"
		fi
	)
	if [ -z "$snapshots" ]; then
		echo "No snapshots found, abort"
		return 1
	fi
	# convert multiline string to bash array
	snapshots=($(echo "$snapshots"))


	local snapshot
	if [ -z $snapshot_index ]; then
		echo "Select which snapshot to restore"
		snapshot=$(choose_from "${snapshots[@]}")
	else
		snapshot="${snapshots[snapshot_index]}"
	fi
	if ! is_in "$snapshot" "${snapshots[@]}" ; then
		echo "No valid snapshot selected, abort"
		return 1
	fi


	echo_debug "Restoring snapshot \"$snapshot\" from \"$backup_dir\""


	# if we restore to PWD, we will overwrite the current world, which might be harmful
	local oldworld_name
	if [ "$dest" = "$PWD" ] && [[ -d "$WORLD_NAME" ]]; then
		echo -n "Preserving old world: "
		oldworld_name="${WORLD_NAME}.old.$(date +'%F_%H-%M-%S.%N')"
		mv -n -v "$PWD/$WORLD_NAME" "$PWD/$oldworld_name"
	fi


	if [ $BACKUP_BACKEND = "bup" ]; then
		bup_restore "$backup_dir" "$snapshot" "$dest"
	elif [ $BACKUP_BACKEND = "borg" ]; then
		borg_restore "$backup_dir" "$snapshot" "$dest"
	else
		tar_restore "$backup_dir" "$snapshot" "$dest"
	fi
	local status=$?


	# if we preseved the current world, but failed to restore the snapshot
	if [ ! -z ${oldworld_name+x} ] && [ $status -ne 0 ]; then
		echo "Failed to restore snapshot, putting old world back where it was:"
		rm -rv "$PWD/$WORLD_NAME"
		mv -v "$PWD/$oldworld_name" "$PWD/$WORLD_NAME"
		return 1
	fi

	echo_debug "Snapshot restored"

	return 0
}

#cd $(dirname $0)

case $1 in
	"start")
		server_start
		;;
	"stop")
		server_stop
		;;
	"attach")
		server_attach
		;;
	"backup")
		server_backup
		;;
	"restore")
		server_restore
		;;
	"status")
		server_status
		;;
	"fbackup")
		server_backup "true"
		;;
	"ls")
		ls_backups
		;;
	*)
		echo "Usage: $0 start|stop|attach|status|backup"
		;;
esac