deploy

#!/bin/bash

set -Eeuo pipefail

trap 'echo "Error on line $LINENO: $BASH_COMMAND" >&2' ERR

readonly SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
readonly CONFIG_DIR="$HOME/.config"
readonly PACMAN_CONF="/etc/pacman.conf"

info() {
  echo "==> $*"
}

error() {
  echo "ERROR: $*" >&2
  exit 1
}

require_command() {
  local command="$1"

  if ! command -v "$command" >/dev/null 2>&1; then
    error "Required command not found: $command"
  fi
}

load_dependencies() {
  local file="$1"
  local -n deps="$2"

  deps=()

  [[ -f "$file" ]] || return 0

  mapfile -t deps < <(
    sed \
      -e 's/#.*//' \
      -e '/^[[:space:]]*$/d' \
      -e 's/^[[:space:]]*//' \
      -e 's/[[:space:]]*$//' \
      "$file"
  )
}

enable_multilib() {
  info "Enabling multilib repository"

  if grep -qE '^[[:space:]]*\[multilib\][[:space:]]*$' "$PACMAN_CONF"; then
    info "Multilib is already enabled."
    return
  fi

  if ! grep -qE '^[[:space:]]*#\[multilib\][[:space:]]*$' "$PACMAN_CONF"; then
    error "Could not find the commented multilib section in $PACMAN_CONF"
  fi

  sudo sed -i \
    -e 's/^[[:space:]]*#\[multilib\][[:space:]]*$/[multilib]/' \
    -e 's/^[[:space:]]*#Include = \/etc\/pacman\.d\/mirrorlist$/Include = \/etc\/pacman.d\/mirrorlist/' \
    "$PACMAN_CONF"

  info "Multilib enabled."
}

configure_reflector() {
  info "Installing and enabling reflector"

  sudo pacman -S --needed --noconfirm reflector
  sudo systemctl enable --now reflector.service
}

update_system() {
  info "Updating system packages"

  sudo pacman -Syu --noconfirm
}

install_yay() {
  info "Installing yay"

  sudo pacman -S --needed --noconfirm git base-devel

  local build_dir
  build_dir="$(mktemp -d)"
  trap 'rm -rf "$build_dir"' EXIT

  git clone https://aur.archlinux.org/yay.git "$build_dir/yay"

  (
    cd "$build_dir/yay"
    makepkg -si --noconfirm
  )

  rm -rf "$build_dir"

  info "yay installed."
}

update_aur_helper() {
  info "Updating AUR packages"

  yay -Syu --noconfirm
}

install_pacman_dependencies() {
  local dependency_file="$SCRIPT_DIR/dependencies/pacman.txt"
  local dependencies=()

  info "Installing official repository packages"

  load_dependencies "$dependency_file" dependencies

  if ((${#dependencies[@]} == 0)); then
    info "No pacman dependencies to install."
    return
  fi

  sudo pacman -S --needed --noconfirm "${dependencies[@]}"
}

install_aur_dependencies() {
  local dependency_file="$SCRIPT_DIR/dependencies/aur.txt"
  local dependencies=()

  info "Installing AUR packages"

  load_dependencies "$dependency_file" dependencies

  if ((${#dependencies[@]} == 0)); then
    info "No AUR dependencies to install."
    return
  fi

  yay -S --needed --noconfirm "${dependencies[@]}"
}

install_packages() {
  install_pacman_dependencies

  if ! command -v yay >/dev/null 2>&1; then
    install_yay
  fi

  update_aur_helper
  install_aur_dependencies
}

create_config_symlinks() {
  local config_dir
  local config_name
  local target
  local backup

  info "Creating configuration symlinks"

  mkdir -p "$CONFIG_DIR"

  # Handle .zshrc
  if [[ -L "$HOME/.zshrc" ]] && [[ "$(readlink "$HOME/.zshrc")" == "$SCRIPT_DIR/.zshrc" ]]; then
    info "Already linked .zshrc"
  else
    if [[ -e "$HOME/.zshrc" || -L "$HOME/.zshrc" ]]; then
      backup="$HOME/.zshrc.backup.$(date +%Y%m%d%H%M%S%N)"
      mv "$HOME/.zshrc" "$backup"
      info "Backed up .zshrc to $backup"
    fi

    ln -s "$SCRIPT_DIR/.zshrc" "$HOME/.zshrc"
    info "Linked .zshrc"
  fi

  shopt -s nullglob

  for config_dir in "$SCRIPT_DIR"/config/*/; do
    config_name="$(basename "$config_dir")"
    target="$CONFIG_DIR/$config_name"

    if [[ -L "$target" ]] && [[ "$(readlink "$target")" == "$config_dir" ]]; then
      info "Already linked $config_name"
      continue
    fi

    if [[ -e "$target" || -L "$target" ]]; then
      backup="$target.backup.$(date +%Y%m%d%H%M%S%N)"
      mv "$target" "$backup"
      info "Backed up $config_name to $backup"
    fi

    ln -s "$config_dir" "$target"
    info "Linked $config_name"
  done

  shopt -u nullglob
}

make_waybar_script_executable() {
  local waybar_script="$CONFIG_DIR/waybar/waybar.sh"

  if [[ -f "$waybar_script" ]]; then
    info "Making Waybar script executable"
    chmod +x "$waybar_script"
  fi
}

change_default_shell() {
  local zsh_path

  info "Configuring default shell"

  if ! command -v zsh >/dev/null 2>&1; then
    info "zsh is not installed; skipping shell configuration."
    return
  fi

  zsh_path="$(command -v zsh)"

  if [[ "$SHELL" == "$zsh_path" ]]; then
    info "zsh is already the default shell."
    return
  fi

  chsh -s "$zsh_path"

  info "Default shell changed to $zsh_path."
}

validate_environment() {
  info "Checking environment"

  if [[ "$EUID" -eq 0 ]]; then
    error "Do not run this script as root."
  fi

  require_command sudo
  require_command pacman
  require_command git

  sudo -v
}

main() {
  info "Starting Arch Linux setup"

  validate_environment

  create_config_symlinks

  enable_multilib
  configure_reflector

  update_system
  install_packages

  change_default_shell
  make_waybar_script_executable

  info "Arch Linux setup completed successfully."
}

main "$@"