1_post_install.sh

#!/bin/bash
#                 _   _                 _     _
#   ___  ___ ___ | |_| |_ ______ _  ___| |__ / |
#  / __|/ __/ _ \| __| __|_  / _` |/ __| '_ \| |
#  \__ \ (_| (_) | |_| |_ / / (_| | (__| | | | |
#  |___/\___\___/ \__|\__/___\__,_|\___|_| |_|_|
#
#       Zac Scott (github.com/scottzach1)
#
# 1_post_install.sh

# Color codes for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Parse command line arguments
VERBOSE=0
while [[ $# -gt 0 ]]; do
  case $1 in
    -v|--verbose)
      VERBOSE=1
      shift
      ;;
    *)
      echo "Unknown option: $1"
      exit 1
      ;;
  esac
done

# Exit on any error
set -e

# Enable command echoing only if verbose mode is on
if [ "$VERBOSE" -eq 1 ]; then
  set -x
fi

# Set up logging
exec 1> >(tee "$(basename --suffix .sh "$0")_$(date +%Y%m%d_%H%M%S).log")
exec 2>&1

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
CLONE_DIR="$HOME/Clone/"

cleanup() {
	local exit_code=$?
	set +x # Turn off command echoing for cleanup
	echo "Script exited with code: $exit_code"
	if [ $exit_code -ne 0 ]; then
		echo "Installation failed! Check the log file for details."
		# Add any necessary cleanup here, like unmounting filesystems
		if mountpoint -q /mnt/boot 2>/dev/null; then
			umount /mnt/boot
		fi
		if mountpoint -q /mnt 2>/dev/null; then
			umount /mnt
		fi
	fi
}

log() {
	local level=$1

	shift
	case "$level" in
	"INFO")
		echo -e "${GREEN}[INFO]${NC} $*"
		;;
	"WARN")
		echo -e "${YELLOW}[WARN]${NC} $*"
		;;
	"ERROR")
		echo -e "${RED}[ERROR]${NC} $*"
		;;
	esac
}

sanity_checks() {
	# Check if script is run as root
  if [ "$EUID" -eq 0 ]; then
      log "ERROR" "This script must not be run as root"
      exit 1
  fi

	# Check internet connectivity
	if ! ping -c 1 archlinux.org >/dev/null 2>&1; then
		error_exit "No internet connection"
	fi

	log "INFO" "All sanity checks passed"
}

setup_things() {
	# Enable time synchronization
	timedatectl set-ntp true
}

copy_configs() {
	log "INFO" "Copying config files to home directory"
	find configs/home -type f -printf '%P\0' |
		while IFS= read -r -d '' item; do
      mkdir -p "$HOME/$(dirname "$item")"
			cp "configs/home/$item" "$HOME/$item"
		done

	log "INFO" "Copy config files to root filesystem"
	find configs/root -type f -printf '%P\0' |
		while IFS= read -r -d '' item; do
      sudo mkdir -p "/$(dirname "$item")"
			sudo cp "configs/root/$item" "/$item"
		done
}

copy_scripts() {
  log "INFO" "Copy scripts to ~/.local/bin"
  mkdir -p "$HOME/.local/bin"
  find scripts -type f -printf '%P\0' |
    while IFS= read -r -d '' item; do
      cp "scripts/$item" "$HOME/.local/bin/$item"
    done

}

install_paru_git() {
  if command -v paru >/dev/null 2>&1; then
    log "INFO" "Paru is already installed (skipping install)"
  else
    log "INFO" "Installing paru from git"
    # https://github.com/Morganamilo/paru
    local target_dir="$CLONE_DIR/Aur/paru"
    mkdir -p "$(dirname "$target_dir")"
    sudo pacman -S --needed base-devel
    git clone https://aur.archlinux.org/paru.git "$target_dir"
    pushd "$target_dir" > /dev/null
    makepkg -si
    popd > /dev/null
  fi
}

install_packages_pacman() {
  log "INFO" "Installing AR packages via pacman"
  # shellcheck disable=SC2046
  sudo pacman -S --needed $(cat packages-pacman.lst)
}

install_packages_paru() {
  log "INFO" "Installing AUR packages via paru"
  # shellcheck disable=SC2046
  paru -S --needed $(cat packages-paru.lst)
}

install_python() {
  log "INFO" "Setting up Python and dependencies"

  if command -v >/dev/null 2>&1; then
    log "INFO" "- uv is already installed (skipping)"
  else
    log "INFO" "- installing astral-sh/uv"
    curl -LsSf https://astral.sh/uv/install.sh | sh
  fi
}

setup_nvim() {
  if ! command -v nvim >/dev/null 2>&1; then
    log "INFO" "Neovim is not installed (skipping setup)"
  else
    log "INFO" "Setting up neovim plugins"

    log "INFO" "- Installing junegunn/vim-plug"
    sh -c 'curl -fLo "${XDG_DATA_HOME:-$HOME/.local/share}"/nvim/site/autoload/plug.vim --create-dirs \
         https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim'

    # Run PlugInstall headless
    nvim --headless +PlugInstall +qall
  fi
}

setup_fish() {
  if ! command -v fish >/dev/null 2>&1; then
    log "INFO" "Fish shell is not installed (skipping setup)"
  else
    log "INFO" "Setting up fish plugins"
    if ! fish -c "omf --version" >/dev/null 2>&1; then
      log "INFO" "- Install oh-my-fish/oh-my-fish"
      curl https://raw.githubusercontent.com/oh-my-fish/oh-my-fish/master/bin/install > omf-install
      fish omf-install --path="$HOME/.local/share/omf" --config="$HOME/.config/omf/config.omf" --noninteractive --yes
    else
      log "INFO" "- oh-my-fish already installed (skipping)"
    fi
    log "INFO" "- Installing scottzach1/dracula-theme-omf"
    if ! fish -c "omf theme" | grep dracula-theme-omf >/dev/null 2>&1; then
      fish -c "omf install https://github.com/scottzach1/dracula-theme-omf.git"
    fi
    fish -c "omf theme dracula-theme-omf"
  fi
}

enable_services() {
  log "INFO" "Enabling servies via systemctl"
  services=("lightdm" "NetworkManager" "bluetooth" "polkit")

  for svc in "${services[@]}"; do
    log "INFO" "- enabling $svc.service"
  done
}

setup_misc() {
  # Setup wallpaper
  log "INFO" "Setting up miscellaneous things"
  log "INFO" "- set falseWallpaper.png background"
  nitrogen --set-auto /usr/share/backgrounds/falseWallpaper.png
  log "INFO" "- set falseWallpaper.png lockscreen"
  betterlockscreen --update /usr/share/backgrounds/falseWallpaper.png
}

# Main installation process
main() {
	sanity_checks
	copy_configs
	copy_scripts
	install_paru_git
	install_packages_pacman
	install_python
	setup_nvim
	setup_fish
	setup_misc
	enable_services

  log "INFO" "Post install setup is complete"
  read -p "Would you like to reboot? (y/N) " -n 1 -r
  echo
  if [[ ! $REPLY =~ ^[Yy]$ ]]; then
    log "INFO" "You may run the following you are done"
    log "INFO" "> reboot"
    log "INFO" "exiting gracefully..."
    exit 0
  fi
  log "INFO" "rebooting..."
  reboot
}

# Run the setup
main