#!/bin/bash

user_groups="audio,video,input,network,storage,wheel"
base_setup glibc
locale_setup /etc/locale.conf

#Pacman setup.
msg_print note "Pacman setup..."
pacman_install="pacman -Suy --needed --noconfirm"

sed -i "s/#Color/Color/" /etc/pacman.conf
[[ $arch_add_i386 == "1" ]] && sed -ie '$!N;s|\#\[multilib\]\n\#Include|\[multilib\]\nInclude|;P;D' /etc/pacman.conf #>_<
mv /etc/pacman.d/mirrorlist{,.pacnew}
mv /etc/pacman.d/mirrorlist{.used,}

msg_print note "Pacman is ready."

#Arch setup.
msg_print note "Installing addational packages..."

to_install="$postinstall" to_enable=''

#Network setup.
if [[ $networkmanager != "1" ]]; then
  msg_print note "Using default network config."
  echo -e "[Match]\nName=eth0\n\n[Network]\nDHCP=yes" >> /etc/systemd/network/20-wired.network
  msg_print warning "Masking udev's network renaming rule."
  ln -s /dev/null /etc/udev/rules.d/80-net-setup-link.rules
  to_enable="$to_enable systemd-networkd"
fi

if [[ $kernel == "1" ]]; then
  case "$kernel_type" in
    vanilla) to_install="$to_install linux linux-firmware linux-headers";;
    virtual) to_install="$to_install linux";;
    *) return_err "Incorrect paramater kernel_type=$kernel_type! Mistake?"
  esac
fi

if [[ $add_soft == "1" ]]; then
  if [[ $networkmanager == "1" ]]; then
    to_install="$to_install networkmanager crda"
    to_enable="$to_enable NetworkManager.service"
  fi
  if [[ $ssh == "1" ]]; then
    to_install="$to_install openssh"
    to_enable="$to_enable sshd"
  fi
  if [[ $pipewire == "1" ]]; then
    to_install="$to_install pipewire pipewire-pulse pipewire-alsa"
    [[ $bluetooth == "1" ]] && to_install="$to_install pulseaudio-bluetooth"
  fi
  if [[ $bluetooth == "1" ]]; then
    to_install="$to_install bluez bluez-utils"
    to_enable="$to_enable bluetooth.service"
  fi
  if [[ $printers == "1" ]]; then
    to_install="$to_install cups cups-filters"
    to_enable="$to_enable cups.socket"
    [[ $bluetooth == "1" ]] && to_install="$to_install bluez-cups"
  fi
fi

if [[ $graphics == "1" ]]; then
  case $graphics_type in
    xorg)
      to_install="$to_install xorg xorg-drivers"
      case $desktop_type in
        DE)
          case $desktop_de in
            plasma) to_install="$to_install plasma kde-applications";;
            xfce4) to_install="$to_install xfce4 xfce4-goodies";;
            cinnamon) to_install="$to_install cinnamon";;
            gnome) to_install="$to_install gnome gnome-extra";;
            *) return_err "Incorrect paramater desktop_de=$desktop_de! Mistake?";;
          esac
          if [[ $pipewire == "1" ]]; then
            case $desktop_de in 
              xfce4|cinnamon|gnome) to_install="$to_install pavucontrol";;
            esac
          fi
          if [[ $bluetooth == "1" ]]; then
            case $desktop_de in 
              xfce4|cinnamon) to_install="$to_install blueberry";;
            esac
          fi
        ;;
        *)
          return_err "Wrong parameter desktop_type=$desktop_type. Mistake?"
        ;;
      esac
    ;;
    wayland)
      to_install="$to_install egl-wayland"
      case $desktop_type in
        DE)
          case $desktop_de in
            plasma) to_install="$to_install plasma kde-applications plasma-wayland-session";;
            gnome) to_install="$to_install gnome gnome-extra";;
            *) return_err "Incorrect paramater desktop_de=$desktop_de! Mistake?";;
          esac
          if [[ $pipewire == "1" ]]; then
            case $desktop_de in 
              gnome) to_install="$to_install pavucontrol";;
            esac
          fi
        ;;
        *)
          return_err "Wrong parameter desktop_type=$desktop_type. Mistake?"
        ;;
      esac
    ;;
    *)
      return_err "Wrong parameter graphics_type=$graphics_type. Mistake?"
    ;;
  esac
  case $desktop_dm in
    gdm)
      to_install="$to_install gdm"
      to_enable="$to_enable gdm.service"
    ;;
    lightdm)
      to_install="$to_install lightdm-gtk-greeter-settings"
      to_enable="$to_enable lightdm.service"
    ;;
    sddm)
      to_install="$to_install sddm"
      to_enable="$to_enable sddm.service"
    ;;
    *) 
      return_err "Incorrect paramater desktop_dm=$desktop_dm! Mistake?"
    ;;
  esac
fi

[[ -n "$to_install" ]] && $pacman_install $to_install
for service in $to_enable; do
  systemctl enable "$service"
done

msg_print note "Packages are installed."

case "$bootloader_name" in
  grub2)
    to_install="grub"
    [[ $bootloader_type = uefi ]] && to_install="$to_install efibootmgr"
    $pacman_install $to_install
    if [[ $removable_disk == "1" ]]; then
      $pacman_install -w os-prober
    else
      $pacman_install os-prober
    fi
    grub_config
  ;;
  *) msg_print note "Bootloader not chosen."
esac

# Keep more space free
msg_print note "Cleaning up..."
rm -rf /var/lib/pacman/sync/* /var/cache/pacman/pkg/*