DietPi-Backup | Add Option to Automatically Restore on First Boot

dietpi.txt

@@ -67,6 +67,9 @@ AUTO_UNMASK_LOGIND=0
 # - Executed script log: /var/tmp/dietpi/logs/dietpi-automation_custom_script.log
 AUTO_SETUP_CUSTOM_SCRIPT_EXEC=0
 
+# - Restore a DietPi-Backup on first boot: 0 => disable | 1 => interactive restore (show list of drives) | 2 => non-interactive restore (restore from first found drive)
+AUTO_SETUP_BACKUP_RESTORE=0
+
 ##### Software Options #####
 # SSH Server Selection: 0=none/custom | -1=Dropbear | -2=OpenSSH
 AUTO_SETUP_SSH_SERVER_INDEX=-1

dietpi/dietpi-software

@@ -16255,6 +16255,7 @@ _EOF_
 		AUTOINSTALL_WEBSERVERINDEX=$(sed -n '/^[[:blank:]]*AUTO_SETUP_WEB_SERVER_INDEX=/{s/^[^=]*=//p;q}' /boot/dietpi.txt)
 		AUTOINSTALL_CUSTOMSCRIPTURL=$(sed -n '/^[[:blank:]]*AUTO_SETUP_CUSTOM_SCRIPT_EXEC=/{s/^[^=]*=//p;q}' /boot/dietpi.txt)
 		AUTOINSTALL_TIMESYNCMODE=$(sed -n '/^[[:blank:]]*CONFIG_NTP_MODE=/{s/^[^=]*=//p;q}' /boot/dietpi.txt)
+		AUTOINSTALL_RESTORE=$(sed -n '/^[[:blank:]]*AUTO_SETUP_BACKUP_RESTORE=/{s/^[^=]*=//p;q}' /boot/dietpi.txt)
 		# Else set defaults
 		[[ $AUTOINSTALL_ENABLED ]] || AUTOINSTALL_ENABLED=0
 		[[ $AUTOINSTALL_AUTOSTARTTARGET ]] || AUTOINSTALL_AUTOSTARTTARGET=0
@@ -16264,6 +16265,7 @@ _EOF_
 		[[ $AUTOINSTALL_WEBSERVERINDEX ]] || AUTOINSTALL_WEBSERVERINDEX=0
 		[[ $AUTOINSTALL_CUSTOMSCRIPTURL ]] || AUTOINSTALL_CUSTOMSCRIPTURL=0
 		[[ $AUTOINSTALL_TIMESYNCMODE ]] || AUTOINSTALL_TIMESYNCMODE=2
+		[[ $AUTOINSTALL_RESTORE ]] || AUTOINSTALL_RESTORE=0
 
 	}
 
@@ -17651,6 +17653,57 @@ List of installed software and their online documentation URLs:
 			# Load all automation vars
 			FirstRun_Automation_Init
 
+			if (( $AUTOINSTALL_RESTORE )); then
+
+					# Find unmounted drives
+					lsblk --noheadings --raw -o NAME,MOUNTPOINT | mawk '$1~/s.*[0-9]/ && $2 == ""' > .lsblk_out_tmp
+					i=0
+					while read -r line
+					do
+
+						# Mount drives to temporary mount points
+						mkdir "/mnt/dietpi-backup$i"
+						mount "/dev/$line" "/mnt/dietpi-backup$i"
+
+						i++
+
+					done < .lsblk_out_tmp
+					rm .lsblk_out_tmp
+
+					mapfile -t alist < <(find /mnt -type f -name "$BACKUP_STATS_FILENAME")
+
+					if [[ $AUTOINSTALL_RESTORE == 1 ]]; then
+						# Do we have any results?
+						if (( ${alist[0]} )); then
+							# Create List for Whiptail
+							G_WHIP_MENU_ARRAY=()
+							for i in "${alist[@]}"
+							do
+
+								last_backup_date=$(sed -n '/ompleted/s/^.*: //p' "$i" | tail -1) # Date of last backup for this backup
+								backup_directory=${i%/$BACKUP_STATS_FILENAME} # Backup directory (minus the backup file), that we can use for target backup directory.
+								G_WHIP_MENU_ARRAY+=("$backup_directory" ": $last_backup_date")
+
+							done
+
+							G_WHIP_MENU 'Please select a previous backup to use:'
+							/boot/dietpi/dietpi-backup -1 "$G_WHIP_RETURNED_VALUE"
+
+						else
+
+							G_WHIP_MSG 'No previous backups were found in /mnt/*'
+							return
+						fi
+					elif [[ $AUTOINSTALL_RESTORE == 2 ]]; then
+						# Do we have any results?
+						if (( ${alist[0]} )); then
+							# Restore first found backup
+							/boot/dietpi/dietpi-backup -1 /mnt/dietpi-backup0
+						fi
+					fi
+
+			fi
+
 			# Global PW
 			# - Automation, apply as per dietpi.txt
 			if (( $AUTOINSTALL_ENABLED )); then