#!/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 custom paru repo ---

log "Setting up Moonarch paru repo..."
PARU_CONF="$HOME/.config/paru/paru.conf"
mkdir -p "$(dirname "$PARU_CONF")"
if ! grep -q '\[moonarch\]' "$PARU_CONF" 2>/dev/null; then
    cat >> "$PARU_CONF" <<'EOCONF'
[moonarch]
Url = https://gitea.moonarch.de/nevaforget/moonarch-pkgbuilds.git
EOCONF
    log "  + Moonarch repo added to paru.conf."
else
    log "  ~ Moonarch repo already in paru.conf."
fi

paru -Sy --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-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
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

# --- 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 ""