#!/bin/sh

# Requires: rofi, and either xfce4-session or elogind (loginctl)

# Power menu, shared between the XFCE session and oxwm. Under XFCE the

# session manager owns these actions: it asks open applications to close and

# saves the session before handing the request on, which is what Ctrl+Alt+Del

# does. Under oxwm there is no session manager, so elogind is asked directly

# and doas is the backstop for a session it does not own. Every action asks

# for confirmation first, defaulting to no — this menu sits a keystroke away

# from the launcher.

set -u

# xfce4-session-logout's per-action flags skip its own dialog on their own;

# --fast is deliberately not passed, because that means "do not save the

# session" rather than "do not ask".

session=""

if [ "${XDG_CURRENT_DESKTOP:-}" = "XFCE" ] && command -v xfce4-session-logout >/dev/null 2>&1; then

    session=xfce4-session-logout

fi

# Only the lockers that are actually installed get offered. None of these

# ship with the system: `doas apk add slock` (or i3lock) to get a lock entry.

# xflock4 is not a locker itself — it asks xfce4-session to lock, which needs

# one of these installed all the same — but going through it means the

# session's own lock settings apply.

locker=""

for candidate in slock i3lock xsecurelock physlock xscreensaver light-locker; do

    if command -v "$candidate" >/dev/null 2>&1; then

        locker=$candidate

        break

    fi

done

if [ -n "$locker" ] && [ -n "$session" ] && command -v xflock4 >/dev/null 2>&1; then

    locker=xflock4

fi

menu() { rofi -dmenu -i -no-custom "$@"; }

confirm() {

    answer=$(printf 'no\nyes\n' | menu -p "$1?" -mesg "This will $2." -l 2) || exit 0

    [ "$answer" = "yes" ]

}

# The session manager when there is one, then loginctl, then doas.

#   $1 xfce4-session-logout flag   $2 loginctl verb   $3 doas fallback

power() {

    if [ -n "$session" ]; then

        "$session" "$1"

    else

        loginctl "$2" 2>/dev/null || doas "$3"

    fi

}

entries="󰤄  suspend

󰜉  reboot

󰐥  shutdown

󰍃  log out"

[ -n "$locker" ] && entries="󰌾  lock

$entries"

CHOICE=$(printf '%s\n' "$entries" | menu -p "power" -l 5) || exit 0

[ -n "$CHOICE" ] || exit 0

case ${CHOICE##*  } in

    lock)     exec "$locker" ;;

    suspend)  confirm "Suspend" "put the machine to sleep" && power --suspend suspend zzz ;;

    reboot)   confirm "Reboot" "close every open window and restart" && power --reboot reboot reboot ;;

    shutdown) confirm "Shut down" "close every open window and power off" && power --halt poweroff poweroff ;;

    "log out")

        confirm "Log out" "end the session and close every window" || exit 0

        if [ -n "$session" ]; then

            "$session" --logout

        else

            # Ending the session takes the whole X server with it, which is

            # what logging out means here; killing oxwm alone would leave X

            # running.

            loginctl terminate-session "${XDG_SESSION_ID:-}" 2>/dev/null ||

                pkill -u "$USER" -x oxwm

        fi

        ;;

esac