#!/bin/bash
#
# Script for automated cloning and renaming of Fuss 9 client computers
# using mac-address as unique identifier.
# Donato Florio 11 April 2019
#
# Last modified 02 May 2019 --> added modification to support cloning on nvme* devices (HP G450)
# Last modified 07 May 2019 --> added various stuff for experimental autojoin to a domain
# Last modified 21 June 2019 --> added various stuff for experimental join to a cluster
# Last modified 01 July 2019 --> added ssh to script and changed disk detection method
# 
# PRT 1 ---------------------------

service ssh start

getNetStuff(){
    if [ $1 -eq 1 ];then
	ls /sys/class/net/ -1 | grep -v lo | grep -v wlan
    else
	cat /sys/class/net/$(getNetStuff 1)/address
    fi
}

getDisk(){
    for i in $(ls /sys/block/ | grep -Ev '(loop|sr|dvd|cd)')
    do
        if [ $(cat /sys/block/$i/removable) -eq 0 ];then
            myDevice="${myDevice} $i"
        fi
    done

    if [ $(echo $myDevice | wc -w) -gt 1 ];then
        echo "ATTENZIONE! Questo computer sembra provvisto di più di un disco!"
        echo ""
        echo "La procedura automatica non prevede l'installazione su sistemi"
        echo "multidisco. Contattare il tecnico informatico."
        echo ""
        echo "Per sicurezza il computer verrà spento tra 10 secondi."
	sleep 10
	shutdown -h now
    else
        diskVar="$(echo $myDevice | tr -d [:blank:])"
    fi
    
    # SSD of HP g450 laptops
    if [ $diskVar == "nvme0n1" ];then
	rootPartition="p$rootPartition"
    fi
    # FINE g450
}

rootPartition=2
getDisk
rootDisk="/dev/$diskVar"
mountPoint="/mnt"
computerList="/home/partimag/computerList.txt"
macAddress=$(getNetStuff 2)
imageName=$(cat $computerList | grep "$macAddress" | awk '{print $3}')

ocs-sr -b -g auto -e1 auto -e2 -r -j2 -scr -p true restoredisk $imageName $diskVar

# # PRT 2 ---------------------------
mount $rootDisk$rootPartition $mountPoint

currentName="$(cat $mountPoint/etc/hostname)"
newName="$(grep "$macAddress" $computerList | awk '{print $1}')"

echo "SETTING HOSTNAME" 

for i in hostname hosts mailname
	 
do
    if [ -e $mountPoint/etc/$i ]; then
	sed -ie "s/$currentName/$newName/g" $mountPoint/etc/$i
    else
	echo "The file $i is not present on this system"
    fi    
done



# ###  Here I insert the part for the joining preparation of the machine

# Verify whether the machine has to be joined to the domain or not.
# If so the .ssh keys are needed and also a script ..... 

joinVar="$(cat $computerList | grep "$macAddress" | awk '{print $4}')"

clusterVar="$(cat $computerList | grep "$macAddress" | awk '{print $5}')"

if [ "$joinVar" == "join" ];then
    rsync -a /home/partimag/.ssh/ /mnt/root/.ssh/
    cp /root/.ssh/known_hosts /mnt/root/.ssh/known_hosts
    cp /home/partimag/clientScripts/rc.local /mnt/etc/
    chmod 770 /mnt/etc/rc.local
    cp /home/partimag/clientScripts/clientScript /mnt/root/
    chmod 770 /mnt/root/clientScript
    touch /mnt/root/reboot
fi

if [ "$clusterVar" != "" ];then
    clusterJoin="-g $clusterVar"
    echo "$clusterJoin" > /mnt/root/cluster
fi



# ### END of the preparation stuff


reboot