setup.sh

#!/bin/bash

if [ "$(id -u)" -ne 0 ]; then
  echo -e "This script must be run as root, use 'sudo' or run it as root."
  exit 1
fi

[ ! -d /sys/firmware/efi ] && echo -e "System not booted in UEFI mode." && exit 1

case "$(readlink -f /sbin/init)" in
  *"runit"*)
    INIT="runit" && echo -e "Init system: $INIT"
    ;;
  *)
    echo -e "Init system: Not supported." && exit 1
    ;;
esac

case "$(grep vendor /proc/cpuinfo)" in
  *"Intel"*)
    UCODE="intel-ucode"
    ;;
  *"AMD"*)
    UCODE="amd-ucode"
    ;;
esac

confirm_password() {
  stty -echo
  until [ "$pass1" = "$pass2" ] && [ "$pass1" ] && [ "$pass2" ]; do
    printf "\n%s\n" "$1" >&2 && read -p $"> " pass1
    printf "\nRe-type %s\n" "$1" >&2 && read -p $"> " pass2
  done
  stty echo
  echo -e "$pass2"
}

# Pacman
sed -i -e '/#DisableSandbox/a DisableDownloadTimeout' /etc/pacman.conf

# Dependencies
command -v parted >/dev/null 2>&1 || pacman -Sy --needed --noconfirm --disable-download-timeout parted

# Load keymap
until [ "$KEYMAP" ]; do
  clear
  echo -e "Load keymap (default: us)" && read -p $"> " KEYMAP
  [ ! "$KEYMAP" ] && KEYMAP="us"
  loadkeys $KEYMAP
  setxkbmap $KEYMAP
done

# Timezone
until [ "$REGION_CITY" ]; do
  clear
  echo -e "Local time (default: Europe/Moscow)" && read -p $"> " REGION_CITY
  [ ! "$REGION_CITY" ] && REGION_CITY="Europe/Moscow"
done

# Host
until [ "$HOST" ]; do
  clear
  echo -e "Hostname (default: localhost)" && read -p $"> " HOST
  [ ! "$HOST" ] && HOST="localhost"
done

# Username
until [ "$USERNAME" ]; do
  clear
  echo -e "Username (default: artix)" && read -p $"> " USERNAME
  [ ! "$USERNAME" ] && USERNAME="artix"
done

# Root
[ ! "$ROOT_PASSWORD" ] && ROOT_PASSWORD=$(confirm_password "Password for superuser (will use same for root)")

# Network
until [ "$SSID" ]; do
  clear
  echo -e "Wi-Fi SSID (leave empty for Ethernet)" && read -p $"> " SSID
  [ ! "$SSID" ] && break
  until [ "$PSK" ]; do
    stty -echo
    echo -e "Password for Wi-Fi" && read -p $"> " PSK
    stty echo
  done
done

# Choose disk
until [ -e "$DISK" ]; do
  clear
  sfdisk -l | grep -E "/dev/"
  echo ""
  echo -e "WARNING: The selected disk will be rewritten."
  echo -e "Disk to install (e.g. /dev/[drive letter])" && read -p $"> " DISK
done

case "$DISK" in
  *"nvme"*)
    PART1="$DISK"p1
    PART2="$DISK"p2
    ;;
  *)
    PART1="$DISK"1
    PART2="$DISK"2
    ;;
esac

ROOT_PART=$PART2

# Encrypt
until [ "$ENCRYPTED" ]; do
  clear
  echo -e "Encrypt filesystem (y/default: n)" && read -p $"> " ENCRYPTED
  [ ! "$ENCRYPTED" ] && ENCRYPTED="n"
  if [ "$ENCRYPTED" = "y" ]; then
    [ ! "$CRYPTPASS" ] && CRYPTPASS=$(confirm_password "Password for encryption (must at least 6 characters)")
  fi
done

# Partition disk
clear
swapoff -a
umount -AR /mnt*
cryptsetup close /dev/mapper/root

if [ "$ENCRYPTED" = "y" ]; then
  dd if=/dev/zero of=$DISK bs=2M status=progress && sync || sync
  dd if=/dev/urandom of=$DISK bs=2M status=progress && sync || sync
fi

parted -s "$DISK" mklabel gpt
parted -s "$DISK" mkpart primary fat32 1MiB 512MiB
parted -s "$DISK" mkpart primary ext4 512MiB 100%
parted -s "$DISK" set 1 boot on

# Encrypt drive
if [ "$ENCRYPTED" = "y" ]; then
  echo -ne "$CRYPTPASS" | cryptsetup -q luksFormat --pbkdf=pbkdf2 "$ROOT_PART"
  echo -ne "$CRYPTPASS" | cryptsetup open "$ROOT_PART" root

  ROOT_PART="/dev/mapper/root"
fi

# Format and mount partitions
mkfs.fat -F 32 "$PART1"
fatlabel "$PART1" ESP
mkfs.ext4 -L root -F -O ^quota,^has_journal,^metadata_csum,uninit_bg -b2048 -m1 "$ROOT_PART"
mount "$ROOT_PART" /mnt
mkdir -p /mnt/boot/efi
mount "$PART1" /mnt/boot/efi
# Create swapfile
SWAP_SIZE=$(echo $(($(free -g | awk '/^Mem:/{print $2}') * 2 + 4)))
mkdir /mnt/swap
fallocate -l "$SWAP_SIZE"G /mnt/swap/swapfile
chmod 600 /mnt/swap/swapfile
mkswap /mnt/swap/swapfile
swapon /mnt/swap/swapfile

# Install base system and kernel
clear
echo -e 'Done with configuration. Installing...'

if [ "$ENCRYPTED" = "y" ]; then
  basestrap /mnt base $INIT elogind-$INIT efibootmgr dbus-$INIT dhcpcd-$INIT grub $UCODE wpa_supplicant-$INIT cryptsetup-$INIT
else
  basestrap /mnt base $INIT elogind-$INIT efibootmgr dbus-$INIT dhcpcd-$INIT grub $UCODE wpa_supplicant-$INIT
fi

basestrap /mnt linux-zen linux-zen-headers linux-firmware mkinitcpio

fstabgen -U /mnt >/mnt/etc/fstab

# Save connection
if [ "$SSID" ]; then
  echo -e "update_config=1
ap_scan=1
fast_reauth=1
network={
ssid=\"$SSID\"
psk=\"$PSK\"
scan_ssid=1  
}" >/mnt/etc/wpa_supplicant/wpa_supplicant.conf
fi

# Chroot
(INIT="$INIT" PART2="$PART2" ROOT_PASSWORD="$ROOT_PASSWORD" ENCRYPTED="$ENCRYPTED" REGION_CITY="$REGION_CITY" HOST="$HOST" USERNAME="$USERNAME" KEYMAP="$KEYMAP" artix-chroot /mnt /bin/bash -c 'bash <(curl -s https://raw.githubusercontent.com/YurinDoctrine/deploy-artix/main/deploy.sh); exit')

# Perform finish
swapoff -a
umount -AR /mnt*
cryptsetup close "$ROOT_PART"

clear
echo -e 'Installation completed successfully. You may now reboot or poweroff...'