#!/bin/sh
echo "Building initial system"
# Get drive values for installation
lsblk
read -p 'Please enter the drive value for the root partition, this partition will be wiped and formatted: ' rootpar
read -p 'Please enter the drive value for the EFI partition: ' efipar
# Format the root partition to ext4
mkfs.ext4 /dev/$rootpar
# Mount root partition
mount /dev/$rootpar /mnt
# Run pacstrap and install base packages
pacstrap -i /mnt base base-devel linux linux-firmware linux-headers intel-ucode git openssh rsync networkmanager grub efibootmgr dosfstools os-prober nano dkms dbus-broker-units mkinitcpio iptables-nft 
# Ask if the EFI partition should be wiped before mounting
read -p "Do you wish to wipe the EFI partition? If you already have an existing OS installed, say No. " yesno
case $yesno in
    [yesYesYy]* ) 
        mkfs.fat -F32 /dev/$efipar
    ;;
    [noNoNn]* ) 
        echo "The EFI partition will not be wiped"
    ;;
    * ) echo "Please answer yes (Yes, y, Y) or no (No, n, N)";;
esac
# Create the /boot/efi directory and mount the EFI partition to it
mkdir /mnt/boot/efi
mount /dev/$efipar /mnt/boot/efi
# Ask if a home partition should be created
read -p "Do you wish to set a home partition? " yesno
case $yesno in
    [yesYesYy]* )
        read -p 'Please enter the drive value for the home partition: ' homepar 
        mkfs.ext4 /dev/$homepar
        mount /dev/$homepar /mnt/home
    ;;
    [noNoNn]* ) 
        echo "A home partition will not be used in this installation"
    ;;
    * ) echo "Please answer yes (Yes, y, Y) or no (No, n, N)";;
esac
# Generate a fstab file
genfstab -U /mnt >> /mnt/etc/fstab
# Copy system config files into the new install
cp files/sys-pacman.conf /mnt/etc/pacman.conf.tmp
cp files/sys-locale.gen /mnt/etc/locale.gen
cp files/sys-environment /mnt/etc/environment
cp files/sys-grub /mnt/etc/default/grub
cp files/sys-hosts /mnt/etc/hosts
cp files/sys-network.conf /etc/sysctl.d/90-network.conf
cp files/sys-mkinitcpio.conf /etc/mkinitcpio.conf
# Create temporary directories to store files for later use
mkdir /mnt/buildscripts /mnt/userfiles
cp scripts/* /mnt/buildscripts
cp files/user-* /mnt/userfiles
# Chroot into the new system and run the next script
arch-chroot /mnt sh /buildscripts/1-root-config.sh