#!/bin/bash
# ABOUTME: Post-install script for Moonarch — installs moonarch-git package and configures the system.
# ABOUTME: Run as regular user after the archinstall base installation.

set -euo pipefail

source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/lib.sh"

# --- Prerequisites ---

check_not_root
check_pacman

# --- Install paru (AUR Helper) ---

if ! command -v paru &>/dev/null; then
    log "Installing paru..."
    sudo pacman -S --needed --noconfirm paru
else
    log "paru already installed."
fi

# --- Set up Moonarch package registry ---

log "Setting up Moonarch package registry..."
if ! grep -q '\[moonarch\]' /etc/pacman.conf 2>/dev/null; then
    sudo tee -a /etc/pacman.conf > /dev/null <<'EOCONF'

[moonarch]
SigLevel = Required DatabaseOptional
Server = https://gitea.moonarch.de/api/packages/nevaforget/arch/$repo/$arch
EOCONF
    log "  + Moonarch repo added to pacman.conf."
else
    log "  ~ Moonarch repo already in pacman.conf."
fi

log "Importing Moonarch registry signing key..."
KEY_FILE=$(mktemp)
curl -sf https://gitea.moonarch.de/api/packages/nevaforget/arch/repository.key -o "$KEY_FILE"
KEY_ID=$(gpg --show-keys --with-colons "$KEY_FILE" 2>/dev/null | awk -F: '/^pub/{print $5}')
if [[ -n "$KEY_ID" ]] && ! sudo pacman-key --list-keys "$KEY_ID" &>/dev/null; then
    sudo pacman-key --add "$KEY_FILE"
    sudo pacman-key --lsign-key "$KEY_ID"
    log "  + Registry key $KEY_ID imported and locally signed."
else
    log "  ~ Registry key already imported."
fi
rm -f "$KEY_FILE"

# --- Set up Moonarch custom paru repo ---

log "Setting up Moonarch paru repo..."
PARU_CONF="/etc/paru.conf"
if ! grep -q 'Mode.*p' "$PARU_CONF" 2>/dev/null; then
    sudo sed -i '/^\[options\]/a Mode = arp' "$PARU_CONF"
    log "  + Mode = arp set in paru.conf."
else
    log "  ~ Mode already includes pkgbuilds."
fi
if ! grep -q '\[moonarch-pkgbuilds\]' "$PARU_CONF" 2>/dev/null; then
    printf '\n[moonarch-pkgbuilds]\nUrl = https://gitea.moonarch.de/nevaforget/moonarch-pkgbuilds.git\n' \
        | sudo tee -a "$PARU_CONF" > /dev/null
    log "  + Moonarch repo added to paru.conf."
else
    log "  ~ Moonarch repo already in paru.conf."
fi

paru -Syu --pkgbuilds --noconfirm

# --- Install moonarch-git (pulls in all configs, scripts, themes as dependencies) ---

log "Installing moonarch-git package..."
paru -S --needed --noconfirm moonarch-git

# --- User-level GTK4 symlinks for libadwaita apps ---

THEME_NAME="Colloid-Grey-Dark-Catppuccin"
THEME_GTK4="/usr/share/themes/$THEME_NAME/gtk-4.0"

if [[ -d "$THEME_GTK4" ]]; then
    log "Creating user-level GTK4 symlinks for $THEME_NAME..."
    USER_GTK4="$HOME/.config/gtk-4.0"
    mkdir -p "$USER_GTK4"
    ln -sf "$THEME_GTK4/gtk-dark.css" "$USER_GTK4/gtk.css"
    ln -sf "$THEME_GTK4/gtk-dark.css" "$USER_GTK4/gtk-dark.css"
    rm -f "$USER_GTK4/assets"
    ln -s "$THEME_GTK4/assets" "$USER_GTK4/assets"
else
    err "GTK4 theme not found: $THEME_GTK4 — libadwaita apps will use fallback theme."
fi

# --- gsettings for GTK theme and dark mode ---

log "Setting gsettings for GTK theme..."
gsettings set org.gnome.desktop.interface gtk-theme "$THEME_NAME"
gsettings set org.gnome.desktop.interface color-scheme 'prefer-dark'
gsettings set org.gnome.desktop.interface icon-theme 'Colloid-Grey-Catppuccin-Dark'
gsettings set org.gnome.desktop.interface font-name 'UbuntuSans Nerd Font 11'

# --- Install user config defaults ---

log "Installing user config defaults to ~/.config/..."
if [[ -d "$USER_DEFAULTS" ]]; then
    # Each subdirectory in user-defaults/ corresponds to a ~/.config/ directory.
    # Files are only copied if they don't exist yet (no overwriting).
    for src_dir in "$USER_DEFAULTS"/*/; do
        app_name="$(basename "$src_dir")"
        dest_dir="$HOME/.config/$app_name"
        mkdir -p "$dest_dir"
        find "$src_dir" -type f -print0 | while IFS= read -r -d '' src_file; do
            rel_path="${src_file#"$src_dir"}"
            dest_file="$dest_dir/$rel_path"
            if [[ ! -f "$dest_file" ]]; then
                mkdir -p "$(dirname "$dest_file")"
                cp "$src_file" "$dest_file"
                log "  + $app_name/$rel_path"
            else
                log "  ~ $app_name/$rel_path already exists, skipped."
            fi
        done
    done
else
    err "USER_DEFAULTS not found: $USER_DEFAULTS — skipping user config defaults."
fi

# --- Zsh user config ---

if [[ ! -f "$HOME/.zshrc" ]]; then
    log "No ~/.zshrc found — sourcing Moonarch defaults."
    mkdir -p "$HOME/.zshrc.d"
    echo "# Load Moonarch defaults, add custom overrides in ~/.zshrc.d/ or below" > "$HOME/.zshrc"
    echo "source /etc/zsh/zshrc.moonarch" >> "$HOME/.zshrc"
fi

# --- Enable systemd user services ---

log "Enabling systemd user services..."
USER_SERVICES=(
    "kanshi"
)

for service in "${USER_SERVICES[@]}"; do
    if systemctl --user cat "${service}.service" &>/dev/null; then
        systemctl --user enable "$service"
        log "  + $service (user)"
    else
        log "  ~ $service (user) not found, skipped."
    fi
done

# --- Enable systemd services ---

log "Enabling services..."
SERVICES=(
    "NetworkManager"
    "bluetooth"
    "docker"
    "greetd"
    "systemd-timesyncd"
    "ufw"
    "auto-cpufreq"
)

for service in "${SERVICES[@]}"; do
    if systemctl list-unit-files "${service}.service" &>/dev/null; then
        sudo systemctl enable "$service"
        log "  + $service"
    else
        log "  ~ $service not found, skipped."
    fi
done

# --- Set shell to zsh ---

if [[ "$SHELL" != */zsh ]]; then
    log "Setting default shell to zsh..."
    chsh -s "$(which zsh)"
fi

# --- Enable firewall ---

log "Configuring UFW..."
sudo ufw default deny incoming
sudo ufw default allow outgoing
sudo ufw --force enable

# --- Boot parameters: add quiet to systemd-boot entries ---

log "Configuring boot parameters..."
for entry in /boot/loader/entries/*.conf; do
    if [[ -f "$entry" ]] && [[ "$(basename "$entry")" != *fallback* ]] && ! grep -q 'quiet' "$entry"; then
        sudo sed -i '/^options/ s/$/ quiet/' "$entry"
        log "  + $(basename "$entry") — quiet added"
    fi
done

# --- Docker-Gruppe ---

if ! groups | grep -q docker; then
    log "Adding user to docker group..."
    sudo usermod -aG docker "$USER"
fi

# --- Screenshots directory ---

mkdir -p "$HOME/Pictures/Screenshots"
mkdir -p "$HOME/Pictures/Wallpaper"

# --- Done ---

log ""
log "============================================"
log "  Moonarch installation complete!"
log "============================================"
log ""
log "Next steps:"
log "  1. Reboot"
log "  2. Place wallpapers in ~/Pictures/Wallpaper/"
log "  3. rustup default stable"
log "  4. User overrides in ~/.config/ or ~/.zshrc.d/"
log ""