#!/usr/bin/env bash

# Ensure pkexec is available
if ! command -v pkexec &>/dev/null; then
    echo "Error: 'pkexec' is not installed. Please install PolicyKit to proceed.";
    exit 1
fi

# Check if the script is being run as root
if [[ "${EUID}" -ne 0 ]]; then
    # Attempt to elevate privileges and rerun the script with pkexec
    exec pkexec /usr/lib/snigdhaos/pkexec-gui "$@";
    exit 1
fi

# Export environment variables from the parent process
# ShellCheck Directive: Allow environment variables to be read and exported from /proc
# shellcheck disable=SC2163
while IFS= read -rd '' var; do
    export "${var}"
done < <(grep --null-data -ae "^\($XDG_CURRENT_DESKTOP\|WAYLAND_DISPLAY\|XDG_RUNTIME_DIR\|XDG_SESSION_TYPE\|XCURSOR_SIZE\|LC_.*\|LANG\|LANGUAGE\|QT_WAYLAND_FORCE_DPI\|QT_QPA_PLATFORMTHEME\|QT_STYLE_OVERRIDE\)=" /proc/$PPID/environ)

# Adjust WAYLAND_DISPLAY if running on Wayland
if [[ -n "${WAYLAND_DISPLAY}" ]]; then
    export WAYLAND_DISPLAY="${XDG_RUNTIME_DIR}/${WAYLAND_DISPLAY}"
fi

# Set the root user's runtime directory
export XDG_RUNTIME_DIR="/run/user/0"

# Execute the provided command with root privileges
exec "$@"