#! /bin/bash

## Install all required packages for this user. It only needs to be run once
## when setting up the HOME directory.
##
## This script is designed not to ask the user over and over again for their
## sudo password. Since it uses yay and makepkg to install AUR and custom
## PKGBUILDs, the somewhat hacky way to do this is as follows:'
## 1. call this script as root
## 2. give the nobody user permission to run pacman with no password via sudo
## 3. set up a custom, temporary HOME directory for yay which is owned by nobody
## 4. run yay as nobody with HOME set to the directory made in (3)
## 5. run makepkg as nobody
## 6. remove the temporary HOME for yay
## 7. remove the pacman sudo privilege for nobody

PKGSRC_dir="$1"
PKGBUILD_dir=/tmp/makepkg
YAYTMP=/tmp/something
NOBODY_SUDO="nobody ALL=(root) NOPASSWD: /usr/bin/pacman"
NOBODY_SUDO_CONF=/etc/sudoers.d/10-nobody-pacman

mk_nobody_perms() {
    echo "Giving user nobody permission to use pacman without password"
    echo "$NOBODY_SUDO" > "$NOBODY_SUDO_CONF"
}

mk_yaytmp() {
    echo "Creating temporary build directory for yay at $YAYTMP"
    mkdir "$YAYTMP"
    chown -R nobody:nobody "$YAYTMP"
}

mk_makepkg_dir() {
    echo "Creating temporary makepkg build directory at $PKGBUILD_dir"
    mkdir -p "$PKGBUILD_dir"
    echo "Populated with packages from $PKGSRC_dir"
    # cp -r "$PKGSRC_dir"/* "$PKGBUILD_dir"
    chown -R nobody:nobody "$PKGBUILD_dir"
}

try_nobody_update() {
    ## Try updating pacman as the nobody user. If this fails then something
    ## went wrong when setting up sudo permissions for nobody
    echo "Testing pacman permissions for user nobody. Trying to update cache."
    if ! sudo -u nobody -n sudo pacman -Syy; then
        echo "Failed to obtain pacman permissions for user nobody. Exiting."
        return 1
    fi
}

call_makepkg() {
    cd "$PKGBUILD_dir/$1" || return 1
    sudo -u nobody makepkg -s -r -i -f --noconfirm
}

run_yay() {
    ## TODO add template switches to control which of these get installed based
    ## on my config
    dunst_pkgs=(dunst)
    emacs_pkgs=(emacs mu)
    flameshot_pkgs=(flameshot)
    gtk_pkgs=(zuki-themes)
    nvidia_pkgs=(optimus-manager)
    r_pkgs=(r docker-rootless-extras-bin gcc-fortran texlive-bin tk)
    redshift_pkgs=(redshift)
    rofi_pkgs=(rofi-git bitwarden-cli libnotify rofi-greenclip
            networkmanager-dmenu-git veracrypt sshfs jmtpfs)
    seafile_pkgs=(seafile)
    urxvt_pkgs=(urxvt-tabbedex rxvt-unicode urxvt-perls)
    xmonad_pkgs=(stack-static autorandr feh xorg-server xorg-xset libpulse
                playerctl wireless_tools acpid ttf-symbola-free ttf-symbola-free
                ttf-dejavu awesome-terminal-fonts numlockx picom i3lock-color
                xorg-xrandr xss-lock)
    zsh_pkgs=(zsh zsh-completions zsh-syntax-highlighting)

    ## AUR pkgs needed for spotify
    spotify_pkgs=(gconf)

    sudo -u nobody HOME="$YAYTMP" \
        yay --noconfirm --removemake -S \
        "${dunst_pkgs[@]}" \
        "${emacs_pkgs[@]}" \
        "${flameshot_pkgs[@]}" \
        "${gtk_pkgs[@]}" \
        "${nvidia_pkgs[@]}" \
        "${nvidia_pkgs[@]}" \
        "${r_pkgs[@]}" \
        "${redshift_pkgs[@]}" \
        "${rofi_pkgs[@]}" \
        "${seafile_pkgs[@]}" \
        "${urxvt_pkgs[@]}" \
        "${xmonad_pkgs[@]}" \
        "${zsh_pkgs[@]}" \
        "${spotify_pkgs[@]}"
}

run_makepkg() {
    ## these are all packages that have some personal customizations and/or are
    ## not in the AUR
    call_makepkg "clevo-xsm-wmi-dkms"
    call_makepkg "conky-lua"
    call_makepkg "spotify"
    call_makepkg "xkb-hypermode"
}

rm_makepkg_dir() {
    echo "Removing temporary makepkg build directory at $PKGBUILD_dir"
    rm -r -f "$PKGBUILD_dir"
}

rm_yaytmp() {
    echo "Removing temporary build directory for yay at $YAYTMP"
    rm -r -f "$YAYTMP"
}

rm_nobody_perms() {
    echo "Removing user nobody's permission to use pacman without password"
    rm -f "$NOBODY_SUDO_CONF"
}

clean_up() {
    rm_yaytmp
    rm_makepkg_dir
    rm_nobody_perms
    exit
}

trap 'clean_up' ERR

if [[ ! "$(id -u)" = "0" ]]; then
    echo "This script must be run as root. Exiting"
    exit 1
fi

mk_nobody_perms
try_nobody_update

# mk_yaytmp
# run_yay
# rm_yaytmp

mk_makepkg_dir
run_makepkg
rm_makepkg_dir

echo "$PKGSRC_dir"
echo "$PKGBUILD_dir"

rm_nobody_perms