#!/bin/bash
set -e

if ! timeout 1 curl -Is tombarrett.xyz | grep "200 OK" > /dev/null; then
    echo "Please first connect to the internet."
    exit
fi

if ! test -b "$1"; then
    echo "Please pass an existing drive as an argument. ie /dev/sda"
    exit
fi

if [ -z "$2" ]; then
    echo "Please pass a hostname also as the second argument."
    exit
fi

parted -s "$1" mklabel gpt 

if ! test -f "/sys/firmware/efi/efivars"; then
    parted "$1" mkpart bios ext4 0% 513MiB
    mkfs.ext4 "$1"1
    parted "$1" set 1 bios_grub on
    parted "$1" mkpart home ext4 513MiB 100%
else
    parted "$1" mkpart efi fat32 0% 513MiB
    mkfs.fat -F 32 "$1"1
    parted "$1" set 1 esp on
    parted "$1" mkpart home ext4 513MiB 100%
fi

mkfs.ext4 "$1"2

mount "$1"2 /mnt
mkdir -p /mnt/boot/efi
mount "$1"1 /mnt/boot/efi

timedatectl set-ntp true

pacman-key --init
pacman-key --populate
pacstrap /mnt base linux linux-firmware git vim grub

genfstab -U /mnt > /mnt/etc/fstab
cp /root/.loader /mnt/root/loader
arch-chroot /mnt /root/loader $1 $2