From 1f0c6d96030c337f1e6a8d03df06d2fde374882f Mon Sep 17 00:00:00 2001
From: UltimateByte <robin.labadie@gmail.com>
Date: Thu, 20 Oct 2016 02:43:31 +0200
Subject: [PATCH] First shot for #1098

---
 lgsm/functions/command_backup.sh | 55 +++++++++++++++++++++++++++++---
 1 file changed, 51 insertions(+), 4 deletions(-)

diff --git a/lgsm/functions/command_backup.sh b/lgsm/functions/command_backup.sh
index ca6fe9607..369228ab1 100644
--- a/lgsm/functions/command_backup.sh
+++ b/lgsm/functions/command_backup.sh
@@ -9,10 +9,24 @@ local commandaction="Backup"
 local function_selfname="$(basename $(readlink -f "${BASH_SOURCE[0]}"))"
 
 check.sh
+# Check if a backup is pending or has been aborted using .backup.lock
+if [ -f "${tmdpdir}/.backup.lock" ]; then
+	fn_print_warning_nl "A backup is currently pending or has been aborted."
+	while true; do
+		read -e -i "y" -p "Continue anyway? [Y/N]" yn
+		case $yn in
+		[Yy]* ) break;;
+		[Nn]* ) echo Exiting; return;;
+		* ) echo "Please answer yes or no.";;
+	esac
+	done
+fi
 info_distro.sh
-backupname="${servicename}-$(date '+%Y-%m-%d-%H%M%S')"
 fn_print_dots ""
 sleep 0.5
+# Prepare backup file name with servicename current date
+backupname="${servicename}-$(date '+%Y-%m-%d-%H%M%S')"
+# Tells how much will be compressed using rootdirduexbackup value from info_distro
 fn_print_info_nl "A total of ${rootdirduexbackup} will be compressed into the following backup:"
 echo "${backupdir}/${backupname}.tar.gz"
 echo ""
@@ -33,8 +47,8 @@ if [ "${status}" != "0" ]; then
 	while true; do
 		read -p "Stop ${servicename} while running the backup? [Y/N]" yn
 		case $yn in
-		[Yy]* ) command_stop.sh; break;;
-		[Nn]* ) break;;
+		[Yy]* ) command_stop.sh; serverstopped="yes"; break;;
+		[Nn]* ) serverstopped="no"; break;;
 		* ) echo "Please answer yes or no.";;
 	esac
 	done
@@ -42,14 +56,47 @@ fi
 
 fn_print_dots "Backup in progress, please wait..."
 fn_script_log_info "Started backup"
-sleep 2
+sleep 1
 if [ ! -d "${backupdir}" ]; then
 	mkdir "${backupdir}"
 fi
+# Create lockfile
+touch "${tmdpdir}/.backup.lock"
+# Compressing files
 tar -czf "${backupdir}/${backupname}.tar.gz" -C "${rootdir}" --exclude "backups" ./*
+# Remove lockfile
+rm "${tmdpdir}/.backup.lock"
+# Check tar exit code and act accordingly
 if [ $? == 0 ]; then
 	fn_print_ok_nl "Backup created: ${backupname}.tar.gz is $(du -sh "${backupdir}/${backupname}.tar.gz" | awk '{print $1}') size"
 	fn_script_log_pass "Backup created: ${backupdir}/${backupname}.tar.gz is $(du -sh "${backupdir}/${backupname}.tar.gz" | awk '{print $1}') size"
+	
+	# Clear old backups if backupdays variable exists
+	if [ -n "${backupdays}" ]; then
+		# Count how many backups can be cleared
+		backupclearcount=$(find "${backupdir}"/ -type f -mtime +"${backupdays}"|wc -l)
+		# Check if there is any backup to clear
+		if [ "${backupclearcount}" -ne "0" ]; then
+			fn_print_info_nl "${backupclearcount} backups older than ${backupdays} days can be cleared."
+			while true; do
+				read -p "Clear older backups? [Y/N]" yn
+				case $yn in
+				[Yy]* ) clearoldbackups="yes"; break;;
+				[Nn]* ) clearoldbackups="no"; break;;
+				* ) echo "Please answer yes or no.";;
+			esac
+			done
+			# If user wants to clear backups
+			if [ "${clearoldbackups}" == "yes" ]; then
+				find "${backupdir}"/ -mtime +"${backupdays}" -type f -exec rm -f {} \;
+				fn_print_ok_nl "Cleared ${backupclearcount} backups."
+			fi
+		fi
+	fi
+	# Restart the server if it was stopped for the backup
+	if [ "${serverstopped}" == "yes" ]; then
+		command_start.sh
+	fi
 else
 	fn_print_error_nl "Backup failed: ${backupname}.tar.gz"
 	fn_script_log_error "Backup failed: ${backupname}.tar.gz"