#! /bin/bash

source ./util/notify

CHOICES=('Shutdown' 'Hibernate' 'Restart')

[ ! -z "$1" ] && selected="$1" \
                || selected="$(echo "${CHOICES[@]}" | tr ' ' '\n' \
                                                    | dmenu -i \
                                                            -nb '#ffffff' \
                                                            -nf '#000080' \
                                                            -sb '#000080' \
                                                            -sf '#ffffff')"

case "$selected" in
    "${CHOICES[0]}")
        shutdown -P 0
    ;;

    "${CHOICES[1]}")
        systemctl suspend
    ;;

    "${CHOICES[2]}")
        shutdown -r 0
    ;;

    '')
        exit 0
    ;;

    *)
        notify 'Invalid Entry'
    ;;
esac