#!/usr/bin/env bash

## Author : Aditya Shakya (adi1090x)
## Github : @adi1090x
## Derived from adi1090x/rofi, GPL-3.0 like upstream. Modifications
## Copyright (C) 2026 Danilo M. <danix@danix.xyz>
#
## Rofi   : Power Menu
#
## Available Styles
#
## style-1   style-2   style-3   style-4   style-5

# Current Theme
dir="$HOME/.config/rofi/udt"
theme='powermenu'

# CMDs
uptime="`uptime -p | sed -e 's/up //g'`"
host=`hostname`

# Options
shutdown=''
reboot=''
logout=''
sleep='󰤄'
yes='󰗠'
no='󰅙'

# Rofi CMD. Both prompts always open on DP-1, whichever monitor has focus.
rofi_cmd() {
	rofi -dmenu -m DP-1 \
		-p "'ndi vidimu ${USER}" \
		-mesg "Uptime: $uptime" \
		-theme ${dir}/${theme}.rasi
}

# Confirmation CMD
confirm_cmd() {
	rofi -dmenu -m DP-1 \
		-p 'Confirmation' \
		-selected-row 1 \
		-mesg 'Simu Sicuri?' \
		-theme "$HOME/.config/rofi/udt/confirm.rasi"
}

# Ask for confirmation
confirm_exit() {
	echo -e "$yes\n$no" | confirm_cmd
}

# Pass variables to rofi dmenu
run_rofi() {
	echo -e "$shutdown\n$reboot\n$sleep\n$logout" | rofi_cmd
}

# Execute Command
run_cmd() {
	selected="$(confirm_exit)"
	if [[ "$selected" == "$yes" ]]; then
		if [[ $1 == '--shutdown' ]]; then
	        notify-send 'system shutdown' 'the system is going to shutdown now'
	        sleep 1
	        udt-ledstrip off
	        udt-bulb off
	        /usr/bin/loginctl poweroff
		elif [[ $1 == '--reboot' ]]; then
	        notify-send 'system reboot' 'the system is going to reboot now'
	        sleep 1
	        udt-ledstrip off
	        udt-bulb off
	        /usr/bin/loginctl reboot
		elif [[ $1 == '--sleep' ]]; then
	        notify-send 'suspend to ram' 'the system is going to sleep now'
	        sleep 1
	        udt-ledstrip sleep
	        udt-bulb off
	        /usr/bin/loginctl suspend
		elif [[ $1 == '--logout' ]]; then
	        hyprctl dispatch 'hl.dsp.exit()'
		fi
	else
		exit 0
	fi
}

# Actions
chosen="$(run_rofi)"
case ${chosen} in
    $shutdown)
		run_cmd --shutdown
        ;;
    $reboot)
		run_cmd --reboot
        ;;
    $sleep)
        run_cmd --sleep
        ;;
    $logout)
		run_cmd --logout
        ;;
esac