From ec767728347b641cbed6412061c4b010e9c89fbd Mon Sep 17 00:00:00 2001
From: l0sted <theext4@yandex.ru>
Date: Mon, 28 May 2018 13:04:40 +0300
Subject: [PATCH] beautified, added some questions and vars, etc...

---
 main.sh | 79 ++++++++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 64 insertions(+), 15 deletions(-)
 mode change 100644 => 100755 main.sh

diff --git a/main.sh b/main.sh
old mode 100644
new mode 100755
index dc92d9a..05f4954
--- a/main.sh
+++ b/main.sh
@@ -1,3 +1,12 @@
+#not working encrypted setup
+
+#vars
+source="rsync@192.168.100.16/rsync/"
+swapSize=$(free -m | grep "Mem" | awk '{print $2}')
+let "swapSize += 512"
+swapSize=$swapSize"M"
+bootSize=256M
+
 #prepare
 ##check for root
 if [ `id -u` != 0 ] ; then 
@@ -5,7 +14,7 @@ if [ `id -u` != 0 ] ; then
     exit 1
 fi
 ##check for network
-ping -c 1 192.168.100.16 &> /dev/null
+ping -c 1 $IP &> /dev/null
 if [ "$?" != 0 ]
 then
   echo "==> Host unavailable, exiting"
@@ -14,31 +23,44 @@ fi
 
 ##check for space
 
-rsync_dir_list=($(rsync --password-file=/home/losted/.rsync_pass rsync://rsync@192.168.100.16/rsync/ | awk '{print $5}' | grep -v .DS_Store | grep -v "@Recently-Snapshot" | sort -r | head -n -1))
+rsync_dir_list=($(rsync --password-file=rsync_pass rsync://$source | awk '{print $5}' | grep -v .DS_Store | grep -v "@Recently-Snapshot" | sort -r | head -n -1))
 
-swapSize=$(free -m | grep "Mem" | awk '{print $2}')
-let "swapSize += 512"
-swapSize=$swapSize"M"
-bootSize=256M
-
-#begin
-
-##chose backup
+##choose backup
 
 echo "==>" ${#rsync_dir_list[@]} backups: ${rsync_dir_list[@]}
 read -p "Input backup number:" backupNum
+
+until [ $backupNum -le ${#rsync_dir_list[@]} ]
+do
+read -p "Input backup number:" backupNum
+done
 let "backupNum -= 1"
 
+
 ##set target drive
 lsblk -o NAME,LABEL,FSTYPE,SIZE,MOUNTPOINT,MODEL
+until [ ! -f $targetDrive ]
+do
 read -p "Target drive is " targetDrive
-
 targetDrive=/dev/$targetDrive
+done
+echo "Encrypted?"
+while read -r -n 1 -s answer ; do
+  if [[ $answer = [YyNn] ]]; then
+    [[ $answer = [Yy] ]] && encryptedDevice=true && read -p "Password: " encryptedPassword
+    [[ $answer = [Nn] ]] && encryptedDevice=false
+    break
+  fi
+done
+#DEPLOY
 
 echo deployin\' ${rsync_dir_list[$backupNum]} on $targetDrive in 15s...
 
 sleep 15
 
+
+trap 'echo "==> Interrupted by user"; exit 1' 2
+
 ##clear mbr for sure
 
 dd if=/dev/zero of=$targetDrive count=512
@@ -74,26 +96,50 @@ echo "mkswap"
 mkswap $targetDrive"1"
 echo "mkfs.vfat"
 mkfs.vfat $targetDrive"2"
+
 echo "mkfs.ext4"
+if $encryptedDevice
+then
+cryptsetup luksFormat -v $targetDrive"3" | echo $encryptedPassword
+cryptsetup open $targetDrive"3" targetLuks | echo $encryptedPassword
+mkfs.ext4 /dev/mapper/targetLuks
+else
 mkfs.ext4 $targetDrive"3"
+fi
+
+sleep 15
 
 ##get disk's uuid
 echo "==> Got UUIDs!"
 UUIDS=($(blkid $targetDrive"1" $targetDrive"2" $targetDrive"3" -o value -s UUID))
+if $encryptedDevice
+then
+UUIDS[3]=UUIDS[2]
+UUIDS[2]=`blkid /dev/mapper/targetLuks -o value -s UUID`
+fi
 
 ##mount disks
 echo "==> Mounting..."
+if $encryptedDevice
+then
+mount /dev/mapper/targetLuks /mnt
+else
 mount $targetDrive"3" /mnt
+fi
+
 mkdir /mnt/boot
 mount $targetDrive"2" /mnt/boot
 
 ##rsync
 echo "==> Copying from NAS..."
-rsync --archive -P --password-file=/home/losted/.rsync_pass rsync://rsync@192.168.100.16/rsync/${rsync_dir_list[$backupNum]}/ /mnt/
+rsync --archive --password-file=rsync_pass rsync://$source${rsync_dir_list[$backupNum]}/ /mnt/
+retVal=$?
+if [ $retVal -ne 0 ]; then
+    exit 1
+fi
 
 ##update fstab
 echo "==> Updating fstab..."
-rm /mnt/etc/fstab.orig
 mv /mnt/etc/fstab /mnt/etc/fstab.orig
 cat /mnt/etc/fstab.orig | grep "ext4" | sed 's/UUID=[A-Fa-f0-9-]*/UUID='${UUIDS[2]}'/' > /mnt/etc/fstab
 cat /mnt/etc/fstab.orig | grep "vfat" | sed 's/UUID=[A-Fa-f0-9-]*/UUID='${UUIDS[1]}'/' >> /mnt/etc/fstab
@@ -103,8 +149,11 @@ rm /mnt/etc/fstab.orig
 echo "==> Result:"
 cat /mnt/etc/fstab
 ##update grub linux options
-
-cat /mnt/etc/default/grub | sed 's/cryptdevice=UUID=[A-Fa-f0-9-]*:cryptroot //' > /mnt/etc/default/grub #remove cryptdevice, no encrypted fs today :c
+if $encryptedDevice
+cat /mnt/etc/default/grub | sed 's/cryptdevice=UUID=[A-Fa-f0-9-]*:cryptroot /cryptdevice=UUID='${UUIDS[3]}':cryptroot/' > /mnt/etc/default/grub 
+else
+cat /mnt/etc/default/grub | sed 's/cryptdevice=UUID=[A-Fa-f0-9-]*:cryptroot //' > /mnt/etc/default/grub
+fi
 
 cat /mnt/etc/default/grub | sed 's/resume=UUID=[A-Fa-f0-9-]*/resume=UUID='${UUIDS[0]}'/' > /mnt/etc/default/grub