#!/bin/bash
#
# Omarchy Final Configurations Installer
#
# This script runs from archinstall's custom_commands after base packages
# and user creation. It switches to the created user and runs install.sh
# to complete package installation and system configuration.
#
# archinstall runs custom_commands as root via:
#   arch-chroot -S /mnt bash /var/tmp/user-command.0.sh
#

set -eEo pipefail

# Find the first non-root user (UID >= 1000, < 60000)
OMARCHY_USER=$(getent passwd | awk -F: '$3 >= 1000 && $3 < 60000 {print $1; exit}')

if [[ -z "$OMARCHY_USER" ]]; then
    echo "ERROR: No non-root user found!"
    echo "Users created:"
    getent passwd | awk -F: '$3 >= 1000 {print $1, $3}'
    exit 1
fi

echo "Setting up Omarchy for user: $OMARCHY_USER"

# Setup passwordless sudo (will be removed by post-install)
echo "Setting up passwordless sudo..."
mkdir -p /etc/sudoers.d
cat >/etc/sudoers.d/99-omarchy-installer <<EOF
root ALL=(ALL:ALL) NOPASSWD: ALL
%wheel ALL=(ALL:ALL) NOPASSWD: ALL
$OMARCHY_USER ALL=(ALL:ALL) NOPASSWD: ALL
EOF
chmod 440 /etc/sudoers.d/99-omarchy-installer

# Get user info from /tmp (written by configurator)
if [[ -f /tmp/omarchy-user-name.txt ]]; then
  OMARCHY_USER_NAME=$(cat /tmp/omarchy-user-name.txt)
else
  OMARCHY_USER_NAME=""
fi

if [[ -f /tmp/omarchy-user-email.txt ]]; then
  OMARCHY_USER_EMAIL=$(cat /tmp/omarchy-user-email.txt)
else
  OMARCHY_USER_EMAIL=""
fi

# Run install.sh as the user
echo "========================================"
echo "Running Omarchy installation as user: $OMARCHY_USER"
echo "========================================"
echo

# Use runuser instead of su for better output handling
# runuser doesn't go through PAM and preserves stdout/stderr better
runuser -u "$OMARCHY_USER" -- bash -c "
    set -eEo pipefail
    export PYTHONUNBUFFERED=1
    export OMARCHY_CHROOT_INSTALL=1
    export OMARCHY_ARCHINSTALL_WRAPPER=1
    export OMARCHY_USER='$OMARCHY_USER'
    export OMARCHY_USER_NAME='$OMARCHY_USER_NAME'
    export OMARCHY_USER_EMAIL='$OMARCHY_USER_EMAIL'
    cd ~
    source /usr/share/omarchy/install.sh
"

exit_code=$?

if [[ $exit_code -eq 0 ]]; then
    echo
    echo "========================================"
    echo "Omarchy install.sh completed successfully!"
    echo "========================================"
else
    echo
    echo "========================================"
    echo "ERROR: Omarchy install.sh exited with code $exit_code"
    echo "========================================"
    exit $exit_code
fi