#!/usr/bin/env bash

################################################################################
# INITIAL SETUP
################################################################################

# for visudo
export EDITOR=/usr/bin/nvim

# setup sudo access for pratham
/usr/bin/sudo -l -U pratham > /dev/null
if [[ $? -ne 0 ]]; then
    doas visudo
fi

# set hostname
WHAT_IS_MY_HOSTNAME=$(cat /etc/hostname)
if [[ $WHAT_IS_MY_HOSTNAME != "flameboi" ]]; then
    hostnamectl set-hostname flameboi
    echo "flameboi" | doas tee /etc/hostname
    WHAT_IS_MY_HOSTNAME=whoopsie
fi

# set timezone
WHAT_IS_MY_TZ=$(readlink /etc/localtime)
if [[ ! $WHAT_IS_MY_TZ =~ "Asia/Kolkata" ]]; then
    timedatectl set-timezone Asia/Kolkata
    WHAT_IS_MY_TZ=whoopsie
fi

# reboot to bring hostname in effect
if [[ $WHAT_IS_MY_TZ == "whoopsie" || $WHAT_IS_MY_HOSTNAME == "whoopsie" ]]; then
    systemctl reboot
fi


################################################################################
# SSH KEYS
################################################################################

# checking func
function generate_keys()
{
    if [[ ! -f "$1" && ! -f "$1"".pub" ]]; then
        ssh-keygen -t ed25519 -f $1
    fi
}

# create ssh keys
if [[ ! -d $HOME/.ssh ]]; then
    mkdir $HOME/.ssh
    chmod 700 $HOME/.ssh
fi
pushd $HOME/.ssh
generate_keys "bluefeds"
generate_keys "flameboi"
generate_keys "gitea"
generate_keys "github"
generate_keys "gitlab"
generate_keys "sentinel"
popd


################################################################################
# CUSTOM HOSTNAME FOR git.thefossguy.com
################################################################################

# check for an empty hostname in ~/.ssh/config
if [[ ! -f $HOME/.ssh/config ]]; then
    EDIT_SSH_CONF=true
else
    CONTENTS_OF_SSH_CONF=$(grep -A 1 "git.thefossguy.com" ~/.ssh/config | tail -n 1 | rev)

    if [[ "${CONTENTS_OF_SSH_CONF::1}" != "5" ]]; then
        EDIT_SSH_CONF=true
    else
        EDIT_SSH_CONF=false
    fi

fi


# set the hostname
if [[ $EDIT_SSH_CONF == true ]]; then
    tput -x clear
    cat <<EOF > $HOME/.ssh/config
Host git.thefossguy.com
    Hostname ::?
    User git
    IdentityFile ~/.ssh/gitea
    Port 22
EOF
    cat $HOME/.ssh/gitea.pub
    echo -e "\n\n\n\nPopulate Hostname (IP addr) for \"git.thefossguy.com\" in ~/.ssh/config"
    nvim ~/.ssh/config
fi


################################################################################
# SETUP DEV ENVIRONMENT
################################################################################

# clone repos
function git_repo_check()
{
    pushd $HOME/my-git-repos
    if [[ ! -d "$1" ]]; then
        git clone git@git.thefossguy.com:thefossguy/"$1"
    else
        pushd "$1"
        git fetch
        git pull
        popd
    fi
    popd
}

# update everything (along with `rustup`)
doas pacman --sync --refresh --refresh --sysupgrade

# rust-lang
rustup default stable
rustup update stable
rustup component add rust-src rust-analyzer rust-analysis


# get dotfiles
echo -ne "\n\n\n\n"
mkdir -p $HOME/my-git-repos
git_repo_check "dotfiles"
git_repo_check "dotfiles-priv"

rsync \
    --verbose --recursive --size-only --human-readable \
    --progress --stats \
    --itemize-changes --checksum \
    --exclude=".git" --exclude=".gitignore" --exclude="README.md" --exclude="run_me.sh" \
    ~/my-git-repos/dotfiles{,-priv}/ ~/

# dark mode (gtk)
gsettings set org.gnome.desktop.interface color-scheme prefer-dark

# podman?
#grep net.ipv4.ping_group_range /etc/sysctl.conf || echo "net.ipv4.ping_group_range=0 $(grep pratham /etc/subuid | awk -F ":" '{print $2 + $3}')" | doas tee -a /etc/sysctl.conf

# flatpak
flatpak --user remote-add --if-not-exists flathub https://flathub.org/repo/flathub.flatpakrepo
flatpak install --user flathub com.brave.Browser com.discordapp.Discord com.github.tchx84.Flatseal io.gitlab.librewolf-community org.raspberrypi.rpi-imager

################################################################################
# VIRSH POOLS + NETWORK
################################################################################

LIBVIRTD_RESTART=no

gropus | grep "libvirt" || doas adduser pratham libvirt
gropus | grep "kvm" || doas adduser pratham kvm

# network
doas virsh net-info default | grep "Autostart" | grep "no" && doas virsh net-autostart default

# storage pool
doas virsh pool-dumpxml default | grep "/flameboi_st/vm-store" > /dev/null
if [[ $? -ne 0 ]]; then
    doas virsh pool-destroy default
    doas virsh pool-undefine default
    doas virsh pool-define-as --name default --type dir --target /flameboi_st/vm-store
    doas virsh pool-autostart default
    doas virsh pool-start default
    LIBVIRTD_RESTART=yes
fi

# restart libvirtd if necessary
if [[ "$LIBVIRTD_RESTART" == "yes" ]]; then
    doas systemctl restart libvirtd
fi


################################################################################
# AUR/PARU
################################################################################

# do I have paru?
if ! command -v paru > /dev/null; then

    # build paru
    doas pacman --sync --refresh --refresh --sysupgrade
    doas pacman --needed base-devel

    git clone --depth 1  https://aur.archlinux.org/paru.git /tmp/paru-tmp-clone
    pushd /tmp/paru-tmp-clone
    makepkg -si

    if [[ $? -ne 0 ]]; then
        echo "paru wasn't installed successfully :("
        exit 1
    fi

    popd
fi

# install packages if not installed
pacman -Qm | grep "ttf-apple-emoji" > /dev/null || paru -S ttf-apple-emoji
pacman -Qm | grep "ttf-fork-awesome" > /dev/null || paru -S ttf-fork-awesome
pacman -Qm | grep "zfs-dkms" > /dev/null || paru -S zfs-dkms

# AUR pkgs
#paru -S noisetorch ssmtp

# wayland-WM
#paru -S hyperland

# intel 
#paru -S libva-intel-driver-g45-h264 intel-hybrid-codec-driver


################################################################################
# WRAP UP
################################################################################

tput -x clear
echo -e "\n\nThe setup appears to have completed (as far as I can tell). Please scroll up and verify yourself too!"
if ! command -v zpool > /dev/null; then
    lsmod | grep zfs
    if [[ $? -ne 0 ]]; then
        echo "ZFS Kernel module is not loaded. Please run the \`doas modprobe zfs\` command and reboot."
    fi
    doas systemctl enable zfs-import-cache.service zfs-import-scan.service zfs-import.service zfs-load-key.service zfs-mount.service zfs-volume-wait.service zfs-zed.service
    doas zpool import 16601987433518749526
    doas zpool import 12327394492612946617
    doas zpool set cachefile=/etc/zfs/zpool.cache heathen_disk
fi