#!/bin/bash

SELECTION="$(printf "󰌾 Lock\n󰤄 Suspend\n󰍃 Log out\n Reboot\n Reboot to UEFI\n󰐥 Shutdown" | fuzzel --dmenu -a top-right -l 6 -w 18 -p "Select an option: ")"
[[ -z "$SELECTION" ]] && exit 0

confirm_action() {
    local action="$1"
    CONFIRMATION="$(printf "No\nYes" | fuzzel --dmenu -a top-right -l 2 -w 18 -p "$action?")"
    [[ "$CONFIRMATION" == *"Yes"* ]]
}

case $SELECTION in
*"󰌾 Lock"*)
    gtklock
    ;;
*"󰤄 Suspend"*)
    if confirm_action "Suspend"; then
        systemctl suspend-then-hibernate
    fi
    ;;
*"󰍃 Log out"*)
    if confirm_action "Log out"; then
        swaymsg exit
    fi
    ;;
*" Reboot"*)
    if confirm_action "Reboot"; then
        systemctl reboot
    fi
    ;;
*" Reboot to UEFI"*)
    if confirm_action "Reboot to UEFI"; then
        systemctl reboot --firmware-setup
    fi
    ;;
*"󰐥 Shutdown"*)
    if confirm_action "Shutdown"; then
        systemctl poweroff
    fi
    ;;
esac