# variables
# Edit these variables or leave them blank to be prompted during setup

# install arch to this drive ex: /dev/sda or /dev/sdb... etc
DRIVE='/dev/nvme0n1'

HOST_NAME='Arun-Laptop'

# has to be all lowercase
USERNAME='arunscape'

# examples:
# America/New_York
# America/Toronto
TIMEZONE='Canada/Mountain'

# amd
# intel
CPU='amd'

# anything for yes, empty for no
WIFI='y'
DOTFILES=''
LAPTOP='y'

# amd
# nvidia
# intel 
GPU='amd'


set -e 
setfont sun12x22
lsblk
read -p "$(tput bold)$(tput setaf 1)WARNING this will wipe $DRIVE Press ENTER to continue, or Ctrl+C to exit$(tput sgr 0)"

sgdisk --zap-all $DRIVE

partprobe $DRIVE
sgdisk --clear \
        --new=1:0:+512MiB --typecode=1:ef00 --change-name=1:EFI \
        --new=2:0:0       --typecode=2:8300 --change-name=2:cryptsystem \
        $DRIVE

partprobe $DRIVE
sleep 1 # surely there's a better way
mkfs.fat -F32 -n EFI /dev/disk/by-partlabel/EFI

cryptsetup luksFormat \
        --pbkdf argon2id \
        --key-size 512 \
        --hash sha512 \
        --cipher aes-xts-plain64 \
        --type luks2 \
        --use-random \
        --iter-time 5000 \
        /dev/disk/by-partlabel/cryptsystem

cryptsetup open /dev/disk/by-partlabel/cryptsystem cryptroot

mkfs.btrfs --force --label arch /dev/mapper/cryptroot
o=defaults,x-mount.mkdir
o_btrfs=$o,compress=lzo,ssd,noatime,nodiratime

# when doing snapshots, also include pacman cache under /var
mount -t btrfs LABEL=arch /mnt
btrfs subvolume create /mnt/@
btrfs subvolume create /mnt/@home
btrfs subvolume create /mnt/@snapshots
umount -R /mnt

mount -t btrfs -o subvol=@,$o_btrfs LABEL=arch /mnt
mount -t btrfs -o subvol=@home,$o_btrfs LABEL=arch /mnt/home
mount -t btrfs -o subvol=@snapshots,$o_btrfs LABEL=arch /mnt/.snapshots

mkdir -p /mnt/var/cache/pacman/
btrfs subvolume create /mnt/var/cache/pacman/pkg
btrfs subvolume create /mnt/var/abs
btrfs subvolume create /mnt/var/tmp
btrfs subvolume create /mnt/srv

mkdir /mnt/boot
mount LABEL=EFI /mnt/boot

sed -i "s/^#Color/Color/" /etc/pacman.conf
sed -i "/^Color/a ILoveCandy" /etc/pacman.conf

echo 'Server = https://cloudflaremirrors.com/archlinux/$repo/os/$arch' | cat - /etc/pacman.d/mirrorlist > tmp && mv tmp /etc/pacman.d/mirrorlist


pacman -Syy
pacstrap /mnt base base-devel \
                    btrfs-progs \
                    $CPU-ucode \
                    linux \
                    linux-headers \
                    linux-lts \
                    linux-lts-headers \
                    linux-zen \
                    linux-zen-headers \
                    linux-firmware \
                    openssh \
                    git \
                    neovim \
                    fish \
                    dhcpcd

# UUID based
genfstab -U /mnt >> /mnt/etc/fstab
# Labels
# genfstab -L -p /mnt >> /mnt/etc/fstab

echo $HOST_NAME > /mnt/etc/hostname
ln -sf /mnt/usr/share/zoneinfo/$TIMEZONE /mnt/etc/localtime
sed -i '/en_CA/ s/^#//' /mnt/etc/locale.gen
sed -i '/en_US/ s/^#//' /mnt/etc/locale.gen
echo LANG=en_CA.UTF-8 >> /mnt/etc/locale.conf
echo LANG=en_US.UTF-8 >> /mnt/etc/locale.conf
echo KEYMAP=us >> /mnt/etc/vconsole.conf
sed -i '/%wheel ALL=(ALL) ALL/ s/^# //' /mnt/etc/sudoers
sed -i "s/^#Color/Color/" /mnt/etc/pacman.conf
sed -i "/^Color/a ILoveCandy" /mnt/etc/pacman.conf

curl -Lo /mnt/install.sh https://raw.githubusercontent.com/Arunscape/arch-install-config/master/02-chroot.sh
chmod +x /mnt/install.sh
arch-chroot /mnt bash install.sh $USERNAME $DRIVE $CPU $GPU $WIFI $DOTFILES $LAPTOP

# exit and reboot
rm /mnt/install.sh
umount -R /mnt
echo "$(tput bold)$(tput setaf 2)Done!!!$(tput sgr 0)"
reboot