#!/bin/bash
# Inspired by https://github.com/cramermarius/rofi-menus/blob/master/scripts/powermenu.sh

# entries with associated commands
declare -A entries
entries=(   [lock]="xset s activate"
            [logout]="i3-msg exit"
            [reboot]="systemctl reboot"
            [shutdown]="systemctl poweroff"
            [suspend]="systemctl suspend"
            [suspend (scheduled)]="scheduled_suspend")

declare -a rofi_args
rofi_args=( -no-config
            -theme /usr/share/rofi/themes/android_notification.rasi
            -lines ${#entries[@]}
            -width 12
            -location 3
            -yoffset 32
            -dmenu
            -no-case-sensitive
            -p) # has to end with -p!

function scheduled_suspend {
    declare -i min=0
    min=$(rofi "${rofi_args[@]}" "minutes: ")

    # make sure the input was a valid number
    # side effect: 0 minutes is not possible
    [[ "$min" -ne 0 ]] || exit 1

    notify-send "suspend in" "$min minutes"
    sleep $((min*60)) && systemctl suspend
}

# Choose option over rofi
# Note: bash does not keep the order of the keys thus they get sorted
chosen="$(printf '%s\n' "${!entries[@]}" | sort | rofi "${rofi_args[@]}" "power: ")"
# exit if nothing was selected
[[ -n "$chosen" ]] || exit 1
# handle scheduled suspend different
[[ "$chosen" != "suspend (scheduled)" ]] || { ${entries[$chosen]}; exit $?; }

# Confirm choice
yesNo="$(echo -e "yes\nno" | rofi  "${rofi_args[@]}" "Are you sure you want to ${chosen}? ")"

# Exit if "No"
[[ "$yesNo" == "yes" ]] || exit 1

# Execute
${entries[$chosen]}