#!/bin/bash

# vim: sw=4 ts=4

sleep 30

diskname="sda,sdb"
HD1=$(echo "$diskname" | awk -F',' '{print $1}')
HD2=$(echo "$diskname" | awk -F',' '{print $2}')

for x in /dev/md/* ; do
  mdadm --stop $x
done

#disable all VG of LVM
vgchange -an

#first disk
dd if=/dev/zero of=/dev/$HD1 bs=1M count=100
parted -s $HD1 mklabel gpt
wipefs -a -f $HD1

#second disk
dd if=/dev/zero of=/dev/$HD2 bs=1M count=100
parted -s $HD2 mklabel gpt
wipefs -a -f $HD2

SWSIZE=$(grep MemTotal /proc/meminfo  | awk '{print int($2/1024)+1}')
if [ ${SWSIZE} -gt 15381 ]; then
        #SWSIZE=32768
        SWSIZE=65536
else
        SWSIZE=2048
fi

SIZE1=$(fdisk -l "/dev/${HD1}"  2>/dev/null  | grep Disk | grep dev |  sed 's/://'|awk '{print $5}')

if [ -z "${HD2}" ]; then
        echo "Can not find second HDD"
        exit 1
fi

rm -f /tmp/part-include

if [ -d /sys/firmware/efi ]; then
        efibootmgr -v | grep BootCurrent | cut -d' ' -f2 > /tmp/efi_booted
        echo "part raid.11 --size=200 --asprimary --ondisk=${HD1}" >> /tmp/part-include
        echo "part raid.21 --size=200 --asprimary --ondisk=${HD2}" >> /tmp/part-include
        echo "raid /boot/efi --fstype efi --level=RAID1 --device=md0 raid.11 raid.21" >> /tmp/part-include
  echo "${HD2}" > /tmp/efi_second_part
elif [ ${SIZE1} -gt 2254857830400 ]; then
        # bios boot partition necessary
        echo "part biosboot --fstype=biosboot --size=1 --ondisk=${HD1}" >> /tmp/part-include
        echo "part biosboot --fstype=biosboot --size=1 --ondisk=${HD2}" >> /tmp/part-include
fi
cat << EOF >> /tmp/part-include
part raid.12 --size=1024 --asprimary --ondisk=${HD1}
part raid.22 --size=1024 --asprimary --ondisk=${HD2}

part raid.13 --size=${SWSIZE} --asprimary --ondisk=${HD1}
part raid.23 --size=${SWSIZE} --asprimary --ondisk=${HD2}

part raid.14 --size=1 --grow --asprimary --ondisk=${HD1}
part raid.24 --size=1 --grow --asprimary --ondisk=${HD2}

raid /boot --fstype xfs --level=RAID1 --device=md1 raid.12 raid.22
raid swap --fstype swap --level=RAID1 --device=md2 raid.13 raid.23
raid / --fstype xfs --level=RAID1 --device=md3 raid.14 raid.24

bootloader --location=mbr --boot-drive=${HD1} --driveorder=${HD1},${HD2} --append=" consoleblank=0 crashkernel=auto selinux=0 intremap=off mitigations=off"
EOF