functions!

README.md

@@ -1,4 +1,8 @@
-# Что это?
+# This is...
 
-Это скрипт, который восстанавливает систему из бекапа, расположенного на rsync сервере, оптимизирован под ArchLinux
+Script that can install OS from previously rsynced backup on NAS
 
+# TODO
+
+* some checks
+* config file support

restore.sh

@@ -1,3 +1,4 @@
+#!/bin/bash
 #not working encrypted setup
 
 #vars
@@ -5,8 +6,8 @@ IP="192.168.100.100"
 source="rsync@"$IP"/mnt/"
 bootSize=256M
 
-#prepare
 ##check for root
+testVars() {
   if [ `id -u` != 0 ] ; then 
       echo "==> Need to be root"
       exit 1
@@ -18,24 +19,24 @@ then
     echo "==> Host unavailable, exiting"
     exit 1
   fi
+}
 
-##check for space
-
+chooseBackup() {
+  ##get list of backups
   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))
 
   ##choose backup
-
   echo "==>" ${#rsync_dir_list[@]} backups: ${rsync_dir_list[@]}
-read -p "Input backup name:" backupName
+  read -p "Input backup number:" backupNum
 
-echo "Selected: " $backupName
-
-#until [ $backupNum -le ${#rsync_dir_list[@]} ]
-#do
-#read -p "Input backup name:" backupName
-#done
-#let "backupNum -= 1"
+  until [ $backupNum -le ${#rsync_dir_list[@]} ]
+  do
+  read -p "Input backup number:" backupNum
+  done
+  let "backupNum -= 1"
+}
 
+setTargetDisk() {
 
   ##set target drive
   lsblk -o NAME,LABEL,FSTYPE,SIZE,MOUNTPOINT,MODEL
@@ -44,9 +45,8 @@ do
   read -p "Target drive is " targetDrive
   targetDrive=/dev/$targetDrive
   done
-#encrypted?
-echo "Encrypted? (Y/N)"
-while read -r -n 1 answer ; do
+  echo "Encrypted?"
+  while read -r -n 1 -s answer ; do
     if [[ $answer = [YyNn] ]]; then
       [[ $answer = [Yy] ]] && encryptedDevice=true && echo "WARNING You will be asked for password later" #read -p "Password: " encryptedPassword
       [[ $answer = [Nn] ]] && encryptedDevice=false
@@ -56,23 +56,21 @@ done
 
   #get swap size
   swapSize=$(free -m | grep "Mem" | awk '{print $2}')
-echo ""
   read -p "Enter extra amount of swap (empty == 512):" swapExtra
-[[ -z "$swapExtra" ]] && swapExtra=512
+  if [ -z "$swapExtra" ] && swapExtra=512
   let "swapSize += swapExtra" #add 512 mbs to swap for swap, not only hibernation
   swapSize=$swapSize"M"
-
+}
 #DEPLOY
 
+warnDeploy() {
   echo deployin\' ${rsync_dir_list[$backupNum]} on $targetDrive in 15s...
-
   sleep 15
-
-
   trap 'echo "==> Interrupted by user"; exit 1' 2
+}
 
+partitionDisk() {
   ##clear mbr for sure
-
   dd if=/dev/zero of=$targetDrive count=512
 
   ##fdisk
@@ -99,36 +97,43 @@ echo
   echo w
   ) | fdisk $targetDrive
   # IMHO this ^ is less obfuscated way
+}
 
+makeFS() {
   ##mkfs's
   echo "==> Making filesystems..."
-echo "=> swap"
+  echo "mkswap"
   mkswap $targetDrive"1"
-echo "=> /boot"
+  echo "mkfs.vfat"
   mkfs.vfat $targetDrive"2"
 
-echo "=> rootfs"
-if $encryptedDevice ; then
+  echo "mkfs.ext4"
+  if $encryptedDevice
+  then
   cryptsetup luksFormat -v $targetDrive"3"
   cryptsetup open $targetDrive"3" targetLuks
-mkfs.ext4 -q /dev/mapper/targetLuks
+  mkfs.ext4 /dev/mapper/targetLuks
   else
-mkfs.ext4 -q $targetDrive"3"
+  mkfs.ext4 $targetDrive"3"
   fi
+}
 
-sleep 15
-
+getUUIDs() {
   ##get disk's uuid
   echo "==> Got UUIDs!"
   UUIDS=($(blkid $targetDrive"1" $targetDrive"2" $targetDrive"3" -o value -s UUID))
-if $encryptedDevice ; then
+  if $encryptedDevice
+  then
   UUIDS[3]=UUIDS[2]
   UUIDS[2]=`blkid /dev/mapper/targetLuks -o value -s UUID`
   fi
+}
 
+mountDisk() {
   ##mount disks
   echo "==> Mounting..."
-if $encryptedDevice ; then
+  if $encryptedDevice
+  then
   mount /dev/mapper/targetLuks /mnt
   else
   mount $targetDrive"3" /mnt
@@ -136,16 +141,19 @@ fi
 
   mkdir /mnt/boot
   mount $targetDrive"2" /mnt/boot
+}
 
+copyFiles() {
   ##rsync
   echo "==> Copying from NAS..."
-#rsync --archive --password-file=rsync_pass rsync://$source${rsync_dir_list[$backupNum]}/ /mnt/
-rsync --archive --password-file=rsync_pass rsync://$source$backupName/ /mnt/
+  rsync --archive --password-file=rsync_pass rsync://$source${rsync_dir_list[$backupNum]}/ /mnt/
   retVal=$?
   if [ $retVal -ne 0 ]; then
       exit 1
   fi
+}
 
+updateFstab() {
   ##update fstab
   echo "==> Updating fstab..."
   mv /mnt/etc/fstab /mnt/etc/fstab.orig
@@ -156,15 +164,22 @@ rm /mnt/etc/fstab.orig #TODO OFC we need to check mountpoint, not only type of F
 
   echo "==> Result:"
   cat /mnt/etc/fstab
+}
+
+configBootloader() 
+{
   ##update grub linux options
-if $encryptedDevice ; then
+  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
+}
 
+updateBootloader()
+{
   ##mount system stuff
   echo "==> Mounting /dev, /sys, /proc..."
   mount /dev /mnt/dev --bind
@@ -177,16 +192,35 @@ echo "==> mkinitcpio and update grub..."
   #cat /mnt/etc/mkinitcpio.conf | sed 's/encrypt //' > /mnt/etc/mkinitcpio.conf #remove cryptdevice, no encrypted fs today :c
 
   #chroot env
-cat << EOF | chroot /mnt /bin/sh
+cat << EOF | chroot /mnt
 mkinitcpio -P
 grub-mkconfig -o /boot/grub/grub.cfg
 grub-install $targetDrive
 exit
 EOF
 
+}
+
+deinit()
+{
   #deinit
   echo "==> Syncing..."
   sync
   echo "==> Unmounting..."
   umount -R /mnt
   echo "==> Done!"
+}
+# funcions
+testVars()
+chooseBackup()
+setTargetDisk()
+warnDeploy()
+partitionDisk()
+makeFS()
+getUUIDs()
+mountDisk()
+copyFiles()
+updateFstab()
+configBootloader()
+updateBootloader()
+deinit()